Update LG_agent.py

LG_agent.py

@@ -37,9 +37,10 @@ def assistant(state: AgentState):
         content=(
             "You are a helpful AI assistant who solves GAIA benchmark questions using step-by-step reasoning.\n"
             "Before answering, always think out loud and plan your approach.\n"
-            "Use tools when you lack information or need external data.\n"
-            "Never guess or make assumptions—if something is unknown, state it clearly or use a tool.\n"
-            "Conclude with a clear and supported final answer.\n"
+            "Use tools when you need external data or must transform text.\n"
+            "After every tool call, always analyze the result and continue reasoning to arrive at a final answer.\n"
+            "Never treat tool outputs as final — interpret them and continue solving the task step-by-step.\n"
+            "Finish with a clear and concise answer, such as 'The answer is: right'.\n"
             "\nAvailable tools:\n"
             f"{tool_descriptions}"
         )
@@ -108,30 +109,6 @@ def build_graph(max_steps: int = 5):
     
         return state
 
-    # def limited_invoke(state):
-    #     steps = 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 not getattr(latest_message, "tool_calls", None):
-    #                 print("✅ Final answer detected. Ending loop.")
-    #                 break
-    #             else:
-    #                 print("🔄 Tool call detected — continuing loop.")
-    #         steps += 1
-    #     return state
-
     return limited_invoke
 
 # 5. BasicAgent class