mirror of
https://github.com/simonlin1212/TradingAgents-astock.git
synced 2026-10-30 09:29:56 +00:00
simon
f3c0fe4cfd
所有 18 个 akshare 调用替换为东财 datacenter/push2、新浪财经、 同花顺 10jqka、财联社 cls.cn 等直连 HTTP API。 彻底消除 akshare+pandas3.0+pyarrow 的 ArrowInvalid 崩溃， 以及 akshare 与 mootdx 的 httpx 版本冲突。 函数签名和返回格式保持不变，对上层 Agent 透明。
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""End-to-end test: run TradingAgents pipeline on A-stock 688017 via Kimi 2.6."""
|
|
|
|
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
|
from tradingagents.default_config import DEFAULT_CONFIG
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
config = DEFAULT_CONFIG.copy()
|
|
|
|
# --- LLM: Kimi 2.6 via Anthropic-compatible API ---
|
|
config["llm_provider"] = "anthropic"
|
|
config["deep_think_llm"] = "claude-sonnet-4-6" # Kimi maps internally
|
|
config["quick_think_llm"] = "claude-sonnet-4-6"
|
|
config["backend_url"] = "https://api.kimi.com/coding/"
|
|
|
|
# --- Data: A-stock vendor (mootdx + tencent + eastmoney + sina) ---
|
|
config["data_vendors"] = {
|
|
"core_stock_apis": "a_stock",
|
|
"technical_indicators": "a_stock",
|
|
"fundamental_data": "a_stock",
|
|
"news_data": "a_stock",
|
|
}
|
|
|
|
# --- Debate settings: minimal for first test ---
|
|
config["max_debate_rounds"] = 1
|
|
config["max_risk_discuss_rounds"] = 1
|
|
config["output_language"] = "Chinese"
|
|
|
|
print("=" * 60)
|
|
print("TradingAgents-Astock E2E Test")
|
|
print("Ticker: 688017")
|
|
print("Trade date: 2026-04-30")
|
|
print("LLM: Kimi 2.6 via Anthropic API")
|
|
print("Data: a_stock (mootdx + tencent + eastmoney + sina)")
|
|
print("=" * 60)
|
|
|
|
ta = TradingAgentsGraph(debug=True, config=config)
|
|
|
|
_, decision = ta.propagate("688017", "2026-04-30")
|
|
print("\n" + "=" * 60)
|
|
print("FINAL DECISION:")
|
|
print("=" * 60)
|
|
print(decision)
|