Update app.py

app.py

@@ -29,7 +29,7 @@ print("Loading LLM...")
 chat = ChatMistralAI(
     model="mistral-medium-2505",
     temperature=0.1,
-    max_retries=5,
+    max_retries=3,
     mistral_api_key = os.getenv("Mistral")
 )
 print(f"Model {chat.model} downloaded!")
@@ -133,7 +133,7 @@ run_code_tool = Tool(
     description="Executes a python code."
 )
 
-tools = [get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool, play_chess_tool, run_code_tool]
+tools = [get_file_path_tool, get_content_tool, search_web_tool, extract_text_tool, play_chess_tool, run_code_tool]
 chat_with_tools = chat.bind_tools(tools, parallel_tool_calls=False)
 
 
@@ -175,6 +175,7 @@ system_prompt = SystemMessage(
         If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.\
         If the question refers to an external content and there is no reference file attached, perform a web search and retrieve relevant information from the internet. \
         If there is a file attached, open the file and read it. \
+        If there is a code, execute it.
         Make sure that each final answer is preceded with 'FINAL ANSWER:' and is short: it should contain a number (without full stop at the end), a string (one or two words only, without full stop at the end) or a comma-separated list (without full stops at the end), nothing else. "
     )
 
@@ -186,7 +187,7 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         message = HumanMessage(content=question)
         print(message)
-        answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 5})['messages'][-1].content
+        answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
         answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))    
         answer = answer.replace('FINAL ANSWER:', '')
         answer = answer.replace('FINAL ANSWER', '')