Agent bug

agent/ALL_node.py

@@ -43,6 +43,7 @@ def user_node(state:MYSTATE):
 
     if_image = bool(image_path)
     d=[{"query": query, "if_image": if_image, "image_path": image_path}]
+    print({"user":d})
     return {"user":d}
 
 
@@ -77,6 +78,7 @@ def coordinator_node(state:MYSTATE):
         "user_input": state["user"][0]["query"],
         "image_status": imagestatus
     })
+    print({"coordinator":[coordinator_output]})
     return {"coordinator":[coordinator_output]}
 
 
@@ -121,6 +123,7 @@ def router_node(state: MYSTATE):
         short_term_output=short_term_checking(state,False)
         
     router_output["short_term_memory"] = short_term_output
+    print(router_output)
     return {"router":[router_output]}
 
 
@@ -182,6 +185,7 @@ def main_llm_node(state:MYSTATE):
     main_history_dict = retrieve_dict("main_history")
     main_history_dict[str(change_index)]["llm_remark"] = main_llm_output
     save_dict("main_history", main_history_dict)
+    print(main_llm_output)
 
     return {"main_llm":[main_llm_output]}
 

routes/chat.py

@@ -40,13 +40,43 @@ async def chat_with_agent(
         # Clean up the temporary image file
         if image_path and os.path.exists(image_path):
             os.remove(image_path)
+
+    # --- FIX STARTS HERE ---
+    
+    # 1. Check if the graph ended early (Coordinator decided to reply directly to user)
+    if result.get("coordinator") and result["coordinator"][0].get("return_to") == "user":
+        coordinator_data = result["coordinator"][0]
         
-    # Extract the final JSON response from the main_llm node
-    final_response_str = result.get("main_llm", ["{}"])[0]
+        # Extract the direct message from the coordinator
+        final_response = {
+            "message_to_user": coordinator_data.get("message", {}).get("message_user", ""),
+            "llm_remarks": coordinator_data.get("llm_remarks", "Responded directly from coordinator."),
+            "navigate_to": None
+        }
+        
+    # 2. Otherwise, the graph went all the way to the end (main_llm)
+    else:
+        final_response_str = result.get("main_llm", ["{}"])[0]
+        try:
+            final_response = json.loads(final_response_str)
+        except json.JSONDecodeError:
+            # Fallback if the LLM output isn't perfect JSON
+            final_response = {
+                "message_to_user": final_response_str, 
+                "llm_remarks": "JSON parse failed",
+                "navigate_to": None
+            }
+            
+    # --- FIX ENDS HERE ---
+            
+    return final_response
+        
+    # # Extract the final JSON response from the main_llm node
+    # final_response_str = result.get("main_llm", ["{}"])[0]
     
-    try:
-        final_response = json.loads(final_response_str)
-    except json.JSONDecodeError:
-        final_response = {"message_to_user": final_response_str, "navigate_to": None}
+    # try:
+    #     final_response = json.loads(final_response_str)
+    # except json.JSONDecodeError:
+    #     final_response = {"message_to_user": final_response_str, "navigate_to": None}
         
-    return final_response
+    # return final_response