Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
1f158db
1
Parent(s):
bd0cde3
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -239,8 +239,14 @@ def create_rag_tool_fn(documents: List[Document], query: str = None) -> Union[Qu
|
|
| 239 |
|
| 240 |
return rag_engine_tool
|
| 241 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 242 |
information_retrieval_tool = FunctionTool.from_defaults(
|
| 243 |
-
fn=
|
| 244 |
name="information_retrieval_tool",
|
| 245 |
description=(
|
| 246 |
"This is the BEST and OPTIMAL tool to query information from documents parsed from URLs or files. "
|
|
|
|
| 239 |
|
| 240 |
return rag_engine_tool
|
| 241 |
|
| 242 |
+
def information_retrieval_fn (paths : List[str], query : str = None) -> Union[QueryEngineTool, str]:
|
| 243 |
+
docs = []
|
| 244 |
+
for path in paths :
|
| 245 |
+
docs.append(read_and_parse_content(path))
|
| 246 |
+
return create_rag_tool_fn(docs,query)
|
| 247 |
+
|
| 248 |
information_retrieval_tool = FunctionTool.from_defaults(
|
| 249 |
+
fn=information_retrieval_fn,
|
| 250 |
name="information_retrieval_tool",
|
| 251 |
description=(
|
| 252 |
"This is the BEST and OPTIMAL tool to query information from documents parsed from URLs or files. "
|