apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-09 21:32:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fix embeddings integration

Browse source
This commit is contained in:
Cyber MacGeddon 2026-03-08 19:38:08 +00:00
parent ea95effbf2
commit 22ba73e680
6 changed files with 31 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/integration/test_graph_rag_integration.py
View file

@ -21,8 +21,12 @@ class TestGraphRagIntegration:
def mock_embeddings_client(self):
"""Mock embeddings client that returns realistic vector embeddings"""
client = AsyncMock()
# New batch format: [[[vectors_for_text1], ...]]
# One text input returns one vector set containing one vector
client.embed.return_value = [
[0.1, 0.2, 0.3, 0.4, 0.5], # Realistic 5-dimensional embedding
[
[0.1, 0.2, 0.3, 0.4, 0.5], # Vector for text
]
]
return client
@ -120,8 +124,8 @@ class TestGraphRagIntegration:
# Assert - Verify service coordination
# 1. Should compute embeddings for query
mock_embeddings_client.embed.assert_called_once_with(query)
# 1. Should compute embeddings for query (now expects list of texts)
mock_embeddings_client.embed.assert_called_once_with([query])
# 2. Should query graph embeddings to find relevant entities
mock_graph_embeddings_client.query.assert_called_once()