Update services/graph_service.py

services/graph_service.py

@@ -16,6 +16,7 @@ class AgentState(TypedDict):
     messages: Annotated[Sequence[BaseMessage], lambda x, y: x + y]
     user_id: str
     job_offer_id: str
+    job_description: str
 
 class GraphInterviewProcessor:
     """
@@ -77,6 +78,19 @@ class GraphInterviewProcessor:
         
         return prompt | llm_with_tools
 
+    def _should_continue(self, state: InterviewState) -> str:
+        """
+        Détermine si l'entretien doit continuer ou se terminer.
+        """
+        messages = state.get('messages', [])
+        last_message = messages[-1]
+        if hasattr(last_message, 'tool_calls') and last_message.tool_calls:
+            for tool_call in last_message.tool_calls:
+                if tool_call.get('name') == 'trigger_interview_analysis':
+                    print("Condition de fin détectée : appel à trigger_interview_analysis.")
+                    return "end" 
+        return "continue"
+
     def _agent_node(self, state: AgentState):
         """Prépare le prompt et appelle le runnable de l'agent."""
 
@@ -109,37 +123,45 @@ class GraphInterviewProcessor:
             "messages": state["messages"],
             "job_description": job_description 
         })
-        return {"messages": [response]}
+        return {
+            "messages": [response],
+            "job_description": job_description_str
+        }
 
-    def _router(self, state: AgentState):
-        """Décide du chemin à suivre après la réponse de l'agent."""
+    def _router(self, state: AgentState) -> str:
+        """
+        Route le flux du graphe en fonction de la dernière réponse de l'agent.
+        - Si un outil d'analyse final est appelé, termine le graphe.
+        - Si un autre outil est appelé, va au noeud d'outils.
+        - Sinon, termine le tour de conversation.
+        """
         last_message = state["messages"][-1]
-        if hasattr(last_message, 'tool_calls') and last_message.tool_calls and len(last_message.tool_calls) > 0:
+        if hasattr(last_message, 'tool_calls') and last_message.tool_calls:
+            if any(tool_call.get('name') == 'trigger_interview_analysis' for tool_call in last_message.tool_calls):
+                print(">>> Routeur : Appel à l'outil final détecté. Terminaison du graphe.")
+                return "call_final_tool"
             return "call_tool"
-        else:
-            return "end_turn"
+        return "end_turn"
 
     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", tool_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]]):
@@ -154,6 +176,7 @@ class GraphInterviewProcessor:
             "user_id": self.user_id,
             "job_offer_id": self.job_offer_id,
             "messages": langchain_messages,
+            "job_description": json.dumps(self.job_offer, ensure_ascii=False),
         }
         
         final_state = self.graph.invoke(initial_state)