Add chess_move tool and update imports in agents and tools modules

agent/agents.py

@@ -15,6 +15,7 @@ from agent.tools import (
     TranscribeYoutubeVideo,
     UnderstandImageBytes,
     WikipediaSearchTool,
+    chess_move,
 )
 from agent.utils import gemini_client, gemini_model_liteLLM
 
@@ -79,6 +80,7 @@ class BasicAgent:
                 understand_image_bytes,
                 code_execution_tool,
                 wiki_retriever,
+                chess_move,
             ],
             model=model,
             additional_authorized_imports=["pandas"],

agent/tools.py

@@ -8,6 +8,14 @@ from markdownify import markdownify as md
 from PIL import Image
 from smolagents import Tool
 
+from smolagents import Tool
+
+chess_move = Tool.from_space(
+    "Agents-MCP-Hackathon/chess-mcp-server",
+    name="chess_server",
+    description="Find the best move for chess",
+    api_name="/predict_2" # top moves
+)
 
 class WikipediaSearchTool(Tool):
     name = "wikipedia_search"
@@ -107,7 +115,7 @@ class UnderstandImageBytes(Tool):
         prompt = "Analyze this image and provide a description of its content."
 
         response = client.models.generate_content(
-            model="gemini-2.5-flash-preview-05-20",
+            model="gemini-2.5-pro",
             contents=[
                 f"{prompt} And answer the question accurately based on the visual information in the image. question: {question} ",
                 types.Part.from_bytes(