Coool2 commited on
Commit
7c8e074
·
verified ·
1 Parent(s): dec17b3

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -295,9 +295,9 @@ def read_and_parse_fn(input_path: str):
295
  """Function compatible avec QueryPipeline"""
296
  return read_and_parse_content(input_path)
297
 
298
- def create_rag_fn(documents):
299
  """Function compatible avec QueryPipeline"""
300
- return create_rag_tool(documents)
301
 
302
  # Créer le pipeline avec FnComponent
303
  def create_forced_rag_pipeline():
 
295
  """Function compatible avec QueryPipeline"""
296
  return read_and_parse_content(input_path)
297
 
298
+ def create_rag_fn(documents, query = None):
299
  """Function compatible avec QueryPipeline"""
300
+ return create_rag_tool(documents, query)
301
 
302
  # Créer le pipeline avec FnComponent
303
  def create_forced_rag_pipeline():