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 |
|