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

feat: structured source document references in graph-rag responses (#1035)

Browse source
This commit is contained in:
Sunny Yang 2026-07-08 16:59:56 -06:00 • committed by GitHub
parent 0374098ee9
commit e9c6a850ad
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
20 changed files with 568 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/test_graph_rag_integration.py
View file

@ -175,7 +175,7 @@ class TestGraphRagIntegration:
assert mock_prompt_client.prompt.call_count == 2
# Verify final response
response, usage = response
response, usage, sources = response
assert response is not None
assert isinstance(response, str)
assert "machine learning" in response.lower()

8
tests/integration/test_graph_rag_streaming_integration.py
View file

@ -127,7 +127,7 @@ class TestGraphRagStreaming:
)
# Assert
response, usage = response
response, usage, sources = response
assert_streaming_chunks_valid(collector.chunks, min_chunks=1)
assert_callback_invoked(AsyncMock(call_count=len(collector.chunks)), min_calls=1)
@ -175,8 +175,8 @@ class TestGraphRagStreaming:
)
# Assert - Results should be equivalent
non_streaming_text, _ = non_streaming_response
streaming_text, _ = streaming_response
non_streaming_text, _, _ = non_streaming_response
streaming_text, _, _ = streaming_response
assert streaming_text == non_streaming_text
assert len(streaming_chunks) > 0
assert "".join(streaming_chunks) == streaming_text
@ -216,7 +216,7 @@ class TestGraphRagStreaming:
# Assert - Should complete without error
assert response is not None
response_text, usage = response
response_text, usage, sources = response
assert isinstance(response_text, str)
@pytest.mark.asyncio