Create agents/nodes

agents/nodes

@@ -0,0 +1,51 @@
+from langchain.agents import AgentExecutor, create_tool_calling_agent
+from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
+
+def news_node(state: GraphState, agent: AgentExecutor) -> GraphState:
+    ticker = state["ticker"]
+    q = f"Research recent news for {ticker}. Focus on price-moving catalysts."
+    res = agent.invoke({"input": q})
+    state["news_summary"] = res["output"]
+    state["completed"] = list(set(state["completed"] + ["news"]))
+    return state
+
+
+def earnings_node(state: GraphState, agent: AgentExecutor) -> GraphState:
+    ticker = state["ticker"]
+    q = f"Analyze earnings for {ticker}. Summarize last and upcoming earnings. Use the tool."
+    res = agent.invoke({"input": q})
+    state["earnings_summary"] = res["output"]
+    state["completed"] = list(set(state["completed"] + ["earnings"]))
+    return state
+
+
+def market_node(state: GraphState, agent: AgentExecutor) -> GraphState:
+    ticker = state["ticker"]
+    q = f"Provide a market snapshot for {ticker}. Use the tool."
+    res = agent.invoke({"input": q})
+    state["market_summary"] = res["output"]
+    state["completed"] = list(set(state["completed"] + ["market"]))
+    return state
+
+
+def synth_node(state: GraphState, synthesizer_chain) -> GraphState:
+    out = synthesizer_chain.invoke(
+        {
+            "ticker": state["ticker"],
+            "news_summary": state.get("news_summary", ""),
+            "earnings_summary": state.get("earnings_summary", ""),
+            "market_summary": state.get("market_summary", ""),
+        }
+    )
+    state["final_recommendation"] = out.content if hasattr(out, "content") else str(out)
+    return state
+    
+def supervisor_node(state: GraphState) -> GraphState:
+    # Do any bookkeeping here if needed; otherwise just pass state through
+    return state
+
+AGENTS = ["news", "earnings", "market"]
+
+def supervisor_router(state: GraphState) -> str:
+    remaining = [a for a in AGENTS if a not in state.get("completed", [])]
+    return remaining[0] if remaining else "synth"