Update app.py

Tool added get_current_stock_price

app.py

@@ -3,7 +3,6 @@ import datetime
 import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
 from tools.stock_price_tool import StockPriceTool
 import yfinance as yf
 from typing import Any, Optional
@@ -40,8 +39,49 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_current_stock_price(ticker_symbol: str) -> Optional[dict]:
+    """
+    Fetches the latest stock price for the given ticker symbol and calculates
+    the price change over the last 5 days.
+
+    Args:
+        ticker_symbol: The stock ticker symbol to fetch the price for.
+
+    Returns:
+        dict: A dictionary containing the current stock price, the price change
+              over the last 5 days, and the percentage change. 
+              Returns None if there's an error or no data is available.
+    """
+    try:
+        # Fetch stock data
+        stock = yf.Ticker(ticker_symbol)
+        stock_data = stock.history(period="5d")  # Fetch data for the last 5 days
+        
+        if stock_data.empty:
+            return {"error": f"No historical data available for {ticker_symbol}"}
+        
+        # Get the current price (last close price)
+        current_price = stock_data['Close'].iloc[-1]
+        
+        # Calculate the price change over the last 5 days
+        price_change = current_price - stock_data['Close'].iloc[0]
+        price_change_percent = (price_change / stock_data['Close'].iloc[0]) * 100
+
+        # Round all values to 2 decimal places
+        return {
+            "price": round(current_price, 2),
+            "price_change": round(price_change, 2),
+            "price_change_percent": round(price_change_percent, 2)
+        }
+
+    except yf.exceptions.YFinanceException as yf_err:
+        logger.error(f"YFinance error for {ticker_symbol}: {str(yf_err)}")
+    except Exception as e:
+        logger.error(f"Unexpected error fetching price for {ticker_symbol}: {str(e)}")
+    return None
+
 final_answer = FinalAnswerTool()
-stock_price_tool = StockPriceTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
@@ -53,7 +93,6 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -62,7 +101,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[stock_price_tool], ## add your tools here (don't remove final answer)
+    tools=[get_current_stock_price], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,