Sample AI agent to get Stock quote - Python

import csv
import openai

# Define the AI agent class

class AI_Agent:
def __init__(self, model):
self.model = model

def get_stock_quote(self, stock_symbol):
# Query the LLM for stock information
query = f"Provide the current stock quote for {stock_symbol}."
response = self.model.process_query(query)
return response

# Define the class for the LLM (OpenAI in this case)

class OpenAI_Model:
def __init__(self, api_key):
openai.api_key = api_key

def process_query(self, query):
try:
# Call OpenAI's GPT model to process the query
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo", # Use your desired model
messages=[{"role": "system", "content": "You are a stock market assistant."}, {"role": "user", "content": query}] )

return response['choices'][0]['message']['content']
except Exception as e:
return f"Error: {e}"

def read_csv_file(file_path):
# Read stock symbols from a CSV file
stock_symbols = []
try:
with open(file_path, newline='') as csvfile:

reader = csv.reader(csvfile)
for row in reader:
stock_symbols.append(row[0]) # Assuming stock symbols are in the first column
return stock_symbols
except Exception as e:
print(f"Error reading CSV file: {e}")
return []





def main():



# Provide your OpenAI API key


api_key = "your_openai_api_key_here"

openai_model = OpenAI_Model(api_key)
ai_agent = AI_Agent(openai_model)
# Path to your CSV file containing stock symbols
csv_file_path = "stock_symbols.csv"
stock_symbols = read_csv_file(csv_file_path)

if stock_symbols:
print("Fetching stock quotes...\n")
for symbol in stock_symbols:
print(f"Stock: {symbol}")
response = ai_agent.get_stock_quote(symbol)
print(f"Quote: {response}\n")
else:
print("No stock symbols found in the CSV file.")

if __name__ == "__main__":
main()

CSV Reader:
The read_csv_file() function reads stock symbols from a CSV file.
Each row's first column is treated as a stock symbol.
AI Agent:
The AI_Agent class queries the LLM for stock quotes.
It sends a specific query for each stock symbol.
LLM Integration:
The OpenAI_Model class interacts with OpenAI's GPT model.
Make sure you replace "your_openai_api_key_here" with your actual OpenAI API key.

Main Function:
The main() function ties everything together, reads the CSV file, queries stock quotes, and displays results.

Requirements: Install the OpenAI Python library using pip install openai.
Prepare a CSV file (stock_symbols.csv) with stock symbols listed (one symbol per row).
Replace the placeholder API key with your own key.