diff --git a/tradingagents/dataflows/yfinance_news.py b/tradingagents/dataflows/yfinance_news.py index a86675b1..335aff84 100644 --- a/tradingagents/dataflows/yfinance_news.py +++ b/tradingagents/dataflows/yfinance_news.py @@ -52,6 +52,7 @@ def get_news_yfinance( ticker: str, start_date: str, end_date: str, + limit: int = 20, ) -> str: """ Retrieve news for a specific stock ticker using yfinance. @@ -60,13 +61,14 @@ def get_news_yfinance( ticker: Stock ticker symbol (e.g., "AAPL") start_date: Start date in yyyy-mm-dd format end_date: End date in yyyy-mm-dd format + limit: Maximum number of news articles to fetch (default: 20) Returns: Formatted string containing news articles """ try: stock = yf.Ticker(ticker) - news = yf_retry(lambda: stock.get_news(count=20)) + news = yf_retry(lambda: stock.get_news(count=limit)) if not news: return f"No news found for {ticker}"