Update agent.py

agent.py

@@ -27,7 +27,7 @@ IMPORTANT:
 - Do not make up tool names.
 - Do not loop unnecessarily.
 - Provide FINAL ANSWER as soon as you are confident.
-
+- If wiki tool doesnt give you enough information, try again with a shorter query, or a more genral query. Dont use a query that is too specific.
 """
 
 TOOLS = [
@@ -51,11 +51,11 @@ class SimpleLangGraphAgent:
         messages.append(HumanMessage(content=question))
 
         state = {"messages": messages}
-        final_state = self.agent.invoke(state, config={"recursion_limit": max_steps})
+        final_state = self.agent.invoke(state)
 
         for msg in final_state["messages"][::-1]:
             if "FINAL ANSWER:" in msg.content.upper():
-                print(f"\n\n msg.content = {msg.content} \n\n")
+                print(f"\n\n thoughts = {msg.content.split('Thought:')[-1].strip()} \n\n")
                 return msg.content.split("FINAL ANSWER:")[-1].strip()
 
         return "No FINAL ANSWER found."