universal-mcp-applications 0.1.21__py3-none-any.whl → 0.1.23__py3-none-any.whl

universal_mcp/applications/yahoo_finance/app.py

@@ -1,5 +1,6 @@
-import yfinance as yf
 from typing import Any
+
+import yfinance as yf
 from universal_mcp.applications.application import APIApplication
 from universal_mcp.integrations import Integration
 
@@ -16,31 +17,31 @@ class YahooFinanceApp(APIApplication):
     def get_stock_info(self, symbol: str) -> dict[str, Any]:
         """
         Gets real-time stock information including current price, market cap, financial ratios, and company details.
-        
+
         Args:
             symbol: Stock ticker symbol (e.g., 'AAPL', 'GOOGL', 'MSFT')
-        
+
         Returns:
             Complete dictionary with all available stock data fields from Yahoo Finance
-        
+
         Raises:
             ValueError: Invalid or empty symbol
             KeyError: Stock symbol not found
             ConnectionError: Network or API issues
-        
+
         Tags:
             stock, info, real-time, price, financials, company-data, important
         """
         if not symbol:
             raise ValueError("Stock symbol cannot be empty")
-        
+
         symbol = symbol.upper().strip()
         ticker = yf.Ticker(symbol)
-        
+
         info = ticker.info
-        if not info or info.get('regularMarketPrice') is None:
+        if not info or info.get("regularMarketPrice") is None:
             raise KeyError(f"Stock symbol '{symbol}' not found or invalid")
-        
+
         return info
 
     def get_stock_history(
@@ -49,75 +50,94 @@ class YahooFinanceApp(APIApplication):
         period: str = "1mo",
         interval: str = "1d",
         start_date: str | None = None,
-        end_date: str | None = None
-    )-> Any:
+        end_date: str | None = None,
+    ) -> dict:
         """
         Gets historical price data for a stock with OHLCV data, dividends, and stock splits.
-        
+
         Args:
             symbol: Stock ticker symbol (e.g., 'AAPL', 'GOOGL', 'MSFT')
             period: Time period ('1d', '5d', '1mo', '3mo', '6mo', '1y', '2y', '5y', '10y', 'ytd', 'max')
             interval: Data interval ('1m', '2m', '5m', '15m', '30m', '60m', '90m', '1h', '1d', '5d', '1wk', '1mo', '3mo')
             start_date: Start date in 'YYYY-MM-DD' format (overrides period)
             end_date: End date in 'YYYY-MM-DD' format (used with start_date)
-        
+
         Returns:
-            Complete DataFrame with Open, High, Low, Close, Volume, Dividends, Stock Splits columns
-        
+            Dictionary with historical stock data, dates as string keys
+
         Tags:
             stock, history, ohlcv, price-data, time-series, important
         """
         if not symbol:
             raise ValueError("Stock symbol cannot be empty")
-        
+
         symbol = symbol.upper().strip()
         ticker = yf.Ticker(symbol)
+
+        df = ticker.history(
+            period=period, interval=interval, start=start_date, end=end_date
+        )
         
-        return ticker.history(period=period, interval=interval, start=start_date, end=end_date)
+        try:
+            data = df.to_dict("index")  # type: ignore
+            if data:
+                converted_data = {}
+                for key, value in data.items():
+                    if hasattr(key, 'strftime'):
+                        converted_key = key.strftime('%Y-%m-%d')
+                    else:
+                        converted_key = str(key)
+                    converted_data[converted_key] = value
+                return converted_data
+            return data
+        except:
+            return {}
 
-    def get_stock_news(self, symbol: str, limit: int = 10)-> list[Any]:
+    def get_stock_news(self, symbol: str, limit: int = 10) -> list[Any]:
         """
         Gets latest news articles for a stock from Yahoo Finance.
-        
+
         Args:
             symbol: Stock ticker symbol (e.g., 'AAPL', 'GOOGL', 'MSFT')
             limit: Maximum number of articles to return (1-50). Defaults to 10
-        
+
         Returns:
             Raw list of news articles from Yahoo Finance
-        
+
         Tags:
             stock, news, articles, sentiment, important
         """
         if not symbol:
             raise ValueError("Stock symbol cannot be empty")
-        
+
         symbol = symbol.upper().strip()
         ticker = yf.Ticker(symbol)
-        
+
         news = ticker.news
         return news[:limit] if news else []
 
-    def get_financial_statements(self, symbol: str, statement_type: str = "income") -> dict:
+    def get_financial_statements(
+        self, symbol: str, statement_type: str = "income"
+    ) -> dict:
         """
         Gets financial statements for a stock from Yahoo Finance.
-        
+
         Args:
             symbol: Stock ticker symbol (e.g., 'AAPL', 'GOOGL', 'MSFT')
             statement_type: Type of statement ('income', 'balance', 'cashflow', 'earnings'). Defaults to 'income'
-        
+
         Returns:
             Dictionary with financial statement data
-        
+
         Tags:
             stock, financial-statements, earnings, important
         """
         if not symbol:
             raise ValueError("Stock symbol cannot be empty")
-        
+
         symbol = symbol.upper().strip()
         ticker = yf.Ticker(symbol)
