@m4ike1/ion-agent-core 0.1.0
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 +622 -0
- package/dist/agent-loop.d.ts +24 -0
- package/dist/agent-loop.d.ts.map +1 -0
- package/dist/agent-loop.js +560 -0
- package/dist/agent-loop.js.map +1 -0
- package/dist/agent.d.ts +122 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +422 -0
- package/dist/agent.js.map +1 -0
- package/dist/harness/agent-harness.d.ts +716 -0
- package/dist/harness/agent-harness.d.ts.map +1 -0
- package/dist/harness/agent-harness.js +6 -0
- package/dist/harness/agent-harness.js.map +1 -0
- package/dist/harness/compaction/branch-summarization.d.ts +68 -0
- package/dist/harness/compaction/branch-summarization.d.ts.map +1 -0
- package/dist/harness/compaction/branch-summarization.js +176 -0
- package/dist/harness/compaction/branch-summarization.js.map +1 -0
- package/dist/harness/compaction/compaction.d.ts +126 -0
- package/dist/harness/compaction/compaction.d.ts.map +1 -0
- package/dist/harness/compaction/compaction.js +560 -0
- package/dist/harness/compaction/compaction.js.map +1 -0
- package/dist/harness/compaction/utils.d.ts +25 -0
- package/dist/harness/compaction/utils.d.ts.map +1 -0
- package/dist/harness/compaction/utils.js +120 -0
- package/dist/harness/compaction/utils.js.map +1 -0
- package/dist/harness/config.d.ts +9 -0
- package/dist/harness/config.d.ts.map +1 -0
- package/dist/harness/config.js +35 -0
- package/dist/harness/config.js.map +1 -0
- package/dist/harness/context.d.ts +9 -0
- package/dist/harness/context.d.ts.map +1 -0
- package/dist/harness/context.js +13 -0
- package/dist/harness/context.js.map +1 -0
- package/dist/harness/env/nodejs.d.ts +43 -0
- package/dist/harness/env/nodejs.d.ts.map +1 -0
- package/dist/harness/env/nodejs.js +865 -0
- package/dist/harness/env/nodejs.js.map +1 -0
- package/dist/harness/events.d.ts +25 -0
- package/dist/harness/events.d.ts.map +1 -0
- package/dist/harness/events.js +250 -0
- package/dist/harness/events.js.map +1 -0
- package/dist/harness/execution/assistant.d.ts +42 -0
- package/dist/harness/execution/assistant.d.ts.map +1 -0
- package/dist/harness/execution/assistant.js +82 -0
- package/dist/harness/execution/assistant.js.map +1 -0
- package/dist/harness/execution/effect-gate.d.ts +22 -0
- package/dist/harness/execution/effect-gate.d.ts.map +1 -0
- package/dist/harness/execution/effect-gate.js +49 -0
- package/dist/harness/execution/effect-gate.js.map +1 -0
- package/dist/harness/execution/tools.d.ts +67 -0
- package/dist/harness/execution/tools.d.ts.map +1 -0
- package/dist/harness/execution/tools.js +121 -0
- package/dist/harness/execution/tools.js.map +1 -0
- package/dist/harness/hooks.d.ts +38 -0
- package/dist/harness/hooks.d.ts.map +1 -0
- package/dist/harness/hooks.js +406 -0
- package/dist/harness/hooks.js.map +1 -0
- package/dist/harness/messages.d.ts +51 -0
- package/dist/harness/messages.d.ts.map +1 -0
- package/dist/harness/messages.js +102 -0
- package/dist/harness/messages.js.map +1 -0
- package/dist/harness/prompt-templates.d.ts +49 -0
- package/dist/harness/prompt-templates.d.ts.map +1 -0
- package/dist/harness/prompt-templates.js +225 -0
- package/dist/harness/prompt-templates.js.map +1 -0
- package/dist/harness/result.d.ts +133 -0
- package/dist/harness/result.d.ts.map +1 -0
- package/dist/harness/result.js +81 -0
- package/dist/harness/result.js.map +1 -0
- package/dist/harness/runtime/drive/boundary.d.ts +28 -0
- package/dist/harness/runtime/drive/boundary.d.ts.map +1 -0
- package/dist/harness/runtime/drive/boundary.js +173 -0
- package/dist/harness/runtime/drive/boundary.js.map +1 -0
- package/dist/harness/runtime/drive/checkpoint.d.ts +8 -0
- package/dist/harness/runtime/drive/checkpoint.d.ts.map +1 -0
- package/dist/harness/runtime/drive/checkpoint.js +133 -0
- package/dist/harness/runtime/drive/checkpoint.js.map +1 -0
- package/dist/harness/runtime/drive/deferred.d.ts +15 -0
- package/dist/harness/runtime/drive/deferred.d.ts.map +1 -0
- package/dist/harness/runtime/drive/deferred.js +179 -0
- package/dist/harness/runtime/drive/deferred.js.map +1 -0
- package/dist/harness/runtime/drive/generation.d.ts +8 -0
- package/dist/harness/runtime/drive/generation.d.ts.map +1 -0
- package/dist/harness/runtime/drive/generation.js +205 -0
- package/dist/harness/runtime/drive/generation.js.map +1 -0
- package/dist/harness/runtime/drive/reconcile.d.ts +5 -0
- package/dist/harness/runtime/drive/reconcile.d.ts.map +1 -0
- package/dist/harness/runtime/drive/reconcile.js +140 -0
- package/dist/harness/runtime/drive/reconcile.js.map +1 -0
- package/dist/harness/runtime/drive/recovery.d.ts +8 -0
- package/dist/harness/runtime/drive/recovery.d.ts.map +1 -0
- package/dist/harness/runtime/drive/recovery.js +85 -0
- package/dist/harness/runtime/drive/recovery.js.map +1 -0
- package/dist/harness/runtime/drive/response.d.ts +23 -0
- package/dist/harness/runtime/drive/response.d.ts.map +1 -0
- package/dist/harness/runtime/drive/response.js +394 -0
- package/dist/harness/runtime/drive/response.js.map +1 -0
- package/dist/harness/runtime/drive/retry.d.ts +4 -0
- package/dist/harness/runtime/drive/retry.d.ts.map +1 -0
- package/dist/harness/runtime/drive/retry.js +34 -0
- package/dist/harness/runtime/drive/retry.js.map +1 -0
- package/dist/harness/runtime/drive/structural.d.ts +32 -0
- package/dist/harness/runtime/drive/structural.d.ts.map +1 -0
- package/dist/harness/runtime/drive/structural.js +929 -0
- package/dist/harness/runtime/drive/structural.js.map +1 -0
- package/dist/harness/runtime/drive/terminal.d.ts +7 -0
- package/dist/harness/runtime/drive/terminal.d.ts.map +1 -0
- package/dist/harness/runtime/drive/terminal.js +49 -0
- package/dist/harness/runtime/drive/terminal.js.map +1 -0
- package/dist/harness/runtime/drive/tool-placement.d.ts +14 -0
- package/dist/harness/runtime/drive/tool-placement.d.ts.map +1 -0
- package/dist/harness/runtime/drive/tool-placement.js +244 -0
- package/dist/harness/runtime/drive/tool-placement.js.map +1 -0
- package/dist/harness/runtime/drive/tools.d.ts +6 -0
- package/dist/harness/runtime/drive/tools.d.ts.map +1 -0
- package/dist/harness/runtime/drive/tools.js +449 -0
- package/dist/harness/runtime/drive/tools.js.map +1 -0
- package/dist/harness/runtime/drive.d.ts +6 -0
- package/dist/harness/runtime/drive.d.ts.map +1 -0
- package/dist/harness/runtime/drive.js +93 -0
- package/dist/harness/runtime/drive.js.map +1 -0
- package/dist/harness/runtime/harness.d.ts +59 -0
- package/dist/harness/runtime/harness.d.ts.map +1 -0
- package/dist/harness/runtime/harness.js +311 -0
- package/dist/harness/runtime/harness.js.map +1 -0
- package/dist/harness/runtime/index.d.ts +2 -0
- package/dist/harness/runtime/index.d.ts.map +1 -0
- package/dist/harness/runtime/index.js +2 -0
- package/dist/harness/runtime/index.js.map +1 -0
- package/dist/harness/runtime/lane.d.ts +101 -0
- package/dist/harness/runtime/lane.d.ts.map +1 -0
- package/dist/harness/runtime/lane.js +1575 -0
- package/dist/harness/runtime/lane.js.map +1 -0
- package/dist/harness/runtime/progress.d.ts +15 -0
- package/dist/harness/runtime/progress.d.ts.map +1 -0
- package/dist/harness/runtime/progress.js +66 -0
- package/dist/harness/runtime/progress.js.map +1 -0
- package/dist/harness/runtime/reducer.d.ts +5 -0
- package/dist/harness/runtime/reducer.d.ts.map +1 -0
- package/dist/harness/runtime/reducer.js +240 -0
- package/dist/harness/runtime/reducer.js.map +1 -0
- package/dist/harness/runtime/restore.d.ts +24 -0
- package/dist/harness/runtime/restore.d.ts.map +1 -0
- package/dist/harness/runtime/restore.js +117 -0
- package/dist/harness/runtime/restore.js.map +1 -0
- package/dist/harness/runtime/transcript.d.ts +21 -0
- package/dist/harness/runtime/transcript.d.ts.map +1 -0
- package/dist/harness/runtime/transcript.js +72 -0
- package/dist/harness/runtime/transcript.js.map +1 -0
- package/dist/harness/runtime/types.d.ts +105 -0
- package/dist/harness/runtime/types.d.ts.map +1 -0
- package/dist/harness/runtime/types.js +61 -0
- package/dist/harness/runtime/types.js.map +1 -0
- package/dist/harness/session/commit.d.ts +53 -0
- package/dist/harness/session/commit.d.ts.map +1 -0
- package/dist/harness/session/commit.js +57 -0
- package/dist/harness/session/commit.js.map +1 -0
- package/dist/harness/session/context.d.ts +10 -0
- package/dist/harness/session/context.d.ts.map +1 -0
- package/dist/harness/session/context.js +49 -0
- package/dist/harness/session/context.js.map +1 -0
- package/dist/harness/session/fork-policy.d.ts +21 -0
- package/dist/harness/session/fork-policy.d.ts.map +1 -0
- package/dist/harness/session/fork-policy.js +55 -0
- package/dist/harness/session/fork-policy.js.map +1 -0
- package/dist/harness/session/fork.d.ts +16 -0
- package/dist/harness/session/fork.d.ts.map +1 -0
- package/dist/harness/session/fork.js +79 -0
- package/dist/harness/session/fork.js.map +1 -0
- package/dist/harness/session/in-memory-storage-state.d.ts +39 -0
- package/dist/harness/session/in-memory-storage-state.d.ts.map +1 -0
- package/dist/harness/session/in-memory-storage-state.js +276 -0
- package/dist/harness/session/in-memory-storage-state.js.map +1 -0
- package/dist/harness/session/index.d.ts +11 -0
- package/dist/harness/session/index.d.ts.map +1 -0
- package/dist/harness/session/index.js +9 -0
- package/dist/harness/session/index.js.map +1 -0
- package/dist/harness/session/jsonl/codec.d.ts +21 -0
- package/dist/harness/session/jsonl/codec.d.ts.map +1 -0
- package/dist/harness/session/jsonl/codec.js +45 -0
- package/dist/harness/session/jsonl/codec.js.map +1 -0
- package/dist/harness/session/jsonl/fork.d.ts +38 -0
- package/dist/harness/session/jsonl/fork.d.ts.map +1 -0
- package/dist/harness/session/jsonl/fork.js +264 -0
- package/dist/harness/session/jsonl/fork.js.map +1 -0
- package/dist/harness/session/jsonl/index.d.ts +4 -0
- package/dist/harness/session/jsonl/index.d.ts.map +1 -0
- package/dist/harness/session/jsonl/index.js +4 -0
- package/dist/harness/session/jsonl/index.js.map +1 -0
- package/dist/harness/session/jsonl/io.d.ts +14 -0
- package/dist/harness/session/jsonl/io.d.ts.map +1 -0
- package/dist/harness/session/jsonl/io.js +86 -0
- package/dist/harness/session/jsonl/io.js.map +1 -0
- package/dist/harness/session/jsonl/legacy-v3.d.ts +43 -0
- package/dist/harness/session/jsonl/legacy-v3.d.ts.map +1 -0
- package/dist/harness/session/jsonl/legacy-v3.js +462 -0
- package/dist/harness/session/jsonl/legacy-v3.js.map +1 -0
- package/dist/harness/session/jsonl/repo.d.ts +35 -0
- package/dist/harness/session/jsonl/repo.d.ts.map +1 -0
- package/dist/harness/session/jsonl/repo.js +301 -0
- package/dist/harness/session/jsonl/repo.js.map +1 -0
- package/dist/harness/session/jsonl/storage.d.ts +40 -0
- package/dist/harness/session/jsonl/storage.d.ts.map +1 -0
- package/dist/harness/session/jsonl/storage.js +200 -0
- package/dist/harness/session/jsonl/storage.js.map +1 -0
- package/dist/harness/session/jsonl/types.d.ts +39 -0
- package/dist/harness/session/jsonl/types.d.ts.map +1 -0
- package/dist/harness/session/jsonl/types.js +3 -0
- package/dist/harness/session/jsonl/types.js.map +1 -0
- package/dist/harness/session/memory.d.ts +48 -0
- package/dist/harness/session/memory.d.ts.map +1 -0
- package/dist/harness/session/memory.js +369 -0
- package/dist/harness/session/memory.js.map +1 -0
- package/dist/harness/session/mutation-line.d.ts +8 -0
- package/dist/harness/session/mutation-line.d.ts.map +1 -0
- package/dist/harness/session/mutation-line.js +21 -0
- package/dist/harness/session/mutation-line.js.map +1 -0
- package/dist/harness/session/session.d.ts +82 -0
- package/dist/harness/session/session.d.ts.map +1 -0
- package/dist/harness/session/session.js +361 -0
- package/dist/harness/session/session.js.map +1 -0
- package/dist/harness/session/testing/benchmark/datasets.d.ts +12 -0
- package/dist/harness/session/testing/benchmark/datasets.d.ts.map +1 -0
- package/dist/harness/session/testing/benchmark/datasets.js +24 -0
- package/dist/harness/session/testing/benchmark/datasets.js.map +1 -0
- package/dist/harness/session/testing/benchmark/session-repo.d.ts +42 -0
- package/dist/harness/session/testing/benchmark/session-repo.d.ts.map +1 -0
- package/dist/harness/session/testing/benchmark/session-repo.js +127 -0
- package/dist/harness/session/testing/benchmark/session-repo.js.map +1 -0
- package/dist/harness/session/testing/benchmark/storage.d.ts +25 -0
- package/dist/harness/session/testing/benchmark/storage.d.ts.map +1 -0
- package/dist/harness/session/testing/benchmark/storage.js +114 -0
- package/dist/harness/session/testing/benchmark/storage.js.map +1 -0
- package/dist/harness/session/testing/conformance/session-repo.d.ts +23 -0
- package/dist/harness/session/testing/conformance/session-repo.d.ts.map +1 -0
- package/dist/harness/session/testing/conformance/session-repo.js +809 -0
- package/dist/harness/session/testing/conformance/session-repo.js.map +1 -0
- package/dist/harness/session/testing/conformance/storage.d.ts +4 -0
- package/dist/harness/session/testing/conformance/storage.d.ts.map +1 -0
- package/dist/harness/session/testing/conformance/storage.js +600 -0
- package/dist/harness/session/testing/conformance/storage.js.map +1 -0
- package/dist/harness/session/testing/gating-storage.d.ts +26 -0
- package/dist/harness/session/testing/gating-storage.d.ts.map +1 -0
- package/dist/harness/session/testing/gating-storage.js +100 -0
- package/dist/harness/session/testing/gating-storage.js.map +1 -0
- package/dist/harness/session/testing/index.d.ts +10 -0
- package/dist/harness/session/testing/index.d.ts.map +1 -0
- package/dist/harness/session/testing/index.js +9 -0
- package/dist/harness/session/testing/index.js.map +1 -0
- package/dist/harness/session/testing/instrumented-storage.d.ts +11 -0
- package/dist/harness/session/testing/instrumented-storage.d.ts.map +1 -0
- package/dist/harness/session/testing/instrumented-storage.js +16 -0
- package/dist/harness/session/testing/instrumented-storage.js.map +1 -0
- package/dist/harness/session/testing/storage-decorator.d.ts +20 -0
- package/dist/harness/session/testing/storage-decorator.d.ts.map +1 -0
- package/dist/harness/session/testing/storage-decorator.js +41 -0
- package/dist/harness/session/testing/storage-decorator.js.map +1 -0
- package/dist/harness/session/testing/types.d.ts +12 -0
- package/dist/harness/session/testing/types.d.ts.map +1 -0
- package/dist/harness/session/testing/types.js +2 -0
- package/dist/harness/session/testing/types.js.map +1 -0
- package/dist/harness/session/types.d.ts +512 -0
- package/dist/harness/session/types.d.ts.map +1 -0
- package/dist/harness/session/types.js +9 -0
- package/dist/harness/session/types.js.map +1 -0
- package/dist/harness/session/values.d.ts +95 -0
- package/dist/harness/session/values.d.ts.map +1 -0
- package/dist/harness/session/values.js +81 -0
- package/dist/harness/session/values.js.map +1 -0
- package/dist/harness/skills.d.ts +46 -0
- package/dist/harness/skills.d.ts.map +1 -0
- package/dist/harness/skills.js +326 -0
- package/dist/harness/skills.js.map +1 -0
- package/dist/harness/system-prompt.d.ts +3 -0
- package/dist/harness/system-prompt.d.ts.map +1 -0
- package/dist/harness/system-prompt.js +30 -0
- package/dist/harness/system-prompt.js.map +1 -0
- package/dist/harness/telemetry.d.ts +1242 -0
- package/dist/harness/telemetry.d.ts.map +1 -0
- package/dist/harness/telemetry.js +527 -0
- package/dist/harness/telemetry.js.map +1 -0
- package/dist/harness/tools/bash.d.ts +27 -0
- package/dist/harness/tools/bash.d.ts.map +1 -0
- package/dist/harness/tools/bash.js +105 -0
- package/dist/harness/tools/bash.js.map +1 -0
- package/dist/harness/tools/edit-diff.d.ts +89 -0
- package/dist/harness/tools/edit-diff.d.ts.map +1 -0
- package/dist/harness/tools/edit-diff.js +386 -0
- package/dist/harness/tools/edit-diff.js.map +1 -0
- package/dist/harness/tools/edit.d.ts +19 -0
- package/dist/harness/tools/edit.d.ts.map +1 -0
- package/dist/harness/tools/edit.js +108 -0
- package/dist/harness/tools/edit.js.map +1 -0
- package/dist/harness/tools/file-mutation-queue.d.ts +5 -0
- package/dist/harness/tools/file-mutation-queue.d.ts.map +1 -0
- package/dist/harness/tools/file-mutation-queue.js +46 -0
- package/dist/harness/tools/file-mutation-queue.js.map +1 -0
- package/dist/harness/tools/image.d.ts +3 -0
- package/dist/harness/tools/image.d.ts.map +1 -0
- package/dist/harness/tools/image.js +106 -0
- package/dist/harness/tools/image.js.map +1 -0
- package/dist/harness/tools/index.d.ts +6 -0
- package/dist/harness/tools/index.d.ts.map +1 -0
- package/dist/harness/tools/index.js +5 -0
- package/dist/harness/tools/index.js.map +1 -0
- package/dist/harness/tools/path-utils.d.ts +5 -0
- package/dist/harness/tools/path-utils.d.ts.map +1 -0
- package/dist/harness/tools/path-utils.js +26 -0
- package/dist/harness/tools/path-utils.js.map +1 -0
- package/dist/harness/tools/read.d.ts +35 -0
- package/dist/harness/tools/read.d.ts.map +1 -0
- package/dist/harness/tools/read.js +108 -0
- package/dist/harness/tools/read.js.map +1 -0
- package/dist/harness/tools/tool-context.d.ts +6 -0
- package/dist/harness/tools/tool-context.d.ts.map +1 -0
- package/dist/harness/tools/tool-context.js +2 -0
- package/dist/harness/tools/tool-context.js.map +1 -0
- package/dist/harness/tools/write.d.ts +11 -0
- package/dist/harness/tools/write.d.ts.map +1 -0
- package/dist/harness/tools/write.js +31 -0
- package/dist/harness/tools/write.js.map +1 -0
- package/dist/harness/types.d.ts +305 -0
- package/dist/harness/types.d.ts.map +1 -0
- package/dist/harness/types.js +75 -0
- package/dist/harness/types.js.map +1 -0
- package/dist/harness/utils/adaptive-publisher.d.ts +24 -0
- package/dist/harness/utils/adaptive-publisher.d.ts.map +1 -0
- package/dist/harness/utils/adaptive-publisher.js +79 -0
- package/dist/harness/utils/adaptive-publisher.js.map +1 -0
- package/dist/harness/utils/output-capture.d.ts +31 -0
- package/dist/harness/utils/output-capture.d.ts.map +1 -0
- package/dist/harness/utils/output-capture.js +217 -0
- package/dist/harness/utils/output-capture.js.map +1 -0
- package/dist/harness/utils/shell-output.d.ts +29 -0
- package/dist/harness/utils/shell-output.d.ts.map +1 -0
- package/dist/harness/utils/shell-output.js +71 -0
- package/dist/harness/utils/shell-output.js.map +1 -0
- package/dist/harness/utils/truncate.d.ts +71 -0
- package/dist/harness/utils/truncate.d.ts.map +1 -0
- package/dist/harness/utils/truncate.js +296 -0
- package/dist/harness/utils/truncate.js.map +1 -0
- package/dist/harness/utils/usage.d.ts +4 -0
- package/dist/harness/utils/usage.d.ts.map +1 -0
- package/dist/harness/utils/usage.js +33 -0
- package/dist/harness/utils/usage.js.map +1 -0
- package/dist/index.d.ts +28 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +26 -0
- package/dist/index.js.map +1 -0
- package/dist/node.d.ts +3 -0
- package/dist/node.d.ts.map +1 -0
- package/dist/node.js +3 -0
- package/dist/node.js.map +1 -0
- package/dist/proxy.d.ts +72 -0
- package/dist/proxy.d.ts.map +1 -0
- package/dist/proxy.js +310 -0
- package/dist/proxy.js.map +1 -0
- package/dist/search/index.d.ts +29 -0
- package/dist/search/index.d.ts.map +1 -0
- package/dist/search/index.js +2 -0
- package/dist/search/index.js.map +1 -0
- package/dist/stream-fn.d.ts +10 -0
- package/dist/stream-fn.d.ts.map +1 -0
- package/dist/stream-fn.js +17 -0
- package/dist/stream-fn.js.map +1 -0
- package/dist/types.d.ts +416 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/package.json +89 -0
|
@@ -0,0 +1,140 @@
|
|
|
1
|
+
import { getTelemetryContext } from "../../context.js";
|
|
2
|
+
import { SessionInvariantError } from "../../session/session.js";
|
|
3
|
+
import { readDeferredSourceHandle } from "./deferred.js";
|
|
4
|
+
import { recoverCancelledAssistantEffect } from "./recovery.js";
|
|
5
|
+
import { operationCleanupWrites, operationResultRecord } from "./terminal.js";
|
|
6
|
+
import { runTools } from "./tools.js";
|
|
7
|
+
async function cancelDeferredBestEffort(lane, drive, deferred, handle) {
|
|
8
|
+
const identity = deferred.configuration.model;
|
|
9
|
+
const model = lane.models.getModel(identity.provider, identity.modelId);
|
|
10
|
+
if (model === undefined)
|
|
11
|
+
return;
|
|
12
|
+
try {
|
|
13
|
+
await lane.models.cancelDeferred(model, handle, {
|
|
14
|
+
signal: drive.closeSignal,
|
|
15
|
+
telemetryContext: getTelemetryContext(drive.context),
|
|
16
|
+
timeoutMs: deferred.streamOptions.timeoutMs,
|
|
17
|
+
maxRetries: deferred.streamOptions.maxRetries,
|
|
18
|
+
maxRetryDelayMs: deferred.streamOptions.maxRetryDelayMs,
|
|
19
|
+
headers: deferred.streamOptions.headers,
|
|
20
|
+
});
|
|
21
|
+
}
|
|
22
|
+
catch {
|
|
23
|
+
// Remote cancellation is best-effort; durable local reconciliation must continue.
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
async function readDeferredHandle(lane, drive, deferred) {
|
|
27
|
+
return lane.settleOperation(deferred, async (_state, _current, _meta, reader) => ({
|
|
28
|
+
kind: "return",
|
|
29
|
+
result: await readDeferredSourceHandle(reader, deferred, drive.context),
|
|
30
|
+
}), drive.context);
|
|
31
|
+
}
|
|
32
|
+
async function publishAbortedTerminal(lane, drive, capability) {
|
|
33
|
+
return lane.settleOperation(capability, async (state, current, meta, reader) => {
|
|
34
|
+
if (current.control.status !== "cancel_requested") {
|
|
35
|
+
throw new SessionInvariantError("Cancellation reconciliation requires cancelled durable control");
|
|
36
|
+
}
|
|
37
|
+
const record = operationResultRecord(meta, "aborted", state.tipId);
|
|
38
|
+
const cleanup = await operationCleanupWrites(reader, drive.operationId, current, drive.context);
|
|
39
|
+
const events = [];
|
|
40
|
+
if (meta.intent.kind === "run") {
|
|
41
|
+
switch (current.at) {
|
|
42
|
+
case "summary.deciding":
|
|
43
|
+
case "summary.ready":
|
|
44
|
+
case "summary.effect_pending":
|
|
45
|
+
case "summary.retry_wait":
|
|
46
|
+
if (current.task.boundary.kind !== "resume_checkpoint" || current.task.reason === undefined) {
|
|
47
|
+
throw new SessionInvariantError("Cancelled run summary has an invalid result boundary");
|
|
48
|
+
}
|
|
49
|
+
events.push({
|
|
50
|
+
type: "compaction_end",
|
|
51
|
+
lane: lane.name,
|
|
52
|
+
runId: drive.operationId,
|
|
53
|
+
reason: current.task.reason,
|
|
54
|
+
status: "aborted",
|
|
55
|
+
endedAt: record.endedAt,
|
|
56
|
+
});
|
|
57
|
+
break;
|
|
58
|
+
default:
|
|
59
|
+
break;
|
|
60
|
+
}
|
|
61
|
+
events.push({
|
|
62
|
+
type: "run_end",
|
|
63
|
+
lane: lane.name,
|
|
64
|
+
runId: drive.operationId,
|
|
65
|
+
status: "aborted",
|
|
66
|
+
fromTipId: meta.sourceTipId,
|
|
67
|
+
tipId: state.tipId,
|
|
68
|
+
endedAt: record.endedAt,
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
else if (meta.intent.kind === "compaction") {
|
|
72
|
+
events.push({
|
|
73
|
+
type: "compaction_end",
|
|
74
|
+
lane: lane.name,
|
|
75
|
+
runId: drive.operationId,
|
|
76
|
+
reason: "manual",
|
|
77
|
+
status: "aborted",
|
|
78
|
+
endedAt: record.endedAt,
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
else {
|
|
82
|
+
events.push({
|
|
83
|
+
type: "navigation_end",
|
|
84
|
+
lane: lane.name,
|
|
85
|
+
runId: drive.operationId,
|
|
86
|
+
status: "aborted",
|
|
87
|
+
fromTipId: meta.sourceTipId,
|
|
88
|
+
tipId: state.tipId,
|
|
89
|
+
endedAt: record.endedAt,
|
|
90
|
+
});
|
|
91
|
+
}
|
|
92
|
+
return {
|
|
93
|
+
kind: "finish",
|
|
94
|
+
writes: cleanup,
|
|
95
|
+
record,
|
|
96
|
+
materialize: () => ({ kind: "settled", outcome: record }),
|
|
97
|
+
events: () => events,
|
|
98
|
+
};
|
|
99
|
+
}, drive.context);
|
|
100
|
+
}
|
|
101
|
+
/** Advance one cancelled durable leaf without starting new ordinary work. */
|
|
102
|
+
export async function reconcileOperation(lane, drive) {
|
|
103
|
+
const operation = lane.state.operation;
|
|
104
|
+
if (operation === null || operation.meta.operationId !== drive.operationId) {
|
|
105
|
+
throw new SessionInvariantError(`Drive ${drive.operationId} has no matching operation to reconcile`);
|
|
106
|
+
}
|
|
107
|
+
if (operation.state.control.status !== "cancel_requested") {
|
|
108
|
+
throw new SessionInvariantError(`Operation ${drive.operationId} is not cancelled`);
|
|
109
|
+
}
|
|
110
|
+
drive.beginAbort(Promise.resolve());
|
|
111
|
+
drive.signalAbort();
|
|
112
|
+
const state = operation.state;
|
|
113
|
+
switch (state.at) {
|
|
114
|
+
case "assistant.effect_pending":
|
|
115
|
+
return recoverCancelledAssistantEffect(lane, drive, state);
|
|
116
|
+
case "tools":
|
|
117
|
+
return runTools(lane, drive, state);
|
|
118
|
+
case "deferred.suspended": {
|
|
119
|
+
const handle = await readDeferredHandle(lane, drive, state);
|
|
120
|
+
await cancelDeferredBestEffort(lane, drive, state, handle);
|
|
121
|
+
return publishAbortedTerminal(lane, drive, state);
|
|
122
|
+
}
|
|
123
|
+
case "deferred.effect_pending": {
|
|
124
|
+
const handle = await readDeferredHandle(lane, drive, state);
|
|
125
|
+
await cancelDeferredBestEffort(lane, drive, state, handle);
|
|
126
|
+
return recoverCancelledAssistantEffect(lane, drive, state);
|
|
127
|
+
}
|
|
128
|
+
case "starting":
|
|
129
|
+
case "checkpoint":
|
|
130
|
+
case "assistant.ready":
|
|
131
|
+
case "assistant.retry_wait":
|
|
132
|
+
case "summary.deciding":
|
|
133
|
+
case "summary.ready":
|
|
134
|
+
case "summary.effect_pending":
|
|
135
|
+
case "summary.retry_wait":
|
|
136
|
+
case "navigation.ready_to_commit":
|
|
137
|
+
return publishAbortedTerminal(lane, drive, state);
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
//# sourceMappingURL=reconcile.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"reconcile.js","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/reconcile.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAQjE,OAAO,EAAE,wBAAwB,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,EAAE,+BAA+B,EAAE,MAAM,eAAe,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAC9E,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEtC,KAAK,UAAU,wBAAwB,CACtC,IAAoB,EACpB,KAAY,EACZ,QAAqE,EACrE,MAAsB,EACN;IAChB,MAAM,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC;IAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;IACxE,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO;IAChC,IAAI,CAAC;QACJ,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE;YAC/C,MAAM,EAAE,KAAK,CAAC,WAAW;YACzB,gBAAgB,EAAE,mBAAmB,CAAC,KAAK,CAAC,OAAO,CAAC;YACpD,SAAS,EAAE,QAAQ,CAAC,aAAa,CAAC,SAAS;YAC3C,UAAU,EAAE,QAAQ,CAAC,aAAa,CAAC,UAAU;YAC7C,eAAe,EAAE,QAAQ,CAAC,aAAa,CAAC,eAAe;YACvD,OAAO,EAAE,QAAQ,CAAC,aAAa,CAAC,OAAO;SACvC,CAAC,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACR,kFAAkF;IACnF,CAAC;AAAA,CACD;AAED,KAAK,UAAU,kBAAkB,CAChC,IAAoB,EACpB,KAAY,EACZ,QAAqE,EAC3C;IAC1B,OAAO,IAAI,CAAC,eAAe,CAC1B,QAAQ,EACR,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC3C,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,MAAM,wBAAwB,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC;KACvE,CAAC,EACF,KAAK,CAAC,OAAO,CACb,CAAC;AAAA,CACF;AAED,KAAK,UAAU,sBAAsB,CACpC,IAAoB,EACpB,KAAY,EACZ,UAA0B,EACC;IAC3B,OAAO,IAAI,CAAC,eAAe,CAC1B,UAAU,EACV,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC;QACvC,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;YACnD,MAAM,IAAI,qBAAqB,CAAC,gEAAgE,CAAC,CAAC;QACnG,CAAC;QACD,MAAM,MAAM,GAAG,qBAAqB,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAChG,MAAM,MAAM,GAAmB,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAChC,QAAQ,OAAO,CAAC,EAAE,EAAE,CAAC;gBACpB,KAAK,kBAAkB,CAAC;gBACxB,KAAK,eAAe,CAAC;gBACrB,KAAK,wBAAwB,CAAC;gBAC9B,KAAK,oBAAoB;oBACxB,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;wBAC7F,MAAM,IAAI,qBAAqB,CAAC,sDAAsD,CAAC,CAAC;oBACzF,CAAC;oBACD,MAAM,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,gBAAgB;wBACtB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,KAAK,EAAE,KAAK,CAAC,WAAW;wBACxB,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM;wBAC3B,MAAM,EAAE,SAAS;wBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;qBACvB,CAAC,CAAC;oBACH,MAAM;gBACP;oBACC,MAAM;YACR,CAAC;YACD,MAAM,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,KAAK,EAAE,KAAK,CAAC,WAAW;gBACxB,MAAM,EAAE,SAAS;gBACjB,SAAS,EAAE,IAAI,CAAC,WAAW;gBAC3B,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,OAAO,EAAE,MAAM,CAAC,OAAO;aACvB,CAAC,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,KAAK,EAAE,KAAK,CAAC,WAAW;gBACxB,MAAM,EAAE,QAAQ;gBAChB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;aACvB,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,KAAK,EAAE,KAAK,CAAC,WAAW;gBACxB,MAAM,EAAE,SAAS;gBACjB,SAAS,EAAE,IAAI,CAAC,WAAW;gBAC3B,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,OAAO,EAAE,MAAM,CAAC,OAAO;aACvB,CAAC,CAAC;QACJ,CAAC;QACD,OAAO;YACN,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,OAAO;YACf,MAAM;YACN,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,CAAU;YAClE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM;SACpB,CAAC;IAAA,CACF,EACD,KAAK,CAAC,OAAO,CACb,CAAC;AAAA,CACF;AAED,6EAA6E;AAC7E,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACvC,IAAoB,EACpB,KAAY,EACe;IAC3B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;IACvC,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,CAAC,IAAI,CAAC,WAAW,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC;QAC5E,MAAM,IAAI,qBAAqB,CAAC,SAAS,KAAK,CAAC,WAAW,yCAAyC,CAAC,CAAC;IACtG,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;QAC3D,MAAM,IAAI,qBAAqB,CAAC,aAAa,KAAK,CAAC,WAAW,mBAAmB,CAAC,CAAC;IACpF,CAAC;IACD,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACpC,KAAK,CAAC,WAAW,EAAE,CAAC;IAEpB,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC;IAC9B,QAAQ,KAAK,CAAC,EAAE,EAAE,CAAC;QAClB,KAAK,0BAA0B;YAC9B,OAAO,+BAA+B,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAC5D,KAAK,OAAO;YACX,OAAO,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QACrC,KAAK,oBAAoB,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;YAC5D,MAAM,wBAAwB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;YAC3D,OAAO,sBAAsB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QACnD,CAAC;QACD,KAAK,yBAAyB,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;YAC5D,MAAM,wBAAwB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;YAC3D,OAAO,+BAA+B,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;QACD,KAAK,UAAU,CAAC;QAChB,KAAK,YAAY,CAAC;QAClB,KAAK,iBAAiB,CAAC;QACvB,KAAK,sBAAsB,CAAC;QAC5B,KAAK,kBAAkB,CAAC;QACxB,KAAK,eAAe,CAAC;QACrB,KAAK,wBAAwB,CAAC;QAC9B,KAAK,oBAAoB,CAAC;QAC1B,KAAK,4BAA4B;YAChC,OAAO,sBAAsB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;AAAA,CACD","sourcesContent":["import type { DeferredHandle } from \"@m4ike1/ion-ai\";\nimport type { HarnessEvent } from \"../../agent-harness.ts\";\nimport { getTelemetryContext } from \"../../context.ts\";\nimport { SessionInvariantError } from \"../../session/session.ts\";\nimport type {\n\tDeferredEffectPendingOperation,\n\tDeferredSuspendedOperation,\n\tOperationState,\n} from \"../../session/types.ts\";\nimport type { Lane } from \"../lane.ts\";\nimport type { Drive, ProcedureResult } from \"../types.ts\";\nimport { readDeferredSourceHandle } from \"./deferred.ts\";\nimport { recoverCancelledAssistantEffect } from \"./recovery.ts\";\nimport { operationCleanupWrites, operationResultRecord } from \"./terminal.ts\";\nimport { runTools } from \"./tools.ts\";\n\nasync function cancelDeferredBestEffort<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tdeferred: DeferredSuspendedOperation | DeferredEffectPendingOperation,\n\thandle: DeferredHandle,\n): Promise<void> {\n\tconst identity = deferred.configuration.model;\n\tconst model = lane.models.getModel(identity.provider, identity.modelId);\n\tif (model === undefined) return;\n\ttry {\n\t\tawait lane.models.cancelDeferred(model, handle, {\n\t\t\tsignal: drive.closeSignal,\n\t\t\ttelemetryContext: getTelemetryContext(drive.context),\n\t\t\ttimeoutMs: deferred.streamOptions.timeoutMs,\n\t\t\tmaxRetries: deferred.streamOptions.maxRetries,\n\t\t\tmaxRetryDelayMs: deferred.streamOptions.maxRetryDelayMs,\n\t\t\theaders: deferred.streamOptions.headers,\n\t\t});\n\t} catch {\n\t\t// Remote cancellation is best-effort; durable local reconciliation must continue.\n\t}\n}\n\nasync function readDeferredHandle<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tdeferred: DeferredSuspendedOperation | DeferredEffectPendingOperation,\n): Promise<DeferredHandle> {\n\treturn lane.settleOperation(\n\t\tdeferred,\n\t\tasync (_state, _current, _meta, reader) => ({\n\t\t\tkind: \"return\",\n\t\t\tresult: await readDeferredSourceHandle(reader, deferred, drive.context),\n\t\t}),\n\t\tdrive.context,\n\t);\n}\n\nasync function publishAbortedTerminal<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tcapability: OperationState,\n): Promise<ProcedureResult> {\n\treturn lane.settleOperation(\n\t\tcapability,\n\t\tasync (state, current, meta, reader) => {\n\t\t\tif (current.control.status !== \"cancel_requested\") {\n\t\t\t\tthrow new SessionInvariantError(\"Cancellation reconciliation requires cancelled durable control\");\n\t\t\t}\n\t\t\tconst record = operationResultRecord(meta, \"aborted\", state.tipId);\n\t\t\tconst cleanup = await operationCleanupWrites(reader, drive.operationId, current, drive.context);\n\t\t\tconst events: HarnessEvent[] = [];\n\t\t\tif (meta.intent.kind === \"run\") {\n\t\t\t\tswitch (current.at) {\n\t\t\t\t\tcase \"summary.deciding\":\n\t\t\t\t\tcase \"summary.ready\":\n\t\t\t\t\tcase \"summary.effect_pending\":\n\t\t\t\t\tcase \"summary.retry_wait\":\n\t\t\t\t\t\tif (current.task.boundary.kind !== \"resume_checkpoint\" || current.task.reason === undefined) {\n\t\t\t\t\t\t\tthrow new SessionInvariantError(\"Cancelled run summary has an invalid result boundary\");\n\t\t\t\t\t\t}\n\t\t\t\t\t\tevents.push({\n\t\t\t\t\t\t\ttype: \"compaction_end\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\treason: current.task.reason,\n\t\t\t\t\t\t\tstatus: \"aborted\",\n\t\t\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tdefault:\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t\tevents.push({\n\t\t\t\t\ttype: \"run_end\",\n\t\t\t\t\tlane: lane.name,\n\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\tstatus: \"aborted\",\n\t\t\t\t\tfromTipId: meta.sourceTipId,\n\t\t\t\t\ttipId: state.tipId,\n\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t});\n\t\t\t} else if (meta.intent.kind === \"compaction\") {\n\t\t\t\tevents.push({\n\t\t\t\t\ttype: \"compaction_end\",\n\t\t\t\t\tlane: lane.name,\n\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\treason: \"manual\",\n\t\t\t\t\tstatus: \"aborted\",\n\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t});\n\t\t\t} else {\n\t\t\t\tevents.push({\n\t\t\t\t\ttype: \"navigation_end\",\n\t\t\t\t\tlane: lane.name,\n\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\tstatus: \"aborted\",\n\t\t\t\t\tfromTipId: meta.sourceTipId,\n\t\t\t\t\ttipId: state.tipId,\n\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t});\n\t\t\t}\n\t\t\treturn {\n\t\t\t\tkind: \"finish\",\n\t\t\t\twrites: cleanup,\n\t\t\t\trecord,\n\t\t\t\tmaterialize: () => ({ kind: \"settled\", outcome: record }) as const,\n\t\t\t\tevents: () => events,\n\t\t\t};\n\t\t},\n\t\tdrive.context,\n\t);\n}\n\n/** Advance one cancelled durable leaf without starting new ordinary work. */\nexport async function reconcileOperation<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n): Promise<ProcedureResult> {\n\tconst operation = lane.state.operation;\n\tif (operation === null || operation.meta.operationId !== drive.operationId) {\n\t\tthrow new SessionInvariantError(`Drive ${drive.operationId} has no matching operation to reconcile`);\n\t}\n\tif (operation.state.control.status !== \"cancel_requested\") {\n\t\tthrow new SessionInvariantError(`Operation ${drive.operationId} is not cancelled`);\n\t}\n\tdrive.beginAbort(Promise.resolve());\n\tdrive.signalAbort();\n\n\tconst state = operation.state;\n\tswitch (state.at) {\n\t\tcase \"assistant.effect_pending\":\n\t\t\treturn recoverCancelledAssistantEffect(lane, drive, state);\n\t\tcase \"tools\":\n\t\t\treturn runTools(lane, drive, state);\n\t\tcase \"deferred.suspended\": {\n\t\t\tconst handle = await readDeferredHandle(lane, drive, state);\n\t\t\tawait cancelDeferredBestEffort(lane, drive, state, handle);\n\t\t\treturn publishAbortedTerminal(lane, drive, state);\n\t\t}\n\t\tcase \"deferred.effect_pending\": {\n\t\t\tconst handle = await readDeferredHandle(lane, drive, state);\n\t\t\tawait cancelDeferredBestEffort(lane, drive, state, handle);\n\t\t\treturn recoverCancelledAssistantEffect(lane, drive, state);\n\t\t}\n\t\tcase \"starting\":\n\t\tcase \"checkpoint\":\n\t\tcase \"assistant.ready\":\n\t\tcase \"assistant.retry_wait\":\n\t\tcase \"summary.deciding\":\n\t\tcase \"summary.ready\":\n\t\tcase \"summary.effect_pending\":\n\t\tcase \"summary.retry_wait\":\n\t\tcase \"navigation.ready_to_commit\":\n\t\t\treturn publishAbortedTerminal(lane, drive, state);\n\t}\n}\n"]}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { AssistantEffectPendingOperation, DeferredEffectPendingOperation } from "../../session/types.ts";
|
|
2
|
+
import type { Lane } from "../lane.ts";
|
|
3
|
+
import type { Drive, ProcedureResult } from "../types.ts";
|
|
4
|
+
/** Settle an orphaned assistant request from its bounded committed frame prefix without another provider call. */
|
|
5
|
+
export declare function recoverAssistantGeneration<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, generation: AssistantEffectPendingOperation): Promise<ProcedureResult>;
|
|
6
|
+
/** Synthetically settle one cancelled orphaned assistant or deferred effect under its reserved ids. */
|
|
7
|
+
export declare function recoverCancelledAssistantEffect<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, effect: AssistantEffectPendingOperation | DeferredEffectPendingOperation): Promise<ProcedureResult>;
|
|
8
|
+
//# sourceMappingURL=recovery.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"recovery.d.ts","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/recovery.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACX,+BAA+B,EAC/B,8BAA8B,EAG9B,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,KAAK,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAiC1D,kHAAkH;AAClH,wBAAsB,0BAA0B,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EACnF,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,+BAA+B,GACzC,OAAO,CAAC,eAAe,CAAC,CAoC1B;AAED,uGAAuG;AACvG,wBAAsB,+BAA+B,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EACxF,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,+BAA+B,GAAG,8BAA8B,GACtE,OAAO,CAAC,eAAe,CAAC,CAmC1B","sourcesContent":["import { type AssistantMessage, reduceAssistantMessageFrames } from \"@m4ike1/ion-ai\";\nimport type {\n\tAssistantEffectPendingOperation,\n\tDeferredEffectPendingOperation,\n\tLaneConfiguration,\n\tSettledAssistantMessage,\n} from \"../../session/types.ts\";\nimport type { Lane } from \"../lane.ts\";\nimport { readAssistantFrames } from \"../progress.ts\";\nimport type { Drive, ProcedureResult } from \"../types.ts\";\nimport { publishResponse } from \"./response.ts\";\n\nconst ZERO_USAGE = {\n\tinput: 0,\n\toutput: 0,\n\tcacheRead: 0,\n\tcacheWrite: 0,\n\ttotalTokens: 0,\n\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },\n};\n\nfunction interruptedAssistantMessage(\n\tidentity: LaneConfiguration[\"model\"],\n\tpartial: AssistantMessage | undefined,\n): SettledAssistantMessage {\n\tconst warning =\n\t\t\"Assistant request was interrupted. The preceding content is the latest committed partial; newer live output may be missing and the external outcome is unknown.\";\n\treturn partial === undefined\n\t\t? {\n\t\t\t\trole: \"assistant\",\n\t\t\t\tcontent: [],\n\t\t\t\tapi: \"unknown\",\n\t\t\t\tprovider: identity.provider,\n\t\t\t\tmodel: identity.modelId,\n\t\t\t\tusage: ZERO_USAGE,\n\t\t\t\tstopReason: \"error\",\n\t\t\t\terrorMessage: warning,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t}\n\t\t: { ...partial, usage: ZERO_USAGE, stopReason: \"error\", errorMessage: warning };\n}\n\n/** Settle an orphaned assistant request from its bounded committed frame prefix without another provider call. */\nexport async function recoverAssistantGeneration<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tgeneration: AssistantEffectPendingOperation,\n): Promise<ProcedureResult> {\n\tconst frames = await lane.continueOperation(\n\t\tgeneration,\n\t\tasync (_state, _current, _meta, reader) => ({\n\t\t\tkind: \"return\",\n\t\t\tresult: await readAssistantFrames(reader, drive.operationId, generation.responseEntryId, drive.context),\n\t\t}),\n\t\tdrive.context,\n\t);\n\tif (frames.kind === \"cancel_requested\") return { kind: \"continue\" };\n\n\tconst message = interruptedAssistantMessage(\n\t\tgeneration.generationContext.configuration.model,\n\t\treduceAssistantMessageFrames(frames.value),\n\t);\n\tawait lane.emitBatch(\n\t\t[\n\t\t\t{\n\t\t\t\ttype: \"message_start\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\ttype: \"message_end\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\tentryId: generation.responseEntryId,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t],\n\t\tdrive.context,\n\t);\n\treturn publishResponse(lane, drive, generation, message, { recovery: true });\n}\n\n/** Synthetically settle one cancelled orphaned assistant or deferred effect under its reserved ids. */\nexport async function recoverCancelledAssistantEffect<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\teffect: AssistantEffectPendingOperation | DeferredEffectPendingOperation,\n): Promise<ProcedureResult> {\n\tconst frames = await lane.settleOperation(\n\t\teffect,\n\t\tasync (_state, _current, _meta, reader) => ({\n\t\t\tkind: \"return\",\n\t\t\tresult: await readAssistantFrames(reader, drive.operationId, effect.responseEntryId, drive.context),\n\t\t}),\n\t\tdrive.context,\n\t);\n\tconst identity =\n\t\teffect.at === \"assistant.effect_pending\"\n\t\t\t? effect.generationContext.configuration.model\n\t\t\t: effect.configuration.model;\n\tconst message = interruptedAssistantMessage(identity, reduceAssistantMessageFrames(frames));\n\tawait lane.emitBatch(\n\t\t[\n\t\t\t{\n\t\t\t\ttype: \"message_start\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\ttype: \"message_end\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\tentryId: effect.responseEntryId,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t],\n\t\tdrive.context,\n\t);\n\treturn publishResponse(lane, drive, effect, message, { recovery: true });\n}\n"]}
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
import { reduceAssistantMessageFrames } from "@m4ike1/ion-ai";
|
|
2
|
+
import { readAssistantFrames } from "../progress.js";
|
|
3
|
+
import { publishResponse } from "./response.js";
|
|
4
|
+
const ZERO_USAGE = {
|
|
5
|
+
input: 0,
|
|
6
|
+
output: 0,
|
|
7
|
+
cacheRead: 0,
|
|
8
|
+
cacheWrite: 0,
|
|
9
|
+
totalTokens: 0,
|
|
10
|
+
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
11
|
+
};
|
|
12
|
+
function interruptedAssistantMessage(identity, partial) {
|
|
13
|
+
const warning = "Assistant request was interrupted. The preceding content is the latest committed partial; newer live output may be missing and the external outcome is unknown.";
|
|
14
|
+
return partial === undefined
|
|
15
|
+
? {
|
|
16
|
+
role: "assistant",
|
|
17
|
+
content: [],
|
|
18
|
+
api: "unknown",
|
|
19
|
+
provider: identity.provider,
|
|
20
|
+
model: identity.modelId,
|
|
21
|
+
usage: ZERO_USAGE,
|
|
22
|
+
stopReason: "error",
|
|
23
|
+
errorMessage: warning,
|
|
24
|
+
timestamp: Date.now(),
|
|
25
|
+
}
|
|
26
|
+
: { ...partial, usage: ZERO_USAGE, stopReason: "error", errorMessage: warning };
|
|
27
|
+
}
|
|
28
|
+
/** Settle an orphaned assistant request from its bounded committed frame prefix without another provider call. */
|
|
29
|
+
export async function recoverAssistantGeneration(lane, drive, generation) {
|
|
30
|
+
const frames = await lane.continueOperation(generation, async (_state, _current, _meta, reader) => ({
|
|
31
|
+
kind: "return",
|
|
32
|
+
result: await readAssistantFrames(reader, drive.operationId, generation.responseEntryId, drive.context),
|
|
33
|
+
}), drive.context);
|
|
34
|
+
if (frames.kind === "cancel_requested")
|
|
35
|
+
return { kind: "continue" };
|
|
36
|
+
const message = interruptedAssistantMessage(generation.generationContext.configuration.model, reduceAssistantMessageFrames(frames.value));
|
|
37
|
+
await lane.emitBatch([
|
|
38
|
+
{
|
|
39
|
+
type: "message_start",
|
|
40
|
+
lane: lane.name,
|
|
41
|
+
runId: drive.operationId,
|
|
42
|
+
message,
|
|
43
|
+
recovery: true,
|
|
44
|
+
},
|
|
45
|
+
{
|
|
46
|
+
type: "message_end",
|
|
47
|
+
lane: lane.name,
|
|
48
|
+
runId: drive.operationId,
|
|
49
|
+
message,
|
|
50
|
+
entryId: generation.responseEntryId,
|
|
51
|
+
recovery: true,
|
|
52
|
+
},
|
|
53
|
+
], drive.context);
|
|
54
|
+
return publishResponse(lane, drive, generation, message, { recovery: true });
|
|
55
|
+
}
|
|
56
|
+
/** Synthetically settle one cancelled orphaned assistant or deferred effect under its reserved ids. */
|
|
57
|
+
export async function recoverCancelledAssistantEffect(lane, drive, effect) {
|
|
58
|
+
const frames = await lane.settleOperation(effect, async (_state, _current, _meta, reader) => ({
|
|
59
|
+
kind: "return",
|
|
60
|
+
result: await readAssistantFrames(reader, drive.operationId, effect.responseEntryId, drive.context),
|
|
61
|
+
}), drive.context);
|
|
62
|
+
const identity = effect.at === "assistant.effect_pending"
|
|
63
|
+
? effect.generationContext.configuration.model
|
|
64
|
+
: effect.configuration.model;
|
|
65
|
+
const message = interruptedAssistantMessage(identity, reduceAssistantMessageFrames(frames));
|
|
66
|
+
await lane.emitBatch([
|
|
67
|
+
{
|
|
68
|
+
type: "message_start",
|
|
69
|
+
lane: lane.name,
|
|
70
|
+
runId: drive.operationId,
|
|
71
|
+
message,
|
|
72
|
+
recovery: true,
|
|
73
|
+
},
|
|
74
|
+
{
|
|
75
|
+
type: "message_end",
|
|
76
|
+
lane: lane.name,
|
|
77
|
+
runId: drive.operationId,
|
|
78
|
+
message,
|
|
79
|
+
entryId: effect.responseEntryId,
|
|
80
|
+
recovery: true,
|
|
81
|
+
},
|
|
82
|
+
], drive.context);
|
|
83
|
+
return publishResponse(lane, drive, effect, message, { recovery: true });
|
|
84
|
+
}
|
|
85
|
+
//# sourceMappingURL=recovery.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"recovery.js","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/recovery.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyB,4BAA4B,EAAE,MAAM,gBAAgB,CAAC;AAQrF,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAErD,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAEhD,MAAM,UAAU,GAAG;IAClB,KAAK,EAAE,CAAC;IACR,MAAM,EAAE,CAAC;IACT,SAAS,EAAE,CAAC;IACZ,UAAU,EAAE,CAAC;IACb,WAAW,EAAE,CAAC;IACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;CACpE,CAAC;AAEF,SAAS,2BAA2B,CACnC,QAAoC,EACpC,OAAqC,EACX;IAC1B,MAAM,OAAO,GACZ,iKAAiK,CAAC;IACnK,OAAO,OAAO,KAAK,SAAS;QAC3B,CAAC,CAAC;YACA,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE;YACX,GAAG,EAAE,SAAS;YACd,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,KAAK,EAAE,QAAQ,CAAC,OAAO;YACvB,KAAK,EAAE,UAAU;YACjB,UAAU,EAAE,OAAO;YACnB,YAAY,EAAE,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACrB;QACF,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC;AAAA,CACjF;AAED,kHAAkH;AAClH,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC/C,IAAoB,EACpB,KAAY,EACZ,UAA2C,EAChB;IAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAC1C,UAAU,EACV,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC3C,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,MAAM,mBAAmB,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,EAAE,UAAU,CAAC,eAAe,EAAE,KAAK,CAAC,OAAO,CAAC;KACvG,CAAC,EACF,KAAK,CAAC,OAAO,CACb,CAAC;IACF,IAAI,MAAM,CAAC,IAAI,KAAK,kBAAkB;QAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IAEpE,MAAM,OAAO,GAAG,2BAA2B,CAC1C,UAAU,CAAC,iBAAiB,CAAC,aAAa,CAAC,KAAK,EAChD,4BAA4B,CAAC,MAAM,CAAC,KAAK,CAAC,CAC1C,CAAC;IACF,MAAM,IAAI,CAAC,SAAS,CACnB;QACC;YACC,IAAI,EAAE,eAAe;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,KAAK,CAAC,WAAW;YACxB,OAAO;YACP,QAAQ,EAAE,IAAI;SACd;QACD;YACC,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,KAAK,CAAC,WAAW;YACxB,OAAO;YACP,OAAO,EAAE,UAAU,CAAC,eAAe;YACnC,QAAQ,EAAE,IAAI;SACd;KACD,EACD,KAAK,CAAC,OAAO,CACb,CAAC;IACF,OAAO,eAAe,CAAC,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;AAAA,CAC7E;AAED,uGAAuG;AACvG,MAAM,CAAC,KAAK,UAAU,+BAA+B,CACpD,IAAoB,EACpB,KAAY,EACZ,MAAwE,EAC7C;IAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CACxC,MAAM,EACN,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC3C,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,MAAM,mBAAmB,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC,eAAe,EAAE,KAAK,CAAC,OAAO,CAAC;KACnG,CAAC,EACF,KAAK,CAAC,OAAO,CACb,CAAC;IACF,MAAM,QAAQ,GACb,MAAM,CAAC,EAAE,KAAK,0BAA0B;QACvC,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,KAAK;QAC9C,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC;IAC/B,MAAM,OAAO,GAAG,2BAA2B,CAAC,QAAQ,EAAE,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC;IAC5F,MAAM,IAAI,CAAC,SAAS,CACnB;QACC;YACC,IAAI,EAAE,eAAe;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,KAAK,CAAC,WAAW;YACxB,OAAO;YACP,QAAQ,EAAE,IAAI;SACd;QACD;YACC,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,KAAK,CAAC,WAAW;YACxB,OAAO;YACP,OAAO,EAAE,MAAM,CAAC,eAAe;YAC/B,QAAQ,EAAE,IAAI;SACd;KACD,EACD,KAAK,CAAC,OAAO,CACb,CAAC;IACF,OAAO,eAAe,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;AAAA,CACzE","sourcesContent":["import { type AssistantMessage, reduceAssistantMessageFrames } from \"@m4ike1/ion-ai\";\nimport type {\n\tAssistantEffectPendingOperation,\n\tDeferredEffectPendingOperation,\n\tLaneConfiguration,\n\tSettledAssistantMessage,\n} from \"../../session/types.ts\";\nimport type { Lane } from \"../lane.ts\";\nimport { readAssistantFrames } from \"../progress.ts\";\nimport type { Drive, ProcedureResult } from \"../types.ts\";\nimport { publishResponse } from \"./response.ts\";\n\nconst ZERO_USAGE = {\n\tinput: 0,\n\toutput: 0,\n\tcacheRead: 0,\n\tcacheWrite: 0,\n\ttotalTokens: 0,\n\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },\n};\n\nfunction interruptedAssistantMessage(\n\tidentity: LaneConfiguration[\"model\"],\n\tpartial: AssistantMessage | undefined,\n): SettledAssistantMessage {\n\tconst warning =\n\t\t\"Assistant request was interrupted. The preceding content is the latest committed partial; newer live output may be missing and the external outcome is unknown.\";\n\treturn partial === undefined\n\t\t? {\n\t\t\t\trole: \"assistant\",\n\t\t\t\tcontent: [],\n\t\t\t\tapi: \"unknown\",\n\t\t\t\tprovider: identity.provider,\n\t\t\t\tmodel: identity.modelId,\n\t\t\t\tusage: ZERO_USAGE,\n\t\t\t\tstopReason: \"error\",\n\t\t\t\terrorMessage: warning,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t}\n\t\t: { ...partial, usage: ZERO_USAGE, stopReason: \"error\", errorMessage: warning };\n}\n\n/** Settle an orphaned assistant request from its bounded committed frame prefix without another provider call. */\nexport async function recoverAssistantGeneration<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tgeneration: AssistantEffectPendingOperation,\n): Promise<ProcedureResult> {\n\tconst frames = await lane.continueOperation(\n\t\tgeneration,\n\t\tasync (_state, _current, _meta, reader) => ({\n\t\t\tkind: \"return\",\n\t\t\tresult: await readAssistantFrames(reader, drive.operationId, generation.responseEntryId, drive.context),\n\t\t}),\n\t\tdrive.context,\n\t);\n\tif (frames.kind === \"cancel_requested\") return { kind: \"continue\" };\n\n\tconst message = interruptedAssistantMessage(\n\t\tgeneration.generationContext.configuration.model,\n\t\treduceAssistantMessageFrames(frames.value),\n\t);\n\tawait lane.emitBatch(\n\t\t[\n\t\t\t{\n\t\t\t\ttype: \"message_start\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\ttype: \"message_end\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\tentryId: generation.responseEntryId,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t],\n\t\tdrive.context,\n\t);\n\treturn publishResponse(lane, drive, generation, message, { recovery: true });\n}\n\n/** Synthetically settle one cancelled orphaned assistant or deferred effect under its reserved ids. */\nexport async function recoverCancelledAssistantEffect<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\teffect: AssistantEffectPendingOperation | DeferredEffectPendingOperation,\n): Promise<ProcedureResult> {\n\tconst frames = await lane.settleOperation(\n\t\teffect,\n\t\tasync (_state, _current, _meta, reader) => ({\n\t\t\tkind: \"return\",\n\t\t\tresult: await readAssistantFrames(reader, drive.operationId, effect.responseEntryId, drive.context),\n\t\t}),\n\t\tdrive.context,\n\t);\n\tconst identity =\n\t\teffect.at === \"assistant.effect_pending\"\n\t\t\t? effect.generationContext.configuration.model\n\t\t\t: effect.configuration.model;\n\tconst message = interruptedAssistantMessage(identity, reduceAssistantMessageFrames(frames));\n\tawait lane.emitBatch(\n\t\t[\n\t\t\t{\n\t\t\t\ttype: \"message_start\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\ttype: \"message_end\",\n\t\t\t\tlane: lane.name,\n\t\t\t\trunId: drive.operationId,\n\t\t\t\tmessage,\n\t\t\t\tentryId: effect.responseEntryId,\n\t\t\t\trecovery: true,\n\t\t\t},\n\t\t],\n\t\tdrive.context,\n\t);\n\treturn publishResponse(lane, drive, effect, message, { recovery: true });\n}\n"]}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type { Context } from "../../context.ts";
|
|
2
|
+
import type { AssistantResponseMetadata, AssistantStreamObserver } from "../../execution/assistant.ts";
|
|
3
|
+
import { type AssistantEffectPendingOperation, type DeferredEffectPendingOperation, type OperationError, type OperationState, type SettledAssistantMessage } from "../../session/types.ts";
|
|
4
|
+
import type { Lane } from "../lane.ts";
|
|
5
|
+
import type { Drive, ProcedureResult } from "../types.ts";
|
|
6
|
+
export type AssistantResponseLifecycle = {
|
|
7
|
+
observer: AssistantStreamObserver;
|
|
8
|
+
afterResponse(message: SettledAssistantMessage, metadata: AssistantResponseMetadata, context: Context): Promise<SettledAssistantMessage>;
|
|
9
|
+
close(): Promise<void>;
|
|
10
|
+
};
|
|
11
|
+
export declare function openAssistantResponse<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, responseEntryId: string, recovery?: boolean): AssistantResponseLifecycle;
|
|
12
|
+
type ResponseIntent = AssistantEffectPendingOperation | DeferredEffectPendingOperation;
|
|
13
|
+
type ConfigurationFailureState = Extract<OperationState, {
|
|
14
|
+
at: "assistant.ready" | "assistant.retry_wait" | "deferred.suspended" | "deferred.effect_pending";
|
|
15
|
+
}>;
|
|
16
|
+
/** Publish a non-retryable request-configuration failure before reserving response ids. */
|
|
17
|
+
export declare function publishConfigurationFailure<TContext extends object | undefined, TState extends ConfigurationFailureState>(lane: Lane<TContext>, drive: Drive, capability: TState, error: OperationError): Promise<ProcedureResult>;
|
|
18
|
+
/** Classify and atomically settle one assistant-generation or deferred-poll response. */
|
|
19
|
+
export declare function publishResponse<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, intent: ResponseIntent, response: SettledAssistantMessage, options?: {
|
|
20
|
+
recovery?: true;
|
|
21
|
+
}): Promise<ProcedureResult>;
|
|
22
|
+
export {};
|
|
23
|
+
//# sourceMappingURL=response.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"response.d.ts","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/response.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,KAAK,EAAE,yBAAyB,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAGvG,OAAO,EACN,KAAK,+BAA+B,EAEpC,KAAK,8BAA8B,EAGnC,KAAK,cAAc,EACnB,KAAK,cAAc,EAEnB,KAAK,uBAAuB,EAI5B,MAAM,wBAAwB,CAAC;AAEhC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,KAAK,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAK1D,MAAM,MAAM,0BAA0B,GAAG;IACxC,QAAQ,EAAE,uBAAuB,CAAC;IAClC,aAAa,CACZ,OAAO,EAAE,uBAAuB,EAChC,QAAQ,EAAE,yBAAyB,EACnC,OAAO,EAAE,OAAO,GACd,OAAO,CAAC,uBAAuB,CAAC,CAAC;IACpC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACvB,CAAC;AAEF,wBAAgB,qBAAqB,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EACxE,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,eAAe,EAAE,MAAM,EACvB,QAAQ,UAAQ,GACd,0BAA0B,CA8C5B;AAED,KAAK,cAAc,GAAG,+BAA+B,GAAG,8BAA8B,CAAC;AACvF,KAAK,yBAAyB,GAAG,OAAO,CACvC,cAAc,EACd;IAAE,EAAE,EAAE,iBAAiB,GAAG,sBAAsB,GAAG,oBAAoB,GAAG,yBAAyB,CAAA;CAAE,CACrG,CAAC;AAEF,2FAA2F;AAC3F,wBAAsB,2BAA2B,CAChD,QAAQ,SAAS,MAAM,GAAG,SAAS,EACnC,MAAM,SAAS,yBAAyB,EACvC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,eAAe,CAAC,CA6BzG;AA2CD,yFAAyF;AACzF,wBAAsB,eAAe,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EACxE,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,cAAc,EACtB,QAAQ,EAAE,uBAAuB,EACjC,OAAO,GAAE;IAAE,QAAQ,CAAC,EAAE,IAAI,CAAA;CAAO,GAC/B,OAAO,CAAC,eAAe,CAAC,CAyS1B","sourcesContent":["import {\n\ttype AssistantMessageEvent,\n\tAssistantMessageFrameEncoder,\n\tisContextOverflow,\n\tisRecoverableLength,\n\tisRetryableAssistantError,\n\tretryDelayMs,\n} from \"@m4ike1/ion-ai\";\nimport type { HarnessEvent } from \"../../agent-harness.ts\";\nimport type { Context } from \"../../context.ts\";\nimport type { AssistantResponseMetadata, AssistantStreamObserver } from \"../../execution/assistant.ts\";\nimport { insertEntry, insertUsage } from \"../../session/commit.ts\";\nimport { SessionInvariantError } from \"../../session/session.ts\";\nimport {\n\ttype AssistantEffectPendingOperation,\n\ttype CommitResult,\n\ttype DeferredEffectPendingOperation,\n\ttype MessageEntry,\n\ttype NewEntry,\n\ttype OperationError,\n\ttype OperationState,\n\toperationScopeOf,\n\ttype SettledAssistantMessage,\n\ttype SummaryDecidingOperation,\n\ttype ToolCall,\n\ttype UsageRow,\n} from \"../../session/types.ts\";\nimport { branchTip, deleteList, operationPreparation, pendingAssistantFrames, setValue } from \"../../session/values.ts\";\nimport type { Lane } from \"../lane.ts\";\nimport { openFrameProgress } from \"../progress.ts\";\nimport type { Drive, ProcedureResult } from \"../types.ts\";\nimport { retryNotBefore } from \"./retry.ts\";\nimport { prepareOverflowCompaction } from \"./structural.ts\";\nimport { operationCleanupWrites, operationResultRecord } from \"./terminal.ts\";\n\nexport type AssistantResponseLifecycle = {\n\tobserver: AssistantStreamObserver;\n\tafterResponse(\n\t\tmessage: SettledAssistantMessage,\n\t\tmetadata: AssistantResponseMetadata,\n\t\tcontext: Context,\n\t): Promise<SettledAssistantMessage>;\n\tclose(): Promise<void>;\n};\n\nexport function openAssistantResponse<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tresponseEntryId: string,\n\trecovery = false,\n): AssistantResponseLifecycle {\n\tconst progress = openFrameProgress(lane, drive, responseEntryId);\n\tconst frameEncoder = new AssistantMessageFrameEncoder();\n\tconst eventContext = {\n\t\tlane: lane.name,\n\t\trunId: drive.operationId,\n\t\t...(recovery ? { recovery: true as const } : {}),\n\t};\n\tconst close = async () => {\n\t\tprogress.seal();\n\t\tawait progress.drain();\n\t};\n\treturn {\n\t\tobserver: {\n\t\t\tstart(message, event, context) {\n\t\t\t\tconst frame = frameEncoder.encode(event);\n\t\t\t\tif (frame !== undefined) progress.write(frame);\n\t\t\t\treturn lane.emitBatch([{ type: \"message_start\", ...eventContext, message }], context);\n\t\t\t},\n\t\t\tupdate(message, event: AssistantMessageEvent, context) {\n\t\t\t\tconst frame = frameEncoder.encode(event);\n\t\t\t\tif (frame !== undefined) progress.write(frame);\n\t\t\t\treturn lane.emitBatch(\n\t\t\t\t\t[{ type: \"message_update\", ...eventContext, message, event, ...(frame === undefined ? {} : { frame }) }],\n\t\t\t\t\tcontext,\n\t\t\t\t);\n\t\t\t},\n\t\t\tend(message, context) {\n\t\t\t\treturn lane.emitBatch(\n\t\t\t\t\t[{ type: \"message_end\", ...eventContext, message, entryId: responseEntryId }],\n\t\t\t\t\tcontext,\n\t\t\t\t);\n\t\t\t},\n\t\t},\n\t\tasync afterResponse(message, metadata, context) {\n\t\t\tawait close();\n\t\t\tconst result = await lane.hooks.runWithGate(\n\t\t\t\t\"after_response\",\n\t\t\t\t{ lane: lane.name, runId: drive.operationId, ...metadata, message },\n\t\t\t\tdrive.gate,\n\t\t\t\tcontext,\n\t\t\t);\n\t\t\treturn result?.message ?? message;\n\t\t},\n\t\tclose,\n\t};\n}\n\ntype ResponseIntent = AssistantEffectPendingOperation | DeferredEffectPendingOperation;\ntype ConfigurationFailureState = Extract<\n\tOperationState,\n\t{ at: \"assistant.ready\" | \"assistant.retry_wait\" | \"deferred.suspended\" | \"deferred.effect_pending\" }\n>;\n\n/** Publish a non-retryable request-configuration failure before reserving response ids. */\nexport async function publishConfigurationFailure<\n\tTContext extends object | undefined,\n\tTState extends ConfigurationFailureState,\n>(lane: Lane<TContext>, drive: Drive, capability: TState, error: OperationError): Promise<ProcedureResult> {\n\tconst result = await lane.continueOperation(\n\t\tcapability,\n\t\tasync (state, current, meta, reader) => {\n\t\t\tif (state.tipId === null) throw new SessionInvariantError(\"Failed run has no Branch tip\");\n\t\t\tconst record = operationResultRecord(meta, \"failed\", state.tipId, error);\n\t\t\tconst cleanup = await operationCleanupWrites(reader, drive.operationId, current, drive.context);\n\t\t\treturn {\n\t\t\t\tkind: \"finish\",\n\t\t\t\twrites: cleanup,\n\t\t\t\trecord,\n\t\t\t\tmaterialize: () => ({ kind: \"settled\", outcome: record }) as const,\n\t\t\t\tevents: () => [\n\t\t\t\t\t{\n\t\t\t\t\t\ttype: \"run_end\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\tstatus: \"failed\",\n\t\t\t\t\t\terror,\n\t\t\t\t\t\tfromTipId: meta.sourceTipId,\n\t\t\t\t\t\ttipId: state.tipId,\n\t\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t\t},\n\t\t\t\t],\n\t\t\t};\n\t\t},\n\t\tdrive.context,\n\t);\n\treturn result.kind === \"cancel_requested\" ? { kind: \"continue\" } : result.value;\n}\n\nfunction uuidV7Timestamp(id: string): number {\n\tconst timestamp = Number.parseInt(id.slice(0, 8) + id.slice(9, 13), 16);\n\tif (!Number.isSafeInteger(timestamp)) throw new SessionInvariantError(`Invalid reserved UUIDv7 ${id}`);\n\treturn timestamp;\n}\n\nfunction providerError(source: \"assistant\" | \"deferred\", message: SettledAssistantMessage): OperationError {\n\treturn {\n\t\tcode: \"assistant_error\",\n\t\tmessage:\n\t\t\tmessage.errorMessage ??\n\t\t\t`${source === \"assistant\" ? \"Assistant\" : \"Deferred\"} request ended with ${message.stopReason}`,\n\t};\n}\n\nfunction normalizeError(message: SettledAssistantMessage, errorMessage: string): SettledAssistantMessage {\n\treturn { ...message, stopReason: \"error\", errorMessage };\n}\n\nfunction normalizeAborted(source: \"assistant\" | \"deferred\", message: SettledAssistantMessage): SettledAssistantMessage {\n\treturn {\n\t\t...message,\n\t\tstopReason: \"aborted\",\n\t\terrorMessage:\n\t\t\tmessage.errorMessage ?? `${source === \"assistant\" ? \"Assistant\" : \"Deferred\"} request was cancelled`,\n\t};\n}\n\nfunction deferredHandleIsValid(message: SettledAssistantMessage, generation: AssistantEffectPendingOperation): boolean {\n\tconst handle = message.deferred;\n\tconst identity = generation.generationContext.configuration.model;\n\treturn (\n\t\tmessage.stopReason === \"deferred\" &&\n\t\thandle !== undefined &&\n\t\thandle.id.length !== 0 &&\n\t\thandle.provider === identity.provider &&\n\t\thandle.modelId === identity.modelId &&\n\t\thandle.api === message.api\n\t);\n}\n\n/** Classify and atomically settle one assistant-generation or deferred-poll response. */\nexport async function publishResponse<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tintent: ResponseIntent,\n\tresponse: SettledAssistantMessage,\n\toptions: { recovery?: true } = {},\n): Promise<ProcedureResult> {\n\tconst overflow =\n\t\tintent.at === \"assistant.effect_pending\" &&\n\t\t(isContextOverflow(response, intent.contextWindow) || isRecoverableLength(response, intent.intendedOutputLimit));\n\tconst overflowPreparation =\n\t\toverflow && !intent.generationContext.overflowRecoveryUsed\n\t\t\t? await prepareOverflowCompaction(lane, drive, intent)\n\t\t\t: undefined;\n\treturn lane.settleOperation<ResponseIntent, ProcedureResult>(\n\t\tintent,\n\t\tasync (state, current, meta, reader) => {\n\t\t\tconst source = current.at === \"assistant.effect_pending\" ? \"assistant\" : \"deferred\";\n\t\t\tconst responseEntryId = intent.responseEntryId;\n\t\t\tconst configuration =\n\t\t\t\tcurrent.at === \"assistant.effect_pending\" ? current.generationContext.configuration : current.configuration;\n\t\t\tconst turnId =\n\t\t\t\tcurrent.at === \"assistant.effect_pending\"\n\t\t\t\t\t? current.generationContext.stepId\n\t\t\t\t\t: `${current.stepId}:poll:${current.poll}`;\n\t\t\tconst scope = { ...operationScopeOf(current), latestAssistantEntryId: responseEntryId };\n\t\t\tlet committed = response;\n\t\t\tlet settled: OperationState | undefined;\n\t\t\tlet failure: OperationError | undefined;\n\n\t\t\tif (current.control.status === \"cancel_requested\") {\n\t\t\t\tcommitted = normalizeAborted(source, response);\n\t\t\t\tsettled = {\n\t\t\t\t\t...scope,\n\t\t\t\t\tat: \"checkpoint\",\n\t\t\t\t\tcontinuation: { kind: \"may_finish\", includeFinalAssistant: true },\n\t\t\t\t\ttriggerEntryId: responseEntryId,\n\t\t\t\t};\n\t\t\t} else if (response.stopReason === \"aborted\") {\n\t\t\t\tthrow new SessionInvariantError(\n\t\t\t\t\t`${source === \"assistant\" ? \"Assistant\" : \"Deferred\"} response is aborted while durable control is running`,\n\t\t\t\t);\n\t\t\t} else if (current.at === \"assistant.effect_pending\" && overflow) {\n\t\t\t\tcommitted = normalizeError(\n\t\t\t\t\tresponse,\n\t\t\t\t\tresponse.errorMessage ?? \"Assistant request exceeded the context window\",\n\t\t\t\t);\n\t\t\t\tif (current.generationContext.overflowRecoveryUsed || overflowPreparation === undefined) {\n\t\t\t\t\tfailure = providerError(source, committed);\n\t\t\t\t} else {\n\t\t\t\t\tconst structural: SummaryDecidingOperation = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"summary.deciding\",\n\t\t\t\t\t\ttask: {\n\t\t\t\t\t\t\ttaskId: overflowPreparation.taskId,\n\t\t\t\t\t\t\treason: \"overflow\",\n\t\t\t\t\t\t\tboundary: {\n\t\t\t\t\t\t\t\tkind: \"resume_checkpoint\",\n\t\t\t\t\t\t\t\tresumeAfter: {\n\t\t\t\t\t\t\t\t\tcontinuation: { kind: \"need_assistant\", overflowRecoveryUsed: true },\n\t\t\t\t\t\t\t\t\ttriggerEntryId: current.generationContext.triggerEntryId,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t};\n\t\t\t\t\tsettled = structural;\n\t\t\t\t}\n\t\t\t} else if (response.stopReason === \"deferred\") {\n\t\t\t\tif (current.at === \"assistant.effect_pending\") {\n\t\t\t\t\tif (deferredHandleIsValid(response, current)) {\n\t\t\t\t\t\tsettled = {\n\t\t\t\t\t\t\t...scope,\n\t\t\t\t\t\t\tat: \"deferred.suspended\",\n\t\t\t\t\t\t\tstepId: current.generationContext.stepId,\n\t\t\t\t\t\t\tsourceEntryId: responseEntryId,\n\t\t\t\t\t\t\tpoll: 0,\n\t\t\t\t\t\t\tconfiguration,\n\t\t\t\t\t\t\tstreamOptions: current.generationContext.streamOptions,\n\t\t\t\t\t\t};\n\t\t\t\t\t} else {\n\t\t\t\t\t\tcommitted = normalizeError(response, \"Provider returned an invalid deferred handle\");\n\t\t\t\t\t\tfailure = providerError(source, committed);\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"deferred.suspended\",\n\t\t\t\t\t\tstepId: current.stepId,\n\t\t\t\t\t\tsourceEntryId: responseEntryId,\n\t\t\t\t\t\tpoll: current.poll,\n\t\t\t\t\t\tconfiguration,\n\t\t\t\t\t\tstreamOptions: current.streamOptions,\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t} else if (response.stopReason === \"error\") {\n\t\t\t\tif (\n\t\t\t\t\tcurrent.at === \"assistant.effect_pending\" &&\n\t\t\t\t\t(options.recovery === true || isRetryableAssistantError(response)) &&\n\t\t\t\t\tcurrent.attempt < current.generationContext.retryPolicy.maxAttempts\n\t\t\t\t) {\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"assistant.retry_wait\",\n\t\t\t\t\t\tgenerationContext: current.generationContext,\n\t\t\t\t\t\tnextAttempt: current.attempt + 1,\n\t\t\t\t\t\tnotBefore: retryNotBefore(current.generationContext.retryPolicy, current.attempt),\n\t\t\t\t\t\terrorMessage: response.errorMessage ?? \"Assistant request failed\",\n\t\t\t\t\t};\n\t\t\t\t} else {\n\t\t\t\t\tfailure = providerError(source, response);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tconst calls = response.content.flatMap((content, sourceIndex) =>\n\t\t\t\t\tcontent.type === \"toolCall\" ? [{ sourceIndex }] : [],\n\t\t\t\t);\n\t\t\t\tif (calls.length !== 0) {\n\t\t\t\t\tconst timestamp = uuidV7Timestamp(responseEntryId);\n\t\t\t\t\tconst planned: ToolCall[] = calls.map(({ sourceIndex }) => ({\n\t\t\t\t\t\tstatus: \"planned\",\n\t\t\t\t\t\tsourceIndex,\n\t\t\t\t\t\tresultEntryId: lane.session.idGenerator.next(timestamp),\n\t\t\t\t\t}));\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"tools\",\n\t\t\t\t\t\tbatch: { assistantEntryId: responseEntryId, configuration, turnId, calls: planned },\n\t\t\t\t\t};\n\t\t\t\t} else if (response.stopReason === \"toolUse\") {\n\t\t\t\t\tcommitted = normalizeError(response, \"Provider reported tool use without any tool calls\");\n\t\t\t\t\tfailure = providerError(source, committed);\n\t\t\t\t} else {\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"checkpoint\",\n\t\t\t\t\t\tcontinuation: { kind: \"may_finish\", includeFinalAssistant: true },\n\t\t\t\t\t\ttriggerEntryId: responseEntryId,\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst responseEntry: NewEntry<MessageEntry> = {\n\t\t\t\tid: responseEntryId,\n\t\t\t\tparentId: state.tipId,\n\t\t\t\ttype: \"message\",\n\t\t\t\tmessage: committed,\n\t\t\t};\n\t\t\tconst usageRow: Omit<UsageRow, \"seq\"> = {\n\t\t\t\tid: intent.usageId,\n\t\t\t\tusage: committed.usage,\n\t\t\t\tentryId: responseEntryId,\n\t\t\t\tadjustment: false,\n\t\t\t};\n\t\t\tif (settled === undefined && failure === undefined) {\n\t\t\t\tthrow new SessionInvariantError(\"Response settlement has no durable disposition\");\n\t\t\t}\n\t\t\tconst record =\n\t\t\t\tfailure === undefined ? undefined : operationResultRecord(meta, \"failed\", responseEntryId, failure);\n\t\t\tconst cleanup =\n\t\t\t\trecord === undefined ? [] : await operationCleanupWrites(reader, drive.operationId, current, drive.context);\n\t\t\tconst writes = [\n\t\t\t\tinsertEntry(responseEntry),\n\t\t\t\tinsertUsage(usageRow),\n\t\t\t\tsetValue(branchTip(lane.name), responseEntryId),\n\t\t\t\t...(record === undefined\n\t\t\t\t\t? [deleteList(pendingAssistantFrames(drive.operationId, responseEntryId))]\n\t\t\t\t\t: cleanup),\n\t\t\t\t...(settled?.at === \"summary.deciding\" && overflowPreparation !== undefined\n\t\t\t\t\t? [\n\t\t\t\t\t\t\tsetValue(\n\t\t\t\t\t\t\t\toperationPreparation(drive.operationId, overflowPreparation.taskId),\n\t\t\t\t\t\t\t\toverflowPreparation.preparation,\n\t\t\t\t\t\t\t),\n\t\t\t\t\t\t]\n\t\t\t\t\t: []),\n\t\t\t];\n\t\t\tconst materializeEntry = (commit: CommitResult): MessageEntry => ({\n\t\t\t\t...responseEntry,\n\t\t\t\tseq: commit.seqs[0]!,\n\t\t\t\ttimestamp: commit.timestamp,\n\t\t\t});\n\t\t\tconst events = (commit: CommitResult): HarnessEvent[] => {\n\t\t\t\tconst entry = materializeEntry(commit);\n\t\t\t\tconst batch: HarnessEvent[] = [\n\t\t\t\t\t{ type: \"entry_added\", lane: lane.name, entry, ...options },\n\t\t\t\t\t{\n\t\t\t\t\t\ttype: \"usage\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trow: { ...usageRow, seq: commit.seqs[1]! },\n\t\t\t\t\t\ttotals: commit.stats.usage,\n\t\t\t\t\t},\n\t\t\t\t];\n\t\t\t\tif (current.at === \"assistant.effect_pending\") {\n\t\t\t\t\tif (options.recovery !== true && current.attempt > 1 && settled?.at !== \"assistant.retry_wait\") {\n\t\t\t\t\t\tconst success = committed.stopReason !== \"error\" && committed.stopReason !== \"aborted\";\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"retry_end\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\tstep: turnId,\n\t\t\t\t\t\t\tattempt: current.attempt,\n\t\t\t\t\t\t\tsuccess,\n\t\t\t\t\t\t\t...(success\n\t\t\t\t\t\t\t\t? {}\n\t\t\t\t\t\t\t\t: {\n\t\t\t\t\t\t\t\t\t\tfinalError:\n\t\t\t\t\t\t\t\t\t\t\tcommitted.errorMessage ?? `Assistant request ended with ${committed.stopReason}`,\n\t\t\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t\tif (options.recovery !== true && settled?.at === \"assistant.retry_wait\") {\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"retry_scheduled\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\tstep: turnId,\n\t\t\t\t\t\t\tattempt: settled.nextAttempt,\n\t\t\t\t\t\t\tmaxAttempts: settled.generationContext.retryPolicy.maxAttempts,\n\t\t\t\t\t\t\tdelayMs: retryDelayMs(current.generationContext.retryPolicy, current.attempt),\n\t\t\t\t\t\t\tnotBefore: settled.notBefore,\n\t\t\t\t\t\t\terrorMessage: settled.errorMessage,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t\tif (options.recovery !== true && settled?.at !== \"tools\" && settled?.at !== \"assistant.retry_wait\") {\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"turn_end\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\tturnId,\n\t\t\t\t\t\t\tmessage: committed,\n\t\t\t\t\t\t\ttoolResults: [],\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t\tif (settled?.at === \"summary.deciding\") {\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"compaction_start\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\treason: \"overflow\",\n\t\t\t\t\t\t\tstartedAt: commit.timestamp,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t} else if (settled?.at !== \"tools\") {\n\t\t\t\t\tbatch.push({\n\t\t\t\t\t\ttype: \"turn_end\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\tturnId,\n\t\t\t\t\t\tmessage: committed,\n\t\t\t\t\t\ttoolResults: [],\n\t\t\t\t\t\t...options,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\tif (\n\t\t\t\t\t(current.at === \"deferred.effect_pending\" || options.recovery !== true) &&\n\t\t\t\t\tsettled?.at === \"deferred.suspended\" &&\n\t\t\t\t\tcommitted.deferred !== undefined\n\t\t\t\t) {\n\t\t\t\t\tbatch.push({\n\t\t\t\t\t\ttype: \"run_suspend\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\treason: \"deferred\",\n\t\t\t\t\t\tdeferred: committed.deferred,\n\t\t\t\t\t\tpoll: settled.poll,\n\t\t\t\t\t\t...(current.at === \"deferred.effect_pending\" ? options : {}),\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\tif (record !== undefined && failure !== undefined) {\n\t\t\t\t\tbatch.push({\n\t\t\t\t\t\ttype: \"run_end\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\tstatus: \"failed\",\n\t\t\t\t\t\terror: failure,\n\t\t\t\t\t\tfromTipId: meta.sourceTipId,\n\t\t\t\t\t\ttipId: responseEntryId,\n\t\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\treturn batch;\n\t\t\t};\n\t\t\tif (record !== undefined) {\n\t\t\t\treturn {\n\t\t\t\t\tkind: \"finish\",\n\t\t\t\t\twrites,\n\t\t\t\t\trecord,\n\t\t\t\t\tlane: { tipId: responseEntryId },\n\t\t\t\t\tmaterialize: () => ({ kind: \"settled\", outcome: record }) as const,\n\t\t\t\t\tevents,\n\t\t\t\t};\n\t\t\t}\n\t\t\tif (settled === undefined) throw new SessionInvariantError(\"Response settlement is missing its next state\");\n\t\t\treturn {\n\t\t\t\tkind: \"commit\",\n\t\t\t\twrites,\n\t\t\t\toperationState: settled,\n\t\t\t\tlane: { tipId: responseEntryId },\n\t\t\t\tmaterialize: () => ({ kind: \"continue\" }) as const,\n\t\t\t\tevents,\n\t\t\t};\n\t\t},\n\t\tdrive.context,\n\t);\n}\n"]}
|