From 80e61eb7a11757f4e541ce5ba6033fb11b1b01f0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mirkan=20=C3=87al=C4=B1=C5=9Fkan?=
 <mirkancaliskan@gmail.com>
Date: Sat, 12 Aug 2023 17:02:28 +0400
Subject: [PATCH] fix(stores): VectorStore k variable was ignored (#110)

---
 packages/langchain/lib/src/documents/vector_stores/base.dart | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/langchain/lib/src/documents/vector_stores/base.dart b/packages/langchain/lib/src/documents/vector_stores/base.dart
index 5b90fe6d..52b054bc 100644
--- 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),
     };
   }