TradingAgents

from tradingagents.graph.trading_graph import TradingAgentsGraph
from tradingagents.default_config import DEFAULT_CONFIG

# 创建自定义配置
config = DEFAULT_CONFIG.copy()
config["deep_think_llm"] = "gpt-4o-mini"  # 使用不同的模型
config["quick_think_llm"] = "gpt-4o-mini"  # 使用不同的模型
config["max_debate_rounds"] = 1  # 设置辩论轮数
config["online_tools"] = True  # 使用在线工具或缓存数据

# 初始化TradingAgents
ta = TradingAgentsGraph(debug=True, config=config)

# 执行交易决策
_, decision = ta.propagate("NVDA", "2024-05-10")
print(decision)

# 高级配置示例
from tradingagents.default_config import DEFAULT_CONFIG

# 创建自定义配置
config = DEFAULT_CONFIG.copy()

# LLM配置
config["llm_provider"] = "openai"  # openai, anthropic, google, ollama
config["deep_think_llm"] = "o1-preview"  # 深度思考模型
config["quick_think_llm"] = "gpt-4o"     # 快速响应模型
config["backend_url"] = "https://api.openai.com/v1"

# 辩论和讨论设置
config["max_debate_rounds"] = 3      # 最大辩论轮数
config["max_risk_discuss_rounds"] = 2 # 风险讨论轮数
config["max_recur_limit"] = 100      # 最大递归限制

# 数据源设置
config["online_tools"] = True       # 使用在线工具或缓存数据
config["data_cache_dir"] = "./cache" # 数据缓存目录

# 结果输出设置
config["results_dir"] = "./results" # 结果保存目录