Update services/graph_service.py

services/graph_service.py

@@ -13,6 +13,7 @@ from langgraph.prebuilt import ToolNode
 
 from tools.analysis_tools import trigger_interview_analysis
 
+
 class AgentState(TypedDict):
     messages: Annotated[Sequence[BaseMessage], lambda x, y: x + y]
     user_id: str
@@ -98,6 +99,7 @@ class GraphInterviewProcessor:
             "conversation_history": state["messages"]
         })
         return {"messages": [response['output']]}
+
     def _router(self, state: AgentState):
         """Décide du chemin à suivre après la réponse de l'agent."""
         last_message = state["messages"][-1]
@@ -105,6 +107,7 @@ class GraphInterviewProcessor:
             return "call_tool"
         else:
             return "end_turn"
+
     def _build_graph(self) -> any:
         """Construit et compile le graphe d'états."""
         tool_node = ToolNode([trigger_interview_analysis])
@@ -114,15 +117,16 @@ class GraphInterviewProcessor:
         graph.add_node("tools", tool_node)
 
         graph.set_entry_point("agent")
+
         graph.add_conditional_edges(
             "agent",
             self._router,
             {
-                "call_tool": "tools", 
-                "end_turn": END     
+                "call_tool": "tools",
+                "end_turn": END
             }
         )
-
+        
         graph.add_edge("tools", "agent")
 
         return graph.compile()
@@ -137,15 +141,24 @@ class GraphInterviewProcessor:
             "messages": langchain_messages,
         }
         
+        logging.info(f"Invoking graph with initial state: {initial_state}")
         final_state = self.graph.invoke(initial_state)
+
+        logging.info(f"Graph finished. Final state: {final_state}")
         
+        if not final_state or not final_state.get('messages'):
+            logging.error("Final state is empty or does not contain messages.")
+            return {"response": "Erreur: Impossible de générer une réponse.", "status": "finished"}
+
         last_message = final_state['messages'][-1]
+        logging.info(f"Last message from graph: {last_message}")
         
-        # Le statut est 'finished' si le dernier message contient un appel à un outil.
-        # Cela signifie que le graphe a pris le chemin de l'analyse.
         status = "finished" if hasattr(last_message, 'tool_calls') and last_message.tool_calls else "interviewing"
         
+        response_content = last_message.content
+        logging.info(f"Final response content: '{response_content}', status: {status}")
+        
         return {
-            "response": last_message.content,
+            "response": response_content,
             "status": status
         }