Dev

ragu/graph/graph_retrieve_backend.py

@@ -1,5 +1,6 @@
 from typing import List, Tuple
 
+import numpy as np
 from ragu.chunker.types import Chunk
 from ragu.graph.types import Entity, Relation
 from ragu.models.embedder import Embedder
@@ -174,4 +175,4 @@ async def build_query_vectors(self, query: str) -> Point:
             if len(sparse_vectors) != 1:
                 raise ValueError("Sparse query embedder must return exactly one vector for one query")
             sparse_query = sparse_vectors[0]
-        return Point(dense_embedding=dense_query, sparse_embedding=sparse_query)
+        return Point(dense_embedding=np.array(dense_query), sparse_embedding=sparse_query)

tests/graph/test_index_crud.py

@@ -1,7 +1,7 @@
 """
 Tests for Index batch CRUD operations and invariant validation.
 """
-
+import numpy as np
 import pytest
 from ragu.graph.types import Community, Entity, Relation
 from ragu.chunker.types import Chunk
@@ -150,7 +150,7 @@ async def test_build_query_vectors_returns_dense_and_sparse_payload(
 ):
     point = await sparse_retriever.build_query_vectors("alpha beta")
 
-    assert point.dense_embedding == [0.1] * 128
+    assert np.array_equal(np.array(point.dense_embedding), np.array([0.1] * 128))
     assert point.sparse_embedding is not None
     assert point.sparse_embedding.indices == [9]
     mock_embedder.embed_text.assert_awaited_once_with("alpha beta")