Update LG_agent.py

LG_agent.py

@@ -41,7 +41,6 @@ class AgentState(TypedDict):
 # 3. Assistant node
 
 def assistant(state: AgentState):
-    # Inject tools description to a system message
     tool_descriptions = "\n".join([
         f"{tool.name}{inspect.signature(tool.func)}:\n    {tool.description.strip()}"
         for tool in all_tools
@@ -55,8 +54,20 @@ def assistant(state: AgentState):
         )
     )
 
+    input_msgs = [sys_msg] + state["messages"]
+    print("\n🧠 Assistant received messages:")
+    for msg in input_msgs:
+        print(f"🔹 {msg.__class__.__name__}: {getattr(msg, 'content', '')[:200]}")
+
+    output = chat_with_tools.invoke(input_msgs)
+
+    print("\n🗣️ Assistant response:")
+    print("-" * 40)
+    print(getattr(output, 'content', '')[:500])
+    print("-" * 40)
+
     return {
-        "messages": [chat_with_tools.invoke([sys_msg] + state["messages"])],
+        "messages": [output],
     }
 
 # 4. Build the agent graph
@@ -75,13 +86,16 @@ def build_graph(max_steps: int = 5):
     def limited_invoke(state):
         steps = 0
         while steps < max_steps:
-            print(f"🔁 Step {steps + 1}")
+            print(f"\U0001f501 Step {steps + 1}")
             state = graph.invoke(state)
 
             for msg in state["messages"]:
                 if isinstance(msg, AIMessage):
-                    print(f"🤔 LLM Thought: {msg.content.strip()}")
-            
+                    print("\n🤖 Assistant says:")
+                    print("-" * 40)
+                    print(msg.content.strip())
+                    print("-" * 40)
+
             latest_message = state["messages"][-1] if state["messages"] else None
 
             if latest_message and isinstance(latest_message, AIMessage):