From 5b8f239ef8a3a1927979b016fbb8bb203041af54 Mon Sep 17 00:00:00 2001 From: AsphodelRem Date: Sat, 9 May 2026 01:07:01 +0700 Subject: [PATCH 1/2] Fix bug with wrong dense embedding type for query --- ragu/graph/graph_retrieve_backend.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ragu/graph/graph_retrieve_backend.py b/ragu/graph/graph_retrieve_backend.py index 6a2ccbc..0dee1e6 100644 --- a/ragu/graph/graph_retrieve_backend.py +++ b/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) From 4158186ec3e7a5cb52d3bee3e84f913ba403ddea Mon Sep 17 00:00:00 2001 From: AsphodelRem Date: Sat, 9 May 2026 01:14:37 +0700 Subject: [PATCH 2/2] Update test --- tests/graph/test_index_crud.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/graph/test_index_crud.py b/tests/graph/test_index_crud.py index 4141a8d..fc71ca1 100644 --- a/tests/graph/test_index_crud.py +++ b/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")