From 5b5a9b9aeeb01b99ea9d8fd3e36a4943ac1c1dcf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?d=20=F0=9F=94=B9?= Date: Tue, 24 Mar 2026 00:06:23 +0000 Subject: [PATCH] refactor: add limit parameter to get_news_yfinance for consistency Address review feedback: make hardcoded count=20 a configurable limit parameter, consistent with get_global_news_yfinance. --- tradingagents/dataflows/yfinance_news.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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}"