diff --git a/test_xiaomi.py b/test_xiaomi.py index 67f415a..4d3f920 100644 --- a/test_xiaomi.py +++ b/test_xiaomi.py @@ -10,9 +10,21 @@ config["backend_url"] = "https://token-plan-cn.xiaomimimo.com/v1" config["deep_think_llm"] = "mimo-v2.5-pro" config["quick_think_llm"] = "mimo-v2.5-pro" +# 配置数据源:Yahoo Finance优先,Alpha Vantage作为备选 +config['data_vendors'] = { + 'core_stock_apis': 'yfinance,alpha_vantage', + 'technical_indicators': 'yfinance,alpha_vantage', + 'fundamental_data': 'yfinance,alpha_vantage', + 'news_data': 'yfinance,alpha_vantage', + 'macro_data': 'fred', + 'prediction_markets': 'polymarket', +} + # 设置 API Key(也可通过环境变量 OPENAI_COMPATIBLE_API_KEY) import os os.environ["OPENAI_COMPATIBLE_API_KEY"] = "tp-ce6eaqg1gh1uev46i4oexh4sahloyfo2hfpxuc2n9m7yl9dd" +os.environ["FRED_API_KEY"] = "825283926dd016ea66e96cc919e047da" +os.environ["ALPHA_VANTAGE_API_KEY"] = "W5PTA2J5OB16W02L" # 减少辩论轮数加速测试 config["max_debate_rounds"] = 1 diff --git a/web_app.py b/web_app.py index 97406d1..c1aa252 100644 --- a/web_app.py +++ b/web_app.py @@ -530,6 +530,7 @@ def run_analysis(task_id, ticker, date): # 设置环境变量 os.environ['OPENAI_COMPATIBLE_API_KEY'] = 'tp-ce6eaqg1gh1uev46i4oexh4sahloyfo2hfpxuc2n9m7yl9dd' os.environ['FRED_API_KEY'] = '825283926dd016ea66e96cc919e047da' + os.environ['ALPHA_VANTAGE_API_KEY'] = 'W5PTA2J5OB16W02L' log(f'📊 开始分析 {ticker} ({date})') log(f'🤖 模型: MiMo-v2.5-pro') @@ -546,6 +547,16 @@ def run_analysis(task_id, ticker, date): config['max_debate_rounds'] = 1 config['max_risk_discuss_rounds'] = 1 + # 配置数据源:Yahoo Finance优先,Alpha Vantage作为备选 + config['data_vendors'] = { + 'core_stock_apis': 'yfinance,alpha_vantage', + 'technical_indicators': 'yfinance,alpha_vantage', + 'fundamental_data': 'yfinance,alpha_vantage', + 'news_data': 'yfinance,alpha_vantage', + 'macro_data': 'fred', + 'prediction_markets': 'polymarket', + } + log('🔧 配置完成,启动分析引擎...') ta = TradingAgentsGraph(debug=True, config=config)