Update SQL #780

neuml · Sep 4, 2024 · ece3b78 · ece3b78
1 parent ddb6c3e
commit ece3b78
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/python/txtai/ann/sqlite.py b/src/python/txtai/ann/sqlite.py
@@ -245,9 +245,7 @@ def searchsql(self):
             SELECT
         """
 
-        return self.tosql(
-            ("SELECT indexid, 1 - distance " "FROM {table} " f"WHERE embedding MATCH {self.embeddingsql()} " "ORDER BY distance " "LIMIT ?")
-        )
+        return self.tosql(("SELECT indexid, 1 - distance FROM {table} " f"WHERE embedding MATCH {self.embeddingsql()} AND k = ? ORDER BY distance"))
 
     def countsql(self):
         """