Upload ai_tools.py

agents/tools/ai_tools.py

@@ -9,8 +9,8 @@ from agents.models.llms import (
     LLM_WEB_BROWSER,
     LLM_IMAGE_ANALYSIS,
     LLM_AUDIO_ANALYSIS,
-    LLM_VIDEO_FILE_ANALYSIS,
-    LLM_VIDEO_URL_ANALYSIS,
+    LLM_VIDEO_ANALYSIS,
+    LLM_YOUTUBE_ANALYSIS,
     LLM_DOCUMENT_ANALYSIS,
     LLM_CODE_GENERATION,
     LLM_CODE_EXECUTION,
@@ -20,7 +20,7 @@ from agents.models.llms import (
 
     THINKING_LEVEL_WEB_SEARCH,
     THINKING_LEVEL_MEDIA_ANALYSIS,
-    THINKING_LEVEL_VIDEO_URL_ANALYSIS,
+    THINKING_LEVEL_YOUTUBE_ANALYSIS,
     THINKING_LEVEL_DOCUMENT_ANALYSIS,
     THINKING_LEVEL_CODE_GENERATION,
     THINKING_LEVEL_CODE_EXECUTION,
@@ -192,8 +192,8 @@ class AITools():
         """
         return AITools._media_analysis_tool("audio_analysis_tool", LLM_AUDIO_ANALYSIS, question, file_path)
     
-    @tool("Video File Analysis Tool")
-    def video_file_analysis_tool(question: str, file_path: str) -> str:
+    @tool("Video Analysis Tool")
+    def video_analysis_tool(question: str, file_path: str) -> str:
         """Given a question and video file, analyze the video to answer the question.
     
         Args:
@@ -206,44 +206,44 @@ class AITools():
         Raises:
             RuntimeError: If processing fails
         """
-        return AITools._media_analysis_tool("video_file_analysis_tool", LLM_VIDEO_FILE_ANALYSIS, question, file_path)
+        return AITools._media_analysis_tool("video_analysis_tool", LLM_VIDEO_ANALYSIS, question, file_path)
             
-    @tool("Video URL Analysis Tool")
-    def video_url_analysis_tool(question: str, url: str) -> str:
-        """Given a question and video URL, analyze the video to answer the question.
+    @tool("YouTube Analysis Tool")
+    def youtube_analysis_tool(question: str, url: str) -> str:
+        """Given a question and YouTube URL, analyze the video to answer the question.
     
         Args:
-            question (str): Question about a video URL
-            url (str): The video URL
+            question (str): Question about a YouTube video
+            url (str): The YouTube URL
             
         Returns:
-            str: Answer to the question about the video URL
+            str: Answer to the question about the YouTube video
             
         Raises:
             RuntimeError: If processing fails
         """
-        print(f"🛠️ AITools: video_url_analysis_tool: question={question}, url={url}")
+        print(f"🛠️ AITools: youtube_analysis_tool: question={question}, url={url}")
         
         try:
             client = AITools._get_client()
     
             result = client.models.generate_content(
-                model=LLM_VIDEO_URL_ANALYSIS,
+                model=LLM_YOUTUBE_ANALYSIS,
                 contents=types.Content(
                     parts=[types.Part(file_data=types.FileData(file_uri=url)),
                            types.Part(text=question)]
                 ),
                 config=types.GenerateContentConfig(
                     thinking_config=types.ThinkingConfig(
-                        thinking_level=THINKING_LEVEL_VIDEO_URL_ANALYSIS
+                        thinking_level=THINKING_LEVEL_YOUTUBE_ANALYSIS
                     )
                 )
             ).strip()
 
-            print(f"🛠️ AITools: video_url_analysis_tool: result={result}")
+            print(f"🛠️ AITools: youtube_analysis_tool: result={result}")
             return result
         except Exception as e:
-            print(f"⚠️ AITools: video_url_analysis_tool: exception={str(e)}")
+            print(f"⚠️ AITools: youtube_analysis_tool: exception={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
     
     @tool("Document Analysis Tool")