Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
3935bca
1
Parent(s):
015887e
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -30,7 +30,7 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
|
|
| 30 |
from llama_index.llms.huggingface import HuggingFaceLLM
|
| 31 |
from llama_index.readers.assemblyai import AssemblyAIAudioTranscriptReader
|
| 32 |
from llama_index.readers.json import JSONReader
|
| 33 |
-
from llama_index.readers.web import
|
| 34 |
from llama_index.readers.youtube_transcript import YoutubeTranscriptReader
|
| 35 |
from llama_index.tools.arxiv import ArxivToolSpec
|
| 36 |
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
|
|
@@ -355,7 +355,7 @@ def search_and_extract_content_from_url(query: str) -> List[Document]:
|
|
| 355 |
loader = YoutubeTranscriptReader()
|
| 356 |
documents = loader.load_data(youtubelinks=[url])
|
| 357 |
else:
|
| 358 |
-
loader =
|
| 359 |
documents = loader.load_data(urls=[url])
|
| 360 |
for doc in documents:
|
| 361 |
doc.metadata["source"] = url
|
|
|
|
| 30 |
from llama_index.llms.huggingface import HuggingFaceLLM
|
| 31 |
from llama_index.readers.assemblyai import AssemblyAIAudioTranscriptReader
|
| 32 |
from llama_index.readers.json import JSONReader
|
| 33 |
+
from llama_index.readers.web import BeautifulSoupWebReader
|
| 34 |
from llama_index.readers.youtube_transcript import YoutubeTranscriptReader
|
| 35 |
from llama_index.tools.arxiv import ArxivToolSpec
|
| 36 |
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
|
|
|
|
| 355 |
loader = YoutubeTranscriptReader()
|
| 356 |
documents = loader.load_data(youtubelinks=[url])
|
| 357 |
else:
|
| 358 |
+
loader = BeautifulSoupWebReader()
|
| 359 |
documents = loader.load_data(urls=[url])
|
| 360 |
for doc in documents:
|
| 361 |
doc.metadata["source"] = url
|