apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-26 00:46:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix tests (#571)

Browse source
This commit is contained in:
cybermaggedon 2025-11-28 16:37:01 +00:00 • committed by GitHub
parent e24de6081f
commit 72cb1c98e0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/integration/test_agent_manager_integration.py
View file

@ -182,8 +182,8 @@ Final Answer: Machine learning is a field of AI that enables computers to learn
assert action.observation == "Machine learning is a subset of AI that enables computers to learn from data."
# Verify callbacks were called
think_callback.assert_called_once_with("I need to search for information about machine learning")
observe_callback.assert_called_once_with("Machine learning is a subset of AI that enables computers to learn from data.")
think_callback.assert_called_once_with("I need to search for information about machine learning", is_final=True)
observe_callback.assert_called_once_with("Machine learning is a subset of AI that enables computers to learn from data.", is_final=True)
# Verify tool was executed
graph_rag_client = mock_flow_context("graph-rag-request")
@ -211,7 +211,7 @@ Final Answer: Machine learning is a branch of artificial intelligence."""
assert action.final == "Machine learning is a branch of artificial intelligence."
# Verify only think callback was called (no observation for final answer)
think_callback.assert_called_once_with("I can provide a direct answer")
think_callback.assert_called_once_with("I can provide a direct answer", is_final=True)
observe_callback.assert_not_called()
@pytest.mark.asyncio
@ -457,7 +457,7 @@ Args: {args_json}"""
# Assert
assert isinstance(action, Action)
assert action.observation == expected_response.strip()
observe_callback.assert_called_with(expected_response.strip())
observe_callback.assert_called_with(expected_response.strip(), is_final=True)
# Reset mocks
mock_flow_context("graph-rag-request").reset_mock()