Isolate plan and task runtime per session

2026-05-13 20:07:09 +08:00
parent f2989cd48b
commit 0eb57f3f42
8 changed files with 146 additions and 22 deletions
@@ -269,11 +269,17 @@ class AgentPromptingTests(unittest.TestCase):
    def test_prompt_builder_mentions_tasks_when_runtime_is_loaded(self) -> None:
        with tempfile.TemporaryDirectory() as tmp_dir:
            workspace = Path(tmp_dir)
-            runtime = TaskRuntime.from_workspace(workspace)
+            scratchpad = workspace / '.port_sessions' / 'accounts' / 'alice' / 'sessions' / 's1' / 'scratchpad'
+            scratchpad.mkdir(parents=True)
+            runtime = TaskRuntime.from_storage_path(scratchpad / 'task_runtime.json')
            runtime.create_task(title='Inspect runtime tasks')
            runtime_config = AgentRuntimeConfig(cwd=workspace)
            model_config = ModelConfig(model='Qwen/Qwen3-Coder-30B-A3B-Instruct')
-            prompt_context = build_prompt_context(runtime_config, model_config)
+            prompt_context = build_prompt_context(
+                runtime_config,
+                model_config,
+                scratchpad_directory=scratchpad,
+            )
            parts = build_system_prompt_parts(
                prompt_context=prompt_context,
                runtime_config=runtime_config,
@@ -305,14 +311,20 @@ class AgentPromptingTests(unittest.TestCase):
    def test_prompt_builder_mentions_planning_when_runtime_is_loaded(self) -> None:
        with tempfile.TemporaryDirectory() as tmp_dir:
            workspace = Path(tmp_dir)
-            runtime = PlanRuntime.from_workspace(workspace)
+            scratchpad = workspace / '.port_sessions' / 'accounts' / 'alice' / 'sessions' / 's1' / 'scratchpad'
+            scratchpad.mkdir(parents=True)
+            runtime = PlanRuntime.from_storage_path(scratchpad / 'plan_runtime.json')
            runtime.update_plan(
                [{'step': 'Inspect runtime planning', 'status': 'pending'}],
                explanation='Track the current plan.',
            )
            runtime_config = AgentRuntimeConfig(cwd=workspace)
            model_config = ModelConfig(model='Qwen/Qwen3-Coder-30B-A3B-Instruct')
-            prompt_context = build_prompt_context(runtime_config, model_config)
+            prompt_context = build_prompt_context(
+                runtime_config,
+                model_config,
+                scratchpad_directory=scratchpad,
+            )
            parts = build_system_prompt_parts(
                prompt_context=prompt_context,
                runtime_config=runtime_config,