cillin/evotraders
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf: asynchronize and parallelize analysis tools to prevent main loop blocking

Browse Source
This commit is contained in:
cillin
2026-04-07 14:59:41 +08:00
parent aae4bc7d40
commit dfc8fda187
1 changed files with 89 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

156
backend/tools/analysis_tools.py
View File

@@ -103,18 +103,28 @@ def _safe_float(value, default=0.0) -> float:
def safe(func): def safe(func):
"""Decorator to catch exceptions in tool functions.""" """Decorator to catch exceptions in both sync and async tool functions."""
@wraps(func) if asyncio.iscoroutinefunction(func):
def wrapper(*args, **kwargs): @wraps(func)
try: async def async_wrapper(*args, **kwargs):
return func(*args, **kwargs) try:
except Exception as e: return await func(*args, **kwargs)
error_msg = f"Error in {func.__name__}: {str(e)}" except Exception as e:
logger.error(f"{error_msg}\n{traceback.format_exc()}") error_msg = f"Error in {func.__name__}: {str(e)}"
return _to_text_response(f"[ERROR] {error_msg}") logger.error(f"{error_msg}\n{traceback.format_exc()}")
return _to_text_response(f"[ERROR] {error_msg}")
return wrapper return async_wrapper
else:
@wraps(func)
def sync_wrapper(*args, **kwargs):
try:
return func(*args, **kwargs)
except Exception as e:
error_msg = f"Error in {func.__name__}: {str(e)}"
logger.error(f"{error_msg}\n{traceback.format_exc()}")
return _to_text_response(f"[ERROR] {error_msg}")
return sync_wrapper
def _fmt(val, fmt=".2f", suffix="") -> str: def _fmt(val, fmt=".2f", suffix="") -> str:
@@ -141,7 +151,7 @@ def _resolved_date(current_date: Optional[str]) -> str:
@safe @safe
def analyze_efficiency_ratios( async def analyze_efficiency_ratios(
tickers: Optional[List[str]] = None, tickers: Optional[List[str]] = None,
current_date: Optional[str] = None, current_date: Optional[str] = None,
) -> ToolResponse: ) -> ToolResponse:
@@ -163,21 +173,26 @@ def analyze_efficiency_ratios(
tickers = _parse_tickers(tickers) tickers = _parse_tickers(tickers)
lines = [f"=== Efficiency Ratios Analysis ({current_date}) ===\n"] lines = [f"=== Efficiency Ratios Analysis ({current_date}) ===\n"]
for ticker in tickers: async def _fetch_one(ticker):
metrics = get_financial_metrics(ticker=ticker, end_date=current_date) try:
if not metrics: metrics = await asyncio.to_thread(get_financial_metrics, ticker=ticker, end_date=current_date)
lines.append(f"{ticker}: No data available\n") if not metrics:
continue return f"{ticker}: No data available\n"
m = metrics[0] m = metrics[0]
lines.append(f"{ticker}:") ticker_lines = [
lines.append(f" Asset Turnover: {_fmt(m.asset_turnover)}") f"{ticker}:",
lines.append(f" Inventory Turnover: {_fmt(m.inventory_turnover)}") f" Asset Turnover: {_fmt(m.asset_turnover)}",
lines.append(f" Receivables Turnover: {_fmt(m.receivables_turnover)}") f" Inventory Turnover: {_fmt(m.inventory_turnover)}",
lines.append( f" Receivables Turnover: {_fmt(m.receivables_turnover)}",
f" Working Capital Turnover: {_fmt(m.working_capital_turnover)}", f" Working Capital Turnover: {_fmt(m.working_capital_turnover)}\n",
) ]
lines.append("") return "\n".join(ticker_lines)
except Exception as e:
return f"{ticker}: Error - {str(e)}\n"
results = await asyncio.gather(*[_fetch_one(t) for t in tickers])
lines.extend(results)
return _to_text_response("\n".join(lines)) return _to_text_response("\n".join(lines))
@@ -310,7 +325,7 @@ def analyze_financial_health(
@safe @safe
def analyze_valuation_ratios( async def analyze_valuation_ratios(
tickers: Optional[List[str]] = None, tickers: Optional[List[str]] = None,
current_date: Optional[str] = None, current_date: Optional[str] = None,
) -> ToolResponse: ) -> ToolResponse:
@@ -332,24 +347,31 @@ def analyze_valuation_ratios(
tickers = _parse_tickers(tickers) tickers = _parse_tickers(tickers)
lines = [f"=== Valuation Ratios Analysis ({current_date}) ===\n"] lines = [f"=== Valuation Ratios Analysis ({current_date}) ===\n"]
for ticker in tickers: async def _fetch_one(ticker):
metrics = get_financial_metrics(ticker=ticker, end_date=current_date) try:
if not metrics: metrics = await asyncio.to_thread(get_financial_metrics, ticker=ticker, end_date=current_date)
lines.append(f"{ticker}: No data available\n") if not metrics:
continue return f"{ticker}: No data available\n"
m = metrics[0] m = metrics[0]
lines.append(f"{ticker}:") ticker_lines = [
lines.append(f" P/E Ratio: {_fmt(m.price_to_earnings_ratio)}") f"{ticker}:",
lines.append(f" P/B Ratio: {_fmt(m.price_to_book_ratio)}") f" P/E Ratio: {_fmt(m.price_to_earnings_ratio)}",
lines.append(f" P/S Ratio: {_fmt(m.price_to_sales_ratio)}") f" P/B Ratio: {_fmt(m.price_to_book_ratio)}",
lines.append("") f" P/S Ratio: {_fmt(m.price_to_sales_ratio)}\n",
]
return "\n".join(ticker_lines)
except Exception as e:
return f"{ticker}: Error - {str(e)}\n"
results = await asyncio.gather(*[_fetch_one(t) for t in tickers])
lines.extend(results)
return _to_text_response("\n".join(lines)) return _to_text_response("\n".join(lines))
@safe @safe
def get_financial_metrics_tool( async def get_financial_metrics_tool(
tickers: Optional[List[str]] = None, tickers: Optional[List[str]] = None,
current_date: Optional[str] = None, current_date: Optional[str] = None,
period: str = "ttm", period: str = "ttm",
@@ -374,35 +396,35 @@ def get_financial_metrics_tool(
f"=== Comprehensive Financial Metrics ({current_date}, {period}) ===\n", f"=== Comprehensive Financial Metrics ({current_date}, {period}) ===\n",
] ]
for ticker in tickers: async def _fetch_one(ticker):
metrics = get_financial_metrics( try:
ticker=ticker, # Offload synchronous data fetching to thread to keep loop snappy
end_date=current_date, metrics = await asyncio.to_thread(
period=period, get_financial_metrics,
) ticker=ticker,
if not metrics: end_date=current_date,
lines.append(f"{ticker}: No data available\n") period=period,
continue )
if not metrics:
return f"{ticker}: No data available\n"
m = metrics[0] m = metrics[0]
lines.append(f"{ticker}:") ticker_lines = [
lines.append(f" Market Cap: ${_fmt(m.market_cap, ',.0f')}") f"{ticker}:",
lines.append( f" Market Cap: ${_fmt(m.market_cap, ',.0f')}",
f" P/E: {_fmt(m.price_to_earnings_ratio)} | P/B: {_fmt(m.price_to_book_ratio)} | P/S: {_fmt(m.price_to_sales_ratio)}", f" P/E: {_fmt(m.price_to_earnings_ratio)} | P/B: {_fmt(m.price_to_book_ratio)} | P/S: {_fmt(m.price_to_sales_ratio)}",
) f" ROE: {_fmt(m.return_on_equity, '.1%')} | Net Margin: {_fmt(m.net_margin, '.1%')}",
lines.append( f" Revenue Growth: {_fmt(m.revenue_growth, '.1%')} | Earnings Growth: {_fmt(m.earnings_growth, '.1%')}",
f" ROE: {_fmt(m.return_on_equity, '.1%')} | Net Margin: {_fmt(m.net_margin, '.1%')}", f" Current Ratio: {_fmt(m.current_ratio)} | D/E: {_fmt(m.debt_to_equity)}",
) f" EPS: ${_fmt(m.earnings_per_share)} | FCF/Share: ${_fmt(m.free_cash_flow_per_share)}\n",
lines.append( ]
f" Revenue Growth: {_fmt(m.revenue_growth, '.1%')} | Earnings Growth: {_fmt(m.earnings_growth, '.1%')}", return "\n".join(ticker_lines)
) except Exception as e:
lines.append( return f"{ticker}: Error fetching data - {str(e)}\n"
f" Current Ratio: {_fmt(m.current_ratio)} | D/E: {_fmt(m.debt_to_equity)}",
) # Parallelize data retrieval for all tickers
lines.append( results = await asyncio.gather(*[_fetch_one(t) for t in tickers])
f" EPS: ${_fmt(m.earnings_per_share)} | FCF/Share: ${_fmt(m.free_cash_flow_per_share)}", lines.extend(results)
)
lines.append("")
return _to_text_response("\n".join(lines)) return _to_text_response("\n".join(lines))