Update code to use environment variables and improve error handling

push_git.bat

@@ -0,0 +1,3 @@
+@echo off
+git push origin main
+git push upstream main 

push_huggingface.bat

@@ -0,0 +1,4 @@
+@echo off
+git add .
+git commit -m "Update code to use environment variables and improve error handling"
+git push origin main 

test_agent.py

@@ -7,6 +7,13 @@ import os
 # Load environment variables from .env file
 load_dotenv()
 
+# Initialize the language model with API key from environment variables
+llm = ChatOpenAI(
+    temperature=0,
+    model_name="gpt-3.5-turbo",
+    openai_api_key=os.getenv("OPENAI_API_KEY")
+)
+
 # Define tools
 tools = [
     Tool(
@@ -16,14 +23,18 @@ tools = [
     ),
     Tool(
         name="BuyingPowerCalculator",
-        func=lambda q: buying_power_tool(q.split(',')[0], float(q.split(',')[1])),
-        description="Calculate how many shares you can buy. Input format: 'AAPL,5000'"
+        func=buying_power_tool,
+        description="Calculate how many shares you can buy. Input format: 'TICKER,AMOUNT' (e.g., 'AAPL,5000')"
     )
 ]
 
-# Create agent
-llm = ChatOpenAI(model="gpt-4", temperature=0)
-agent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
+# Initialize the agent
+agent = initialize_agent(
+    tools,
+    llm,
+    agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+    verbose=True
+)
 
 # Run a test
 response = agent.run("How much is AAPL?")