Cover runtime bootstrap source contract
- SHA
d67b8f99f005b49de6752069d8a61f84284f2dc0- Parents
-
3c26874 - Tree
ebfca5e
tenseleyflow/loader
/
d67b8f9
Browse files
6 changed files
| Status | File | + | - |
|---|---|---|---|
| M |
tests/test_response_route_handlers.py
|
1 | 1 |
| M |
tests/test_response_routing.py
|
1 | 1 |
| M |
tests/test_runtime_bootstrap.py
|
21 | 7 |
| M |
tests/test_runtime_context.py
|
5 | 3 |
| M |
tests/test_runtime_launcher.py
|
4 | 1 |
| M |
tests/test_turn_iteration.py
|
1 | 1 |
tests/test_response_route_handlers.pymodified@@ -51,7 +51,7 @@ async def _prepare_context( | ||
| 51 | 51 | task=prepared.task, |
| 52 | 52 | effective_task=prepared.effective_task, |
| 53 | 53 | iterations=1, |
| 54 | - max_iterations=runtime.agent.config.max_iterations, | |
| 54 | + max_iterations=runtime.context.config.max_iterations, | |
| 55 | 55 | actions_taken=[], |
| 56 | 56 | continuation_count=continuation_count, |
| 57 | 57 | consecutive_errors=consecutive_errors, |
tests/test_response_routing.pymodified@@ -53,7 +53,7 @@ async def _prepare_context( | ||
| 53 | 53 | task=prepared.task, |
| 54 | 54 | effective_task=prepared.effective_task, |
| 55 | 55 | iterations=1, |
| 56 | - max_iterations=runtime.agent.config.max_iterations, | |
| 56 | + max_iterations=runtime.context.config.max_iterations, | |
| 57 | 57 | actions_taken=[], |
| 58 | 58 | continuation_count=continuation_count, |
| 59 | 59 | consecutive_errors=consecutive_errors, |
tests/test_runtime_bootstrap.pymodified@@ -5,7 +5,12 @@ from __future__ import annotations | ||
| 5 | 5 | from pathlib import Path |
| 6 | 6 | |
| 7 | 7 | from loader.agent.loop import Agent, AgentConfig |
| 8 | -from loader.runtime.bootstrap import build_runtime_context, sync_runtime_context | |
| 8 | +from loader.runtime.bootstrap import ( | |
| 9 | + RuntimeBootstrapView, | |
| 10 | + build_runtime_bootstrap_source, | |
| 11 | + build_runtime_context, | |
| 12 | + sync_runtime_context, | |
| 13 | +) | |
| 9 | 14 | from loader.runtime.conversation import ConversationRuntime |
| 10 | 15 | from loader.runtime.explore import ExploreRuntime |
| 11 | 16 | from loader.runtime.launcher import RuntimeLauncher, build_runtime_launcher |
@@ -21,8 +26,9 @@ def test_build_runtime_context_uses_shared_bootstrap_contract( | ||
| 21 | 26 | config=AgentConfig(auto_context=False), |
| 22 | 27 | project_root=temp_dir, |
| 23 | 28 | ) |
| 29 | + source = build_runtime_bootstrap_source(agent) | |
| 24 | 30 | |
| 25 | - context = build_runtime_context(agent) | |
| 31 | + context = build_runtime_context(source) | |
| 26 | 32 | |
| 27 | 33 | assert context.project_root == temp_dir.resolve() |
| 28 | 34 | assert context.backend is agent.backend |
@@ -35,6 +41,7 @@ def test_build_runtime_context_uses_shared_bootstrap_contract( | ||
| 35 | 41 | assert context.workflow_mode == agent.workflow_mode |
| 36 | 42 | assert context.prompt_format == agent.prompt_format |
| 37 | 43 | assert context.prompt_sections == agent.prompt_sections |
| 44 | + assert source.metadata == {"owner_type": "Agent"} | |
| 38 | 45 | |
| 39 | 46 | |
| 40 | 47 | def test_sync_runtime_context_refreshes_prompt_and_capability_state( |
@@ -46,7 +53,8 @@ def test_sync_runtime_context_refreshes_prompt_and_capability_state( | ||
| 46 | 53 | config=AgentConfig(auto_context=False), |
| 47 | 54 | project_root=temp_dir, |
| 48 | 55 | ) |
| 49 | - context = build_runtime_context(agent) | |
| 56 | + source = build_runtime_bootstrap_source(agent) | |
| 57 | + context = build_runtime_context(source) | |
| 50 | 58 | |
| 51 | 59 | agent.prompt_format = "native" |
| 52 | 60 | agent.prompt_sections = ["Workflow Context", "Runtime Config"] |
@@ -54,7 +62,7 @@ def test_sync_runtime_context_refreshes_prompt_and_capability_state( | ||
| 54 | 62 | backend._supports_native_tools = False # type: ignore[attr-defined] |
| 55 | 63 | agent.refresh_capability_profile() |
| 56 | 64 | |
| 57 | - sync_runtime_context(context, agent) | |
| 65 | + sync_runtime_context(context, source) | |
| 58 | 66 | |
| 59 | 67 | assert context.workflow_mode == "clarify" |
| 60 | 68 | assert context.prompt_format == "native" |
@@ -83,10 +91,12 @@ def test_conversation_runtime_uses_shared_bootstrap_factory( | ||
| 83 | 91 | fake_build_runtime_context, |
| 84 | 92 | ) |
| 85 | 93 | |
| 86 | - runtime = ConversationRuntime(agent) | |
| 94 | + source = build_runtime_bootstrap_source(agent) | |
| 95 | + runtime = ConversationRuntime(source) | |
| 87 | 96 | |
| 88 | 97 | assert calls == ["conversation"] |
| 89 | 98 | assert runtime.context.project_root == temp_dir.resolve() |
| 99 | + assert runtime.source is source | |
| 90 | 100 | |
| 91 | 101 | |
| 92 | 102 | def test_explore_runtime_uses_shared_bootstrap_factory( |
@@ -110,10 +120,12 @@ def test_explore_runtime_uses_shared_bootstrap_factory( | ||
| 110 | 120 | fake_build_runtime_context, |
| 111 | 121 | ) |
| 112 | 122 | |
| 113 | - runtime = ExploreRuntime(agent) | |
| 123 | + source = build_runtime_bootstrap_source(agent) | |
| 124 | + runtime = ExploreRuntime(source) | |
| 114 | 125 | |
| 115 | 126 | assert calls == ["explore"] |
| 116 | 127 | assert runtime.context.project_root == temp_dir.resolve() |
| 128 | + assert runtime.source is source | |
| 117 | 129 | |
| 118 | 130 | |
| 119 | 131 | def test_build_runtime_launcher_wraps_shared_bootstrap_source( |
@@ -128,4 +140,6 @@ def test_build_runtime_launcher_wraps_shared_bootstrap_source( | ||
| 128 | 140 | launcher = build_runtime_launcher(agent) |
| 129 | 141 | |
| 130 | 142 | assert isinstance(launcher, RuntimeLauncher) |
| 131 | - assert launcher.source is agent | |
| 143 | + assert isinstance(launcher.source, RuntimeBootstrapView) | |
| 144 | + assert launcher.source is not agent | |
| 145 | + assert launcher.source.metadata == {"owner_type": "Agent"} | |
tests/test_runtime_context.pymodified@@ -5,7 +5,7 @@ from __future__ import annotations | ||
| 5 | 5 | from pathlib import Path |
| 6 | 6 | |
| 7 | 7 | from loader.agent.loop import Agent, AgentConfig |
| 8 | -from loader.runtime.bootstrap import build_runtime_context | |
| 8 | +from loader.runtime.bootstrap import build_runtime_bootstrap_source, build_runtime_context | |
| 9 | 9 | from loader.runtime.context import RuntimeContext |
| 10 | 10 | from loader.runtime.recovery import RecoveryContext |
| 11 | 11 | from tests.helpers.runtime_harness import ScriptedBackend |
@@ -18,8 +18,9 @@ def test_agent_builds_typed_runtime_context(temp_dir: Path) -> None: | ||
| 18 | 18 | config=AgentConfig(auto_context=False), |
| 19 | 19 | project_root=temp_dir, |
| 20 | 20 | ) |
| 21 | + source = build_runtime_bootstrap_source(agent) | |
| 21 | 22 | |
| 22 | - context = build_runtime_context(agent) | |
| 23 | + context = build_runtime_context(source) | |
| 23 | 24 | |
| 24 | 25 | assert isinstance(context, RuntimeContext) |
| 25 | 26 | assert context.project_root == temp_dir.resolve() |
@@ -47,8 +48,9 @@ def test_runtime_context_control_callbacks_stay_in_sync(temp_dir: Path) -> None: | ||
| 47 | 48 | config=AgentConfig(auto_context=False), |
| 48 | 49 | project_root=temp_dir, |
| 49 | 50 | ) |
| 51 | + source = build_runtime_bootstrap_source(agent) | |
| 50 | 52 | |
| 51 | - context = build_runtime_context(agent) | |
| 53 | + context = build_runtime_context(source) | |
| 52 | 54 | context.queue_steering_message("Re-check the current task.") |
| 53 | 55 | |
| 54 | 56 | assert context.drain_steering_messages() == ["Re-check the current task."] |
tests/test_runtime_launcher.pymodified@@ -8,6 +8,7 @@ import pytest | ||
| 8 | 8 | |
| 9 | 9 | from loader.agent.loop import Agent, AgentConfig, ReasoningConfig |
| 10 | 10 | from loader.llm.base import CompletionResponse, StreamChunk |
| 11 | +from loader.runtime.bootstrap import RuntimeBootstrapView | |
| 11 | 12 | from loader.runtime.launcher import RuntimeLauncher, build_runtime_launcher |
| 12 | 13 | from tests.helpers.runtime_harness import ScriptedBackend |
| 13 | 14 | |
@@ -24,7 +25,9 @@ def test_build_runtime_launcher_returns_launcher_for_agent_source( | ||
| 24 | 25 | launcher = build_runtime_launcher(agent) |
| 25 | 26 | |
| 26 | 27 | assert isinstance(launcher, RuntimeLauncher) |
| 27 | - assert launcher.source is agent | |
| 28 | + assert isinstance(launcher.source, RuntimeBootstrapView) | |
| 29 | + assert launcher.source is not agent | |
| 30 | + assert launcher.source.metadata == {"owner_type": "Agent"} | |
| 28 | 31 | |
| 29 | 32 | |
| 30 | 33 | @pytest.mark.asyncio |
tests/test_turn_iteration.pymodified@@ -44,7 +44,7 @@ async def _run_iteration( | ||
| 44 | 44 | original_task=original_task, |
| 45 | 45 | effective_max_tokens=prepared.effective_max_tokens, |
| 46 | 46 | iterations=1, |
| 47 | - max_iterations=runtime.agent.config.max_iterations, | |
| 47 | + max_iterations=runtime.context.config.max_iterations, | |
| 48 | 48 | actions_taken=[], |
| 49 | 49 | continuation_count=0, |
| 50 | 50 | empty_retry_count=0, |