Coool2 commited on
Commit
dec17b3
·
verified ·
1 Parent(s): 449486e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
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=lambda input_path: forced_rag_pipeline.run(input_path),
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. "