-        
+
         if statement_type == "income":
             df = ticker.income_stmt
         elif statement_type == "balance":
@@ -128,96 +148,121 @@ class YahooFinanceApp(APIApplication):
             df = ticker.earnings
         else:
             df = ticker.income_stmt
-            
+
         try:
-            return df.to_dict('dict')  # type: ignore
+            data = df.to_dict("dict")  # type: ignore
+            if data:
+                converted_data = {}
+                for key, value in data.items():
+                    if hasattr(key, 'strftime'):
+                        converted_key = key.strftime('%Y-%m-%d')
+                    else:
+                        converted_key = str(key)
+                    converted_data[converted_key] = value
+                return converted_data
+            return data
         except:
             return {}
 
-    def get_stock_recommendations(self, symbol: str, rec_type: str = "recommendations") -> list[dict]:
+    def get_stock_recommendations(
+        self, symbol: str, rec_type: str = "recommendations"
+    ) -> list[dict]:
         """
         Gets analyst recommendations for a stock from Yahoo Finance.
-        
+
         Args:
             symbol: Stock ticker symbol (e.g., 'AAPL', 'GOOGL', 'MSFT')
             rec_type: Type of recommendation data ('recommendations', 'upgrades_downgrades'). Defaults to 'recommendations'
-        
+
         Returns:
             List of dictionaries with analyst recommendation data
-        
+
         Tags:
             stock, recommendations, analyst-ratings, important
         """
         if not symbol:
             raise ValueError("Stock symbol cannot be empty")
-        
+
         symbol = symbol.upper().strip()
         ticker = yf.Ticker(symbol)
-        
+
         if rec_type == "upgrades_downgrades":
             df = ticker.upgrades_downgrades
         else:
             df = ticker.recommendations
-            
+
         try:
-            return df.to_dict('records')  # type: ignore
+            return df.to_dict("records")  # type: ignore
         except:
             return []
 
-    def search(self, query: str, max_results: int = 10, news_count: int = 5, include_research: bool = False) -> dict[str, Any]:
+    def search(
+        self,
+        query: str,
+        max_results: int = 10,
+        news_count: int = 5,
+        include_research: bool = False,
+    ) -> dict[str, Any]:
         """
         Search Yahoo Finance for quotes, news, and research using yfinance Search.
-        
+
         Args:
             query: Search query (company name, ticker symbol, or keyword)
             max_results: Maximum number of quote results to return. Defaults to 10
             news_count: Number of news articles to return. Defaults to 5
             include_research: Whether to include research data. Defaults to False
-        
+
         Returns:
             Dictionary containing quotes, news, and optionally research data
-        
+
         Tags:
             search, quotes, news, research, important
         """
         if not query:
             raise ValueError("Search query cannot be empty")
-        
-        search = yf.Search(query, max_results=max_results, news_count=news_count, include_research=include_research)
-        
+
+        search = yf.Search(
+            query,
+            max_results=max_results,
+            news_count=news_count,
+            include_research=include_research,
+        )
+
         result = {}
         for attr in dir(search):
-            if not attr.startswith('_'): 
+            if not attr.startswith("_"):
                 try:
                     value = getattr(search, attr)
-                    if not callable(value):  
+                    if not callable(value):
                         result[attr] = value
                 except:
                     continue
-                    
+
         return result
 
-    def lookup_ticker(self, query: str, lookup_type: str = "all", count: int = 25) -> list[dict]:
+    def lookup_ticker(
+        self, query: str, lookup_type: str = "all", count: int = 25
+    ) -> list[dict]:
         """
         Look up ticker symbols by type using yfinance Lookup.
-        
+
         Args:
             query: Search query for ticker lookup
             lookup_type: Type of lookup ('all', 'stock', 'mutualfund', 'etf', 'index', 'future', 'currency', 'cryptocurrency'). Defaults to 'all'
             count: Maximum number of results to return. Defaults to 25
-        
+
         Returns:
             List of dictionaries with ticker lookup results
-        
+
         Tags:
             lookup, ticker, symbols, important
         """
         if not query:
             raise ValueError("Lookup query cannot be empty")
-        
+
         try:
             lookup = yf.Lookup(query)
-            
+
             if lookup_type == "stock":
                 results = lookup.get_stock(count=count)
             elif lookup_type == "mutualfund":
@@ -234,17 +279,14 @@ class YahooFinanceApp(APIApplication):
                 results = lookup.get_cryptocurrency(count=count)
             else:  # default to 'all'
                 results = lookup.get_all(count=count)
-            
+
             try:
-                return results.to_dict('records')  # type: ignore
+                return results.to_dict("records")  # type: ignore
             except:
                 return []
-                
+
         except Exception as e:
-            return [{
-                "query": query,
-                "error": f"Lookup failed: {str(e)}"
-            }]
+            return [{"query": query, "error": f"Lookup failed: {str(e)}"}]
 
     def list_tools(self):
         return [
@@ -255,4 +297,4 @@ class YahooFinanceApp(APIApplication):
             self.get_stock_recommendations,
             self.search,
             self.lookup_ticker,
-        ]
+        ]