Zeggai Abdellah commited on
Commit
b761483
·
1 Parent(s): f336d8d
Files changed (1) hide show
  1. prepare_env.py +1 -1
prepare_env.py CHANGED
@@ -284,7 +284,7 @@ def create_section_tools(embedding_function, llm):
284
  if os.path.exists(path):
285
  print(f"📁 Creating retriever for section {section} from {path}")
286
  vstore, docs = create_vectorstore_from_json(path, f"Guide_2023_{section}", embedding_function)
287
- section_retrievers[section] = create_retriever(vstore, docs, llm,)
288
  print(f"✅ Successfully created retriever for section {section}")
289
  else:
290
  print(f"⚠️ Warning: File not found for section {section}: {path}")
 
284
  if os.path.exists(path):
285
  print(f"📁 Creating retriever for section {section} from {path}")
286
  vstore, docs = create_vectorstore_from_json(path, f"Guide_2023_{section}", embedding_function)
287
+ section_retrievers[section] = create_retriever(vstore, docs, llm, bm25_k=7, vector_k=10)
288
  print(f"✅ Successfully created retriever for section {section}")
289
  else:
290
  print(f"⚠️ Warning: File not found for section {section}: {path}")