Create stock_price_tool.py

tools/stock_price_tool.py

@@ -0,0 +1,45 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import yfinance as yf
+
+class StockPriceTool(Tool):
+    """
+    A tool that fetches the latest stock price for a given stock ticker.
+    This class extends the Tool base class from smolagents.
+    """
+
+    name = "stock_price_tool"
+    description = "Fetches the latest stock price for a given stock ticker using yfinance."
+    inputs = {'ticker_symbol': {'type': 'any', 'description': 'Stock Ticker Symbol'}}
+    output_type = "Any"
+
+    def _call(self, ticker_symbol: str) -> Any:
+        """
+        Fetches the latest stock price for the given ticker symbol.
+        Args:
+            ticker_symbol (str): The stock ticker symbol (e.g., "AAPL" for Apple, "GOOGL" for Google).
+        Returns:
+            float: The current stock price of the given ticker symbol.
+            None: If an error occurs or data is unavailable.
+        Example:
+            >>> StockPriceTool()._call("AAPL")
+            150.25
+        """
+        try:
+            # Fetch the stock data using yfinance
+            stock = yf.Ticker(ticker_symbol)
+            
+            # Get the current price (the last close price)
+            stock_data = stock.history(period="1d")
+            if stock_data.empty:
+                return None
+            
+            current_price = stock_data['Close'].iloc[0]
+            return float(current_price)
+        except Exception as e:
+            # If there's an error, log it and return None
+            print(f"Error fetching price for {ticker_symbol}: {str(e)}")
+            return None
+            
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False