70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
"""
|
|
简化版 TradingAgents 测试
|
|
跳过可能有问题的全球新闻分析，只测试核心功能
|
|
"""
|
|
|
|
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
|
from tradingagents.default_config import DEFAULT_CONFIG
|
|
from dotenv import load_dotenv
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
# Create a custom config for DeepSeek API
|
|
config = DEFAULT_CONFIG.copy()
|
|
|
|
# Configure DeepSeek API
|
|
config["llm_provider"] = "openai" # DeepSeek uses OpenAI-compatible API
|
|
config["backend_url"] = "https://api.deepseek.com" # DeepSeek API endpoint
|
|
config["deep_think_llm"] = "deepseek-reasoner" # DeepSeek reasoning model (思考模式)
|
|
config["quick_think_llm"] = "deepseek-chat" # DeepSeek chat model (非思考模式)
|
|
config["max_debate_rounds"] = 1 # Debate rounds
|
|
|
|
# Configure data vendors - 使用 yfinance 作为主要数据源
|
|
config["data_vendors"] = {
|
|
"core_stock_apis": "yfinance", # Options: yfinance, alpha_vantage, local
|
|
"technical_indicators": "yfinance", # Options: yfinance, alpha_vantage, local
|
|
"fundamental_data": "yfinance", # 改用 yfinance 避免 API 限制
|
|
"news_data": "yfinance", # 改用 yfinance 避免 API 限制
|
|
}
|
|
|
|
print("=" * 60)
|
|
print("初始化 TradingAgents (使用 DeepSeek API)")
|
|
print("=" * 60)
|
|
print(f"LLM Provider: {config['llm_provider']}")
|
|
print(f"Backend URL: {config['backend_url']}")
|
|
print(f"Deep Think Model: {config['deep_think_llm']}")
|
|
print(f"Quick Think Model: {config['quick_think_llm']}")
|
|
print("=" * 60)
|
|
|
|
# 只使用市场分析师，跳过新闻分析以避免 API 问题
|
|
# selected_analysts = ["market", "social", "fundamentals"] # 跳过 "news"
|
|
selected_analysts = ["market"] # 先只测试市场分析
|
|
|
|
print(f"\n选择的分析师: {selected_analysts}")
|
|
print("\n开始初始化 TradingAgents Graph...")
|
|
|
|
# Initialize with custom config
|
|
ta = TradingAgentsGraph(
|
|
debug=True,
|
|
config=config,
|
|
selected_analysts=selected_analysts
|
|
)
|
|
|
|
print("✓ TradingAgents Graph 初始化成功!")
|
|
print("\n开始分析 NVDA (英伟达) 在 2024-05-10 的交易决策...")
|
|
print("=" * 60)
|
|
|
|
# forward propagate
|
|
try:
|
|
_, decision = ta.propagate("NVDA", "2024-05-10")
|
|
print("\n" + "=" * 60)
|
|
print("✅ 分析完成!")
|
|
print("=" * 60)
|
|
print(f"交易决策: {decision}")
|
|
print("=" * 60)
|
|
except Exception as e:
|
|
print(f"\n❌ 分析过程中出现错误: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|