Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -1,12 +1,31 @@
 import os
-
-from agents.llms import WEB_SEARCH_MODEL, IMAGE_ANALYSIS_MODEL, AUDIO_ANALYSIS_MODEL, VIDEO_ANALYSIS_MODEL, YOUTUBE_ANALYSIS_MODEL, DOCUMENT_ANALYSIS_MODEL, ARITHMETIC_MODEL, CODE_GENERATION_MODEL, CODE_EXECUTION_MODEL, WEB_BROWSER_MODEL, IMG_TO_FEN_PROMPT
-from agents.mcp.mcp_client import call_mcp_tool, MCP_SSE_URL_CHESS_ANALYSIS, MCP_TOOL_CHESS_ANALYSIS
+from agents.llms import (
+    WEB_SEARCH_MODEL,
+    IMAGE_ANALYSIS_MODEL,
+    AUDIO_ANALYSIS_MODEL,
+    VIDEO_ANALYSIS_MODEL,
+    YOUTUBE_ANALYSIS_MODEL,
+    DOCUMENT_ANALYSIS_MODEL,
+    ARITHMETIC_MODEL,
+    CODE_GENERATION_MODEL,
+    CODE_EXECUTION_MODEL,
+    WEB_BROWSER_MODEL,
+    IMG_TO_FEN_PROMPT
+)
+from agents.mcp.mcp_client import (
+    call_mcp_tool,
+    MCP_SSE_URL_CHESS_ANALYSIS,
+    MCP_TOOL_CHESS_ANALYSIS
+)
 from crewai.tools import tool
 from crewai_tools import StagehandTool
 from google import genai
 from google.genai import types
-from utils.utils import read_docx_text, read_pptx_text, is_ext
+from utils.utils import (
+    read_docx_text,
+    read_pptx_text,
+    is_ext
+)
 
 class AITools():
     @tool("Web Search Tool")