diff --git a/tests/unit/test_storage/test_graph_embeddings_milvus_storage.py b/tests/unit/test_storage/test_graph_embeddings_milvus_storage.py index c28bead2..8a8e1090 100644 --- a/tests/unit/test_storage/test_graph_embeddings_milvus_storage.py +++ b/tests/unit/test_storage/test_graph_embeddings_milvus_storage.py @@ -136,7 +136,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor: message.metadata.collection = 'test_collection' entity = EntityEmbeddings( - entity=Value(value='', is_uri=False), + entity=Term(type=LITERAL, value=''), vectors=[[0.1, 0.2, 0.3]] ) message.entities = [entity] @@ -155,7 +155,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor: message.metadata.collection = 'test_collection' entity = EntityEmbeddings( - entity=Value(value=None, is_uri=False), + entity=Term(type=LITERAL, value=None), vectors=[[0.1, 0.2, 0.3]] ) message.entities = [entity] @@ -174,15 +174,15 @@ class TestMilvusGraphEmbeddingsStorageProcessor: message.metadata.collection = 'test_collection' valid_entity = EntityEmbeddings( - entity=Value(value='http://example.com/valid', is_uri=True), + entity=Term(type=IRI, iri='http://example.com/valid'), vectors=[[0.1, 0.2, 0.3]] ) empty_entity = EntityEmbeddings( - entity=Value(value='', is_uri=False), + entity=Term(type=LITERAL, value=''), vectors=[[0.4, 0.5, 0.6]] ) none_entity = EntityEmbeddings( - entity=Value(value=None, is_uri=False), + entity=Term(type=LITERAL, value=None), vectors=[[0.7, 0.8, 0.9]] ) message.entities = [valid_entity, empty_entity, none_entity] @@ -217,7 +217,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor: message.metadata.collection = 'test_collection' entity = EntityEmbeddings( - entity=Value(value='http://example.com/entity', is_uri=True), + entity=Term(type=IRI, iri='http://example.com/entity'), vectors=[] ) message.entities = [entity] @@ -236,7 +236,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor: message.metadata.collection = 'test_collection' entity = EntityEmbeddings( - entity=Value(value='http://example.com/entity', is_uri=True), + entity=Term(type=IRI, iri='http://example.com/entity'), vectors=[ [0.1, 0.2], # 2D vector [0.3, 0.4, 0.5, 0.6], # 4D vector @@ -269,11 +269,11 @@ class TestMilvusGraphEmbeddingsStorageProcessor: message.metadata.collection = 'test_collection' uri_entity = EntityEmbeddings( - entity=Value(value='http://example.com/uri_entity', is_uri=True), + entity=Term(type=IRI, iri='http://example.com/uri_entity'), vectors=[[0.1, 0.2, 0.3]] ) literal_entity = EntityEmbeddings( - entity=Value(value='literal entity text', is_uri=False), + entity=Term(type=LITERAL, value='literal entity text'), vectors=[[0.4, 0.5, 0.6]] ) message.entities = [uri_entity, literal_entity]