Spaces:
Runtime error
Runtime error
change doc search type to mmr
Browse files
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,
|