feat: add Chief Analyst as terminal node in LangGraph pipeline

2026-03-24 14:07:51 +01:00 · 2026-03-24 14:07:51 +01:00 · 0f74540e04
parent 1d7273d765
commit 0f74540e04
1 changed files with 5 additions and 1 deletions
--- a/tradingagents/graph/setup.py
+++ b/tradingagents/graph/setup.py
@ -200,7 +200,11 @@ class GraphSetup:
            },
        )

-        workflow.add_edge("Risk Judge", END)
+        from tradingagents.agents.managers.chief_analyst import create_chief_analyst
+        chief_analyst_node = create_chief_analyst(self.deep_thinking_llm)
+        workflow.add_node("Chief Analyst", chief_analyst_node)
+        workflow.add_edge("Risk Judge", "Chief Analyst")
+        workflow.add_edge("Chief Analyst", END)

        # Compile and return
        return workflow.compile(checkpointer=checkpointer)