sqfoo commited on
Commit
dd9278b
·
verified ·
1 Parent(s): 523374b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -3
agent.py CHANGED
@@ -15,6 +15,8 @@ from pypdf import PdfReader
15
  from langchain_community.tools import WikipediaQueryRun
16
  from langchain_community.utilities import WikipediaAPIWrapper
17
  from langchain_community.document_loaders import YoutubeLoader
 
 
18
 
19
  @tool
20
  def web_search(query: str) -> str:
@@ -76,8 +78,6 @@ def read_file(dir: str) -> str:
76
  return f.read()
77
 
78
  @tool
79
- from langchain_community.document_loaders import UnstructuredExcelLoader
80
- # from langchain_core.documents import Document
81
  def excel_read(dir: str) -> str:
82
  """Read the content of the excel file
83
  Args:
@@ -89,7 +89,6 @@ def excel_read(dir: str) -> str:
89
  return "\n".join(contents)
90
 
91
  @tool
92
- from langchain_community.document_loaders import AssemblyAIAudioTranscriptLoader
93
  def mp3_listen(dir: str) -> str:
94
  """Listen to the provided mp3 file
95
  Args:
 
15
  from langchain_community.tools import WikipediaQueryRun
16
  from langchain_community.utilities import WikipediaAPIWrapper
17
  from langchain_community.document_loaders import YoutubeLoader
18
+ from langchain_community.document_loaders import UnstructuredExcelLoader
19
+ from langchain_community.document_loaders import AssemblyAIAudioTranscriptLoader
20
 
21
  @tool
22
  def web_search(query: str) -> str:
 
78
  return f.read()
79
 
80
  @tool
 
 
81
  def excel_read(dir: str) -> str:
82
  """Read the content of the excel file
83
  Args:
 
89
  return "\n".join(contents)
90
 
91
  @tool
 
92
  def mp3_listen(dir: str) -> str:
93
  """Listen to the provided mp3 file
94
  Args: