Update services/graph_service.py

services/graph_service.py

@@ -10,7 +10,7 @@ from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langgraph.graph import StateGraph, END
 from langgraph.prebuilt import ToolNode
 
-from tools.analysis_tools import trigger_interview_analysis
+from tools.analysis_tools import trigger_interview_analysis, _final_analysis_node
 
 class AgentState(TypedDict):
     messages: Annotated[Sequence[BaseMessage], lambda x, y: x + y]
@@ -144,7 +144,7 @@ class GraphInterviewProcessor:
                 return "call_final_tool"
             return "call_tool"
         return "end_turn"
-
+'''
     def _build_graph(self) -> any:
         """Construit et compile le graphe d'états."""
         tool_node = ToolNode([trigger_interview_analysis])
@@ -165,6 +165,27 @@ class GraphInterviewProcessor:
         graph.add_edge("tools", "agent")
         graph.add_edge("final_tool_node", END)
         return graph.compile()
+'''
+    def _build_graph(self) -> any:
+    """Construit et compile le graphe d'états."""
+        tool_node = ToolNode([trigger_interview_analysis])
+        graph = StateGraph(AgentState)
+        graph.add_node("agent", self._agent_node)
+        graph.add_node("tools", tool_node)
+        graph.add_node("final_tool_node", self._final_analysis_node) 
+        graph.set_entry_point("agent")
+        graph.add_conditional_edges(
+            "agent",
+            self._router,
+            {
+                "call_tool": "tools",
+                "call_final_tool": "final_tool_node",
+                "end_turn": END
+            }
+        )
+        graph.add_edge("tools", "agent")
+        graph.add_edge("final_tool_node", END)
+        return graph.compile()
 
     def invoke(self, messages: List[Dict[str, Any]]):
         """Point d'entrée pour lancer une conversation dans le graphe."""