From cce2a99ed055e905312e43fb9891bc27656f85b1 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 9 Feb 2026 14:01:22 +0000 Subject: [PATCH] Fix tests --- tests/integration/test_kg_extract_store_integration.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/integration/test_kg_extract_store_integration.py b/tests/integration/test_kg_extract_store_integration.py index dd13789f..7a3af18c 100644 --- a/tests/integration/test_kg_extract_store_integration.py +++ b/tests/integration/test_kg_extract_store_integration.py @@ -405,9 +405,14 @@ class TestKnowledgeGraphPipelineIntegration: collection="test_collection", metadata=[] ), - entities=[] + entities=[ + EntityEmbeddings( + entity=Value(value="http://example.org/entity", is_uri=True), + vectors=[[0.1, 0.2, 0.3]] + ) + ] ) - + mock_msg = MagicMock() mock_msg.value.return_value = sample_embeddings