Update app.py

app.py

@@ -29,7 +29,7 @@ print("Loading LLM...")
 chat = ChatMistralAI(
     model="mistral-medium-2505",
     temperature=0.1,
-    max_retries=3,
+    max_retries=5,
     mistral_api_key = os.getenv("Mistral")
 )
 print(f"Model {chat.model} downloaded!")
@@ -90,8 +90,12 @@ def extract_text_from_image(img_path: str) -> str:
         return ""
 
 def play_chess():
-    return chess.Board()
+    board = chess.Board()
+    return board
 
+def run_code(code: str):
+    return exec(code)
+    
 # Initialize the tool
 get_file_path_tool = Tool(
     name="file_path_retriever",
@@ -123,7 +127,13 @@ play_chess_tool = Tool(
     description="Sets a chess board."
 )
 
-tools = [get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool, play_chess_tool]
+run_code_tool = Tool(
+    name="run_code_retriever",
+    func=run_code,
+    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]
 chat_with_tools = chat.bind_tools(tools, parallel_tool_calls=False)