anpigon commited on
Commit
c81eed9
1 Parent(s): 3d9ec47

feat(app.py): add run names to retrievers in ensemble retriever configuration for better logging and debugging

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -184,7 +184,11 @@ chroma_retriever = chroma_db.as_retriever(
184
  bm25_retriever = BM25Retriever.from_documents(combined_documents)
185
  bm25_retriever.k = 10
186
  ensemble_retriever = EnsembleRetriever(
187
- retrievers=[bm25_retriever, faiss_retriever, chroma_retriever],
 
 
 
 
188
  weights=[0.4, 0.3, 0.3],
189
  )
190
 
 
184
  bm25_retriever = BM25Retriever.from_documents(combined_documents)
185
  bm25_retriever.k = 10
186
  ensemble_retriever = EnsembleRetriever(
187
+ retrievers=[
188
+ bm25_retriever.with_config(run_name="bm25"),
189
+ faiss_retriever.with_config(run_name="faiss"),
190
+ chroma_retriever.with_config(run_name="chroma"),
191
+ ],
192
  weights=[0.4, 0.3, 0.3],
193
  )
194