Update agent.py

agent.py

@@ -357,7 +357,6 @@ def tool_node(state: AgentState) -> Dict[str, Any]:
         new_state = {"messages": []}
         return propagate_state(new_state, state)
     last = wrap_message(messages_list[-1])
-    # Safely retrieve pending tool_calls from the message's __dict__
     tool_calls = last.__dict__.get("tool_calls")
     if not (isinstance(last, AIMessage) and tool_calls):
         logger.warning("tool_node invoked without pending tool_calls")
@@ -462,9 +461,9 @@ def should_continue(state: AgentState) -> str:
     if "consultation complete" in last.content.lower():
         state["done"] = True
         return "end_conversation_turn"
-    state["done"] = False
-    # Return "start" to loop back.
-    return "start"
+    # If no tool calls are present, terminate the conversation instead of looping.
+    state["done"] = True
+    return "end_conversation_turn"
 
 def after_tools_router(state: AgentState) -> str:
     if state.get("interaction_warnings"):
@@ -480,17 +479,14 @@ class ClinicalAgent:
         wf.add_node("tools", tool_node)
         wf.add_node("reflection", reflection_node)
         wf.set_entry_point("start")
-        # Note: Added a "start" branch in the conditional edges.
         wf.add_conditional_edges("start", should_continue, {
             "continue_tools": "tools",
-            "start": "start",
             "end_conversation_turn": END
         })
         wf.add_conditional_edges("tools", after_tools_router, {
             "reflection": "reflection",
             "end_conversation_turn": END
         })
-        # Removed edge from reflection back to start to break the cycle.
         self.graph_app = wf.compile()
         logger.info("ClinicalAgent ready")