agent-hitch 0.2.6 → 0.2.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +14 -1
- package/dist/scripts/canary-buildkit-secrets.js +166 -0
- package/dist/scripts/canary-buildkit-secrets.js.map +1 -0
- package/dist/scripts/canary-collision-domains.js +77 -0
- package/dist/scripts/canary-collision-domains.js.map +1 -0
- package/dist/scripts/canary-docker-image-observation.js +181 -0
- package/dist/scripts/canary-docker-image-observation.js.map +1 -0
- package/dist/scripts/canary-gpu-hardware.js +95 -0
- package/dist/scripts/canary-gpu-hardware.js.map +1 -0
- package/dist/scripts/canary-harbor-load.js +365 -0
- package/dist/scripts/canary-harbor-load.js.map +1 -0
- package/dist/scripts/canary-packaged-harness.js +75 -0
- package/dist/scripts/canary-packaged-harness.js.map +1 -0
- package/dist/scripts/canary-resource-load.js +188 -0
- package/dist/scripts/canary-resource-load.js.map +1 -0
- package/dist/scripts/check-architecture.js +8 -4
- package/dist/scripts/check-architecture.js.map +1 -1
- package/dist/scripts/docker-reaper-crash-child.js +21 -0
- package/dist/scripts/docker-reaper-crash-child.js.map +1 -0
- package/dist/src/adapters/catalog.js +1 -0
- package/dist/src/adapters/catalog.js.map +1 -1
- package/dist/src/adapters/discovery.js +1 -1
- package/dist/src/adapters/discovery.js.map +1 -1
- package/dist/src/adapters/providers/claude.js +6 -0
- package/dist/src/adapters/providers/claude.js.map +1 -1
- package/dist/src/adapters/providers/codex.js +6 -0
- package/dist/src/adapters/providers/codex.js.map +1 -1
- package/dist/src/adapters/providers/deepseek.js +6 -0
- package/dist/src/adapters/providers/deepseek.js.map +1 -1
- package/dist/src/adapters/providers/opencode.js +6 -0
- package/dist/src/adapters/providers/opencode.js.map +1 -1
- package/dist/src/adapters/providers/pi.js +6 -0
- package/dist/src/adapters/providers/pi.js.map +1 -1
- package/dist/src/artifacts/index.js +1 -0
- package/dist/src/artifacts/index.js.map +1 -1
- package/dist/src/backends/harbor/backend.js +53 -103
- package/dist/src/backends/harbor/backend.js.map +1 -1
- package/dist/src/backends/harbor/environment-config.js +92 -0
- package/dist/src/backends/harbor/environment-config.js.map +1 -0
- package/dist/src/backends/harbor/index.js +5 -1
- package/dist/src/backends/harbor/index.js.map +1 -1
- package/dist/src/backends/harbor/model-proxy-config.js +23 -0
- package/dist/src/backends/harbor/model-proxy-config.js.map +1 -0
- package/dist/src/backends/harbor/process.js +177 -0
- package/dist/src/backends/harbor/process.js.map +1 -0
- package/dist/src/backends/harbor/runtime-toolchain.js +30 -0
- package/dist/src/backends/harbor/runtime-toolchain.js.map +1 -0
- package/dist/src/backends/harbor/supervisor.js +74 -0
- package/dist/src/backends/harbor/supervisor.js.map +1 -0
- package/dist/src/backends/harbor/task-resources.js +206 -0
- package/dist/src/backends/harbor/task-resources.js.map +1 -0
- package/dist/src/backends/harbor/tools.js +6 -29
- package/dist/src/backends/harbor/tools.js.map +1 -1
- package/dist/src/backends/index.js +4 -1
- package/dist/src/backends/index.js.map +1 -1
- package/dist/src/cli/commands/daemon.js +135 -8
- package/dist/src/cli/commands/daemon.js.map +1 -1
- package/dist/src/cli/commands/eval.js +214 -4
- package/dist/src/cli/commands/eval.js.map +1 -1
- package/dist/src/cli/commands/images.js +40 -0
- package/dist/src/cli/commands/images.js.map +1 -0
- package/dist/src/cli/commands/run.js +6 -3
- package/dist/src/cli/commands/run.js.map +1 -1
- package/dist/src/cli/commands/runs.js +25 -3
- package/dist/src/cli/commands/runs.js.map +1 -1
- package/dist/src/cli/commands/worker.js +79 -0
- package/dist/src/cli/commands/worker.js.map +1 -0
- package/dist/src/cli/main.js +4 -0
- package/dist/src/cli/main.js.map +1 -1
- package/dist/src/cli/output.js +71 -3
- package/dist/src/cli/output.js.map +1 -1
- package/dist/src/control-plane/build-admission.js +69 -0
- package/dist/src/control-plane/build-admission.js.map +1 -0
- package/dist/src/control-plane/build-records.js +6 -0
- package/dist/src/control-plane/build-records.js.map +1 -0
- package/dist/src/control-plane/collisions.js +40 -0
- package/dist/src/control-plane/collisions.js.map +1 -0
- package/dist/src/control-plane/eval-control-state.js +22 -0
- package/dist/src/control-plane/eval-control-state.js.map +1 -0
- package/dist/src/control-plane/eval-control-work.js +39 -0
- package/dist/src/control-plane/eval-control-work.js.map +1 -0
- package/dist/src/control-plane/eval-image-resolution.js +52 -0
- package/dist/src/control-plane/eval-image-resolution.js.map +1 -0
- package/dist/src/control-plane/eval-image-services.js +30 -0
- package/dist/src/control-plane/eval-image-services.js.map +1 -0
- package/dist/src/control-plane/eval-records.js +252 -0
- package/dist/src/control-plane/eval-records.js.map +1 -0
- package/dist/src/control-plane/eval-recovery.js +181 -0
- package/dist/src/control-plane/eval-recovery.js.map +1 -0
- package/dist/src/control-plane/eval-scheduler.js +447 -0
- package/dist/src/control-plane/eval-scheduler.js.map +1 -0
- package/dist/src/control-plane/index.js +20 -0
- package/dist/src/control-plane/index.js.map +1 -0
- package/dist/src/control-plane/local-worker.js +49 -0
- package/dist/src/control-plane/local-worker.js.map +1 -0
- package/dist/src/control-plane/model-capture-planning.js +12 -0
- package/dist/src/control-plane/model-capture-planning.js.map +1 -0
- package/dist/src/control-plane/remote-result-transport.js +172 -0
- package/dist/src/control-plane/remote-result-transport.js.map +1 -0
- package/dist/src/control-plane/remote-work-coordinator.js +332 -0
- package/dist/src/control-plane/remote-work-coordinator.js.map +1 -0
- package/dist/src/control-plane/remote-work-inputs.js +266 -0
- package/dist/src/control-plane/remote-work-inputs.js.map +1 -0
- package/dist/src/control-plane/remote-work-recovery.js +181 -0
- package/dist/src/control-plane/remote-work-recovery.js.map +1 -0
- package/dist/src/control-plane/remote-worker-artifacts.js +135 -0
- package/dist/src/control-plane/remote-worker-artifacts.js.map +1 -0
- package/dist/src/control-plane/remote-worker-client.js +202 -0
- package/dist/src/control-plane/remote-worker-client.js.map +1 -0
- package/dist/src/control-plane/remote-worker-configuration.js +34 -0
- package/dist/src/control-plane/remote-worker-configuration.js.map +1 -0
- package/dist/src/control-plane/remote-worker-credentials.js +58 -0
- package/dist/src/control-plane/remote-worker-credentials.js.map +1 -0
- package/dist/src/control-plane/remote-worker-protocol.js +510 -0
- package/dist/src/control-plane/remote-worker-protocol.js.map +1 -0
- package/dist/src/control-plane/remote-worker-runner.js +214 -0
- package/dist/src/control-plane/remote-worker-runner.js.map +1 -0
- package/dist/src/control-plane/remote-workers.js +377 -0
- package/dist/src/control-plane/remote-workers.js.map +1 -0
- package/dist/src/control-plane/rerun-scheduler.js +411 -0
- package/dist/src/control-plane/rerun-scheduler.js.map +1 -0
- package/dist/src/control-plane/rerun-submission.js +56 -0
- package/dist/src/control-plane/rerun-submission.js.map +1 -0
- package/dist/src/control-plane/resources.js +182 -0
- package/dist/src/control-plane/resources.js.map +1 -0
- package/dist/src/control-plane/scheduler-eval-entry.js +32 -0
- package/dist/src/control-plane/scheduler-eval-entry.js.map +1 -0
- package/dist/src/control-plane/synthetic-result.js +21 -0
- package/dist/src/control-plane/synthetic-result.js.map +1 -0
- package/dist/src/control-plane/work-admission.js +21 -0
- package/dist/src/control-plane/work-admission.js.map +1 -0
- package/dist/src/control-plane/work-dispatcher.js +229 -0
- package/dist/src/control-plane/work-dispatcher.js.map +1 -0
- package/dist/src/controller-runtime/hash.js +12 -1
- package/dist/src/controller-runtime/hash.js.map +1 -1
- package/dist/src/controller-runtime/index.js +1 -1
- package/dist/src/controller-runtime/index.js.map +1 -1
- package/dist/src/controller-runtime/store.js +15 -6
- package/dist/src/controller-runtime/store.js.map +1 -1
- package/dist/src/daemon/client.js.map +1 -1
- package/dist/src/daemon/event-stream.js +46 -0
- package/dist/src/daemon/event-stream.js.map +1 -0
- package/dist/src/daemon/health.js +66 -0
- package/dist/src/daemon/health.js.map +1 -0
- package/dist/src/daemon/launcher.js +28 -1
- package/dist/src/daemon/launcher.js.map +1 -1
- package/dist/src/daemon/logging.js +23 -0
- package/dist/src/daemon/logging.js.map +1 -0
- package/dist/src/daemon/scheduler.js +26 -5
- package/dist/src/daemon/scheduler.js.map +1 -1
- package/dist/src/daemon/server.js +255 -98
- package/dist/src/daemon/server.js.map +1 -1
- package/dist/src/daemon/telemetry.js +163 -0
- package/dist/src/daemon/telemetry.js.map +1 -0
- package/dist/src/daemon/worker-routes.js +219 -0
- package/dist/src/daemon/worker-routes.js.map +1 -0
- package/dist/src/domain/bundles.js +2 -0
- package/dist/src/domain/bundles.js.map +1 -0
- package/dist/src/domain/control-plane.js +2 -0
- package/dist/src/domain/control-plane.js.map +1 -0
- package/dist/src/domain/execution-evidence.js +2 -0
- package/dist/src/domain/execution-evidence.js.map +1 -0
- package/dist/src/domain/execution-plan.js +2 -0
- package/dist/src/domain/execution-plan.js.map +1 -0
- package/dist/src/domain/images.js +2 -0
- package/dist/src/domain/images.js.map +1 -0
- package/dist/src/domain/index.js +9 -0
- package/dist/src/domain/index.js.map +1 -1
- package/dist/src/domain/interactions.js +2 -0
- package/dist/src/domain/interactions.js.map +1 -0
- package/dist/src/domain/providers.js +2 -0
- package/dist/src/domain/providers.js.map +1 -0
- package/dist/src/domain/resources.js +2 -0
- package/dist/src/domain/resources.js.map +1 -0
- package/dist/src/domain/workers.js +2 -0
- package/dist/src/domain/workers.js.map +1 -0
- package/dist/src/evals/collect-only-rerun.js +182 -0
- package/dist/src/evals/collect-only-rerun.js.map +1 -0
- package/dist/src/evals/directory.js +30 -0
- package/dist/src/evals/directory.js.map +1 -0
- package/dist/src/evals/docker-engine-stats.js +89 -0
- package/dist/src/evals/docker-engine-stats.js.map +1 -0
- package/dist/src/evals/docker-ownership.js +51 -0
- package/dist/src/evals/docker-ownership.js.map +1 -0
- package/dist/src/evals/docker-reaper.js +144 -0
- package/dist/src/evals/docker-reaper.js.map +1 -0
- package/dist/src/evals/docker-resource-observer.js +207 -0
- package/dist/src/evals/docker-resource-observer.js.map +1 -0
- package/dist/src/evals/environment-image-planning.js +164 -0
- package/dist/src/evals/environment-image-planning.js.map +1 -0
- package/dist/src/evals/eval-artifact-planning.js +56 -0
- package/dist/src/evals/eval-artifact-planning.js.map +1 -0
- package/dist/src/evals/eval-finalization.js +10 -0
- package/dist/src/evals/eval-finalization.js.map +1 -0
- package/dist/src/evals/eval-lifecycle-events.js +16 -0
- package/dist/src/evals/eval-lifecycle-events.js.map +1 -0
- package/dist/src/evals/events.js +53 -5
- package/dist/src/evals/events.js.map +1 -1
- package/dist/src/evals/execution-evidence.js +116 -0
- package/dist/src/evals/execution-evidence.js.map +1 -0
- package/dist/src/evals/execution-leases.js +342 -0
- package/dist/src/evals/execution-leases.js.map +1 -0
- package/dist/src/evals/execution-plan-artifacts.js +72 -0
- package/dist/src/evals/execution-plan-artifacts.js.map +1 -0
- package/dist/src/evals/execution-plan-images.js +69 -0
- package/dist/src/evals/execution-plan-images.js.map +1 -0
- package/dist/src/evals/execution-plan-resources.js +174 -0
- package/dist/src/evals/execution-plan-resources.js.map +1 -0
- package/dist/src/evals/execution-plan.js +415 -0
- package/dist/src/evals/execution-plan.js.map +1 -0
- package/dist/src/evals/harbor-artifact-builder-image.js +99 -0
- package/dist/src/evals/harbor-artifact-builder-image.js.map +1 -0
- package/dist/src/evals/harbor-artifact-builder.js +354 -0
- package/dist/src/evals/harbor-artifact-builder.js.map +1 -0
- package/dist/src/evals/harbor-bridge-error.js +18 -3
- package/dist/src/evals/harbor-bridge-error.js.map +1 -1
- package/dist/src/evals/harbor-phase-timings.js +60 -0
- package/dist/src/evals/harbor-phase-timings.js.map +1 -0
- package/dist/src/evals/index.js +25 -2
- package/dist/src/evals/index.js.map +1 -1
- package/dist/src/evals/infrastructure-retry.js +100 -63
- package/dist/src/evals/infrastructure-retry.js.map +1 -1
- package/dist/src/evals/interaction-capture-import.js +49 -0
- package/dist/src/evals/interaction-capture-import.js.map +1 -0
- package/dist/src/evals/local-docker-provider.js +372 -0
- package/dist/src/evals/local-docker-provider.js.map +1 -0
- package/dist/src/evals/local-eval-planning.js +80 -0
- package/dist/src/evals/local-eval-planning.js.map +1 -0
- package/dist/src/evals/model-capture-plan.js +22 -0
- package/dist/src/evals/model-capture-plan.js.map +1 -0
- package/dist/src/evals/model-capture-runtime.js +114 -0
- package/dist/src/evals/model-capture-runtime.js.map +1 -0
- package/dist/src/evals/model-proxy-runtime-state.js +53 -0
- package/dist/src/evals/model-proxy-runtime-state.js.map +1 -0
- package/dist/src/evals/planned-execution-support.js +54 -0
- package/dist/src/evals/planned-execution-support.js.map +1 -0
- package/dist/src/evals/planned-execution.js +461 -0
- package/dist/src/evals/planned-execution.js.map +1 -0
- package/dist/src/evals/planned-remote-execution.js +40 -0
- package/dist/src/evals/planned-remote-execution.js.map +1 -0
- package/dist/src/evals/planned-retry-lifecycle.js +186 -0
- package/dist/src/evals/planned-retry-lifecycle.js.map +1 -0
- package/dist/src/evals/prepared-harness.js +76 -0
- package/dist/src/evals/prepared-harness.js.map +1 -0
- package/dist/src/evals/progress.js +4 -4
- package/dist/src/evals/progress.js.map +1 -1
- package/dist/src/evals/records.js +20 -3
- package/dist/src/evals/records.js.map +1 -1
- package/dist/src/evals/recovery.js +192 -0
- package/dist/src/evals/recovery.js.map +1 -0
- package/dist/src/evals/remote-infrastructure-retry.js +99 -0
- package/dist/src/evals/remote-infrastructure-retry.js.map +1 -0
- package/dist/src/evals/rerun-inputs.js +89 -0
- package/dist/src/evals/rerun-inputs.js.map +1 -0
- package/dist/src/evals/rerun-slots.js +4 -3
- package/dist/src/evals/rerun-slots.js.map +1 -1
- package/dist/src/evals/rerun-types.js +49 -0
- package/dist/src/evals/rerun-types.js.map +1 -0
- package/dist/src/evals/rerun.js +113 -75
- package/dist/src/evals/rerun.js.map +1 -1
- package/dist/src/evals/result-helpers.js +65 -0
- package/dist/src/evals/result-helpers.js.map +1 -0
- package/dist/src/evals/resume-state.js +36 -0
- package/dist/src/evals/resume-state.js.map +1 -0
- package/dist/src/evals/service-types.js +2 -0
- package/dist/src/evals/service-types.js.map +1 -0
- package/dist/src/evals/service.js +292 -257
- package/dist/src/evals/service.js.map +1 -1
- package/dist/src/evals/task-resources.js +209 -0
- package/dist/src/evals/task-resources.js.map +1 -0
- package/dist/src/evals/trial-environment-evidence.js +87 -0
- package/dist/src/evals/trial-environment-evidence.js.map +1 -0
- package/dist/src/evals/trial-execution-evidence.js +13 -0
- package/dist/src/evals/trial-execution-evidence.js.map +1 -0
- package/dist/src/evals/trial-import-identity.js +31 -0
- package/dist/src/evals/trial-import-identity.js.map +1 -0
- package/dist/src/evals/trial-import.js +143 -145
- package/dist/src/evals/trial-import.js.map +1 -1
- package/dist/src/evals/trial-publication-recovery.js +104 -0
- package/dist/src/evals/trial-publication-recovery.js.map +1 -0
- package/dist/src/evals/trial-publication.js +25 -0
- package/dist/src/evals/trial-publication.js.map +1 -0
- package/dist/src/evals/work-item-artifacts.js +23 -0
- package/dist/src/evals/work-item-artifacts.js.map +1 -0
- package/dist/src/foundation/config.js +17 -0
- package/dist/src/foundation/config.js.map +1 -1
- package/dist/src/foundation/credential-redaction.js +137 -0
- package/dist/src/foundation/credential-redaction.js.map +1 -0
- package/dist/src/foundation/environment-image-references.js +59 -0
- package/dist/src/foundation/environment-image-references.js.map +1 -0
- package/dist/src/foundation/executable.js +45 -5
- package/dist/src/foundation/executable.js.map +1 -1
- package/dist/src/foundation/index.js +5 -2
- package/dist/src/foundation/index.js.map +1 -1
- package/dist/src/foundation/process-identity.js +69 -0
- package/dist/src/foundation/process-identity.js.map +1 -0
- package/dist/src/images/context.js +47 -0
- package/dist/src/images/context.js.map +1 -0
- package/dist/src/images/docker-buildkit.js +137 -0
- package/dist/src/images/docker-buildkit.js.map +1 -0
- package/dist/src/images/dockerfile.js +67 -0
- package/dist/src/images/dockerfile.js.map +1 -0
- package/dist/src/images/gc.js +261 -0
- package/dist/src/images/gc.js.map +1 -0
- package/dist/src/images/index.js +10 -0
- package/dist/src/images/index.js.map +1 -0
- package/dist/src/images/manifest.js +109 -0
- package/dist/src/images/manifest.js.map +1 -0
- package/dist/src/images/ownership.js +5 -0
- package/dist/src/images/ownership.js.map +1 -0
- package/dist/src/images/records.js +32 -0
- package/dist/src/images/records.js.map +1 -0
- package/dist/src/images/registry.js +154 -0
- package/dist/src/images/registry.js.map +1 -0
- package/dist/src/images/service.js +224 -0
- package/dist/src/images/service.js.map +1 -0
- package/dist/src/model-access/capture.js +255 -0
- package/dist/src/model-access/capture.js.map +1 -0
- package/dist/src/model-access/index.js +5 -0
- package/dist/src/model-access/index.js.map +1 -0
- package/dist/src/model-access/policy.js +49 -0
- package/dist/src/model-access/policy.js.map +1 -0
- package/dist/src/model-access/proxy.js +336 -0
- package/dist/src/model-access/proxy.js.map +1 -0
- package/dist/src/model-access/records.js +167 -0
- package/dist/src/model-access/records.js.map +1 -0
- package/dist/src/runs/bundle.js +450 -0
- package/dist/src/runs/bundle.js.map +1 -0
- package/dist/src/runs/events.js +12 -1
- package/dist/src/runs/events.js.map +1 -1
- package/dist/src/runs/executor.js +24 -17
- package/dist/src/runs/executor.js.map +1 -1
- package/dist/src/runs/index.js +3 -1
- package/dist/src/runs/index.js.map +1 -1
- package/dist/src/runs/manifest.js +5 -5
- package/dist/src/runs/manifest.js.map +1 -1
- package/dist/src/runs/request.js +6 -1
- package/dist/src/runs/request.js.map +1 -1
- package/dist/src/runs/training-candidate.js +161 -0
- package/dist/src/runs/training-candidate.js.map +1 -0
- package/dist/src/trajectories/provider-capture.js +15 -24
- package/dist/src/trajectories/provider-capture.js.map +1 -1
- package/dist/src/trajectories/providers/deepseek.js +3 -3
- package/dist/src/trajectories/providers/deepseek.js.map +1 -1
- package/dist/src/workers/index.js +3 -0
- package/dist/src/workers/index.js.map +1 -0
- package/dist/src/workers/remote-harbor-work-spec.js +97 -0
- package/dist/src/workers/remote-harbor-work-spec.js.map +1 -0
- package/dist/src/workers/remote-harbor-worker.js +220 -0
- package/dist/src/workers/remote-harbor-worker.js.map +1 -0
- package/docs/schemas/environment-build-index.schema.json +13 -0
- package/docs/schemas/environment-build-record.schema.json +25 -0
- package/docs/schemas/environment-image-manifest.schema.json +62 -0
- package/docs/schemas/eval-control.schema.json +36 -0
- package/docs/schemas/eval-execution-plan.schema.json +216 -0
- package/docs/schemas/eval-rerun-result.schema.json +53 -0
- package/docs/schemas/eval-rerun-state.schema.json +61 -0
- package/docs/schemas/eval-rerun-submission.schema.json +58 -0
- package/docs/schemas/eval-result.schema.json +34 -1
- package/docs/schemas/eval-submission-input.schema.json +14 -0
- package/docs/schemas/eval-submission.schema.json +68 -0
- package/docs/schemas/eval-trial-publication.schema.json +39 -0
- package/docs/schemas/execution-evidence.schema.json +96 -0
- package/docs/schemas/execution-lease.schema.json +50 -0
- package/docs/schemas/execution-provider-status.schema.json +65 -0
- package/docs/schemas/execution-worker.schema.json +53 -0
- package/docs/schemas/harbor-process-exit.schema.json +14 -0
- package/docs/schemas/interaction-capture-ref.schema.json +45 -0
- package/docs/schemas/local-provider-execution.schema.json +33 -0
- package/docs/schemas/model-interaction.schema.json +41 -0
- package/docs/schemas/remote-result-envelope.schema.json +32 -0
- package/docs/schemas/remote-tree-envelope.schema.json +25 -0
- package/docs/schemas/remote-work-offer.schema.json +58 -0
- package/docs/schemas/remote-work-receipt.schema.json +50 -0
- package/docs/schemas/remote-worker-artifact.schema.json +17 -0
- package/docs/schemas/remote-worker-event.schema.json +18 -0
- package/docs/schemas/remote-worker-heartbeat.schema.json +37 -0
- package/docs/schemas/remote-worker-registration.schema.json +52 -0
- package/docs/schemas/result-bundle-index.schema.json +133 -0
- package/docs/schemas/run-environment-images.schema.json +39 -0
- package/docs/schemas/training-data-candidate.schema.json +46 -0
- package/integrations/harbor/hitch_harbor_agent.py +171 -539
- package/integrations/harbor/hitch_harbor_environment.py +331 -0
- package/integrations/harbor/hitch_harbor_task_resources.py +275 -0
- package/integrations/harbor/hitch_harbor_verifier.py +23 -0
- package/package.json +14 -1
|
@@ -0,0 +1,182 @@
|
|
|
1
|
+
import { readdir } from "node:fs/promises";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { readHarborRawResult } from "../backends/index.js";
|
|
4
|
+
import { HitchError, SCHEMA_VERSION, atomicWriteJSON, readJSON } from "../foundation/index.js";
|
|
5
|
+
import { summarizeTrialRefs } from "./result-helpers.js";
|
|
6
|
+
import { parseEvalExecutionPlan } from "./execution-plan.js";
|
|
7
|
+
import { replaceInvalidEvalProgressTrial, writeEvalProgress } from "./progress.js";
|
|
8
|
+
import { evalRerunSemantics } from "./rerun-types.js";
|
|
9
|
+
import { formatSlot, invalidTrialSlots, slotKey, sortSlots, uniqueTasks } from "./rerun-slots.js";
|
|
10
|
+
import { importEvalTrialRun, TrialBundlePendingError, validateEvalTrialReferences } from "./trial-import.js";
|
|
11
|
+
export async function collectOnlyEvalRerun(input) {
|
|
12
|
+
const executionPlan = parseEvalExecutionPlan(await readJSON(path.join(input.evalDirectory, "execution-plan.json")));
|
|
13
|
+
if (executionPlan.eval_id !== input.evalId || executionPlan.membership !== "known")
|
|
14
|
+
throw unavailable("collect-only requires a known-task execution plan");
|
|
15
|
+
const resolution = await readJSON(path.join(input.evalDirectory, "resolution.json"));
|
|
16
|
+
if (resolution.identity !== requiredString(input.plan.candidate.revision_identity, "candidate revision identity"))
|
|
17
|
+
throw unavailable("collect-only resolution identity changed");
|
|
18
|
+
const runtimeId = requiredString(input.plan.controllerRuntime.runtime_id, "controller runtime id");
|
|
19
|
+
let progress = input.progress;
|
|
20
|
+
const baseRefs = progress.trials.filter((ref) => !input.selectedTrials.some((slot) => slotKey(slot) === slotKey(ref)));
|
|
21
|
+
const collectedRefs = [];
|
|
22
|
+
const sources = [];
|
|
23
|
+
for (const slot of input.selectedTrials) {
|
|
24
|
+
const collected = await collectSlot(input, executionPlan, resolution, runtimeId, slot, [...baseRefs, ...collectedRefs]);
|
|
25
|
+
collectedRefs.push(collected.ref);
|
|
26
|
+
sources.push(collected.source);
|
|
27
|
+
}
|
|
28
|
+
for (const ref of collectedRefs)
|
|
29
|
+
progress = replaceInvalidEvalProgressTrial(progress, ref);
|
|
30
|
+
if (progress.generation !== input.progress.generation)
|
|
31
|
+
await writeEvalProgress(input.evalDirectory, progress);
|
|
32
|
+
const remainingTrials = invalidTrialSlots(input.plan.tasks, input.plan.attempts, progress);
|
|
33
|
+
const remainingTasks = uniqueTasks(remainingTrials);
|
|
34
|
+
const repairedTrials = sortSlots(collectedRefs.map((ref) => ({ task_id: ref.task_id, attempt: ref.attempt })));
|
|
35
|
+
const repairedTasks = uniqueTasks(repairedTrials);
|
|
36
|
+
const succeeded = remainingTrials.length === 0;
|
|
37
|
+
const completedAt = new Date().toISOString();
|
|
38
|
+
await writeEvalResult(input, progress, remainingTrials, succeeded, completedAt);
|
|
39
|
+
await atomicWriteJSON(path.join(input.rerunDirectory, "state.json"), {
|
|
40
|
+
schema_version: SCHEMA_VERSION,
|
|
41
|
+
rerun_id: input.rerunId,
|
|
42
|
+
eval_id: input.evalId,
|
|
43
|
+
rerun_type: "collect-only",
|
|
44
|
+
semantics: evalRerunSemantics("collect-only"),
|
|
45
|
+
status: "completed",
|
|
46
|
+
tasks: uniqueTasks(input.selectedTrials),
|
|
47
|
+
trials: sortSlots(input.selectedTrials),
|
|
48
|
+
repaired_tasks: repairedTasks,
|
|
49
|
+
repaired_trials: repairedTrials,
|
|
50
|
+
remaining_invalid_tasks: remainingTasks,
|
|
51
|
+
remaining_invalid_trials: remainingTrials,
|
|
52
|
+
sources,
|
|
53
|
+
eval_status: succeeded ? "succeeded" : "failed",
|
|
54
|
+
started_at: input.startedAt,
|
|
55
|
+
completed_at: completedAt,
|
|
56
|
+
updated_at: completedAt,
|
|
57
|
+
});
|
|
58
|
+
return {
|
|
59
|
+
schema_version: "1",
|
|
60
|
+
kind: "eval-rerun",
|
|
61
|
+
rerun_id: input.rerunId,
|
|
62
|
+
rerun_type: "collect-only",
|
|
63
|
+
semantics: evalRerunSemantics("collect-only"),
|
|
64
|
+
eval_id: input.evalId,
|
|
65
|
+
status: "completed",
|
|
66
|
+
selected_tasks: uniqueTasks(input.selectedTrials),
|
|
67
|
+
repaired_tasks: repairedTasks,
|
|
68
|
+
remaining_invalid_tasks: remainingTasks,
|
|
69
|
+
selected_trials: sortSlots(input.selectedTrials),
|
|
70
|
+
repaired_trials: repairedTrials,
|
|
71
|
+
remaining_invalid_trials: remainingTrials,
|
|
72
|
+
sources,
|
|
73
|
+
eval_status: succeeded ? "succeeded" : "failed",
|
|
74
|
+
started_at: input.startedAt,
|
|
75
|
+
completed_at: completedAt,
|
|
76
|
+
};
|
|
77
|
+
}
|
|
78
|
+
async function collectSlot(input, executionPlan, resolution, runtimeId, slot, existingRefs) {
|
|
79
|
+
const plannedSlot = executionPlan.slots.find((entry) => entry.task_id === slot.task_id && entry.attempt === slot.attempt);
|
|
80
|
+
const work = plannedSlot && executionPlan.work_items.find((entry) => entry.slots.includes(plannedSlot.slot_id));
|
|
81
|
+
if (!plannedSlot || !work || work.task_ids.length !== 1 || work.task_ids[0] !== slot.task_id) {
|
|
82
|
+
throw unavailable(`collect-only has no isolated work item for ${formatSlot(slot)}`);
|
|
83
|
+
}
|
|
84
|
+
const workDirectory = path.join(input.evalDirectory, "harbor", "work-items", work.work_id);
|
|
85
|
+
let entries;
|
|
86
|
+
try {
|
|
87
|
+
entries = await readdir(workDirectory, { withFileTypes: true });
|
|
88
|
+
}
|
|
89
|
+
catch (error) {
|
|
90
|
+
if (error.code === "ENOENT")
|
|
91
|
+
throw unavailable(`collect-only source is missing for ${formatSlot(slot)}`);
|
|
92
|
+
throw error;
|
|
93
|
+
}
|
|
94
|
+
const epochs = entries.filter((entry) => entry.isDirectory() && /^epoch-[0-9]{6}$/.test(entry.name))
|
|
95
|
+
.map((entry) => entry.name).sort().reverse();
|
|
96
|
+
for (const epoch of epochs) {
|
|
97
|
+
const relativeBackend = path.join("harbor", "work-items", work.work_id, epoch);
|
|
98
|
+
const jobDirectory = path.join(input.evalDirectory, relativeBackend, "job");
|
|
99
|
+
const raw = await readHarborRawResult(jobDirectory);
|
|
100
|
+
const trials = Array.isArray(raw?.trial_results) ? raw.trial_results : [];
|
|
101
|
+
if (!raw || trials.length === 0)
|
|
102
|
+
continue;
|
|
103
|
+
const imported = [...existingRefs];
|
|
104
|
+
for (const [index, trial] of trials.entries()) {
|
|
105
|
+
try {
|
|
106
|
+
const ref = await importEvalTrialRun({
|
|
107
|
+
root: input.root,
|
|
108
|
+
evalId: input.evalId,
|
|
109
|
+
evalDirectory: input.evalDirectory,
|
|
110
|
+
harborJobDirectory: jobDirectory,
|
|
111
|
+
expectedAttempt: slot.attempt,
|
|
112
|
+
request: input.request,
|
|
113
|
+
resolvedRevision: resolution,
|
|
114
|
+
benchmarkId: input.request.benchmark_id,
|
|
115
|
+
benchmarkRevision: input.request.benchmark_revision,
|
|
116
|
+
publicationMode: "replace-invalid",
|
|
117
|
+
runtimeId,
|
|
118
|
+
env: input.env ?? process.env,
|
|
119
|
+
requireCompleteMarker: true,
|
|
120
|
+
}, trial, index, imported);
|
|
121
|
+
if (!imported.some((entry) => entry.trial_id === ref.trial_id))
|
|
122
|
+
imported.push(ref);
|
|
123
|
+
}
|
|
124
|
+
catch (error) {
|
|
125
|
+
if (error instanceof TrialBundlePendingError)
|
|
126
|
+
continue;
|
|
127
|
+
throw error;
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
const ref = imported.find((entry) => entry.task_id === slot.task_id && entry.attempt === slot.attempt && entry.observation_status === "valid");
|
|
131
|
+
if (!ref)
|
|
132
|
+
continue;
|
|
133
|
+
await validateEvalTrialReferences(input.root, input.evalId, [ref], {
|
|
134
|
+
benchmarkId: input.request.benchmark_id,
|
|
135
|
+
benchmarkRevision: input.request.benchmark_revision,
|
|
136
|
+
});
|
|
137
|
+
return {
|
|
138
|
+
ref,
|
|
139
|
+
source: {
|
|
140
|
+
source_trial_id: ref.trial_id,
|
|
141
|
+
source_run_id: ref.run_id,
|
|
142
|
+
source_work_id: work.work_id,
|
|
143
|
+
source_backend_directory: relativeBackend.split(path.sep).join("/"),
|
|
144
|
+
},
|
|
145
|
+
};
|
|
146
|
+
}
|
|
147
|
+
throw new HitchError(`collect-only found no complete valid result for ${formatSlot(slot)}`, {
|
|
148
|
+
code: "eval_collect_only_result_unavailable",
|
|
149
|
+
exitCode: 12,
|
|
150
|
+
});
|
|
151
|
+
}
|
|
152
|
+
async function writeEvalResult(input, progress, remainingTrials, succeeded, completedAt) {
|
|
153
|
+
const result = {
|
|
154
|
+
...(input.previousResult ?? {}),
|
|
155
|
+
schema_version: SCHEMA_VERSION,
|
|
156
|
+
eval_id: input.evalId,
|
|
157
|
+
status: succeeded ? "succeeded" : "failed",
|
|
158
|
+
exit_code: succeeded ? 0 : 13,
|
|
159
|
+
candidate: input.plan.candidate,
|
|
160
|
+
dataset: input.request.dataset,
|
|
161
|
+
benchmark_id: input.request.benchmark_id,
|
|
162
|
+
benchmark_revision: input.request.benchmark_revision,
|
|
163
|
+
generation: progress.generation,
|
|
164
|
+
trials: progress.trials,
|
|
165
|
+
summary: summarizeTrialRefs(progress.trials),
|
|
166
|
+
...(succeeded ? {} : { error: { code: "eval_has_invalid_tasks", message: `eval has invalid or missing trials: ${remainingTrials.map(formatSlot).join(", ")}` } }),
|
|
167
|
+
started_at: typeof input.previousResult?.started_at === "string" ? input.previousResult.started_at : progress.started_at,
|
|
168
|
+
completed_at: completedAt,
|
|
169
|
+
};
|
|
170
|
+
if (succeeded)
|
|
171
|
+
delete result.error;
|
|
172
|
+
await atomicWriteJSON(path.join(input.evalDirectory, "result.json"), result);
|
|
173
|
+
}
|
|
174
|
+
function requiredString(value, label) {
|
|
175
|
+
if (typeof value !== "string" || !value)
|
|
176
|
+
throw unavailable(`${label} is missing`);
|
|
177
|
+
return value;
|
|
178
|
+
}
|
|
179
|
+
function unavailable(message) {
|
|
180
|
+
return new HitchError(message, { code: "eval_collect_only_unavailable", exitCode: 2 });
|
|
181
|
+
}
|
|
182
|
+
//# sourceMappingURL=collect-only-rerun.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"collect-only-rerun.js","sourceRoot":"","sources":["../../../src/evals/collect-only-rerun.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAE3D,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAC/F,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,+BAA+B,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AACnF,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAEtD,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAElG,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAC;AA6B7G,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,KAA4B;IACrE,MAAM,aAAa,GAAG,sBAAsB,CAAC,MAAM,QAAQ,CAAU,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7H,IAAI,aAAa,CAAC,OAAO,KAAK,KAAK,CAAC,MAAM,IAAI,aAAa,CAAC,UAAU,KAAK,OAAO;QAAE,MAAM,WAAW,CAAC,mDAAmD,CAAC,CAAC;IAC3J,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAmB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC;IACvG,IAAI,UAAU,CAAC,QAAQ,KAAK,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,6BAA6B,CAAC;QAAE,MAAM,WAAW,CAAC,0CAA0C,CAAC,CAAC;IACjL,MAAM,SAAS,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,uBAAuB,CAAC,CAAC;IACnG,IAAI,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;IAC9B,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACvH,MAAM,aAAa,GAAqB,EAAE,CAAC;IAC3C,MAAM,OAAO,GAAsB,EAAE,CAAC;IACtC,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,GAAG,QAAQ,EAAE,GAAG,aAAa,CAAC,CAAC,CAAC;QACxH,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAClC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IACD,KAAK,MAAM,GAAG,IAAI,aAAa;QAAE,QAAQ,GAAG,+BAA+B,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC3F,IAAI,QAAQ,CAAC,UAAU,KAAK,KAAK,CAAC,QAAQ,CAAC,UAAU;QAAE,MAAM,iBAAiB,CAAC,KAAK,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;IAC9G,MAAM,eAAe,GAAG,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC3F,MAAM,cAAc,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IACpD,MAAM,cAAc,GAAG,SAAS,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;IAC/G,MAAM,aAAa,GAAG,WAAW,CAAC,cAAc,CAAC,CAAC;IAClD,MAAM,SAAS,GAAG,eAAe,CAAC,MAAM,KAAK,CAAC,CAAC;IAC/C,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC7C,MAAM,eAAe,CAAC,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;IAChF,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,YAAY,CAAC,EAAE;QACnE,cAAc,EAAE,cAAc;QAC9B,QAAQ,EAAE,KAAK,CAAC,OAAO;QACvB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,UAAU,EAAE,cAAc;QAC1B,SAAS,EAAE,kBAAkB,CAAC,cAAc,CAAC;QAC7C,MAAM,EAAE,WAAW;QACnB,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC;QACxC,MAAM,EAAE,SAAS,CAAC,KAAK,CAAC,cAAc,CAAC;QACvC,cAAc,EAAE,aAAa;QAC7B,eAAe,EAAE,cAAc;QAC/B,uBAAuB,EAAE,cAAc;QACvC,wBAAwB,EAAE,eAAe;QACzC,OAAO;QACP,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;QAC/C,UAAU,EAAE,KAAK,CAAC,SAAS;QAC3B,YAAY,EAAE,WAAW;QACzB,UAAU,EAAE,WAAW;KACxB,CAAC,CAAC;IACH,OAAO;QACL,cAAc,EAAE,GAAG;QACnB,IAAI,EAAE,YAAY;QAClB,QAAQ,EAAE,KAAK,CAAC,OAAO;QACvB,UAAU,EAAE,cAAc;QAC1B,SAAS,EAAE,kBAAkB,CAAC,cAAc,CAAC;QAC7C,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,MAAM,EAAE,WAAW;QACnB,cAAc,EAAE,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC;QACjD,cAAc,EAAE,aAAa;QAC7B,uBAAuB,EAAE,cAAc;QACvC,eAAe,EAAE,SAAS,CAAC,KAAK,CAAC,cAAc,CAAC;QAChD,eAAe,EAAE,cAAc;QAC/B,wBAAwB,EAAE,eAAe;QACzC,OAAO;QACP,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;QAC/C,UAAU,EAAE,KAAK,CAAC,SAAS;QAC3B,YAAY,EAAE,WAAW;KAC1B,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,WAAW,CACxB,KAA4B,EAC5B,aAAkC,EAClC,UAA4B,EAC5B,SAAiB,EACjB,IAAmB,EACnB,YAA8B;IAE9B,MAAM,WAAW,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1H,MAAM,IAAI,GAAG,WAAW,IAAI,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;IAChH,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;QAC7F,MAAM,WAAW,CAAC,8CAA8C,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACtF,CAAC;IACD,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3F,IAAI,OAAO,CAAC;IACZ,IAAI,CAAC;QAAC,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;IAAC,CAAC;IACxE,OAAO,KAAK,EAAE,CAAC;QACb,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ;YAAE,MAAM,WAAW,CAAC,sCAAsC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpI,MAAM,KAAK,CAAC;IACd,CAAC;IACD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;SACjG,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;IAC/C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC/E,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,eAAe,EAAE,KAAK,CAAC,CAAC;QAC5E,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,YAAY,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,aAA0C,CAAC,CAAC,CAAC,EAAE,CAAC;QACvG,IAAI,CAAC,GAAG,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAC1C,MAAM,QAAQ,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;QACnC,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;YAC9C,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,MAAM,kBAAkB,CAAC;oBACnC,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,aAAa,EAAE,KAAK,CAAC,aAAa;oBAClC,kBAAkB,EAAE,YAAY;oBAChC,eAAe,EAAE,IAAI,CAAC,OAAO;oBAC7B,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,gBAAgB,EAAE,UAAU;oBAC5B,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY;oBACvC,iBAAiB,EAAE,KAAK,CAAC,OAAO,CAAC,kBAAkB;oBACnD,eAAe,EAAE,iBAAiB;oBAClC,SAAS;oBACT,GAAG,EAAE,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG;oBAC7B,qBAAqB,EAAE,IAAI;iBAC5B,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBAC3B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,KAAK,GAAG,CAAC,QAAQ,CAAC;oBAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrF,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,uBAAuB;oBAAE,SAAS;gBACvD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC,kBAAkB,KAAK,OAAO,CAAC,CAAC;QAC/I,IAAI,CAAC,GAAG;YAAE,SAAS;QACnB,MAAM,2BAA2B,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE;YACjE,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY;YACvC,iBAAiB,EAAE,KAAK,CAAC,OAAO,CAAC,kBAAkB;SACpD,CAAC,CAAC;QACH,OAAO;YACL,GAAG;YACH,MAAM,EAAE;gBACN,eAAe,EAAE,GAAG,CAAC,QAAQ;gBAC7B,aAAa,EAAE,GAAG,CAAC,MAAM;gBACzB,cAAc,EAAE,IAAI,CAAC,OAAO;gBAC5B,wBAAwB,EAAE,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;aACpE;SACF,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,UAAU,CAAC,mDAAmD,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE;QAC1F,IAAI,EAAE,sCAAsC;QAC5C,QAAQ,EAAE,EAAE;KACb,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,KAA4B,EAC5B,QAAwB,EACxB,eAAgC,EAChC,SAAkB,EAClB,WAAmB;IAEnB,MAAM,MAAM,GAA4B;QACtC,GAAG,CAAC,KAAK,CAAC,cAAc,IAAI,EAAE,CAAC;QAC/B,cAAc,EAAE,cAAc;QAC9B,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;QAC1C,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;QAC7B,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS;QAC/B,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO;QAC9B,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY;QACxC,kBAAkB,EAAE,KAAK,CAAC,OAAO,CAAC,kBAAkB;QACpD,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,OAAO,EAAE,kBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC;QAC5C,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,OAAO,EAAE,uCAAuC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC;QACjK,UAAU,EAAE,OAAO,KAAK,CAAC,cAAc,EAAE,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU;QACxH,YAAY,EAAE,WAAW;KAC1B,CAAC;IACF,IAAI,SAAS;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC;IACnC,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,CAAC;AAC/E,CAAC;AAED,SAAS,cAAc,CAAC,KAAc,EAAE,KAAa;IACnD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK;QAAE,MAAM,WAAW,CAAC,GAAG,KAAK,aAAa,CAAC,CAAC;IAClF,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,WAAW,CAAC,OAAe;IAClC,OAAO,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,+BAA+B,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;AACzF,CAAC"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import { mkdir } from "node:fs/promises";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { HitchError, atomicWriteJSON, readJSON } from "../foundation/index.js";
|
|
4
|
+
export async function prepareEvalDirectory(input) {
|
|
5
|
+
const directory = path.join(input.evalsDirectory, input.evalId);
|
|
6
|
+
if (input.precreated) {
|
|
7
|
+
const existingRequest = await readJSON(path.join(directory, "request.json"), null);
|
|
8
|
+
if (!existingRequest)
|
|
9
|
+
throw new HitchError(`precreated eval request is missing: ${input.evalId}`, { code: "eval_not_found", exitCode: 3 });
|
|
10
|
+
if (JSON.stringify(existingRequest) !== JSON.stringify(input.request)) {
|
|
11
|
+
throw new HitchError(`precreated eval request does not match: ${input.evalId}`, { code: "eval_request_conflict", exitCode: 2 });
|
|
12
|
+
}
|
|
13
|
+
if (await readJSON(path.join(directory, "result.json"), null)) {
|
|
14
|
+
throw new HitchError(`eval is already terminal: ${input.evalId}`, { code: "eval_id_conflict", exitCode: 2 });
|
|
15
|
+
}
|
|
16
|
+
return directory;
|
|
17
|
+
}
|
|
18
|
+
try {
|
|
19
|
+
await mkdir(directory, { mode: 0o700 });
|
|
20
|
+
}
|
|
21
|
+
catch (error) {
|
|
22
|
+
if (error.code === "EEXIST") {
|
|
23
|
+
throw new HitchError(`eval ID already exists: ${input.evalId}`, { code: "eval_id_conflict", exitCode: 2 });
|
|
24
|
+
}
|
|
25
|
+
throw error;
|
|
26
|
+
}
|
|
27
|
+
await atomicWriteJSON(path.join(directory, "request.json"), input.request);
|
|
28
|
+
return directory;
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=directory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"directory.js","sourceRoot":"","sources":["../../../src/evals/directory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAE/E,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,KAK1C;IACC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAChE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACrB,MAAM,eAAe,GAAG,MAAM,QAAQ,CAAqB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,EAAE,IAAI,CAAC,CAAC;QACvG,IAAI,CAAC,eAAe;YAAE,MAAM,IAAI,UAAU,CAAC,uCAAuC,KAAK,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAC3I,IAAI,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YACtE,MAAM,IAAI,UAAU,CAAC,2CAA2C,KAAK,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,uBAAuB,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAClI,CAAC;QACD,IAAI,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC;YAC9D,MAAM,IAAI,UAAU,CAAC,6BAA6B,KAAK,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/G,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACvD,MAAM,IAAI,UAAU,CAAC,2BAA2B,KAAK,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7G,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;IACD,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3E,OAAO,SAAS,CAAC;AACnB,CAAC"}
|
|
@@ -0,0 +1,89 @@
|
|
|
1
|
+
import { request } from "node:http";
|
|
2
|
+
const MAX_RESPONSE_BYTES = 1024 * 1024;
|
|
3
|
+
export async function readDockerEngineContainerStats(containerId, options = {}) {
|
|
4
|
+
if (!/^[a-f0-9]{12,64}$/.test(containerId))
|
|
5
|
+
throw new TypeError("Docker Engine stats container ID is invalid");
|
|
6
|
+
const endpoint = dockerEndpoint(options.env ?? process.env);
|
|
7
|
+
const timeoutMs = options.timeoutMs ?? 2_000;
|
|
8
|
+
if (!Number.isSafeInteger(timeoutMs) || timeoutMs < 1 || timeoutMs > 60_000)
|
|
9
|
+
throw new TypeError("Docker Engine stats timeout is invalid");
|
|
10
|
+
const payload = await requestJson(endpoint, `/containers/${containerId}/stats?stream=false`, timeoutMs, options.signal);
|
|
11
|
+
return parseDockerEngineContainerStats(payload);
|
|
12
|
+
}
|
|
13
|
+
export function parseDockerEngineContainerStats(value) {
|
|
14
|
+
if (!value || typeof value !== "object" || Array.isArray(value))
|
|
15
|
+
throw new TypeError("Docker Engine stats response is invalid");
|
|
16
|
+
const stats = value;
|
|
17
|
+
const cpuStats = record(stats.cpu_stats);
|
|
18
|
+
const cpuUsage = record(cpuStats?.cpu_usage);
|
|
19
|
+
const totalUsage = cpuUsage?.total_usage;
|
|
20
|
+
const memoryUsage = record(stats.memory_stats)?.usage;
|
|
21
|
+
if (totalUsage !== undefined && (!Number.isSafeInteger(totalUsage) || totalUsage < 0))
|
|
22
|
+
throw new TypeError("Docker Engine cumulative CPU time is invalid");
|
|
23
|
+
if (memoryUsage !== undefined && (!Number.isSafeInteger(memoryUsage) || memoryUsage < 0))
|
|
24
|
+
throw new TypeError("Docker Engine memory usage is invalid");
|
|
25
|
+
return {
|
|
26
|
+
...(totalUsage === undefined ? {} : { cpu_time_ns: totalUsage }),
|
|
27
|
+
...(memoryUsage === undefined ? {} : { memory_bytes: memoryUsage }),
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
function dockerEndpoint(env) {
|
|
31
|
+
const raw = env.HITCH_DOCKER_SOCKET || env.DOCKER_HOST;
|
|
32
|
+
if (!raw)
|
|
33
|
+
return { socketPath: process.platform === "win32" ? "//./pipe/docker_engine" : "/var/run/docker.sock" };
|
|
34
|
+
if (raw.startsWith("unix://"))
|
|
35
|
+
return { socketPath: decodeURIComponent(new URL(raw).pathname) };
|
|
36
|
+
if (raw.startsWith("npipe://"))
|
|
37
|
+
return { socketPath: raw.slice("npipe://".length).replace(/^\/+/, "//") };
|
|
38
|
+
if (raw.startsWith("tcp://") || raw.startsWith("http://")) {
|
|
39
|
+
const url = new URL(raw.replace(/^tcp:/, "http:"));
|
|
40
|
+
return { protocol: "http:", hostname: url.hostname, port: Number(url.port || 2375) };
|
|
41
|
+
}
|
|
42
|
+
throw new TypeError("Docker Engine endpoint is unsupported for cumulative stats");
|
|
43
|
+
}
|
|
44
|
+
function requestJson(endpoint, pathname, timeoutMs, signal) {
|
|
45
|
+
return new Promise((resolve, reject) => {
|
|
46
|
+
const requestOptions = endpoint.socketPath
|
|
47
|
+
? { socketPath: endpoint.socketPath, path: pathname, method: "GET" }
|
|
48
|
+
: { protocol: endpoint.protocol, hostname: endpoint.hostname, port: endpoint.port, path: pathname, method: "GET" };
|
|
49
|
+
const req = request(requestOptions, (response) => {
|
|
50
|
+
const chunks = [];
|
|
51
|
+
let size = 0;
|
|
52
|
+
response.on("data", (chunk) => {
|
|
53
|
+
size += chunk.byteLength;
|
|
54
|
+
if (size > MAX_RESPONSE_BYTES)
|
|
55
|
+
req.destroy(new Error("Docker Engine stats response is too large"));
|
|
56
|
+
else
|
|
57
|
+
chunks.push(chunk);
|
|
58
|
+
});
|
|
59
|
+
response.on("end", () => {
|
|
60
|
+
if (response.statusCode !== 200)
|
|
61
|
+
return reject(new Error(`Docker Engine stats returned HTTP ${response.statusCode ?? 0}`));
|
|
62
|
+
try {
|
|
63
|
+
resolve(JSON.parse(Buffer.concat(chunks).toString("utf8")));
|
|
64
|
+
}
|
|
65
|
+
catch (error) {
|
|
66
|
+
reject(new TypeError(`Docker Engine stats JSON is invalid: ${error.message}`));
|
|
67
|
+
}
|
|
68
|
+
});
|
|
69
|
+
});
|
|
70
|
+
const timer = setTimeout(() => req.destroy(new Error("Docker Engine stats timed out")), timeoutMs);
|
|
71
|
+
timer.unref();
|
|
72
|
+
const abort = () => { req.destroy(Object.assign(new Error("Docker Engine stats cancelled"), { code: "cancelled" })); };
|
|
73
|
+
signal?.addEventListener("abort", abort, { once: true });
|
|
74
|
+
req.on("error", reject);
|
|
75
|
+
req.on("close", () => {
|
|
76
|
+
clearTimeout(timer);
|
|
77
|
+
signal?.removeEventListener("abort", abort);
|
|
78
|
+
});
|
|
79
|
+
if (signal?.aborted) {
|
|
80
|
+
abort();
|
|
81
|
+
return;
|
|
82
|
+
}
|
|
83
|
+
req.end();
|
|
84
|
+
});
|
|
85
|
+
}
|
|
86
|
+
function record(value) {
|
|
87
|
+
return value && typeof value === "object" && !Array.isArray(value) ? value : null;
|
|
88
|
+
}
|
|
89
|
+
//# sourceMappingURL=docker-engine-stats.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"docker-engine-stats.js","sourceRoot":"","sources":["../../../src/evals/docker-engine-stats.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEpC,MAAM,kBAAkB,GAAG,IAAI,GAAG,IAAI,CAAC;AAOvC,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAClD,WAAmB,EACnB,UAAiF,EAAE;IAEnF,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,6CAA6C,CAAC,CAAC;IAC/G,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,CAAC,CAAC;IAC5D,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,KAAK,CAAC;IAC7C,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,MAAM;QAAE,MAAM,IAAI,SAAS,CAAC,wCAAwC,CAAC,CAAC;IAC3I,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,QAAQ,EAAE,eAAe,WAAW,qBAAqB,EAAE,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACxH,OAAO,+BAA+B,CAAC,OAAO,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,+BAA+B,CAAC,KAAc;IAC5D,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,yCAAyC,CAAC,CAAC;IAChI,MAAM,KAAK,GAAG,KAAgC,CAAC;IAC/C,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7C,MAAM,UAAU,GAAG,QAAQ,EAAE,WAAW,CAAC;IACzC,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,KAAK,CAAC;IACtD,IAAI,UAAU,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,IAAK,UAAqB,GAAG,CAAC,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,8CAA8C,CAAC,CAAC;IACvK,IAAI,WAAW,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,IAAK,WAAsB,GAAG,CAAC,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,uCAAuC,CAAC,CAAC;IACnK,OAAO;QACL,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,UAAoB,EAAE,CAAC;QAC1E,GAAG,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,WAAqB,EAAE,CAAC;KAC9E,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,GAAsB;IAC5C,MAAM,GAAG,GAAG,GAAG,CAAC,mBAAmB,IAAI,GAAG,CAAC,WAAW,CAAC;IACvD,IAAI,CAAC,GAAG;QAAE,OAAO,EAAE,UAAU,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,EAAE,CAAC;IAClH,IAAI,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,EAAE,UAAU,EAAE,kBAAkB,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;IAChG,IAAI,GAAG,CAAC,UAAU,CAAC,UAAU,CAAC;QAAE,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC;IAC1G,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1D,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QACnD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;IACvF,CAAC;IACD,MAAM,IAAI,SAAS,CAAC,4DAA4D,CAAC,CAAC;AACpF,CAAC;AAED,SAAS,WAAW,CAClB,QAA2C,EAC3C,QAAgB,EAChB,SAAiB,EACjB,MAAoB;IAEpB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,cAAc,GAAG,QAAQ,CAAC,UAAU;YACxC,CAAC,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE;YACpE,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QACrH,MAAM,GAAG,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,QAAQ,EAAE,EAAE;YAC/C,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBACpC,IAAI,IAAI,KAAK,CAAC,UAAU,CAAC;gBACzB,IAAI,IAAI,GAAG,kBAAkB;oBAAE,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC,CAAC;;oBAC9F,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACtB,IAAI,QAAQ,CAAC,UAAU,KAAK,GAAG;oBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,qCAAqC,QAAQ,CAAC,UAAU,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC3H,IAAI,CAAC;oBAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAY,CAAC,CAAC;gBAAC,CAAC;gBAC/E,OAAO,KAAK,EAAE,CAAC;oBAAC,MAAM,CAAC,IAAI,SAAS,CAAC,wCAAyC,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAAC,CAAC;YAC9G,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACnG,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,MAAM,KAAK,GAAG,GAAS,EAAE,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,+BAA+B,CAAC,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7H,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxB,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACnB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QACH,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,KAAK,EAAE,CAAC;YACR,OAAO;QACT,CAAC;QACD,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,MAAM,CAAC,KAAc;IAC5B,OAAO,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAgC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC/G,CAAC"}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
import { hitchRootId } from "../foundation/index.js";
|
|
2
|
+
export const DOCKER_OWNERSHIP_LABELS = {
|
|
3
|
+
rootId: "io.hitch.root-id",
|
|
4
|
+
provider: "io.hitch.provider",
|
|
5
|
+
evalId: "io.hitch.eval-id",
|
|
6
|
+
workId: "io.hitch.work-id",
|
|
7
|
+
leaseId: "io.hitch.lease-id",
|
|
8
|
+
leaseEpoch: "io.hitch.lease-epoch",
|
|
9
|
+
taskId: "io.hitch.task-id",
|
|
10
|
+
};
|
|
11
|
+
export function dockerResourceOwnership(root, lease, taskId) {
|
|
12
|
+
if (!validProvider(lease.provider))
|
|
13
|
+
throw new TypeError("Docker ownership provider is invalid");
|
|
14
|
+
if (taskId !== undefined && (!taskId || taskId.length > 4_096 || /[\0\r\n]/.test(taskId))) {
|
|
15
|
+
throw new TypeError("Docker ownership task id is invalid");
|
|
16
|
+
}
|
|
17
|
+
return {
|
|
18
|
+
root_id: hitchRootId(root),
|
|
19
|
+
provider: lease.provider,
|
|
20
|
+
eval_id: lease.eval_id,
|
|
21
|
+
work_id: lease.work_id,
|
|
22
|
+
lease_id: lease.lease_id,
|
|
23
|
+
lease_epoch: lease.epoch,
|
|
24
|
+
...(taskId === undefined ? {} : { task_id: taskId }),
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
export function dockerOwnershipLabelMap(ownership) {
|
|
28
|
+
validateDockerResourceOwnership(ownership);
|
|
29
|
+
return {
|
|
30
|
+
[DOCKER_OWNERSHIP_LABELS.rootId]: ownership.root_id,
|
|
31
|
+
[DOCKER_OWNERSHIP_LABELS.provider]: ownership.provider,
|
|
32
|
+
[DOCKER_OWNERSHIP_LABELS.evalId]: ownership.eval_id,
|
|
33
|
+
[DOCKER_OWNERSHIP_LABELS.workId]: ownership.work_id,
|
|
34
|
+
[DOCKER_OWNERSHIP_LABELS.leaseId]: ownership.lease_id,
|
|
35
|
+
[DOCKER_OWNERSHIP_LABELS.leaseEpoch]: String(ownership.lease_epoch),
|
|
36
|
+
...(ownership.task_id === undefined ? {} : { [DOCKER_OWNERSHIP_LABELS.taskId]: ownership.task_id }),
|
|
37
|
+
};
|
|
38
|
+
}
|
|
39
|
+
export function validateDockerResourceOwnership(value) {
|
|
40
|
+
if (!/^[a-f0-9]{24}$/.test(value.root_id) || !validProvider(value.provider)
|
|
41
|
+
|| !/^eval_[a-f0-9]{32}$/.test(value.eval_id) || !/^work_[a-f0-9]{32}$/.test(value.work_id)
|
|
42
|
+
|| !/^lease_[a-f0-9]{32}$/.test(value.lease_id) || !Number.isSafeInteger(value.lease_epoch) || value.lease_epoch < 1
|
|
43
|
+
|| (value.task_id !== undefined && (!value.task_id || value.task_id.length > 4_096 || /[\0\r\n]/.test(value.task_id)))) {
|
|
44
|
+
throw new TypeError("Docker resource ownership is invalid");
|
|
45
|
+
}
|
|
46
|
+
return value;
|
|
47
|
+
}
|
|
48
|
+
function validProvider(value) {
|
|
49
|
+
return Boolean(value) && value.length <= 128 && /^[a-z0-9][a-z0-9._-]*$/.test(value);
|
|
50
|
+
}
|
|
51
|
+
//# sourceMappingURL=docker-ownership.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"docker-ownership.js","sourceRoot":"","sources":["../../../src/evals/docker-ownership.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAErD,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,MAAM,EAAE,kBAAkB;IAC1B,QAAQ,EAAE,mBAAmB;IAC7B,MAAM,EAAE,kBAAkB;IAC1B,MAAM,EAAE,kBAAkB;IAC1B,OAAO,EAAE,mBAAmB;IAC5B,UAAU,EAAE,sBAAsB;IAClC,MAAM,EAAE,kBAAkB;CAClB,CAAC;AAEX,MAAM,UAAU,uBAAuB,CAAC,IAAY,EAAE,KAAuB,EAAE,MAAe;IAC5F,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,sCAAsC,CAAC,CAAC;IAChG,IAAI,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,KAAK,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;QAC1F,MAAM,IAAI,SAAS,CAAC,qCAAqC,CAAC,CAAC;IAC7D,CAAC;IACD,OAAO;QACL,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;QAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,KAAK,CAAC,KAAK;QACxB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;KACrD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,SAAoC;IAC1E,+BAA+B,CAAC,SAAS,CAAC,CAAC;IAC3C,OAAO;QACL,CAAC,uBAAuB,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,OAAO;QACnD,CAAC,uBAAuB,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,QAAQ;QACtD,CAAC,uBAAuB,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,OAAO;QACnD,CAAC,uBAAuB,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,OAAO;QACnD,CAAC,uBAAuB,CAAC,OAAO,CAAC,EAAE,SAAS,CAAC,QAAQ;QACrD,CAAC,uBAAuB,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC;QACnE,GAAG,CAAC,SAAS,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,uBAAuB,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC;KACpG,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,+BAA+B,CAAC,KAAgC;IAC9E,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC;WACtE,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;WACxF,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,WAAW,GAAG,CAAC;WACjH,CAAC,KAAK,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,KAAK,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QACzH,MAAM,IAAI,SAAS,CAAC,sCAAsC,CAAC,CAAC;IAC9D,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IAClC,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,IAAI,GAAG,IAAI,wBAAwB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACvF,CAAC"}
|
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
import { lstat } from "node:fs/promises";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { hitchRootId, readJSON, runCommand, withFileLock } from "../foundation/index.js";
|
|
4
|
+
import { DOCKER_OWNERSHIP_LABELS, validateDockerResourceOwnership } from "./docker-ownership.js";
|
|
5
|
+
import { parseExecutionLease } from "./execution-leases.js";
|
|
6
|
+
const RESOURCES = [
|
|
7
|
+
{ kind: "container", list: ["container", "ls", "--all"], remove: ["container", "rm", "--force"], format: "{{.ID}}" },
|
|
8
|
+
{ kind: "network", list: ["network", "ls"], remove: ["network", "rm"], format: "{{.ID}}" },
|
|
9
|
+
{ kind: "volume", list: ["volume", "ls"], remove: ["volume", "rm"], format: "{{.Name}}" },
|
|
10
|
+
];
|
|
11
|
+
export async function reapOwnedDockerResources(options) {
|
|
12
|
+
if (!options.root)
|
|
13
|
+
throw new TypeError("Docker reaper root is required");
|
|
14
|
+
const rootId = hitchRootId(options.root);
|
|
15
|
+
const selectedLeases = options.leaseIds === undefined ? null : validateLeaseIds(options.leaseIds);
|
|
16
|
+
const command = options.run ?? ((args) => runCommand(options.dockerExecutable || options.env?.HITCH_DOCKER_PATH || "docker", args, {
|
|
17
|
+
...(options.env ? { env: options.env } : {}),
|
|
18
|
+
...(options.signal ? { signal: options.signal } : {}),
|
|
19
|
+
timeoutMs: 10_000,
|
|
20
|
+
failureCode: "docker_reaper_command_failed",
|
|
21
|
+
}));
|
|
22
|
+
const report = { schema_version: "1", root_id: rootId, scanned: 0, deleted: [], retained: [], issues: [] };
|
|
23
|
+
for (const resource of RESOURCES) {
|
|
24
|
+
let ids;
|
|
25
|
+
try {
|
|
26
|
+
const listed = await command([...resource.list, "--filter", `label=${DOCKER_OWNERSHIP_LABELS.rootId}=${rootId}`, "--format", resource.format]);
|
|
27
|
+
ids = parseListedIds(listed.stdout);
|
|
28
|
+
}
|
|
29
|
+
catch (error) {
|
|
30
|
+
report.issues.push({ kind: resource.kind, stage: "list", message: errorMessage(error) });
|
|
31
|
+
continue;
|
|
32
|
+
}
|
|
33
|
+
for (const id of ids)
|
|
34
|
+
await inspectAndMaybeDelete(options.root, resource, id, rootId, selectedLeases, command, report);
|
|
35
|
+
}
|
|
36
|
+
return report;
|
|
37
|
+
}
|
|
38
|
+
async function inspectAndMaybeDelete(root, resource, id, rootId, selectedLeases, command, report) {
|
|
39
|
+
report.scanned += 1;
|
|
40
|
+
let inspected;
|
|
41
|
+
let ownership;
|
|
42
|
+
try {
|
|
43
|
+
inspected = await inspectResource(resource.kind, id, command);
|
|
44
|
+
id = inspected.id;
|
|
45
|
+
}
|
|
46
|
+
catch (error) {
|
|
47
|
+
report.issues.push({ kind: resource.kind, id, stage: "inspect", message: errorMessage(error) });
|
|
48
|
+
return;
|
|
49
|
+
}
|
|
50
|
+
try {
|
|
51
|
+
ownership = ownershipFromLabels(inspected.labels);
|
|
52
|
+
if (ownership.root_id !== rootId)
|
|
53
|
+
throw new TypeError("root label does not match this Hitch root");
|
|
54
|
+
if (selectedLeases && !selectedLeases.has(ownership.lease_id)) {
|
|
55
|
+
report.retained.push({ kind: resource.kind, id, reason: "lease_not_selected" });
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
const lease = await readOwnedLease(root, ownership);
|
|
59
|
+
assertLeaseAllowsDeletion(lease, ownership);
|
|
60
|
+
}
|
|
61
|
+
catch (error) {
|
|
62
|
+
report.retained.push({ kind: resource.kind, id, reason: errorMessage(error) });
|
|
63
|
+
return;
|
|
64
|
+
}
|
|
65
|
+
const evalDirectory = path.join(root, "evals", ownership.eval_id);
|
|
66
|
+
try {
|
|
67
|
+
await withFileLock(path.join(evalDirectory, "leases", ".locks"), ownership.lease_id, async () => {
|
|
68
|
+
const current = await inspectResource(resource.kind, id, command);
|
|
69
|
+
const currentOwnership = ownershipFromLabels(current.labels);
|
|
70
|
+
if (JSON.stringify(currentOwnership) !== JSON.stringify(ownership))
|
|
71
|
+
throw new TypeError("resource ownership changed after inspection");
|
|
72
|
+
assertLeaseAllowsDeletion(await readOwnedLease(root, ownership), ownership);
|
|
73
|
+
await command([...resource.remove, id]);
|
|
74
|
+
}, { timeoutCode: "docker_reaper_lease_locked", timeoutExitCode: 12 });
|
|
75
|
+
report.deleted.push({ kind: resource.kind, id, lease_id: ownership.lease_id, lease_epoch: ownership.lease_epoch });
|
|
76
|
+
}
|
|
77
|
+
catch (error) {
|
|
78
|
+
report.issues.push({ kind: resource.kind, id, stage: "delete", message: errorMessage(error) });
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
async function inspectResource(kind, id, command) {
|
|
82
|
+
if (!id || id.length > 4_096 || /[\0\r\n]/.test(id))
|
|
83
|
+
throw new TypeError("Docker resource id is invalid");
|
|
84
|
+
const value = JSON.parse((await command([kind, "inspect", id])).stdout);
|
|
85
|
+
if (!Array.isArray(value) || value.length !== 1 || !record(value[0]))
|
|
86
|
+
throw new TypeError("Docker inspect response is invalid");
|
|
87
|
+
const item = value[0];
|
|
88
|
+
const observedId = kind === "volume" ? item.Name : item.Id;
|
|
89
|
+
if (typeof observedId !== "string" || !observedId || (kind === "volume" ? observedId !== id : observedId !== id && !observedId.startsWith(id))) {
|
|
90
|
+
throw new TypeError("Docker inspect identity does not match listing");
|
|
91
|
+
}
|
|
92
|
+
const rawLabels = kind === "container" && record(item.Config) ? item.Config.Labels : item.Labels;
|
|
93
|
+
if (!record(rawLabels) || Object.values(rawLabels).some((entry) => typeof entry !== "string"))
|
|
94
|
+
throw new TypeError("Docker ownership labels are missing or invalid");
|
|
95
|
+
return { kind, id: observedId, labels: rawLabels };
|
|
96
|
+
}
|
|
97
|
+
function ownershipFromLabels(labels) {
|
|
98
|
+
const epochLabel = labels[DOCKER_OWNERSHIP_LABELS.leaseEpoch];
|
|
99
|
+
const epoch = Number(epochLabel);
|
|
100
|
+
if (epochLabel !== String(epoch))
|
|
101
|
+
throw new TypeError("Docker resource ownership epoch is not canonical");
|
|
102
|
+
return validateDockerResourceOwnership({
|
|
103
|
+
root_id: labels[DOCKER_OWNERSHIP_LABELS.rootId] || "",
|
|
104
|
+
provider: labels[DOCKER_OWNERSHIP_LABELS.provider] || "",
|
|
105
|
+
eval_id: labels[DOCKER_OWNERSHIP_LABELS.evalId] || "",
|
|
106
|
+
work_id: labels[DOCKER_OWNERSHIP_LABELS.workId] || "",
|
|
107
|
+
lease_id: labels[DOCKER_OWNERSHIP_LABELS.leaseId] || "",
|
|
108
|
+
lease_epoch: epoch,
|
|
109
|
+
...(labels[DOCKER_OWNERSHIP_LABELS.taskId] === undefined ? {} : { task_id: labels[DOCKER_OWNERSHIP_LABELS.taskId] }),
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
async function readOwnedLease(root, ownership) {
|
|
113
|
+
const file = path.join(root, "evals", ownership.eval_id, "leases", `${ownership.lease_id}.json`);
|
|
114
|
+
if (!(await lstat(file)).isFile())
|
|
115
|
+
throw new TypeError("execution lease is not a regular file");
|
|
116
|
+
return parseExecutionLease(await readJSON(file));
|
|
117
|
+
}
|
|
118
|
+
function assertLeaseAllowsDeletion(lease, ownership) {
|
|
119
|
+
if (lease.lease_id !== ownership.lease_id || lease.eval_id !== ownership.eval_id || lease.work_id !== ownership.work_id || lease.provider !== ownership.provider) {
|
|
120
|
+
throw new TypeError("resource labels do not match the persisted lease");
|
|
121
|
+
}
|
|
122
|
+
if (lease.state !== "released" && lease.state !== "expired" && lease.state !== "lost")
|
|
123
|
+
throw new TypeError(`lease is not terminal or expired: ${lease.state}`);
|
|
124
|
+
if (!(lease.resource_epochs ?? [lease.epoch]).includes(ownership.lease_epoch))
|
|
125
|
+
throw new TypeError("resource epoch is not authorized by the persisted lease");
|
|
126
|
+
}
|
|
127
|
+
function parseListedIds(stdout) {
|
|
128
|
+
const values = stdout.split(/\r?\n/).map((entry) => entry.trim()).filter(Boolean);
|
|
129
|
+
if (values.some((entry) => entry.length > 4_096 || /[\0\r\n]/.test(entry)))
|
|
130
|
+
throw new TypeError("Docker list returned an invalid resource id");
|
|
131
|
+
return [...new Set(values)].sort();
|
|
132
|
+
}
|
|
133
|
+
function validateLeaseIds(values) {
|
|
134
|
+
if (values.some((value) => !/^lease_[a-f0-9]{32}$/.test(value)))
|
|
135
|
+
throw new TypeError("Docker reaper lease selection is invalid");
|
|
136
|
+
return new Set(values);
|
|
137
|
+
}
|
|
138
|
+
function errorMessage(error) {
|
|
139
|
+
return (error?.message || String(error)).slice(0, 1_024);
|
|
140
|
+
}
|
|
141
|
+
function record(value) {
|
|
142
|
+
return Boolean(value) && typeof value === "object" && !Array.isArray(value);
|
|
143
|
+
}
|
|
144
|
+
//# sourceMappingURL=docker-reaper.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"docker-reaper.js","sourceRoot":"","sources":["../../../src/evals/docker-reaper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACzF,OAAO,EAAE,uBAAuB,EAAE,+BAA+B,EAAE,MAAM,uBAAuB,CAAC;AACjG,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AA4B5D,MAAM,SAAS,GAKV;IACH,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,WAAW,EAAE,IAAI,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE;IACpH,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE;IAC1F,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE;CAC1F,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,OAA4B;IACzE,IAAI,CAAC,OAAO,CAAC,IAAI;QAAE,MAAM,IAAI,SAAS,CAAC,gCAAgC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACzC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClG,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC,IAAc,EAAE,EAAE,CAAC,UAAU,CAC5D,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,EAAE,iBAAiB,IAAI,QAAQ,EACtE,IAAI,EACJ;QACE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5C,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,SAAS,EAAE,MAAM;QACjB,WAAW,EAAE,8BAA8B;KAC5C,CACF,CAAC,CAAC;IACH,MAAM,MAAM,GAAyB,EAAE,cAAc,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACjI,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,GAAa,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,SAAS,uBAAuB,CAAC,MAAM,IAAI,MAAM,EAAE,EAAE,UAAU,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/I,GAAG,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACzF,SAAS;QACX,CAAC;QACD,KAAK,MAAM,EAAE,IAAI,GAAG;YAAE,MAAM,qBAAqB,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACzH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,IAAY,EACZ,QAAkC,EAClC,EAAU,EACV,MAAc,EACd,cAAkC,EAClC,OAAyE,EACzE,MAA4B;IAE5B,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC;IACpB,IAAI,SAA4B,CAAC;IACjC,IAAI,SAAoC,CAAC;IACzC,IAAI,CAAC;QACH,SAAS,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;QAC9D,EAAE,GAAG,SAAS,CAAC,EAAE,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAChG,OAAO;IACT,CAAC;IACD,IAAI,CAAC;QACH,SAAS,GAAG,mBAAmB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAClD,IAAI,SAAS,CAAC,OAAO,KAAK,MAAM;YAAE,MAAM,IAAI,SAAS,CAAC,2CAA2C,CAAC,CAAC;QACnG,IAAI,cAAc,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9D,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,oBAAoB,EAAE,CAAC,CAAC;YAChF,OAAO;QACT,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACpD,yBAAyB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAC9C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/E,OAAO;IACT,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,CAAC;QACH,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,QAAQ,CAAC,EAAE,SAAS,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YAC9F,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;YAClE,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC7D,IAAI,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;gBAAE,MAAM,IAAI,SAAS,CAAC,6CAA6C,CAAC,CAAC;YACvI,yBAAyB,CAAC,MAAM,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC;YAC5E,MAAM,OAAO,CAAC,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC,EAAE,EAAE,WAAW,EAAE,4BAA4B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;QACvE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,SAAS,CAAC,QAAQ,EAAE,WAAW,EAAE,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC;IACrH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;AACH,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,IAAgC,EAChC,EAAU,EACV,OAAyE;IAEzE,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,MAAM,GAAG,KAAK,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,+BAA+B,CAAC,CAAC;IAC1G,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,OAAO,CAAC,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAY,CAAC;IACnF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,oCAAoC,CAAC,CAAC;IAChI,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACtB,MAAM,UAAU,GAAG,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IAC3D,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,KAAK,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAC/I,MAAM,IAAI,SAAS,CAAC,gDAAgD,CAAC,CAAC;IACxE,CAAC;IACD,MAAM,SAAS,GAAG,IAAI,KAAK,WAAW,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;IACjG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,gDAAgD,CAAC,CAAC;IACrK,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,SAAmC,EAAE,CAAC;AAC/E,CAAC;AAED,SAAS,mBAAmB,CAAC,MAA8B;IACzD,MAAM,UAAU,GAAG,MAAM,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;IAC9D,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;IACjC,IAAI,UAAU,KAAK,MAAM,CAAC,KAAK,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,kDAAkD,CAAC,CAAC;IAC1G,OAAO,+BAA+B,CAAC;QACrC,OAAO,EAAE,MAAM,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,EAAE;QACrD,QAAQ,EAAE,MAAM,CAAC,uBAAuB,CAAC,QAAQ,CAAC,IAAI,EAAE;QACxD,OAAO,EAAE,MAAM,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,EAAE;QACrD,OAAO,EAAE,MAAM,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,EAAE;QACrD,QAAQ,EAAE,MAAM,CAAC,uBAAuB,CAAC,OAAO,CAAC,IAAI,EAAE;QACvD,WAAW,EAAE,KAAK;QAClB,GAAG,CAAC,MAAM,CAAC,uBAAuB,CAAC,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,uBAAuB,CAAC,MAAM,CAAC,EAAE,CAAC;KACrH,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,IAAY,EAAE,SAAoC;IAC9E,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,CAAC,OAAO,EAAE,QAAQ,EAAE,GAAG,SAAS,CAAC,QAAQ,OAAO,CAAC,CAAC;IACjG,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,EAAE;QAAE,MAAM,IAAI,SAAS,CAAC,uCAAuC,CAAC,CAAC;IAChG,OAAO,mBAAmB,CAAC,MAAM,QAAQ,CAAU,IAAI,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,yBAAyB,CAAC,KAAuB,EAAE,SAAoC;IAC9F,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,QAAQ,EAAE,CAAC;QACjK,MAAM,IAAI,SAAS,CAAC,kDAAkD,CAAC,CAAC;IAC1E,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,KAAK,UAAU,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,MAAM;QAAE,MAAM,IAAI,SAAS,CAAC,qCAAqC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IAC/J,IAAI,CAAC,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,WAAW,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,yDAAyD,CAAC,CAAC;AAChK,CAAC;AAED,SAAS,cAAc,CAAC,MAAc;IACpC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClF,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,KAAK,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,6CAA6C,CAAC,CAAC;IAC/I,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAyB;IACjD,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,0CAA0C,CAAC,CAAC;IACjI,OAAO,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,CAAE,KAAe,EAAE,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;AACtE,CAAC;AAED,SAAS,MAAM,CAAC,KAAc;IAC5B,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC"}
|