mirror of
https://github.com/TauricResearch/TradingAgents.git
synced 2026-09-27 15:02:39 +03:00
Yijia-Xiao
a1b3b5bab8
- TradingAgentsGraph.run_settings(): version, provider, models, analysts, debate rounds, language and vendors; no endpoints, keys or paths - complete_report.md opens with the analysis date and those settings; the saved state log carries them as run_settings
107 lines
4.3 KiB
Python
107 lines
4.3 KiB
Python
"""Report parity: the shared writer produces the report tree for the CLI and the
|
|
programmatic API alike (#1037)."""
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
|
from tradingagents.reporting import write_report_tree
|
|
|
|
|
|
def _state():
|
|
return {
|
|
"market_report": "MKT",
|
|
"news_report": "NEWS",
|
|
"investment_debate_state": {"bull_history": "BULL"},
|
|
"investment_plan": "RM PLAN",
|
|
"trader_investment_plan": "TRADE",
|
|
"risk_debate_state": {"neutral_history": "NEUTRAL"},
|
|
"final_trade_decision": "PM DECISION",
|
|
}
|
|
|
|
|
|
SETTINGS = {"version": "0.5.2", "llm_provider": "openai", "deep_think_llm": "gpt-6-sol",
|
|
"quick_think_llm": "gpt-6-luna", "analysts": ["market", "news"], "max_debate_rounds": 1,
|
|
"max_risk_discuss_rounds": 2, "data_vendors": {"core_stock_apis": "yfinance"}}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_write_report_tree_creates_files(tmp_path):
|
|
out = write_report_tree(_state(), "AAPL", tmp_path)
|
|
assert out.name == "complete_report.md"
|
|
assert (tmp_path / "1_analysts" / "market.md").read_text() == "MKT"
|
|
assert (tmp_path / "1_analysts" / "news.md").read_text() == "NEWS"
|
|
assert (tmp_path / "2_research" / "manager.md").read_text() == "RM PLAN"
|
|
assert (tmp_path / "3_trading" / "trader.md").read_text() == "TRADE"
|
|
assert (tmp_path / "5_portfolio" / "decision.md").read_text() == "PM DECISION"
|
|
complete = out.read_text()
|
|
assert "Trading Analysis Report: AAPL" in complete
|
|
assert "MKT" in complete and "PM DECISION" in complete
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_save_reports_explicit_path(tmp_path):
|
|
graph = SimpleNamespace(run_settings=lambda: SETTINGS)
|
|
out = TradingAgentsGraph.save_reports(graph, _state(), "AAPL", save_path=tmp_path)
|
|
assert (tmp_path / "complete_report.md").exists()
|
|
assert out == tmp_path / "complete_report.md"
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_save_reports_defaults_under_results_dir(tmp_path):
|
|
mock_self = SimpleNamespace(config={"results_dir": str(tmp_path)}, run_settings=lambda: SETTINGS)
|
|
out = TradingAgentsGraph.save_reports(mock_self, _state(), "AAPL")
|
|
assert out.exists()
|
|
assert out.parent.parent.name == "reports" # results_dir/reports/AAPL_<stamp>/...
|
|
assert out.parent.name.startswith("AAPL_")
|
|
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_the_report_names_the_analysis_date_and_what_produced_it(tmp_path):
|
|
state = dict(_state(), trade_date="2026-09-23")
|
|
|
|
header = write_report_tree(state, "NVDA", tmp_path, settings=SETTINGS).read_text().split("## ")[0]
|
|
|
|
assert "Analysis date: 2026-09-23" in header
|
|
assert "TradingAgents 0.5.2" in header
|
|
assert "openai, deep gpt-6-sol, quick gpt-6-luna" in header
|
|
assert "Analysts: market, news" in header
|
|
assert "research debate rounds 1, risk debate rounds 2" in header
|
|
assert "core_stock_apis yfinance" in header
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_run_settings_record_the_run_without_endpoints_or_paths():
|
|
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
|
|
|
graph = object.__new__(TradingAgentsGraph)
|
|
graph.selected_analysts = ("market", "news")
|
|
graph.config = {"llm_provider": "openai", "deep_think_llm": "gpt-6-sol", "quick_think_llm": "gpt-6-luna",
|
|
"max_debate_rounds": 1, "max_risk_discuss_rounds": 1, "output_language": "English",
|
|
"data_vendors": {"core_stock_apis": "yfinance"}, "tool_vendors": {},
|
|
"backend_url": "https://user:secret@relay.example/v1", "results_dir": "/home/me/results"}
|
|
|
|
settings = graph.run_settings()
|
|
|
|
assert settings["analysts"] == ["market", "news"]
|
|
assert settings["deep_think_llm"] == "gpt-6-sol"
|
|
assert "secret" not in str(settings) and "/home/me" not in str(settings)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_a_partial_settings_dict_still_writes_the_report(tmp_path):
|
|
out = write_report_tree(_state(), "AAPL", tmp_path, settings={"llm_provider": "openai"})
|
|
assert "openai" in out.read_text()
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_run_settings_name_the_version_of_the_running_code():
|
|
import tradingagents
|
|
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
|
|
|
graph = object.__new__(TradingAgentsGraph)
|
|
graph.selected_analysts, graph.config = ("market",), {}
|
|
assert graph.run_settings()["version"] == tradingagents.__version__
|