Create agent.py

agent.py

@@ -0,0 +1,37 @@
+from smolagents import ToolCallingAgent
+from smolagents.models import HfApiModel
+from smolagents.tools import tool
+import requests
+
+# Tool: Stock price lookup via Yahoo Finance
+@tool
+def get_stock_price(ticker: str) -> str:
+    """Returns the current stock price for the given ticker symbol (e.g., AAPL)."""
+    try:
+        url = f"https://query1.finance.yahoo.com/v7/finance/quote?symbols={ticker}"
+        response = requests.get(url)
+        data = response.json()
+        result = data['quoteResponse']['result']
+        if not result:
+            return f"No stock data found for '{ticker}'."
+        price = result[0]['regularMarketPrice']
+        name = result[0]['shortName']
+        return f"{name} ({ticker.upper()}) is currently trading at ${price:.2f}"
+    except Exception as e:
+        return f"Error fetching stock data: {e}"
+
+# Model
+model = HfApiModel(model_id="mistralai/Mistral-7B-Instruct-v0.2")
+
+# Agent
+agent = ToolCallingAgent(
+    tools=[get_stock_price],
+    model=model,
+    max_steps=3,
+    name="stock_price_agent",
+    description="Answers questions about stock prices using Yahoo Finance."
+)
+
+# Interface function
+def answer_question(question: str) -> str:
+    return agent.run(question).strip()