hbui commited on
Commit
bc0e176
·
verified ·
1 Parent(s): 319f50d

Update models/langOpen.py

Browse files
Files changed (1) hide show
  1. 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 = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 10})
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