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:
@@ -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 = [
|
||||
|
||||
Reference in New Issue
Block a user