Spaces:

itachi-ai
/

Chat-Bot

Sleeping

itachi-ai commited on Mar 26, 2024

Commit

a3cd6cc

•

1 Parent(s): 221b1c2

updated

Files changed (1) hide show

embed_with_db.py CHANGED Viewed

@@ -62,7 +62,7 @@ class VECTORDB_STORE:
         return None
     def chain(self):
-        retriever = self.vectordb_store.as_retriever(search_kwargs={"k": 5})
         chain = {'context': retriever, 'question': RunnablePassthrough()} | prompt | llm_model | parser
         return chain

         return None
     def chain(self):
+        retriever = self.vectordb_store.as_retriever(search_kwargs={"k": 10})
         chain = {'context': retriever, 'question': RunnablePassthrough()} | prompt | llm_model | parser
         return chain