Update app.py

tool StockPriceTool added

app.py

@@ -39,49 +39,8 @@ 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' 
@@ -102,7 +61,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_current_stock_price ], ## add your tools here (don't remove final answer)
+    tools=[stock_price_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,