Spaces:

MikeMann
/

PrototypGrundschutzChatbot

Paused

MikeMann commited on Jan 1

Commit

c6b35f5

1 Parent(s): bafa1b0

added Hybrid Search with BM25

Files changed (1) hide show

app.py CHANGED Viewed

@@ -315,7 +315,7 @@ class BSIChatbot:
         global vectorstore
         if hybridSearch == True:
             allDocs = self.retrieveDocFromFaiss()
-            bm25_retriever = BM25Retriever.from_documents(allDocs.page_content)
             #TODO!
             bm25_retriever.k= 4
             vectordb = vectorstore.as_retriever(search_kwargs={"k":4})

         global vectorstore
         if hybridSearch == True:
             allDocs = self.retrieveDocFromFaiss()
+            bm25_retriever = BM25Retriever.from_documents(allDocs)
             #TODO!
             bm25_retriever.k= 4
             vectordb = vectorstore.as_retriever(search_kwargs={"k":4})