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, AnalyzeVideo, read_excel, read_python, wiki_search, web_search, arxiv_search, download_youtube_video
+from tools import ExtractTextFromImage, DescribeImage, TranscribeAudio, read_excel, read_python, wiki_search, web_search, arxiv_search, download_youtube_video, extract_audio_from_video
 
 
 class AgentState(TypedDict):
@@ -24,9 +24,8 @@ class BasicAgent():
         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, analyze_video, read_excel, read_python, wiki_search, web_search, arxiv_search, download_youtube_video]
+        self.tools = [extract_text_from_image, describe_image, transcribe_audio, read_excel, read_python, wiki_search, web_search, arxiv_search, download_youtube_video, extract_audio_from_video]
         self.chat_with_tools = chat.bind_tools(self.tools)
         self._initialize_graph()
         self._initialize_telemetry()