Spaces:
Runtime error
Runtime error
change k results to 5
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ class AASLDConversationalAgent():
|
|
| 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":
|
| 42 |
self.llm = ChatOpenAI(temperature = 0, model_name=LLM_MODEL)
|
| 43 |
tool = create_retriever_tool(
|
| 44 |
self.doc_retriever,
|
|
|
|
| 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": 5})
|
| 42 |
self.llm = ChatOpenAI(temperature = 0, model_name=LLM_MODEL)
|
| 43 |
tool = create_retriever_tool(
|
| 44 |
self.doc_retriever,
|