Create get_stock_news.py

get_stock_news.py

@@ -0,0 +1,80 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+
+from datetime import datetime, timedelta
+import pytz
+import yfinance as yf
+
+class StockNewsTool(Tool):
+    name = "stock_news_items"
+    description="Searches Yahoo Finance for stock news items and quotes related to the stock named in the query, then returns a string of metadata about news items"
+    inputs= {'query': {'type': 'string', 'description': 'The stock name or symbol to look up data for.'}}
+    output_type = "string"
+
+    def date_from_utime(self, utime:str)->str:
+        ts = int(utime)
+        dt = datetime.utcfromtimestamp(ts)
+        eastern = pytz.timezone('US/Eastern')
+        return dt.astimezone(eastern).strftime('%Y-%m-%d')
+
+    def etz_now(self):
+        eastern = pytz.timezone('US/Eastern')
+        ltime = datetime.now(eastern)
+        return ltime
+
+    def get_last_closing(self, symbol, offset=0, timeout=10):
+        try:
+            etime = self.etz_now()
+            if etime.hour >= 16:
+                etime = etime + timedelta(days=1)
+            if offset > 0:
+                etime = etime - timedelta(weeks=offset)
+            five_days_ago = etime - timedelta(days=6)
+            end = etime.strftime('%Y-%m-%d')
+            start = five_days_ago.strftime('%Y-%m-%d')
+            df = yf.download(symbol,
+                            start = start,
+                            end = end,
+                            progress = False,
+                            timeout=timeout,
+                            )
+            # print(df)
+            closing_date = 'unknown'
+            data_top = df.tail(1)
+            for row in data_top.index:
+                closing_date = row.strftime('%Y-%m-%d')
+                # print(closing_date)
+            return (df.iat[-1,0], closing_date)
+        except:
+            return ("unknown", "0000-00-00")
+
+
+# class StockNewsTool(Tool):
+
+    def forward(self, query:str)->str:
+        fuzzy = True
+        last_closing = 'unknown'
+        closing_date = 'n/a'
+        try:
+            yf_search = yf.Search(query, news_count= 5, max_results = 3, enable_fuzzy_query=fuzzy)
+            news = yf_search.news
+            quotes = yf_search.quotes
+        except:
+            return f'No results for search term {query}\n'
+        rv = 'News:\n'
+        for item in news:
+            rv += f'Title: {item["title"]}\n'
+            rv += f'Publisher: {item["publisher"]}\n'
+            rv += f'Date published: {self.date_from_utime(item["providerPublishTime"])}\n'
+            rv += f'Link: [URL]({item["link"]})\n\n'
+        for quote in quotes:
+            exchange = quote['exchange']
+            symbol = quote['symbol']
+            if exchange == 'NYQ' or exchange == 'NYS':
+                (last_closing, closing_date) = self.get_last_closing(symbol) 
+                break
+        rv += f'\n\nPrice quote at last closing on {closing_date} is {last_closing}\n'
+        return rv
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False