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 -1
View File
@@ -189,11 +189,17 @@ class TestMilvusDB:
]
result = milvus_db.get(vector_id)
assert result.id == vector_id
assert result.payload == {"user_id": "alice"}
assert result.score is None
def test_get_missing_returns_none(self, milvus_db, mock_milvus_client):
"""get() must return None (not raise IndexError) for an unknown id."""
mock_milvus_client.get.return_value = []
assert milvus_db.get("missing") is None
def test_list_with_filters(self, milvus_db, mock_milvus_client):
"""Test listing memories with filters."""
mock_milvus_client.query.return_value = [