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

davidmigloz · Aug 12, 2023 · 80e61eb · 80e61eb
1 parent 5b9bb06
commit 80e61eb
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),
     };
   }