Update agent.py
Browse files
agent.py
CHANGED
@@ -316,9 +316,11 @@ def create_forced_rag_pipeline():
|
|
316 |
|
317 |
forced_rag_pipeline = create_forced_rag_pipeline()
|
318 |
|
|
|
|
|
319 |
# Remplacer les tools individuels par le pipeline
|
320 |
information_retrieval_tool = FunctionTool.from_defaults(
|
321 |
-
fn=
|
322 |
name="information_retrieval_tool",
|
323 |
description=(
|
324 |
"This tool is the PRIMARY and MOST EFFECTIVE method for answering user queries by extracting and retrieving information from URLs or documents. "
|
|
|
316 |
|
317 |
forced_rag_pipeline = create_forced_rag_pipeline()
|
318 |
|
319 |
+
def forced_rag_pipeline_fn(input_path, query) :
|
320 |
+
return forced_rag_pipeline.run(input_path,query)
|
321 |
# Remplacer les tools individuels par le pipeline
|
322 |
information_retrieval_tool = FunctionTool.from_defaults(
|
323 |
+
fn=forced_rag_pipeline_fn,
|
324 |
name="information_retrieval_tool",
|
325 |
description=(
|
326 |
"This tool is the PRIMARY and MOST EFFECTIVE method for answering user queries by extracting and retrieving information from URLs or documents. "
|