Added get_stock_price_tool

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+import yfinance as yf
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -18,6 +19,49 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def get_stock_price_tool(stock_symbol:str)-> str: 
+    """A tool that fetches the current price of a specified stock.
+    
+    Args:
+        stock_symbol: A string representing a valid stock on the NASDAQ stock exchange (E.g., AMZN, GOOGL).
+      """
+    try:
+        #1. Create a ticker object (container) for the stock symbol (i.e., an instance of the yf.Ticker class,
+        #whereby the stock symbol initializes the object). It holds the logic and methods for fetching data.
+        ticker = yf.Ticker(stock_symbol)
+
+        #2. API call (executing a network request) to fetch (return) the stock data.
+        #Accessing the.info attribute via dot notation, triggers the API call.
+        info = ticker.info
+        #The info variable now holds a Python dictionary (data structure) of all available fields/keys
+        #(price, volume, name, sector, etc.) associated with that specific stock symbol.
+      
+      
+        #3. Extract the price, using the 'regularMarketPrice' field, and the .get(...) method.
+        # This price is typically the current trading price during market hours.
+        price = info.get('regularMarketPrice')
+        
+        if price is None or price == 0:
+            # Fallback check, sometimes the price is in 'currentPrice' or not available
+            return f"Error: The stock price for '{stock_symbol}' could not be found. Please ensure the stock symbol is correct."
+        
+        #4. Format the stock price into US dollars (e.g., $1,234.56).
+        formatted_US_dollars = "${:,.2f}".format(price)
+
+        #5. Use the canonical symbol from the Ticker object for the output.
+        #Canonical symbol refers to the single, officially recognized, and standardized identifier for an entity within a specific system.
+        #ticker.ticker is an attribute (a piece of data) of this ticker object itself.
+        #This attribute holds the canonical stock symbol string (e.g., 'AMZN') that the external API recognizes
+        #for the object that was initialized with or identified, ensuring our output is accurate.
+        symbol_used = ticker.ticker
+      
+        return  f"The current stock price for {symbol_used} is {formatted_US_dollars}."
+    
+    except Exception as e:
+        #Catch general errors (network issues, invalid symbol format, etc.)
+        return f"Error fetching price for stock '{stock_symbol}'. Details: The request failed."
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -84,7 +128,8 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_current_time_in_timezone_non_IANA], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, get_current_time_in_timezone_non_IANA, get_stock_price_tool],
+    ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,