Update agent.py

agent.py

@@ -10,7 +10,7 @@ from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langfuse.callback import CallbackHandler
 
-from tools import ExtractTextFromImage, DescribeImage, TranscribeAudio, read_excel, read_python, wiki_search, web_search, arxiv_search
+from tools import ExtractTextFromImage, DescribeImage, TranscribeAudio, AnalyzeVideo, read_excel, read_python, wiki_search, web_search, arxiv_search, download_youtube_video
 
 
 class AgentState(TypedDict):
@@ -19,13 +19,14 @@ class AgentState(TypedDict):
 
 class BasicAgent():
     def __init__(self, chat):
-        self.vision_llm = ChatOpenAI(model="gpt-4o")
+        self.multimodal_model = ChatOpenAI(model="gpt-4o")
         
-        extract_text_from_image = tool(ExtractTextFromImage(self.vision_llm).__call__)
-        describe_image = tool(DescribeImage(self.vision_llm).__call__)
-        transcribe_audio = tool(TranscribeAudio(self.vision_llm).__call__)
+        extract_text_from_image = tool(ExtractTextFromImage(self.multimodal_model).__call__)
+        describe_image = tool(DescribeImage(self.multimodal_model).__call__)
+        transcribe_audio = tool(TranscribeAudio(self.multimodal_model).__call__)
+        analyze_video = tool(AnalyzeVideo(self.multimodal_model).__call__)
 
-        self.tools = [extract_text_from_image, describe_image, transcribe_audio, read_excel, read_python, wiki_search, web_search, arxiv_search]
+        self.tools = [extract_text_from_image, describe_image, transcribe_audio, analyze_video, read_excel, read_python, wiki_search, web_search, arxiv_search, download_youtube_video]
         self.chat_with_tools = chat.bind_tools(self.tools)
         self._initialize_graph()
         self._initialize_telemetry()