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
|