diff --git a/libs/community/langchain_community/vectorstores/kdbai.py b/libs/community/langchain_community/vectorstores/kdbai.py
index d996df12dfe..9b1310293b0 100644
--- a/libs/community/langchain_community/vectorstores/kdbai.py
+++ b/libs/community/langchain_community/vectorstores/kdbai.py
@@ -217,10 +217,12 @@ class KDBAI(VectorStore):
         """
         if "n" in kwargs:
             k = kwargs.pop("n")
-        matches = self._table.search(vectors=[embedding], n=k, filter=filter, **kwargs)[
-            0
-        ]
-        docs = []
+        matches = self._table.search(vectors=[embedding], n=k, filter=filter, **kwargs)
+        docs: list = []
+        if isinstance(matches, list):
+            matches = matches[0]
+        else:
+            return docs
         for row in matches.to_dict(orient="records"):
             text = row.pop("text")
             score = row.pop("__nn_distance")