Merge branch 'main' of https://huggingface.co/spaces/Shipmaster1/Agent_Langgraph into main

test_agent.py

@@ -0,0 +1,30 @@
+from langchain.chat_models import ChatOpenAI
+from langchain.agents import initialize_agent, Tool, AgentType
+from tools import get_price, buying_power_tool
+from dotenv import load_dotenv
+import os
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Define tools
+tools = [
+    Tool(
+        name="GetStockPrice",
+        func=get_price,
+        description="Get the current price of a stock by its ticker symbol, e.g., 'AAPL'"
+    ),
+    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'"
+    )
+]
+
+# Create agent
+llm = ChatOpenAI(model="gpt-4", temperature=0)
+agent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
+
+# Run a test
+response = agent.run("How much is AAPL?")
+print(response)