Update models/langOpen.py
Browse files- models/langOpen.py +1 -1
models/langOpen.py
CHANGED
|
@@ -38,7 +38,7 @@ class LangOpen:
|
|
| 38 |
print("query_str: ", query_str)
|
| 39 |
print("model_name: ", self.llm.model_name)
|
| 40 |
#docs = self.index.similarity_search(query_str, k=4)
|
| 41 |
-
vectorstore_retriever =
|
| 42 |
compressor = CohereRerank()
|
| 43 |
compression_retriever = ContextualCompressionRetriever(
|
| 44 |
base_compressor=compressor, base_retriever=vectorstore_retriever
|
|
|
|
| 38 |
print("query_str: ", query_str)
|
| 39 |
print("model_name: ", self.llm.model_name)
|
| 40 |
#docs = self.index.similarity_search(query_str, k=4)
|
| 41 |
+
vectorstore_retriever = self.index.as_retriever(search_type="similarity", search_kwargs={"k": 10})
|
| 42 |
compressor = CohereRerank()
|
| 43 |
compression_retriever = ContextualCompressionRetriever(
|
| 44 |
base_compressor=compressor, base_retriever=vectorstore_retriever
|