80 lines
3.0 KiB
Python
80 lines
3.0 KiB
Python
import json
|
|
|
|
from tradingagents.default_config import DEFAULT_CONFIG
|
|
from tradingagents.graph.trading_graph import TradingAgentsGraph, _merge_with_default_config
|
|
|
|
|
|
def test_merge_with_default_config_keeps_required_defaults():
|
|
merged = _merge_with_default_config({
|
|
"llm_provider": "anthropic",
|
|
"backend_url": "https://example.com/api",
|
|
})
|
|
|
|
assert merged["llm_provider"] == "anthropic"
|
|
assert merged["backend_url"] == "https://example.com/api"
|
|
assert merged["project_dir"] == DEFAULT_CONFIG["project_dir"]
|
|
assert merged["results_dir"] == DEFAULT_CONFIG["results_dir"]
|
|
|
|
|
|
def test_merge_with_default_config_merges_nested_vendor_settings():
|
|
merged = _merge_with_default_config({
|
|
"data_vendors": {
|
|
"news_data": "alpha_vantage",
|
|
},
|
|
"tool_vendors": {
|
|
"get_stock_data": "alpha_vantage",
|
|
},
|
|
})
|
|
|
|
assert merged["data_vendors"]["news_data"] == "alpha_vantage"
|
|
assert merged["data_vendors"]["core_stock_apis"] == DEFAULT_CONFIG["data_vendors"]["core_stock_apis"]
|
|
assert merged["tool_vendors"]["get_stock_data"] == "alpha_vantage"
|
|
|
|
|
|
def test_log_state_persists_research_provenance(tmp_path):
|
|
graph = TradingAgentsGraph.__new__(TradingAgentsGraph)
|
|
graph.config = {"results_dir": str(tmp_path)}
|
|
graph.ticker = "AAPL"
|
|
graph.log_states_dict = {}
|
|
|
|
final_state = {
|
|
"company_of_interest": "AAPL",
|
|
"trade_date": "2026-04-11",
|
|
"market_report": "",
|
|
"sentiment_report": "",
|
|
"news_report": "",
|
|
"fundamentals_report": "",
|
|
"investment_debate_state": {
|
|
"bull_history": "Bull Analyst: case",
|
|
"bear_history": "Bear Analyst: case",
|
|
"history": "Bull Analyst: case\nBear Analyst: case",
|
|
"current_response": "Recommendation: HOLD",
|
|
"judge_decision": "Recommendation: HOLD",
|
|
"research_status": "degraded",
|
|
"research_mode": "degraded_synthesis",
|
|
"timed_out_nodes": ["Bull Researcher"],
|
|
"degraded_reason": "bull_researcher_timeout",
|
|
"covered_dimensions": ["market"],
|
|
"manager_confidence": 0.0,
|
|
},
|
|
"trader_investment_plan": "",
|
|
"risk_debate_state": {
|
|
"aggressive_history": "",
|
|
"conservative_history": "",
|
|
"neutral_history": "",
|
|
"history": "",
|
|
"judge_decision": "",
|
|
},
|
|
"investment_plan": "Recommendation: HOLD",
|
|
"final_trade_decision": "HOLD",
|
|
}
|
|
|
|
TradingAgentsGraph._log_state(graph, "2026-04-11", final_state)
|
|
|
|
log_path = tmp_path / "AAPL" / "TradingAgentsStrategy_logs" / "full_states_log_2026-04-11.json"
|
|
payload = json.loads(log_path.read_text(encoding="utf-8"))
|
|
assert payload["investment_debate_state"]["research_status"] == "degraded"
|
|
assert payload["investment_debate_state"]["research_mode"] == "degraded_synthesis"
|
|
assert payload["investment_debate_state"]["timed_out_nodes"] == ["Bull Researcher"]
|
|
assert payload["investment_debate_state"]["manager_confidence"] == 0.0
|