coalesce-transform-mcp 0.2.3 → 0.4.1
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 +84 -2
- package/dist/client.d.ts +1 -0
- package/dist/client.d.ts.map +1 -1
- package/dist/client.js +27 -3
- package/dist/client.js.map +1 -1
- package/dist/coalesce/api/environments.d.ts +0 -12
- package/dist/coalesce/api/environments.d.ts.map +1 -1
- package/dist/coalesce/api/environments.js +0 -4
- package/dist/coalesce/api/environments.js.map +1 -1
- package/dist/coalesce/api/jobs.d.ts +3 -5
- package/dist/coalesce/api/jobs.d.ts.map +1 -1
- package/dist/coalesce/api/jobs.js +3 -6
- package/dist/coalesce/api/jobs.js.map +1 -1
- package/dist/coalesce/api/nodes.d.ts +3 -3
- package/dist/coalesce/api/nodes.d.ts.map +1 -1
- package/dist/coalesce/api/nodes.js +6 -4
- package/dist/coalesce/api/nodes.js.map +1 -1
- package/dist/coalesce/api/runs.d.ts.map +1 -1
- package/dist/coalesce/api/runs.js +11 -1
- package/dist/coalesce/api/runs.js.map +1 -1
- package/dist/coalesce/api/scan.d.ts +14 -0
- package/dist/coalesce/api/scan.d.ts.map +1 -0
- package/dist/coalesce/api/scan.js +64 -0
- package/dist/coalesce/api/scan.js.map +1 -0
- package/dist/coalesce/api/subgraphs.d.ts +3 -2
- package/dist/coalesce/api/subgraphs.d.ts.map +1 -1
- package/dist/coalesce/api/subgraphs.js +3 -2
- package/dist/coalesce/api/subgraphs.js.map +1 -1
- package/dist/coalesce/run-schemas.d.ts.map +1 -1
- package/dist/coalesce/run-schemas.js +26 -16
- package/dist/coalesce/run-schemas.js.map +1 -1
- package/dist/coalesce/tool-response.d.ts +1 -13
- package/dist/coalesce/tool-response.d.ts.map +1 -1
- package/dist/coalesce/tool-response.js +20 -6
- package/dist/coalesce/tool-response.js.map +1 -1
- package/dist/coalesce/tool-schemas.d.ts +1 -2
- package/dist/coalesce/tool-schemas.d.ts.map +1 -1
- package/dist/coalesce/tool-schemas.js +368 -5
- package/dist/coalesce/tool-schemas.js.map +1 -1
- package/dist/coalesce/types.d.ts +8 -0
- package/dist/coalesce/types.d.ts.map +1 -1
- package/dist/coalesce/types.js +8 -3
- package/dist/coalesce/types.js.map +1 -1
- package/dist/constants.d.ts +18 -0
- package/dist/constants.d.ts.map +1 -0
- package/dist/constants.js +21 -0
- package/dist/constants.js.map +1 -0
- package/dist/mcp/cache.d.ts +2 -1
- package/dist/mcp/cache.d.ts.map +1 -1
- package/dist/mcp/cache.js +122 -138
- package/dist/mcp/cache.js.map +1 -1
- package/dist/mcp/environments.d.ts +2 -1
- package/dist/mcp/environments.d.ts.map +1 -1
- package/dist/mcp/environments.js +56 -112
- package/dist/mcp/environments.js.map +1 -1
- package/dist/mcp/git-accounts.d.ts +2 -1
- package/dist/mcp/git-accounts.d.ts.map +1 -1
- package/dist/mcp/git-accounts.js +74 -96
- package/dist/mcp/git-accounts.js.map +1 -1
- package/dist/mcp/jobs.d.ts +2 -1
- package/dist/mcp/jobs.d.ts.map +1 -1
- package/dist/mcp/jobs.js +68 -122
- package/dist/mcp/jobs.js.map +1 -1
- package/dist/mcp/lineage.d.ts +5 -0
- package/dist/mcp/lineage.d.ts.map +1 -0
- package/dist/mcp/lineage.js +410 -0
- package/dist/mcp/lineage.js.map +1 -0
- package/dist/mcp/node-type-corpus.d.ts +2 -1
- package/dist/mcp/node-type-corpus.d.ts.map +1 -1
- package/dist/mcp/node-type-corpus.js +148 -151
- package/dist/mcp/node-type-corpus.js.map +1 -1
- package/dist/mcp/nodes.d.ts +2 -1
- package/dist/mcp/nodes.d.ts.map +1 -1
- package/dist/mcp/nodes.js +358 -464
- package/dist/mcp/nodes.js.map +1 -1
- package/dist/mcp/pipelines.d.ts +2 -1
- package/dist/mcp/pipelines.d.ts.map +1 -1
- package/dist/mcp/pipelines.js +514 -314
- package/dist/mcp/pipelines.js.map +1 -1
- package/dist/mcp/projects.d.ts +2 -1
- package/dist/mcp/projects.d.ts.map +1 -1
- package/dist/mcp/projects.js +66 -100
- package/dist/mcp/projects.js.map +1 -1
- package/dist/mcp/repo-node-types.d.ts +2 -1
- package/dist/mcp/repo-node-types.d.ts.map +1 -1
- package/dist/mcp/repo-node-types.js +92 -121
- package/dist/mcp/repo-node-types.js.map +1 -1
- package/dist/mcp/runs.d.ts +4 -1
- package/dist/mcp/runs.d.ts.map +1 -1
- package/dist/mcp/runs.js +94 -147
- package/dist/mcp/runs.js.map +1 -1
- package/dist/mcp/skills.d.ts +13 -0
- package/dist/mcp/skills.d.ts.map +1 -0
- package/dist/mcp/skills.js +85 -0
- package/dist/mcp/skills.js.map +1 -0
- package/dist/mcp/subgraphs.d.ts +2 -1
- package/dist/mcp/subgraphs.d.ts.map +1 -1
- package/dist/mcp/subgraphs.js +61 -98
- package/dist/mcp/subgraphs.js.map +1 -1
- package/dist/mcp/tool-helpers.d.ts +37 -0
- package/dist/mcp/tool-helpers.d.ts.map +1 -0
- package/dist/mcp/tool-helpers.js +82 -0
- package/dist/mcp/tool-helpers.js.map +1 -0
- package/dist/mcp/users.d.ts +2 -1
- package/dist/mcp/users.d.ts.map +1 -1
- package/dist/mcp/users.js +92 -145
- package/dist/mcp/users.js.map +1 -1
- package/dist/mcp/workshop.d.ts +2 -1
- package/dist/mcp/workshop.d.ts.map +1 -1
- package/dist/mcp/workshop.js +66 -101
- package/dist/mcp/workshop.js.map +1 -1
- package/dist/mcp/workspaces.d.ts +2 -1
- package/dist/mcp/workspaces.d.ts.map +1 -1
- package/dist/mcp/workspaces.js +19 -34
- package/dist/mcp/workspaces.js.map +1 -1
- package/dist/prompts/index.d.ts.map +1 -1
- package/dist/prompts/index.js +85 -0
- package/dist/prompts/index.js.map +1 -1
- package/dist/resources/context/pipeline-workshop-guide.md +1 -1
- package/dist/resources/context/tool-usage.md +7 -0
- package/dist/resources/index.d.ts +13 -0
- package/dist/resources/index.d.ts.map +1 -1
- package/dist/resources/index.js +105 -5
- package/dist/resources/index.js.map +1 -1
- package/dist/schemas/node-payloads.d.ts +2 -2
- package/dist/server.d.ts +2 -1
- package/dist/server.d.ts.map +1 -1
- package/dist/server.js +173 -45
- package/dist/server.js.map +1 -1
- package/dist/services/cache/snapshots.d.ts.map +1 -1
- package/dist/services/cache/snapshots.js +9 -5
- package/dist/services/cache/snapshots.js.map +1 -1
- package/dist/services/config/schema-resolver.d.ts.map +1 -1
- package/dist/services/config/schema-resolver.js +3 -6
- package/dist/services/config/schema-resolver.js.map +1 -1
- package/dist/services/lineage/lineage-cache.d.ts +53 -0
- package/dist/services/lineage/lineage-cache.d.ts.map +1 -0
- package/dist/services/lineage/lineage-cache.js +335 -0
- package/dist/services/lineage/lineage-cache.js.map +1 -0
- package/dist/services/lineage/lineage-documentation.d.ts +29 -0
- package/dist/services/lineage/lineage-documentation.d.ts.map +1 -0
- package/dist/services/lineage/lineage-documentation.js +80 -0
- package/dist/services/lineage/lineage-documentation.js.map +1 -0
- package/dist/services/lineage/lineage-propagation.d.ts +47 -0
- package/dist/services/lineage/lineage-propagation.d.ts.map +1 -0
- package/dist/services/lineage/lineage-propagation.js +176 -0
- package/dist/services/lineage/lineage-propagation.js.map +1 -0
- package/dist/services/lineage/lineage-search.d.ts +33 -0
- package/dist/services/lineage/lineage-search.d.ts.map +1 -0
- package/dist/services/lineage/lineage-search.js +133 -0
- package/dist/services/lineage/lineage-search.js.map +1 -0
- package/dist/services/lineage/lineage-traversal.d.ts +34 -0
- package/dist/services/lineage/lineage-traversal.d.ts.map +1 -0
- package/dist/services/lineage/lineage-traversal.js +283 -0
- package/dist/services/lineage/lineage-traversal.js.map +1 -0
- package/dist/services/pipelines/clause-extraction.d.ts +3 -0
- package/dist/services/pipelines/clause-extraction.d.ts.map +1 -0
- package/dist/services/pipelines/clause-extraction.js +27 -0
- package/dist/services/pipelines/clause-extraction.js.map +1 -0
- package/dist/services/pipelines/column-helpers.d.ts +8 -0
- package/dist/services/pipelines/column-helpers.d.ts.map +1 -0
- package/dist/services/pipelines/column-helpers.js +125 -0
- package/dist/services/pipelines/column-helpers.js.map +1 -0
- package/dist/services/pipelines/cte-parsing.d.ts +29 -0
- package/dist/services/pipelines/cte-parsing.d.ts.map +1 -0
- package/dist/services/pipelines/cte-parsing.js +160 -0
- package/dist/services/pipelines/cte-parsing.js.map +1 -0
- package/dist/services/pipelines/cte-planning.d.ts +22 -0
- package/dist/services/pipelines/cte-planning.d.ts.map +1 -0
- package/dist/services/pipelines/cte-planning.js +206 -0
- package/dist/services/pipelines/cte-planning.js.map +1 -0
- package/dist/services/pipelines/execution.d.ts.map +1 -1
- package/dist/services/pipelines/execution.js +0 -1
- package/dist/services/pipelines/execution.js.map +1 -1
- package/dist/services/pipelines/intent-parsing.d.ts +24 -0
- package/dist/services/pipelines/intent-parsing.d.ts.map +1 -0
- package/dist/services/pipelines/intent-parsing.js +245 -0
- package/dist/services/pipelines/intent-parsing.js.map +1 -0
- package/dist/services/pipelines/intent-resolution.d.ts +24 -0
- package/dist/services/pipelines/intent-resolution.d.ts.map +1 -0
- package/dist/services/pipelines/intent-resolution.js +141 -0
- package/dist/services/pipelines/intent-resolution.js.map +1 -0
- package/dist/services/pipelines/intent.d.ts +4 -45
- package/dist/services/pipelines/intent.d.ts.map +1 -1
- package/dist/services/pipelines/intent.js +14 -408
- package/dist/services/pipelines/intent.js.map +1 -1
- package/dist/services/pipelines/node-type-candidates.d.ts +6 -0
- package/dist/services/pipelines/node-type-candidates.d.ts.map +1 -0
- package/dist/services/pipelines/node-type-candidates.js +165 -0
- package/dist/services/pipelines/node-type-candidates.js.map +1 -0
- package/dist/services/pipelines/node-type-intent.d.ts +1 -5
- package/dist/services/pipelines/node-type-intent.d.ts.map +1 -1
- package/dist/services/pipelines/node-type-intent.js +1 -5
- package/dist/services/pipelines/node-type-intent.js.map +1 -1
- package/dist/services/pipelines/node-type-scoring.d.ts +13 -0
- package/dist/services/pipelines/node-type-scoring.d.ts.map +1 -0
- package/dist/services/pipelines/node-type-scoring.js +322 -0
- package/dist/services/pipelines/node-type-scoring.js.map +1 -0
- package/dist/services/pipelines/node-type-selection.d.ts +22 -2
- package/dist/services/pipelines/node-type-selection.d.ts.map +1 -1
- package/dist/services/pipelines/node-type-selection.js +16 -538
- package/dist/services/pipelines/node-type-selection.js.map +1 -1
- package/dist/services/pipelines/plan-builder.d.ts +33 -0
- package/dist/services/pipelines/plan-builder.d.ts.map +1 -0
- package/dist/services/pipelines/plan-builder.js +224 -0
- package/dist/services/pipelines/plan-builder.js.map +1 -0
- package/dist/services/pipelines/planning-types.d.ts +543 -0
- package/dist/services/pipelines/planning-types.d.ts.map +1 -0
- package/dist/services/pipelines/planning-types.js +85 -0
- package/dist/services/pipelines/planning-types.js.map +1 -0
- package/dist/services/pipelines/planning.d.ts +8 -537
- package/dist/services/pipelines/planning.d.ts.map +1 -1
- package/dist/services/pipelines/planning.js +10 -1956
- package/dist/services/pipelines/planning.js.map +1 -1
- package/dist/services/pipelines/review.d.ts.map +1 -1
- package/dist/services/pipelines/review.js +3 -8
- package/dist/services/pipelines/review.js.map +1 -1
- package/dist/services/pipelines/select-parsing.d.ts +7 -0
- package/dist/services/pipelines/select-parsing.d.ts.map +1 -0
- package/dist/services/pipelines/select-parsing.js +185 -0
- package/dist/services/pipelines/select-parsing.js.map +1 -0
- package/dist/services/pipelines/source-parsing.d.ts +8 -0
- package/dist/services/pipelines/source-parsing.d.ts.map +1 -0
- package/dist/services/pipelines/source-parsing.js +151 -0
- package/dist/services/pipelines/source-parsing.js.map +1 -0
- package/dist/services/pipelines/sql-parsing.d.ts +8 -0
- package/dist/services/pipelines/sql-parsing.d.ts.map +1 -0
- package/dist/services/pipelines/sql-parsing.js +9 -0
- package/dist/services/pipelines/sql-parsing.js.map +1 -0
- package/dist/services/pipelines/sql-tokenizer.d.ts +42 -0
- package/dist/services/pipelines/sql-tokenizer.d.ts.map +1 -0
- package/dist/services/pipelines/sql-tokenizer.js +493 -0
- package/dist/services/pipelines/sql-tokenizer.js.map +1 -0
- package/dist/services/pipelines/sql-utils.d.ts +30 -0
- package/dist/services/pipelines/sql-utils.d.ts.map +1 -0
- package/dist/services/pipelines/sql-utils.js +62 -0
- package/dist/services/pipelines/sql-utils.js.map +1 -0
- package/dist/services/pipelines/workshop.d.ts.map +1 -1
- package/dist/services/pipelines/workshop.js +53 -25
- package/dist/services/pipelines/workshop.js.map +1 -1
- package/dist/services/pipelines/workspace-resolution.d.ts +18 -0
- package/dist/services/pipelines/workspace-resolution.d.ts.map +1 -0
- package/dist/services/pipelines/workspace-resolution.js +279 -0
- package/dist/services/pipelines/workspace-resolution.js.map +1 -0
- package/dist/services/runs/diagnostics.d.ts.map +1 -1
- package/dist/services/runs/diagnostics.js +3 -8
- package/dist/services/runs/diagnostics.js.map +1 -1
- package/dist/services/shared/elicitation.d.ts +14 -0
- package/dist/services/shared/elicitation.d.ts.map +1 -0
- package/dist/services/shared/elicitation.js +56 -0
- package/dist/services/shared/elicitation.js.map +1 -0
- package/dist/services/workspace/join-operations.d.ts +56 -0
- package/dist/services/workspace/join-operations.d.ts.map +1 -0
- package/dist/services/workspace/join-operations.js +303 -0
- package/dist/services/workspace/join-operations.js.map +1 -0
- package/dist/services/workspace/mutations.d.ts +3 -99
- package/dist/services/workspace/mutations.d.ts.map +1 -1
- package/dist/services/workspace/mutations.js +13 -1220
- package/dist/services/workspace/mutations.js.map +1 -1
- package/dist/services/workspace/node-creation.d.ts +48 -0
- package/dist/services/workspace/node-creation.d.ts.map +1 -0
- package/dist/services/workspace/node-creation.js +448 -0
- package/dist/services/workspace/node-creation.js.map +1 -0
- package/dist/services/workspace/node-update-helpers.d.ts +89 -0
- package/dist/services/workspace/node-update-helpers.d.ts.map +1 -0
- package/dist/services/workspace/node-update-helpers.js +492 -0
- package/dist/services/workspace/node-update-helpers.js.map +1 -0
- package/dist/tasks/index.d.ts +5 -0
- package/dist/tasks/index.d.ts.map +1 -0
- package/dist/tasks/index.js +10 -0
- package/dist/tasks/index.js.map +1 -0
- package/dist/tasks/retry-and-wait-task.d.ts +4 -0
- package/dist/tasks/retry-and-wait-task.d.ts.map +1 -0
- package/dist/tasks/retry-and-wait-task.js +61 -0
- package/dist/tasks/retry-and-wait-task.js.map +1 -0
- package/dist/tasks/run-and-wait-task.d.ts +4 -0
- package/dist/tasks/run-and-wait-task.d.ts.map +1 -0
- package/dist/tasks/run-and-wait-task.js +61 -0
- package/dist/tasks/run-and-wait-task.js.map +1 -0
- package/dist/tasks/start-run-task.d.ts +4 -0
- package/dist/tasks/start-run-task.d.ts.map +1 -0
- package/dist/tasks/start-run-task.js +47 -0
- package/dist/tasks/start-run-task.js.map +1 -0
- package/dist/tasks/store.d.ts +2 -0
- package/dist/tasks/store.d.ts.map +1 -0
- package/dist/tasks/store.js +2 -0
- package/dist/tasks/store.js.map +1 -0
- package/dist/utils.d.ts +11 -0
- package/dist/utils.d.ts.map +1 -1
- package/dist/utils.js +20 -1
- package/dist/utils.js.map +1 -1
- package/dist/workflows/get-environment-health.d.ts +49 -0
- package/dist/workflows/get-environment-health.d.ts.map +1 -0
- package/dist/workflows/get-environment-health.js +310 -0
- package/dist/workflows/get-environment-health.js.map +1 -0
- package/dist/workflows/get-environment-overview.d.ts +2 -1
- package/dist/workflows/get-environment-overview.d.ts.map +1 -1
- package/dist/workflows/get-environment-overview.js +13 -19
- package/dist/workflows/get-environment-overview.js.map +1 -1
- package/dist/workflows/get-run-details.d.ts +2 -2
- package/dist/workflows/get-run-details.d.ts.map +1 -1
- package/dist/workflows/get-run-details.js +14 -19
- package/dist/workflows/get-run-details.js.map +1 -1
- package/dist/workflows/poll-run.d.ts +14 -0
- package/dist/workflows/poll-run.d.ts.map +1 -0
- package/dist/workflows/poll-run.js +86 -0
- package/dist/workflows/poll-run.js.map +1 -0
- package/dist/workflows/progress.d.ts +1 -0
- package/dist/workflows/progress.d.ts.map +1 -1
- package/dist/workflows/progress.js +3 -0
- package/dist/workflows/progress.js.map +1 -1
- package/dist/workflows/retry-and-wait.d.ts +1 -1
- package/dist/workflows/retry-and-wait.d.ts.map +1 -1
- package/dist/workflows/retry-and-wait.js +18 -89
- package/dist/workflows/retry-and-wait.js.map +1 -1
- package/dist/workflows/run-and-wait.d.ts +1 -1
- package/dist/workflows/run-and-wait.d.ts.map +1 -1
- package/dist/workflows/run-and-wait.js +18 -89
- package/dist/workflows/run-and-wait.js.map +1 -1
- package/package.json +2 -2
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"progress.js","sourceRoot":"","sources":["../../src/workflows/progress.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAEhD,MAAM,UAAU,eAAe,CAAC,SAAiB,EAAE,cAAsB;IACvE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAAc;IAClD,IAAI,KAAK,YAAY,gBAAgB,EAAE,CAAC;QACtC,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChE,CAAC;IACJ,CAAC;IACD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;IACpC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,6BAA6B,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;AACnE,CAAC;AA2BD,SAAS,gBAAgB;IACvB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IACjD,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC;IAC1B,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAoB;IACjD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;QACpB,MAAM,gBAAgB,EAAE,CAAC;IAC3B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,EAAU,EACV,MAAoB;IAEpB,cAAc,CAAC,MAAM,CAAC,CAAC;IAEvB,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QACxD,OAAO;IACT,CAAC;IAED,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,aAAa,CAAC,CAAC;YAC5B,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,KAA6B;IAE7B,MAAM,aAAa,GAAG,KAAK,EAAE,KAAK,EAAE,aAAa,CAAC;IAClD,MAAM,gBAAgB,GAAG,KAAK,EAAE,gBAAgB,CAAC;IAEjD,IAAI,aAAa,KAAK,SAAS,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACrD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,QAAQ,GAAG,CAAC,CAAC;IAEjB,OAAO,KAAK,EAAE,OAAe,EAAE,KAAc,EAAE,EAAE;QAC/C,QAAQ,IAAI,CAAC,CAAC;QAEd,IAAI,CAAC;YACH,MAAM,gBAAgB,CAAC;gBACrB,MAAM,EAAE,wBAAwB;gBAChC,MAAM,EAAE;oBACN,aAAa;oBACb,QAAQ;oBACR,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACzC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChC;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,4DAA4D;YAC5D,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACtE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,aAAa,MAAM,MAAM,IAAI,CAAC,CAAC;QAC/F,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}
|
|
1
|
+
{"version":3,"file":"progress.js","sourceRoot":"","sources":["../../src/workflows/progress.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAEhD,MAAM,UAAU,eAAe,CAAC,SAAiB,EAAE,cAAsB;IACvE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAAc;IAClD,IAAI,KAAK,YAAY,gBAAgB,EAAE,CAAC;QACtC,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChE,CAAC;IACJ,CAAC;IACD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;IACpC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,6BAA6B,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;AACnE,CAAC;AA2BD,SAAS,gBAAgB;IACvB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IACjD,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC;IAC1B,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAc;IACzC,OAAO,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC;AAC/D,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAoB;IACjD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;QACpB,MAAM,gBAAgB,EAAE,CAAC;IAC3B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,EAAU,EACV,MAAoB;IAEpB,cAAc,CAAC,MAAM,CAAC,CAAC;IAEvB,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QACxD,OAAO;IACT,CAAC;IAED,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,aAAa,CAAC,CAAC;YAC5B,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,KAA6B;IAE7B,MAAM,aAAa,GAAG,KAAK,EAAE,KAAK,EAAE,aAAa,CAAC;IAClD,MAAM,gBAAgB,GAAG,KAAK,EAAE,gBAAgB,CAAC;IAEjD,IAAI,aAAa,KAAK,SAAS,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACrD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,QAAQ,GAAG,CAAC,CAAC;IAEjB,OAAO,KAAK,EAAE,OAAe,EAAE,KAAc,EAAE,EAAE;QAC/C,QAAQ,IAAI,CAAC,CAAC;QAEd,IAAI,CAAC;YACH,MAAM,gBAAgB,CAAC;gBACrB,MAAM,EAAE,wBAAwB;gBAChC,MAAM,EAAE;oBACN,aAAa;oBACb,QAAQ;oBACR,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACzC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChC;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,4DAA4D;YAC5D,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACtE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,aAAa,MAAM,MAAM,IAAI,CAAC,CAAC;QAC/F,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { z } from "zod";
|
|
2
2
|
import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
3
|
-
import {
|
|
3
|
+
import type { CoalesceClient } from "../client.js";
|
|
4
4
|
import { RerunParams } from "../coalesce/types.js";
|
|
5
5
|
import { type WorkflowProgressReporter } from "./progress.js";
|
|
6
6
|
export declare function retryAndWait(client: CoalesceClient, params: z.infer<typeof RerunParams> & {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"retry-and-wait.d.ts","sourceRoot":"","sources":["../../src/workflows/retry-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,
|
|
1
|
+
{"version":3,"file":"retry-and-wait.d.ts","sourceRoot":"","sources":["../../src/workflows/retry-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EACL,WAAW,EAOZ,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAKL,KAAK,wBAAwB,EAC9B,MAAM,eAAe,CAAC;AAWvB,wBAAsB,YAAY,CAChC,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,WAAW,CAAC,GAAG;IACpC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,EACD,OAAO,GAAE;IACP,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,wBAAwB,CAAC;CACtC,GACL,OAAO,CAAC,OAAO,CAAC,CAoClB;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,cAAc,GAAG,IAAI,CA6BpF"}
|
|
@@ -1,106 +1,35 @@
|
|
|
1
1
|
import { z } from "zod";
|
|
2
|
-
import { CoalesceApiError } from "../client.js";
|
|
3
2
|
import { RerunParams, buildJsonToolResponse, buildRerunBody, WRITE_ANNOTATIONS, sanitizeResponse, handleToolError, getToolOutputSchema, } from "../coalesce/types.js";
|
|
4
|
-
import { createWorkflowProgressReporter, remainingTimeMs,
|
|
5
|
-
import {
|
|
3
|
+
import { createWorkflowProgressReporter, remainingTimeMs, throwIfAborted, } from "./progress.js";
|
|
4
|
+
import { pollRunToCompletion } from "./poll-run.js";
|
|
5
|
+
import { POLL_INTERVAL_MIN_S, POLL_INTERVAL_DEFAULT_S, POLL_INTERVAL_MAX_S, WORKFLOW_TIMEOUT_MIN_S, WORKFLOW_TIMEOUT_DEFAULT_S, WORKFLOW_TIMEOUT_MAX_S, } from "../constants.js";
|
|
6
6
|
export async function retryAndWait(client, params, options = {}) {
|
|
7
|
-
const
|
|
8
|
-
const
|
|
7
|
+
const pollIntervalMs = Math.max(POLL_INTERVAL_MIN_S, Math.min(params.pollInterval ?? POLL_INTERVAL_DEFAULT_S, POLL_INTERVAL_MAX_S)) * 1000;
|
|
8
|
+
const timeoutMs = Math.max(WORKFLOW_TIMEOUT_MIN_S, Math.min(params.timeout ?? WORKFLOW_TIMEOUT_DEFAULT_S, WORKFLOW_TIMEOUT_MAX_S)) * 1000;
|
|
9
9
|
const startedAt = Date.now();
|
|
10
10
|
const { signal, reportProgress } = options;
|
|
11
11
|
throwIfAborted(signal);
|
|
12
12
|
// Retry the run — response is { runCounter: number }
|
|
13
13
|
const body = buildRerunBody(params);
|
|
14
14
|
const rerunResult = (await client.post("/scheduler/rerun", body, undefined, {
|
|
15
|
-
timeoutMs: remainingTimeMs(startedAt,
|
|
15
|
+
timeoutMs: remainingTimeMs(startedAt, timeoutMs),
|
|
16
|
+
signal,
|
|
16
17
|
}));
|
|
17
18
|
if (typeof rerunResult.runCounter !== "number") {
|
|
18
19
|
throw new Error(`rerun response did not include a numeric runCounter (got ${typeof rerunResult.runCounter})`);
|
|
19
20
|
}
|
|
20
21
|
const runCounter = rerunResult.runCounter;
|
|
21
|
-
await reportProgress?.(`Started retry run ${runCounter}. Polling every ${
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
let status;
|
|
33
|
-
try {
|
|
34
|
-
status = (await client.get("/scheduler/runStatus", {
|
|
35
|
-
runCounter,
|
|
36
|
-
}, { timeoutMs: statusTimeoutMs }));
|
|
37
|
-
}
|
|
38
|
-
catch (error) {
|
|
39
|
-
if (error instanceof CoalesceApiError && error.status === 408) {
|
|
40
|
-
pollCount += 1;
|
|
41
|
-
await reportProgress?.(`Status check ${pollCount} for retry run ${runCounter} timed out. Retrying while time remains.`);
|
|
42
|
-
continue;
|
|
43
|
-
}
|
|
44
|
-
throw error;
|
|
45
|
-
}
|
|
46
|
-
lastStatus = status;
|
|
47
|
-
pollCount += 1;
|
|
48
|
-
const runStatus = status.runStatus;
|
|
49
|
-
await reportProgress?.(`Status check ${pollCount} for retry run ${runCounter}: ${formatRunStatusForMessage(runStatus)}.`);
|
|
50
|
-
const validatedRunStatus = validateRunStatus(runCounter, runStatus);
|
|
51
|
-
if (isTerminalRunStatus(validatedRunStatus)) {
|
|
52
|
-
// Fetch run results — runCounter is the numeric run ID
|
|
53
|
-
await reportProgress?.(`Retry run ${runCounter} reached terminal status ${validatedRunStatus}. Fetching results.`);
|
|
54
|
-
const resultsTimeoutMs = remainingTimeMs(startedAt, timeout);
|
|
55
|
-
if (resultsTimeoutMs <= 0) {
|
|
56
|
-
await reportProgress?.(`Workflow deadline reached before results could be fetched for retry run ${runCounter}.`);
|
|
57
|
-
return {
|
|
58
|
-
status,
|
|
59
|
-
results: null,
|
|
60
|
-
resultsError: {
|
|
61
|
-
message: "Workflow timeout reached before run results could be fetched",
|
|
62
|
-
status: 408,
|
|
63
|
-
},
|
|
64
|
-
incomplete: true,
|
|
65
|
-
};
|
|
66
|
-
}
|
|
67
|
-
try {
|
|
68
|
-
const results = await client.get(`/api/v1/runs/${runCounter}/results`, undefined, { timeoutMs: resultsTimeoutMs });
|
|
69
|
-
await reportProgress?.(`Fetched results for retry run ${runCounter}.`);
|
|
70
|
-
return { status, results };
|
|
71
|
-
}
|
|
72
|
-
catch (error) {
|
|
73
|
-
const serializedError = serializeResultsError(error);
|
|
74
|
-
await reportProgress?.(`Retry run ${runCounter} finished, but fetching results failed: ${serializedError.message}.`);
|
|
75
|
-
return {
|
|
76
|
-
status,
|
|
77
|
-
results: null,
|
|
78
|
-
resultsError: serializedError,
|
|
79
|
-
incomplete: true,
|
|
80
|
-
};
|
|
81
|
-
}
|
|
82
|
-
}
|
|
83
|
-
}
|
|
84
|
-
// Timeout — return last known status
|
|
85
|
-
let finalStatus = lastStatus;
|
|
86
|
-
const finalStatusTimeoutMs = remainingTimeMs(startedAt, timeout);
|
|
87
|
-
if (finalStatusTimeoutMs > 0) {
|
|
88
|
-
try {
|
|
89
|
-
finalStatus = await client.get("/scheduler/runStatus", {
|
|
90
|
-
runCounter: runCounter,
|
|
91
|
-
}, { timeoutMs: finalStatusTimeoutMs });
|
|
92
|
-
}
|
|
93
|
-
catch (error) {
|
|
94
|
-
if (!(error instanceof CoalesceApiError && error.status === 408)) {
|
|
95
|
-
throw error;
|
|
96
|
-
}
|
|
97
|
-
}
|
|
98
|
-
}
|
|
99
|
-
if (finalStatus && typeof finalStatus === "object") {
|
|
100
|
-
validateRunStatus(runCounter, finalStatus.runStatus);
|
|
101
|
-
}
|
|
102
|
-
await reportProgress?.(`Timed out waiting for retry run ${runCounter}. Returning the last known status.`);
|
|
103
|
-
return { status: finalStatus, results: null, timedOut: true };
|
|
22
|
+
await reportProgress?.(`Started retry run ${runCounter}. Polling every ${pollIntervalMs / 1000}s for up to ${timeoutMs / 1000}s.`);
|
|
23
|
+
return pollRunToCompletion({
|
|
24
|
+
client,
|
|
25
|
+
runCounter,
|
|
26
|
+
label: "retry run",
|
|
27
|
+
pollIntervalMs,
|
|
28
|
+
timeoutMs,
|
|
29
|
+
startedAt,
|
|
30
|
+
signal,
|
|
31
|
+
reportProgress,
|
|
32
|
+
});
|
|
104
33
|
}
|
|
105
34
|
export function registerRetryAndWait(server, client) {
|
|
106
35
|
server.registerTool("retry_and_wait", {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"retry-and-wait.js","sourceRoot":"","sources":["../../src/workflows/retry-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;
|
|
1
|
+
{"version":3,"file":"retry-and-wait.js","sourceRoot":"","sources":["../../src/workflows/retry-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EACL,WAAW,EACX,qBAAqB,EACrB,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAChB,eAAe,EACf,mBAAmB,GACpB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,8BAA8B,EAC9B,eAAe,EACf,cAAc,GAGf,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EACL,mBAAmB,EACnB,uBAAuB,EACvB,mBAAmB,EACnB,sBAAsB,EACtB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,iBAAiB,CAAC;AAEzB,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,MAAsB,EACtB,MAGC,EACD,UAGI,EAAE;IAEN,MAAM,cAAc,GAClB,IAAI,CAAC,GAAG,CAAC,mBAAmB,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,IAAI,uBAAuB,EAAE,mBAAmB,CAAC,CAAC,GAAG,IAAI,CAAC;IACtH,MAAM,SAAS,GACb,IAAI,CAAC,GAAG,CAAC,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,0BAA0B,EAAE,sBAAsB,CAAC,CAAC,GAAG,IAAI,CAAC;IAC1H,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;IAE3C,cAAc,CAAC,MAAM,CAAC,CAAC;IAEvB,qDAAqD;IACrD,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,WAAW,GAAG,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,EAAE,SAAS,EAAE;QAC1E,SAAS,EAAE,eAAe,CAAC,SAAS,EAAE,SAAS,CAAC;QAChD,MAAM;KACP,CAAC,CAA4B,CAAC;IAC/B,IAAI,OAAO,WAAW,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC/C,MAAM,IAAI,KAAK,CACb,4DAA4D,OAAO,WAAW,CAAC,UAAU,GAAG,CAC7F,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAW,WAAW,CAAC,UAAU,CAAC;IAClD,MAAM,cAAc,EAAE,CACpB,qBAAqB,UAAU,mBAAmB,cAAc,GAAG,IAAI,eAAe,SAAS,GAAG,IAAI,IAAI,CAC3G,CAAC;IAEF,OAAO,mBAAmB,CAAC;QACzB,MAAM;QACN,UAAU;QACV,KAAK,EAAE,WAAW;QAClB,cAAc;QACd,SAAS;QACT,SAAS;QACT,MAAM;QACN,cAAc;KACf,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAiB,EAAE,MAAsB;IAC5E,MAAM,CAAC,YAAY,CACjB,gBAAgB,EAChB;QACE,KAAK,EAAE,gBAAgB;QACvB,WAAW,EACT,srBAAsrB;QACxrB,WAAW,EAAE,WAAW,CAAC,MAAM,CAAC;YAC9B,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+DAA+D,CAAC;YAC7G,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yDAAyD,CAAC;SACnG,CAAC;QACF,YAAY,EAAE,mBAAmB,CAAC,gBAAgB,CAAC;QACnD,WAAW,EAAE,iBAAiB;KAC/B,EACD,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE;QACtB,IAAI,CAAC;YACH,MAAM,gBAAgB,GAAG,8BAA8B,CACrD,KAA0C,CAC3C,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE;gBAChD,MAAM,EAAE,KAAK,EAAE,MAAM;gBACrB,cAAc,EAAE,gBAAgB;aACjC,CAAC,CAAC;YACH,OAAO,qBAAqB,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { z } from "zod";
|
|
2
2
|
import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
3
|
-
import {
|
|
3
|
+
import type { CoalesceClient } from "../client.js";
|
|
4
4
|
import { StartRunParams } from "../coalesce/types.js";
|
|
5
5
|
import { type WorkflowProgressReporter } from "./progress.js";
|
|
6
6
|
export declare function runAndWait(client: CoalesceClient, params: z.infer<typeof StartRunParams> & {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"run-and-wait.d.ts","sourceRoot":"","sources":["../../src/workflows/run-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,
|
|
1
|
+
{"version":3,"file":"run-and-wait.d.ts","sourceRoot":"","sources":["../../src/workflows/run-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EACL,cAAc,EAOf,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAKL,KAAK,wBAAwB,EAC9B,MAAM,eAAe,CAAC;AAWvB,wBAAsB,UAAU,CAC9B,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,GAAG;IACvC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,EACD,OAAO,GAAE;IACP,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,wBAAwB,CAAC;CACtC,GACL,OAAO,CAAC,OAAO,CAAC,CAoClB;AAED,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,cAAc,GAAG,IAAI,CA6BlF"}
|
|
@@ -1,106 +1,35 @@
|
|
|
1
1
|
import { z } from "zod";
|
|
2
|
-
import { CoalesceApiError } from "../client.js";
|
|
3
2
|
import { StartRunParams, buildJsonToolResponse, buildStartRunBody, WRITE_ANNOTATIONS, sanitizeResponse, handleToolError, getToolOutputSchema, } from "../coalesce/types.js";
|
|
4
|
-
import { createWorkflowProgressReporter, remainingTimeMs,
|
|
5
|
-
import {
|
|
3
|
+
import { createWorkflowProgressReporter, remainingTimeMs, throwIfAborted, } from "./progress.js";
|
|
4
|
+
import { pollRunToCompletion } from "./poll-run.js";
|
|
5
|
+
import { POLL_INTERVAL_MIN_S, POLL_INTERVAL_DEFAULT_S, POLL_INTERVAL_MAX_S, WORKFLOW_TIMEOUT_MIN_S, WORKFLOW_TIMEOUT_DEFAULT_S, WORKFLOW_TIMEOUT_MAX_S, } from "../constants.js";
|
|
6
6
|
export async function runAndWait(client, params, options = {}) {
|
|
7
|
-
const
|
|
8
|
-
const
|
|
7
|
+
const pollIntervalMs = Math.max(POLL_INTERVAL_MIN_S, Math.min(params.pollInterval ?? POLL_INTERVAL_DEFAULT_S, POLL_INTERVAL_MAX_S)) * 1000;
|
|
8
|
+
const timeoutMs = Math.max(WORKFLOW_TIMEOUT_MIN_S, Math.min(params.timeout ?? WORKFLOW_TIMEOUT_DEFAULT_S, WORKFLOW_TIMEOUT_MAX_S)) * 1000;
|
|
9
9
|
const startedAt = Date.now();
|
|
10
10
|
const { signal, reportProgress } = options;
|
|
11
11
|
throwIfAborted(signal);
|
|
12
12
|
// Start the run — response is { runCounter: number }
|
|
13
13
|
const body = buildStartRunBody(params);
|
|
14
14
|
const startResult = (await client.post("/scheduler/startRun", body, undefined, {
|
|
15
|
-
timeoutMs: remainingTimeMs(startedAt,
|
|
15
|
+
timeoutMs: remainingTimeMs(startedAt, timeoutMs),
|
|
16
|
+
signal,
|
|
16
17
|
}));
|
|
17
18
|
if (typeof startResult.runCounter !== "number") {
|
|
18
19
|
throw new Error(`startRun response did not include a numeric runCounter (got ${typeof startResult.runCounter})`);
|
|
19
20
|
}
|
|
20
21
|
const runCounter = startResult.runCounter;
|
|
21
|
-
await reportProgress?.(`Started run ${runCounter}. Polling every ${
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
let status;
|
|
33
|
-
try {
|
|
34
|
-
status = (await client.get("/scheduler/runStatus", {
|
|
35
|
-
runCounter,
|
|
36
|
-
}, { timeoutMs: statusTimeoutMs }));
|
|
37
|
-
}
|
|
38
|
-
catch (error) {
|
|
39
|
-
if (error instanceof CoalesceApiError && error.status === 408) {
|
|
40
|
-
pollCount += 1;
|
|
41
|
-
await reportProgress?.(`Status check ${pollCount} for run ${runCounter} timed out. Retrying while time remains.`);
|
|
42
|
-
continue;
|
|
43
|
-
}
|
|
44
|
-
throw error;
|
|
45
|
-
}
|
|
46
|
-
lastStatus = status;
|
|
47
|
-
pollCount += 1;
|
|
48
|
-
const runStatus = status.runStatus;
|
|
49
|
-
await reportProgress?.(`Status check ${pollCount} for run ${runCounter}: ${formatRunStatusForMessage(runStatus)}.`);
|
|
50
|
-
const validatedRunStatus = validateRunStatus(runCounter, runStatus);
|
|
51
|
-
if (isTerminalRunStatus(validatedRunStatus)) {
|
|
52
|
-
// Fetch run results — runCounter is the numeric run ID
|
|
53
|
-
await reportProgress?.(`Run ${runCounter} reached terminal status ${validatedRunStatus}. Fetching results.`);
|
|
54
|
-
const resultsTimeoutMs = remainingTimeMs(startedAt, timeout);
|
|
55
|
-
if (resultsTimeoutMs <= 0) {
|
|
56
|
-
await reportProgress?.(`Workflow deadline reached before results could be fetched for run ${runCounter}.`);
|
|
57
|
-
return {
|
|
58
|
-
status,
|
|
59
|
-
results: null,
|
|
60
|
-
resultsError: {
|
|
61
|
-
message: "Workflow timeout reached before run results could be fetched",
|
|
62
|
-
status: 408,
|
|
63
|
-
},
|
|
64
|
-
incomplete: true,
|
|
65
|
-
};
|
|
66
|
-
}
|
|
67
|
-
try {
|
|
68
|
-
const results = await client.get(`/api/v1/runs/${runCounter}/results`, undefined, { timeoutMs: resultsTimeoutMs });
|
|
69
|
-
await reportProgress?.(`Fetched results for run ${runCounter}.`);
|
|
70
|
-
return { status, results };
|
|
71
|
-
}
|
|
72
|
-
catch (error) {
|
|
73
|
-
const serializedError = serializeResultsError(error);
|
|
74
|
-
await reportProgress?.(`Run ${runCounter} finished, but fetching results failed: ${serializedError.message}.`);
|
|
75
|
-
return {
|
|
76
|
-
status,
|
|
77
|
-
results: null,
|
|
78
|
-
resultsError: serializedError,
|
|
79
|
-
incomplete: true,
|
|
80
|
-
};
|
|
81
|
-
}
|
|
82
|
-
}
|
|
83
|
-
}
|
|
84
|
-
// Timeout — return last known status
|
|
85
|
-
let finalStatus = lastStatus;
|
|
86
|
-
const finalStatusTimeoutMs = remainingTimeMs(startedAt, timeout);
|
|
87
|
-
if (finalStatusTimeoutMs > 0) {
|
|
88
|
-
try {
|
|
89
|
-
finalStatus = await client.get("/scheduler/runStatus", {
|
|
90
|
-
runCounter,
|
|
91
|
-
}, { timeoutMs: finalStatusTimeoutMs });
|
|
92
|
-
}
|
|
93
|
-
catch (error) {
|
|
94
|
-
if (!(error instanceof CoalesceApiError && error.status === 408)) {
|
|
95
|
-
throw error;
|
|
96
|
-
}
|
|
97
|
-
}
|
|
98
|
-
}
|
|
99
|
-
if (finalStatus && typeof finalStatus === "object") {
|
|
100
|
-
validateRunStatus(runCounter, finalStatus.runStatus);
|
|
101
|
-
}
|
|
102
|
-
await reportProgress?.(`Timed out waiting for run ${runCounter}. Returning the last known status.`);
|
|
103
|
-
return { status: finalStatus, results: null, timedOut: true };
|
|
22
|
+
await reportProgress?.(`Started run ${runCounter}. Polling every ${pollIntervalMs / 1000}s for up to ${timeoutMs / 1000}s.`);
|
|
23
|
+
return pollRunToCompletion({
|
|
24
|
+
client,
|
|
25
|
+
runCounter,
|
|
26
|
+
label: "run",
|
|
27
|
+
pollIntervalMs,
|
|
28
|
+
timeoutMs,
|
|
29
|
+
startedAt,
|
|
30
|
+
signal,
|
|
31
|
+
reportProgress,
|
|
32
|
+
});
|
|
104
33
|
}
|
|
105
34
|
export function registerRunAndWait(server, client) {
|
|
106
35
|
server.registerTool("run_and_wait", {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"run-and-wait.js","sourceRoot":"","sources":["../../src/workflows/run-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;
|
|
1
|
+
{"version":3,"file":"run-and-wait.js","sourceRoot":"","sources":["../../src/workflows/run-and-wait.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EACL,cAAc,EACd,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EACjB,gBAAgB,EAChB,eAAe,EACf,mBAAmB,GACpB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,8BAA8B,EAC9B,eAAe,EACf,cAAc,GAGf,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EACL,mBAAmB,EACnB,uBAAuB,EACvB,mBAAmB,EACnB,sBAAsB,EACtB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,iBAAiB,CAAC;AAEzB,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,MAAsB,EACtB,MAGC,EACD,UAGI,EAAE;IAEN,MAAM,cAAc,GAClB,IAAI,CAAC,GAAG,CAAC,mBAAmB,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,IAAI,uBAAuB,EAAE,mBAAmB,CAAC,CAAC,GAAG,IAAI,CAAC;IACtH,MAAM,SAAS,GACb,IAAI,CAAC,GAAG,CAAC,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,0BAA0B,EAAE,sBAAsB,CAAC,CAAC,GAAG,IAAI,CAAC;IAC1H,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;IAE3C,cAAc,CAAC,MAAM,CAAC,CAAC;IAEvB,qDAAqD;IACrD,MAAM,IAAI,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACvC,MAAM,WAAW,GAAG,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,IAAI,EAAE,SAAS,EAAE;QAC7E,SAAS,EAAE,eAAe,CAAC,SAAS,EAAE,SAAS,CAAC;QAChD,MAAM;KACP,CAAC,CAA4B,CAAC;IAC/B,IAAI,OAAO,WAAW,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC/C,MAAM,IAAI,KAAK,CACb,+DAA+D,OAAO,WAAW,CAAC,UAAU,GAAG,CAChG,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAW,WAAW,CAAC,UAAU,CAAC;IAClD,MAAM,cAAc,EAAE,CACpB,eAAe,UAAU,mBAAmB,cAAc,GAAG,IAAI,eAAe,SAAS,GAAG,IAAI,IAAI,CACrG,CAAC;IAEF,OAAO,mBAAmB,CAAC;QACzB,MAAM;QACN,UAAU;QACV,KAAK,EAAE,KAAK;QACZ,cAAc;QACd,SAAS;QACT,SAAS;QACT,MAAM;QACN,cAAc;KACf,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAiB,EAAE,MAAsB;IAC1E,MAAM,CAAC,YAAY,CACjB,cAAc,EACd;QACE,KAAK,EAAE,cAAc;QACrB,WAAW,EACT,8pCAA8pC;QAChqC,WAAW,EAAE,cAAc,CAAC,MAAM,CAAC;YACjC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+DAA+D,CAAC;YAC7G,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yDAAyD,CAAC;SACnG,CAAC;QACF,YAAY,EAAE,mBAAmB,CAAC,cAAc,CAAC;QACjD,WAAW,EAAE,iBAAiB;KAC/B,EACD,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE;QACtB,IAAI,CAAC;YACH,MAAM,gBAAgB,GAAG,8BAA8B,CACrD,KAA0C,CAC3C,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE;gBAC9C,MAAM,EAAE,KAAK,EAAE,MAAM;gBACrB,cAAc,EAAE,gBAAgB;aACjC,CAAC,CAAC;YACH,OAAO,qBAAqB,CAAC,cAAc,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "coalesce-transform-mcp",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.4.1",
|
|
4
4
|
"mcpName": "io.github.jessemarshall/coalesce-transform",
|
|
5
5
|
"description": "MCP server for the Coalesce Transform API; run tools support Snowflake Key Pair auth only",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -17,7 +17,7 @@
|
|
|
17
17
|
"import:node-type-corpus": "node scripts/import-node-type-corpus.mjs",
|
|
18
18
|
"test": "vitest run",
|
|
19
19
|
"test:watch": "vitest",
|
|
20
|
-
"preversion": "npm run build && npm
|
|
20
|
+
"preversion": "npm run build && npm audit --omit=dev && node scripts/sync-version.mjs",
|
|
21
21
|
"postversion": "node scripts/sync-version.mjs && git add server.json && git commit --amend --no-edit",
|
|
22
22
|
"prepublishOnly": "npm run build",
|
|
23
23
|
"prepare": "husky"
|