venthan commited on
Commit
b462580
·
1 Parent(s): cd8e4f7

change doc search type to mmr

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -37,7 +37,8 @@ class AASLDConversationalAgent():
37
  encode_kwargs={'normalize_embeddings': False}
38
  )
39
  self.doc_search = FAISS.from_documents(pages, embeddings_model)
40
- self.doc_retriever = self.doc_search.as_retriever()
 
41
  self.llm = ChatOpenAI(temperature = 0, model_name=LLM_MODEL)
42
  tool = create_retriever_tool(
43
  self.doc_retriever,
 
37
  encode_kwargs={'normalize_embeddings': False}
38
  )
39
  self.doc_search = FAISS.from_documents(pages, embeddings_model)
40
+ self.doc_retriever = self.doc_search.as_retriever(
41
+ search_type="mmr", search_kwargs={"k": 100})
42
  self.llm = ChatOpenAI(temperature = 0, model_name=LLM_MODEL)
43
  tool = create_retriever_tool(
44
  self.doc_retriever,