Update LG_agent.py

LG_agent.py

@@ -76,37 +76,37 @@ def build_graph(max_steps: int = 5):
     graph = builder.compile()
 
     def limited_invoke(state, max_steps: int = 5, max_reasoning_steps_after_tool: int = 2):
-    steps = 0
-    reasoning_steps_since_last_tool = 0
-
-    while steps < max_steps:
-        print(f"\U0001f501 Step {steps + 1}")
-        state = graph.invoke(state)
-
-        for msg in state["messages"]:
-            if isinstance(msg, AIMessage):
-                print("\n🤖 Assistant says:")
-                print("-" * 40)
-                print(msg.content.strip())
-                print("-" * 40)
-
-        latest_message = state["messages"][-1] if state["messages"] else None
-
-        if isinstance(latest_message, AIMessage):
-            if latest_message.tool_calls:
-                print("🔄 Tool call detected — continuing loop.")
-                reasoning_steps_since_last_tool = 0  # reset counter
-            else:
-                reasoning_steps_since_last_tool += 1
-                print(f"🧠 No tool call — reasoning step #{reasoning_steps_since_last_tool}")
-
-                if reasoning_steps_since_last_tool >= max_reasoning_steps_after_tool:
-                    print("✅ Final answer assumed after sufficient reasoning.")
-                    break
-
-        steps += 1
-
-    return state
+        steps = 0
+        reasoning_steps_since_last_tool = 0
+    
+        while steps < max_steps:
+            print(f"\U0001f501 Step {steps + 1}")
+            state = graph.invoke(state)
+    
+            for msg in state["messages"]:
+                if isinstance(msg, AIMessage):
+                    print("\n🤖 Assistant says:")
+                    print("-" * 40)
+                    print(msg.content.strip())
+                    print("-" * 40)
+    
+            latest_message = state["messages"][-1] if state["messages"] else None
+    
+            if isinstance(latest_message, AIMessage):
+                if latest_message.tool_calls:
+                    print("🔄 Tool call detected — continuing loop.")
+                    reasoning_steps_since_last_tool = 0  # reset counter
+                else:
+                    reasoning_steps_since_last_tool += 1
+                    print(f"🧠 No tool call — reasoning step #{reasoning_steps_since_last_tool}")
+    
+                    if reasoning_steps_since_last_tool >= max_reasoning_steps_after_tool:
+                        print("✅ Final answer assumed after sufficient reasoning.")
+                        break
+    
+            steps += 1
+    
+        return state
 
     # def limited_invoke(state):
     #     steps = 0
@@ -132,7 +132,7 @@ def build_graph(max_steps: int = 5):
     #         steps += 1
     #     return state
 
-    # return limited_invoke
+    return limited_invoke
 
 # 5. BasicAgent class