charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update interface.py

This commit is contained in:
Joseph O'Brien 2025-12-02 20:40:56 -05:00
parent f998d59b5e
commit 43e7d98ff1
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tradingagents/dataflows/interface.py
View File

@ -1,5 +1,6 @@
from typing import Annotated, List, Dict, Any, Optional from typing import Annotated, List, Dict, Any, Optional
from datetime import datetime, timedelta from datetime import datetime, timedelta
import threading
from .local import get_YFin_data, get_finnhub_news, get_finnhub_company_insider_sentiment, get_finnhub_company_insider_transactions, get_simfin_balance_sheet, get_simfin_cashflow, get_simfin_income_statements, get_reddit_global_news, get_reddit_company_news from .local import get_YFin_data, get_finnhub_news, get_finnhub_company_insider_sentiment, get_finnhub_company_insider_transactions, get_simfin_balance_sheet, get_simfin_cashflow, get_simfin_income_statements, get_reddit_global_news, get_reddit_company_news
from .y_finance import get_YFin_data_online, get_stock_stats_indicators_window, get_balance_sheet as get_yfinance_balance_sheet, get_cashflow as get_yfinance_cashflow, get_income_statement as get_yfinance_income_statement, get_insider_transactions as get_yfinance_insider_transactions from .y_finance import get_YFin_data_online, get_stock_stats_indicators_window, get_balance_sheet as get_yfinance_balance_sheet, get_cashflow as get_yfinance_cashflow, get_income_statement as get_yfinance_income_statement, get_insider_transactions as get_yfinance_insider_transactions
@ -121,6 +122,7 @@ VENDOR_METHODS = {
CACHE_TTL_SECONDS = 300 CACHE_TTL_SECONDS = 300
_bulk_news_cache: Dict[str, Dict[str, Any]] = {} _bulk_news_cache: Dict[str, Dict[str, Any]] = {}
_bulk_news_cache_lock = threading.Lock()
def parse_lookback_period(lookback: str) -> int: def parse_lookback_period(lookback: str) -> int:
@ -140,20 +142,22 @@ def parse_lookback_period(lookback: str) -> int:
def _get_cached_bulk_news(lookback_period: str) -> Optional[List[NewsArticle]]: def _get_cached_bulk_news(lookback_period: str) -> Optional[List[NewsArticle]]:
cache_key = lookback_period cache_key = lookback_period
if cache_key in _bulk_news_cache: with _bulk_news_cache_lock:
cached = _bulk_news_cache[cache_key] if cache_key in _bulk_news_cache:
cached_time = cached.get("timestamp") cached = _bulk_news_cache[cache_key]
if cached_time and (datetime.now() - cached_time).total_seconds() < CACHE_TTL_SECONDS: cached_time = cached.get("timestamp")
return cached.get("articles") if cached_time and (datetime.now() - cached_time).total_seconds() < CACHE_TTL_SECONDS:
return cached.get("articles")
return None return None
def _set_cached_bulk_news(lookback_period: str, articles: List[NewsArticle]) -> None: def _set_cached_bulk_news(lookback_period: str, articles: List[NewsArticle]) -> None:
cache_key = lookback_period cache_key = lookback_period
_bulk_news_cache[cache_key] = { with _bulk_news_cache_lock:
"timestamp": datetime.now(), _bulk_news_cache[cache_key] = {
"articles": articles, "timestamp": datetime.now(),
} "articles": articles,
}
def _convert_to_news_articles(raw_articles: List[Dict[str, Any]]) -> List[NewsArticle]: def _convert_to_news_articles(raw_articles: List[Dict[str, Any]]) -> List[NewsArticle]: