mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-09-12 06:59:07 +00:00
ff0a6768c2
* feat(subagents): add capacity controls and durable batches * fix(helm): sync subagent config schema version * fix(subagents): preserve batch history without worker * fix(subagents): support explicit factory runtimes * fix: address durable batch review findings
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from deerflow.config.subagent_batches_config import SubagentBatchesConfig
|
|
from deerflow.config.subagent_runtime_config import SubagentRuntimeConfig
|
|
from deerflow.subagents import SubagentRuntime
|
|
|
|
|
|
def test_runtime_rejects_batch_repository_without_enabled_batch_config() -> None:
|
|
with pytest.raises(ValueError, match="batch_config.enabled"):
|
|
SubagentRuntime(
|
|
SubagentRuntimeConfig(),
|
|
batch_repository=MagicMock(),
|
|
batch_config=SubagentBatchesConfig(enabled=False),
|
|
)
|
|
|
|
|
|
def test_runtime_rejects_batch_repository_without_app_config_snapshot() -> None:
|
|
with pytest.raises(ValueError, match="explicit app_config snapshot"):
|
|
SubagentRuntime(
|
|
SubagentRuntimeConfig(),
|
|
batch_repository=MagicMock(),
|
|
batch_config=SubagentBatchesConfig(enabled=True),
|
|
)
|
|
|
|
|
|
def test_runtime_uses_one_caller_owned_app_config_snapshot() -> None:
|
|
app_config = SimpleNamespace(
|
|
subagent_runtime=SubagentRuntimeConfig(max_running=11),
|
|
subagents=SimpleNamespace(max_total_per_run=14),
|
|
subagent_batches=SubagentBatchesConfig(enabled=False),
|
|
)
|
|
|
|
runtime = SubagentRuntime.from_app_config(app_config)
|
|
|
|
assert runtime.config.max_running == 11
|
|
assert runtime.max_total_per_run == 14
|
|
assert runtime.app_config is app_config
|
|
assert runtime.batch_submitter is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_runtime_owns_batch_worker_lifecycle_and_shared_capacity() -> None:
|
|
service = MagicMock()
|
|
service.start = AsyncMock()
|
|
service.stop = AsyncMock()
|
|
repository = MagicMock()
|
|
app_config = MagicMock()
|
|
|
|
with patch(
|
|
"deerflow.subagents.batch_service.SubagentBatchService",
|
|
return_value=service,
|
|
) as service_type:
|
|
runtime = SubagentRuntime(
|
|
SubagentRuntimeConfig(max_running=9),
|
|
batch_repository=repository,
|
|
batch_config=SubagentBatchesConfig(enabled=True),
|
|
app_config=app_config,
|
|
)
|
|
assert runtime.batch_submitter is None
|
|
|
|
async with runtime:
|
|
assert runtime.batch_submitter is service
|
|
|
|
assert runtime.batch_submitter is None
|
|
|
|
service_type.assert_called_once_with(
|
|
repository=repository,
|
|
config=runtime.batch_config,
|
|
runtime_config=runtime.config,
|
|
app_config=app_config,
|
|
execution_capacity=runtime.execution_capacity,
|
|
)
|
|
service.start.assert_awaited_once_with()
|
|
service.stop.assert_awaited_once_with()
|