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

Fix tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-11-28 16:36:29 +00:00
parent e24de6081f
commit a9a1da115c
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." assert action.observation == "Machine learning is a subset of AI that enables computers to learn from data."
# Verify callbacks were called # Verify callbacks were called
think_callback.assert_called_once_with("I need to search for information about machine learning") 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.") 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 # Verify tool was executed
graph_rag_client = mock_flow_context("graph-rag-request") 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." assert action.final == "Machine learning is a branch of artificial intelligence."
# Verify only think callback was called (no observation for final answer) # 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() observe_callback.assert_not_called()
@pytest.mark.asyncio @pytest.mark.asyncio
@ -457,7 +457,7 @@ Args: {args_json}"""
# Assert # Assert
assert isinstance(action, Action) assert isinstance(action, Action)
assert action.observation == expected_response.strip() 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 # Reset mocks
mock_flow_context("graph-rag-request").reset_mock() mock_flow_context("graph-rag-request").reset_mock()

4
trustgraph-flow/trustgraph/agent/react/agent_manager.py
View file

@ -346,7 +346,7 @@ class AgentManager:
# In non-streaming mode, send complete thought # In non-streaming mode, send complete thought
# In streaming mode, thoughts were already sent as chunks # In streaming mode, thoughts were already sent as chunks
if not streaming: if not streaming:
await think(act.thought) await think(act.thought, is_final=True)
return act return act
else: else:
@ -354,7 +354,7 @@ class AgentManager:
# In non-streaming mode, send complete thought # In non-streaming mode, send complete thought
# In streaming mode, thoughts were already sent as chunks # In streaming mode, thoughts were already sent as chunks
if not streaming: if not streaming:
await think(act.thought) await think(act.thought, is_final=True)
logger.debug(f"ACTION: {act.name}") logger.debug(f"ACTION: {act.name}")