fix(vector_stores): return None from get() for missing IDs (milvus/weaviate/supabase) (#5562)

Co-authored-by: ly-wang19 <ly-wang19@users.noreply.github.com>
This commit is contained in:
ly-wang19
2026-06-16 11:52:59 +05:30
committed by GitHub
co-authored by ly-wang19
parent b55c51e004
commit bb69b036b5
6 changed files with 29 additions and 15 deletions
+7
View File
@@ -111,6 +111,13 @@ def test_get_vector(supabase_instance, mock_collection):
assert result.payload == {"name": "vector1"}
def test_get_missing_returns_none(supabase_instance, mock_collection):
# An unknown id yields an empty fetch; get() must return None (not []).
mock_collection.fetch.return_value = []
assert supabase_instance.get(vector_id="missing") is None
def test_list_vectors(supabase_instance, mock_collection):
mock_query_results = [("id1", 0.9, {}), ("id2", 0.8, {})]
mock_fetch_results = [("id1", [0.1, 0.2, 0.3], {"name": "vector1"}), ("id2", [0.4, 0.5, 0.6], {"name": "vector2"})]