Initial commit

graph.py

@@ -18,6 +18,7 @@ from prompts import SUPERVISOR_PROMPT, ADS_ANALYST_PROMPT
 class AgentState(TypedDict):
     messages: Annotated[list[BaseMessage], add_messages]
     next_node: str
+    agent_responded: bool
 
 # ==========================================
 # 2. ИНИЦИАЛИЗАЦИЯ LLM (OpenRouter)
@@ -45,7 +46,7 @@ async def ads_node(state: AgentState):
     # Отдаем агенту готовый массив
     result = await ads_agent_runnable.ainvoke({"messages": messages_with_prompt})
     
-    return {"messages": [result["messages"][-1]]}
+    return {"messages": [result["messages"][-1]], "agent_responded": True}
 
 # ==========================================
 # 4. УЗЕЛ: СУПЕРВАЙЗЕР (Оркестратор)
@@ -54,9 +55,12 @@ class RouterOutput(BaseModel):
     next_node: Literal["Ads_Analyst", "FINISH"]
 
 async def supervisor_node(state: AgentState):
+    # Если агент уже ответил — детерминированно завершаем, без вызова LLM
+    if state.get("agent_responded"):
+        return {"next_node": "FINISH", "agent_responded": False}
     messages = [SystemMessage(content=SUPERVISOR_PROMPT)] + state["messages"]
     response = await llm.with_structured_output(RouterOutput).ainvoke(messages)
-    return {"next_node": response.next_node}
+    return {"next_node": response.next_node, "agent_responded": False}
 
 # ==========================================
 # 5. СБОРКА ГРАФА СОСТОЯНИЙ