PlantBasedTen commited on
Commit
9158776
·
verified ·
1 Parent(s): 2e00cd6

argument 'k' changed to 'limit'

Browse files
Files changed (1) hide show
  1. financial_bot/chains.py +1 -1
financial_bot/chains.py CHANGED
@@ -122,7 +122,7 @@ class ContextExtractorChain(Chain):
122
  # (or other time frame).
123
  matches = self.vector_store.search(
124
  query_vector=embeddings,
125
- k=self.top_k,
126
  collection_name=self.vector_collection,
127
  )
128
 
 
122
  # (or other time frame).
123
  matches = self.vector_store.search(
124
  query_vector=embeddings,
125
+ limit=self.top_k,
126
  collection_name=self.vector_collection,
127
  )
128