didericis
74efb1c143
Manual refactor into the rough shape we want/how we want the resolve_plan logic to be consolidated. Needs subsequent fixes.
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
"""Prepare step for the Docker bottle backend.
|
|
|
|
`resolve_plan` does all host-side resolution (image and container
|
|
names, prompt-file, proxy plan, runtime detection) and returns a
|
|
frozen DockerBottlePlan. No Docker resources are created; the only
|
|
side effects are scratch files under `stage_dir` and a probe of
|
|
`docker info`. Cross-backend host-side validation has already run
|
|
via the base class's `prepare` template before this is called.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from . import util as docker_mod
|
|
from .bottle_plan import DockerBottlePlan
|
|
from .. import BottleSpec
|
|
from ...env import ResolvedEnv
|
|
from ...agent_provider import AgentProvisionPlan
|
|
from ...egress import EgressPlan
|
|
from ...supervise import SupervisePlan
|
|
from ...git_gate import GitGatePlan
|
|
|
|
def preflight():
|
|
docker_mod.require_docker()
|
|
|
|
def build_guest_env(resolved_env: ResolvedEnv):
|
|
# resolved = resolve_env(spec.manifest, spec.agent_name)
|
|
# forwarded_env: dict[str, str] = dict(resolved.forwarded)
|
|
return dict(resolved_env.literals)
|
|
|
|
|
|
def resolve_plan(
|
|
spec: BottleSpec,
|
|
slug: str,
|
|
resolved_env: ResolvedEnv,
|
|
instance_name: str,
|
|
agent_image: str,
|
|
agent_dockerfile_path: str,
|
|
prompt_file: Path,
|
|
agent_provision_plan: AgentProvisionPlan,
|
|
egress_plan: EgressPlan,
|
|
supervise_plan: SupervisePlan,
|
|
git_gate_plan: GitGatePlan,
|
|
stage_dir: Path,
|
|
) -> DockerBottlePlan:
|
|
"""Resolve Docker-specific names and write scratch files. Trusts
|
|
that the agent and its skills/git-gate keys are present —
|
|
validation already ran in the base class."""
|
|
|
|
# ==== docker specific setup ====
|
|
use_runsc = docker_mod.runsc_available()
|
|
|
|
return DockerBottlePlan(
|
|
spec=spec,
|
|
stage_dir=stage_dir,
|
|
slug=slug,
|
|
container_name=instance_name,
|
|
# container_name_pinned=container_name_pinned,
|
|
image=agent_image,
|
|
dockerfile_path=agent_dockerfile_path,
|
|
forwarded_env=dict(resolved_env.forwarded),
|
|
prompt_file=prompt_file,
|
|
git_gate_plan=git_gate_plan,
|
|
egress_plan=egress_plan,
|
|
supervise_plan=supervise_plan,
|
|
use_runsc=use_runsc,
|
|
agent_provision=agent_provision_plan,
|
|
# workspace_plan=workspace_plan,
|
|
)
|
|
|
|
|