Update agent.py

agent.py

@@ -102,17 +102,17 @@ def build_graph():
 
 
     def assistant(state: MessagesState):
-    """Assistant node: invoke LLM with tools."""
-    print("---Calling Assistant---") # Added print for debugging
-    
-    # 确保系统消息在消息列表的开头
-    messages = state["messages"]
-    if not any(isinstance(m, SystemMessage) for m in messages):
-        messages = [SystemMessage(content=system_prompt)] + messages
-    
-    result = llm_with_tools.invoke(messages)
-    print(f"---Assistant Response: {result}") # Added print for debugging
-    return {"messages": [result]}
+        """Assistant node: invoke LLM with tools."""
+        print("---Calling Assistant---") # Added print for debugging
+        
+        # 确保系统消息在消息列表的开头
+        messages = state["messages"]
+        if not any(isinstance(m, SystemMessage) for m in messages):
+            messages = [SystemMessage(content=system_prompt)] + messages
+        
+        result = llm_with_tools.invoke(messages)
+        print(f"---Assistant Response: {result}") # Added print for debugging
+        return {"messages": [result]}
 
     builder = StateGraph(MessagesState)
     builder.add_node("assistant", assistant)