Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
4faf04f
1
Parent(s):
5a13635
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. "
|