fix(stores): VectorStore k variable was ignored (davidmigloz#110)

KennethKnudsen97 · Sep 29, 2023 · ae8adf5 · ae8adf5
1 parent 8476eed
commit ae8adf5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/langchain/lib/src/documents/vector_stores/base.dart b/packages/langchain/lib/src/documents/vector_stores/base.dart
@@ -89,8 +89,8 @@ abstract class VectorStore {
     final int k = 4,
   }) {
     return switch (searchType) {
-      VectorStoreSearchType.similarity => similaritySearch(query: query),
-      VectorStoreSearchType.mmr => maxMarginalRelevanceSearch(query: query),
+      VectorStoreSearchType.similarity => similaritySearch(query: query, k: k),
+      VectorStoreSearchType.mmr => maxMarginalRelevanceSearch(query: query, k: k),
     };
   }