Spaces:
Runtime error
Runtime error
argument 'k' changed to 'limit'
Browse files- 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 |
-
|
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 |
|