fix(storage): require topK in vector index query

src/storage/src/storage3/_async/vectors.py

@@ -147,7 +147,7 @@ async def list(
     async def query(
         self,
         query_vector: VectorData,
-        topK: int | None = None,
+        topK: int,
         filter: VectorFilter | None = None,
         return_distance: bool = True,
         return_metadata: bool = True,

src/storage/src/storage3/_sync/vectors.py

@@ -139,7 +139,7 @@ def list(
     def query(
         self,
         query_vector: VectorData,
-        topK: int | None = None,
+        topK: int,
         filter: VectorFilter | None = None,
         return_distance: bool = True,
         return_metadata: bool = True,

src/storage/tests/test_vectors.py

@@ -326,6 +326,22 @@ async def test_query_vectors_sends_query_and_parses_distance(use_async: bool) ->
     assert result.vectors[0].distance == 0.25
 
 
+@parametrize_clients
+async def test_query_vectors_requires_top_k(use_async: bool) -> None:
+    recorder = Recorder()
+    with pytest.raises(TypeError, match="topK"):
+        await run(
+            use_async,
+            recorder,
+            lambda c: c.vectors()
+            .from_("embeddings")
+            .index("docs")
+            .query(VectorData(float32=[0.1, 0.2])),
+        )
+
+    assert recorder.requests == []
+
+
 @parametrize_clients
 async def test_put_vectors_omits_missing_metadata(use_async: bool) -> None:
     recorder = Recorder()