@namzu/sdk 0.6.0 → 1.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/CHANGELOG.md +529 -0
- package/dist/advisory/executor.d.ts.map +1 -1
- package/dist/advisory/executor.js +9 -2
- package/dist/advisory/executor.js.map +1 -1
- package/dist/advisory/executor.test.d.ts +2 -1
- package/dist/advisory/executor.test.d.ts.map +1 -1
- package/dist/advisory/executor.test.js +7 -4
- package/dist/advisory/executor.test.js.map +1 -1
- package/dist/agents/ReactiveAgent.d.ts.map +1 -1
- package/dist/agents/ReactiveAgent.js +7 -0
- package/dist/agents/ReactiveAgent.js.map +1 -1
- package/dist/agents/SupervisorAgent.d.ts +22 -1
- package/dist/agents/SupervisorAgent.d.ts.map +1 -1
- package/dist/agents/SupervisorAgent.js +54 -14
- package/dist/agents/SupervisorAgent.js.map +1 -1
- package/dist/agents/__tests__/supervisor-ledger.test.d.ts +2 -0
- package/dist/agents/__tests__/supervisor-ledger.test.d.ts.map +1 -0
- package/dist/agents/__tests__/supervisor-ledger.test.js +131 -0
- package/dist/agents/__tests__/supervisor-ledger.test.js.map +1 -0
- package/dist/bridge/sse/mapper.test.js +2 -2
- package/dist/config/runtime.d.ts +24 -0
- package/dist/config/runtime.d.ts.map +1 -1
- package/dist/config/runtime.js +9 -0
- package/dist/config/runtime.js.map +1 -1
- package/dist/connector/builtins/webhook.test.js +17 -0
- package/dist/connector/builtins/webhook.test.js.map +1 -1
- package/dist/connector/index.d.ts +1 -0
- package/dist/connector/index.d.ts.map +1 -1
- package/dist/connector/index.js +1 -0
- package/dist/connector/index.js.map +1 -1
- package/dist/connector/mcp/client.d.ts.map +1 -1
- package/dist/connector/mcp/client.js +4 -0
- package/dist/connector/mcp/client.js.map +1 -1
- package/dist/connector/mcp/index.d.ts +1 -0
- package/dist/connector/mcp/index.d.ts.map +1 -1
- package/dist/connector/mcp/index.js +1 -0
- package/dist/connector/mcp/index.js.map +1 -1
- package/dist/connector/mcp/streamable-http.d.ts +22 -0
- package/dist/connector/mcp/streamable-http.d.ts.map +1 -0
- package/dist/connector/mcp/streamable-http.js +128 -0
- package/dist/connector/mcp/streamable-http.js.map +1 -0
- package/dist/connector/mcp/streamable-http.test.d.ts +2 -0
- package/dist/connector/mcp/streamable-http.test.d.ts.map +1 -0
- package/dist/connector/mcp/streamable-http.test.js +132 -0
- package/dist/connector/mcp/streamable-http.test.js.map +1 -0
- package/dist/constants/compaction/index.d.ts.map +1 -1
- package/dist/constants/compaction/index.js +8 -3
- package/dist/constants/compaction/index.js.map +1 -1
- package/dist/constants/sandbox/index.d.ts +21 -0
- package/dist/constants/sandbox/index.d.ts.map +1 -1
- package/dist/constants/sandbox/index.js +30 -0
- package/dist/constants/sandbox/index.js.map +1 -1
- package/dist/constants/tools/index.d.ts.map +1 -1
- package/dist/constants/tools/index.js +33 -2
- package/dist/constants/tools/index.js.map +1 -1
- package/dist/manager/run/persistence.d.ts.map +1 -1
- package/dist/manager/run/persistence.js +35 -5
- package/dist/manager/run/persistence.js.map +1 -1
- package/dist/persona/assembler.d.ts +1 -0
- package/dist/persona/assembler.d.ts.map +1 -1
- package/dist/persona/assembler.js +37 -6
- package/dist/persona/assembler.js.map +1 -1
- package/dist/provider/collect.d.ts.map +1 -1
- package/dist/provider/collect.js +4 -1
- package/dist/provider/collect.js.map +1 -1
- package/dist/provider/collect.test.js +2 -2
- package/dist/public-runtime.d.ts +6 -5
- package/dist/public-runtime.d.ts.map +1 -1
- package/dist/public-runtime.js +6 -5
- package/dist/public-runtime.js.map +1 -1
- package/dist/public-tools.d.ts +2 -0
- package/dist/public-tools.d.ts.map +1 -1
- package/dist/public-tools.js +2 -0
- package/dist/public-tools.js.map +1 -1
- package/dist/public-types.d.ts +3 -0
- package/dist/public-types.d.ts.map +1 -1
- package/dist/registry/index.d.ts +2 -0
- package/dist/registry/index.d.ts.map +1 -1
- package/dist/registry/index.js +1 -0
- package/dist/registry/index.js.map +1 -1
- package/dist/registry/tool/execute.d.ts +27 -0
- package/dist/registry/tool/execute.d.ts.map +1 -1
- package/dist/registry/tool/execute.js +192 -7
- package/dist/registry/tool/execute.js.map +1 -1
- package/dist/registry/tool/execute.test.d.ts +9 -4
- package/dist/registry/tool/execute.test.d.ts.map +1 -1
- package/dist/registry/tool/execute.test.js +178 -7
- package/dist/registry/tool/execute.test.js.map +1 -1
- package/dist/registry/toolset/catalog.d.ts +42 -0
- package/dist/registry/toolset/catalog.d.ts.map +1 -0
- package/dist/registry/toolset/catalog.js +217 -0
- package/dist/registry/toolset/catalog.js.map +1 -0
- package/dist/registry/toolset/catalog.test.d.ts +2 -0
- package/dist/registry/toolset/catalog.test.d.ts.map +1 -0
- package/dist/registry/toolset/catalog.test.js +147 -0
- package/dist/registry/toolset/catalog.test.js.map +1 -0
- package/dist/runtime/query/__tests__/deferred-tools.test.d.ts +2 -0
- package/dist/runtime/query/__tests__/deferred-tools.test.d.ts.map +1 -0
- package/dist/runtime/query/__tests__/deferred-tools.test.js +185 -0
- package/dist/runtime/query/__tests__/deferred-tools.test.js.map +1 -0
- package/dist/runtime/query/__tests__/executor-concurrency.test.d.ts +2 -0
- package/dist/runtime/query/__tests__/executor-concurrency.test.d.ts.map +1 -0
- package/dist/runtime/query/__tests__/executor-concurrency.test.js +98 -0
- package/dist/runtime/query/__tests__/executor-concurrency.test.js.map +1 -0
- package/dist/runtime/query/__tests__/executor-plugin-hooks.test.js +38 -3
- package/dist/runtime/query/__tests__/executor-plugin-hooks.test.js.map +1 -1
- package/dist/runtime/query/__tests__/long-document-flow.test.d.ts +2 -0
- package/dist/runtime/query/__tests__/long-document-flow.test.d.ts.map +1 -0
- package/dist/runtime/query/__tests__/long-document-flow.test.js +154 -0
- package/dist/runtime/query/__tests__/long-document-flow.test.js.map +1 -0
- package/dist/runtime/query/__tests__/prompt.test.js +51 -2
- package/dist/runtime/query/__tests__/prompt.test.js.map +1 -1
- package/dist/runtime/query/__tests__/stream-recovery.test.d.ts +2 -0
- package/dist/runtime/query/__tests__/stream-recovery.test.d.ts.map +1 -0
- package/dist/runtime/query/__tests__/stream-recovery.test.js +126 -0
- package/dist/runtime/query/__tests__/stream-recovery.test.js.map +1 -0
- package/dist/runtime/query/continuation.d.ts +16 -0
- package/dist/runtime/query/continuation.d.ts.map +1 -0
- package/dist/runtime/query/continuation.js +16 -0
- package/dist/runtime/query/continuation.js.map +1 -0
- package/dist/runtime/query/executor.d.ts +3 -0
- package/dist/runtime/query/executor.d.ts.map +1 -1
- package/dist/runtime/query/executor.js +71 -3
- package/dist/runtime/query/executor.js.map +1 -1
- package/dist/runtime/query/index.d.ts +9 -0
- package/dist/runtime/query/index.d.ts.map +1 -1
- package/dist/runtime/query/index.js +42 -5
- package/dist/runtime/query/index.js.map +1 -1
- package/dist/runtime/query/iteration/index.d.ts +24 -0
- package/dist/runtime/query/iteration/index.d.ts.map +1 -1
- package/dist/runtime/query/iteration/index.js +326 -66
- package/dist/runtime/query/iteration/index.js.map +1 -1
- package/dist/runtime/query/iteration/phases/checkpoint.d.ts.map +1 -1
- package/dist/runtime/query/iteration/phases/checkpoint.js +2 -2
- package/dist/runtime/query/iteration/phases/checkpoint.js.map +1 -1
- package/dist/runtime/query/iteration/phases/compaction.d.ts +8 -0
- package/dist/runtime/query/iteration/phases/compaction.d.ts.map +1 -1
- package/dist/runtime/query/iteration/phases/compaction.js +84 -3
- package/dist/runtime/query/iteration/phases/compaction.js.map +1 -1
- package/dist/runtime/query/iteration/phases/context.d.ts +22 -0
- package/dist/runtime/query/iteration/phases/context.d.ts.map +1 -1
- package/dist/runtime/query/iteration/phases/context.js +53 -0
- package/dist/runtime/query/iteration/phases/context.js.map +1 -1
- package/dist/runtime/query/iteration/phases/hitl-abort-race.test.d.ts +2 -0
- package/dist/runtime/query/iteration/phases/hitl-abort-race.test.d.ts.map +1 -0
- package/dist/runtime/query/iteration/phases/hitl-abort-race.test.js +73 -0
- package/dist/runtime/query/iteration/phases/hitl-abort-race.test.js.map +1 -0
- package/dist/runtime/query/iteration/phases/hitl-answer-question.test.d.ts +2 -0
- package/dist/runtime/query/iteration/phases/hitl-answer-question.test.d.ts.map +1 -0
- package/dist/runtime/query/iteration/phases/hitl-answer-question.test.js +120 -0
- package/dist/runtime/query/iteration/phases/hitl-answer-question.test.js.map +1 -0
- package/dist/runtime/query/iteration/phases/tool-review.d.ts +1 -1
- package/dist/runtime/query/iteration/phases/tool-review.d.ts.map +1 -1
- package/dist/runtime/query/iteration/phases/tool-review.js +7 -2
- package/dist/runtime/query/iteration/phases/tool-review.js.map +1 -1
- package/dist/runtime/query/iteration/phases/working-memory-compaction.test.d.ts +19 -0
- package/dist/runtime/query/iteration/phases/working-memory-compaction.test.d.ts.map +1 -0
- package/dist/runtime/query/iteration/phases/working-memory-compaction.test.js +177 -0
- package/dist/runtime/query/iteration/phases/working-memory-compaction.test.js.map +1 -0
- package/dist/runtime/query/iteration/phases/working-memory.d.ts +31 -0
- package/dist/runtime/query/iteration/phases/working-memory.d.ts.map +1 -0
- package/dist/runtime/query/iteration/phases/working-memory.js +81 -0
- package/dist/runtime/query/iteration/phases/working-memory.js.map +1 -0
- package/dist/runtime/query/prompt.d.ts.map +1 -1
- package/dist/runtime/query/prompt.js +21 -1
- package/dist/runtime/query/prompt.js.map +1 -1
- package/dist/runtime/query/tooling.d.ts +1 -0
- package/dist/runtime/query/tooling.d.ts.map +1 -1
- package/dist/runtime/query/tooling.js +1 -0
- package/dist/runtime/query/tooling.js.map +1 -1
- package/dist/sandbox/provider/local.d.ts.map +1 -1
- package/dist/sandbox/provider/local.js +32 -1
- package/dist/sandbox/provider/local.js.map +1 -1
- package/dist/session/workspace/__tests__/shared-run.test.d.ts +2 -0
- package/dist/session/workspace/__tests__/shared-run.test.d.ts.map +1 -0
- package/dist/session/workspace/__tests__/shared-run.test.js +147 -0
- package/dist/session/workspace/__tests__/shared-run.test.js.map +1 -0
- package/dist/session/workspace/index.d.ts +2 -0
- package/dist/session/workspace/index.d.ts.map +1 -1
- package/dist/session/workspace/index.js +1 -0
- package/dist/session/workspace/index.js.map +1 -1
- package/dist/session/workspace/shared-run.d.ts +81 -0
- package/dist/session/workspace/shared-run.d.ts.map +1 -0
- package/dist/session/workspace/shared-run.js +252 -0
- package/dist/session/workspace/shared-run.js.map +1 -0
- package/dist/skills/loader.d.ts.map +1 -1
- package/dist/skills/loader.js +36 -6
- package/dist/skills/loader.js.map +1 -1
- package/dist/skills/loader.test.d.ts +2 -0
- package/dist/skills/loader.test.d.ts.map +1 -0
- package/dist/skills/loader.test.js +65 -0
- package/dist/skills/loader.test.js.map +1 -0
- package/dist/streaming/coalesce.test.js +1 -1
- package/dist/streaming/coalesce.test.js.map +1 -1
- package/dist/tools/builtins/__tests__/edit.test.d.ts +2 -0
- package/dist/tools/builtins/__tests__/edit.test.d.ts.map +1 -0
- package/dist/tools/builtins/__tests__/edit.test.js +38 -0
- package/dist/tools/builtins/__tests__/edit.test.js.map +1 -0
- package/dist/tools/builtins/__tests__/payload-budget.test.d.ts +2 -0
- package/dist/tools/builtins/__tests__/payload-budget.test.d.ts.map +1 -0
- package/dist/tools/builtins/__tests__/payload-budget.test.js +70 -0
- package/dist/tools/builtins/__tests__/payload-budget.test.js.map +1 -0
- package/dist/tools/builtins/__tests__/read-file.test.d.ts +2 -0
- package/dist/tools/builtins/__tests__/read-file.test.d.ts.map +1 -0
- package/dist/tools/builtins/__tests__/read-file.test.js +33 -0
- package/dist/tools/builtins/__tests__/read-file.test.js.map +1 -0
- package/dist/tools/builtins/__tests__/verify-outputs.test.d.ts +2 -0
- package/dist/tools/builtins/__tests__/verify-outputs.test.d.ts.map +1 -0
- package/dist/tools/builtins/__tests__/verify-outputs.test.js +52 -0
- package/dist/tools/builtins/__tests__/verify-outputs.test.js.map +1 -0
- package/dist/tools/builtins/__tests__/write-file.test.d.ts +2 -0
- package/dist/tools/builtins/__tests__/write-file.test.d.ts.map +1 -0
- package/dist/tools/builtins/__tests__/write-file.test.js +74 -0
- package/dist/tools/builtins/__tests__/write-file.test.js.map +1 -0
- package/dist/tools/builtins/bash.d.ts.map +1 -1
- package/dist/tools/builtins/bash.js +40 -7
- package/dist/tools/builtins/bash.js.map +1 -1
- package/dist/tools/builtins/edit.d.ts +5 -2
- package/dist/tools/builtins/edit.d.ts.map +1 -1
- package/dist/tools/builtins/edit.js +114 -18
- package/dist/tools/builtins/edit.js.map +1 -1
- package/dist/tools/builtins/index.d.ts +1 -0
- package/dist/tools/builtins/index.d.ts.map +1 -1
- package/dist/tools/builtins/index.js +13 -13
- package/dist/tools/builtins/index.js.map +1 -1
- package/dist/tools/builtins/read-file.d.ts +1 -0
- package/dist/tools/builtins/read-file.d.ts.map +1 -1
- package/dist/tools/builtins/read-file.js +73 -10
- package/dist/tools/builtins/read-file.js.map +1 -1
- package/dist/tools/builtins/search-tools.d.ts.map +1 -1
- package/dist/tools/builtins/search-tools.js +35 -7
- package/dist/tools/builtins/search-tools.js.map +1 -1
- package/dist/tools/builtins/verify-outputs.d.ts +5 -0
- package/dist/tools/builtins/verify-outputs.d.ts.map +1 -0
- package/dist/tools/builtins/verify-outputs.js +103 -0
- package/dist/tools/builtins/verify-outputs.js.map +1 -0
- package/dist/tools/builtins/write-file.d.ts +3 -2
- package/dist/tools/builtins/write-file.d.ts.map +1 -1
- package/dist/tools/builtins/write-file.js +72 -12
- package/dist/tools/builtins/write-file.js.map +1 -1
- package/dist/tools/coordinator/__tests__/agent.test.d.ts +15 -0
- package/dist/tools/coordinator/__tests__/agent.test.d.ts.map +1 -0
- package/dist/tools/coordinator/__tests__/agent.test.js +142 -0
- package/dist/tools/coordinator/__tests__/agent.test.js.map +1 -0
- package/dist/tools/coordinator/__tests__/approve-plan.test.d.ts +2 -0
- package/dist/tools/coordinator/__tests__/approve-plan.test.d.ts.map +1 -0
- package/dist/tools/coordinator/__tests__/approve-plan.test.js +114 -0
- package/dist/tools/coordinator/__tests__/approve-plan.test.js.map +1 -0
- package/dist/tools/coordinator/__tests__/ask-user-question.test.d.ts +2 -0
- package/dist/tools/coordinator/__tests__/ask-user-question.test.d.ts.map +1 -0
- package/dist/tools/coordinator/__tests__/ask-user-question.test.js +309 -0
- package/dist/tools/coordinator/__tests__/ask-user-question.test.js.map +1 -0
- package/dist/tools/coordinator/__tests__/task-list.test.d.ts +13 -0
- package/dist/tools/coordinator/__tests__/task-list.test.d.ts.map +1 -0
- package/dist/tools/coordinator/__tests__/task-list.test.js +162 -0
- package/dist/tools/coordinator/__tests__/task-list.test.js.map +1 -0
- package/dist/tools/coordinator/agent.d.ts +34 -0
- package/dist/tools/coordinator/agent.d.ts.map +1 -0
- package/dist/tools/coordinator/agent.js +107 -0
- package/dist/tools/coordinator/agent.js.map +1 -0
- package/dist/tools/coordinator/index.d.ts +15 -0
- package/dist/tools/coordinator/index.d.ts.map +1 -1
- package/dist/tools/coordinator/index.js +305 -35
- package/dist/tools/coordinator/index.js.map +1 -1
- package/dist/types/agent/base.d.ts +8 -0
- package/dist/types/agent/base.d.ts.map +1 -1
- package/dist/types/agent/index.d.ts +1 -0
- package/dist/types/agent/index.d.ts.map +1 -1
- package/dist/types/agent/index.js +1 -0
- package/dist/types/agent/index.js.map +1 -1
- package/dist/types/agent/reactive.d.ts +38 -0
- package/dist/types/agent/reactive.d.ts.map +1 -1
- package/dist/types/agent/supervisor.d.ts +32 -0
- package/dist/types/agent/supervisor.d.ts.map +1 -1
- package/dist/types/agent/working-memory.d.ts +29 -0
- package/dist/types/agent/working-memory.d.ts.map +1 -0
- package/dist/types/agent/working-memory.js +2 -0
- package/dist/types/agent/working-memory.js.map +1 -0
- package/dist/types/common/index.d.ts +10 -0
- package/dist/types/common/index.d.ts.map +1 -1
- package/dist/types/common/index.js +18 -0
- package/dist/types/common/index.js.map +1 -1
- package/dist/types/common/usage-merge.test.d.ts +2 -0
- package/dist/types/common/usage-merge.test.d.ts.map +1 -0
- package/dist/types/common/usage-merge.test.js +48 -0
- package/dist/types/common/usage-merge.test.js.map +1 -0
- package/dist/types/connector/mcp.d.ts +9 -2
- package/dist/types/connector/mcp.d.ts.map +1 -1
- package/dist/types/hitl/index.d.ts +42 -0
- package/dist/types/hitl/index.d.ts.map +1 -1
- package/dist/types/hitl/index.js +11 -0
- package/dist/types/hitl/index.js.map +1 -1
- package/dist/types/message/index.d.ts +22 -1
- package/dist/types/message/index.d.ts.map +1 -1
- package/dist/types/message/index.js +7 -2
- package/dist/types/message/index.js.map +1 -1
- package/dist/types/provider/chat.d.ts +11 -9
- package/dist/types/provider/chat.d.ts.map +1 -1
- package/dist/types/run/events.d.ts +6 -0
- package/dist/types/run/events.d.ts.map +1 -1
- package/dist/types/run/events.js.map +1 -1
- package/dist/types/sandbox/index.d.ts +193 -0
- package/dist/types/sandbox/index.d.ts.map +1 -1
- package/dist/types/sandbox/index.js.map +1 -1
- package/dist/types/skills/index.d.ts +2 -0
- package/dist/types/skills/index.d.ts.map +1 -1
- package/dist/types/tool/index.d.ts +22 -0
- package/dist/types/tool/index.d.ts.map +1 -1
- package/dist/types/toolset/index.d.ts +71 -0
- package/dist/types/toolset/index.d.ts.map +1 -0
- package/dist/types/toolset/index.js +2 -0
- package/dist/types/toolset/index.js.map +1 -0
- package/dist/types/workspace/index.d.ts +1 -0
- package/dist/types/workspace/index.d.ts.map +1 -1
- package/dist/types/workspace/shared-run.d.ts +61 -0
- package/dist/types/workspace/shared-run.d.ts.map +1 -0
- package/dist/types/workspace/shared-run.js +2 -0
- package/dist/types/workspace/shared-run.js.map +1 -0
- package/dist/verification/index.d.ts +1 -0
- package/dist/verification/index.d.ts.map +1 -1
- package/dist/verification/index.js +1 -0
- package/dist/verification/index.js.map +1 -1
- package/dist/verification/presets.d.ts +53 -0
- package/dist/verification/presets.d.ts.map +1 -0
- package/dist/verification/presets.js +70 -0
- package/dist/verification/presets.js.map +1 -0
- package/dist/verification/presets.test.d.ts +16 -0
- package/dist/verification/presets.test.d.ts.map +1 -0
- package/dist/verification/presets.test.js +79 -0
- package/dist/verification/presets.test.js.map +1 -0
- package/dist/version.d.ts.map +1 -1
- package/dist/version.js +25 -4
- package/dist/version.js.map +1 -1
- package/package.json +3 -2
- package/src/advisory/executor.test.ts +7 -4
- package/src/advisory/executor.ts +11 -2
- package/src/agents/ReactiveAgent.ts +7 -0
- package/src/agents/SupervisorAgent.ts +63 -16
- package/src/agents/__tests__/supervisor-ledger.test.ts +185 -0
- package/src/bridge/sse/mapper.test.ts +2 -2
- package/src/config/runtime.ts +9 -0
- package/src/connector/builtins/webhook.test.ts +21 -0
- package/src/connector/index.ts +1 -0
- package/src/connector/mcp/client.ts +4 -0
- package/src/connector/mcp/index.ts +1 -0
- package/src/connector/mcp/streamable-http.test.ts +169 -0
- package/src/connector/mcp/streamable-http.ts +152 -0
- package/src/constants/compaction/index.ts +8 -3
- package/src/constants/sandbox/index.ts +37 -0
- package/src/constants/tools/index.ts +33 -2
- package/src/manager/run/persistence.ts +34 -6
- package/src/persona/assembler.ts +43 -8
- package/src/provider/collect.test.ts +2 -2
- package/src/provider/collect.ts +4 -1
- package/src/public-runtime.ts +15 -1
- package/src/public-tools.ts +2 -0
- package/src/public-types.ts +7 -0
- package/src/registry/index.ts +7 -0
- package/src/registry/tool/execute.test.ts +211 -7
- package/src/registry/tool/execute.ts +199 -11
- package/src/registry/toolset/catalog.test.ts +165 -0
- package/src/registry/toolset/catalog.ts +283 -0
- package/src/runtime/query/__tests__/deferred-tools.test.ts +230 -0
- package/src/runtime/query/__tests__/executor-concurrency.test.ts +122 -0
- package/src/runtime/query/__tests__/executor-plugin-hooks.test.ts +48 -3
- package/src/runtime/query/__tests__/long-document-flow.test.ts +180 -0
- package/src/runtime/query/__tests__/prompt.test.ts +55 -2
- package/src/runtime/query/__tests__/stream-recovery.test.ts +156 -0
- package/src/runtime/query/continuation.ts +16 -0
- package/src/runtime/query/executor.ts +82 -13
- package/src/runtime/query/index.ts +58 -5
- package/src/runtime/query/iteration/index.ts +361 -74
- package/src/runtime/query/iteration/phases/checkpoint.ts +7 -2
- package/src/runtime/query/iteration/phases/compaction.ts +89 -3
- package/src/runtime/query/iteration/phases/context.ts +68 -0
- package/src/runtime/query/iteration/phases/hitl-abort-race.test.ts +86 -0
- package/src/runtime/query/iteration/phases/hitl-answer-question.test.ts +139 -0
- package/src/runtime/query/iteration/phases/tool-review.ts +7 -3
- package/src/runtime/query/iteration/phases/working-memory-compaction.test.ts +215 -0
- package/src/runtime/query/iteration/phases/working-memory.ts +86 -0
- package/src/runtime/query/prompt.ts +17 -1
- package/src/runtime/query/tooling.ts +2 -0
- package/src/sandbox/provider/local.ts +33 -0
- package/src/session/workspace/__tests__/shared-run.test.ts +181 -0
- package/src/session/workspace/index.ts +6 -0
- package/src/session/workspace/shared-run.ts +317 -0
- package/src/skills/loader.test.ts +89 -0
- package/src/skills/loader.ts +37 -6
- package/src/streaming/coalesce.test.ts +4 -4
- package/src/tools/builtins/__tests__/edit.test.ts +57 -0
- package/src/tools/builtins/__tests__/payload-budget.test.ts +92 -0
- package/src/tools/builtins/__tests__/read-file.test.ts +43 -0
- package/src/tools/builtins/__tests__/verify-outputs.test.ts +71 -0
- package/src/tools/builtins/__tests__/write-file.test.ts +97 -0
- package/src/tools/builtins/bash.ts +48 -7
- package/src/tools/builtins/edit.ts +162 -27
- package/src/tools/builtins/index.ts +13 -13
- package/src/tools/builtins/read-file.ts +81 -10
- package/src/tools/builtins/search-tools.ts +42 -7
- package/src/tools/builtins/verify-outputs.ts +126 -0
- package/src/tools/builtins/write-file.ts +83 -14
- package/src/tools/coordinator/__tests__/agent.test.ts +172 -0
- package/src/tools/coordinator/__tests__/approve-plan.test.ts +139 -0
- package/src/tools/coordinator/__tests__/ask-user-question.test.ts +370 -0
- package/src/tools/coordinator/__tests__/task-list.test.ts +182 -0
- package/src/tools/coordinator/agent.ts +157 -0
- package/src/tools/coordinator/index.ts +355 -39
- package/src/types/agent/base.ts +8 -0
- package/src/types/agent/index.ts +1 -0
- package/src/types/agent/reactive.ts +42 -0
- package/src/types/agent/supervisor.ts +36 -0
- package/src/types/agent/working-memory.ts +29 -0
- package/src/types/common/index.ts +19 -0
- package/src/types/common/usage-merge.test.ts +53 -0
- package/src/types/connector/mcp.ts +14 -2
- package/src/types/hitl/index.ts +52 -1
- package/src/types/message/index.ts +32 -2
- package/src/types/provider/chat.ts +12 -9
- package/src/types/run/events.ts +6 -0
- package/src/types/sandbox/index.ts +219 -0
- package/src/types/skills/index.ts +4 -0
- package/src/types/tool/index.ts +24 -0
- package/src/types/toolset/index.ts +86 -0
- package/src/types/workspace/index.ts +9 -0
- package/src/types/workspace/shared-run.ts +65 -0
- package/src/verification/index.ts +1 -0
- package/src/verification/presets.test.ts +112 -0
- package/src/verification/presets.ts +72 -0
- package/src/version.ts +26 -4
|
@@ -0,0 +1,185 @@
|
|
|
1
|
+
import { mkdtemp, rm } from 'node:fs/promises';
|
|
2
|
+
import { tmpdir } from 'node:os';
|
|
3
|
+
import { join } from 'node:path';
|
|
4
|
+
import { afterEach, describe, expect, it } from 'vitest';
|
|
5
|
+
import { z } from 'zod';
|
|
6
|
+
import { ToolRegistry } from '../../../registry/tool/execute.js';
|
|
7
|
+
import { SearchToolsTool } from '../../../tools/builtins/search-tools.js';
|
|
8
|
+
import { createUserMessage } from '../../../types/message/index.js';
|
|
9
|
+
import { drainQuery } from '../index.js';
|
|
10
|
+
const ZERO_USAGE = {
|
|
11
|
+
promptTokens: 0,
|
|
12
|
+
completionTokens: 0,
|
|
13
|
+
totalTokens: 0,
|
|
14
|
+
cachedTokens: 0,
|
|
15
|
+
cacheWriteTokens: 0,
|
|
16
|
+
};
|
|
17
|
+
class CapturingProvider {
|
|
18
|
+
id = 'capturing';
|
|
19
|
+
name = 'Capturing Provider';
|
|
20
|
+
lastParams;
|
|
21
|
+
async *chatStream(params) {
|
|
22
|
+
this.lastParams = params;
|
|
23
|
+
yield {
|
|
24
|
+
id: 'msg_1',
|
|
25
|
+
delta: { content: 'done' },
|
|
26
|
+
};
|
|
27
|
+
yield {
|
|
28
|
+
id: 'msg_1',
|
|
29
|
+
delta: {},
|
|
30
|
+
finishReason: 'stop',
|
|
31
|
+
usage: ZERO_USAGE,
|
|
32
|
+
};
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
function registerDeferredDocumentTool(tools, name = 'generate_document') {
|
|
36
|
+
tools.register({
|
|
37
|
+
name,
|
|
38
|
+
description: 'Generate a project document by document id.',
|
|
39
|
+
inputSchema: z.object({
|
|
40
|
+
documentId: z.string(),
|
|
41
|
+
}),
|
|
42
|
+
execute: async () => ({ success: true, output: 'generated' }),
|
|
43
|
+
}, 'deferred');
|
|
44
|
+
}
|
|
45
|
+
describe('query deferred tool discovery', () => {
|
|
46
|
+
let workdirs = [];
|
|
47
|
+
afterEach(async () => {
|
|
48
|
+
await Promise.all(workdirs.map((dir) => rm(dir, { recursive: true, force: true })));
|
|
49
|
+
workdirs = [];
|
|
50
|
+
});
|
|
51
|
+
it('auto-exposes search_tools when deferred tools are registered', async () => {
|
|
52
|
+
const provider = new CapturingProvider();
|
|
53
|
+
const tools = new ToolRegistry();
|
|
54
|
+
registerDeferredDocumentTool(tools);
|
|
55
|
+
const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-deferred-tools-'));
|
|
56
|
+
workdirs.push(workingDirectory);
|
|
57
|
+
const run = await drainQuery({
|
|
58
|
+
provider,
|
|
59
|
+
tools,
|
|
60
|
+
runConfig: {
|
|
61
|
+
model: 'mock-model',
|
|
62
|
+
timeoutMs: 5_000,
|
|
63
|
+
tokenBudget: 100_000,
|
|
64
|
+
maxIterations: 1,
|
|
65
|
+
maxResponseTokens: 256,
|
|
66
|
+
},
|
|
67
|
+
agentId: 'agent_test',
|
|
68
|
+
agentName: 'Test Agent',
|
|
69
|
+
messages: [createUserMessage('what tools can you use?')],
|
|
70
|
+
workingDirectory,
|
|
71
|
+
sessionId: 'ses_deferred_tools',
|
|
72
|
+
threadId: 'thd_deferred_tools',
|
|
73
|
+
projectId: 'prj_deferred_tools',
|
|
74
|
+
tenantId: 'tnt_deferred_tools',
|
|
75
|
+
});
|
|
76
|
+
expect(run.status).toBe('completed');
|
|
77
|
+
expect(tools.has(SearchToolsTool.name)).toBe(true);
|
|
78
|
+
expect(tools.getAvailability(SearchToolsTool.name)).toBe('active');
|
|
79
|
+
expect(tools.getAvailability('generate_document')).toBe('deferred');
|
|
80
|
+
const toolNames = provider.lastParams?.tools?.map((tool) => tool.function.name).sort() ?? [];
|
|
81
|
+
expect(toolNames).toEqual([SearchToolsTool.name]);
|
|
82
|
+
const systemPrompt = (provider.lastParams?.messages ?? [])
|
|
83
|
+
.filter((message) => message.role === 'system')
|
|
84
|
+
.map((message) => message.content)
|
|
85
|
+
.join('\n');
|
|
86
|
+
expect(systemPrompt).toContain('Use search_tools to load these before use:');
|
|
87
|
+
expect(systemPrompt).toContain('- generate_document');
|
|
88
|
+
expect(systemPrompt).not.toContain('Deferred tools are discoverable but not executable until the runtime activates them');
|
|
89
|
+
});
|
|
90
|
+
it('keeps search_tools executable when allowedTools names a deferred tool', async () => {
|
|
91
|
+
const provider = new CapturingProvider();
|
|
92
|
+
const tools = new ToolRegistry();
|
|
93
|
+
registerDeferredDocumentTool(tools);
|
|
94
|
+
const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-deferred-tools-'));
|
|
95
|
+
workdirs.push(workingDirectory);
|
|
96
|
+
const run = await drainQuery({
|
|
97
|
+
provider,
|
|
98
|
+
tools,
|
|
99
|
+
allowedTools: ['generate_document'],
|
|
100
|
+
runConfig: {
|
|
101
|
+
model: 'mock-model',
|
|
102
|
+
timeoutMs: 5_000,
|
|
103
|
+
tokenBudget: 100_000,
|
|
104
|
+
maxIterations: 1,
|
|
105
|
+
maxResponseTokens: 256,
|
|
106
|
+
},
|
|
107
|
+
agentId: 'agent_test',
|
|
108
|
+
agentName: 'Test Agent',
|
|
109
|
+
messages: [createUserMessage('generate D-01')],
|
|
110
|
+
workingDirectory,
|
|
111
|
+
sessionId: 'ses_deferred_allowed_tools',
|
|
112
|
+
threadId: 'thd_deferred_allowed_tools',
|
|
113
|
+
projectId: 'prj_deferred_allowed_tools',
|
|
114
|
+
tenantId: 'tnt_deferred_allowed_tools',
|
|
115
|
+
});
|
|
116
|
+
expect(run.status).toBe('completed');
|
|
117
|
+
expect(tools.getAvailability('generate_document')).toBe('deferred');
|
|
118
|
+
const toolNames = provider.lastParams?.tools?.map((tool) => tool.function.name).sort() ?? [];
|
|
119
|
+
expect(toolNames).toEqual([SearchToolsTool.name]);
|
|
120
|
+
const systemPrompt = (provider.lastParams?.messages ?? [])
|
|
121
|
+
.filter((message) => message.role === 'system')
|
|
122
|
+
.map((message) => message.content)
|
|
123
|
+
.join('\n');
|
|
124
|
+
expect(systemPrompt).toContain('Use search_tools to load these before use:');
|
|
125
|
+
expect(systemPrompt).toContain('- generate_document');
|
|
126
|
+
});
|
|
127
|
+
it('does not let search_tools reveal or activate deferred tools outside allowedTools', async () => {
|
|
128
|
+
const tools = new ToolRegistry();
|
|
129
|
+
registerDeferredDocumentTool(tools);
|
|
130
|
+
registerDeferredDocumentTool(tools, 'dangerous_purge_document');
|
|
131
|
+
// 'dangerous' matches only the out-of-allowlist tool ('delete'-style
|
|
132
|
+
// CRUD verbs are stop tokens and never match anything by themselves).
|
|
133
|
+
const result = await SearchToolsTool.execute({ query: 'dangerous' }, {
|
|
134
|
+
runId: 'run_deferred_allowed_tools',
|
|
135
|
+
workingDirectory: '/tmp',
|
|
136
|
+
abortSignal: new AbortController().signal,
|
|
137
|
+
env: {},
|
|
138
|
+
log: () => undefined,
|
|
139
|
+
toolRegistry: tools,
|
|
140
|
+
allowedTools: ['generate_document', SearchToolsTool.name],
|
|
141
|
+
});
|
|
142
|
+
expect(result.success).toBe(true);
|
|
143
|
+
expect(result.output).toContain('No deferred tools matching "dangerous"');
|
|
144
|
+
expect(result.output).not.toContain('dangerous_purge_document');
|
|
145
|
+
expect(tools.getAvailability('generate_document')).toBe('deferred');
|
|
146
|
+
expect(tools.getAvailability('dangerous_purge_document')).toBe('deferred');
|
|
147
|
+
});
|
|
148
|
+
it('activates only the top-5 ranked matches and reports near-misses without activating', async () => {
|
|
149
|
+
const tools = new ToolRegistry();
|
|
150
|
+
// Eight deferred tools that all match "invoice" equally by name; the
|
|
151
|
+
// alphabetical tie-break makes the top-5 cut deterministic.
|
|
152
|
+
const names = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'].map((s) => `invoice_${s}`);
|
|
153
|
+
for (const name of names) {
|
|
154
|
+
tools.register({
|
|
155
|
+
name,
|
|
156
|
+
description: `Billing helper ${name.slice(-1)}.`,
|
|
157
|
+
inputSchema: z.object({ id: z.string() }),
|
|
158
|
+
execute: async () => ({ success: true, output: 'ok' }),
|
|
159
|
+
}, 'deferred');
|
|
160
|
+
}
|
|
161
|
+
const result = await SearchToolsTool.execute({ query: 'invoice' }, {
|
|
162
|
+
runId: 'run_deferred_top_k',
|
|
163
|
+
workingDirectory: '/tmp',
|
|
164
|
+
abortSignal: new AbortController().signal,
|
|
165
|
+
env: {},
|
|
166
|
+
log: () => undefined,
|
|
167
|
+
toolRegistry: tools,
|
|
168
|
+
});
|
|
169
|
+
expect(result.success).toBe(true);
|
|
170
|
+
expect(result.output).toContain('Activated 5 tool(s)');
|
|
171
|
+
expect(result.output).toContain('NOT loaded');
|
|
172
|
+
expect(result.data).toMatchObject({
|
|
173
|
+
activated: ['invoice_a', 'invoice_b', 'invoice_c', 'invoice_d', 'invoice_e'],
|
|
174
|
+
count: 5,
|
|
175
|
+
nearMisses: ['invoice_f', 'invoice_g', 'invoice_h'],
|
|
176
|
+
});
|
|
177
|
+
for (const name of ['invoice_a', 'invoice_b', 'invoice_c', 'invoice_d', 'invoice_e']) {
|
|
178
|
+
expect(tools.getAvailability(name)).toBe('active');
|
|
179
|
+
}
|
|
180
|
+
for (const name of ['invoice_f', 'invoice_g', 'invoice_h']) {
|
|
181
|
+
expect(tools.getAvailability(name)).toBe('deferred');
|
|
182
|
+
}
|
|
183
|
+
});
|
|
184
|
+
});
|
|
185
|
+
//# sourceMappingURL=deferred-tools.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"deferred-tools.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/deferred-tools.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACxD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAA;AAEzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAOnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,iBAAiB;IACb,EAAE,GAAG,WAAW,CAAA;IAChB,IAAI,GAAG,oBAAoB,CAAA;IACpC,UAAU,CAAuB;IAEjC,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;QACxB,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;SAC1B,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,SAAS,4BAA4B,CAAC,KAAmB,EAAE,IAAI,GAAG,mBAAmB;IACpF,KAAK,CAAC,QAAQ,CACb;QACC,IAAI;QACJ,WAAW,EAAE,6CAA6C;QAC1D,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;SACtB,CAAC;QACF,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;KAC7D,EACD,UAAU,CACV,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,CAAC;YACxD,gBAAgB;YAChB,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;YAC1C,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;SAC1C,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAClD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAClE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CACjC,qFAAqF,CACrF,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,YAAY,EAAE,CAAC,mBAAmB,CAAC;YACnC,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;YAC9C,gBAAgB;YAChB,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;YAClD,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;SAClD,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QACnC,4BAA4B,CAAC,KAAK,EAAE,0BAA0B,CAAC,CAAA;QAE/D,qEAAqE;QACrE,sEAAsE;QACtE,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,WAAW,EAAE,EACtB;YACC,KAAK,EAAE,4BAAqC;YAC5C,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;YACnB,YAAY,EAAE,CAAC,mBAAmB,EAAE,eAAe,CAAC,IAAI,CAAC;SACzD,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wCAAwC,CAAC,CAAA;QACzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAA;QAC/D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACnE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QACnG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,qEAAqE;QACrE,4DAA4D;QAC5D,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAA;QACjF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,KAAK,CAAC,QAAQ,CACb;gBACC,IAAI;gBACJ,WAAW,EAAE,kBAAkB,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG;gBAChD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACzC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aACtD,EACD,UAAU,CACV,CAAA;QACF,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,SAAS,EAAE,EACpB;YACC,KAAK,EAAE,oBAA6B;YACpC,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;SACnB,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAA;QAC7C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;YACjC,SAAS,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;YAC5E,KAAK,EAAE,CAAC;YACR,UAAU,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;SACnD,CAAC,CAAA;QACF,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACnD,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACrD,CAAC;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor-concurrency.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-concurrency.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
|
2
|
+
import { ActivityStore } from '../../../store/activity/memory.js';
|
|
3
|
+
import { ToolExecutor } from '../executor.js';
|
|
4
|
+
const mockRunId = 'run_test';
|
|
5
|
+
function makeLogger() {
|
|
6
|
+
const stub = { info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() };
|
|
7
|
+
return { ...stub, child: vi.fn(() => ({ ...stub, child: vi.fn() })) };
|
|
8
|
+
}
|
|
9
|
+
const delay = (ms) => new Promise((r) => setTimeout(r, ms));
|
|
10
|
+
/** Two tool calls in one assistant message (one batch). */
|
|
11
|
+
function twoCallResponse(name, a, b) {
|
|
12
|
+
return {
|
|
13
|
+
message: {
|
|
14
|
+
role: 'assistant',
|
|
15
|
+
content: null,
|
|
16
|
+
toolCalls: [
|
|
17
|
+
{ id: 'c1', type: 'function', function: { name, arguments: JSON.stringify(a) } },
|
|
18
|
+
{ id: 'c2', type: 'function', function: { name, arguments: JSON.stringify(b) } },
|
|
19
|
+
],
|
|
20
|
+
},
|
|
21
|
+
finishReason: 'tool_calls',
|
|
22
|
+
usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },
|
|
23
|
+
};
|
|
24
|
+
}
|
|
25
|
+
describe('ToolExecutor — concurrencySafe batching', () => {
|
|
26
|
+
let activityStore;
|
|
27
|
+
let emitEvent;
|
|
28
|
+
beforeEach(() => {
|
|
29
|
+
activityStore = new ActivityStore(mockRunId, {
|
|
30
|
+
enabled: true,
|
|
31
|
+
trackToolCalls: true,
|
|
32
|
+
trackLlmTurns: true,
|
|
33
|
+
});
|
|
34
|
+
emitEvent = async () => { };
|
|
35
|
+
});
|
|
36
|
+
it('serializes concurrency-unsafe tools so read-modify-write does not race', async () => {
|
|
37
|
+
// Shared mutable state, mutated via read → await → write (like `edit`).
|
|
38
|
+
let file = 'A';
|
|
39
|
+
const execute = vi.fn(async (_name, input) => {
|
|
40
|
+
const current = file; // read
|
|
41
|
+
await delay(10); // window a parallel run would exploit
|
|
42
|
+
file = current + input.add; // write
|
|
43
|
+
return { success: true, output: 'ok' };
|
|
44
|
+
});
|
|
45
|
+
const tools = {
|
|
46
|
+
register: vi.fn(),
|
|
47
|
+
unregister: vi.fn(),
|
|
48
|
+
execute,
|
|
49
|
+
// edit/write/bash declare concurrencySafe:false → isConcurrencySafe()=>false
|
|
50
|
+
get: vi.fn(() => ({ isConcurrencySafe: () => false })),
|
|
51
|
+
has: vi.fn(() => true),
|
|
52
|
+
listNames: vi.fn(() => []),
|
|
53
|
+
getAvailability: vi.fn(),
|
|
54
|
+
};
|
|
55
|
+
const exec = new ToolExecutor({
|
|
56
|
+
tools,
|
|
57
|
+
runId: mockRunId,
|
|
58
|
+
workingDirectory: '/tmp',
|
|
59
|
+
permissionMode: 'auto',
|
|
60
|
+
env: {},
|
|
61
|
+
abortSignal: new AbortController().signal,
|
|
62
|
+
}, activityStore, emitEvent, makeLogger());
|
|
63
|
+
await exec.executeBatch(twoCallResponse('edit', { add: 'B' }, { add: 'C' }));
|
|
64
|
+
// Serialized: A→AB→ABC. A racing run would lose one append (e.g. 'AC').
|
|
65
|
+
expect(file).toBe('ABC');
|
|
66
|
+
});
|
|
67
|
+
it('runs concurrency-safe tools in parallel', async () => {
|
|
68
|
+
let active = 0;
|
|
69
|
+
let maxActive = 0;
|
|
70
|
+
const execute = vi.fn(async () => {
|
|
71
|
+
active++;
|
|
72
|
+
maxActive = Math.max(maxActive, active);
|
|
73
|
+
await delay(10);
|
|
74
|
+
active--;
|
|
75
|
+
return { success: true, output: 'ok' };
|
|
76
|
+
});
|
|
77
|
+
const tools = {
|
|
78
|
+
register: vi.fn(),
|
|
79
|
+
unregister: vi.fn(),
|
|
80
|
+
execute,
|
|
81
|
+
get: vi.fn(() => ({ isConcurrencySafe: () => true })),
|
|
82
|
+
has: vi.fn(() => true),
|
|
83
|
+
listNames: vi.fn(() => []),
|
|
84
|
+
getAvailability: vi.fn(),
|
|
85
|
+
};
|
|
86
|
+
const exec = new ToolExecutor({
|
|
87
|
+
tools,
|
|
88
|
+
runId: mockRunId,
|
|
89
|
+
workingDirectory: '/tmp',
|
|
90
|
+
permissionMode: 'auto',
|
|
91
|
+
env: {},
|
|
92
|
+
abortSignal: new AbortController().signal,
|
|
93
|
+
}, activityStore, emitEvent, makeLogger());
|
|
94
|
+
await exec.executeBatch(twoCallResponse('grep', { p: '1' }, { p: '2' }));
|
|
95
|
+
expect(maxActive).toBe(2); // both ran at once
|
|
96
|
+
});
|
|
97
|
+
});
|
|
98
|
+
//# sourceMappingURL=executor-concurrency.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor-concurrency.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-concurrency.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAA;AAMjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAE7C,MAAM,SAAS,GAAG,UAAmB,CAAA;AAErC,SAAS,UAAU;IAClB,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAA;IAC7E,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAuB,CAAA;AAC3F,CAAC;AAED,MAAM,KAAK,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAA;AAEnE,2DAA2D;AAC3D,SAAS,eAAe,CAAC,IAAY,EAAE,CAAS,EAAE,CAAS;IAC1D,OAAO;QACN,OAAO,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE;gBACV,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChF,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE;aAChF;SACD;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;KACrC,CAAA;AAC5B,CAAC;AAED,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;IACxD,IAAI,aAA4B,CAAA;IAChC,IAAI,SAAyC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACf,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,EAAE;YAC5C,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACnB,CAAC,CAAA;QACF,SAAS,GAAG,KAAK,IAAI,EAAE,GAAE,CAAC,CAAA;IAC3B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACvF,wEAAwE;QACxE,IAAI,IAAI,GAAG,GAAG,CAAA;QACd,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAa,EAAE,KAAc,EAAE,EAAE;YAC7D,MAAM,OAAO,GAAG,IAAI,CAAA,CAAC,OAAO;YAC5B,MAAM,KAAK,CAAC,EAAE,CAAC,CAAA,CAAC,sCAAsC;YACtD,IAAI,GAAG,OAAO,GAAI,KAAyB,CAAC,GAAG,CAAA,CAAC,QAAQ;YACxD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,OAAO;YACP,6EAA6E;YAC7E,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;YACtD,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACW,CAAA;QAEpC,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;QAC5E,wEAAwE;QACxE,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACzB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACxD,IAAI,MAAM,GAAG,CAAC,CAAA;QACd,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YAChC,MAAM,EAAE,CAAA;YACR,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;YACvC,MAAM,KAAK,CAAC,EAAE,CAAC,CAAA;YACf,MAAM,EAAE,CAAA;YACR,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,OAAO;YACP,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YACrD,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACW,CAAA;QAEpC,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;QACxE,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA,CAAC,mBAAmB;IAC9C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -9,7 +9,10 @@ function makeLogger() {
|
|
|
9
9
|
error: vi.fn(),
|
|
10
10
|
debug: vi.fn(),
|
|
11
11
|
};
|
|
12
|
-
return {
|
|
12
|
+
return {
|
|
13
|
+
...stub,
|
|
14
|
+
child: vi.fn(() => ({ ...stub, child: vi.fn() })),
|
|
15
|
+
};
|
|
13
16
|
}
|
|
14
17
|
function makeToolRegistry(execute) {
|
|
15
18
|
return {
|
|
@@ -70,6 +73,32 @@ describe('ToolExecutor plugin hooks', () => {
|
|
|
70
73
|
const batch = await exec.executeBatch(buildResponse('echo', { msg: 'hi' }));
|
|
71
74
|
expect(batch.results[0]?.output).toBe('ok');
|
|
72
75
|
});
|
|
76
|
+
it('preserves tool stdout/stderr when a tool exits unsuccessfully', async () => {
|
|
77
|
+
const tools = makeToolRegistry(vi.fn(async () => ({
|
|
78
|
+
success: false,
|
|
79
|
+
output: 'STDOUT:\npartial result\n\nSTDERR:\nboom',
|
|
80
|
+
error: 'Command exited with code 1',
|
|
81
|
+
})));
|
|
82
|
+
const exec = new ToolExecutor({
|
|
83
|
+
tools,
|
|
84
|
+
runId: mockRunId,
|
|
85
|
+
workingDirectory: '/tmp',
|
|
86
|
+
permissionMode: 'auto',
|
|
87
|
+
env: {},
|
|
88
|
+
abortSignal: new AbortController().signal,
|
|
89
|
+
}, activityStore, emitEvent, makeLogger());
|
|
90
|
+
const batch = await exec.executeBatch(buildResponse('bash', { command: 'false' }));
|
|
91
|
+
expect(batch.results[0]?.output).toContain('STDOUT:\npartial result');
|
|
92
|
+
expect(batch.results[0]?.output).toContain('STDERR:\nboom');
|
|
93
|
+
expect(batch.results[0]?.output).toContain('Error: Command exited with code 1');
|
|
94
|
+
const completed = emitted.find((e) => e.type === 'tool_completed');
|
|
95
|
+
expect(completed).toMatchObject({
|
|
96
|
+
type: 'tool_completed',
|
|
97
|
+
toolName: 'bash',
|
|
98
|
+
result: expect.stringContaining('STDOUT:\npartial result'),
|
|
99
|
+
isError: true,
|
|
100
|
+
});
|
|
101
|
+
});
|
|
73
102
|
it('replaces input on pre_tool_use modify', async () => {
|
|
74
103
|
const executeMock = vi.fn(async () => ({ success: true, output: 'ok' }));
|
|
75
104
|
const tools = makeToolRegistry(executeMock);
|
|
@@ -89,7 +118,10 @@ describe('ToolExecutor plugin hooks', () => {
|
|
|
89
118
|
expect(executeMock).toHaveBeenCalledWith('echo', { replaced: true }, expect.any(Object));
|
|
90
119
|
});
|
|
91
120
|
it('skips registry execution and synthesizes output on pre_tool_use skip', async () => {
|
|
92
|
-
const executeMock = vi.fn(async () => ({
|
|
121
|
+
const executeMock = vi.fn(async () => ({
|
|
122
|
+
success: true,
|
|
123
|
+
output: 'should-not-run',
|
|
124
|
+
}));
|
|
93
125
|
const tools = makeToolRegistry(executeMock);
|
|
94
126
|
const pluginManager = makePluginManager(async (event) => event === 'pre_tool_use'
|
|
95
127
|
? [{ action: 'skip', reason: 'policy' }]
|
|
@@ -169,7 +201,10 @@ describe('ToolExecutor plugin hooks', () => {
|
|
|
169
201
|
]);
|
|
170
202
|
});
|
|
171
203
|
it('carries modified input into synthetic skip outcome (modify -> skip chain)', async () => {
|
|
172
|
-
const executeMock = vi.fn(async () => ({
|
|
204
|
+
const executeMock = vi.fn(async () => ({
|
|
205
|
+
success: true,
|
|
206
|
+
output: 'should-not-run',
|
|
207
|
+
}));
|
|
173
208
|
const tools = makeToolRegistry(executeMock);
|
|
174
209
|
const { PluginLifecycleManager } = await import('../../../plugin/lifecycle.js');
|
|
175
210
|
const realManager = new PluginLifecycleManager({
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"executor-plugin-hooks.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-plugin-hooks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAA;AAOjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAE7C,MAAM,SAAS,GAAG,UAAmB,CAAA;AAErC,SAAS,UAAU;IAClB,MAAM,IAAI,GAAG;QACZ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACd,CAAA;IACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAuB,CAAA;AAC3F,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAwC;IACjE,OAAO;QACN,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,OAAO;QACP,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC3B,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;QACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;KACW,CAAA;AACrC,CAAC;AAED,SAAS,iBAAiB,CACzB,YAAoD;IAEpD,OAAO,EAAE,YAAY,EAAuC,CAAA;AAC7D,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,IAAY;IACpD,OAAO;QACN,OAAO,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE;gBACV;oBACC,EAAE,EAAE,QAAQ;oBACZ,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;iBAC7D;aACD;SACD;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;KACrC,CAAA;AAC5B,CAAC;AAED,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IAC1C,IAAI,aAA4B,CAAA;IAChC,IAAI,OAAmB,CAAA;IACvB,IAAI,SAAyC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACf,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,EAAE;YAC5C,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACnB,CAAC,CAAA;QACF,OAAO,GAAG,EAAE,CAAA;QACZ,SAAS,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE;YACvB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAChB,CAAC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAC3E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QACxE,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAwB;YAC3E,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAA;IACzF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAA;QACpF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAwB;YAChE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAA;QACrF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,eAAe;YACxB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAwB;YACpE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAChD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACtB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,QAAQ,GAAG;YAChB,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;YACD,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;SACD,CAAA;QACD,oFAAoF;QACpF,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;YACtD,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;SACtD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1B,EAAE,IAAI,EAAE,CAAC,EAAE;SACX,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAA;QACpF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAA;gBACtD,CAAC,CAAQ;aACT;YACD;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,CAAA;gBACjD,CAAC,CAAQ;aACT;SACD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAChF,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAA;QACjF,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC7B,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SAC5B,CAAC,CAAA;QACF,oFAAoF;QACpF,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAA;QACvE,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,MAAM;YAChB,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc,CAAC,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAwB,CAAC,CAAC,CAAC,EAAE,CAC7E,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzE,4BAA4B,CAC5B,CAAA;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
1
|
+
{"version":3,"file":"executor-plugin-hooks.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-plugin-hooks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAA;AAOjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAE7C,MAAM,SAAS,GAAG,UAAmB,CAAA;AAErC,SAAS,UAAU;IAClB,MAAM,IAAI,GAAG;QACZ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACd,CAAA;IACD,OAAO;QACN,GAAG,IAAI;QACP,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;KAC5B,CAAA;AACvB,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAwC;IACjE,OAAO;QACN,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,OAAO;QACP,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC3B,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;QACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;KACW,CAAA;AACrC,CAAC;AAED,SAAS,iBAAiB,CACzB,YAAoD;IAEpD,OAAO,EAAE,YAAY,EAAuC,CAAA;AAC7D,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,IAAY;IACpD,OAAO;QACN,OAAO,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE;gBACV;oBACC,EAAE,EAAE,QAAQ;oBACZ,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;iBAC7D;aACD;SACD;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;KACrC,CAAA;AAC5B,CAAC;AAED,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IAC1C,IAAI,aAA4B,CAAA;IAChC,IAAI,OAAmB,CAAA;IACvB,IAAI,SAAyC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACf,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,EAAE;YAC5C,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACnB,CAAC,CAAA;QACF,OAAO,GAAG,EAAE,CAAA;QACZ,SAAS,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE;YACvB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAChB,CAAC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAC3E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,KAAK,GAAG,gBAAgB,CAC7B,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YAClB,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,0CAA0C;YAClD,KAAK,EAAE,4BAA4B;SACnC,CAAC,CAAC,CACH,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;QAClF,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAA;QACrE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAA;QAC3D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAA;QAE/E,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAA;QAClE,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,CAAC;YAC/B,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,MAAM;YAChB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,yBAAyB,CAAC;YAC1D,OAAO,EAAE,IAAI;SACb,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QACxE,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAwB;YAC3E,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAA;IACzF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACtC,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,gBAAgB;SACxB,CAAC,CAAC,CAAA;QACH,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAwB;YAChE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAA;QACrF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,eAAe;YACxB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAwB;YACpE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAChD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACtB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,QAAQ,GAAG;YAChB,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;YACD,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;SACD,CAAA;QACD,oFAAoF;QACpF,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;YACtD,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;SACtD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1B,EAAE,IAAI,EAAE,CAAC,EAAE;SACX,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACtC,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,gBAAgB;SACxB,CAAC,CAAC,CAAA;QACH,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAA;gBACtD,CAAC,CAAQ;aACT;YACD;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,CAAA;gBACjD,CAAC,CAAQ;aACT;SACD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAChF,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAA;QACjF,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC7B,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SAC5B,CAAC,CAAA;QACF,oFAAoF;QACpF,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAA;QACvE,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,MAAM;YAChB,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc,CAAC,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAwB,CAAC,CAAC,CAAC,EAAE,CAC7E,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzE,4BAA4B,CAC5B,CAAA;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"long-document-flow.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/long-document-flow.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,154 @@
|
|
|
1
|
+
import { readFile, rm } from 'node:fs/promises';
|
|
2
|
+
import { mkdtemp } from 'node:fs/promises';
|
|
3
|
+
import { tmpdir } from 'node:os';
|
|
4
|
+
import { join } from 'node:path';
|
|
5
|
+
import { afterEach, describe, expect, it } from 'vitest';
|
|
6
|
+
import { ToolRegistry } from '../../../registry/tool/execute.js';
|
|
7
|
+
import { EditTool, WriteFileTool } from '../../../tools/builtins/index.js';
|
|
8
|
+
import { createUserMessage } from '../../../types/message/index.js';
|
|
9
|
+
import { drainQuery } from '../index.js';
|
|
10
|
+
const ZERO_USAGE = {
|
|
11
|
+
promptTokens: 0,
|
|
12
|
+
completionTokens: 0,
|
|
13
|
+
totalTokens: 0,
|
|
14
|
+
cachedTokens: 0,
|
|
15
|
+
cacheWriteTokens: 0,
|
|
16
|
+
};
|
|
17
|
+
class ScriptedLongDocumentProvider {
|
|
18
|
+
steps;
|
|
19
|
+
id = 'scripted-long-document';
|
|
20
|
+
name = 'Scripted Long Document Provider';
|
|
21
|
+
calls = 0;
|
|
22
|
+
constructor(steps) {
|
|
23
|
+
this.steps = steps;
|
|
24
|
+
}
|
|
25
|
+
async *chatStream() {
|
|
26
|
+
const step = this.steps[this.calls];
|
|
27
|
+
this.calls += 1;
|
|
28
|
+
if (!step) {
|
|
29
|
+
yield {
|
|
30
|
+
id: 'msg_done',
|
|
31
|
+
delta: { content: 'Long document created and verified.' },
|
|
32
|
+
finishReason: 'stop',
|
|
33
|
+
usage: ZERO_USAGE,
|
|
34
|
+
};
|
|
35
|
+
return;
|
|
36
|
+
}
|
|
37
|
+
yield {
|
|
38
|
+
id: `msg_${this.calls}`,
|
|
39
|
+
delta: {
|
|
40
|
+
toolCalls: [
|
|
41
|
+
{
|
|
42
|
+
index: 0,
|
|
43
|
+
id: `toolu_${this.calls}`,
|
|
44
|
+
type: 'function',
|
|
45
|
+
function: { name: step.name },
|
|
46
|
+
},
|
|
47
|
+
],
|
|
48
|
+
},
|
|
49
|
+
};
|
|
50
|
+
yield {
|
|
51
|
+
id: `msg_${this.calls}`,
|
|
52
|
+
delta: {
|
|
53
|
+
toolCalls: [
|
|
54
|
+
{
|
|
55
|
+
index: 0,
|
|
56
|
+
id: `toolu_${this.calls}`,
|
|
57
|
+
function: { arguments: JSON.stringify(step.input) },
|
|
58
|
+
},
|
|
59
|
+
],
|
|
60
|
+
},
|
|
61
|
+
};
|
|
62
|
+
yield {
|
|
63
|
+
id: `msg_${this.calls}`,
|
|
64
|
+
delta: {},
|
|
65
|
+
finishReason: 'tool_calls',
|
|
66
|
+
usage: ZERO_USAGE,
|
|
67
|
+
};
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
describe('query long-document tool flow', () => {
|
|
71
|
+
const workdirs = [];
|
|
72
|
+
afterEach(async () => {
|
|
73
|
+
await Promise.all(workdirs.map((dir) => rm(dir, { recursive: true, force: true })));
|
|
74
|
+
workdirs.length = 0;
|
|
75
|
+
});
|
|
76
|
+
it('executes the canonical write skeleton plus bounded edit chunks flow end to end', async () => {
|
|
77
|
+
const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-query-long-doc-'));
|
|
78
|
+
workdirs.push(workingDirectory);
|
|
79
|
+
const chunks = Array.from({ length: 4 }, (_, sectionIndex) => {
|
|
80
|
+
const lines = Array.from({ length: 40 }, (_, lineIndex) => {
|
|
81
|
+
return `Section ${sectionIndex + 1}.${lineIndex + 1}: this content is intentionally long enough to require repeated bounded edit calls.`;
|
|
82
|
+
});
|
|
83
|
+
const chunk = [`## Section ${sectionIndex + 1}`, ...lines, ''].join('\n');
|
|
84
|
+
expect(chunk.length).toBeLessThan(12_000);
|
|
85
|
+
return chunk;
|
|
86
|
+
});
|
|
87
|
+
const provider = new ScriptedLongDocumentProvider([
|
|
88
|
+
{
|
|
89
|
+
name: 'write',
|
|
90
|
+
input: {
|
|
91
|
+
path: 'outputs/long-document-flow.md',
|
|
92
|
+
content: '# Long document flow\n\n{{BODY}}\n',
|
|
93
|
+
},
|
|
94
|
+
},
|
|
95
|
+
{
|
|
96
|
+
name: 'edit',
|
|
97
|
+
input: {
|
|
98
|
+
path: 'outputs/long-document-flow.md',
|
|
99
|
+
oldStr: '{{BODY}}',
|
|
100
|
+
newStr: chunks[0],
|
|
101
|
+
replace_all: false,
|
|
102
|
+
},
|
|
103
|
+
},
|
|
104
|
+
...chunks.slice(1).map((chunk) => ({
|
|
105
|
+
name: 'edit',
|
|
106
|
+
input: {
|
|
107
|
+
path: 'outputs/long-document-flow.md',
|
|
108
|
+
insertLine: 'end',
|
|
109
|
+
newStr: chunk,
|
|
110
|
+
replace_all: false,
|
|
111
|
+
},
|
|
112
|
+
})),
|
|
113
|
+
]);
|
|
114
|
+
const tools = new ToolRegistry();
|
|
115
|
+
tools.register(WriteFileTool);
|
|
116
|
+
tools.register(EditTool);
|
|
117
|
+
const events = [];
|
|
118
|
+
const run = await drainQuery({
|
|
119
|
+
provider,
|
|
120
|
+
tools,
|
|
121
|
+
runConfig: {
|
|
122
|
+
model: 'mock-model',
|
|
123
|
+
timeoutMs: 5_000,
|
|
124
|
+
tokenBudget: 100_000,
|
|
125
|
+
maxIterations: 8,
|
|
126
|
+
maxResponseTokens: 256,
|
|
127
|
+
},
|
|
128
|
+
agentId: 'agent_long_doc',
|
|
129
|
+
agentName: 'Long Document Agent',
|
|
130
|
+
messages: [createUserMessage('create a long document')],
|
|
131
|
+
workingDirectory,
|
|
132
|
+
sessionId: 'ses_long_document_flow',
|
|
133
|
+
threadId: 'thd_long_document_flow',
|
|
134
|
+
projectId: 'prj_long_document_flow',
|
|
135
|
+
tenantId: 'tnt_long_document_flow',
|
|
136
|
+
}, (event) => {
|
|
137
|
+
events.push(event);
|
|
138
|
+
});
|
|
139
|
+
const final = await readFile(join(workingDirectory, 'outputs/long-document-flow.md'), 'utf-8');
|
|
140
|
+
const executingTools = events
|
|
141
|
+
.filter((event) => event.type === 'tool_executing')
|
|
142
|
+
.map((event) => event.toolName);
|
|
143
|
+
expect(run.status).toBe('completed');
|
|
144
|
+
expect(run.result).toBe('Long document created and verified.');
|
|
145
|
+
expect(provider.calls).toBe(6);
|
|
146
|
+
expect(executingTools).toEqual(['write', 'edit', 'edit', 'edit', 'edit']);
|
|
147
|
+
expect(final).not.toContain('{{BODY}}');
|
|
148
|
+
expect(final.split('\n').length).toBeGreaterThan(160);
|
|
149
|
+
expect(final).toContain('## Section 1');
|
|
150
|
+
expect(final).toContain('## Section 4');
|
|
151
|
+
expect(final).toContain('Section 4.40');
|
|
152
|
+
});
|
|
153
|
+
});
|
|
154
|
+
//# sourceMappingURL=long-document-flow.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"long-document-flow.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/long-document-flow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAA;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExD,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAA;AAE1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAOD,MAAM,4BAA4B;IAKJ;IAJpB,EAAE,GAAG,wBAAwB,CAAA;IAC7B,IAAI,GAAG,iCAAiC,CAAA;IACjD,KAAK,GAAG,CAAC,CAAA;IAET,YAA6B,KAA0B;QAA1B,UAAK,GAAL,KAAK,CAAqB;IAAG,CAAC;IAE3D,KAAK,CAAC,CAAC,UAAU;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnC,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QAEf,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,MAAM;gBACL,EAAE,EAAE,UAAU;gBACd,KAAK,EAAE,EAAE,OAAO,EAAE,qCAAqC,EAAE;gBACzD,YAAY,EAAE,MAAM;gBACpB,KAAK,EAAE,UAAU;aACjB,CAAA;YACD,OAAM;QACP,CAAC;QAED,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;qBAC7B;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;qBACnD;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,YAAY;YAC1B,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,MAAM,QAAQ,GAAa,EAAE,CAAA;IAE7B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE;YAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE;gBACzD,OAAO,WAAW,YAAY,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,qFAAqF,CAAA;YACzI,CAAC,CAAC,CAAA;YACF,MAAM,KAAK,GAAG,CAAC,cAAc,YAAY,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACzE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;YACzC,OAAO,KAAK,CAAA;QACb,CAAC,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,IAAI,4BAA4B,CAAC;YACjD;gBACC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,OAAO,EAAE,oCAAoC;iBAC7C;aACD;YACD;gBACC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,MAAM,EAAE,UAAU;oBAClB,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;oBACjB,WAAW,EAAE,KAAK;iBAClB;aACD;YACD,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAClC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,UAAU,EAAE,KAAK;oBACjB,MAAM,EAAE,KAAK;oBACb,WAAW,EAAE,KAAK;iBAClB;aACD,CAAC,CAAC;SACH,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;QAC7B,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;QACxB,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,gBAAgB;YACzB,SAAS,EAAE,qBAAqB;YAChC,QAAQ,EAAE,CAAC,iBAAiB,CAAC,wBAAwB,CAAC,CAAC;YACvD,gBAAgB;YAChB,SAAS,EAAE,wBAAqC;YAChD,QAAQ,EAAE,wBAAoC;YAC9C,SAAS,EAAE,wBAAqC;YAChD,QAAQ,EAAE,wBAAoC;SAC9C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE,+BAA+B,CAAC,EAAE,OAAO,CAAC,CAAA;QAC9F,MAAM,cAAc,GAAG,MAAM;aAC3B,MAAM,CACN,CAAC,KAAK,EAA0D,EAAE,CACjE,KAAK,CAAC,IAAI,KAAK,gBAAgB,CAChC;aACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;QAEhC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAA;QAC9D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAA;QACzE,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAA;QACrD,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|