Spaces:
Configuration error
Configuration error
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
|