giulia-fontanella commited on
Commit
958874c
·
verified ·
1 Parent(s): 2d91d8b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -2
agent.py CHANGED
@@ -10,7 +10,7 @@ from langgraph.prebuilt import tools_condition
10
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
11
  from langfuse.callback import CallbackHandler
12
 
13
- from tools import ExtractTextFromImage, DescribeImage, read_excel, read_python, wiki_search, web_search, arxiv_search
14
 
15
 
16
  class AgentState(TypedDict):
@@ -23,8 +23,9 @@ class BasicAgent():
23
 
24
  extract_text_from_image = tool(ExtractTextFromImage(self.vision_llm).__call__)
25
  describe_image = tool(DescribeImage(self.vision_llm).__call__)
 
26
 
27
- self.tools = [extract_text_from_image, describe_image, read_excel, read_python, wiki_search, web_search, arxiv_search]
28
  self.chat_with_tools = chat.bind_tools(self.tools)
29
  self._initialize_graph()
30
  self._initialize_telemetry()
 
10
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
11
  from langfuse.callback import CallbackHandler
12
 
13
+ from tools import ExtractTextFromImage, DescribeImage, TranscribeAudio, read_excel, read_python, wiki_search, web_search, arxiv_search
14
 
15
 
16
  class AgentState(TypedDict):
 
23
 
24
  extract_text_from_image = tool(ExtractTextFromImage(self.vision_llm).__call__)
25
  describe_image = tool(DescribeImage(self.vision_llm).__call__)
26
+ transcribe_audio = tool(TranscribeAudio(self.vision_llm).__call__)
27
 
28
+ self.tools = [extract_text_from_image, describe_image, transcribe_audio, read_excel, read_python, wiki_search, web_search, arxiv_search]
29
  self.chat_with_tools = chat.bind_tools(self.tools)
30
  self._initialize_graph()
31
  self._initialize_telemetry()