小钰
9a6c476ff1
- Alpha Vantage API Key: W5PTA2J5OB16W02L - 数据源优先级: Yahoo Finance → Alpha Vantage - 当Yahoo Finance限流时自动fallback到Alpha Vantage
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
"""使用小米 mimo-v2.5-pro 模型运行 TradingAgents 示例"""
|
||
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
||
from tradingagents.default_config import DEFAULT_CONFIG
|
||
|
||
config = DEFAULT_CONFIG.copy()
|
||
|
||
# 使用 openai_compatible 对接小米 API
|
||
config["llm_provider"] = "openai_compatible"
|
||
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
|
||
config["max_risk_discuss_rounds"] = 1
|
||
|
||
print("配置完成，开始运行 TradingAgents...")
|
||
print(f"Provider: {config['llm_provider']}")
|
||
print(f"Backend: {config['backend_url']}")
|
||
print(f"Model: {config['deep_think_llm']}")
|
||
|
||
ta = TradingAgentsGraph(debug=True, config=config)
|
||
|
||
# 测试分析茅台（A股）
|
||
_, decision = ta.propagate("600519.SS", "2026-01-15")
|
||
print("\n=== 交易决策 ===")
|
||
print(decision)
|