Coool2 commited on
Commit
1d035ba
·
verified ·
1 Parent(s): 5adab1e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -5
agent.py CHANGED
@@ -9,8 +9,8 @@ from llama_index.core.query_engine import RetrieverQueryEngine
9
  from llama_index.readers.file import PDFReader, DocxReader, CSVReader, ImageReader
10
  import os
11
  from typing import List, Dict, Any
12
- from llama_index.core.tools.ondemand_loader_tool import OnDemandLoaderTool
13
  from llama_index.tools.arxiv import ArxivToolSpec
 
14
  import duckduckgo_search as ddg
15
  import re
16
  from llama_index.core.agent.workflow import ReActAgent
@@ -239,10 +239,6 @@ analysis_agent = FunctionAgent(
239
  max_steps=5
240
  )
241
 
242
-
243
- from llama_index.tools.arxiv import ArxivToolSpec
244
- from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
245
-
246
  class IntelligentSourceRouter:
247
  def __init__(self):
248
  # Initialize ArXiv and DuckDuckGo as LlamaIndex tools
 
9
  from llama_index.readers.file import PDFReader, DocxReader, CSVReader, ImageReader
10
  import os
11
  from typing import List, Dict, Any
 
12
  from llama_index.tools.arxiv import ArxivToolSpec
13
+ from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
14
  import duckduckgo_search as ddg
15
  import re
16
  from llama_index.core.agent.workflow import ReActAgent
 
239
  max_steps=5
240
  )
241
 
 
 
 
 
242
  class IntelligentSourceRouter:
243
  def __init__(self):
244
  # Initialize ArXiv and DuckDuckGo as LlamaIndex tools