alex.a.alves/voiceagent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: simplifica o adapter DeepAgents

Browse Source
This commit is contained in:
alex alves
2026-11-05 16:22:23 -03:00
parent fbbe47d90d
commit 85fc4c4d46
3 changed files with 56 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_deep_agent.py
View File

@@ -18,20 +18,20 @@ def test_deep_agent_session_config():
adapter = object.__new__(DeepAgentsAdapter)
assert adapter._config("session-a") == {
assert adapter._session_config("session-a") == {
"configurable": {
"thread_id": "session-a",
}
}
assert adapter._config("session-b") == {
assert adapter._session_config("session-b") == {
"configurable": {
"thread_id": "session-b",
}
}
with pytest.raises(ValueError):
adapter._config("")
adapter._session_config("")
@@ -41,8 +41,8 @@ def test_deep_agent_cleanup_is_idempotent():
from voice_assistant.agent.deep_agent import DeepAgentsAdapter
adapter = object.__new__(DeepAgentsAdapter)
adapter.model = MagicMock()
adapter.model.aclose = AsyncMock()
adapter._model = MagicMock()
adapter._model.aclose = AsyncMock()
adapter._closed = False
async def scenario():
@@ -51,7 +51,7 @@ def test_deep_agent_cleanup_is_idempotent():
asyncio.run(scenario())
adapter.model.aclose.assert_awaited_once()
adapter._model.aclose.assert_awaited_once()
assert adapter._closed is True
@@ -90,7 +90,7 @@ def test_deep_agent_stream_can_be_cancelled():
self.content = content
self.tool_call_chunks = []
adapter.agent = FakeAgent()
adapter._graph = FakeAgent()
async def consume():
async for _ in adapter.stream(
@@ -389,6 +389,6 @@ def test_deep_agent_factory_configures_authentication(monkeypatch, options, expe
model_factory.assert_called_once()
kwargs = model_factory.call_args.kwargs
assert (kwargs["auth_type"], kwargs["auth_profile"], kwargs["auth_file_location"]) == expected_auth
assert adapter.model is model_factory.return_value
assert agent_factory.call_args.kwargs["model"] is adapter.model
assert adapter._model is model_factory.return_value
assert agent_factory.call_args.kwargs["model"] is adapter._model