fix: hybrid search

This commit is contained in:
Timothy J. Baek
2024-09-13 01:21:47 -04:00
parent 939bfd153e
commit 823093eea6
3 changed files with 3 additions and 6 deletions

View File

@@ -97,8 +97,8 @@ def query_doc_with_hybrid_search(
result = VECTOR_DB_CLIENT.get(collection_name=collection_name)
bm25_retriever = BM25Retriever.from_texts(
texts=result.documents,
metadatas=result.metadatas,
texts=result.documents[0],
metadatas=result.metadatas[0],
)
bm25_retriever.k = k