Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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:
|