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