agent-hitch 0.2.8 → 0.2.10
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 +206 -137
- package/README.zh-CN.md +193 -119
- package/dist/scripts/canary-native-training.js +221 -0
- package/dist/scripts/canary-native-training.js.map +1 -0
- package/dist/scripts/canary-remote-training.js +246 -0
- package/dist/scripts/canary-remote-training.js.map +1 -0
- package/dist/scripts/canary-remote-verifier.js +124 -0
- package/dist/scripts/canary-remote-verifier.js.map +1 -0
- package/dist/scripts/canary-worker-host.js +113 -0
- package/dist/scripts/canary-worker-host.js.map +1 -0
- package/dist/scripts/canary-worker-peer.js +188 -0
- package/dist/scripts/canary-worker-peer.js.map +1 -0
- package/dist/scripts/check-architecture.js +4 -3
- package/dist/scripts/check-architecture.js.map +1 -1
- package/dist/src/adapters/catalog.js +2 -1
- package/dist/src/adapters/catalog.js.map +1 -1
- package/dist/src/adapters/providers/codex.js +65 -2
- package/dist/src/adapters/providers/codex.js.map +1 -1
- package/dist/src/adapters/providers/model-call.js +17 -1
- package/dist/src/adapters/providers/model-call.js.map +1 -1
- package/dist/src/adapters/providers/training-tool.js +29 -0
- package/dist/src/adapters/providers/training-tool.js.map +1 -0
- package/dist/src/artifacts/index.js +1 -1
- package/dist/src/artifacts/index.js.map +1 -1
- package/dist/src/artifacts/store.js +15 -10
- package/dist/src/artifacts/store.js.map +1 -1
- package/dist/src/backends/harbor/backend.js +24 -15
- package/dist/src/backends/harbor/backend.js.map +1 -1
- package/dist/src/backends/harbor/bridge-environment.js +12 -0
- package/dist/src/backends/harbor/bridge-environment.js.map +1 -0
- package/dist/src/backends/harbor/environment-config.js +2 -2
- package/dist/src/backends/harbor/environment-config.js.map +1 -1
- package/dist/src/backends/harbor/host-credential-helper.js +81 -0
- package/dist/src/backends/harbor/host-credential-helper.js.map +1 -0
- package/dist/src/backends/harbor/index.js +3 -0
- package/dist/src/backends/harbor/index.js.map +1 -1
- package/dist/src/backends/harbor/model-proxy-config.js +18 -1
- package/dist/src/backends/harbor/model-proxy-config.js.map +1 -1
- package/dist/src/backends/harbor/observation.js +61 -0
- package/dist/src/backends/harbor/observation.js.map +1 -0
- package/dist/src/backends/harbor/portable-regrade.js +150 -0
- package/dist/src/backends/harbor/portable-regrade.js.map +1 -0
- package/dist/src/backends/harbor/process.js +26 -5
- package/dist/src/backends/harbor/process.js.map +1 -1
- package/dist/src/backends/harbor/regrade.js +4 -1
- package/dist/src/backends/harbor/regrade.js.map +1 -1
- package/dist/src/backends/harbor/supervisor.js +22 -0
- package/dist/src/backends/harbor/supervisor.js.map +1 -1
- package/dist/src/backends/harbor/tools.js +2 -0
- package/dist/src/backends/harbor/tools.js.map +1 -1
- package/dist/src/backends/index.js +3 -0
- package/dist/src/backends/index.js.map +1 -1
- package/dist/src/cli/arguments.js +33 -0
- package/dist/src/cli/arguments.js.map +1 -1
- package/dist/src/cli/commands/capabilities.js +14 -1
- package/dist/src/cli/commands/capabilities.js.map +1 -1
- package/dist/src/cli/commands/daemon.js +19 -0
- package/dist/src/cli/commands/daemon.js.map +1 -1
- package/dist/src/cli/commands/eval.js +38 -10
- package/dist/src/cli/commands/eval.js.map +1 -1
- package/dist/src/cli/commands/local.js +130 -0
- package/dist/src/cli/commands/local.js.map +1 -0
- package/dist/src/cli/commands/model-node.js +26 -0
- package/dist/src/cli/commands/model-node.js.map +1 -0
- package/dist/src/cli/commands/models.js +72 -0
- package/dist/src/cli/commands/models.js.map +1 -0
- package/dist/src/cli/commands/run.js +36 -1
- package/dist/src/cli/commands/run.js.map +1 -1
- package/dist/src/cli/commands/training.js +32 -0
- package/dist/src/cli/commands/training.js.map +1 -0
- package/dist/src/cli/commands/worker.js +30 -2
- package/dist/src/cli/commands/worker.js.map +1 -1
- package/dist/src/cli/main.js +8 -0
- package/dist/src/cli/main.js.map +1 -1
- package/dist/src/cli/output.js +24 -2
- package/dist/src/cli/output.js.map +1 -1
- package/dist/src/control-plane/eval-records.js +2 -0
- package/dist/src/control-plane/eval-records.js.map +1 -1
- package/dist/src/control-plane/eval-recovery.js +7 -1
- package/dist/src/control-plane/eval-recovery.js.map +1 -1
- package/dist/src/control-plane/eval-scheduler.js +24 -50
- package/dist/src/control-plane/eval-scheduler.js.map +1 -1
- package/dist/src/control-plane/eval-submission-persist.js +30 -0
- package/dist/src/control-plane/eval-submission-persist.js.map +1 -0
- package/dist/src/control-plane/execution-observation-contract.js +60 -0
- package/dist/src/control-plane/execution-observation-contract.js.map +1 -0
- package/dist/src/control-plane/index.js +10 -0
- package/dist/src/control-plane/index.js.map +1 -1
- package/dist/src/control-plane/inference-manager.js +357 -0
- package/dist/src/control-plane/inference-manager.js.map +1 -0
- package/dist/src/control-plane/managed-gateway-receipt.js +26 -0
- package/dist/src/control-plane/managed-gateway-receipt.js.map +1 -0
- package/dist/src/control-plane/managed-service-recovery.js +158 -0
- package/dist/src/control-plane/managed-service-recovery.js.map +1 -0
- package/dist/src/control-plane/model-capture-planning.js +15 -1
- package/dist/src/control-plane/model-capture-planning.js.map +1 -1
- package/dist/src/control-plane/ordered-eval-control.js +221 -0
- package/dist/src/control-plane/ordered-eval-control.js.map +1 -0
- package/dist/src/control-plane/remote-backend-result.js +18 -0
- package/dist/src/control-plane/remote-backend-result.js.map +1 -0
- package/dist/src/control-plane/remote-execution-lease.js +15 -0
- package/dist/src/control-plane/remote-execution-lease.js.map +1 -0
- package/dist/src/control-plane/remote-execution-observation.js +108 -0
- package/dist/src/control-plane/remote-execution-observation.js.map +1 -0
- package/dist/src/control-plane/remote-model-routes.js +107 -0
- package/dist/src/control-plane/remote-model-routes.js.map +1 -0
- package/dist/src/control-plane/remote-observation-reporter.js +37 -0
- package/dist/src/control-plane/remote-observation-reporter.js.map +1 -0
- package/dist/src/control-plane/remote-offer-contract.js +48 -0
- package/dist/src/control-plane/remote-offer-contract.js.map +1 -0
- package/dist/src/control-plane/remote-rerun-journal.js +61 -0
- package/dist/src/control-plane/remote-rerun-journal.js.map +1 -0
- package/dist/src/control-plane/remote-result-transport.js +55 -9
- package/dist/src/control-plane/remote-result-transport.js.map +1 -1
- package/dist/src/control-plane/remote-verifier-import.js +52 -0
- package/dist/src/control-plane/remote-verifier-import.js.map +1 -0
- package/dist/src/control-plane/remote-verifier-result.js +61 -0
- package/dist/src/control-plane/remote-verifier-result.js.map +1 -0
- package/dist/src/control-plane/remote-verifier-source-transport.js +68 -0
- package/dist/src/control-plane/remote-verifier-source-transport.js.map +1 -0
- package/dist/src/control-plane/remote-work-coordinator.js +103 -47
- package/dist/src/control-plane/remote-work-coordinator.js.map +1 -1
- package/dist/src/control-plane/remote-work-inputs.js +26 -5
- package/dist/src/control-plane/remote-work-inputs.js.map +1 -1
- package/dist/src/control-plane/remote-work-recovery.js +123 -34
- package/dist/src/control-plane/remote-work-recovery.js.map +1 -1
- package/dist/src/control-plane/remote-worker-artifacts.js +10 -7
- package/dist/src/control-plane/remote-worker-artifacts.js.map +1 -1
- package/dist/src/control-plane/remote-worker-cleanup.js +104 -0
- package/dist/src/control-plane/remote-worker-cleanup.js.map +1 -0
- package/dist/src/control-plane/remote-worker-client.js +156 -14
- package/dist/src/control-plane/remote-worker-client.js.map +1 -1
- package/dist/src/control-plane/remote-worker-codecs.js +160 -0
- package/dist/src/control-plane/remote-worker-codecs.js.map +1 -0
- package/dist/src/control-plane/remote-worker-generation-recovery.js +67 -0
- package/dist/src/control-plane/remote-worker-generation-recovery.js.map +1 -0
- package/dist/src/control-plane/remote-worker-ownership.js +51 -0
- package/dist/src/control-plane/remote-worker-ownership.js.map +1 -0
- package/dist/src/control-plane/remote-worker-protocol.js +85 -119
- package/dist/src/control-plane/remote-worker-protocol.js.map +1 -1
- package/dist/src/control-plane/remote-worker-runner.js +95 -18
- package/dist/src/control-plane/remote-worker-runner.js.map +1 -1
- package/dist/src/control-plane/remote-workers.js +33 -5
- package/dist/src/control-plane/remote-workers.js.map +1 -1
- package/dist/src/control-plane/rerun-recovery.js +89 -0
- package/dist/src/control-plane/rerun-recovery.js.map +1 -0
- package/dist/src/control-plane/rerun-scheduler.js +70 -96
- package/dist/src/control-plane/rerun-scheduler.js.map +1 -1
- package/dist/src/control-plane/rerun-source.js +57 -0
- package/dist/src/control-plane/rerun-source.js.map +1 -0
- package/dist/src/control-plane/worker-model-relay.js +76 -0
- package/dist/src/control-plane/worker-model-relay.js.map +1 -0
- package/dist/src/controller-runtime/hash.js +3 -0
- package/dist/src/controller-runtime/hash.js.map +1 -1
- package/dist/src/controller-runtime/index.js +2 -0
- package/dist/src/controller-runtime/index.js.map +1 -1
- package/dist/src/controller-runtime/observation.js +29 -0
- package/dist/src/controller-runtime/observation.js.map +1 -0
- package/dist/src/controller-runtime/resident.js +22 -0
- package/dist/src/controller-runtime/resident.js.map +1 -0
- package/dist/src/daemon/auth.js +7 -2
- package/dist/src/daemon/auth.js.map +1 -1
- package/dist/src/daemon/client.js +34 -1
- package/dist/src/daemon/client.js.map +1 -1
- package/dist/src/daemon/execution-observation.js +38 -0
- package/dist/src/daemon/execution-observation.js.map +1 -0
- package/dist/src/daemon/http.js +57 -0
- package/dist/src/daemon/http.js.map +1 -0
- package/dist/src/daemon/inference-prepare.js +47 -0
- package/dist/src/daemon/inference-prepare.js.map +1 -0
- package/dist/src/daemon/scheduler.js +4 -1
- package/dist/src/daemon/scheduler.js.map +1 -1
- package/dist/src/daemon/server.js +80 -82
- package/dist/src/daemon/server.js.map +1 -1
- package/dist/src/daemon/worker-observation.js +34 -0
- package/dist/src/daemon/worker-observation.js.map +1 -0
- package/dist/src/daemon/worker-ownership.js +71 -0
- package/dist/src/daemon/worker-ownership.js.map +1 -0
- package/dist/src/daemon/worker-routes.js +83 -15
- package/dist/src/daemon/worker-routes.js.map +1 -1
- package/dist/src/domain/execution-observation.js +2 -0
- package/dist/src/domain/execution-observation.js.map +1 -0
- package/dist/src/domain/index.js +1 -0
- package/dist/src/domain/index.js.map +1 -1
- package/dist/src/domain/inference.js +2 -0
- package/dist/src/domain/inference.js.map +1 -0
- package/dist/src/domain/runs.js.map +1 -1
- package/dist/src/domain/training.js +2 -0
- package/dist/src/domain/training.js.map +1 -0
- package/dist/src/domain/validation.js +11 -0
- package/dist/src/domain/validation.js.map +1 -1
- package/dist/src/domain/verifier-source.js +2 -0
- package/dist/src/domain/verifier-source.js.map +1 -0
- package/dist/src/domain/worker-recovery.js +2 -0
- package/dist/src/domain/worker-recovery.js.map +1 -0
- package/dist/src/evals/eval-logical-plan.js +8 -1
- package/dist/src/evals/eval-logical-plan.js.map +1 -1
- package/dist/src/evals/execution-leases.js +15 -2
- package/dist/src/evals/execution-leases.js.map +1 -1
- package/dist/src/evals/execution-plan.js +11 -1
- package/dist/src/evals/execution-plan.js.map +1 -1
- package/dist/src/evals/harbor-bridge-error.js +9 -0
- package/dist/src/evals/harbor-bridge-error.js.map +1 -1
- package/dist/src/evals/index.js +19 -0
- package/dist/src/evals/index.js.map +1 -1
- package/dist/src/evals/model-capture-plan.js +12 -0
- package/dist/src/evals/model-capture-plan.js.map +1 -1
- package/dist/src/evals/model-capture-runtime.js +25 -0
- package/dist/src/evals/model-capture-runtime.js.map +1 -1
- package/dist/src/evals/physical-work-plan.js +57 -0
- package/dist/src/evals/physical-work-plan.js.map +1 -0
- package/dist/src/evals/preparation-rerun.js +6 -0
- package/dist/src/evals/preparation-rerun.js.map +1 -1
- package/dist/src/evals/recovery.js +8 -1
- package/dist/src/evals/recovery.js.map +1 -1
- package/dist/src/evals/regrade-evidence.js +4 -4
- package/dist/src/evals/regrade-evidence.js.map +1 -1
- package/dist/src/evals/remote-execution-ownership.js +50 -0
- package/dist/src/evals/remote-execution-ownership.js.map +1 -0
- package/dist/src/evals/remote-infrastructure-retry.js +2 -0
- package/dist/src/evals/remote-infrastructure-retry.js.map +1 -1
- package/dist/src/evals/remote-lease-release.js +58 -0
- package/dist/src/evals/remote-lease-release.js.map +1 -0
- package/dist/src/evals/remote-rerun-execution.js +85 -0
- package/dist/src/evals/remote-rerun-execution.js.map +1 -0
- package/dist/src/evals/remote-trial.js +6 -0
- package/dist/src/evals/remote-trial.js.map +1 -0
- package/dist/src/evals/remote-verifier-assessment.js +85 -0
- package/dist/src/evals/remote-verifier-assessment.js.map +1 -0
- package/dist/src/evals/remote-verifier-contract.js +43 -0
- package/dist/src/evals/remote-verifier-contract.js.map +1 -0
- package/dist/src/evals/remote-verifier-execution.js +105 -0
- package/dist/src/evals/remote-verifier-execution.js.map +1 -0
- package/dist/src/evals/remote-verifier-journal.js +21 -0
- package/dist/src/evals/remote-verifier-journal.js.map +1 -0
- package/dist/src/evals/remote-verifier-outcome.js +51 -0
- package/dist/src/evals/remote-verifier-outcome.js.map +1 -0
- package/dist/src/evals/remote-verifier-rerun.js +134 -0
- package/dist/src/evals/remote-verifier-rerun.js.map +1 -0
- package/dist/src/evals/remote-verifier-selection.js +61 -0
- package/dist/src/evals/remote-verifier-selection.js.map +1 -0
- package/dist/src/evals/remote-verifier-source.js +33 -0
- package/dist/src/evals/remote-verifier-source.js.map +1 -0
- package/dist/src/evals/remote-worker-cleanup.js +46 -0
- package/dist/src/evals/remote-worker-cleanup.js.map +1 -0
- package/dist/src/evals/request.js +58 -1
- package/dist/src/evals/request.js.map +1 -1
- package/dist/src/evals/rerun-completion.js +39 -0
- package/dist/src/evals/rerun-completion.js.map +1 -0
- package/dist/src/evals/rerun-inputs.js +54 -0
- package/dist/src/evals/rerun-inputs.js.map +1 -1
- package/dist/src/evals/rerun-resume.js +70 -0
- package/dist/src/evals/rerun-resume.js.map +1 -0
- package/dist/src/evals/rerun-state.js +25 -0
- package/dist/src/evals/rerun-state.js.map +1 -0
- package/dist/src/evals/rerun-types.js.map +1 -1
- package/dist/src/evals/rerun.js +94 -101
- package/dist/src/evals/rerun.js.map +1 -1
- package/dist/src/evals/service.js +59 -5
- package/dist/src/evals/service.js.map +1 -1
- package/dist/src/evals/verifier-only-rerun.js +2 -16
- package/dist/src/evals/verifier-only-rerun.js.map +1 -1
- package/dist/src/evals/verifier-rerun-result.js +24 -0
- package/dist/src/evals/verifier-rerun-result.js.map +1 -0
- package/dist/src/evals/verifier-source-import.js +48 -0
- package/dist/src/evals/verifier-source-import.js.map +1 -0
- package/dist/src/evals/verifier-source.js +139 -0
- package/dist/src/evals/verifier-source.js.map +1 -0
- package/dist/src/foundation/base64.js +19 -0
- package/dist/src/foundation/base64.js.map +1 -0
- package/dist/src/foundation/config.js +10 -0
- package/dist/src/foundation/config.js.map +1 -1
- package/dist/src/foundation/credential-redaction.js +22 -9
- package/dist/src/foundation/credential-redaction.js.map +1 -1
- package/dist/src/foundation/index.js +1 -0
- package/dist/src/foundation/index.js.map +1 -1
- package/dist/src/foundation/process-identity.js +13 -9
- package/dist/src/foundation/process-identity.js.map +1 -1
- package/dist/src/inference/daemon-capacity.js +41 -0
- package/dist/src/inference/daemon-capacity.js.map +1 -0
- package/dist/src/inference/device-reservation.js +58 -0
- package/dist/src/inference/device-reservation.js.map +1 -0
- package/dist/src/inference/doctor.js +95 -0
- package/dist/src/inference/doctor.js.map +1 -0
- package/dist/src/inference/generation-recovery.js +83 -0
- package/dist/src/inference/generation-recovery.js.map +1 -0
- package/dist/src/inference/index.js +24 -0
- package/dist/src/inference/index.js.map +1 -0
- package/dist/src/inference/lock.js +331 -0
- package/dist/src/inference/lock.js.map +1 -0
- package/dist/src/inference/managed-launcher.js +35 -0
- package/dist/src/inference/managed-launcher.js.map +1 -0
- package/dist/src/inference/manifest.js +262 -0
- package/dist/src/inference/manifest.js.map +1 -0
- package/dist/src/inference/materialize.js +66 -0
- package/dist/src/inference/materialize.js.map +1 -0
- package/dist/src/inference/model-location.js +34 -0
- package/dist/src/inference/model-location.js.map +1 -0
- package/dist/src/inference/model-store.js +439 -0
- package/dist/src/inference/model-store.js.map +1 -0
- package/dist/src/inference/node-client.js +167 -0
- package/dist/src/inference/node-client.js.map +1 -0
- package/dist/src/inference/node-registry.js +79 -0
- package/dist/src/inference/node-registry.js.map +1 -0
- package/dist/src/inference/observation.js +79 -0
- package/dist/src/inference/observation.js.map +1 -0
- package/dist/src/inference/preflight.js +132 -0
- package/dist/src/inference/preflight.js.map +1 -0
- package/dist/src/inference/process-sglang.js +169 -0
- package/dist/src/inference/process-sglang.js.map +1 -0
- package/dist/src/inference/protocol-probe.js +29 -0
- package/dist/src/inference/protocol-probe.js.map +1 -0
- package/dist/src/inference/remote-model-store.js +24 -0
- package/dist/src/inference/remote-model-store.js.map +1 -0
- package/dist/src/inference/runtime-catalog.js +41 -0
- package/dist/src/inference/runtime-catalog.js.map +1 -0
- package/dist/src/inference/runtime-store.js +93 -0
- package/dist/src/inference/runtime-store.js.map +1 -0
- package/dist/src/inference/service-attachment.js +46 -0
- package/dist/src/inference/service-attachment.js.map +1 -0
- package/dist/src/inference/service-inspection.js +38 -0
- package/dist/src/inference/service-inspection.js.map +1 -0
- package/dist/src/inference/sglang.js +279 -0
- package/dist/src/inference/sglang.js.map +1 -0
- package/dist/src/inference/supervisor.js +460 -0
- package/dist/src/inference/supervisor.js.map +1 -0
- package/dist/src/model-access/environment.js +28 -0
- package/dist/src/model-access/environment.js.map +1 -0
- package/dist/src/model-access/index.js +4 -0
- package/dist/src/model-access/index.js.map +1 -1
- package/dist/src/model-access/local-gateway-request.js +139 -0
- package/dist/src/model-access/local-gateway-request.js.map +1 -0
- package/dist/src/model-access/local-gateway.js +323 -0
- package/dist/src/model-access/local-gateway.js.map +1 -0
- package/dist/src/model-access/proxy.js +90 -5
- package/dist/src/model-access/proxy.js.map +1 -1
- package/dist/src/model-access/remote-model.js +58 -0
- package/dist/src/model-access/remote-model.js.map +1 -0
- package/dist/src/model-access/training.js +115 -0
- package/dist/src/model-access/training.js.map +1 -0
- package/dist/src/runs/adapter-process.js +2 -2
- package/dist/src/runs/adapter-process.js.map +1 -1
- package/dist/src/runs/executor-types.js +2 -0
- package/dist/src/runs/executor-types.js.map +1 -0
- package/dist/src/runs/executor.js +54 -34
- package/dist/src/runs/executor.js.map +1 -1
- package/dist/src/runs/index.js +1 -0
- package/dist/src/runs/index.js.map +1 -1
- package/dist/src/runs/local-inference-environment.js +59 -0
- package/dist/src/runs/local-inference-environment.js.map +1 -0
- package/dist/src/runs/local-inference-run.js +53 -0
- package/dist/src/runs/local-inference-run.js.map +1 -0
- package/dist/src/runs/outcome.js +22 -0
- package/dist/src/runs/outcome.js.map +1 -1
- package/dist/src/runs/records.js +9 -0
- package/dist/src/runs/records.js.map +1 -1
- package/dist/src/runs/request.js +58 -3
- package/dist/src/runs/request.js.map +1 -1
- package/dist/src/runs/training-evidence.js +22 -0
- package/dist/src/runs/training-evidence.js.map +1 -0
- package/dist/src/runs/training-runtime.js +20 -0
- package/dist/src/runs/training-runtime.js.map +1 -0
- package/dist/src/workers/execution-observation.js +32 -0
- package/dist/src/workers/execution-observation.js.map +1 -0
- package/dist/src/workers/index.js +2 -0
- package/dist/src/workers/index.js.map +1 -1
- package/dist/src/workers/remote-harbor-ownership.js +266 -0
- package/dist/src/workers/remote-harbor-ownership.js.map +1 -0
- package/dist/src/workers/remote-harbor-verifier.js +76 -0
- package/dist/src/workers/remote-harbor-verifier.js.map +1 -0
- package/dist/src/workers/remote-harbor-work-spec.js +66 -10
- package/dist/src/workers/remote-harbor-work-spec.js.map +1 -1
- package/dist/src/workers/remote-harbor-worker.js +56 -49
- package/dist/src/workers/remote-harbor-worker.js.map +1 -1
- package/dist/src/workers/remote-worker-host.js +33 -0
- package/dist/src/workers/remote-worker-host.js.map +1 -0
- package/docs/local-model-inference-spec.zh-CN.md +666 -0
- package/docs/schemas/controller-runtime-observation.schema.json +21 -0
- package/docs/schemas/eval-execution-plan.schema.json +1 -0
- package/docs/schemas/eval-request.schema.json +3 -1
- package/docs/schemas/execution-lease.schema.json +95 -1
- package/docs/schemas/execution-provider-status.schema.json +162 -28
- package/docs/schemas/inference-execution-evidence.schema.json +487 -0
- package/docs/schemas/inference-lock.schema.json +487 -0
- package/docs/schemas/inference-runtime-manifest.schema.json +156 -0
- package/docs/schemas/inference-service-record.schema.json +227 -0
- package/docs/schemas/local-inference-selection.schema.json +98 -0
- package/docs/schemas/local-model-manifest.schema.json +43 -0
- package/docs/schemas/model-node-binding.schema.json +34 -0
- package/docs/schemas/model-node-generation-release.schema.json +46 -0
- package/docs/schemas/ordered-eval-control.schema.json +39 -0
- package/docs/schemas/remote-execution-lease-grant.schema.json +45 -0
- package/docs/schemas/remote-harbor-work-spec.schema.json +325 -0
- package/docs/schemas/remote-model-binding.schema.json +21 -0
- package/docs/schemas/remote-physical-execution.schema.json +108 -0
- package/docs/schemas/remote-result-envelope.schema.json +9 -3
- package/docs/schemas/remote-tree-envelope.schema.json +17 -3
- package/docs/schemas/remote-verifier-result.schema.json +203 -0
- package/docs/schemas/remote-verifier-source.schema.json +52 -0
- package/docs/schemas/remote-verifier-work.schema.json +125 -0
- package/docs/schemas/remote-work-offer.schema.json +170 -34
- package/docs/schemas/remote-worker-execution-admission.schema.json +214 -0
- package/docs/schemas/remote-worker-generation-cleanup.schema.json +199 -0
- package/docs/schemas/remote-worker-registration.schema.json +226 -25
- package/docs/schemas/run-manifest.schema.json +211 -44
- package/docs/schemas/run-request.schema.json +2 -1
- package/docs/schemas/training-external-binding.schema.json +26 -0
- package/integrations/harbor/hitch_harbor_agent.py +366 -38
- package/integrations/harbor/hitch_harbor_environment.py +22 -1
- package/integrations/harbor/hitch_host_credentials.py +354 -0
- package/integrations/harbor/hitch_private_exec.py +150 -0
- package/integrations/model-call/cli.js +13 -5
- package/integrations/training-tool/cli.js +72 -0
- package/package.json +8 -2
|
@@ -0,0 +1,357 @@
|
|
|
1
|
+
import { randomUUID } from "node:crypto";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { HitchError, atomicWriteJSON, ensureDir, readJSON, sha256JSON, statePaths, withFileLock } from "../foundation/index.js";
|
|
4
|
+
import { prepareLocalInference, SGLangServiceSupervisor } from "../inference/index.js";
|
|
5
|
+
import { LocalModelGateway } from "../model-access/index.js";
|
|
6
|
+
import { sealManagedGateway } from "./managed-gateway-receipt.js";
|
|
7
|
+
export class LocalInferenceManager {
|
|
8
|
+
root;
|
|
9
|
+
resources;
|
|
10
|
+
supervisor;
|
|
11
|
+
preflight;
|
|
12
|
+
startGateway;
|
|
13
|
+
onEvent;
|
|
14
|
+
services = new Map();
|
|
15
|
+
serviceById = new Map();
|
|
16
|
+
gatewayPending = new Map();
|
|
17
|
+
resourceLeases = new Map();
|
|
18
|
+
recovery;
|
|
19
|
+
recovered = new Map();
|
|
20
|
+
recoveryPollIntervalMs;
|
|
21
|
+
recoveryTimer;
|
|
22
|
+
recoveryChecking;
|
|
23
|
+
closed = false;
|
|
24
|
+
constructor(options) {
|
|
25
|
+
this.root = options.root;
|
|
26
|
+
this.resources = options.resources;
|
|
27
|
+
this.onEvent = options.onEvent;
|
|
28
|
+
this.recovery = options.recovery;
|
|
29
|
+
this.recoveryPollIntervalMs = options.recoveryPollIntervalMs ?? 1_000;
|
|
30
|
+
if (!Number.isSafeInteger(this.recoveryPollIntervalMs) || this.recoveryPollIntervalMs < 1)
|
|
31
|
+
throw new TypeError("invalid inference recovery polling interval");
|
|
32
|
+
this.supervisor = options.supervisor ?? new SGLangServiceSupervisor({ root: options.root, ...(options.onEvent ? { onEvent: options.onEvent } : {}) });
|
|
33
|
+
this.preflight = options.preflight ?? prepareLocalInference;
|
|
34
|
+
this.startGateway = options.startGateway ?? LocalModelGateway.start;
|
|
35
|
+
this.supervisor.subscribeTerminal(async (record, released) => {
|
|
36
|
+
const service = this.serviceById.get(record.service_id);
|
|
37
|
+
if (service) {
|
|
38
|
+
service.invalidated = true;
|
|
39
|
+
await service.gateway.close();
|
|
40
|
+
if (released)
|
|
41
|
+
await this.cleanupManagedService(service);
|
|
42
|
+
}
|
|
43
|
+
else if (released) {
|
|
44
|
+
const key = `${record.inference_id}:${record.isolation_key}`;
|
|
45
|
+
this.resourceLeases.get(key)?.release();
|
|
46
|
+
this.resourceLeases.delete(key);
|
|
47
|
+
}
|
|
48
|
+
});
|
|
49
|
+
}
|
|
50
|
+
async initialize() {
|
|
51
|
+
const plans = await this.recovery?.select(await this.supervisor.list()) ?? [];
|
|
52
|
+
const claims = new Map();
|
|
53
|
+
for (const plan of plans)
|
|
54
|
+
claims.set(plan.record.service_id, [...claims.get(plan.record.service_id) ?? [], plan.ownerId]);
|
|
55
|
+
const attached = await this.supervisor.recover([...claims].map(([service_id, owner_ids]) => ({ service_id, owner_ids })));
|
|
56
|
+
for (const restored of attached) {
|
|
57
|
+
const selected = plans.filter(plan => plan.record.service_id === restored.record.service_id);
|
|
58
|
+
if (!selected.length || selected.some(plan => plan.port !== selected[0].port || sha256JSON(plan.lock) !== sha256JSON(restored.lock))) {
|
|
59
|
+
throw new HitchError("recovered model service has conflicting gateway evidence", { code: "inference_recovery_ambiguous", exitCode: 12 });
|
|
60
|
+
}
|
|
61
|
+
const first = restored.owners.get(selected[0].ownerId);
|
|
62
|
+
const gateway = await this.startGateway({ upstreamBaseUrl: first.base_url, engineToken: first.engine_token, wireModel: first.wire_model,
|
|
63
|
+
lock: restored.lock, bindHost: "127.0.0.1", listenPort: selected[0].port });
|
|
64
|
+
const key = `${restored.record.inference_id}:${restored.record.isolation_key}`;
|
|
65
|
+
const managed = { key, serviceId: restored.record.service_id, gateway, refs: selected.length, lock: restored.lock };
|
|
66
|
+
this.services.set(key, managed);
|
|
67
|
+
this.serviceById.set(managed.serviceId, managed);
|
|
68
|
+
for (const plan of selected) {
|
|
69
|
+
const service = restored.owners.get(plan.ownerId);
|
|
70
|
+
let owner;
|
|
71
|
+
const registration = gateway.restore(plan.ownerId, plan.target.binding, plan.target.credential, async () => service.isReady() && (owner.claimed || await plan.state() === "active"));
|
|
72
|
+
let released = false;
|
|
73
|
+
const lease = { binding: registration.binding, credential: registration.credential, lock: restored.lock,
|
|
74
|
+
service_id: service.service_id, service_epoch: service.epoch,
|
|
75
|
+
release: async () => {
|
|
76
|
+
if (released)
|
|
77
|
+
return;
|
|
78
|
+
registration.revoke();
|
|
79
|
+
await service.release();
|
|
80
|
+
managed.refs = Math.max(0, managed.refs - 1);
|
|
81
|
+
released = true;
|
|
82
|
+
} };
|
|
83
|
+
owner = { plan, service, registration, lease, claimed: false };
|
|
84
|
+
this.recovered.set(`${key}:${plan.ownerId}`, owner);
|
|
85
|
+
}
|
|
86
|
+
this.onEvent?.({ type: "inference.gateway.reattached", service_id: managed.serviceId, inference_id: restored.record.inference_id, epoch: restored.record.epoch });
|
|
87
|
+
}
|
|
88
|
+
if (this.recovered.size) {
|
|
89
|
+
this.recoveryTimer = setInterval(() => {
|
|
90
|
+
this.recoveryChecking ??= this.reconcileRecovered().catch(() => {
|
|
91
|
+
this.onEvent?.({ type: "inference.recovery.unresolved", code: "inference_recovery_ambiguous" });
|
|
92
|
+
}).finally(() => { this.recoveryChecking = undefined; });
|
|
93
|
+
}, this.recoveryPollIntervalMs);
|
|
94
|
+
this.recoveryTimer.unref?.();
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
async reconcileRecovered() {
|
|
98
|
+
for (const [key, owner] of this.recovered) {
|
|
99
|
+
const state = owner.service.isReady() ? await owner.plan.state() : "finished";
|
|
100
|
+
if (owner.claimed || this.recovered.get(key) !== owner)
|
|
101
|
+
continue;
|
|
102
|
+
if (state !== "finished")
|
|
103
|
+
continue;
|
|
104
|
+
owner.retiring ??= owner.lease.release().then(() => { this.recovered.delete(key); });
|
|
105
|
+
await owner.retiring;
|
|
106
|
+
}
|
|
107
|
+
if (!this.recovered.size && this.recoveryTimer) {
|
|
108
|
+
clearInterval(this.recoveryTimer);
|
|
109
|
+
delete this.recoveryTimer;
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
async acquire(input) {
|
|
113
|
+
if (this.closed)
|
|
114
|
+
throw new HitchError("local inference manager is closed", { code: "inference_route_unavailable", exitCode: 12 });
|
|
115
|
+
const prepared = await this.prepareInputs(input);
|
|
116
|
+
return (await this.acquirePrepared(input, prepared, true)).lease;
|
|
117
|
+
}
|
|
118
|
+
prepareInputs(input) {
|
|
119
|
+
return this.preflight({
|
|
120
|
+
root: this.root,
|
|
121
|
+
selection: input.selection,
|
|
122
|
+
harnessRef: input.harness_ref,
|
|
123
|
+
reusableLocks: [...this.services.values()].filter((s) => !s.invalidated).map((s) => s.lock),
|
|
124
|
+
...(input.signal ? { runtime: { signal: input.signal } } : {}),
|
|
125
|
+
...(input.on_event ? { onProgress: (message) => input.on_event?.({ type: "inference.preparing", message }) } : {}),
|
|
126
|
+
});
|
|
127
|
+
}
|
|
128
|
+
async acquirePrepared(input, prepared, persistEvidence) {
|
|
129
|
+
if (this.closed)
|
|
130
|
+
throw new HitchError("local inference manager is closed", { code: "inference_route_unavailable", exitCode: 12 });
|
|
131
|
+
if (input.signal?.aborted)
|
|
132
|
+
throw new HitchError("inference acquisition cancelled", { code: "cancelled", exitCode: 9 });
|
|
133
|
+
const cacheScopeOwner = !prepared.lock.model_node && prepared.lock.execution.prefix_cache.mode === "disabled"
|
|
134
|
+
? "prefix-cache-disabled"
|
|
135
|
+
: input.cache_scope_owner;
|
|
136
|
+
const isolationKey = sha256JSON({ inference_id: prepared.lock.inference_id, cache_scope_owner: cacheScopeOwner });
|
|
137
|
+
const serviceKey = `${prepared.lock.inference_id}:${isolationKey}`;
|
|
138
|
+
const recoveredKey = `${serviceKey}:${input.run_id}`;
|
|
139
|
+
await this.recovered.get(recoveredKey)?.retiring;
|
|
140
|
+
if (this.closed || input.signal?.aborted)
|
|
141
|
+
throw new HitchError("inference acquisition interrupted during recovery", { code: "inference_route_unavailable", exitCode: 12 });
|
|
142
|
+
const recovered = this.recovered.get(recoveredKey);
|
|
143
|
+
if (recovered) {
|
|
144
|
+
if (!recovered.service.isReady() || recovered.plan.cacheScopeOwner !== input.cache_scope_owner
|
|
145
|
+
|| sha256JSON(recovered.plan.lock) !== sha256JSON(prepared.lock)
|
|
146
|
+
|| sha256JSON(recovered.plan.evidenceOwner) !== sha256JSON(input.evidence_owner ?? null)) {
|
|
147
|
+
throw new HitchError("resumed inference owner differs from its original service", { code: "inference_recovery_ambiguous", exitCode: 12 });
|
|
148
|
+
}
|
|
149
|
+
recovered.claimed = true;
|
|
150
|
+
this.recovered.delete(recoveredKey);
|
|
151
|
+
return { lease: recovered.lease, observation: recovered.service.observation ?? null };
|
|
152
|
+
}
|
|
153
|
+
let allocation = this.resourceLeases.get(serviceKey);
|
|
154
|
+
// Model-node process capacity/ownership is enforced by that node's shared
|
|
155
|
+
// training/evaluation ledger, never by the controller host's GPU inventory.
|
|
156
|
+
if (!allocation && this.resources && !prepared.lock.model_node) {
|
|
157
|
+
if (!this.resources.canEverFit(prepared.lock.resources)) {
|
|
158
|
+
const missing = Object.entries(prepared.lock.resources).filter(([key, value]) => value > (this.resources.capacity[key] ?? 0))
|
|
159
|
+
.map(([key, value]) => `${key} requires ${value}`);
|
|
160
|
+
throw new HitchError(`local inference resources exceed daemon capacity (${missing.join(", ")}); restart the daemon with sufficient --capacity-* limits`, { code: "inference_capacity_exceeded", exitCode: 12 });
|
|
161
|
+
}
|
|
162
|
+
allocation = this.resources.tryAcquire(serviceKey, "inference", prepared.lock.resources) ?? undefined;
|
|
163
|
+
if (!allocation)
|
|
164
|
+
throw new HitchError("local inference resources are currently unavailable", { code: "inference_capacity_exceeded", exitCode: 12 });
|
|
165
|
+
this.resourceLeases.set(serviceKey, allocation);
|
|
166
|
+
}
|
|
167
|
+
// Only a supervisor terminal event confirms that the engine no longer owns
|
|
168
|
+
// these resources. An acquisition failure alone is not such confirmation.
|
|
169
|
+
const serviceLease = await this.supervisor.acquire({
|
|
170
|
+
lock: prepared.lock,
|
|
171
|
+
model: prepared.model,
|
|
172
|
+
runtime: prepared.runtime,
|
|
173
|
+
isolationKey,
|
|
174
|
+
ownerId: input.run_id,
|
|
175
|
+
...(input.signal ? { signal: input.signal } : {}),
|
|
176
|
+
});
|
|
177
|
+
const managed = await this.ensureGateway(serviceKey, serviceLease, prepared.lock, allocation, input);
|
|
178
|
+
managed.refs += 1;
|
|
179
|
+
let registration;
|
|
180
|
+
try {
|
|
181
|
+
if (!serviceLease.isReady() || input.signal?.aborted)
|
|
182
|
+
throw new HitchError("inference service is no longer available", { code: "inference_route_unavailable", exitCode: 12 });
|
|
183
|
+
registration = managed.gateway.register(input.run_id);
|
|
184
|
+
if (prepared.lock.model_node)
|
|
185
|
+
await sealManagedGateway(this.root, { serviceId: serviceLease.service_id, epoch: serviceLease.epoch,
|
|
186
|
+
inferenceId: prepared.lock.inference_id, isolationKey, ownerId: input.run_id, cacheScopeOwner: input.cache_scope_owner,
|
|
187
|
+
evidenceOwner: input.evidence_owner, binding: registration.binding, credential: registration.credential });
|
|
188
|
+
if (persistEvidence)
|
|
189
|
+
await writeInferenceEvidence(this.root, input.run_id, input.evidence_owner, {
|
|
190
|
+
lock: prepared.lock,
|
|
191
|
+
model: prepared.model,
|
|
192
|
+
runtime: prepared.runtime,
|
|
193
|
+
service: { service_id: serviceLease.service_id, epoch: serviceLease.epoch, isolation_key: isolationKey },
|
|
194
|
+
doctor: prepared.doctor ?? null,
|
|
195
|
+
model_node_observation: prepared.node_observation ?? null,
|
|
196
|
+
observation: serviceLease.observation ?? null,
|
|
197
|
+
});
|
|
198
|
+
input.on_event?.({
|
|
199
|
+
type: "inference.ready", run_id: input.run_id, service_id: serviceLease.service_id,
|
|
200
|
+
inference_id: prepared.lock.inference_id, backend: prepared.lock.execution.platform.backend,
|
|
201
|
+
});
|
|
202
|
+
}
|
|
203
|
+
catch (error) {
|
|
204
|
+
registration?.revoke();
|
|
205
|
+
managed.refs = Math.max(0, managed.refs - 1);
|
|
206
|
+
await serviceLease.release();
|
|
207
|
+
throw error;
|
|
208
|
+
}
|
|
209
|
+
let released = false;
|
|
210
|
+
const lease = {
|
|
211
|
+
binding: registration.binding,
|
|
212
|
+
credential: registration.credential,
|
|
213
|
+
lock: prepared.lock,
|
|
214
|
+
service_id: serviceLease.service_id,
|
|
215
|
+
service_epoch: serviceLease.epoch,
|
|
216
|
+
release: async () => {
|
|
217
|
+
if (released)
|
|
218
|
+
return;
|
|
219
|
+
released = true;
|
|
220
|
+
registration.revoke();
|
|
221
|
+
managed.refs = Math.max(0, managed.refs - 1);
|
|
222
|
+
await serviceLease.release();
|
|
223
|
+
},
|
|
224
|
+
};
|
|
225
|
+
return { lease, observation: serviceLease.observation ?? null };
|
|
226
|
+
}
|
|
227
|
+
async prepare(selection, signal, onEvent) {
|
|
228
|
+
if (this.closed)
|
|
229
|
+
throw new HitchError("local inference manager is closed", { code: "inference_route_unavailable", exitCode: 12 });
|
|
230
|
+
const id = `run_${randomUUID().replaceAll("-", "")}`;
|
|
231
|
+
const input = {
|
|
232
|
+
run_id: id, harness_ref: "model-call", selection, cache_scope_owner: id,
|
|
233
|
+
...(signal ? { signal } : {}), ...(onEvent ? { on_event: onEvent } : {}),
|
|
234
|
+
};
|
|
235
|
+
const prepared = await this.prepareInputs(input);
|
|
236
|
+
const { lease, observation } = await this.acquirePrepared(input, prepared, false);
|
|
237
|
+
try {
|
|
238
|
+
const result = { schema_version: "1", ...prepared, observation };
|
|
239
|
+
await atomicWriteJSON(path.join(statePaths(this.root).inferenceLocks, prepared.lock.inference_id.slice(7), "validation.json"), result);
|
|
240
|
+
return result;
|
|
241
|
+
}
|
|
242
|
+
finally {
|
|
243
|
+
await lease.release();
|
|
244
|
+
try {
|
|
245
|
+
await this.stop(lease.service_id);
|
|
246
|
+
}
|
|
247
|
+
catch (error) {
|
|
248
|
+
if (error.code !== "inference_in_use")
|
|
249
|
+
throw error;
|
|
250
|
+
}
|
|
251
|
+
}
|
|
252
|
+
}
|
|
253
|
+
async list() { return this.supervisor.list(); }
|
|
254
|
+
async stop(serviceId, force = false) {
|
|
255
|
+
await this.supervisor.stop(serviceId, force);
|
|
256
|
+
}
|
|
257
|
+
async close() {
|
|
258
|
+
if (this.closed)
|
|
259
|
+
return;
|
|
260
|
+
this.closed = true;
|
|
261
|
+
if (this.recoveryTimer)
|
|
262
|
+
clearInterval(this.recoveryTimer);
|
|
263
|
+
await this.recoveryChecking;
|
|
264
|
+
await this.supervisor.close();
|
|
265
|
+
}
|
|
266
|
+
async ensureGateway(key, lease, lock, resources, input) {
|
|
267
|
+
const existing = this.services.get(key);
|
|
268
|
+
if (existing && existing.serviceId === lease.service_id && !existing.invalidated)
|
|
269
|
+
return existing;
|
|
270
|
+
let pending = this.gatewayPending.get(lease.service_id);
|
|
271
|
+
if (!pending) {
|
|
272
|
+
pending = (async () => {
|
|
273
|
+
const gateway = await this.startGateway({
|
|
274
|
+
upstreamBaseUrl: lease.base_url,
|
|
275
|
+
engineToken: lease.engine_token,
|
|
276
|
+
wireModel: lease.wire_model,
|
|
277
|
+
lock,
|
|
278
|
+
...(input.on_event ? { onRequest: input.on_event } : {}),
|
|
279
|
+
});
|
|
280
|
+
if (!lease.isReady()) {
|
|
281
|
+
await gateway.close();
|
|
282
|
+
throw new HitchError("inference service failed while opening its gateway", { code: "inference_route_unavailable", exitCode: 12 });
|
|
283
|
+
}
|
|
284
|
+
const managed = { key, serviceId: lease.service_id, gateway, lock, refs: 0, ...(resources ? { resources } : {}) };
|
|
285
|
+
this.services.set(key, managed);
|
|
286
|
+
this.serviceById.set(lease.service_id, managed);
|
|
287
|
+
return managed;
|
|
288
|
+
})();
|
|
289
|
+
this.gatewayPending.set(lease.service_id, pending);
|
|
290
|
+
pending.finally(() => { if (this.gatewayPending.get(lease.service_id) === pending)
|
|
291
|
+
this.gatewayPending.delete(lease.service_id); }).catch(() => { });
|
|
292
|
+
}
|
|
293
|
+
try {
|
|
294
|
+
return await pending;
|
|
295
|
+
}
|
|
296
|
+
catch (error) {
|
|
297
|
+
await lease.release();
|
|
298
|
+
try {
|
|
299
|
+
await this.supervisor.stop(lease.service_id);
|
|
300
|
+
}
|
|
301
|
+
catch { /* Other owners or an ambiguous stop retain the reservation. */ }
|
|
302
|
+
throw error;
|
|
303
|
+
}
|
|
304
|
+
}
|
|
305
|
+
async cleanupManagedService(service) {
|
|
306
|
+
if (this.services.get(service.key) !== service)
|
|
307
|
+
return;
|
|
308
|
+
this.services.delete(service.key);
|
|
309
|
+
this.serviceById.delete(service.serviceId);
|
|
310
|
+
if (this.resourceLeases.get(service.key) === service.resources)
|
|
311
|
+
this.resourceLeases.delete(service.key);
|
|
312
|
+
await service.gateway.close().catch(() => { });
|
|
313
|
+
service.resources?.release();
|
|
314
|
+
}
|
|
315
|
+
}
|
|
316
|
+
async function writeInferenceEvidence(root, runId, owner, evidence) {
|
|
317
|
+
if (!/^run_[a-f0-9]{32}$/.test(runId))
|
|
318
|
+
throw new TypeError("inference evidence run ID is invalid");
|
|
319
|
+
if (owner && (!/^eval_[a-f0-9]{32}$/.test(owner.eval_id)
|
|
320
|
+
|| owner.rerun_id !== undefined && !/^rerun_[a-f0-9]{32}$/.test(owner.rerun_id))) {
|
|
321
|
+
throw new TypeError("inference evidence eval identity is invalid");
|
|
322
|
+
}
|
|
323
|
+
const directory = await ensureDir(owner
|
|
324
|
+
? owner.rerun_id
|
|
325
|
+
? path.join(statePaths(root).evals, owner.eval_id, "reruns", owner.rerun_id, "inference")
|
|
326
|
+
: path.join(statePaths(root).evals, owner.eval_id, "inference")
|
|
327
|
+
: path.join(statePaths(root).runs, runId, "inference"));
|
|
328
|
+
const execution = {
|
|
329
|
+
schema_version: evidence.model_node_observation ? "2" : "1", run_id: runId,
|
|
330
|
+
...(owner ? { eval_id: owner.eval_id, ...(owner.rerun_id ? { rerun_id: owner.rerun_id } : {}) } : {}),
|
|
331
|
+
service: evidence.service, doctor: evidence.doctor, observation: evidence.observation,
|
|
332
|
+
...(evidence.model_node_observation ? { model_node_observation: evidence.model_node_observation } : {}),
|
|
333
|
+
prepared_at: new Date().toISOString(),
|
|
334
|
+
};
|
|
335
|
+
const files = [["lock.json", evidence.lock], ["model.manifest.json", evidence.model],
|
|
336
|
+
["runtime.manifest.json", evidence.runtime], ["execution.json", execution]];
|
|
337
|
+
await withFileLock(statePaths(root).inferenceOperationLocks, sha256JSON({ inference_evidence: directory }), async () => {
|
|
338
|
+
const previous = await readJSON(path.join(directory, "execution.json"), null);
|
|
339
|
+
if (execution.schema_version === "2" && previous && sha256JSON(previous.service) === sha256JSON(execution.service)) {
|
|
340
|
+
// Recovery can finish the worker and retire its gateway before the eval
|
|
341
|
+
// scheduler reacquires the same service to finalize. Retain the original
|
|
342
|
+
// evidence, including its time, instead of making that service look new.
|
|
343
|
+
if (typeof previous.prepared_at !== "string" || !Number.isFinite(Date.parse(previous.prepared_at))
|
|
344
|
+
|| sha256JSON(previous) !== sha256JSON({ ...execution, prepared_at: previous.prepared_at })) {
|
|
345
|
+
throw new HitchError("original managed inference execution evidence changed", { code: "inference_recovery_ambiguous", exitCode: 12 });
|
|
346
|
+
}
|
|
347
|
+
for (const [name, value] of files.slice(0, -1)) {
|
|
348
|
+
if (sha256JSON(await readJSON(path.join(directory, name), null)) !== sha256JSON(value)) {
|
|
349
|
+
throw new HitchError("original managed inference input evidence changed", { code: "inference_recovery_ambiguous", exitCode: 12 });
|
|
350
|
+
}
|
|
351
|
+
}
|
|
352
|
+
return;
|
|
353
|
+
}
|
|
354
|
+
await Promise.all(files.map(([name, value]) => atomicWriteJSON(path.join(directory, name), value)));
|
|
355
|
+
});
|
|
356
|
+
}
|
|
357
|
+
//# sourceMappingURL=inference-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"inference-manager.js","sourceRoot":"","sources":["../../../src/control-plane/inference-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAQ7B,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAChI,OAAO,EAAE,qBAAqB,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAEvF,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAI7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAgClE,MAAM,OAAO,qBAAqB;IACf,IAAI,CAAS;IACb,SAAS,CAA6B;IACtC,UAAU,CAA0B;IACpC,SAAS,CAAwF;IACjG,YAAY,CAAoE;IAChF,OAAO,CAAyD;IAChE,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC7C,WAAW,GAAG,IAAI,GAAG,EAA0B,CAAC;IAChD,cAAc,GAAG,IAAI,GAAG,EAAmC,CAAC;IAC5D,cAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;IAClD,QAAQ,CAAqC;IAC7C,SAAS,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC9C,sBAAsB,CAAS;IACxC,aAAa,CAAkB;IAC/B,gBAAgB,CAA4B;IAC5C,MAAM,GAAG,KAAK,CAAC;IAEvB,YAAY,OAAqC;QAC/C,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,IAAI,KAAK,CAAC;QACtE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,IAAI,CAAC,sBAAsB,GAAG,CAAC;YAAE,MAAM,IAAI,SAAS,CAAC,6CAA6C,CAAC,CAAC;QAC9J,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,IAAI,uBAAuB,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACtJ,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,qBAAqB,CAAC;QAC5D,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,iBAAiB,CAAC,KAAK,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC3D,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACxD,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;gBAC3B,MAAM,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC9B,IAAI,QAAQ;oBAAE,MAAM,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAC1D,CAAC;iBAAM,IAAI,QAAQ,EAAE,CAAC;gBACpB,MAAM,GAAG,GAAG,GAAG,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBAC7D,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC;gBACxC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;QAC9E,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoB,CAAC;QAC3C,KAAK,MAAM,IAAI,IAAI,KAAK;YAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1H,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1H,KAAK,MAAM,QAAQ,IAAI,QAAQ,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,KAAK,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC7F,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAC,IAAI,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;gBACtI,MAAM,IAAI,UAAU,CAAC,0DAA0D,EAAE,EAAE,IAAI,EAAE,8BAA8B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAC3I,CAAC;YACD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,OAAO,CAAE,CAAC;YACzD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,eAAe,EAAE,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,UAAU;gBACrI,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAC/E,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,YAAY,IAAI,QAAQ,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC/E,MAAM,OAAO,GAAmB,EAAE,GAAG,EAAE,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,UAAU,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC;YACpI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAClF,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;gBAC5B,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAE,CAAC;gBACnD,IAAI,KAAqB,CAAC;gBAC1B,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAC5F,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,MAAM,IAAI,CAAC,KAAK,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACvF,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,MAAM,KAAK,GAA4B,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,EAAE,UAAU,EAAE,YAAY,CAAC,UAAU,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI;oBAC9H,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,aAAa,EAAE,OAAO,CAAC,KAAK;oBAC5D,OAAO,EAAE,KAAK,IAAI,EAAE;wBAClB,IAAI,QAAQ;4BAAE,OAAO;wBACrB,YAAY,CAAC,MAAM,EAAE,CAAC;wBACtB,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;wBACxB,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;wBAAC,QAAQ,GAAG,IAAI,CAAC;oBAChE,CAAC,EAAE,CAAC;gBACN,KAAK,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBAC/D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAC;YACtD,CAAC;YACD,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,UAAU,EAAE,OAAO,CAAC,SAAS,EAAE,YAAY,EAAE,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QACpK,CAAC;QACD,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;YACxB,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE;gBACpC,IAAI,CAAC,gBAAgB,KAAK,IAAI,CAAC,kBAAkB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;oBAC7D,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,+BAA+B,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC,CAAC;gBAClG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,CAAC,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAChC,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,EAAE,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,kBAAkB;QAC9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1C,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;YAC9E,IAAI,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,KAAK;gBAAE,SAAS;YACjE,IAAI,KAAK,KAAK,UAAU;gBAAE,SAAS;YACnC,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACrF,MAAM,KAAK,CAAC,QAAQ,CAAC;QACvB,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAAC,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAAC,OAAO,IAAI,CAAC,aAAa,CAAC;QAAC,CAAC;IACnH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,KAAqC;QACjD,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,UAAU,CAAC,mCAAmC,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAClI,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACjD,OAAO,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;IACnE,CAAC;IAEO,aAAa,CAAC,KAAqC;QACzD,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,KAAK,CAAC,WAAW;YAC7B,aAAa,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YAC3F,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACnH,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,KAAqC,EAAE,QAAyC,EAAE,eAAwB;QACtI,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,UAAU,CAAC,mCAAmC,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAClI,IAAI,KAAK,CAAC,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,UAAU,CAAC,iCAAiC,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QACvH,MAAM,eAAe,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,KAAK,UAAU;YAC3G,CAAC,CAAC,uBAAuB;YACzB,CAAC,CAAC,KAAK,CAAC,iBAAiB,CAAC;QAC5B,MAAM,YAAY,GAAG,UAAU,CAAC,EAAE,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC;QAClH,MAAM,UAAU,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,YAAY,IAAI,YAAY,EAAE,CAAC;QACnE,MAAM,YAAY,GAAG,GAAG,UAAU,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;QACrD,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,QAAQ,CAAC;QACjD,IAAI,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,UAAU,CAAC,mDAAmD,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAC3K,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,SAAS,CAAC,IAAI,CAAC,eAAe,KAAK,KAAK,CAAC,iBAAiB;mBACzF,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;mBAC7D,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,UAAU,CAAC,KAAK,CAAC,cAAc,IAAI,IAAI,CAAC,EAAE,CAAC;gBAC3F,MAAM,IAAI,UAAU,CAAC,2DAA2D,EAAE,EAAE,IAAI,EAAE,8BAA8B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAC5I,CAAC;YACD,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC;YAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAC9D,OAAO,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,EAAE,WAAW,EAAE,SAAS,CAAC,OAAO,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QACxF,CAAC;QACD,IAAI,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACrD,0EAA0E;QAC1E,4EAA4E;QAC5E,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YAC/D,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACxD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,SAAU,CAAC,QAAQ,CAAC,GAA2C,CAAC,IAAI,CAAC,CAAC,CAAC;qBACnK,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,aAAa,KAAK,EAAE,CAAC,CAAC;gBACrD,MAAM,IAAI,UAAU,CAAC,qDAAqD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,2DAA2D,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAClN,CAAC;YACD,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,UAAU,EAAE,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC;YACtG,IAAI,CAAC,UAAU;gBAAE,MAAM,IAAI,UAAU,CAAC,qDAAqD,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACpJ,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAClD,CAAC;QACD,2EAA2E;QAC3E,0EAA0E;QAC1E,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YACjD,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,YAAY;YACZ,OAAO,EAAE,KAAK,CAAC,MAAM;YACrB,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAClD,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,YAAY,EAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;QACrG,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC;QAClB,IAAI,YAAY,CAAC;QACjB,IAAI,CAAC;YACH,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,MAAM,EAAE,OAAO;gBAAE,MAAM,IAAI,UAAU,CAAC,0CAA0C,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9K,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtD,IAAI,QAAQ,CAAC,IAAI,CAAC,UAAU;gBAAE,MAAM,kBAAkB,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,YAAY,CAAC,UAAU,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK;oBAC/H,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,CAAC,MAAM,EAAE,eAAe,EAAE,KAAK,CAAC,iBAAiB;oBACtH,aAAa,EAAE,KAAK,CAAC,cAAc,EAAE,OAAO,EAAE,YAAY,CAAC,OAAO,EAAE,UAAU,EAAE,YAAY,CAAC,UAAU,EAAE,CAAC,CAAC;YAC7G,IAAI,eAAe;gBAAE,MAAM,sBAAsB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,cAAc,EAAE;oBAC/F,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,OAAO,EAAE,EAAE,UAAU,EAAE,YAAY,CAAC,UAAU,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE;oBACxG,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,IAAI;oBAC/B,sBAAsB,EAAE,QAAQ,CAAC,gBAAgB,IAAI,IAAI;oBACzD,WAAW,EAAE,YAAY,CAAC,WAAW,IAAI,IAAI;iBAC9C,CAAC,CAAC;YACH,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACf,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,UAAU,EAAE,YAAY,CAAC,UAAU;gBAClF,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO;aAC5F,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,EAAE,MAAM,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;YAC7C,MAAM,YAAY,CAAC,OAAO,EAAE,CAAC;YAC7B,MAAM,KAAK,CAAC;QACd,CAAC;QACD,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,MAAM,KAAK,GAA4B;YACrC,OAAO,EAAE,YAAY,CAAC,OAAO;YAC7B,UAAU,EAAE,YAAY,CAAC,UAAU;YACnC,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,UAAU,EAAE,YAAY,CAAC,UAAU;YACnC,aAAa,EAAE,YAAY,CAAC,KAAK;YACjC,OAAO,EAAE,KAAK,IAAI,EAAE;gBAClB,IAAI,QAAQ;oBAAE,OAAO;gBACrB,QAAQ,GAAG,IAAI,CAAC;gBAChB,YAAY,CAAC,MAAM,EAAE,CAAC;gBACtB,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;gBAC7C,MAAM,YAAY,CAAC,OAAO,EAAE,CAAC;YAC/B,CAAC;SACF,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAoC,EAAE,MAAoB,EAAE,OAAkD;QAC1H,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,UAAU,CAAC,mCAAmC,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAClI,MAAM,EAAE,GAAG,OAAO,UAAU,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC;QACrD,MAAM,KAAK,GAAmC;YAC5C,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,EAAE;YACvE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzE,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QAClF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,EAAE,cAAc,EAAE,GAAG,EAAE,GAAG,QAAQ,EAAE,WAAW,EAAE,CAAC;YACjE,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC;YACvI,OAAO,MAAM,CAAC;QAChB,CAAC;gBAAS,CAAC;YACT,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;YACtB,IAAI,CAAC;gBAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAAC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACxD,IAAK,KAA2B,CAAC,IAAI,KAAK,kBAAkB;oBAAE,MAAM,KAAK,CAAC;YAC5E,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAE/C,KAAK,CAAC,IAAI,CAAC,SAAkB,EAAE,KAAK,GAAG,KAAK;QAC1C,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,IAAI,CAAC,aAAa;YAAE,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,MAAM,IAAI,CAAC,gBAAgB,CAAC;QAC5B,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,GAAW,EACX,KAAyB,EACzB,IAAqC,EACrC,SAAoC,EACpC,KAAqC;QAErC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,QAAQ,IAAI,QAAQ,CAAC,SAAS,KAAK,KAAK,CAAC,UAAU,IAAI,CAAC,QAAQ,CAAC,WAAW;YAAE,OAAO,QAAQ,CAAC;QAClG,IAAI,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACxD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;gBACpB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC;oBACtC,eAAe,EAAE,KAAK,CAAC,QAAQ;oBAC/B,WAAW,EAAE,KAAK,CAAC,YAAY;oBAC/B,SAAS,EAAE,KAAK,CAAC,UAAU;oBAC3B,IAAI;oBACJ,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACzD,CAAC,CAAC;gBACH,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;oBACrB,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;oBACtB,MAAM,IAAI,UAAU,CAAC,oDAAoD,EAAE,EAAE,IAAI,EAAE,6BAA6B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;gBACpI,CAAC;gBACD,MAAM,OAAO,GAAmB,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,UAAU,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;gBAClI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;gBAChC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAChD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACnD,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,OAAO;gBAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACtJ,CAAC;QACD,IAAI,CAAC;YAAC,OAAO,MAAM,OAAO,CAAC;QAAC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAC3C,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;YACtB,IAAI,CAAC;gBAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,+DAA+D,CAAC,CAAC;YAC/H,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAAC,OAAuB;QACzD,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,OAAO;YAAE,OAAO;QACvD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,OAAO,CAAC,SAAS;YAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACxG,MAAM,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC9C,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC;IAC/B,CAAC;CACF;AAED,KAAK,UAAU,sBAAsB,CACnC,IAAY,EACZ,KAAa,EACb,KAAuD,EACvD,QAAiC;IAEjC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,sCAAsC,CAAC,CAAC;IACnG,IAAI,KAAK,IAAI,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;WACnD,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;QACnF,MAAM,IAAI,SAAS,CAAC,6CAA6C,CAAC,CAAC;IACrE,CAAC;IACD,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,KAAK;QACrC,CAAC,CAAC,KAAK,CAAC,QAAQ;YACd,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,WAAW,CAAC;YACzF,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC;QACjE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC;IAC1D,MAAM,SAAS,GAAG;QACd,cAAc,EAAE,QAAQ,CAAC,sBAAsB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK;QAC1E,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrG,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW;QACrF,GAAG,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC,CAAC,EAAE,sBAAsB,EAAE,QAAQ,CAAC,sBAAsB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvG,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACxC,CAAC;IACF,MAAM,KAAK,GAA6B,CAAC,CAAC,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,qBAAqB,EAAE,QAAQ,CAAC,KAAK,CAAC;QAC5G,CAAC,uBAAuB,EAAE,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC,CAAC;IAC9E,MAAM,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,uBAAuB,EAAE,UAAU,CAAC,EAAE,kBAAkB,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,IAAI,EAAE;QACrH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAiC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9G,IAAI,SAAS,CAAC,cAAc,KAAK,GAAG,IAAI,QAAQ,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YACnH,wEAAwE;YACxE,yEAAyE;YACzE,yEAAyE;YACzE,IAAI,OAAO,QAAQ,CAAC,WAAW,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;mBAC7F,UAAU,CAAC,QAAQ,CAAC,KAAK,UAAU,CAAC,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBAC9F,MAAM,IAAI,UAAU,CAAC,uDAAuD,EAAE,EAAE,IAAI,EAAE,8BAA8B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACxI,CAAC;YACD,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/C,IAAI,UAAU,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,KAAK,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;oBACvF,MAAM,IAAI,UAAU,CAAC,mDAAmD,EAAE,EAAE,IAAI,EAAE,8BAA8B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;gBACpI,CAAC;YACH,CAAC;YACD,OAAO;QACT,CAAC;QACD,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IACtG,CAAC,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import path from "node:path";
|
|
2
|
+
import { HitchError, atomicWriteJSON, readJSON, sha256JSON, statePaths } from "../foundation/index.js";
|
|
3
|
+
function receipt(input) {
|
|
4
|
+
return { schema_version: "2", service_id: input.serviceId, epoch: input.epoch, inference_id: input.inferenceId,
|
|
5
|
+
isolation_key: input.isolationKey, owner_id: input.ownerId,
|
|
6
|
+
scope_digest: sha256JSON({ cache_scope_owner: input.cacheScopeOwner, evidence_owner: input.evidenceOwner ?? null }),
|
|
7
|
+
target_digest: sha256JSON({ binding: input.binding, credential: input.credential }) };
|
|
8
|
+
}
|
|
9
|
+
function file(root, input) {
|
|
10
|
+
if (!/^inference_[a-f0-9]{32}$/.test(input.serviceId) || !/^run_[a-f0-9]{32}$/.test(input.ownerId))
|
|
11
|
+
throw ambiguous();
|
|
12
|
+
return path.join(statePaths(root).inferenceServices, input.serviceId, "gateways", receipt(input).target_digest.slice(7) + ".json");
|
|
13
|
+
}
|
|
14
|
+
export async function sealManagedGateway(root, input) {
|
|
15
|
+
const expected = receipt(input), previous = await readJSON(file(root, input), null);
|
|
16
|
+
if (previous && sha256JSON(previous) !== sha256JSON(expected))
|
|
17
|
+
throw ambiguous();
|
|
18
|
+
if (!previous)
|
|
19
|
+
await atomicWriteJSON(file(root, input), expected);
|
|
20
|
+
}
|
|
21
|
+
export async function verifyManagedGateway(root, input) {
|
|
22
|
+
if (sha256JSON(await readJSON(file(root, input), null)) !== sha256JSON(receipt(input)))
|
|
23
|
+
throw ambiguous();
|
|
24
|
+
}
|
|
25
|
+
function ambiguous() { return new HitchError("original private gateway receipt is missing or changed", { code: "inference_recovery_ambiguous", exitCode: 12 }); }
|
|
26
|
+
//# sourceMappingURL=managed-gateway-receipt.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"managed-gateway-receipt.js","sourceRoot":"","sources":["../../../src/control-plane/managed-gateway-receipt.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAOvG,SAAS,OAAO,CAAC,KAAiC;IAChD,OAAO,EAAE,cAAc,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,CAAC,SAAS,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,CAAC,WAAW;QAC5G,aAAa,EAAE,KAAK,CAAC,YAAY,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO;QAC1D,YAAY,EAAE,UAAU,CAAC,EAAE,iBAAiB,EAAE,KAAK,CAAC,eAAe,EAAE,cAAc,EAAE,KAAK,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACnH,aAAa,EAAE,UAAU,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,EAAE,CAAC;AAC1F,CAAC;AACD,SAAS,IAAI,CAAC,IAAY,EAAE,KAAiC;IAC3D,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;QAAE,MAAM,SAAS,EAAE,CAAC;IACtH,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,iBAAiB,EAAE,KAAK,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC;AACrI,CAAC;AACD,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,IAAY,EAAE,KAAiC;IACtF,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,GAAG,MAAM,QAAQ,CAAiB,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;IACpG,IAAI,QAAQ,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,UAAU,CAAC,QAAQ,CAAC;QAAE,MAAM,SAAS,EAAE,CAAC;IACjF,IAAI,CAAC,QAAQ;QAAE,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,QAAQ,CAAC,CAAC;AACpE,CAAC;AACD,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,IAAY,EAAE,KAAiC;IACxF,IAAI,UAAU,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC,KAAK,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAAE,MAAM,SAAS,EAAE,CAAC;AAC5G,CAAC;AACD,SAAS,SAAS,KAAK,OAAO,IAAI,UAAU,CAAC,wDAAwD,EAAE,EAAE,IAAI,EAAE,8BAA8B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,158 @@
|
|
|
1
|
+
import { readdir } from "node:fs/promises";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { parseExecutionLease, parsePhysicalExecution, readExecutionLeases } from "../evals/index.js";
|
|
4
|
+
import { HitchError, readJSON, sha256JSON, statePaths } from "../foundation/index.js";
|
|
5
|
+
import { validateInferenceLockShape } from "../inference/index.js";
|
|
6
|
+
import { remoteModelBinding } from "../model-access/index.js";
|
|
7
|
+
import { parseEvalControl } from "./eval-records.js";
|
|
8
|
+
import { RemoteWorkInputStore } from "./remote-work-inputs.js";
|
|
9
|
+
import { verifyManagedGateway } from "./managed-gateway-receipt.js";
|
|
10
|
+
import { orderedEvalCancelled } from "./ordered-eval-control.js";
|
|
11
|
+
/** Private routes already seal the original gateway credentials before dispatch. */
|
|
12
|
+
export class ManagedServiceRecovery {
|
|
13
|
+
root;
|
|
14
|
+
registry;
|
|
15
|
+
protocol;
|
|
16
|
+
constructor(root, registry, protocol) {
|
|
17
|
+
this.root = root;
|
|
18
|
+
this.registry = registry;
|
|
19
|
+
this.protocol = protocol;
|
|
20
|
+
}
|
|
21
|
+
async select(records) {
|
|
22
|
+
const groups = new Map();
|
|
23
|
+
const evals = await readdir(statePaths(this.root).evals, { withFileTypes: true }).catch(error => {
|
|
24
|
+
if (error.code === "ENOENT")
|
|
25
|
+
return [];
|
|
26
|
+
throw error;
|
|
27
|
+
});
|
|
28
|
+
const inputs = new RemoteWorkInputStore(this.root);
|
|
29
|
+
for (const directory of evals) {
|
|
30
|
+
if (!directory.isDirectory() || !/^eval_[a-f0-9]{32}$/.test(directory.name))
|
|
31
|
+
continue;
|
|
32
|
+
const evalDirectory = path.join(statePaths(this.root).evals, directory.name);
|
|
33
|
+
for (const lease of await readExecutionLeases(evalDirectory)) {
|
|
34
|
+
if (lease.provider === "local-docker" || !["offered", "accepted", "running"].includes(lease.state) || Date.parse(lease.expires_at) <= Date.now())
|
|
35
|
+
continue;
|
|
36
|
+
const offer = await this.protocol.findOfferForLease(lease.worker_id, lease.lease_id);
|
|
37
|
+
if (!offer || offer.state !== "accepted")
|
|
38
|
+
continue;
|
|
39
|
+
const worker = await this.registry.get(offer.worker_id);
|
|
40
|
+
if (!worker || worker.revoked_at || worker.generation !== offer.generation)
|
|
41
|
+
continue;
|
|
42
|
+
const ref = offer.inputs?.find(item => item.kind === "work-spec");
|
|
43
|
+
if (!ref)
|
|
44
|
+
continue;
|
|
45
|
+
const spec = await readJSON((await inputs.verify(ref)).path);
|
|
46
|
+
if (spec.model_binding?.kind !== "managed-inference")
|
|
47
|
+
continue;
|
|
48
|
+
const target = await this.protocol.modelRoutes.recoveryTarget(offer);
|
|
49
|
+
if (!target || target.kind !== "managed-inference" || target.binding.kind !== "managed-node"
|
|
50
|
+
|| spec.schema_version !== "2" || sha256JSON(spec.work) !== sha256JSON(offer.work)
|
|
51
|
+
|| sha256JSON(spec.model_binding) !== sha256JSON(remoteModelBinding(target)))
|
|
52
|
+
throw ambiguous("managed route differs from its sealed work input");
|
|
53
|
+
const url = new URL(target.binding.base_url), match = url.pathname.match(/^\/runs\/(run_[a-f0-9]{32})\/v1\/$/);
|
|
54
|
+
if (!match || url.protocol !== "http:" || url.hostname !== "127.0.0.1" || !url.port || url.username || url.password || url.search || url.hash
|
|
55
|
+
|| !/^[a-f0-9]{64}$/.test(target.credential))
|
|
56
|
+
throw ambiguous("original private gateway address or credential is invalid");
|
|
57
|
+
const ownerId = match[1];
|
|
58
|
+
const physical = spec.physical_execution ? parsePhysicalExecution(spec.physical_execution) : undefined;
|
|
59
|
+
if (physical?.kind === "verifier-only")
|
|
60
|
+
throw ambiguous("verifier work cannot recover a model gateway");
|
|
61
|
+
const rerunId = physical && physical.kind !== "physical-infrastructure-retry" ? physical.rerun_id : undefined;
|
|
62
|
+
if (ownerId !== `run_${(rerunId ?? lease.eval_id).slice((rerunId ? "rerun_" : "eval_").length)}`)
|
|
63
|
+
throw ambiguous("gateway owner differs from its original eval or rerun");
|
|
64
|
+
const cacheScopeOwner = rerunId ? `${lease.eval_id}:${rerunId}` : lease.eval_id;
|
|
65
|
+
const evidenceOwner = { kind: "eval", eval_id: lease.eval_id, ...(rerunId ? { rerun_id: rerunId } : {}) };
|
|
66
|
+
if (await this.cancelled(evidenceOwner))
|
|
67
|
+
continue;
|
|
68
|
+
const candidates = records.filter(record => record.inference_id === target.binding.inference_id && record.lease_owner_ids.includes(ownerId)
|
|
69
|
+
&& sha256JSON(record.model_node ?? null) === sha256JSON(target.binding.model_node));
|
|
70
|
+
if (candidates.length !== 1 || candidates[0].state !== "ready")
|
|
71
|
+
throw ambiguous("accepted model work has no unique original ready service");
|
|
72
|
+
const record = candidates[0];
|
|
73
|
+
await verifyManagedGateway(this.root, { serviceId: record.service_id, epoch: record.epoch, inferenceId: record.inference_id,
|
|
74
|
+
isolationKey: record.isolation_key, ownerId, cacheScopeOwner, evidenceOwner, binding: target.binding, credential: target.credential });
|
|
75
|
+
const scopeDirectory = rerunId ? path.join(evalDirectory, "reruns", rerunId) : evalDirectory;
|
|
76
|
+
const lock = validateInferenceLockShape(await readJSON(path.join(scopeDirectory, "inference", "lock.json")));
|
|
77
|
+
const execution = await readJSON(path.join(scopeDirectory, "inference", "execution.json"));
|
|
78
|
+
if (record.inference_id !== lock.inference_id || lock.model_id !== target.modelId
|
|
79
|
+
|| lock.protocol.api !== target.binding.api || lock.generation.max_output_tokens !== target.maxOutputTokens
|
|
80
|
+
|| sha256JSON(lock.model_node ?? null) !== sha256JSON(record.model_node)
|
|
81
|
+
|| record.isolation_key !== sha256JSON({ inference_id: lock.inference_id, cache_scope_owner: cacheScopeOwner })
|
|
82
|
+
|| execution.schema_version !== "2" || execution.run_id !== ownerId || execution.eval_id !== lease.eval_id || execution.rerun_id !== rerunId
|
|
83
|
+
|| sha256JSON(execution.service) !== sha256JSON({ service_id: record.service_id, epoch: record.epoch, isolation_key: record.isolation_key })) {
|
|
84
|
+
throw ambiguous("original inference execution evidence differs from the active gateway");
|
|
85
|
+
}
|
|
86
|
+
const key = `${record.service_id}:${ownerId}`;
|
|
87
|
+
const prior = groups.get(key);
|
|
88
|
+
if (prior) {
|
|
89
|
+
if (sha256JSON(prior.plan.target) !== sha256JSON(target))
|
|
90
|
+
throw ambiguous("one model owner has conflicting private gateways");
|
|
91
|
+
prior.seeds.push(offer);
|
|
92
|
+
}
|
|
93
|
+
else {
|
|
94
|
+
const seeds = [offer];
|
|
95
|
+
const plan = { record, ownerId, lock, cacheScopeOwner, evidenceOwner, target, port: Number(url.port),
|
|
96
|
+
state: () => this.state(seeds, target, evidenceOwner) };
|
|
97
|
+
groups.set(key, { plan, seeds });
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
const plans = [];
|
|
102
|
+
for (const { plan } of groups.values())
|
|
103
|
+
if (await plan.state() !== "finished")
|
|
104
|
+
plans.push(plan);
|
|
105
|
+
return plans;
|
|
106
|
+
}
|
|
107
|
+
async cancelled(owner) {
|
|
108
|
+
if (await orderedEvalCancelled(this.root, owner.eval_id, owner.rerun_id))
|
|
109
|
+
return true;
|
|
110
|
+
const directory = path.join(statePaths(this.root).evals, owner.eval_id);
|
|
111
|
+
if (owner.rerun_id) {
|
|
112
|
+
const state = await readJSON(path.join(directory, "reruns", owner.rerun_id, "state.json"));
|
|
113
|
+
if (state.schema_version !== "1" || state.eval_id !== owner.eval_id || state.rerun_id !== owner.rerun_id
|
|
114
|
+
|| !["queued", "running", "completed", "failed", "cancelled"].includes(String(state.status)))
|
|
115
|
+
throw ambiguous("rerun state identity changed");
|
|
116
|
+
const marker = await readJSON(path.join(directory, "reruns", owner.rerun_id, "cancellation.json"), null);
|
|
117
|
+
if (marker && (marker.schema_version !== "1" || marker.eval_id !== owner.eval_id || marker.rerun_id !== owner.rerun_id))
|
|
118
|
+
throw ambiguous("rerun cancellation identity changed");
|
|
119
|
+
return !!marker || ["completed", "failed", "cancelled"].includes(String(state.status));
|
|
120
|
+
}
|
|
121
|
+
const control = parseEvalControl(await readJSON(path.join(directory, "control.json")));
|
|
122
|
+
if (control.eval_id !== owner.eval_id)
|
|
123
|
+
throw ambiguous("eval control identity changed");
|
|
124
|
+
return !!control.cancel_requested_at || ["cancelling", "cancelled", "failed", "succeeded"].includes(control.state);
|
|
125
|
+
}
|
|
126
|
+
async state(seeds, target, owner) {
|
|
127
|
+
if (await this.cancelled(owner))
|
|
128
|
+
return "finished";
|
|
129
|
+
let waiting = false;
|
|
130
|
+
for (const seed of seeds) {
|
|
131
|
+
const offer = await this.protocol.getOffer(seed.worker_id, seed.offer_id);
|
|
132
|
+
if (!offer)
|
|
133
|
+
throw ambiguous("accepted model offer disappeared during recovery");
|
|
134
|
+
if (sha256JSON(identity(offer)) !== sha256JSON(identity(seed)))
|
|
135
|
+
throw ambiguous("accepted model offer identity changed");
|
|
136
|
+
if (offer.state !== "accepted")
|
|
137
|
+
continue;
|
|
138
|
+
const lease = parseExecutionLease(await readJSON(path.join(statePaths(this.root).evals, owner.eval_id, "leases", `${offer.lease.lease_id}.json`)));
|
|
139
|
+
if (lease.epoch !== offer.lease.epoch || !["offered", "accepted", "running"].includes(lease.state) || Date.parse(lease.expires_at) <= Date.now())
|
|
140
|
+
continue;
|
|
141
|
+
const worker = await this.registry.get(offer.worker_id);
|
|
142
|
+
if (!worker || worker.revoked_at || worker.generation !== offer.generation)
|
|
143
|
+
continue;
|
|
144
|
+
await this.protocol.modelRoutes.authorize(offer);
|
|
145
|
+
if (sha256JSON(await this.protocol.modelRoutes.recoveryTarget(offer)) !== sha256JSON(target))
|
|
146
|
+
throw ambiguous("private model route changed during recovery");
|
|
147
|
+
if (worker.worker.status === "ready")
|
|
148
|
+
return "active";
|
|
149
|
+
waiting = true;
|
|
150
|
+
}
|
|
151
|
+
return waiting ? "waiting" : "finished";
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
function identity(offer) {
|
|
155
|
+
return { worker: offer.worker_id, generation: offer.generation, offer: offer.offer_id, nonce: offer.nonce, lease: offer.lease, work: offer.work, inputs: offer.inputs };
|
|
156
|
+
}
|
|
157
|
+
function ambiguous(message) { return new HitchError(message, { code: "inference_recovery_ambiguous", exitCode: 12 }); }
|
|
158
|
+
//# sourceMappingURL=managed-service-recovery.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"managed-service-recovery.js","sourceRoot":"","sources":["../../../src/control-plane/managed-service-recovery.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACrG,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACtF,OAAO,EAAE,0BAA0B,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAE9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAGrD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAejE,oFAAoF;AACpF,MAAM,OAAO,sBAAsB;IACZ;IAAuB;IAAyC;IAArF,YAAqB,IAAY,EAAW,QAA8B,EAAW,QAA8B;QAA9F,SAAI,GAAJ,IAAI,CAAQ;QAAW,aAAQ,GAAR,QAAQ,CAAsB;QAAW,aAAQ,GAAR,QAAQ,CAAsB;IAAG,CAAC;IAEvH,KAAK,CAAC,MAAM,CAAC,OAAmC;QAC9C,MAAM,MAAM,GAAG,IAAI,GAAG,EAA4E,CAAC;QACnG,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;YAC9F,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,EAAE,CAAC;YAAC,MAAM,KAAK,CAAC;QACjF,CAAC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,KAAK,MAAM,SAAS,IAAI,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAAE,SAAS;YACtF,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;YAC7E,KAAK,MAAM,KAAK,IAAI,MAAM,mBAAmB,CAAC,aAAa,CAAC,EAAE,CAAC;gBAC7D,IAAI,KAAK,CAAC,QAAQ,KAAK,cAAc,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE;oBAAE,SAAS;gBAC3J,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACrF,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,KAAK,UAAU;oBAAE,SAAS;gBACnD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBACxD,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,UAAU;oBAAE,SAAS;gBACrF,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;gBAAC,IAAI,CAAC,GAAG;oBAAE,SAAS;gBACtF,MAAM,IAAI,GAAG,MAAM,QAAQ,CAA0B,CAAC,MAAM,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACtF,IAAK,IAAI,CAAC,aAAgD,EAAE,IAAI,KAAK,mBAAmB;oBAAE,SAAS;gBACnG,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;gBACrE,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,mBAAmB,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,KAAK,cAAc;uBACvF,IAAI,CAAC,cAAc,KAAK,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC;uBAC/E,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,UAAU,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;oBAAE,MAAM,SAAS,CAAC,kDAAkD,CAAC,CAAC;gBACpJ,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;gBAC/G,IAAI,CAAC,KAAK,IAAI,GAAG,CAAC,QAAQ,KAAK,OAAO,IAAI,GAAG,CAAC,QAAQ,KAAK,WAAW,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,IAAI;uBACxI,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;oBAAE,MAAM,SAAS,CAAC,2DAA2D,CAAC,CAAC;gBAC7H,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACvG,IAAI,QAAQ,EAAE,IAAI,KAAK,eAAe;oBAAE,MAAM,SAAS,CAAC,8CAA8C,CAAC,CAAC;gBACxG,MAAM,OAAO,GAAG,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,+BAA+B,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC9G,IAAI,OAAO,KAAK,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,EAAE;oBAAE,MAAM,SAAS,CAAC,uDAAuD,CAAC,CAAC;gBAC3K,MAAM,eAAe,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,OAAO,IAAI,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;gBAChF,MAAM,aAAa,GAAG,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;gBACnH,IAAI,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;oBAAE,SAAS;gBAClD,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,KAAK,MAAM,CAAC,OAAO,CAAC,YAAY,IAAI,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC;uBACtI,UAAU,CAAC,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,KAAK,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;gBACtF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,CAAC,CAAE,CAAC,KAAK,KAAK,OAAO;oBAAE,MAAM,SAAS,CAAC,0DAA0D,CAAC,CAAC;gBAC7I,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;gBAC9B,MAAM,oBAAoB,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,YAAY;oBACzH,YAAY,EAAE,MAAM,CAAC,aAAa,EAAE,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;gBACzI,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC7F,MAAM,IAAI,GAAG,0BAA0B,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;gBAC7G,MAAM,SAAS,GAAG,MAAM,QAAQ,CAA0B,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC,CAAC;gBACpH,IAAI,MAAM,CAAC,YAAY,KAAK,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,OAAO;uBAC5E,IAAI,CAAC,QAAQ,CAAC,GAAG,KAAK,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,iBAAiB,KAAK,MAAM,CAAC,eAAe;uBACxG,UAAU,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,KAAK,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC;uBACrE,MAAM,CAAC,aAAa,KAAK,UAAU,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,iBAAiB,EAAE,eAAe,EAAE,CAAC;uBAC5G,SAAS,CAAC,cAAc,KAAK,GAAG,IAAI,SAAS,CAAC,MAAM,KAAK,OAAO,IAAI,SAAS,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC,QAAQ,KAAK,OAAO;uBACzI,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,UAAU,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,EAAE,CAAC;oBAC/I,MAAM,SAAS,CAAC,uEAAuE,CAAC,CAAC;gBAC3F,CAAC;gBACD,MAAM,GAAG,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,OAAO,EAAE,CAAC;gBAC9C,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAC9B,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,UAAU,CAAC,MAAM,CAAC;wBAAE,MAAM,SAAS,CAAC,kDAAkD,CAAC,CAAC;oBAC9H,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC1B,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;oBACtB,MAAM,IAAI,GAA+B,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;wBAC9H,KAAK,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,aAAa,CAAC,EAAE,CAAC;oBAC1D,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;QACH,CAAC;QACD,MAAM,KAAK,GAAiC,EAAE,CAAC;QAC/C,KAAK,MAAM,EAAE,IAAI,EAAE,IAAI,MAAM,CAAC,MAAM,EAAE;YAAE,IAAI,MAAM,IAAI,CAAC,KAAK,EAAE,KAAK,UAAU;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChG,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,KAAkD;QACxE,IAAI,MAAM,oBAAoB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC;YAAE,OAAO,IAAI,CAAC;QACtF,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACxE,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAA0B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC;YACpH,IAAI,KAAK,CAAC,cAAc,KAAK,GAAG,IAAI,KAAK,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ;mBACnG,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAAE,MAAM,SAAS,CAAC,8BAA8B,CAAC,CAAC;YAChJ,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAiC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,mBAAmB,CAAC,EAAE,IAAI,CAAC,CAAC;YACzI,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,KAAK,GAAG,IAAI,MAAM,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO,IAAI,MAAM,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ,CAAC;gBAAE,MAAM,SAAS,CAAC,qCAAqC,CAAC,CAAC;YAChL,OAAO,CAAC,CAAC,MAAM,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QACzF,CAAC;QACD,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC;QACvF,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO;YAAE,MAAM,SAAS,CAAC,+BAA+B,CAAC,CAAC;QACxF,OAAO,CAAC,CAAC,OAAO,CAAC,mBAAmB,IAAI,CAAC,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACrH,CAAC;IAEO,KAAK,CAAC,KAAK,CAAC,KAA0B,EAAE,MAAc,EAAE,KAAkD;QAChH,IAAI,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YAAE,OAAO,UAAU,CAAC;QACnD,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1E,IAAI,CAAC,KAAK;gBAAE,MAAM,SAAS,CAAC,kDAAkD,CAAC,CAAC;YAChF,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,KAAK,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAAE,MAAM,SAAS,CAAC,uCAAuC,CAAC,CAAC;YACzH,IAAI,KAAK,CAAC,KAAK,KAAK,UAAU;gBAAE,SAAS;YACzC,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,QAAQ,OAAO,CAAC,CAAC,CAAC,CAAC;YACnJ,IAAI,KAAK,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE;gBAAE,SAAS;YAC3J,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YACxD,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,UAAU;gBAAE,SAAS;YACrF,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACjD,IAAI,UAAU,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,KAAK,UAAU,CAAC,MAAM,CAAC;gBAAE,MAAM,SAAS,CAAC,6CAA6C,CAAC,CAAC;YAC7J,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,OAAO;gBAAE,OAAO,QAAQ,CAAC;YACtD,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC;QACD,OAAO,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC;IAC1C,CAAC;CACF;AACD,SAAS,QAAQ,CAAC,KAAwB;IACxC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,SAAS,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;AAC1K,CAAC;AACD,SAAS,SAAS,CAAC,OAAe,IAAI,OAAO,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,8BAA8B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC"}
|
|
@@ -1,12 +1,26 @@
|
|
|
1
1
|
import { getAdapter } from "../adapters/index.js";
|
|
2
2
|
import { planModelCapture } from "../model-access/index.js";
|
|
3
|
+
import { forceLocalInferenceCapturePlan } from "../evals/index.js";
|
|
4
|
+
import { HitchError } from "../foundation/index.js";
|
|
3
5
|
export function modelCapturePlanForEval(request, execution, provider) {
|
|
4
6
|
const separator = request.harness_ref.indexOf("@");
|
|
5
7
|
const harnessId = separator < 0 ? request.harness_ref : request.harness_ref.slice(0, separator);
|
|
6
|
-
|
|
8
|
+
const plan = planModelCapture({
|
|
7
9
|
policy: execution.model_capture,
|
|
8
10
|
adapter: getAdapter(harnessId).requirements,
|
|
9
11
|
provider,
|
|
10
12
|
});
|
|
13
|
+
if (request.local_inference || request.training_binding) {
|
|
14
|
+
const remote = execution.provider !== "local-docker";
|
|
15
|
+
const feature = request.training_binding ? "training_external_binding" : "managed_model_node";
|
|
16
|
+
if (remote && (!provider.features.model_proxy || provider.features[feature] !== "2"
|
|
17
|
+
|| request.local_inference && (!request.local_inference.model_node || !request.local_inference.inference_id))) {
|
|
18
|
+
throw new HitchError(`remote provider requires ${feature}=2 and an immutable model route`, {
|
|
19
|
+
code: "remote_model_binding_unsupported", exitCode: 10,
|
|
20
|
+
});
|
|
21
|
+
}
|
|
22
|
+
return forceLocalInferenceCapturePlan(plan, remote ? "in-sandbox" : "host-side");
|
|
23
|
+
}
|
|
24
|
+
return plan;
|
|
11
25
|
}
|
|
12
26
|
//# sourceMappingURL=model-capture-planning.js.map
|