fix: Fixed some test suite bugs

Author: deepakdgupta1

tests/unit/api/test_api_endpoints.py

@@ -8,7 +8,7 @@
 
 
 class DummySearchEngine:
-    def search(self, _query, limit=5, expand_deps=True):
+    def search(self, query, limit=5, expand_deps=True, **_kwargs):
         return [CodeChunk(id="c1", entity_id="e1", content="hi", tokens=["hi"])]
 
 

tests/unit/storage/test_vector_store.py

@@ -18,7 +18,8 @@ def test_vector_store_save_load(tmp_path) -> None:
     path = tmp_path / "vectors"
     store.save(path)
 
-    loaded = VectorStore(dimension=2, index_path=path)
+    loaded = VectorStore(dimension=2)
+    loaded.load(path)
     assert loaded.id_map
     results = loaded.search([1.0, 0.0], limit=1)
     assert results[0][0] == "c1"