@code-yeongyu/senpi-ai 2026.8.12 → 2026.8.13
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 +26 -8
- package/dist/api/anthropic-messages.js.map +1 -1
- package/dist/api/azure-openai-responses.js +5 -1
- package/dist/api/azure-openai-responses.js.map +1 -1
- package/dist/api/bedrock-converse-stream.js.map +1 -1
- package/dist/api/bedrock-converse-stream.lazy.js.map +1 -1
- package/dist/api/google-generative-ai.d.ts.map +1 -1
- package/dist/api/google-generative-ai.js +2 -2
- package/dist/api/google-generative-ai.js.map +1 -1
- package/dist/api/google-shared.d.ts +15 -1
- package/dist/api/google-shared.d.ts.map +1 -1
- package/dist/api/google-shared.js +103 -4
- package/dist/api/google-shared.js.map +1 -1
- package/dist/api/google-vertex.d.ts.map +1 -1
- package/dist/api/google-vertex.js +2 -2
- package/dist/api/google-vertex.js.map +1 -1
- package/dist/api/lazy.d.ts +5 -1
- package/dist/api/lazy.d.ts.map +1 -1
- package/dist/api/lazy.js +19 -7
- package/dist/api/lazy.js.map +1 -1
- package/dist/api/openai-codex-responses.js.map +1 -1
- package/dist/api/openai-completions.d.ts +3 -1
- package/dist/api/openai-completions.d.ts.map +1 -1
- package/dist/api/openai-completions.js +47 -7
- package/dist/api/openai-completions.js.map +1 -1
- package/dist/api/openai-responses-shared.d.ts.map +1 -1
- package/dist/api/openai-responses-shared.js +22 -9
- package/dist/api/openai-responses-shared.js.map +1 -1
- package/dist/api/openai-responses.js +5 -1
- package/dist/api/openai-responses.js.map +1 -1
- package/dist/api/pi-messages.d.ts.map +1 -1
- package/dist/api/pi-messages.js.map +1 -1
- package/dist/api/simple-options.d.ts +2 -0
- package/dist/api/simple-options.d.ts.map +1 -1
- package/dist/api/simple-options.js +8 -2
- package/dist/api/simple-options.js.map +1 -1
- package/dist/api/transform-messages.js.map +1 -1
- package/dist/api/warm-prompt-cache.js.map +1 -1
- package/dist/auth/credential-store.d.ts +6 -6
- package/dist/auth/credential-store.d.ts.map +1 -1
- package/dist/auth/credential-store.js +22 -11
- package/dist/auth/credential-store.js.map +1 -1
- package/dist/auth/helpers.d.ts +1 -0
- package/dist/auth/helpers.d.ts.map +1 -1
- package/dist/auth/helpers.js +7 -2
- package/dist/auth/helpers.js.map +1 -1
- package/dist/auth/oauth/anthropic.d.ts.map +1 -1
- package/dist/auth/oauth/anthropic.js +14 -8
- package/dist/auth/oauth/anthropic.js.map +1 -1
- package/dist/auth/oauth/device-code.d.ts +1 -1
- package/dist/auth/oauth/device-code.d.ts.map +1 -1
- package/dist/auth/oauth/device-code.js +4 -4
- package/dist/auth/oauth/device-code.js.map +1 -1
- package/dist/auth/oauth/github-copilot.d.ts.map +1 -1
- package/dist/auth/oauth/github-copilot.js +43 -30
- package/dist/auth/oauth/github-copilot.js.map +1 -1
- package/dist/auth/oauth/kimi-coding.d.ts.map +1 -1
- package/dist/auth/oauth/kimi-coding.js +17 -5
- package/dist/auth/oauth/kimi-coding.js.map +1 -1
- package/dist/auth/oauth/load.d.ts.map +1 -1
- package/dist/auth/oauth/load.js.map +1 -1
- package/dist/auth/oauth/openai-codex.d.ts.map +1 -1
- package/dist/auth/oauth/openai-codex.js +12 -5
- package/dist/auth/oauth/openai-codex.js.map +1 -1
- package/dist/auth/oauth/openrouter.d.ts.map +1 -1
- package/dist/auth/oauth/openrouter.js +10 -10
- package/dist/auth/oauth/openrouter.js.map +1 -1
- package/dist/auth/oauth/radius.d.ts.map +1 -1
- package/dist/auth/oauth/radius.js +8 -7
- package/dist/auth/oauth/radius.js.map +1 -1
- package/dist/auth/oauth/xai.d.ts.map +1 -1
- package/dist/auth/oauth/xai.js +3 -2
- package/dist/auth/oauth/xai.js.map +1 -1
- package/dist/auth/resolve.d.ts +1 -0
- package/dist/auth/resolve.d.ts.map +1 -1
- package/dist/auth/resolve.js +24 -13
- package/dist/auth/resolve.js.map +1 -1
- package/dist/auth/types.d.ts +21 -8
- package/dist/auth/types.d.ts.map +1 -1
- package/dist/auth/types.js.map +1 -1
- package/dist/cli.js +1 -0
- package/dist/cli.js.map +1 -1
- package/dist/compat.js.map +1 -1
- package/dist/env-api-keys.d.ts.map +1 -1
- package/dist/env-api-keys.js +3 -0
- package/dist/env-api-keys.js.map +1 -1
- package/dist/image-models.generated.d.ts +69 -54
- package/dist/image-models.generated.d.ts.map +1 -1
- package/dist/image-models.generated.js +15 -0
- package/dist/image-models.generated.js.map +1 -1
- package/dist/images-models.d.ts.map +1 -1
- package/dist/images-models.js +1 -0
- package/dist/images-models.js.map +1 -1
- package/dist/model.d.ts +2 -0
- package/dist/model.d.ts.map +1 -1
- package/dist/model.js.map +1 -1
- package/dist/models-store.d.ts +9 -12
- package/dist/models-store.d.ts.map +1 -1
- package/dist/models-store.js +6 -3
- package/dist/models-store.js.map +1 -1
- package/dist/models.d.ts +40 -18
- package/dist/models.d.ts.map +1 -1
- package/dist/models.generated.d.ts +6 -0
- package/dist/models.generated.d.ts.map +1 -1
- package/dist/models.generated.js +6 -0
- package/dist/models.generated.js.map +1 -1
- package/dist/models.js +247 -91
- package/dist/models.js.map +1 -1
- package/dist/node/provider-scope.d.ts.map +1 -1
- package/dist/providers/all.d.ts.map +1 -1
- package/dist/providers/all.js +6 -0
- package/dist/providers/all.js.map +1 -1
- package/dist/providers/amazon-bedrock.d.ts.map +1 -1
- package/dist/providers/amazon-bedrock.js +15 -7
- package/dist/providers/amazon-bedrock.js.map +1 -1
- package/dist/providers/anthropic.d.ts.map +1 -1
- package/dist/providers/anthropic.js +15 -6
- package/dist/providers/anthropic.js.map +1 -1
- package/dist/providers/baseten.d.ts +3 -0
- package/dist/providers/baseten.d.ts.map +1 -0
- package/dist/providers/baseten.js +15 -0
- package/dist/providers/baseten.js.map +1 -0
- package/dist/providers/baseten.models.d.ts +4 -0
- package/dist/providers/baseten.models.d.ts.map +1 -0
- package/dist/providers/baseten.models.js +6 -0
- package/dist/providers/baseten.models.js.map +1 -0
- package/dist/providers/cloudflare-auth.d.ts.map +1 -1
- package/dist/providers/cloudflare-auth.js +15 -10
- package/dist/providers/cloudflare-auth.js.map +1 -1
- package/dist/providers/data/.manifest.json +1 -1
- package/dist/providers/data/baseten.json +1 -0
- package/dist/providers/data/fireworks.json +1 -1
- package/dist/providers/data/github-copilot.json +1 -1
- package/dist/providers/data/opencode-go.json +1 -1
- package/dist/providers/data/opencode.json +1 -1
- package/dist/providers/data/opengateway.json +1 -0
- package/dist/providers/data/openrouter.json +1 -1
- package/dist/providers/data/qwen-token-plan-cn.json +1 -1
- package/dist/providers/data/qwen-token-plan-individual.json +1 -0
- package/dist/providers/data/qwen-token-plan.json +1 -1
- package/dist/providers/data/vercel-ai-gateway.json +1 -1
- package/dist/providers/data/xai.json +1 -1
- package/dist/providers/faux.d.ts +18 -13
- package/dist/providers/faux.d.ts.map +1 -1
- package/dist/providers/faux.js +95 -5
- package/dist/providers/faux.js.map +1 -1
- package/dist/providers/github-copilot.js +1 -1
- package/dist/providers/github-copilot.js.map +1 -1
- package/dist/providers/google-vertex.d.ts.map +1 -1
- package/dist/providers/google-vertex.js +15 -7
- package/dist/providers/google-vertex.js.map +1 -1
- package/dist/providers/kimi-coding.d.ts.map +1 -1
- package/dist/providers/kimi-coding.js +1 -0
- package/dist/providers/kimi-coding.js.map +1 -1
- package/dist/providers/ollama.d.ts.map +1 -1
- package/dist/providers/ollama.js +1 -2
- package/dist/providers/ollama.js.map +1 -1
- package/dist/providers/openai-codex.d.ts.map +1 -1
- package/dist/providers/openai-codex.js +5 -1
- package/dist/providers/openai-codex.js.map +1 -1
- package/dist/providers/opengateway.d.ts +3 -0
- package/dist/providers/opengateway.d.ts.map +1 -0
- package/dist/providers/opengateway.js +15 -0
- package/dist/providers/opengateway.js.map +1 -0
- package/dist/providers/opengateway.models.d.ts +4 -0
- package/dist/providers/opengateway.models.d.ts.map +1 -0
- package/dist/providers/opengateway.models.js +6 -0
- package/dist/providers/opengateway.models.js.map +1 -0
- package/dist/providers/qwen-token-plan-individual.d.ts +3 -0
- package/dist/providers/qwen-token-plan-individual.d.ts.map +1 -0
- package/dist/providers/qwen-token-plan-individual.js +15 -0
- package/dist/providers/qwen-token-plan-individual.js.map +1 -0
- package/dist/providers/qwen-token-plan-individual.models.d.ts +4 -0
- package/dist/providers/qwen-token-plan-individual.models.d.ts.map +1 -0
- package/dist/providers/qwen-token-plan-individual.models.js +6 -0
- package/dist/providers/qwen-token-plan-individual.models.js.map +1 -0
- package/dist/providers/radius.d.ts.map +1 -1
- package/dist/providers/radius.js +36 -26
- package/dist/providers/radius.js.map +1 -1
- package/dist/providers/xai.d.ts.map +1 -1
- package/dist/providers/xai.js +1 -0
- package/dist/providers/xai.js.map +1 -1
- package/dist/tool-call-middleware/context-transformer.js.map +1 -1
- package/dist/tool-call-middleware/protocols/anthropic-xml/recovery-wrapper-state.d.ts.map +1 -1
- package/dist/tool-call-middleware/protocols/kimi-xtml/parse.js.map +1 -1
- package/dist/tool-call-middleware/recovery-content-lifecycle.d.ts.map +1 -1
- package/dist/tool-call-middleware/recovery-event-stream.d.ts.map +1 -1
- package/dist/tool-call-middleware/recovery-native-projection.d.ts.map +1 -1
- package/dist/tool-call-middleware/recovery-native-projection.js.map +1 -1
- package/dist/tool-call-middleware/recovery-stream-terminal.d.ts.map +1 -1
- package/dist/tool-call-middleware/recovery-stream-wrapper.d.ts.map +1 -1
- package/dist/tool-call-middleware/recovery-stream-wrapper.js +5 -0
- package/dist/tool-call-middleware/recovery-stream-wrapper.js.map +1 -1
- package/dist/tool-call-middleware/recovery-text-projection.d.ts.map +1 -1
- package/dist/tool-call-middleware/recovery-text-projection.js.map +1 -1
- package/dist/tool-call-middleware/stream-thinking-projection.d.ts.map +1 -1
- package/dist/tool-call-middleware/stream-wrapper-shared.d.ts.map +1 -1
- package/dist/tool-call-middleware/stream-wrapper-shared.js.map +1 -1
- package/dist/tool-call-middleware/stream-wrapper.js.map +1 -1
- package/dist/types.d.ts +94 -87
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js.map +1 -1
- package/dist/utils/abort.d.ts +8 -0
- package/dist/utils/abort.d.ts.map +1 -0
- package/dist/utils/abort.js +49 -0
- package/dist/utils/abort.js.map +1 -0
- package/dist/utils/deferred-tools.js.map +1 -1
- package/dist/utils/event-stream.d.ts.map +1 -1
- package/dist/utils/overflow.d.ts +14 -1
- package/dist/utils/overflow.d.ts.map +1 -1
- package/dist/utils/overflow.js +18 -1
- package/dist/utils/overflow.js.map +1 -1
- package/dist/utils/prompt-cache-ttl.d.ts +3 -1
- package/dist/utils/prompt-cache-ttl.d.ts.map +1 -1
- package/dist/utils/prompt-cache-ttl.js +2 -0
- package/dist/utils/prompt-cache-ttl.js.map +1 -1
- package/dist/utils/provider-retry.js.map +1 -1
- package/dist/utils/retry.d.ts +25 -0
- package/dist/utils/retry.d.ts.map +1 -1
- package/dist/utils/retry.js +64 -0
- package/dist/utils/retry.js.map +1 -1
- package/dist/utils/tool-schema-compat.d.ts.map +1 -1
- package/dist/utils/tool-schema-compat.js +1 -0
- package/dist/utils/tool-schema-compat.js.map +1 -1
- package/dist/utils/validation.d.ts.map +1 -1
- package/dist/utils/validation.js +6 -0
- package/dist/utils/validation.js.map +1 -1
- package/package.json +2 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"recovery-stream-terminal.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-terminal.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC7F,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAM1E,iFAAiF;AACjF,qBAAa,sBAAsB;IAClC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;
|
|
1
|
+
{"version":3,"file":"recovery-stream-terminal.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-terminal.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC7F,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAM1E,iFAAiF;AACjF,qBAAa,sBAAsB;IAClC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;gBAEzC,MAAM,EAAE,2BAA2B;IAI/C,IAAI,CACH,UAAU,EAAE,uBAAuB,EACnC,MAAM,EAAE,gBAAgB,EACxB,WAAW,EAAE,OAAO,EACpB,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC,GACxD,IAAI;IASP,WAAW,CACV,UAAU,EAAE,uBAAuB,EACnC,MAAM,EAAE,gBAAgB,EACxB,WAAW,EAAE,OAAO,EACpB,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,SAAS,GAAG,OAAO,CAAC,GAC9C,IAAI;IAoBP,eAAe,CAAC,UAAU,EAAE,uBAAuB,GAAG,IAAI,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI;IAcjF,SAAS,CAAC,UAAU,EAAE,uBAAuB,GAAG,IAAI,GAAG,IAAI;IAa3D,SAAS,CAAC,UAAU,EAAE,uBAAuB,GAAG,IAAI,GAAG,IAAI;IAc3D,OAAO,CAAC,IAAI;CAMZ"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"recovery-stream-wrapper.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAA2C,2BAA2B,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAC9G,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAC;AASzF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,MAAM,WAAW,qBAAqB;IACrC,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,KAAK,oBAAoB,CAAC;IAChE,QAAQ,CAAC,EAAE,cAAc,CAAC;CAC1B;AAED,wBAAgB,4BAA4B,CAC3C,WAAW,EAAE,2BAA2B,EACxC,KAAK,EAAE,SAAS,IAAI,EAAE,EACtB,OAAO,GAAE,qBAAqB,GAAG,CAAC,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,KAAK,oBAAoB,CAAM,GACtF,2BAA2B,
|
|
1
|
+
{"version":3,"file":"recovery-stream-wrapper.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAA2C,2BAA2B,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAC9G,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAC;AASzF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,MAAM,WAAW,qBAAqB;IACrC,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,KAAK,oBAAoB,CAAC;IAChE,QAAQ,CAAC,EAAE,cAAc,CAAC;CAC1B;AAED,wBAAgB,4BAA4B,CAC3C,WAAW,EAAE,2BAA2B,EACxC,KAAK,EAAE,SAAS,IAAI,EAAE,EACtB,OAAO,GAAE,qBAAqB,GAAG,CAAC,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,KAAK,oBAAoB,CAAM,GACtF,2BAA2B,CAmO7B"}
|
|
@@ -141,6 +141,11 @@ export function wrapStreamWithInvokeRecovery(innerStream, tools, options = {}) {
|
|
|
141
141
|
return;
|
|
142
142
|
if (!outerStream.markSourceClosed())
|
|
143
143
|
return;
|
|
144
|
+
if (event.reason === "deferred") {
|
|
145
|
+
outerStream.push(event);
|
|
146
|
+
outerStream.end(event.message);
|
|
147
|
+
return;
|
|
148
|
+
}
|
|
144
149
|
terminal.done(projection, event.message, sawToolCall, event.reason);
|
|
145
150
|
return;
|
|
146
151
|
case "error":
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"recovery-stream-wrapper.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-wrapper.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,qCAAqC,EAAE,MAAM,sCAAsC,CAAC;AAC7F,OAAO,EAA4B,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AACrG,OAAO,EAAE,mCAAmC,EAAE,MAAM,4BAA4B,CAAC;AACjF,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAA8B,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AAC7G,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAQrE,MAAM,UAAU,4BAA4B,CAC3C,WAAwC,EACxC,KAAsB,EACtB,OAAO,GAA+E,EAAE;IAExF,MAAM,WAAW,GAAG,IAAI,mCAAmC,EAAE,CAAC;IAC9D,MAAM,eAAe,GAAG,OAAO,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IAC5F,MAAM,YAAY,GAAG,eAAe,CAAC,YAAY,IAAI,qCAAqC,CAAC;IAC3F,MAAM,QAAQ,GAAG,eAAe,CAAC,QAAQ,IAAI,OAAO,CAAC;IAErD,KAAK,CAAC,KAAK,IAAmB,EAAE;QAC/B,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAC1D,MAAM,WAAW,GAAyC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,aAAa,EAAE,CAAC;QAC1G,IAAI,UAAU,GAAmC,IAAI,CAAC;QACtD,IAAI,gBAAgB,GAAoC,IAAI,CAAC;QAC7D,IAAI,cAAc,GAAkC,IAAI,CAAC;QACzD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAClC,MAAM,gBAAgB,GAAG,IAAI,wBAAwB,EAAE,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CAAC,WAAW,CAAC,CAAC;QAEzD,MAAM,UAAU,GAAG,GAAS,EAAE;YAC7B,IAAI,CAAC,cAAc;gBAAE,OAAO;YAC5B,WAAW,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,WAAW,CAAC;YACrD,cAAc,GAAG,IAAI,CAAC;QACvB,CAAC,CAAC;QAEF,WAAW,CAAC,sBAAsB,CAAC,KAAK,IAAI,EAAE;YAC7C,qBAAqB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC;gBACJ,MAAM,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC/B,UAAU,EAAE,CAAC;gBACb,QAAQ,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,UAAU,EAAE,CAAC;gBACb,QAAQ,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC5C,MAAM,KAAK,CAAC;YACb,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAE,OAA8B,EAAQ,EAAE;YAC9F,IAAI,CAAC,UAAU,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YAC3D,UAAU,EAAE,CAAC;YACb,iCAAiC,CAAC,WAAW,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC3F,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAE,YAAoB,EAAW,EAAE;YACvF,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;gBAAE,OAAO,KAAK,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,IAAI,YAAY,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACtG,mBAAmB,CAAC,MAAM,EAAE,4BAA4B,CAAC,CAAC;gBAC1D,OAAO,KAAK,CAAC;YACd,CAAC;YACD,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,gBAAgB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,EAAE,YAAY,CAAC;gBAAE,OAAO,IAAI,CAAC;YACzE,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,CAChB,MAAwB,EACxB,YAAoB,EACpB,IAAyB,EACzB,OAAO,GAA0B,6BAA6B,EACpD,EAAE;YACZ,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,IAAI,YAAY,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACtG,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBACrC,OAAO,KAAK,CAAC;YACd,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,YAAY,GACjB,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,CAAC;gBAC3C,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,CAAC;gBACnD,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,CAAC,CAAC;YACrD,IAAI,YAAY,IAAI,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAAE,OAAO,IAAI,CAAC;YACzE,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,CAChB,MAAwB,EACxB,YAAoB,EACpB,IAAyB,EACzB,OAAO,GAA0B,6BAA6B,EACpD,EAAE;YACZ,IAAI,gBAAgB,CAAC,QAAQ,CAAC,YAAY,EAAE,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC/D,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAW,EAAE;YACjE,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,sBAAsB,EAAE,IAAI,EAAE,CAAC,CAAC;YAClG,gBAAgB,KAAK,IAAI,wBAAwB,CAAC,WAAW,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC7F,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,gBAAgB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC3C,UAAU,EAAE,CAAC;YACb,IAAI,gBAAgB,CAAC,oBAAoB,CAAC,MAAM,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC/D,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;gBACvC,IAAI,qBAAqB;oBAAE,SAAS;gBACpC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACpB,KAAK,OAAO;wBACX,UAAU,GAAG,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,EAAE;4BACpE,sBAAsB,EAAE,IAAI;yBAC5B,CAAC,CAAC;wBACH,gBAAgB,GAAG,IAAI,wBAAwB,CAAC,WAAW,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;wBAC3F,gBAAgB,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAClD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;wBACjE,MAAM;oBACP,KAAK,YAAY,EAAE,CAAC;wBACnB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;4BAC9F,OAAO;wBACR,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CAAC,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,KAAK,CAAC,YAAY,EAAE;4BACpG,YAAY;4BACZ,QAAQ;yBACR,CAAC,CAAC;wBACH,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;4BACpC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,cAAc,GAAG,QAAQ,CAAC;wBAC1B,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;wBACnD,MAAM;oBACP,CAAC;oBACD,KAAK,YAAY;wBAChB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc;4BAAE,OAAO;wBACvF,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC;wBAC9D,MAAM;oBACP,KAAK,UAAU;wBACd,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC;4BAAE,OAAO;wBACpE,UAAU,EAAE,CAAC;wBACb,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;wBACjD,MAAM;oBACP,KAAK,MAAM;wBACV,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBAC/D,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;4BAAE,OAAO;wBAC5C,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;wBACpE,OAAO;oBACR,KAAK,OAAO;wBACX,IAAI,CAAC,UAAU,EAAE,CAAC;4BACjB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gCAAE,OAAO;4BAC5C,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,EAAE,CAAC;4BAClB,OAAO;wBACR,CAAC;wBACD,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC;4BAAE,OAAO;wBAC9C,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;4BAAE,OAAO;wBAC5C,QAAQ,CAAC,WAAW,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;wBACzE,OAAO;oBACR,KAAK,gBAAgB,EAAE,CAAC;wBACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,MAAM,MAAM,GAAG,gBAAgB,CAAC,kBAAkB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;wBACtF,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;4BAC5B,mBAAmB,CAClB,KAAK,CAAC,OAAO,EACb,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,4BAA4B,CACnE,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACvD,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB;wBACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;4BAC1F,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC5E,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACrD,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM;oBACP,KAAK,gBAAgB,EAAE,CAAC;wBACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;4BAC9F,OAAO;wBACR,MAAM,UAAU,GAAG,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/E,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,EAAE,CAAC;4BAC5E,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACvD,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB;wBACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBACnF,UAAU,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAChF,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBACnF,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5E,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACrD,MAAM;gBACR,CAAC;YACF,CAAC;YAED,IAAI,qBAAqB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YACrE,UAAU,EAAE,CAAC;YACb,QAAQ,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,qBAAqB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YACrE,UAAU,EAAE,CAAC;YACb,QAAQ,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACpB,CAAC","sourcesContent":["import type { AssistantMessage, AssistantMessageEvent, AssistantMessageEventStream, Tool } from \"../types.ts\";\nimport type { RecoveryStreamParser } from \"./protocols/anthropic-xml/recovery-stream.ts\";\nimport { createAntmlInvokeRecoveryStreamParser } from \"./protocols/antml/recovery-stream.ts\";\nimport { type RecoveryContentKind, RecoveryContentLifecycle } from \"./recovery-content-lifecycle.ts\";\nimport { RecoveryAssistantMessageEventStream } from \"./recovery-event-stream.ts\";\nimport { RecoveryNativeProjection } from \"./recovery-native-projection.ts\";\nimport { type RecoveryStreamFailure, terminateRecoveryStreamForFailure } from \"./recovery-stream-failure.ts\";\nimport { RecoveryStreamTerminal } from \"./recovery-stream-terminal.ts\";\nimport { RecoveryTextProjection } from \"./recovery-text-projection.ts\";\nimport { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\nimport type { ToolCallFormat } from \"./types.ts\";\n\nexport interface InvokeRecoveryOptions {\n\tcreateParser?: (tools: readonly Tool[]) => RecoveryStreamParser;\n\tprotocol?: ToolCallFormat;\n}\n\nexport function wrapStreamWithInvokeRecovery(\n\tinnerStream: AssistantMessageEventStream,\n\ttools: readonly Tool[],\n\toptions: InvokeRecoveryOptions | ((tools: readonly Tool[]) => RecoveryStreamParser) = {},\n): AssistantMessageEventStream {\n\tconst outerStream = new RecoveryAssistantMessageEventStream();\n\tconst resolvedOptions = typeof options === \"function\" ? { createParser: options } : options;\n\tconst createParser = resolvedOptions.createParser ?? createAntmlInvokeRecoveryStreamParser;\n\tconst protocol = resolvedOptions.protocol ?? \"antml\";\n\n\tvoid (async (): Promise<void> => {\n\t\tconst innerIterator = innerStream[Symbol.asyncIterator]();\n\t\tconst innerEvents: AsyncIterable<AssistantMessageEvent> = { [Symbol.asyncIterator]: () => innerIterator };\n\t\tlet projection: StreamMessageProjection | null = null;\n\t\tlet nativeProjection: RecoveryNativeProjection | null = null;\n\t\tlet textProjection: RecoveryTextProjection | null = null;\n\t\tlet sawToolCall = false;\n\t\tlet cancellationRequested = false;\n\t\tconst contentLifecycle = new RecoveryContentLifecycle();\n\t\tconst terminal = new RecoveryStreamTerminal(outerStream);\n\n\t\tconst finishText = (): void => {\n\t\t\tif (!textProjection) return;\n\t\t\tsawToolCall = textProjection.finish() || sawToolCall;\n\t\t\ttextProjection = null;\n\t\t};\n\n\t\touterStream.setCancellationHandler(async () => {\n\t\t\tcancellationRequested = true;\n\t\t\ttry {\n\t\t\t\tawait innerIterator.return?.();\n\t\t\t\tfinishText();\n\t\t\t\tterminal.cancelled(projection);\n\t\t\t} catch (error) {\n\t\t\t\tfinishText();\n\t\t\t\tterminal.iteratorFailure(projection, error);\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t});\n\n\t\tconst terminateForFailure = (source: AssistantMessage, failure: RecoveryStreamFailure): void => {\n\t\t\tif (!projection || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminateRecoveryStreamForFailure(outerStream, projection, { source, failure, protocol });\n\t\t};\n\n\t\tconst prepareContentEvent = (source: AssistantMessage, contentIndex: number): boolean => {\n\t\t\tif (!projection || !nativeProjection) return false;\n\t\t\tif (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {\n\t\t\t\tterminateForFailure(source, \"invalid_native_event_order\");\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tprojection.sync(source);\n\t\t\tnativeProjection.reserveVisibleIds(source);\n\t\t\tif (nativeProjection.synchronizeLower(source, contentIndex)) return true;\n\t\t\tterminateForFailure(source, \"collision\");\n\t\t\treturn false;\n\t\t};\n\n\t\tconst canStart = (\n\t\t\tsource: AssistantMessage,\n\t\t\tcontentIndex: number,\n\t\t\tkind: RecoveryContentKind,\n\t\t\tfailure: RecoveryStreamFailure = \"invalid_content_event_order\",\n\t\t): boolean => {\n\t\t\tif (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {\n\t\t\t\tterminateForFailure(source, failure);\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tconst block = source.content[contentIndex];\n\t\t\tconst blockMatches =\n\t\t\t\t(kind === \"text\" && block?.type === \"text\") ||\n\t\t\t\t(kind === \"thinking\" && block?.type === \"thinking\") ||\n\t\t\t\t(kind === \"toolCall\" && block?.type === \"toolCall\");\n\t\t\tif (blockMatches && contentLifecycle.canStart(contentIndex)) return true;\n\t\t\tterminateForFailure(source, failure);\n\t\t\treturn false;\n\t\t};\n\n\t\tconst isActive = (\n\t\t\tsource: AssistantMessage,\n\t\t\tcontentIndex: number,\n\t\t\tkind: RecoveryContentKind,\n\t\t\tfailure: RecoveryStreamFailure = \"invalid_content_event_order\",\n\t\t): boolean => {\n\t\t\tif (contentLifecycle.isActive(contentIndex, kind)) return true;\n\t\t\tterminateForFailure(source, failure);\n\t\t\treturn false;\n\t\t};\n\n\t\tconst synchronizeTerminal = (source: AssistantMessage): boolean => {\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, source, { preserveSourceMetadata: true });\n\t\t\tnativeProjection ??= new RecoveryNativeProjection(outerStream, projection.message, protocol);\n\t\t\tprojection.sync(source);\n\t\t\tnativeProjection.reserveVisibleIds(source);\n\t\t\tfinishText();\n\t\t\tif (nativeProjection.synchronizeRemaining(source)) return true;\n\t\t\tterminateForFailure(source, \"collision\");\n\t\t\treturn false;\n\t\t};\n\n\t\ttry {\n\t\t\tfor await (const event of innerEvents) {\n\t\t\t\tif (cancellationRequested) continue;\n\t\t\t\tswitch (event.type) {\n\t\t\t\t\tcase \"start\":\n\t\t\t\t\t\tprojection = new StreamMessageProjection(outerStream, event.partial, {\n\t\t\t\t\t\t\tpreserveSourceMetadata: true,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tnativeProjection = new RecoveryNativeProjection(outerStream, projection.message, protocol);\n\t\t\t\t\t\tnativeProjection.reserveVisibleIds(event.partial);\n\t\t\t\t\t\touterStream.push({ type: \"start\", partial: projection.message });\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\tconst nextText = new RecoveryTextProjection(tools, projection, nativeProjection, event.contentIndex, {\n\t\t\t\t\t\t\tcreateParser,\n\t\t\t\t\t\t\tprotocol,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tif (!nextText.start(event.partial)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\ttextProjection = nextText;\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"text\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_delta\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !textProjection) return;\n\t\t\t\t\t\tsawToolCall = textProjection.feed(event.delta) || sawToolCall;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex)) return;\n\t\t\t\t\t\tfinishText();\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"text\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"done\":\n\t\t\t\t\t\tif (!synchronizeTerminal(event.message) || !projection) return;\n\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\tterminal.done(projection, event.message, sawToolCall, event.reason);\n\t\t\t\t\t\treturn;\n\t\t\t\t\tcase \"error\":\n\t\t\t\t\t\tif (!projection) {\n\t\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end();\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (!synchronizeTerminal(event.error)) return;\n\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\tterminal.sourceError(projection, event.error, sawToolCall, event.reason);\n\t\t\t\t\t\treturn;\n\t\t\t\t\tcase \"toolcall_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tconst status = nativeProjection.projectNativeStart(event.partial, event.contentIndex);\n\t\t\t\t\t\tif (status !== \"projected\") {\n\t\t\t\t\t\t\tterminateForFailure(\n\t\t\t\t\t\t\t\tevent.partial,\n\t\t\t\t\t\t\t\tstatus === \"collision\" ? \"collision\" : \"invalid_native_event_order\",\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"toolCall\");\n\t\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tif (!nativeProjection.projectNativeDelta(event.partial, event.contentIndex, event.delta)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"toolcall_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tif (!nativeProjection.projectNativeEnd(event.contentIndex, event.toolCall)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"toolCall\");\n\t\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\tconst outerIndex = projection.startThinking(event.contentIndex, event.partial);\n\t\t\t\t\t\tif (!nativeProjection.recordProjectedBlock(event.contentIndex, outerIndex)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"thinking\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"thinking_delta\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection) return;\n\t\t\t\t\t\tprojection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection) return;\n\t\t\t\t\t\tprojection.finishThinking(event.contentIndex, event.content, event.partial);\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"thinking\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif (cancellationRequested || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminal.exhausted(projection);\n\t\t} catch (error) {\n\t\t\tif (cancellationRequested || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminal.iteratorFailure(projection, error);\n\t\t}\n\t})();\n\n\treturn outerStream;\n}\n"]}
|
|
1
|
+
{"version":3,"file":"recovery-stream-wrapper.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-stream-wrapper.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,qCAAqC,EAAE,MAAM,sCAAsC,CAAC;AAC7F,OAAO,EAA4B,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AACrG,OAAO,EAAE,mCAAmC,EAAE,MAAM,4BAA4B,CAAC;AACjF,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAA8B,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AAC7G,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAQrE,MAAM,UAAU,4BAA4B,CAC3C,WAAwC,EACxC,KAAsB,EACtB,UAAsF,EAAE;IAExF,MAAM,WAAW,GAAG,IAAI,mCAAmC,EAAE,CAAC;IAC9D,MAAM,eAAe,GAAG,OAAO,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IAC5F,MAAM,YAAY,GAAG,eAAe,CAAC,YAAY,IAAI,qCAAqC,CAAC;IAC3F,MAAM,QAAQ,GAAG,eAAe,CAAC,QAAQ,IAAI,OAAO,CAAC;IAErD,KAAK,CAAC,KAAK,IAAmB,EAAE;QAC/B,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAC1D,MAAM,WAAW,GAAyC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,aAAa,EAAE,CAAC;QAC1G,IAAI,UAAU,GAAmC,IAAI,CAAC;QACtD,IAAI,gBAAgB,GAAoC,IAAI,CAAC;QAC7D,IAAI,cAAc,GAAkC,IAAI,CAAC;QACzD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAClC,MAAM,gBAAgB,GAAG,IAAI,wBAAwB,EAAE,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CAAC,WAAW,CAAC,CAAC;QAEzD,MAAM,UAAU,GAAG,GAAS,EAAE;YAC7B,IAAI,CAAC,cAAc;gBAAE,OAAO;YAC5B,WAAW,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,WAAW,CAAC;YACrD,cAAc,GAAG,IAAI,CAAC;QACvB,CAAC,CAAC;QAEF,WAAW,CAAC,sBAAsB,CAAC,KAAK,IAAI,EAAE;YAC7C,qBAAqB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC;gBACJ,MAAM,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC/B,UAAU,EAAE,CAAC;gBACb,QAAQ,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,UAAU,EAAE,CAAC;gBACb,QAAQ,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC5C,MAAM,KAAK,CAAC;YACb,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAE,OAA8B,EAAQ,EAAE;YAC9F,IAAI,CAAC,UAAU,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YAC3D,UAAU,EAAE,CAAC;YACb,iCAAiC,CAAC,WAAW,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC3F,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAE,YAAoB,EAAW,EAAE;YACvF,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;gBAAE,OAAO,KAAK,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,IAAI,YAAY,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACtG,mBAAmB,CAAC,MAAM,EAAE,4BAA4B,CAAC,CAAC;gBAC1D,OAAO,KAAK,CAAC;YACd,CAAC;YACD,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,gBAAgB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,EAAE,YAAY,CAAC;gBAAE,OAAO,IAAI,CAAC;YACzE,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,CAChB,MAAwB,EACxB,YAAoB,EACpB,IAAyB,EACzB,UAAiC,6BAA6B,EACpD,EAAE;YACZ,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,IAAI,YAAY,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACtG,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBACrC,OAAO,KAAK,CAAC;YACd,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,YAAY,GACjB,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,CAAC;gBAC3C,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,CAAC;gBACnD,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,CAAC,CAAC;YACrD,IAAI,YAAY,IAAI,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAAE,OAAO,IAAI,CAAC;YACzE,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,CAChB,MAAwB,EACxB,YAAoB,EACpB,IAAyB,EACzB,UAAiC,6BAA6B,EACpD,EAAE;YACZ,IAAI,gBAAgB,CAAC,QAAQ,CAAC,YAAY,EAAE,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC/D,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,mBAAmB,GAAG,CAAC,MAAwB,EAAW,EAAE;YACjE,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,sBAAsB,EAAE,IAAI,EAAE,CAAC,CAAC;YAClG,gBAAgB,KAAK,IAAI,wBAAwB,CAAC,WAAW,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC7F,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,gBAAgB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC3C,UAAU,EAAE,CAAC;YACb,IAAI,gBAAgB,CAAC,oBAAoB,CAAC,MAAM,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC/D,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;gBACvC,IAAI,qBAAqB;oBAAE,SAAS;gBACpC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACpB,KAAK,OAAO;wBACX,UAAU,GAAG,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,EAAE;4BACpE,sBAAsB,EAAE,IAAI;yBAC5B,CAAC,CAAC;wBACH,gBAAgB,GAAG,IAAI,wBAAwB,CAAC,WAAW,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;wBAC3F,gBAAgB,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAClD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;wBACjE,MAAM;oBACP,KAAK,YAAY,CAAC,CAAC,CAAC;wBACnB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;4BAC9F,OAAO;wBACR,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CAAC,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,KAAK,CAAC,YAAY,EAAE;4BACpG,YAAY;4BACZ,QAAQ;yBACR,CAAC,CAAC;wBACH,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;4BACpC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,cAAc,GAAG,QAAQ,CAAC;wBAC1B,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;wBACnD,MAAM;oBACP,CAAC;oBACD,KAAK,YAAY;wBAChB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc;4BAAE,OAAO;wBACvF,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC;wBAC9D,MAAM;oBACP,KAAK,UAAU;wBACd,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC;4BAAE,OAAO;wBACjE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC;4BAAE,OAAO;wBACpE,UAAU,EAAE,CAAC;wBACb,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;wBACjD,MAAM;oBACP,KAAK,MAAM;wBACV,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBAC/D,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;4BAAE,OAAO;wBAC5C,IAAI,KAAK,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;4BACjC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC/B,OAAO;wBACR,CAAC;wBACD,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;wBACpE,OAAO;oBACR,KAAK,OAAO;wBACX,IAAI,CAAC,UAAU,EAAE,CAAC;4BACjB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gCAAE,OAAO;4BAC5C,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,EAAE,CAAC;4BAClB,OAAO;wBACR,CAAC;wBACD,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC;4BAAE,OAAO;wBAC9C,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;4BAAE,OAAO;wBAC5C,QAAQ,CAAC,WAAW,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;wBACzE,OAAO;oBACR,KAAK,gBAAgB,CAAC,CAAC,CAAC;wBACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,MAAM,MAAM,GAAG,gBAAgB,CAAC,kBAAkB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;wBACtF,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;4BAC5B,mBAAmB,CAClB,KAAK,CAAC,OAAO,EACb,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,4BAA4B,CACnE,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACvD,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB;wBACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;4BAC1F,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,EAAE,4BAA4B,CAAC;4BAAE,OAAO;wBACnG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB;4BAAE,OAAO;wBACzF,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC5E,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACrD,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM;oBACP,KAAK,gBAAgB,CAAC,CAAC,CAAC;wBACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,gBAAgB;4BAC9F,OAAO;wBACR,MAAM,UAAU,GAAG,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/E,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,EAAE,CAAC;4BAC5E,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAC;4BACjE,OAAO;wBACR,CAAC;wBACD,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACvD,MAAM;oBACP,CAAC;oBACD,KAAK,gBAAgB;wBACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBACnF,UAAU,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAChF,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC;4BAAE,OAAO;wBACrE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU;4BAAE,OAAO;wBACnF,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5E,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBACrD,MAAM;gBACR,CAAC;YACF,CAAC;YAED,IAAI,qBAAqB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YACrE,UAAU,EAAE,CAAC;YACb,QAAQ,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,qBAAqB,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE;gBAAE,OAAO;YACrE,UAAU,EAAE,CAAC;YACb,QAAQ,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACpB,CAAC","sourcesContent":["import type { AssistantMessage, AssistantMessageEvent, AssistantMessageEventStream, Tool } from \"../types.ts\";\nimport type { RecoveryStreamParser } from \"./protocols/anthropic-xml/recovery-stream.ts\";\nimport { createAntmlInvokeRecoveryStreamParser } from \"./protocols/antml/recovery-stream.ts\";\nimport { type RecoveryContentKind, RecoveryContentLifecycle } from \"./recovery-content-lifecycle.ts\";\nimport { RecoveryAssistantMessageEventStream } from \"./recovery-event-stream.ts\";\nimport { RecoveryNativeProjection } from \"./recovery-native-projection.ts\";\nimport { type RecoveryStreamFailure, terminateRecoveryStreamForFailure } from \"./recovery-stream-failure.ts\";\nimport { RecoveryStreamTerminal } from \"./recovery-stream-terminal.ts\";\nimport { RecoveryTextProjection } from \"./recovery-text-projection.ts\";\nimport { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\nimport type { ToolCallFormat } from \"./types.ts\";\n\nexport interface InvokeRecoveryOptions {\n\tcreateParser?: (tools: readonly Tool[]) => RecoveryStreamParser;\n\tprotocol?: ToolCallFormat;\n}\n\nexport function wrapStreamWithInvokeRecovery(\n\tinnerStream: AssistantMessageEventStream,\n\ttools: readonly Tool[],\n\toptions: InvokeRecoveryOptions | ((tools: readonly Tool[]) => RecoveryStreamParser) = {},\n): AssistantMessageEventStream {\n\tconst outerStream = new RecoveryAssistantMessageEventStream();\n\tconst resolvedOptions = typeof options === \"function\" ? { createParser: options } : options;\n\tconst createParser = resolvedOptions.createParser ?? createAntmlInvokeRecoveryStreamParser;\n\tconst protocol = resolvedOptions.protocol ?? \"antml\";\n\n\tvoid (async (): Promise<void> => {\n\t\tconst innerIterator = innerStream[Symbol.asyncIterator]();\n\t\tconst innerEvents: AsyncIterable<AssistantMessageEvent> = { [Symbol.asyncIterator]: () => innerIterator };\n\t\tlet projection: StreamMessageProjection | null = null;\n\t\tlet nativeProjection: RecoveryNativeProjection | null = null;\n\t\tlet textProjection: RecoveryTextProjection | null = null;\n\t\tlet sawToolCall = false;\n\t\tlet cancellationRequested = false;\n\t\tconst contentLifecycle = new RecoveryContentLifecycle();\n\t\tconst terminal = new RecoveryStreamTerminal(outerStream);\n\n\t\tconst finishText = (): void => {\n\t\t\tif (!textProjection) return;\n\t\t\tsawToolCall = textProjection.finish() || sawToolCall;\n\t\t\ttextProjection = null;\n\t\t};\n\n\t\touterStream.setCancellationHandler(async () => {\n\t\t\tcancellationRequested = true;\n\t\t\ttry {\n\t\t\t\tawait innerIterator.return?.();\n\t\t\t\tfinishText();\n\t\t\t\tterminal.cancelled(projection);\n\t\t\t} catch (error) {\n\t\t\t\tfinishText();\n\t\t\t\tterminal.iteratorFailure(projection, error);\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t});\n\n\t\tconst terminateForFailure = (source: AssistantMessage, failure: RecoveryStreamFailure): void => {\n\t\t\tif (!projection || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminateRecoveryStreamForFailure(outerStream, projection, { source, failure, protocol });\n\t\t};\n\n\t\tconst prepareContentEvent = (source: AssistantMessage, contentIndex: number): boolean => {\n\t\t\tif (!projection || !nativeProjection) return false;\n\t\t\tif (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {\n\t\t\t\tterminateForFailure(source, \"invalid_native_event_order\");\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tprojection.sync(source);\n\t\t\tnativeProjection.reserveVisibleIds(source);\n\t\t\tif (nativeProjection.synchronizeLower(source, contentIndex)) return true;\n\t\t\tterminateForFailure(source, \"collision\");\n\t\t\treturn false;\n\t\t};\n\n\t\tconst canStart = (\n\t\t\tsource: AssistantMessage,\n\t\t\tcontentIndex: number,\n\t\t\tkind: RecoveryContentKind,\n\t\t\tfailure: RecoveryStreamFailure = \"invalid_content_event_order\",\n\t\t): boolean => {\n\t\t\tif (!Number.isSafeInteger(contentIndex) || contentIndex < 0 || contentIndex >= source.content.length) {\n\t\t\t\tterminateForFailure(source, failure);\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tconst block = source.content[contentIndex];\n\t\t\tconst blockMatches =\n\t\t\t\t(kind === \"text\" && block?.type === \"text\") ||\n\t\t\t\t(kind === \"thinking\" && block?.type === \"thinking\") ||\n\t\t\t\t(kind === \"toolCall\" && block?.type === \"toolCall\");\n\t\t\tif (blockMatches && contentLifecycle.canStart(contentIndex)) return true;\n\t\t\tterminateForFailure(source, failure);\n\t\t\treturn false;\n\t\t};\n\n\t\tconst isActive = (\n\t\t\tsource: AssistantMessage,\n\t\t\tcontentIndex: number,\n\t\t\tkind: RecoveryContentKind,\n\t\t\tfailure: RecoveryStreamFailure = \"invalid_content_event_order\",\n\t\t): boolean => {\n\t\t\tif (contentLifecycle.isActive(contentIndex, kind)) return true;\n\t\t\tterminateForFailure(source, failure);\n\t\t\treturn false;\n\t\t};\n\n\t\tconst synchronizeTerminal = (source: AssistantMessage): boolean => {\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, source, { preserveSourceMetadata: true });\n\t\t\tnativeProjection ??= new RecoveryNativeProjection(outerStream, projection.message, protocol);\n\t\t\tprojection.sync(source);\n\t\t\tnativeProjection.reserveVisibleIds(source);\n\t\t\tfinishText();\n\t\t\tif (nativeProjection.synchronizeRemaining(source)) return true;\n\t\t\tterminateForFailure(source, \"collision\");\n\t\t\treturn false;\n\t\t};\n\n\t\ttry {\n\t\t\tfor await (const event of innerEvents) {\n\t\t\t\tif (cancellationRequested) continue;\n\t\t\t\tswitch (event.type) {\n\t\t\t\t\tcase \"start\":\n\t\t\t\t\t\tprojection = new StreamMessageProjection(outerStream, event.partial, {\n\t\t\t\t\t\t\tpreserveSourceMetadata: true,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tnativeProjection = new RecoveryNativeProjection(outerStream, projection.message, protocol);\n\t\t\t\t\t\tnativeProjection.reserveVisibleIds(event.partial);\n\t\t\t\t\t\touterStream.push({ type: \"start\", partial: projection.message });\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\tconst nextText = new RecoveryTextProjection(tools, projection, nativeProjection, event.contentIndex, {\n\t\t\t\t\t\t\tcreateParser,\n\t\t\t\t\t\t\tprotocol,\n\t\t\t\t\t\t});\n\t\t\t\t\t\tif (!nextText.start(event.partial)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\ttextProjection = nextText;\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"text\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_delta\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !textProjection) return;\n\t\t\t\t\t\tsawToolCall = textProjection.feed(event.delta) || sawToolCall;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"text\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex)) return;\n\t\t\t\t\t\tfinishText();\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"text\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"done\":\n\t\t\t\t\t\tif (!synchronizeTerminal(event.message) || !projection) return;\n\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\tif (event.reason === \"deferred\") {\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end(event.message);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tterminal.done(projection, event.message, sawToolCall, event.reason);\n\t\t\t\t\t\treturn;\n\t\t\t\t\tcase \"error\":\n\t\t\t\t\t\tif (!projection) {\n\t\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end();\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (!synchronizeTerminal(event.error)) return;\n\t\t\t\t\t\tif (!outerStream.markSourceClosed()) return;\n\t\t\t\t\t\tterminal.sourceError(projection, event.error, sawToolCall, event.reason);\n\t\t\t\t\t\treturn;\n\t\t\t\t\tcase \"toolcall_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tconst status = nativeProjection.projectNativeStart(event.partial, event.contentIndex);\n\t\t\t\t\t\tif (status !== \"projected\") {\n\t\t\t\t\t\t\tterminateForFailure(\n\t\t\t\t\t\t\t\tevent.partial,\n\t\t\t\t\t\t\t\tstatus === \"collision\" ? \"collision\" : \"invalid_native_event_order\",\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"toolCall\");\n\t\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tif (!nativeProjection.projectNativeDelta(event.partial, event.contentIndex, event.delta)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"toolcall_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"toolCall\", \"invalid_native_event_order\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !nativeProjection) return;\n\t\t\t\t\t\tif (!nativeProjection.projectNativeEnd(event.contentIndex, event.toolCall)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"toolCall\");\n\t\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_start\": {\n\t\t\t\t\t\tif (!canStart(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection || !nativeProjection)\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\tconst outerIndex = projection.startThinking(event.contentIndex, event.partial);\n\t\t\t\t\t\tif (!nativeProjection.recordProjectedBlock(event.contentIndex, outerIndex)) {\n\t\t\t\t\t\t\tterminateForFailure(event.partial, \"invalid_native_event_order\");\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcontentLifecycle.start(event.contentIndex, \"thinking\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"thinking_delta\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection) return;\n\t\t\t\t\t\tprojection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_end\":\n\t\t\t\t\t\tif (!isActive(event.partial, event.contentIndex, \"thinking\")) return;\n\t\t\t\t\t\tif (!prepareContentEvent(event.partial, event.contentIndex) || !projection) return;\n\t\t\t\t\t\tprojection.finishThinking(event.contentIndex, event.content, event.partial);\n\t\t\t\t\t\tcontentLifecycle.end(event.contentIndex, \"thinking\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif (cancellationRequested || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminal.exhausted(projection);\n\t\t} catch (error) {\n\t\t\tif (cancellationRequested || !outerStream.markSourceClosed()) return;\n\t\t\tfinishText();\n\t\t\tterminal.iteratorFailure(projection, error);\n\t\t}\n\t})();\n\n\treturn outerStream;\n}\n"]}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"recovery-text-projection.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-text-projection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAC1D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAC;AAIzF,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAChF,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAC1E,OAAO,KAAK,EAAqB,cAAc,EAAE,MAAM,YAAY,CAAC;AAEpE,iFAAiF;AACjF,qBAAa,sBAAsB;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA0B;IACrD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA2B;IAC5D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAuB;IAC9C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;IAC1C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA4B;IACjD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,UAAU,CAAM;IACxB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,QAAQ,CAAS;
|
|
1
|
+
{"version":3,"file":"recovery-text-projection.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-text-projection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAC1D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAC;AAIzF,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAChF,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAC1E,OAAO,KAAK,EAAqB,cAAc,EAAE,MAAM,YAAY,CAAC;AAEpE,iFAAiF;AACjF,qBAAa,sBAAsB;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA0B;IACrD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA2B;IAC5D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAuB;IAC9C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;IAC1C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA4B;IACjD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,UAAU,CAAM;IACxB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,QAAQ,CAAS;gBAGxB,KAAK,EAAE,SAAS,IAAI,EAAE,EACtB,UAAU,EAAE,uBAAuB,EACnC,gBAAgB,EAAE,wBAAwB,EAC1C,UAAU,EAAE,MAAM,EAClB,OAAO,GAAE;QACR,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,KAAK,oBAAoB,CAAC;QAChE,QAAQ,CAAC,EAAE,cAAc,CAAC;KACrB;IASP,KAAK,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO;IAKxC,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAU3B,MAAM,IAAI,OAAO;IAWjB,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,mBAAmB;IAoB3B,OAAO,CAAC,SAAS;CAKjB"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"recovery-text-projection.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-text-projection.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,qCAAqC,EAAE,MAAM,sCAAsC,CAAC;AAC7F,OAAO,EAAE,sBAAsB,EAAgC,MAAM,yBAAyB,CAAC;AAC/F,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAKrE,iFAAiF;AACjF,MAAM,OAAO,sBAAsB;IAYlC,YACC,KAAsB,EACtB,UAAmC,EACnC,gBAA0C,EAC1C,UAAkB,EAClB,
|
|
1
|
+
{"version":3,"file":"recovery-text-projection.js","sourceRoot":"","sources":["../../src/tool-call-middleware/recovery-text-projection.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,qCAAqC,EAAE,MAAM,sCAAsC,CAAC;AAC7F,OAAO,EAAE,sBAAsB,EAAgC,MAAM,yBAAyB,CAAC;AAC/F,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAKrE,iFAAiF;AACjF,MAAM,OAAO,sBAAsB;IAYlC,YACC,KAAsB,EACtB,UAAmC,EACnC,gBAA0C,EAC1C,UAAkB,EAClB,UAGI,EAAE;QAdU,SAAI,GAAG,sBAAsB,EAAE,CAAC;QACzC,iBAAY,GAAG,KAAK,CAAC;QACrB,eAAU,GAAG,EAAE,CAAC;QAChB,gBAAW,GAAG,KAAK,CAAC;QACpB,aAAQ,GAAG,KAAK,CAAC;QAYxB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,qCAAqC,CAAC,CAAC,KAAK,CAAC,CAAC;QACrF,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,MAAwB;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QAC/F,OAAO,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IACrE,CAAC;IAED,IAAI,CAAC,IAAY;QAChB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACrD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACrC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;YACvE,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC;gBAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACxF,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC;IACzB,CAAC;IAED,MAAM;QACL,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC,WAAW,CAAC;QAC3C,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACvE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC;QAC7B,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,OAAO,IAAI,CAAC,WAAW,CAAC;IACzB,CAAC;IAEO,cAAc,CAAC,OAAgC;QACtD,IAAI,OAAO,CAAC,gBAAgB;YAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAChF,IAAI,OAAO,CAAC,IAAI;YAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;;YACtE,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IACtC,CAAC;IAEO,mBAAmB,CAAC,MAAoC;QAC/D,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC;YACtE,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC3B,IAAI,CAAC,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC;gBAC9B,SAAS;YACV,CAAC;YACD,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YAC5D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;YAC1D,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB;gBAAE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC9D,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gBACnC,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,kBAAkB,EAAE,CAAC;oBAClD,wBAAwB,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACpE,CAAC;YACF,CAAC;QACF,CAAC;QACD,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACnD,CAAC;IAEO,SAAS;QAChB,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACzC,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QAC/E,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;IACtB,CAAC;CACD","sourcesContent":["import type { AssistantMessage, Tool } from \"../types.ts\";\nimport type { RecoveryStreamParser } from \"./protocols/anthropic-xml/recovery-stream.ts\";\nimport { createAntmlInvokeRecoveryStreamParser } from \"./protocols/antml/recovery-stream.ts\";\nimport { createRecoveryCodeMask, type RecoveryCodeMaskSegment } from \"./recovery-code-mask.ts\";\nimport { appendRecoveryDiagnostic } from \"./recovery-diagnostics.ts\";\nimport type { RecoveryNativeProjection } from \"./recovery-native-projection.ts\";\nimport type { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\nimport type { StreamParserEvent, ToolCallFormat } from \"./types.ts\";\n\n/** Owns one ordinary assistant text block's mask/parser/projection lifecycle. */\nexport class RecoveryTextProjection {\n\tprivate readonly projection: StreamMessageProjection;\n\tprivate readonly nativeProjection: RecoveryNativeProjection;\n\tprivate readonly innerIndex: number;\n\tprivate readonly parser: RecoveryStreamParser;\n\tprivate readonly protocol: ToolCallFormat;\n\tprivate readonly mask = createRecoveryCodeMask();\n\tprivate activeInvoke = false;\n\tprivate textBuffer = \"\";\n\tprivate sawToolCall = false;\n\tprivate finished = false;\n\n\tconstructor(\n\t\ttools: readonly Tool[],\n\t\tprojection: StreamMessageProjection,\n\t\tnativeProjection: RecoveryNativeProjection,\n\t\tinnerIndex: number,\n\t\toptions: {\n\t\t\tcreateParser?: (tools: readonly Tool[]) => RecoveryStreamParser;\n\t\t\tprotocol?: ToolCallFormat;\n\t\t} = {},\n\t) {\n\t\tthis.projection = projection;\n\t\tthis.nativeProjection = nativeProjection;\n\t\tthis.innerIndex = innerIndex;\n\t\tthis.parser = (options.createParser ?? createAntmlInvokeRecoveryStreamParser)(tools);\n\t\tthis.protocol = options.protocol ?? \"antml\";\n\t}\n\n\tstart(source: AssistantMessage): boolean {\n\t\tconst outerIndex = this.projection.startText(this.innerIndex, source.content[this.innerIndex]);\n\t\treturn this.nativeProjection.startText(this.innerIndex, outerIndex);\n\t}\n\n\tfeed(text: string): boolean {\n\t\tfor (let index = 0; index < text.length; index += 1) {\n\t\t\tconst character = text.charAt(index);\n\t\t\tconst options = this.activeInvoke ? { activeInvoke: true } : undefined;\n\t\t\tfor (const segment of this.mask.feed(character, options)) this.processSegment(segment);\n\t\t}\n\t\tthis.flushText();\n\t\treturn this.sawToolCall;\n\t}\n\n\tfinish(): boolean {\n\t\tif (this.finished) return this.sawToolCall;\n\t\tfor (const segment of this.mask.finish()) this.processSegment(segment);\n\t\tthis.projectParserEvents(this.parser.finish());\n\t\tthis.flushText();\n\t\tthis.projection.finishText();\n\t\tthis.nativeProjection.extendText(this.innerIndex);\n\t\tthis.finished = true;\n\t\treturn this.sawToolCall;\n\t}\n\n\tprivate processSegment(segment: RecoveryCodeMaskSegment): void {\n\t\tif (segment.recoveryBoundary) this.projectParserEvents(this.parser.interrupt());\n\t\tif (segment.scan) this.projectParserEvents(this.parser.feed(segment.text));\n\t\telse this.textBuffer += segment.text;\n\t}\n\n\tprivate projectParserEvents(events: readonly StreamParserEvent[]): void {\n\t\tfor (const event of this.nativeProjection.assignRecoveredIds(events)) {\n\t\t\tif (event.type === \"text\") {\n\t\t\t\tthis.textBuffer += event.text;\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tthis.flushText();\n\t\t\tconst result = this.projection.projectParserEvents([event]);\n\t\t\tthis.sawToolCall = this.sawToolCall || result.sawToolCall;\n\t\t\tif (event.type === \"toolcall_start\") this.activeInvoke = true;\n\t\t\tif (event.type === \"toolcall_end\") {\n\t\t\t\tthis.activeInvoke = false;\n\t\t\t\tfor (const toolCall of result.completedToolCalls) {\n\t\t\t\t\tappendRecoveryDiagnostic(this.projection, toolCall, this.protocol);\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tthis.nativeProjection.extendText(this.innerIndex);\n\t}\n\n\tprivate flushText(): void {\n\t\tif (this.textBuffer.length === 0) return;\n\t\tthis.projection.projectParserEvents([{ type: \"text\", text: this.textBuffer }]);\n\t\tthis.textBuffer = \"\";\n\t}\n}\n"]}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream-thinking-projection.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-thinking-projection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAmB,MAAM,aAAa,CAAC;AAMlG,wFAAwF;AACxF,qBAAa,wBAAwB;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IACrD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAA6B;IACpE,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAU;
|
|
1
|
+
{"version":3,"file":"stream-thinking-projection.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-thinking-projection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAmB,MAAM,aAAa,CAAC;AAMlG,wFAAwF;AACxF,qBAAa,wBAAwB;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IACrD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAA6B;IACpE,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAU;gBAErC,MAAM,EAAE,2BAA2B,EAAE,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,OAAO;IAM3G,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,MAAM;IAa3D,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAaxE,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;CAYxE"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream-wrapper-shared.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper-shared.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAe,QAAQ,EAAE,MAAM,aAAa,CAAC;AACxG,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AAI1E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAIpD,MAAM,MAAM,sBAAsB,GAAG;IACpC,QAAQ,CAAC,WAAW,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,kBAAkB,EAAE,SAAS,QAAQ,EAAE,CAAC;CACjD,CAAC;AAEF,KAAK,8BAA8B,GAAG;IAAE,QAAQ,CAAC,sBAAsB,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC;AAUpF,wEAAwE;AACxE,qBAAa,uBAAuB;IACnC,QAAQ,CAAC,OAAO,EAAE,gBAAgB,CAAC;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IACrD,OAAO,CAAC,qBAAqB,CAAuB;IACpD,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAAoC;IAC/E,OAAO,CAAC,QAAQ,CAAC,8BAA8B,CAA6B;IAC5E,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAA6B;IACxE,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAoC;IACzE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA2B;IACpD,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAU;IACjD,OAAO,CAAC,MAAM,CAAmB;
|
|
1
|
+
{"version":3,"file":"stream-wrapper-shared.d.ts","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper-shared.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,2BAA2B,EAAe,QAAQ,EAAE,MAAM,aAAa,CAAC;AACxG,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AAI1E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAIpD,MAAM,MAAM,sBAAsB,GAAG;IACpC,QAAQ,CAAC,WAAW,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,kBAAkB,EAAE,SAAS,QAAQ,EAAE,CAAC;CACjD,CAAC;AAEF,KAAK,8BAA8B,GAAG;IAAE,QAAQ,CAAC,sBAAsB,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC;AAUpF,wEAAwE;AACxE,qBAAa,uBAAuB;IACnC,QAAQ,CAAC,OAAO,EAAE,gBAAgB,CAAC;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8B;IACrD,OAAO,CAAC,qBAAqB,CAAuB;IACpD,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAAoC;IAC/E,OAAO,CAAC,QAAQ,CAAC,8BAA8B,CAA6B;IAC5E,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAA6B;IACxE,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAoC;IACzE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA2B;IACpD,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAU;IACjD,OAAO,CAAC,MAAM,CAAmB;gBAGhC,MAAM,EAAE,2BAA2B,EACnC,MAAM,EAAE,gBAAgB,EACxB,OAAO,GAAE,8BAAmC;IAS7C,IAAI,CAAC,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAKpC,gBAAgB,CAAC,UAAU,EAAE,0BAA0B,GAAG,IAAI;IAK9D,SAAS,CAAC,YAAY,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,GAAG,MAAM;IAgB1F,aAAa,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,MAAM;IAIrE,oBAAoB,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAIzF,cAAc,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IAIrF,mBAAmB,CAAC,MAAM,EAAE,SAAS,iBAAiB,EAAE,GAAG,sBAAsB;IA6BjF,UAAU,IAAI,IAAI;IAuBlB,yBAAyB,IAAI,IAAI;IAmBjC,QAAQ,CAAC,WAAW,EAAE,gBAAgB,EAAE,WAAW,EAAE,OAAO,GAAG,gBAAgB;IAgB/E,2BAA2B,IAAI,OAAO;IAItC,OAAO,CAAC,QAAQ;IAgBhB,OAAO,CAAC,aAAa;IA2BrB,OAAO,CAAC,cAAc;IAUtB,OAAO,CAAC,WAAW;CAgBnB"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream-wrapper-shared.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper-shared.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,6BAA6B,EAAE,4BAA4B,EAAE,MAAM,8BAA8B,CAAC;AAC3G,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAY3E,SAAS,iBAAiB,CAAC,KAAsD;IAChF,OAAO,KAAK,EAAE,IAAI,KAAK,UAAU,IAAI,aAAa,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ,CAAC;AACtG,CAAC;AAED,SAAS,WAAW,CAAC,KAAY;IAChC,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;AACtE,CAAC;AAED,wEAAwE;AACxE,MAAM,OAAO,uBAAuB;IAYnC,YACC,MAAmC,EACnC,MAAwB,EACxB,OAAO,GAAmC,EAAE;QAZrC,0BAAqB,GAAkB,IAAI,CAAC;QACnC,+BAA0B,GAAG,IAAI,GAAG,EAAyB,CAAC;QAC9D,mCAA8B,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC3D,+BAA0B,GAAG,IAAI,GAAG,EAAkB,CAAC;QACvD,yBAAoB,GAAiC,EAAE,CAAC;QAUxE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,KAAK,IAAI,CAAC;QACtE,IAAI,CAAC,OAAO,GAAG,6BAA6B,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACjH,IAAI,CAAC,QAAQ,GAAG,IAAI,wBAAwB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACjG,CAAC;IAED,IAAI,CAAC,MAAwB;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,4BAA4B,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC5G,CAAC;IAED,gBAAgB,CAAC,UAAsC;QACtD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,4BAA4B,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACjH,CAAC;IAED,SAAS,CAAC,YAAoB,EAAE,WAAiD;QAChF,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QAC/C,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,MAAM,KAAK,GACV,WAAW,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;YAC1F,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACzD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1F,OAAO,UAAU,CAAC;IACnB,CAAC;IAED,aAAa,CAAC,YAAoB,EAAE,MAAwB;QAC3D,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,oBAAoB,CAAC,YAAoB,EAAE,KAAa,EAAE,MAAwB;QACjF,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,cAAc,CAAC,YAAoB,EAAE,OAAe,EAAE,MAAwB;QAC7E,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,mBAAmB,CAAC,MAAoC;QACvD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,MAAM,kBAAkB,GAAe,EAAE,CAAC;QAC1C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACpB,KAAK,MAAM;oBACV,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC1B,MAAM;gBACP,KAAK,gBAAgB;oBACpB,WAAW,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;oBAC1B,MAAM;gBACP,KAAK,gBAAgB;oBACpB,WAAW,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;oBACvD,MAAM;gBACP,KAAK,cAAc,EAAE,CAAC;oBACrB,WAAW,GAAG,IAAI,CAAC;oBACnB,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;oBAC1C,IAAI,SAAS;wBAAE,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBAClD,MAAM;gBACP,CAAC;gBACD;oBACC,WAAW,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACF,CAAC;QACD,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC;IAC5C,CAAC;IAED,UAAU;QACT,MAAM,iBAAiB,GACtB,IAAI,CAAC,qBAAqB,IAAI,IAAI;YACjC,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC;gBACjE,CAAC,IAAI,CAAC,sBAAsB;oBAC3B,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC;oBACrE,CAAC,CAAC,IAAI,CAAC;gBACR,IAAI,CAAC,CAAC;QACT,IAAI,iBAAiB,IAAI,IAAI,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACtD,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBAChB,IAAI,EAAE,UAAU;oBAChB,YAAY,EAAE,iBAAiB;oBAC/B,OAAO,EAAE,KAAK,CAAC,IAAI;oBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;iBACrB,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,yBAAyB;QACxB,KAAK,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpE,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC;gBAAE,SAAS;YACxC,MAAM,aAAa,GAAa;gBAC/B,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,UAAU,EAAE,IAAI;gBAChB,YAAY,EAAE,0CAA0C;aACxD,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;YACnD,KAAK,MAAM,CAAC,WAAW,EAAE,kBAAkB,CAAC,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBACjF,IAAI,kBAAkB,KAAK,YAAY;oBAAE,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC9F,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1G,CAAC;IACF,CAAC;IAED,QAAQ,CAAC,WAA6B,EAAE,WAAoB;QAC3D,MAAM,YAAY,GAAG,6BAA6B,CACjD,WAAW,EACX,IAAI,CAAC,OAAO,CAAC,OAAO,EACpB,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,sBAAsB,CAC3B,CAAC;QACF,IACC,CAAC,WAAW,IAAI,IAAI,CAAC,2BAA2B,EAAE,CAAC;YACnD,CAAC,YAAY,CAAC,UAAU,KAAK,MAAM,IAAI,YAAY,CAAC,UAAU,KAAK,QAAQ,CAAC,EAC3E,CAAC;YACF,YAAY,CAAC,UAAU,GAAG,SAAS,CAAC;QACrC,CAAC;QACD,OAAO,YAAY,CAAC;IACrB,CAAC;IAED,2BAA2B;QAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;IACrG,CAAC;IAEO,QAAQ,CAAC,IAAY;QAC5B,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI;YAAE,OAAO;QAC/C,IAAI,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,IAAI,CAAC;QAC3F,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;YAC1B,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YAC3C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;YAC9E,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QACnF,CAAC;aAAM,CAAC;YACP,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACjD,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM;gBAAE,OAAO;YACnC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5F,CAAC;IAEO,aAAa,CAAC,KAA6D;QAClF,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAClF,MAAM,KAAK,GAAG,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC9E,IACC,SAAS,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;gBAC7C,KAAK,EAAE,IAAI,KAAK,MAAM;gBACtB,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBACvB,KAAK,CAAC,aAAa,KAAK,SAAS,EAChC,CAAC;gBACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YAC5B,CAAC;QACF,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,QAAQ,GAAoB;YACjC,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,EAAE;YACb,WAAW,EAAE,EAAE;SACf,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;QAC/D,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI;YAAE,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;QAC9G,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IACnF,CAAC;IAEO,cAAc,CAAC,WAAmB,EAAE,KAAa;QACxD,MAAM,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACtE,IAAI,YAAY,IAAI,IAAI;YAAE,OAAO;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACjD,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC;YAAE,OAAO;QACtC,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC;QAC3B,KAAK,CAAC,SAAS,GAAG,kBAAkB,CAA0B,KAAK,CAAC,WAAW,CAAC,CAAC;QACjF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC1F,CAAC;IAEO,WAAW,CAAC,KAA2D;QAC9E,MAAM,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACtE,IAAI,YAAY,IAAI,IAAI;YAAE,OAAO,SAAS,CAAC;QAC3C,MAAM,aAAa,GAAa;YAC/B,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1D,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;SACjF,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;QACnD,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACzG,OAAO,aAAa,CAAC;IACtB,CAAC;CACD","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, TextContent, ToolCall } from \"../types.ts\";\nimport type { AssistantMessageDiagnostic } from \"../utils/diagnostics.ts\";\nimport { parseStreamingJson } from \"../utils/json-parse.ts\";\nimport { cloneAssistantMessageMetadata, syncAssistantMessageMetadata } from \"./stream-message-metadata.ts\";\nimport { StreamThinkingProjection } from \"./stream-thinking-projection.ts\";\nimport type { StreamParserEvent } from \"./types.ts\";\n\ntype PartialToolCall = ToolCall & { partialJson: string };\n\nexport type ParserProjectionResult = {\n\treadonly sawToolCall: boolean;\n\treadonly completedToolCalls: readonly ToolCall[];\n};\n\ntype StreamMessageProjectionOptions = { readonly preserveSourceMetadata?: boolean };\n\nfunction isPartialToolCall(block: AssistantMessage[\"content\"][number] | undefined): block is PartialToolCall {\n\treturn block?.type === \"toolCall\" && \"partialJson\" in block && typeof block.partialJson === \"string\";\n}\n\nfunction assertNever(value: never): never {\n\tthrow new Error(`Unexpected parser event: ${JSON.stringify(value)}`);\n}\n\n/** Projects parser events into the canonical outer assistant stream. */\nexport class StreamMessageProjection {\n\treadonly message: AssistantMessage;\n\tprivate readonly stream: AssistantMessageEventStream;\n\tprivate currentInnerTextIndex: number | null = null;\n\tprivate readonly textBlockIndexByInnerIndex = new Map<number, number | null>();\n\tprivate readonly lastTextBlockIndexByInnerIndex = new Map<number, number>();\n\tprivate readonly toolCallIndexByParserIndex = new Map<number, number>();\n\tprivate readonly projectedDiagnostics: AssistantMessageDiagnostic[] = [];\n\tprivate readonly thinking: StreamThinkingProjection;\n\tprivate readonly preserveSourceMetadata: boolean;\n\tprivate source: AssistantMessage;\n\n\tconstructor(\n\t\tstream: AssistantMessageEventStream,\n\t\tsource: AssistantMessage,\n\t\toptions: StreamMessageProjectionOptions = {},\n\t) {\n\t\tthis.stream = stream;\n\t\tthis.source = source;\n\t\tthis.preserveSourceMetadata = options.preserveSourceMetadata === true;\n\t\tthis.message = cloneAssistantMessageMetadata(source, [], this.projectedDiagnostics, this.preserveSourceMetadata);\n\t\tthis.thinking = new StreamThinkingProjection(stream, this.message, this.preserveSourceMetadata);\n\t}\n\n\tsync(source: AssistantMessage): void {\n\t\tthis.source = source;\n\t\tsyncAssistantMessageMetadata(this.message, source, this.projectedDiagnostics, this.preserveSourceMetadata);\n\t}\n\n\tappendDiagnostic(diagnostic: AssistantMessageDiagnostic): void {\n\t\tthis.projectedDiagnostics.push(diagnostic);\n\t\tsyncAssistantMessageMetadata(this.message, this.source, this.projectedDiagnostics, this.preserveSourceMetadata);\n\t}\n\n\tstartText(contentIndex: number, sourceBlock?: AssistantMessage[\"content\"][number]): number {\n\t\tthis.currentInnerTextIndex = contentIndex;\n\t\tconst outerIndex = this.message.content.length;\n\t\tif (this.preserveSourceMetadata) {\n\t\t\tconst block: TextContent =\n\t\t\t\tsourceBlock?.type === \"text\" ? { ...sourceBlock, text: \"\" } : { type: \"text\", text: \"\" };\n\t\t\tthis.message.content.push(block);\n\t\t\tthis.textBlockIndexByInnerIndex.set(contentIndex, outerIndex);\n\t\t\tthis.lastTextBlockIndexByInnerIndex.set(contentIndex, outerIndex);\n\t\t} else {\n\t\t\tthis.textBlockIndexByInnerIndex.set(contentIndex, null);\n\t\t}\n\t\tthis.stream.push({ type: \"text_start\", contentIndex: outerIndex, partial: this.message });\n\t\treturn outerIndex;\n\t}\n\n\tstartThinking(contentIndex: number, source: AssistantMessage): number {\n\t\treturn this.thinking.start(contentIndex, source);\n\t}\n\n\tprojectThinkingDelta(contentIndex: number, delta: string, source: AssistantMessage): void {\n\t\tthis.thinking.delta(contentIndex, delta, source);\n\t}\n\n\tfinishThinking(contentIndex: number, content: string, source: AssistantMessage): void {\n\t\tthis.thinking.end(contentIndex, content, source);\n\t}\n\n\tprojectParserEvents(events: readonly StreamParserEvent[]): ParserProjectionResult {\n\t\tlet sawToolCall = false;\n\t\tconst completedToolCalls: ToolCall[] = [];\n\t\tfor (const event of events) {\n\t\t\tswitch (event.type) {\n\t\t\t\tcase \"text\":\n\t\t\t\t\tthis.pushText(event.text);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_start\":\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tthis.startToolCall(event);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tthis.updateToolCall(event.index, event.argumentsDelta);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_end\": {\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tconst completed = this.endToolCall(event);\n\t\t\t\t\tif (completed) completedToolCalls.push(completed);\n\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t\tdefault:\n\t\t\t\t\tassertNever(event);\n\t\t\t}\n\t\t}\n\t\treturn { sawToolCall, completedToolCalls };\n\t}\n\n\tfinishText(): void {\n\t\tconst outerContentIndex =\n\t\t\tthis.currentInnerTextIndex == null\n\t\t\t\t? null\n\t\t\t\t: (this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ??\n\t\t\t\t\t(this.preserveSourceMetadata\n\t\t\t\t\t\t? this.lastTextBlockIndexByInnerIndex.get(this.currentInnerTextIndex)\n\t\t\t\t\t\t: null) ??\n\t\t\t\t\tnull);\n\t\tif (outerContentIndex != null) {\n\t\t\tconst block = this.message.content[outerContentIndex];\n\t\t\tif (block?.type === \"text\") {\n\t\t\t\tthis.stream.push({\n\t\t\t\t\ttype: \"text_end\",\n\t\t\t\t\tcontentIndex: outerContentIndex,\n\t\t\t\t\tcontent: block.text,\n\t\t\t\t\tpartial: this.message,\n\t\t\t\t});\n\t\t\t}\n\t\t}\n\t\tthis.currentInnerTextIndex = null;\n\t}\n\n\tfinalizeDanglingToolCalls(): void {\n\t\tfor (const [contentIndex, block] of this.message.content.entries()) {\n\t\t\tif (!isPartialToolCall(block)) continue;\n\t\t\tconst finalToolCall: ToolCall = {\n\t\t\t\ttype: \"toolCall\",\n\t\t\t\tid: block.id,\n\t\t\t\tname: block.name,\n\t\t\t\targuments: block.arguments,\n\t\t\t\tincomplete: true,\n\t\t\t\terrorMessage: \"Tool call stream ended before completion\",\n\t\t\t};\n\t\t\tthis.message.content[contentIndex] = finalToolCall;\n\t\t\tfor (const [parserIndex, mappedContentIndex] of this.toolCallIndexByParserIndex) {\n\t\t\t\tif (mappedContentIndex === contentIndex) this.toolCallIndexByParserIndex.delete(parserIndex);\n\t\t\t}\n\t\t\tthis.stream.push({ type: \"toolcall_end\", contentIndex, toolCall: finalToolCall, partial: this.message });\n\t\t}\n\t}\n\n\tfinalize(doneMessage: AssistantMessage, sawToolCall: boolean): AssistantMessage {\n\t\tconst finalMessage = cloneAssistantMessageMetadata(\n\t\t\tdoneMessage,\n\t\t\tthis.message.content,\n\t\t\tthis.projectedDiagnostics,\n\t\t\tthis.preserveSourceMetadata,\n\t\t);\n\t\tif (\n\t\t\t(sawToolCall || this.hasFinalizedToolCallContent()) &&\n\t\t\t(finalMessage.stopReason === \"stop\" || finalMessage.stopReason === \"length\")\n\t\t) {\n\t\t\tfinalMessage.stopReason = \"toolUse\";\n\t\t}\n\t\treturn finalMessage;\n\t}\n\n\thasFinalizedToolCallContent(): boolean {\n\t\treturn this.message.content.some((block) => block.type === \"toolCall\" && !isPartialToolCall(block));\n\t}\n\n\tprivate pushText(text: string): void {\n\t\tif (this.currentInnerTextIndex == null) return;\n\t\tlet contentIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ?? null;\n\t\tif (contentIndex == null) {\n\t\t\tcontentIndex = this.message.content.length;\n\t\t\tthis.message.content.push({ type: \"text\", text });\n\t\t\tthis.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);\n\t\t\tthis.lastTextBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);\n\t\t} else {\n\t\t\tconst block = this.message.content[contentIndex];\n\t\t\tif (block?.type !== \"text\") return;\n\t\t\tblock.text += text;\n\t\t}\n\t\tthis.stream.push({ type: \"text_delta\", contentIndex, delta: text, partial: this.message });\n\t}\n\n\tprivate startToolCall(event: Extract<StreamParserEvent, { type: \"toolcall_start\" }>): void {\n\t\tif (this.currentInnerTextIndex != null) {\n\t\t\tconst textIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex);\n\t\t\tconst block = textIndex == null ? undefined : this.message.content[textIndex];\n\t\t\tif (\n\t\t\t\ttextIndex === this.message.content.length - 1 &&\n\t\t\t\tblock?.type === \"text\" &&\n\t\t\t\tblock.text.length === 0 &&\n\t\t\t\tblock.textSignature === undefined\n\t\t\t) {\n\t\t\t\tthis.message.content.pop();\n\t\t\t}\n\t\t}\n\t\tconst contentIndex = this.message.content.length;\n\t\tconst toolCall: PartialToolCall = {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: event.id,\n\t\t\tname: event.name,\n\t\t\targuments: {},\n\t\t\tpartialJson: \"\",\n\t\t};\n\t\tthis.message.content.push(toolCall);\n\t\tthis.toolCallIndexByParserIndex.set(event.index, contentIndex);\n\t\tif (this.currentInnerTextIndex != null) this.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, null);\n\t\tthis.stream.push({ type: \"toolcall_start\", contentIndex, partial: this.message });\n\t}\n\n\tprivate updateToolCall(parserIndex: number, delta: string): void {\n\t\tconst contentIndex = this.toolCallIndexByParserIndex.get(parserIndex);\n\t\tif (contentIndex == null) return;\n\t\tconst block = this.message.content[contentIndex];\n\t\tif (!isPartialToolCall(block)) return;\n\t\tblock.partialJson += delta;\n\t\tblock.arguments = parseStreamingJson<Record<string, unknown>>(block.partialJson);\n\t\tthis.stream.push({ type: \"toolcall_delta\", contentIndex, delta, partial: this.message });\n\t}\n\n\tprivate endToolCall(event: Extract<StreamParserEvent, { type: \"toolcall_end\" }>): ToolCall | undefined {\n\t\tconst contentIndex = this.toolCallIndexByParserIndex.get(event.index);\n\t\tif (contentIndex == null) return undefined;\n\t\tconst finalToolCall: ToolCall = {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: event.id,\n\t\t\tname: event.name,\n\t\t\targuments: event.arguments,\n\t\t\t...(event.incomplete === true ? { incomplete: true } : {}),\n\t\t\t...(event.errorMessage === undefined ? {} : { errorMessage: event.errorMessage }),\n\t\t};\n\t\tthis.message.content[contentIndex] = finalToolCall;\n\t\tthis.toolCallIndexByParserIndex.delete(event.index);\n\t\tthis.stream.push({ type: \"toolcall_end\", contentIndex, toolCall: finalToolCall, partial: this.message });\n\t\treturn finalToolCall;\n\t}\n}\n"]}
|
|
1
|
+
{"version":3,"file":"stream-wrapper-shared.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper-shared.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,6BAA6B,EAAE,4BAA4B,EAAE,MAAM,8BAA8B,CAAC;AAC3G,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAY3E,SAAS,iBAAiB,CAAC,KAAsD;IAChF,OAAO,KAAK,EAAE,IAAI,KAAK,UAAU,IAAI,aAAa,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ,CAAC;AACtG,CAAC;AAED,SAAS,WAAW,CAAC,KAAY;IAChC,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;AACtE,CAAC;AAED,wEAAwE;AACxE,MAAM,OAAO,uBAAuB;IAYnC,YACC,MAAmC,EACnC,MAAwB,EACxB,UAA0C,EAAE;QAZrC,0BAAqB,GAAkB,IAAI,CAAC;QACnC,+BAA0B,GAAG,IAAI,GAAG,EAAyB,CAAC;QAC9D,mCAA8B,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC3D,+BAA0B,GAAG,IAAI,GAAG,EAAkB,CAAC;QACvD,yBAAoB,GAAiC,EAAE,CAAC;QAUxE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,KAAK,IAAI,CAAC;QACtE,IAAI,CAAC,OAAO,GAAG,6BAA6B,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACjH,IAAI,CAAC,QAAQ,GAAG,IAAI,wBAAwB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACjG,CAAC;IAED,IAAI,CAAC,MAAwB;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,4BAA4B,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC5G,CAAC;IAED,gBAAgB,CAAC,UAAsC;QACtD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,4BAA4B,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACjH,CAAC;IAED,SAAS,CAAC,YAAoB,EAAE,WAAiD;QAChF,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QAC/C,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,MAAM,KAAK,GACV,WAAW,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;YAC1F,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACzD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1F,OAAO,UAAU,CAAC;IACnB,CAAC;IAED,aAAa,CAAC,YAAoB,EAAE,MAAwB;QAC3D,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,oBAAoB,CAAC,YAAoB,EAAE,KAAa,EAAE,MAAwB;QACjF,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,cAAc,CAAC,YAAoB,EAAE,OAAe,EAAE,MAAwB;QAC7E,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,mBAAmB,CAAC,MAAoC;QACvD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,MAAM,kBAAkB,GAAe,EAAE,CAAC;QAC1C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACpB,KAAK,MAAM;oBACV,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC1B,MAAM;gBACP,KAAK,gBAAgB;oBACpB,WAAW,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;oBAC1B,MAAM;gBACP,KAAK,gBAAgB;oBACpB,WAAW,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;oBACvD,MAAM;gBACP,KAAK,cAAc,CAAC,CAAC,CAAC;oBACrB,WAAW,GAAG,IAAI,CAAC;oBACnB,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;oBAC1C,IAAI,SAAS;wBAAE,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBAClD,MAAM;gBACP,CAAC;gBACD;oBACC,WAAW,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACF,CAAC;QACD,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC;IAC5C,CAAC;IAED,UAAU;QACT,MAAM,iBAAiB,GACtB,IAAI,CAAC,qBAAqB,IAAI,IAAI;YACjC,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC;gBACjE,CAAC,IAAI,CAAC,sBAAsB;oBAC3B,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC;oBACrE,CAAC,CAAC,IAAI,CAAC;gBACR,IAAI,CAAC,CAAC;QACT,IAAI,iBAAiB,IAAI,IAAI,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACtD,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBAChB,IAAI,EAAE,UAAU;oBAChB,YAAY,EAAE,iBAAiB;oBAC/B,OAAO,EAAE,KAAK,CAAC,IAAI;oBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;iBACrB,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,yBAAyB;QACxB,KAAK,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpE,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC;gBAAE,SAAS;YACxC,MAAM,aAAa,GAAa;gBAC/B,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,UAAU,EAAE,IAAI;gBAChB,YAAY,EAAE,0CAA0C;aACxD,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;YACnD,KAAK,MAAM,CAAC,WAAW,EAAE,kBAAkB,CAAC,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBACjF,IAAI,kBAAkB,KAAK,YAAY;oBAAE,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC9F,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1G,CAAC;IACF,CAAC;IAED,QAAQ,CAAC,WAA6B,EAAE,WAAoB;QAC3D,MAAM,YAAY,GAAG,6BAA6B,CACjD,WAAW,EACX,IAAI,CAAC,OAAO,CAAC,OAAO,EACpB,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,sBAAsB,CAC3B,CAAC;QACF,IACC,CAAC,WAAW,IAAI,IAAI,CAAC,2BAA2B,EAAE,CAAC;YACnD,CAAC,YAAY,CAAC,UAAU,KAAK,MAAM,IAAI,YAAY,CAAC,UAAU,KAAK,QAAQ,CAAC,EAC3E,CAAC;YACF,YAAY,CAAC,UAAU,GAAG,SAAS,CAAC;QACrC,CAAC;QACD,OAAO,YAAY,CAAC;IACrB,CAAC;IAED,2BAA2B;QAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;IACrG,CAAC;IAEO,QAAQ,CAAC,IAAY;QAC5B,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI;YAAE,OAAO;QAC/C,IAAI,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,IAAI,CAAC;QAC3F,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;YAC1B,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YAC3C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;YAC9E,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QACnF,CAAC;aAAM,CAAC;YACP,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACjD,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM;gBAAE,OAAO;YACnC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5F,CAAC;IAEO,aAAa,CAAC,KAA6D;QAClF,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAClF,MAAM,KAAK,GAAG,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC9E,IACC,SAAS,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;gBAC7C,KAAK,EAAE,IAAI,KAAK,MAAM;gBACtB,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBACvB,KAAK,CAAC,aAAa,KAAK,SAAS,EAChC,CAAC;gBACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YAC5B,CAAC;QACF,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,QAAQ,GAAoB;YACjC,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,EAAE;YACb,WAAW,EAAE,EAAE;SACf,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;QAC/D,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI;YAAE,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;QAC9G,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IACnF,CAAC;IAEO,cAAc,CAAC,WAAmB,EAAE,KAAa;QACxD,MAAM,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACtE,IAAI,YAAY,IAAI,IAAI;YAAE,OAAO;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACjD,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC;YAAE,OAAO;QACtC,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC;QAC3B,KAAK,CAAC,SAAS,GAAG,kBAAkB,CAA0B,KAAK,CAAC,WAAW,CAAC,CAAC;QACjF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC1F,CAAC;IAEO,WAAW,CAAC,KAA2D;QAC9E,MAAM,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACtE,IAAI,YAAY,IAAI,IAAI;YAAE,OAAO,SAAS,CAAC;QAC3C,MAAM,aAAa,GAAa;YAC/B,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1D,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;SACjF,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;QACnD,IAAI,CAAC,0BAA0B,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACzG,OAAO,aAAa,CAAC;IACtB,CAAC;CACD","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, TextContent, ToolCall } from \"../types.ts\";\nimport type { AssistantMessageDiagnostic } from \"../utils/diagnostics.ts\";\nimport { parseStreamingJson } from \"../utils/json-parse.ts\";\nimport { cloneAssistantMessageMetadata, syncAssistantMessageMetadata } from \"./stream-message-metadata.ts\";\nimport { StreamThinkingProjection } from \"./stream-thinking-projection.ts\";\nimport type { StreamParserEvent } from \"./types.ts\";\n\ntype PartialToolCall = ToolCall & { partialJson: string };\n\nexport type ParserProjectionResult = {\n\treadonly sawToolCall: boolean;\n\treadonly completedToolCalls: readonly ToolCall[];\n};\n\ntype StreamMessageProjectionOptions = { readonly preserveSourceMetadata?: boolean };\n\nfunction isPartialToolCall(block: AssistantMessage[\"content\"][number] | undefined): block is PartialToolCall {\n\treturn block?.type === \"toolCall\" && \"partialJson\" in block && typeof block.partialJson === \"string\";\n}\n\nfunction assertNever(value: never): never {\n\tthrow new Error(`Unexpected parser event: ${JSON.stringify(value)}`);\n}\n\n/** Projects parser events into the canonical outer assistant stream. */\nexport class StreamMessageProjection {\n\treadonly message: AssistantMessage;\n\tprivate readonly stream: AssistantMessageEventStream;\n\tprivate currentInnerTextIndex: number | null = null;\n\tprivate readonly textBlockIndexByInnerIndex = new Map<number, number | null>();\n\tprivate readonly lastTextBlockIndexByInnerIndex = new Map<number, number>();\n\tprivate readonly toolCallIndexByParserIndex = new Map<number, number>();\n\tprivate readonly projectedDiagnostics: AssistantMessageDiagnostic[] = [];\n\tprivate readonly thinking: StreamThinkingProjection;\n\tprivate readonly preserveSourceMetadata: boolean;\n\tprivate source: AssistantMessage;\n\n\tconstructor(\n\t\tstream: AssistantMessageEventStream,\n\t\tsource: AssistantMessage,\n\t\toptions: StreamMessageProjectionOptions = {},\n\t) {\n\t\tthis.stream = stream;\n\t\tthis.source = source;\n\t\tthis.preserveSourceMetadata = options.preserveSourceMetadata === true;\n\t\tthis.message = cloneAssistantMessageMetadata(source, [], this.projectedDiagnostics, this.preserveSourceMetadata);\n\t\tthis.thinking = new StreamThinkingProjection(stream, this.message, this.preserveSourceMetadata);\n\t}\n\n\tsync(source: AssistantMessage): void {\n\t\tthis.source = source;\n\t\tsyncAssistantMessageMetadata(this.message, source, this.projectedDiagnostics, this.preserveSourceMetadata);\n\t}\n\n\tappendDiagnostic(diagnostic: AssistantMessageDiagnostic): void {\n\t\tthis.projectedDiagnostics.push(diagnostic);\n\t\tsyncAssistantMessageMetadata(this.message, this.source, this.projectedDiagnostics, this.preserveSourceMetadata);\n\t}\n\n\tstartText(contentIndex: number, sourceBlock?: AssistantMessage[\"content\"][number]): number {\n\t\tthis.currentInnerTextIndex = contentIndex;\n\t\tconst outerIndex = this.message.content.length;\n\t\tif (this.preserveSourceMetadata) {\n\t\t\tconst block: TextContent =\n\t\t\t\tsourceBlock?.type === \"text\" ? { ...sourceBlock, text: \"\" } : { type: \"text\", text: \"\" };\n\t\t\tthis.message.content.push(block);\n\t\t\tthis.textBlockIndexByInnerIndex.set(contentIndex, outerIndex);\n\t\t\tthis.lastTextBlockIndexByInnerIndex.set(contentIndex, outerIndex);\n\t\t} else {\n\t\t\tthis.textBlockIndexByInnerIndex.set(contentIndex, null);\n\t\t}\n\t\tthis.stream.push({ type: \"text_start\", contentIndex: outerIndex, partial: this.message });\n\t\treturn outerIndex;\n\t}\n\n\tstartThinking(contentIndex: number, source: AssistantMessage): number {\n\t\treturn this.thinking.start(contentIndex, source);\n\t}\n\n\tprojectThinkingDelta(contentIndex: number, delta: string, source: AssistantMessage): void {\n\t\tthis.thinking.delta(contentIndex, delta, source);\n\t}\n\n\tfinishThinking(contentIndex: number, content: string, source: AssistantMessage): void {\n\t\tthis.thinking.end(contentIndex, content, source);\n\t}\n\n\tprojectParserEvents(events: readonly StreamParserEvent[]): ParserProjectionResult {\n\t\tlet sawToolCall = false;\n\t\tconst completedToolCalls: ToolCall[] = [];\n\t\tfor (const event of events) {\n\t\t\tswitch (event.type) {\n\t\t\t\tcase \"text\":\n\t\t\t\t\tthis.pushText(event.text);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_start\":\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tthis.startToolCall(event);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tthis.updateToolCall(event.index, event.argumentsDelta);\n\t\t\t\t\tbreak;\n\t\t\t\tcase \"toolcall_end\": {\n\t\t\t\t\tsawToolCall = true;\n\t\t\t\t\tconst completed = this.endToolCall(event);\n\t\t\t\t\tif (completed) completedToolCalls.push(completed);\n\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t\tdefault:\n\t\t\t\t\tassertNever(event);\n\t\t\t}\n\t\t}\n\t\treturn { sawToolCall, completedToolCalls };\n\t}\n\n\tfinishText(): void {\n\t\tconst outerContentIndex =\n\t\t\tthis.currentInnerTextIndex == null\n\t\t\t\t? null\n\t\t\t\t: (this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ??\n\t\t\t\t\t(this.preserveSourceMetadata\n\t\t\t\t\t\t? this.lastTextBlockIndexByInnerIndex.get(this.currentInnerTextIndex)\n\t\t\t\t\t\t: null) ??\n\t\t\t\t\tnull);\n\t\tif (outerContentIndex != null) {\n\t\t\tconst block = this.message.content[outerContentIndex];\n\t\t\tif (block?.type === \"text\") {\n\t\t\t\tthis.stream.push({\n\t\t\t\t\ttype: \"text_end\",\n\t\t\t\t\tcontentIndex: outerContentIndex,\n\t\t\t\t\tcontent: block.text,\n\t\t\t\t\tpartial: this.message,\n\t\t\t\t});\n\t\t\t}\n\t\t}\n\t\tthis.currentInnerTextIndex = null;\n\t}\n\n\tfinalizeDanglingToolCalls(): void {\n\t\tfor (const [contentIndex, block] of this.message.content.entries()) {\n\t\t\tif (!isPartialToolCall(block)) continue;\n\t\t\tconst finalToolCall: ToolCall = {\n\t\t\t\ttype: \"toolCall\",\n\t\t\t\tid: block.id,\n\t\t\t\tname: block.name,\n\t\t\t\targuments: block.arguments,\n\t\t\t\tincomplete: true,\n\t\t\t\terrorMessage: \"Tool call stream ended before completion\",\n\t\t\t};\n\t\t\tthis.message.content[contentIndex] = finalToolCall;\n\t\t\tfor (const [parserIndex, mappedContentIndex] of this.toolCallIndexByParserIndex) {\n\t\t\t\tif (mappedContentIndex === contentIndex) this.toolCallIndexByParserIndex.delete(parserIndex);\n\t\t\t}\n\t\t\tthis.stream.push({ type: \"toolcall_end\", contentIndex, toolCall: finalToolCall, partial: this.message });\n\t\t}\n\t}\n\n\tfinalize(doneMessage: AssistantMessage, sawToolCall: boolean): AssistantMessage {\n\t\tconst finalMessage = cloneAssistantMessageMetadata(\n\t\t\tdoneMessage,\n\t\t\tthis.message.content,\n\t\t\tthis.projectedDiagnostics,\n\t\t\tthis.preserveSourceMetadata,\n\t\t);\n\t\tif (\n\t\t\t(sawToolCall || this.hasFinalizedToolCallContent()) &&\n\t\t\t(finalMessage.stopReason === \"stop\" || finalMessage.stopReason === \"length\")\n\t\t) {\n\t\t\tfinalMessage.stopReason = \"toolUse\";\n\t\t}\n\t\treturn finalMessage;\n\t}\n\n\thasFinalizedToolCallContent(): boolean {\n\t\treturn this.message.content.some((block) => block.type === \"toolCall\" && !isPartialToolCall(block));\n\t}\n\n\tprivate pushText(text: string): void {\n\t\tif (this.currentInnerTextIndex == null) return;\n\t\tlet contentIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex) ?? null;\n\t\tif (contentIndex == null) {\n\t\t\tcontentIndex = this.message.content.length;\n\t\t\tthis.message.content.push({ type: \"text\", text });\n\t\t\tthis.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);\n\t\t\tthis.lastTextBlockIndexByInnerIndex.set(this.currentInnerTextIndex, contentIndex);\n\t\t} else {\n\t\t\tconst block = this.message.content[contentIndex];\n\t\t\tif (block?.type !== \"text\") return;\n\t\t\tblock.text += text;\n\t\t}\n\t\tthis.stream.push({ type: \"text_delta\", contentIndex, delta: text, partial: this.message });\n\t}\n\n\tprivate startToolCall(event: Extract<StreamParserEvent, { type: \"toolcall_start\" }>): void {\n\t\tif (this.currentInnerTextIndex != null) {\n\t\t\tconst textIndex = this.textBlockIndexByInnerIndex.get(this.currentInnerTextIndex);\n\t\t\tconst block = textIndex == null ? undefined : this.message.content[textIndex];\n\t\t\tif (\n\t\t\t\ttextIndex === this.message.content.length - 1 &&\n\t\t\t\tblock?.type === \"text\" &&\n\t\t\t\tblock.text.length === 0 &&\n\t\t\t\tblock.textSignature === undefined\n\t\t\t) {\n\t\t\t\tthis.message.content.pop();\n\t\t\t}\n\t\t}\n\t\tconst contentIndex = this.message.content.length;\n\t\tconst toolCall: PartialToolCall = {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: event.id,\n\t\t\tname: event.name,\n\t\t\targuments: {},\n\t\t\tpartialJson: \"\",\n\t\t};\n\t\tthis.message.content.push(toolCall);\n\t\tthis.toolCallIndexByParserIndex.set(event.index, contentIndex);\n\t\tif (this.currentInnerTextIndex != null) this.textBlockIndexByInnerIndex.set(this.currentInnerTextIndex, null);\n\t\tthis.stream.push({ type: \"toolcall_start\", contentIndex, partial: this.message });\n\t}\n\n\tprivate updateToolCall(parserIndex: number, delta: string): void {\n\t\tconst contentIndex = this.toolCallIndexByParserIndex.get(parserIndex);\n\t\tif (contentIndex == null) return;\n\t\tconst block = this.message.content[contentIndex];\n\t\tif (!isPartialToolCall(block)) return;\n\t\tblock.partialJson += delta;\n\t\tblock.arguments = parseStreamingJson<Record<string, unknown>>(block.partialJson);\n\t\tthis.stream.push({ type: \"toolcall_delta\", contentIndex, delta, partial: this.message });\n\t}\n\n\tprivate endToolCall(event: Extract<StreamParserEvent, { type: \"toolcall_end\" }>): ToolCall | undefined {\n\t\tconst contentIndex = this.toolCallIndexByParserIndex.get(event.index);\n\t\tif (contentIndex == null) return undefined;\n\t\tconst finalToolCall: ToolCall = {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: event.id,\n\t\t\tname: event.name,\n\t\t\targuments: event.arguments,\n\t\t\t...(event.incomplete === true ? { incomplete: true } : {}),\n\t\t\t...(event.errorMessage === undefined ? {} : { errorMessage: event.errorMessage }),\n\t\t};\n\t\tthis.message.content[contentIndex] = finalToolCall;\n\t\tthis.toolCallIndexByParserIndex.delete(event.index);\n\t\tthis.stream.push({ type: \"toolcall_end\", contentIndex, toolCall: finalToolCall, partial: this.message });\n\t\treturn finalToolCall;\n\t}\n}\n"]}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream-wrapper.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,2BAA2B,IAAI,+BAA+B,EAAE,MAAM,0BAA0B,CAAC;AAC1G,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAGrE,MAAM,UAAU,gCAAgC,CAC/C,WAAwC,EACxC,QAA0B,EAC1B,KAAa;IAEb,MAAM,WAAW,GAAG,IAAI,+BAA+B,EAAE,CAAC;IAC1D,MAAM,MAAM,GAAG,QAAQ,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAElD,KAAK,CAAC,KAAK,IAAmB,EAAE;QAC/B,IAAI,UAAU,GAAmC,IAAI,CAAC;QACtD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAElC,MAAM,WAAW,GAAG,GAAS,EAAE;YAC9B,IAAI,CAAC,UAAU,IAAI,CAAC,qBAAqB;gBAAE,OAAO;YAClD,MAAM,MAAM,GAAG,UAAU,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/D,WAAW,GAAG,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;YAChD,UAAU,CAAC,UAAU,EAAE,CAAC;YACxB,qBAAqB,GAAG,KAAK,CAAC;QAC/B,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;gBACvC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACpB,KAAK,OAAO;wBACX,UAAU,GAAG,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBACrE,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;wBACjE,MAAM;oBACP,KAAK,YAAY;wBAChB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;wBACpF,MAAM;oBACP,KAAK,YAAY,EAAE,CAAC;wBACnB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;4BAAE,qBAAqB,GAAG,IAAI,CAAC;wBACzD,MAAM,MAAM,GAAG,UAAU,CAAC,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;wBACxE,WAAW,GAAG,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;wBAChD,MAAM;oBACP,CAAC;oBACD,KAAK,UAAU;wBACd,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,WAAW,EAAE,CAAC;wBACd,MAAM;oBACP,KAAK,gBAAgB;wBACpB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5D,MAAM;oBACP,KAAK,gBAAgB;wBACpB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAChF,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5E,MAAM;oBACP,KAAK,MAAM,EAAE,CAAC;wBACb,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBACvE,WAAW,EAAE,CAAC;wBACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;wBAChE,MAAM,SAAS,GAAG,WAAW,IAAI,UAAU,CAAC,2BAA2B,EAAE,CAAC;wBAC1E,MAAM,MAAM,GACX,SAAS,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;wBAChG,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;wBACpD,WAAW,CAAC,GAAG,EAAE,CAAC;wBAClB,OAAO;oBACR,CAAC;oBACD,KAAK,OAAO,EAAE,CAAC;wBACd,IAAI,CAAC,UAAU,EAAE,CAAC;4BACjB,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;4BAC7B,OAAO;wBACR,CAAC;wBACD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;wBAC7B,WAAW,EAAE,CAAC;wBACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;wBAC9D,IAAI,UAAU,CAAC,2BAA2B,EAAE,EAAE,CAAC;4BAC9C,MAAM,SAAS,GAAqB,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;4BAC1E,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;4BAC1E,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;4BAC3B,OAAO;wBACR,CAAC;wBACD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;wBAC1E,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACzB,OAAO;oBACR,CAAC;oBACD,KAAK,gBAAgB,CAAC;oBACtB,KAAK,gBAAgB,CAAC;oBACtB,KAAK,cAAc;wBAClB,MAAM;gBACR,CAAC;YACF,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;YAChD,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YACtE,WAAW,EAAE,CAAC;YACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;YACvC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;YACpF,UAAU,CAAC,yBAAyB,EAAE,CAAC;YACvC,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC;YACpC,QAAQ,CAAC,UAAU,GAAG,OAAO,CAAC;YAC9B,QAAQ,CAAC,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACvF,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YACtE,WAAW,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACpB,CAAC","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, Tool } from \"../types.ts\";\nimport { AssistantMessageEventStream as AssistantMessageEventStreamImpl } from \"../utils/event-stream.ts\";\nimport { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\nimport type { ToolCallProtocol } from \"./types.ts\";\n\nexport function wrapStreamWithToolCallMiddleware(\n\tinnerStream: AssistantMessageEventStream,\n\tprotocol: ToolCallProtocol,\n\ttools: Tool[],\n): AssistantMessageEventStream {\n\tconst outerStream = new AssistantMessageEventStreamImpl();\n\tconst parser = protocol.createStreamParser(tools);\n\n\tvoid (async (): Promise<void> => {\n\t\tlet projection: StreamMessageProjection | null = null;\n\t\tlet sawToolCall = false;\n\t\tlet parserHasPendingInput = false;\n\n\t\tconst flushParser = (): void => {\n\t\t\tif (!projection || !parserHasPendingInput) return;\n\t\t\tconst result = projection.projectParserEvents(parser.finish());\n\t\t\tsawToolCall = sawToolCall || result.sawToolCall;\n\t\t\tprojection.finishText();\n\t\t\tparserHasPendingInput = false;\n\t\t};\n\n\t\ttry {\n\t\t\tfor await (const event of innerStream) {\n\t\t\t\tswitch (event.type) {\n\t\t\t\t\tcase \"start\":\n\t\t\t\t\t\tprojection = new StreamMessageProjection(outerStream, event.partial);\n\t\t\t\t\t\touterStream.push({ type: \"start\", partial: projection.message });\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_start\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.startText(event.contentIndex, event.partial.content[event.contentIndex]);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_delta\": {\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tif (event.delta.length > 0) parserHasPendingInput = true;\n\t\t\t\t\t\tconst result = projection.projectParserEvents(parser.feed(event.delta));\n\t\t\t\t\t\tsawToolCall = sawToolCall || result.sawToolCall;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_end\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_start\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.startThinking(event.contentIndex, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_delta\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_end\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.finishThinking(event.contentIndex, event.content, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"done\": {\n\t\t\t\t\t\tprojection ??= new StreamMessageProjection(outerStream, event.message);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\t\t\t\tconst message = projection.finalize(event.message, sawToolCall);\n\t\t\t\t\t\tconst recovered = sawToolCall || projection.hasFinalizedToolCallContent();\n\t\t\t\t\t\tconst reason =\n\t\t\t\t\t\t\trecovered && (event.reason === \"stop\" || event.reason === \"length\") ? \"toolUse\" : event.reason;\n\t\t\t\t\t\touterStream.push({ type: \"done\", reason, message });\n\t\t\t\t\t\touterStream.end();\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"error\": {\n\t\t\t\t\t\tif (!projection) {\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end(event.error);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tprojection.sync(event.error);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\t\t\t\tconst message = projection.finalize(event.error, sawToolCall);\n\t\t\t\t\t\tif (projection.hasFinalizedToolCallContent()) {\n\t\t\t\t\t\t\tconst recovered: AssistantMessage = { ...message, stopReason: \"toolUse\" };\n\t\t\t\t\t\t\touterStream.push({ type: \"done\", reason: \"toolUse\", message: recovered });\n\t\t\t\t\t\t\touterStream.end(recovered);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\touterStream.push({ type: \"error\", reason: event.reason, error: message });\n\t\t\t\t\t\touterStream.end(message);\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"toolcall_start\":\n\t\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\tcase \"toolcall_end\":\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst innerMessage = await innerStream.result();\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, innerMessage);\n\t\t\tflushParser();\n\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\touterStream.end(projection.finalize(innerMessage, sawToolCall));\n\t\t} catch (error) {\n\t\t\tflushParser();\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, await innerStream.result());\n\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\tconst fallback = projection.message;\n\t\t\tfallback.stopReason = \"error\";\n\t\t\tfallback.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);\n\t\t\touterStream.push({ type: \"error\", reason: \"error\", error: fallback });\n\t\t\touterStream.end();\n\t\t}\n\t})();\n\n\treturn outerStream;\n}\n"]}
|
|
1
|
+
{"version":3,"file":"stream-wrapper.js","sourceRoot":"","sources":["../../src/tool-call-middleware/stream-wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,2BAA2B,IAAI,+BAA+B,EAAE,MAAM,0BAA0B,CAAC;AAC1G,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAGrE,MAAM,UAAU,gCAAgC,CAC/C,WAAwC,EACxC,QAA0B,EAC1B,KAAa;IAEb,MAAM,WAAW,GAAG,IAAI,+BAA+B,EAAE,CAAC;IAC1D,MAAM,MAAM,GAAG,QAAQ,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAElD,KAAK,CAAC,KAAK,IAAmB,EAAE;QAC/B,IAAI,UAAU,GAAmC,IAAI,CAAC;QACtD,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAElC,MAAM,WAAW,GAAG,GAAS,EAAE;YAC9B,IAAI,CAAC,UAAU,IAAI,CAAC,qBAAqB;gBAAE,OAAO;YAClD,MAAM,MAAM,GAAG,UAAU,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/D,WAAW,GAAG,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;YAChD,UAAU,CAAC,UAAU,EAAE,CAAC;YACxB,qBAAqB,GAAG,KAAK,CAAC;QAC/B,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;gBACvC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACpB,KAAK,OAAO;wBACX,UAAU,GAAG,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBACrE,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;wBACjE,MAAM;oBACP,KAAK,YAAY;wBAChB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;wBACpF,MAAM;oBACP,KAAK,YAAY,CAAC,CAAC,CAAC;wBACnB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;4BAAE,qBAAqB,GAAG,IAAI,CAAC;wBACzD,MAAM,MAAM,GAAG,UAAU,CAAC,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;wBACxE,WAAW,GAAG,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;wBAChD,MAAM;oBACP,CAAC;oBACD,KAAK,UAAU;wBACd,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,WAAW,EAAE,CAAC;wBACd,MAAM;oBACP,KAAK,gBAAgB;wBACpB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5D,MAAM;oBACP,KAAK,gBAAgB;wBACpB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAChF,MAAM;oBACP,KAAK,cAAc;wBAClB,IAAI,CAAC,UAAU;4BAAE,MAAM;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC/B,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBAC5E,MAAM;oBACP,KAAK,MAAM,CAAC,CAAC,CAAC;wBACb,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;wBACvE,WAAW,EAAE,CAAC;wBACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;wBAChE,MAAM,SAAS,GAAG,WAAW,IAAI,UAAU,CAAC,2BAA2B,EAAE,CAAC;wBAC1E,MAAM,MAAM,GACX,SAAS,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;wBAChG,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;wBACpD,WAAW,CAAC,GAAG,EAAE,CAAC;wBAClB,OAAO;oBACR,CAAC;oBACD,KAAK,OAAO,CAAC,CAAC,CAAC;wBACd,IAAI,CAAC,UAAU,EAAE,CAAC;4BACjB,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACxB,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;4BAC7B,OAAO;wBACR,CAAC;wBACD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;wBAC7B,WAAW,EAAE,CAAC;wBACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;wBAC9D,IAAI,UAAU,CAAC,2BAA2B,EAAE,EAAE,CAAC;4BAC9C,MAAM,SAAS,GAAqB,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;4BAC1E,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;4BAC1E,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;4BAC3B,OAAO;wBACR,CAAC;wBACD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;wBAC1E,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACzB,OAAO;oBACR,CAAC;oBACD,KAAK,gBAAgB,CAAC;oBACtB,KAAK,gBAAgB,CAAC;oBACtB,KAAK,cAAc;wBAClB,MAAM;gBACR,CAAC;YACF,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;YAChD,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YACtE,WAAW,EAAE,CAAC;YACd,UAAU,CAAC,yBAAyB,EAAE,CAAC;YACvC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,UAAU,KAAK,IAAI,uBAAuB,CAAC,WAAW,EAAE,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;YACpF,UAAU,CAAC,yBAAyB,EAAE,CAAC;YACvC,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC;YACpC,QAAQ,CAAC,UAAU,GAAG,OAAO,CAAC;YAC9B,QAAQ,CAAC,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACvF,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YACtE,WAAW,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACpB,CAAC","sourcesContent":["import type { AssistantMessage, AssistantMessageEventStream, Tool } from \"../types.ts\";\nimport { AssistantMessageEventStream as AssistantMessageEventStreamImpl } from \"../utils/event-stream.ts\";\nimport { StreamMessageProjection } from \"./stream-wrapper-shared.ts\";\nimport type { ToolCallProtocol } from \"./types.ts\";\n\nexport function wrapStreamWithToolCallMiddleware(\n\tinnerStream: AssistantMessageEventStream,\n\tprotocol: ToolCallProtocol,\n\ttools: Tool[],\n): AssistantMessageEventStream {\n\tconst outerStream = new AssistantMessageEventStreamImpl();\n\tconst parser = protocol.createStreamParser(tools);\n\n\tvoid (async (): Promise<void> => {\n\t\tlet projection: StreamMessageProjection | null = null;\n\t\tlet sawToolCall = false;\n\t\tlet parserHasPendingInput = false;\n\n\t\tconst flushParser = (): void => {\n\t\t\tif (!projection || !parserHasPendingInput) return;\n\t\t\tconst result = projection.projectParserEvents(parser.finish());\n\t\t\tsawToolCall = sawToolCall || result.sawToolCall;\n\t\t\tprojection.finishText();\n\t\t\tparserHasPendingInput = false;\n\t\t};\n\n\t\ttry {\n\t\t\tfor await (const event of innerStream) {\n\t\t\t\tswitch (event.type) {\n\t\t\t\t\tcase \"start\":\n\t\t\t\t\t\tprojection = new StreamMessageProjection(outerStream, event.partial);\n\t\t\t\t\t\touterStream.push({ type: \"start\", partial: projection.message });\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_start\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.startText(event.contentIndex, event.partial.content[event.contentIndex]);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"text_delta\": {\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tif (event.delta.length > 0) parserHasPendingInput = true;\n\t\t\t\t\t\tconst result = projection.projectParserEvents(parser.feed(event.delta));\n\t\t\t\t\t\tsawToolCall = sawToolCall || result.sawToolCall;\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"text_end\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_start\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.startThinking(event.contentIndex, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_delta\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.projectThinkingDelta(event.contentIndex, event.delta, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"thinking_end\":\n\t\t\t\t\t\tif (!projection) break;\n\t\t\t\t\t\tprojection.sync(event.partial);\n\t\t\t\t\t\tprojection.finishThinking(event.contentIndex, event.content, event.partial);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\tcase \"done\": {\n\t\t\t\t\t\tprojection ??= new StreamMessageProjection(outerStream, event.message);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\t\t\t\tconst message = projection.finalize(event.message, sawToolCall);\n\t\t\t\t\t\tconst recovered = sawToolCall || projection.hasFinalizedToolCallContent();\n\t\t\t\t\t\tconst reason =\n\t\t\t\t\t\t\trecovered && (event.reason === \"stop\" || event.reason === \"length\") ? \"toolUse\" : event.reason;\n\t\t\t\t\t\touterStream.push({ type: \"done\", reason, message });\n\t\t\t\t\t\touterStream.end();\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"error\": {\n\t\t\t\t\t\tif (!projection) {\n\t\t\t\t\t\t\touterStream.push(event);\n\t\t\t\t\t\t\touterStream.end(event.error);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tprojection.sync(event.error);\n\t\t\t\t\t\tflushParser();\n\t\t\t\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\t\t\t\tconst message = projection.finalize(event.error, sawToolCall);\n\t\t\t\t\t\tif (projection.hasFinalizedToolCallContent()) {\n\t\t\t\t\t\t\tconst recovered: AssistantMessage = { ...message, stopReason: \"toolUse\" };\n\t\t\t\t\t\t\touterStream.push({ type: \"done\", reason: \"toolUse\", message: recovered });\n\t\t\t\t\t\t\touterStream.end(recovered);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\touterStream.push({ type: \"error\", reason: event.reason, error: message });\n\t\t\t\t\t\touterStream.end(message);\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"toolcall_start\":\n\t\t\t\t\tcase \"toolcall_delta\":\n\t\t\t\t\tcase \"toolcall_end\":\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst innerMessage = await innerStream.result();\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, innerMessage);\n\t\t\tflushParser();\n\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\touterStream.end(projection.finalize(innerMessage, sawToolCall));\n\t\t} catch (error) {\n\t\t\tflushParser();\n\t\t\tprojection ??= new StreamMessageProjection(outerStream, await innerStream.result());\n\t\t\tprojection.finalizeDanglingToolCalls();\n\t\t\tconst fallback = projection.message;\n\t\t\tfallback.stopReason = \"error\";\n\t\t\tfallback.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);\n\t\t\touterStream.push({ type: \"error\", reason: \"error\", error: fallback });\n\t\t\touterStream.end();\n\t\t}\n\t})();\n\n\treturn outerStream;\n}\n"]}
|