Coool2 commited on
Commit
3935bca
·
1 Parent(s): 015887e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
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 TrafilaturaWebReader
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 = TrafilaturaWebReader()
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