@cloudflare/tanstack-ai 0.1.3 → 0.1.5
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 +36 -0
- package/dist/adapters/anthropic.cjs +3 -3
- package/dist/adapters/anthropic.js +2 -2
- package/dist/adapters/grok.cjs +3 -3
- package/dist/adapters/grok.js +2 -2
- package/dist/adapters/openai.cjs +3 -3
- package/dist/adapters/openai.js +2 -2
- package/dist/adapters/openrouter.cjs +3 -3
- package/dist/adapters/openrouter.js +2 -2
- package/dist/adapters/workers-ai-image.cjs +3 -3
- package/dist/adapters/workers-ai-image.js +2 -2
- package/dist/adapters/workers-ai-summarize.cjs +3 -3
- package/dist/adapters/workers-ai-summarize.js +2 -2
- package/dist/adapters/workers-ai-transcription.cjs +3 -3
- package/dist/adapters/workers-ai-transcription.js +2 -2
- package/dist/adapters/workers-ai-tts.cjs +3 -3
- package/dist/adapters/workers-ai-tts.js +2 -2
- package/dist/adapters/workers-ai.cjs +3 -3
- package/dist/adapters/workers-ai.js +2 -2
- package/dist/{chunk-CPOXZIMA.cjs → chunk-2YEF5YWC.cjs} +6 -6
- package/dist/{chunk-CPOXZIMA.cjs.map → chunk-2YEF5YWC.cjs.map} +1 -1
- package/dist/{chunk-53RO5I22.cjs → chunk-3L6IRVAV.cjs} +15 -12
- package/dist/chunk-3L6IRVAV.cjs.map +1 -0
- package/dist/{chunk-IL6PTKJE.cjs → chunk-5WAK4STT.cjs} +6 -6
- package/dist/{chunk-IL6PTKJE.cjs.map → chunk-5WAK4STT.cjs.map} +1 -1
- package/dist/{chunk-45DEJBFY.cjs → chunk-72MGG2FX.cjs} +6 -6
- package/dist/{chunk-45DEJBFY.cjs.map → chunk-72MGG2FX.cjs.map} +1 -1
- package/dist/{chunk-HFZK4C5B.js → chunk-73UI5P36.js} +2 -2
- package/dist/{chunk-QUS7INYA.cjs → chunk-CEQA4ZVD.cjs} +5 -5
- package/dist/{chunk-QUS7INYA.cjs.map → chunk-CEQA4ZVD.cjs.map} +1 -1
- package/dist/{chunk-MPVUETJY.cjs → chunk-GT3TZXZX.cjs} +3 -3
- package/dist/{chunk-MPVUETJY.cjs.map → chunk-GT3TZXZX.cjs.map} +1 -1
- package/dist/{chunk-Y2PUBTP5.cjs → chunk-HPMTYA35.cjs} +6 -6
- package/dist/{chunk-Y2PUBTP5.cjs.map → chunk-HPMTYA35.cjs.map} +1 -1
- package/dist/{chunk-DTGQYEJ2.cjs → chunk-IQAGFHRT.cjs} +7 -7
- package/dist/{chunk-DTGQYEJ2.cjs.map → chunk-IQAGFHRT.cjs.map} +1 -1
- package/dist/{chunk-RKBJLYVO.js → chunk-JDDH3EZD.js} +2 -2
- package/dist/{chunk-DQGO754X.cjs → chunk-JHZAPIW5.cjs} +5 -5
- package/dist/{chunk-DQGO754X.cjs.map → chunk-JHZAPIW5.cjs.map} +1 -1
- package/dist/{chunk-AJMCHPPR.js → chunk-JK4WXQVL.js} +2 -2
- package/dist/{chunk-HDDZEGLC.js → chunk-KQ3JSPII.js} +2 -2
- package/dist/{chunk-GST6AT3R.js → chunk-MBDHL2I4.js} +2 -2
- package/dist/{chunk-627RVSSB.js → chunk-MM5TKP3U.js} +2 -2
- package/dist/{chunk-727MRLUF.js → chunk-NGOJB274.js} +6 -3
- package/dist/chunk-NGOJB274.js.map +1 -0
- package/dist/{chunk-XCLZJGUI.js → chunk-OU57O7JK.js} +2 -2
- package/dist/{chunk-PMPGIUVA.js → chunk-RQLWAWHC.js} +2 -2
- package/dist/{chunk-4HOZDNW6.js → chunk-W24YURD6.js} +2 -2
- package/dist/{chunk-7YINQQCY.cjs → chunk-ZLPTIWRB.cjs} +3 -3
- package/dist/{chunk-7YINQQCY.cjs.map → chunk-ZLPTIWRB.cjs.map} +1 -1
- package/dist/index.cjs +11 -11
- package/dist/index.js +10 -10
- package/package.json +1 -1
- package/dist/chunk-53RO5I22.cjs.map +0 -1
- package/dist/chunk-727MRLUF.js.map +0 -1
- /package/dist/{chunk-HFZK4C5B.js.map → chunk-73UI5P36.js.map} +0 -0
- /package/dist/{chunk-RKBJLYVO.js.map → chunk-JDDH3EZD.js.map} +0 -0
- /package/dist/{chunk-AJMCHPPR.js.map → chunk-JK4WXQVL.js.map} +0 -0
- /package/dist/{chunk-HDDZEGLC.js.map → chunk-KQ3JSPII.js.map} +0 -0
- /package/dist/{chunk-GST6AT3R.js.map → chunk-MBDHL2I4.js.map} +0 -0
- /package/dist/{chunk-627RVSSB.js.map → chunk-MM5TKP3U.js.map} +0 -0
- /package/dist/{chunk-XCLZJGUI.js.map → chunk-OU57O7JK.js.map} +0 -0
- /package/dist/{chunk-PMPGIUVA.js.map → chunk-RQLWAWHC.js.map} +0 -0
- /package/dist/{chunk-4HOZDNW6.js.map → chunk-W24YURD6.js.map} +0 -0
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-5WAK4STT.cjs","../src/adapters/workers-ai-image.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACjBA,iDAAiC;AA6B1B,IAAM,sBAAA,EAAN,MAAA,QAAoC,2BAAsC;AAAA,EAIhF,WAAA,CAAY,MAAA,EAAgC,KAAA,EAA4B;AACvE,IAAA,KAAA,CAAM,CAAC,CAAA,EAAG,KAAK,CAAA;AAJhB,IAAA,6CAAA,IAAA,EAAS,MAAA,EAAO,kBAAA,CAAA;AAChB,IAAA,6CAAA,IAAA,EAAQ,eAAA,CAAA;AAIP,IAAA,uDAAA,MAA8B,CAAA;AAC9B,IAAA,IAAA,CAAK,cAAA,EAAgB,MAAA;AAAA,EACtB;AAAA,EAEA,MAAM,cAAA,CAAe,OAAA,EAAiE;AACrF,IAAA,MAAM,EAAE,MAAA,EAAQ,IAAA,EAAM,aAAa,EAAA,EAAI,OAAA;AACvC,IAAA,MAAM,MAAA,EAAiC,EAAE,GAAG,aAAa,CAAA;AAOzD,IAAA,GAAA,CAAI,IAAA,EAAM;AACT,MAAA,MAAM,CAAC,CAAA,EAAG,CAAC,EAAA,EAAI,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA;AAC7B,MAAA,GAAA,CAAI,CAAA,EAAG,KAAA,CAAM,MAAA,EAAQ,MAAA,CAAO,CAAC,CAAA;AAC7B,MAAA,GAAA,CAAI,CAAA,EAAG,KAAA,CAAM,OAAA,EAAS,MAAA,CAAO,CAAC,CAAA;AAAA,IAC/B;AAEA,IAAA,GAAA,CAAI,qDAAA,IAAsB,CAAK,aAAa,CAAA,EAAG;AAC9C,MAAA,OAAO,IAAA,CAAK,kBAAA,CAAmB,MAAA,EAAQ,KAAK,CAAA;AAAA,IAC7C;AAEA,IAAA,GAAA,CAAI,yDAAA,IAA0B,CAAK,aAAa,CAAA,EAAG;AAClD,MAAA,OAAO,IAAA,CAAK,eAAA,CAAgB,MAAA,EAAQ,KAAK,CAAA;AAAA,IAC1C;AAGA,IAAA,OAAO,IAAA,CAAK,kBAAA,CAAmB,MAAA,EAAQ,KAAK,CAAA;AAAA,EAC7C;AAAA,EAEA,MAAc,kBAAA,CACb,MAAA,EACA,OAAA,EACiC;AACjC,IAAA,MAAM,GAAA,EAAM,IAAA,CAAK,aAAA,CAA+C,OAAA;AAChE,IAAA,MAAM,OAAA,EAAS,MAAM,EAAA,CAAG,GAAA,CAAI,IAAA,CAAK,KAAA,EAAO,EAAE,MAAA,EAAQ,GAAG,QAAQ,CAAC,CAAA;AAE9D,IAAA,MAAM,IAAA,EAAM,MAAM,8CAAA,MAAe,EAAQ,OAAO,CAAA;AAChD,IAAA,OAAO,IAAA,CAAK,UAAA,CAAW,GAAG,CAAA;AAAA,EAC3B;AAAA,EAEA,MAAc,eAAA,CACb,MAAA,EACA,OAAA,EACiC;AACjC,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AACpB,IAAA,MAAM,SAAA,EAAW,MAAM,kDAAA;AAAA,MACtB,MAAA;AAAA,MACA,IAAA,CAAK,KAAA;AAAA,MACL,EAAE,MAAA,EAAQ,GAAG,QAAQ,CAAA;AAAA,MACrB,EAAE,KAAA,EAAO,kBAAA,EAAoB,MAAA,EAAS,OAAA,CAAqC,OAAO;AAAA,IACnF,CAAA;AAEA,IAAA,MAAM,OAAA,EAAS,MAAM,QAAA,CAAS,WAAA,CAAY,CAAA;AAC1C,IAAA,OAAO,IAAA,CAAK,UAAA,CAAW,kDAAA,IAAuB,UAAA,CAAW,MAAM,CAAC,CAAC,CAAA;AAAA,EAClE;AAAA,EAEA,MAAc,kBAAA,CACb,MAAA,EACA,OAAA,EACiC;AACjC,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,aAAA;AAC3B,IAAA,MAAM,aAAA,EAAe,kDAAA,YAAmB,EAAc,aAAa,CAAA;AAKnE,IAAA,MAAM,SAAA,EAAW,MAAM,YAAA,CAAa,kDAAA,EAAoD;AAAA,MACvF,MAAA,EAAQ,MAAA;AAAA,MACR,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU;AAAA,QACpB,KAAA,EAAO,IAAA,CAAK,KAAA;AAAA,QACZ,MAAA;AAAA,QACA,GAAG;AAAA,MACJ,CAAC;AAAA,IACF,CAAC,CAAA;AAED,IAAA,GAAA,CAAI,CAAC,QAAA,CAAS,EAAA,EAAI;AACjB,MAAA,MAAM,UAAA,EAAY,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AACtC,MAAA,MAAM,IAAI,KAAA;AAAA,QACT,CAAA,yCAAA,EAA4C,QAAA,CAAS,MAAM,CAAA,GAAA,EAAM,SAAS,CAAA;AAAA,MAAA;AAC3E,IAAA;AAGD,IAAA;AACA,IAAA;AAAiE,EAAA;AAClE;AAAA,EAAA;AAIC,IAAA;AAAO,MAAA;AACc,MAAA;AACR,MAAA;AACa,IAAA;AAC1B,EAAA;AAEF;AAyBO;AACN,EAAA;AACD;ADlEA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-5WAK4STT.cjs","sourcesContent":[null,"import { BaseImageAdapter } from \"@tanstack/ai/adapters\";\nimport type { ImageGenerationOptions, ImageGenerationResult } from \"@tanstack/ai\";\nimport type { AiModels, BaseAiTextToImage } from \"@cloudflare/workers-types\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch } from \"../utils/workers-ai-rest\";\nimport { binaryToBase64, uint8ArrayToBase64 } from \"../utils/binary\";\nimport type { WorkersAiDirectCredentialsConfig } from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model type derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiImageModel =\n\t| { [K in keyof AiModels]: AiModels[K] extends BaseAiTextToImage ? K : never }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiImageAdapter: image generation via Workers AI\n// Extends BaseImageAdapter so it works with TanStack AI's generateImage()\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiImageAdapter extends BaseImageAdapter<WorkersAiImageModel> {\n\treadonly name = \"workers-ai-image\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiImageModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync generateImages(options: ImageGenerationOptions): Promise<ImageGenerationResult> {\n\t\tconst { prompt, size, modelOptions } = options;\n\t\tconst extra: Record<string, unknown> = { ...modelOptions };\n\n\t\t// Note: Workers AI Stable Diffusion models only generate a single image\n\t\t// per request — there is no multi-image parameter. `numberOfImages` from\n\t\t// the TanStack AI options is intentionally not forwarded. Use\n\t\t// `modelOptions.num_steps` to control the number of diffusion steps.\n\n\t\tif (size) {\n\t\t\tconst [w, h] = size.split(\"x\");\n\t\t\tif (w) extra.width = Number(w);\n\t\t\tif (h) extra.height = Number(h);\n\t\t}\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaBinding(prompt, extra);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaRest(prompt, extra);\n\t\t}\n\n\t\t// Gateway mode\n\t\treturn this.generateViaGateway(prompt, extra);\n\t}\n\n\tprivate async generateViaBinding(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = await ai.run(this.model, { prompt, ...options });\n\n\t\tconst b64 = await binaryToBase64(result, \"image\");\n\t\treturn this.wrapResult(b64);\n\t}\n\n\tprivate async generateViaRest(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst response = await workersAiRestFetch(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\t{ prompt, ...options },\n\t\t\t{ label: \"Workers AI image\", signal: (options as { signal?: AbortSignal }).signal },\n\t\t);\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\tprivate async generateViaGateway(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/images/generations\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\tprompt,\n\t\t\t\t...options,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI image gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\t/** Wrap a base64 image string into the standard ImageGenerationResult. */\n\tprivate wrapResult(b64: string): ImageGenerationResult {\n\t\treturn {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\timages: [{ b64Json: b64 }],\n\t\t};\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI image generation adapter.\n *\n * Works with TanStack AI's `generateImage()` activity function:\n * ```ts\n * import { generateImage } from \"@tanstack/ai\";\n * import { createWorkersAiImage } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiImage(\n * \"@cf/stabilityai/stable-diffusion-xl-base-1.0\",\n * { binding: env.AI },\n * );\n *\n * const result = await generateImage({ adapter, prompt: \"a cat in space\" });\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiImage(model: WorkersAiImageModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiImageAdapter(config, model);\n}\n"]}
|
|
@@ -10,7 +10,7 @@ var _chunk7HSUHP63cjs = require('./chunk-7HSUHP63.cjs');
|
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
|
|
13
|
-
var
|
|
13
|
+
var _chunk3L6IRVAVcjs = require('./chunk-3L6IRVAV.cjs');
|
|
14
14
|
|
|
15
15
|
|
|
16
16
|
var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
|
|
@@ -22,7 +22,7 @@ var WorkersAiTranscriptionAdapter = class extends _adapters.BaseTranscriptionAda
|
|
|
22
22
|
super({}, model);
|
|
23
23
|
_chunk4DE2IREAcjs.__publicField.call(void 0, this, "name", "workers-ai-transcription");
|
|
24
24
|
_chunk4DE2IREAcjs.__publicField.call(void 0, this, "adapterConfig");
|
|
25
|
-
|
|
25
|
+
_chunk3L6IRVAVcjs.validateWorkersAiConfig.call(void 0, config);
|
|
26
26
|
this.adapterConfig = config;
|
|
27
27
|
}
|
|
28
28
|
async transcribe(options) {
|
|
@@ -32,10 +32,10 @@ var WorkersAiTranscriptionAdapter = class extends _adapters.BaseTranscriptionAda
|
|
|
32
32
|
if (language) extra.language = language;
|
|
33
33
|
if (prompt) extra.initial_prompt = prompt;
|
|
34
34
|
const audioPayload = this.buildAudioPayload(audioBytes, audio);
|
|
35
|
-
if (
|
|
35
|
+
if (_chunk3L6IRVAVcjs.isDirectBindingConfig.call(void 0, this.adapterConfig)) {
|
|
36
36
|
return this.transcribeViaBinding(audioPayload, extra);
|
|
37
37
|
}
|
|
38
|
-
if (
|
|
38
|
+
if (_chunk3L6IRVAVcjs.isDirectCredentialsConfig.call(void 0, this.adapterConfig)) {
|
|
39
39
|
if (this.model === "@cf/deepgram/nova-3") {
|
|
40
40
|
return this.transcribeViaRestBinary(audioBytes, audio, extra);
|
|
41
41
|
}
|
|
@@ -106,7 +106,7 @@ var WorkersAiTranscriptionAdapter = class extends _adapters.BaseTranscriptionAda
|
|
|
106
106
|
}
|
|
107
107
|
async transcribeViaGateway(audioPayload, options) {
|
|
108
108
|
const gatewayConfig = this.adapterConfig;
|
|
109
|
-
const gatewayFetch =
|
|
109
|
+
const gatewayFetch = _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "workers-ai", gatewayConfig);
|
|
110
110
|
const response = await gatewayFetch("https://api.cloudflare.com/v1/audio/transcriptions", {
|
|
111
111
|
method: "POST",
|
|
112
112
|
body: JSON.stringify({
|
|
@@ -215,4 +215,4 @@ function detectAudioContentType(audio) {
|
|
|
215
215
|
|
|
216
216
|
|
|
217
217
|
exports.WorkersAiTranscriptionAdapter = WorkersAiTranscriptionAdapter; exports.createWorkersAiTranscription = createWorkersAiTranscription;
|
|
218
|
-
//# sourceMappingURL=chunk-
|
|
218
|
+
//# sourceMappingURL=chunk-72MGG2FX.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-45DEJBFY.cjs","../src/adapters/workers-ai-transcription.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACjBA,iDAAyC;AAyClC,IAAM,8BAAA,EAAN,MAAA,QAA4C,mCAAsD;AAAA,EAIxG,WAAA,CAAY,MAAA,EAAgC,KAAA,EAAoC;AAC/E,IAAA,KAAA,CAAM,CAAC,CAAA,EAAG,KAAK,CAAA;AAJhB,IAAA,6CAAA,IAAA,EAAS,MAAA,EAAO,0BAAA,CAAA;AAChB,IAAA,6CAAA,IAAA,EAAQ,eAAA,CAAA;AAIP,IAAA,uDAAA,MAA8B,CAAA;AAC9B,IAAA,IAAA,CAAK,cAAA,EAAgB,MAAA;AAAA,EACtB;AAAA,EAEA,MAAM,UAAA,CAAW,OAAA,EAA6D;AAC7E,IAAA,MAAM,EAAE,KAAA,EAAO,QAAA,EAAU,MAAA,EAAQ,aAAa,EAAA,EAAI,OAAA;AAGlD,IAAA,MAAM,WAAA,EAAa,MAAM,qBAAA,CAAsB,KAAK,CAAA;AAEpD,IAAA,MAAM,MAAA,EAAiC,EAAE,GAAG,aAAa,CAAA;AACzD,IAAA,GAAA,CAAI,QAAA,EAAU,KAAA,CAAM,SAAA,EAAW,QAAA;AAC/B,IAAA,GAAA,CAAI,MAAA,EAAQ,KAAA,CAAM,eAAA,EAAiB,MAAA;AAOnC,IAAA,MAAM,aAAA,EAAe,IAAA,CAAK,iBAAA,CAAkB,UAAA,EAAY,KAAK,CAAA;AAE7D,IAAA,GAAA,CAAI,qDAAA,IAAsB,CAAK,aAAa,CAAA,EAAG;AAC9C,MAAA,OAAO,IAAA,CAAK,oBAAA,CAAqB,YAAA,EAAc,KAAK,CAAA;AAAA,IACrD;AAEA,IAAA,GAAA,CAAI,yDAAA,IAA0B,CAAK,aAAa,CAAA,EAAG;AAElD,MAAA,GAAA,CAAI,IAAA,CAAK,MAAA,IAAU,qBAAA,EAAuB;AACzC,QAAA,OAAO,IAAA,CAAK,uBAAA,CAAwB,UAAA,EAAY,KAAA,EAAO,KAAK,CAAA;AAAA,MAC7D;AACA,MAAA,OAAO,IAAA,CAAK,iBAAA,CAAkB,YAAA,EAAc,KAAK,CAAA;AAAA,IAClD;AAEA,IAAA,OAAO,IAAA,CAAK,oBAAA,CAAqB,YAAA,EAAc,KAAK,CAAA;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASQ,iBAAA,CACP,UAAA,EACA,aAAA,EAC0B;AAC1B,IAAA,GAAA,CAAI,IAAA,CAAK,MAAA,IAAU,qBAAA,EAAuB;AACzC,MAAA,MAAM,IAAA,EAAM,kDAAA,IAAuB,UAAA,CAAW,UAAU,CAAC,CAAA;AACzD,MAAA,MAAM,YAAA,EAAc,sBAAA,CAAuB,aAAa,CAAA;AACxD,MAAA,OAAO,EAAE,KAAA,EAAO,EAAE,IAAA,EAAM,GAAA,EAAK,YAAY,EAAE,CAAA;AAAA,IAC5C;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK,MAAA,IAAU,mCAAA,EAAqC;AACvD,MAAA,OAAO,EAAE,KAAA,EAAO,kDAAA,IAAuB,UAAA,CAAW,UAAU,CAAC,EAAE,CAAA;AAAA,IAChE;AAEA,IAAA,OAAO,EAAE,KAAA,EAAO,WAAW,CAAA;AAAA,EAC5B;AAAA,EAEA,MAAc,oBAAA,CACb,YAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,GAAA,EAAM,IAAA,CAAK,aAAA,CAA+C,OAAA;AAChE,IAAA,MAAM,OAAA,EAAU,MAAM,EAAA,CAAG,GAAA,CAAI,IAAA,CAAK,KAAA,EAAO;AAAA,MACxC,GAAG,YAAA;AAAA,MACH,GAAG;AAAA,IACJ,CAAC,CAAA;AACD,IAAA,OAAO,IAAA,CAAK,eAAA,CAAgB,MAAM,CAAA;AAAA,EACnC;AAAA,EAEA,MAAc,iBAAA,CACb,YAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AAEpB,IAAA,MAAM,SAAA,EAAW,MAAM,kDAAA;AAAA,MACtB,MAAA;AAAA,MACA,IAAA,CAAK,KAAA;AAAA,MACL,EAAE,GAAG,YAAA,EAAc,GAAG,QAAQ,CAAA;AAAA,MAC9B;AAAA,QACC,KAAA,EAAO,0BAAA;AAAA,QACP,MAAA,EAAS,OAAA,CAAqC;AAAA,MAC/C;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,KAAA,EAAQ,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AAMlC,IAAA,OAAO,IAAA,CAAK,eAAA,kBAAgB,IAAA,CAAK,MAAA,UAAU,MAAI,CAAA;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAc,uBAAA,CACb,UAAA,EACA,aAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AACpB,IAAA,MAAM,YAAA,EAAc,sBAAA,CAAuB,aAAa,CAAA;AAExD,IAAA,MAAM,SAAA,EAAW,MAAM,wDAAA;AAAA,MACtB,MAAA;AAAA,MACA,IAAA,CAAK,KAAA;AAAA,MACL,IAAI,UAAA,CAAW,UAAU,CAAA;AAAA,MACzB,WAAA;AAAA,MACA;AAAA,QACC,KAAA,EAAO,0BAAA;AAAA,QACP,MAAA,EAAS,OAAA,CAAqC;AAAA,MAC/C;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,KAAA,EAAQ,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AAIlC,IAAA,OAAO,IAAA,CAAK,eAAA,kBAAgB,IAAA,CAAK,MAAA,UAAU,MAAI,CAAA;AAAA,EAChD;AAAA,EAEA,MAAc,oBAAA,CACb,YAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,aAAA;AAC3B,IAAA,MAAM,aAAA,EAAe,kDAAA,YAAmB,EAAc,aAAa,CAAA;AAKnE,IAAA,MAAM,SAAA,EAAW,MAAM,YAAA,CAAa,oDAAA,EAAsD;AAAA,MACzF,MAAA,EAAQ,MAAA;AAAA,MACR,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU;AAAA,QACpB,KAAA,EAAO,IAAA,CAAK,KAAA;AAAA,QACZ,GAAG,YAAA;AAAA,QACH,GAAG;AAAA,MACJ,CAAC;AAAA,IACF,CAAC,CAAA;AAED,IAAA,GAAA,CAAI,CAAC,QAAA,CAAS,EAAA,EAAI;AACjB,MAAA,MAAM,UAAA,EAAY,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AACtC,MAAA,MAAM,IAAI,KAAA;AAAA,QACT,CAAA,iDAAA,EAAoD,QAAA,CAAS,MAAM,CAAA,GAAA,EAAM,SAAS,CAAA;AAAA,MAAA;AACnF,IAAA;AAGD,IAAA;AACA,IAAA;AAAgC,EAAA;AACjC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAaC,IAAA;AACA,IAAA;AACC,MAAA;AAOA,MAAA;AACA,MAAA;AACA,MAAA;AAAoC,QAAA;AACf,QAAA;AACR,QAAA;AACZ,MAAA;AAED,MAAA;AACC,QAAA;AAAqC,UAAA;AACpB,UAAA;AACE,UAAA;AACJ,QAAA;AACb,MAAA;AAEH,MAAA;AAAO,IAAA;AAKR,IAAA;AAAoC,MAAA;AACf,MAAA;AACR,MAAA;AACkB,IAAA;AAI/B,IAAA;AACA,IAAA;AACC,MAAA;AAAoC,IAAA;AAIrC,IAAA;AACC,MAAA;AAAoC,IAAA;AAIrC,IAAA;AACC,MAAA;AAAmF,QAAA;AAC9E,QAAA;AAC0B,QAAA;AACE,QAAA;AACJ,MAAA;AAC3B,IAAA;AAIH,IAAA;AACC,MAAA;AAA8D,QAAA;AACjC,QAAA;AACE,QAAA;AACJ,MAAA;AACzB,IAAA;AAGH,IAAA;AAAO,EAAA;AAET;AA0BO;AAIN,EAAA;AACD;AAaA;AACC,EAAA;AACC,IAAA;AAAuC,EAAA;AAGxC,EAAA;AAEC,IAAA;AACA,IAAA;AAAwC,EAAA;AAGzC,EAAA;AAEC,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AAA8B,IAAA;AAE/B,IAAA;AAAuB,EAAA;AAGxB,EAAA;AACD;AASA;AAEC,EAAA;AACC,IAAA;AAAa,EAAA;AAKd,EAAA;AACD;AD1JA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-45DEJBFY.cjs","sourcesContent":[null,"import { BaseTranscriptionAdapter } from \"@tanstack/ai/adapters\";\nimport type { TranscriptionOptions, TranscriptionResult } from \"@tanstack/ai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype WorkersAiDirectCredentialsConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch, workersAiRestFetchBinary } from \"../utils/workers-ai-rest\";\nimport { uint8ArrayToBase64 } from \"../utils/binary\";\n\n// ---------------------------------------------------------------------------\n// Model types\n// ---------------------------------------------------------------------------\n\n/**\n * Workers AI models that support speech-to-text transcription.\n *\n * Note: the typed `AiModels` interface in `@cloudflare/workers-types` may lag\n * behind what's deployed. We use a string union here that matches the known\n * models including Deepgram partner models.\n *\n * **Nova-3 note:** `@cf/deepgram/nova-3` uses a different input format than the\n * Whisper models. Via binding it accepts `{ audio: { body: base64, contentType } }`.\n * Via REST it requires multipart form data (not JSON). The adapter handles both.\n */\nexport type WorkersAiTranscriptionModel =\n\t| \"@cf/openai/whisper\"\n\t| \"@cf/openai/whisper-tiny-en\"\n\t| \"@cf/openai/whisper-large-v3-turbo\"\n\t| \"@cf/deepgram/nova-3\"\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiTranscriptionAdapter\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiTranscriptionAdapter extends BaseTranscriptionAdapter<WorkersAiTranscriptionModel> {\n\treadonly name = \"workers-ai-transcription\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiTranscriptionModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync transcribe(options: TranscriptionOptions): Promise<TranscriptionResult> {\n\t\tconst { audio, language, prompt, modelOptions } = options;\n\n\t\t// Normalize audio to raw bytes\n\t\tconst audioBytes = await normalizeAudioToBytes(audio);\n\n\t\tconst extra: Record<string, unknown> = { ...modelOptions };\n\t\tif (language) extra.language = language;\n\t\tif (prompt) extra.initial_prompt = prompt;\n\n\t\t// Build the model-specific audio payload:\n\t\t// - Deepgram Nova-3 (binding): { audio: { body: base64, contentType: \"audio/...\" } }\n\t\t// - Deepgram Nova-3 (REST): multipart FormData (handled separately)\n\t\t// - Whisper Large v3 Turbo (REST/gateway): { audio: base64string }\n\t\t// - Other Whisper models (binding): { audio: number[] }\n\t\tconst audioPayload = this.buildAudioPayload(audioBytes, audio);\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.transcribeViaBinding(audioPayload, extra);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\t// Nova-3 REST requires raw binary audio, not JSON\n\t\t\tif (this.model === \"@cf/deepgram/nova-3\") {\n\t\t\t\treturn this.transcribeViaRestBinary(audioBytes, audio, extra);\n\t\t\t}\n\t\t\treturn this.transcribeViaRest(audioPayload, extra);\n\t\t}\n\n\t\treturn this.transcribeViaGateway(audioPayload, extra);\n\t}\n\n\t/**\n\t * Build the audio field for the request payload, handling model-specific formats.\n\t *\n\t * - `@cf/deepgram/nova-3` requires `{ body: base64, contentType: \"audio/...\" }`\n\t * - `@cf/openai/whisper-large-v3-turbo` REST/gateway accepts a base64 string\n\t * - Other Whisper models accept `number[]` (binding) or base64 (REST)\n\t */\n\tprivate buildAudioPayload(\n\t\taudioBytes: number[],\n\t\toriginalAudio: string | File | Blob | ArrayBuffer,\n\t): Record<string, unknown> {\n\t\tif (this.model === \"@cf/deepgram/nova-3\") {\n\t\t\tconst b64 = uint8ArrayToBase64(new Uint8Array(audioBytes));\n\t\t\tconst contentType = detectAudioContentType(originalAudio);\n\t\t\treturn { audio: { body: b64, contentType } };\n\t\t}\n\n\t\tif (this.model === \"@cf/openai/whisper-large-v3-turbo\") {\n\t\t\treturn { audio: uint8ArrayToBase64(new Uint8Array(audioBytes)) };\n\t\t}\n\n\t\treturn { audio: audioBytes };\n\t}\n\n\tprivate async transcribeViaBinding(\n\t\taudioPayload: Record<string, unknown>,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = (await ai.run(this.model, {\n\t\t\t...audioPayload,\n\t\t\t...options,\n\t\t})) as Record<string, unknown>;\n\t\treturn this.normalizeResult(result);\n\t}\n\n\tprivate async transcribeViaRest(\n\t\taudioPayload: Record<string, unknown>,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\n\t\tconst response = await workersAiRestFetch(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\t{ ...audioPayload, ...options },\n\t\t\t{\n\t\t\t\tlabel: \"Workers AI transcription\",\n\t\t\t\tsignal: (options as { signal?: AbortSignal }).signal,\n\t\t\t},\n\t\t);\n\n\t\tconst data = (await response.json()) as {\n\t\t\tresult?: Record<string, unknown>;\n\t\t} & Record<string, unknown>;\n\n\t\t// Cloudflare REST API wraps responses in { success, result: {...} }.\n\t\t// Use `data.result` when present, fall back to `data` for direct responses.\n\t\treturn this.normalizeResult(data.result ?? data);\n\t}\n\n\t/**\n\t * Transcribe via REST using raw binary audio.\n\t * Required for models like Deepgram Nova-3 that expect raw audio bytes\n\t * with a Content-Type header (e.g. \"audio/wav\") instead of JSON.\n\t */\n\tprivate async transcribeViaRestBinary(\n\t\taudioBytes: number[],\n\t\toriginalAudio: string | File | Blob | ArrayBuffer,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst contentType = detectAudioContentType(originalAudio);\n\n\t\tconst response = await workersAiRestFetchBinary(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\tnew Uint8Array(audioBytes),\n\t\t\tcontentType,\n\t\t\t{\n\t\t\t\tlabel: \"Workers AI transcription\",\n\t\t\t\tsignal: (options as { signal?: AbortSignal }).signal,\n\t\t\t},\n\t\t);\n\n\t\tconst data = (await response.json()) as {\n\t\t\tresult?: Record<string, unknown>;\n\t\t} & Record<string, unknown>;\n\n\t\treturn this.normalizeResult(data.result ?? data);\n\t}\n\n\tprivate async transcribeViaGateway(\n\t\taudioPayload: Record<string, unknown>,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/audio/transcriptions\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\t...audioPayload,\n\t\t\t\t...options,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI transcription gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst data = (await response.json()) as Record<string, unknown>;\n\t\treturn this.normalizeResult(data);\n\t}\n\n\t/**\n\t * Normalize Workers AI transcription results into the standard\n\t * TanStack AI TranscriptionResult shape.\n\t *\n\t * Handles three response formats:\n\t * - Whisper: `{ text, words?, vtt? }`\n\t * - Whisper v3-turbo: `{ text, segments?, transcription_info? }`\n\t * - Deepgram Nova-3: `{ results: { channels: [{ alternatives: [{ transcript, words }] }] } }`\n\t */\n\tprivate normalizeResult(raw: Record<string, unknown>): TranscriptionResult {\n\t\t// Deepgram Nova-3 format: { results: { channels: [{ alternatives: [{ transcript, words }] }] } }\n\t\tconst results = raw.results as Record<string, unknown> | undefined;\n\t\tif (results?.channels) {\n\t\t\tconst channels = results.channels as Array<{\n\t\t\t\talternatives?: Array<{\n\t\t\t\t\ttranscript?: string;\n\t\t\t\t\tconfidence?: number;\n\t\t\t\t\twords?: Array<{ word: string; start: number; end: number; confidence: number }>;\n\t\t\t\t}>;\n\t\t\t}>;\n\t\t\tconst alt = channels?.[0]?.alternatives?.[0];\n\t\t\tconst text = alt?.transcript ?? \"\";\n\t\t\tconst result: TranscriptionResult = {\n\t\t\t\tid: this.generateId(),\n\t\t\t\tmodel: this.model,\n\t\t\t\ttext,\n\t\t\t};\n\t\t\tif (alt?.words && Array.isArray(alt.words)) {\n\t\t\t\tresult.words = alt.words.map((w) => ({\n\t\t\t\t\tword: w.word ?? \"\",\n\t\t\t\t\tstart: w.start ?? 0,\n\t\t\t\t\tend: w.end ?? 0,\n\t\t\t\t}));\n\t\t\t}\n\t\t\treturn result;\n\t\t}\n\n\t\t// Whisper format: { text, words?, vtt? }\n\t\t// Whisper v3-turbo format: { text, segments?, transcription_info? }\n\t\tconst result: TranscriptionResult = {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\ttext: (raw.text as string) ?? \"\",\n\t\t};\n\n\t\t// Language from transcription_info (whisper-large-v3-turbo)\n\t\tconst transcriptionInfo = raw.transcription_info as Record<string, unknown> | undefined;\n\t\tif (transcriptionInfo?.language) {\n\t\t\tresult.language = transcriptionInfo.language as string;\n\t\t}\n\n\t\t// Duration\n\t\tif (transcriptionInfo?.duration != null) {\n\t\t\tresult.duration = transcriptionInfo.duration as number;\n\t\t}\n\n\t\t// Segments (whisper-large-v3-turbo returns these)\n\t\tif (raw.segments && Array.isArray(raw.segments)) {\n\t\t\tresult.segments = raw.segments.map((seg: Record<string, unknown>, idx: number) => ({\n\t\t\t\tid: idx,\n\t\t\t\ttext: (seg.text as string) ?? \"\",\n\t\t\t\tstart: (seg.start as number) ?? 0,\n\t\t\t\tend: (seg.end as number) ?? 0,\n\t\t\t}));\n\t\t}\n\n\t\t// Words — basic whisper returns top-level words[], v3-turbo nests them in segments\n\t\tif (raw.words && Array.isArray(raw.words)) {\n\t\t\tresult.words = raw.words.map((w: Record<string, unknown>) => ({\n\t\t\t\tword: (w.word as string) ?? \"\",\n\t\t\t\tstart: (w.start as number) ?? 0,\n\t\t\t\tend: (w.end as number) ?? 0,\n\t\t\t}));\n\t\t}\n\n\t\treturn result;\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI transcription adapter for speech-to-text.\n *\n * Works with TanStack AI's `generateTranscription()` activity function:\n * ```ts\n * import { generateTranscription } from \"@tanstack/ai\";\n * import { createWorkersAiTranscription } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiTranscription(\n * \"@cf/openai/whisper-large-v3-turbo\",\n * { binding: env.AI },\n * );\n *\n * const result = await generateTranscription({ adapter, audio: audioData });\n * // result.text — the transcribed text\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiTranscription(\n\tmodel: WorkersAiTranscriptionModel,\n\tconfig: WorkersAiAdapterConfig,\n) {\n\treturn new WorkersAiTranscriptionAdapter(config, model);\n}\n\n// ---------------------------------------------------------------------------\n// Utilities\n// ---------------------------------------------------------------------------\n\n/**\n * Normalize various audio input formats into a number[] (raw bytes).\n *\n * Note: `File extends Blob`, so `File` instances are handled by the\n * `instanceof Blob` branch. `Blob.arrayBuffer()` always reads the full\n * contents regardless of any prior reads — there's no cursor to worry about.\n */\nasync function normalizeAudioToBytes(audio: string | File | Blob | ArrayBuffer): Promise<number[]> {\n\tif (audio instanceof ArrayBuffer) {\n\t\treturn Array.from(new Uint8Array(audio));\n\t}\n\n\tif (audio instanceof Blob) {\n\t\t// This also handles `File` (which extends Blob)\n\t\tconst buffer = await audio.arrayBuffer();\n\t\treturn Array.from(new Uint8Array(buffer));\n\t}\n\n\tif (typeof audio === \"string\") {\n\t\t// Assume base64 string — decode to bytes\n\t\tconst binary = atob(audio);\n\t\tconst bytes = new Uint8Array(binary.length);\n\t\tfor (let i = 0; i < binary.length; i++) {\n\t\t\tbytes[i] = binary.charCodeAt(i);\n\t\t}\n\t\treturn Array.from(bytes);\n\t}\n\n\tthrow new Error(\"Unsupported audio format. Expected string, File, Blob, or ArrayBuffer.\");\n}\n\n/**\n * Detect the MIME type of the audio input for models that require it\n * (e.g., Deepgram Nova-3).\n *\n * - `File` / `Blob`: use the `.type` property (e.g., \"audio/wav\")\n * - `ArrayBuffer` / `string`: defaults to \"audio/wav\"\n */\nfunction detectAudioContentType(audio: string | File | Blob | ArrayBuffer): string {\n\t// File and Blob carry their own MIME type\n\tif (audio instanceof Blob && audio.type) {\n\t\treturn audio.type;\n\t}\n\n\t// For raw bytes, default to audio/wav — this is the most common\n\t// format for transcription inputs and what the E2E tests use.\n\treturn \"audio/wav\";\n}\n"]}
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-72MGG2FX.cjs","../src/adapters/workers-ai-transcription.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACjBA,iDAAyC;AAyClC,IAAM,8BAAA,EAAN,MAAA,QAA4C,mCAAsD;AAAA,EAIxG,WAAA,CAAY,MAAA,EAAgC,KAAA,EAAoC;AAC/E,IAAA,KAAA,CAAM,CAAC,CAAA,EAAG,KAAK,CAAA;AAJhB,IAAA,6CAAA,IAAA,EAAS,MAAA,EAAO,0BAAA,CAAA;AAChB,IAAA,6CAAA,IAAA,EAAQ,eAAA,CAAA;AAIP,IAAA,uDAAA,MAA8B,CAAA;AAC9B,IAAA,IAAA,CAAK,cAAA,EAAgB,MAAA;AAAA,EACtB;AAAA,EAEA,MAAM,UAAA,CAAW,OAAA,EAA6D;AAC7E,IAAA,MAAM,EAAE,KAAA,EAAO,QAAA,EAAU,MAAA,EAAQ,aAAa,EAAA,EAAI,OAAA;AAGlD,IAAA,MAAM,WAAA,EAAa,MAAM,qBAAA,CAAsB,KAAK,CAAA;AAEpD,IAAA,MAAM,MAAA,EAAiC,EAAE,GAAG,aAAa,CAAA;AACzD,IAAA,GAAA,CAAI,QAAA,EAAU,KAAA,CAAM,SAAA,EAAW,QAAA;AAC/B,IAAA,GAAA,CAAI,MAAA,EAAQ,KAAA,CAAM,eAAA,EAAiB,MAAA;AAOnC,IAAA,MAAM,aAAA,EAAe,IAAA,CAAK,iBAAA,CAAkB,UAAA,EAAY,KAAK,CAAA;AAE7D,IAAA,GAAA,CAAI,qDAAA,IAAsB,CAAK,aAAa,CAAA,EAAG;AAC9C,MAAA,OAAO,IAAA,CAAK,oBAAA,CAAqB,YAAA,EAAc,KAAK,CAAA;AAAA,IACrD;AAEA,IAAA,GAAA,CAAI,yDAAA,IAA0B,CAAK,aAAa,CAAA,EAAG;AAElD,MAAA,GAAA,CAAI,IAAA,CAAK,MAAA,IAAU,qBAAA,EAAuB;AACzC,QAAA,OAAO,IAAA,CAAK,uBAAA,CAAwB,UAAA,EAAY,KAAA,EAAO,KAAK,CAAA;AAAA,MAC7D;AACA,MAAA,OAAO,IAAA,CAAK,iBAAA,CAAkB,YAAA,EAAc,KAAK,CAAA;AAAA,IAClD;AAEA,IAAA,OAAO,IAAA,CAAK,oBAAA,CAAqB,YAAA,EAAc,KAAK,CAAA;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASQ,iBAAA,CACP,UAAA,EACA,aAAA,EAC0B;AAC1B,IAAA,GAAA,CAAI,IAAA,CAAK,MAAA,IAAU,qBAAA,EAAuB;AACzC,MAAA,MAAM,IAAA,EAAM,kDAAA,IAAuB,UAAA,CAAW,UAAU,CAAC,CAAA;AACzD,MAAA,MAAM,YAAA,EAAc,sBAAA,CAAuB,aAAa,CAAA;AACxD,MAAA,OAAO,EAAE,KAAA,EAAO,EAAE,IAAA,EAAM,GAAA,EAAK,YAAY,EAAE,CAAA;AAAA,IAC5C;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK,MAAA,IAAU,mCAAA,EAAqC;AACvD,MAAA,OAAO,EAAE,KAAA,EAAO,kDAAA,IAAuB,UAAA,CAAW,UAAU,CAAC,EAAE,CAAA;AAAA,IAChE;AAEA,IAAA,OAAO,EAAE,KAAA,EAAO,WAAW,CAAA;AAAA,EAC5B;AAAA,EAEA,MAAc,oBAAA,CACb,YAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,GAAA,EAAM,IAAA,CAAK,aAAA,CAA+C,OAAA;AAChE,IAAA,MAAM,OAAA,EAAU,MAAM,EAAA,CAAG,GAAA,CAAI,IAAA,CAAK,KAAA,EAAO;AAAA,MACxC,GAAG,YAAA;AAAA,MACH,GAAG;AAAA,IACJ,CAAC,CAAA;AACD,IAAA,OAAO,IAAA,CAAK,eAAA,CAAgB,MAAM,CAAA;AAAA,EACnC;AAAA,EAEA,MAAc,iBAAA,CACb,YAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AAEpB,IAAA,MAAM,SAAA,EAAW,MAAM,kDAAA;AAAA,MACtB,MAAA;AAAA,MACA,IAAA,CAAK,KAAA;AAAA,MACL,EAAE,GAAG,YAAA,EAAc,GAAG,QAAQ,CAAA;AAAA,MAC9B;AAAA,QACC,KAAA,EAAO,0BAAA;AAAA,QACP,MAAA,EAAS,OAAA,CAAqC;AAAA,MAC/C;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,KAAA,EAAQ,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AAMlC,IAAA,OAAO,IAAA,CAAK,eAAA,kBAAgB,IAAA,CAAK,MAAA,UAAU,MAAI,CAAA;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAc,uBAAA,CACb,UAAA,EACA,aAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AACpB,IAAA,MAAM,YAAA,EAAc,sBAAA,CAAuB,aAAa,CAAA;AAExD,IAAA,MAAM,SAAA,EAAW,MAAM,wDAAA;AAAA,MACtB,MAAA;AAAA,MACA,IAAA,CAAK,KAAA;AAAA,MACL,IAAI,UAAA,CAAW,UAAU,CAAA;AAAA,MACzB,WAAA;AAAA,MACA;AAAA,QACC,KAAA,EAAO,0BAAA;AAAA,QACP,MAAA,EAAS,OAAA,CAAqC;AAAA,MAC/C;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,KAAA,EAAQ,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AAIlC,IAAA,OAAO,IAAA,CAAK,eAAA,kBAAgB,IAAA,CAAK,MAAA,UAAU,MAAI,CAAA;AAAA,EAChD;AAAA,EAEA,MAAc,oBAAA,CACb,YAAA,EACA,OAAA,EAC+B;AAC/B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,aAAA;AAC3B,IAAA,MAAM,aAAA,EAAe,kDAAA,YAAmB,EAAc,aAAa,CAAA;AAKnE,IAAA,MAAM,SAAA,EAAW,MAAM,YAAA,CAAa,oDAAA,EAAsD;AAAA,MACzF,MAAA,EAAQ,MAAA;AAAA,MACR,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU;AAAA,QACpB,KAAA,EAAO,IAAA,CAAK,KAAA;AAAA,QACZ,GAAG,YAAA;AAAA,QACH,GAAG;AAAA,MACJ,CAAC;AAAA,IACF,CAAC,CAAA;AAED,IAAA,GAAA,CAAI,CAAC,QAAA,CAAS,EAAA,EAAI;AACjB,MAAA,MAAM,UAAA,EAAY,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AACtC,MAAA,MAAM,IAAI,KAAA;AAAA,QACT,CAAA,iDAAA,EAAoD,QAAA,CAAS,MAAM,CAAA,GAAA,EAAM,SAAS,CAAA;AAAA,MAAA;AACnF,IAAA;AAGD,IAAA;AACA,IAAA;AAAgC,EAAA;AACjC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAaC,IAAA;AACA,IAAA;AACC,MAAA;AAOA,MAAA;AACA,MAAA;AACA,MAAA;AAAoC,QAAA;AACf,QAAA;AACR,QAAA;AACZ,MAAA;AAED,MAAA;AACC,QAAA;AAAqC,UAAA;AACpB,UAAA;AACE,UAAA;AACJ,QAAA;AACb,MAAA;AAEH,MAAA;AAAO,IAAA;AAKR,IAAA;AAAoC,MAAA;AACf,MAAA;AACR,MAAA;AACkB,IAAA;AAI/B,IAAA;AACA,IAAA;AACC,MAAA;AAAoC,IAAA;AAIrC,IAAA;AACC,MAAA;AAAoC,IAAA;AAIrC,IAAA;AACC,MAAA;AAAmF,QAAA;AAC9E,QAAA;AAC0B,QAAA;AACE,QAAA;AACJ,MAAA;AAC3B,IAAA;AAIH,IAAA;AACC,MAAA;AAA8D,QAAA;AACjC,QAAA;AACE,QAAA;AACJ,MAAA;AACzB,IAAA;AAGH,IAAA;AAAO,EAAA;AAET;AA0BO;AAIN,EAAA;AACD;AAaA;AACC,EAAA;AACC,IAAA;AAAuC,EAAA;AAGxC,EAAA;AAEC,IAAA;AACA,IAAA;AAAwC,EAAA;AAGzC,EAAA;AAEC,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AAA8B,IAAA;AAE/B,IAAA;AAAuB,EAAA;AAGxB,EAAA;AACD;AASA;AAEC,EAAA;AACC,IAAA;AAAa,EAAA;AAKd,EAAA;AACD;AD1JA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-72MGG2FX.cjs","sourcesContent":[null,"import { BaseTranscriptionAdapter } from \"@tanstack/ai/adapters\";\nimport type { TranscriptionOptions, TranscriptionResult } from \"@tanstack/ai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype WorkersAiDirectCredentialsConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch, workersAiRestFetchBinary } from \"../utils/workers-ai-rest\";\nimport { uint8ArrayToBase64 } from \"../utils/binary\";\n\n// ---------------------------------------------------------------------------\n// Model types\n// ---------------------------------------------------------------------------\n\n/**\n * Workers AI models that support speech-to-text transcription.\n *\n * Note: the typed `AiModels` interface in `@cloudflare/workers-types` may lag\n * behind what's deployed. We use a string union here that matches the known\n * models including Deepgram partner models.\n *\n * **Nova-3 note:** `@cf/deepgram/nova-3` uses a different input format than the\n * Whisper models. Via binding it accepts `{ audio: { body: base64, contentType } }`.\n * Via REST it requires multipart form data (not JSON). The adapter handles both.\n */\nexport type WorkersAiTranscriptionModel =\n\t| \"@cf/openai/whisper\"\n\t| \"@cf/openai/whisper-tiny-en\"\n\t| \"@cf/openai/whisper-large-v3-turbo\"\n\t| \"@cf/deepgram/nova-3\"\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiTranscriptionAdapter\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiTranscriptionAdapter extends BaseTranscriptionAdapter<WorkersAiTranscriptionModel> {\n\treadonly name = \"workers-ai-transcription\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiTranscriptionModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync transcribe(options: TranscriptionOptions): Promise<TranscriptionResult> {\n\t\tconst { audio, language, prompt, modelOptions } = options;\n\n\t\t// Normalize audio to raw bytes\n\t\tconst audioBytes = await normalizeAudioToBytes(audio);\n\n\t\tconst extra: Record<string, unknown> = { ...modelOptions };\n\t\tif (language) extra.language = language;\n\t\tif (prompt) extra.initial_prompt = prompt;\n\n\t\t// Build the model-specific audio payload:\n\t\t// - Deepgram Nova-3 (binding): { audio: { body: base64, contentType: \"audio/...\" } }\n\t\t// - Deepgram Nova-3 (REST): multipart FormData (handled separately)\n\t\t// - Whisper Large v3 Turbo (REST/gateway): { audio: base64string }\n\t\t// - Other Whisper models (binding): { audio: number[] }\n\t\tconst audioPayload = this.buildAudioPayload(audioBytes, audio);\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.transcribeViaBinding(audioPayload, extra);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\t// Nova-3 REST requires raw binary audio, not JSON\n\t\t\tif (this.model === \"@cf/deepgram/nova-3\") {\n\t\t\t\treturn this.transcribeViaRestBinary(audioBytes, audio, extra);\n\t\t\t}\n\t\t\treturn this.transcribeViaRest(audioPayload, extra);\n\t\t}\n\n\t\treturn this.transcribeViaGateway(audioPayload, extra);\n\t}\n\n\t/**\n\t * Build the audio field for the request payload, handling model-specific formats.\n\t *\n\t * - `@cf/deepgram/nova-3` requires `{ body: base64, contentType: \"audio/...\" }`\n\t * - `@cf/openai/whisper-large-v3-turbo` REST/gateway accepts a base64 string\n\t * - Other Whisper models accept `number[]` (binding) or base64 (REST)\n\t */\n\tprivate buildAudioPayload(\n\t\taudioBytes: number[],\n\t\toriginalAudio: string | File | Blob | ArrayBuffer,\n\t): Record<string, unknown> {\n\t\tif (this.model === \"@cf/deepgram/nova-3\") {\n\t\t\tconst b64 = uint8ArrayToBase64(new Uint8Array(audioBytes));\n\t\t\tconst contentType = detectAudioContentType(originalAudio);\n\t\t\treturn { audio: { body: b64, contentType } };\n\t\t}\n\n\t\tif (this.model === \"@cf/openai/whisper-large-v3-turbo\") {\n\t\t\treturn { audio: uint8ArrayToBase64(new Uint8Array(audioBytes)) };\n\t\t}\n\n\t\treturn { audio: audioBytes };\n\t}\n\n\tprivate async transcribeViaBinding(\n\t\taudioPayload: Record<string, unknown>,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = (await ai.run(this.model, {\n\t\t\t...audioPayload,\n\t\t\t...options,\n\t\t})) as Record<string, unknown>;\n\t\treturn this.normalizeResult(result);\n\t}\n\n\tprivate async transcribeViaRest(\n\t\taudioPayload: Record<string, unknown>,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\n\t\tconst response = await workersAiRestFetch(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\t{ ...audioPayload, ...options },\n\t\t\t{\n\t\t\t\tlabel: \"Workers AI transcription\",\n\t\t\t\tsignal: (options as { signal?: AbortSignal }).signal,\n\t\t\t},\n\t\t);\n\n\t\tconst data = (await response.json()) as {\n\t\t\tresult?: Record<string, unknown>;\n\t\t} & Record<string, unknown>;\n\n\t\t// Cloudflare REST API wraps responses in { success, result: {...} }.\n\t\t// Use `data.result` when present, fall back to `data` for direct responses.\n\t\treturn this.normalizeResult(data.result ?? data);\n\t}\n\n\t/**\n\t * Transcribe via REST using raw binary audio.\n\t * Required for models like Deepgram Nova-3 that expect raw audio bytes\n\t * with a Content-Type header (e.g. \"audio/wav\") instead of JSON.\n\t */\n\tprivate async transcribeViaRestBinary(\n\t\taudioBytes: number[],\n\t\toriginalAudio: string | File | Blob | ArrayBuffer,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst contentType = detectAudioContentType(originalAudio);\n\n\t\tconst response = await workersAiRestFetchBinary(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\tnew Uint8Array(audioBytes),\n\t\t\tcontentType,\n\t\t\t{\n\t\t\t\tlabel: \"Workers AI transcription\",\n\t\t\t\tsignal: (options as { signal?: AbortSignal }).signal,\n\t\t\t},\n\t\t);\n\n\t\tconst data = (await response.json()) as {\n\t\t\tresult?: Record<string, unknown>;\n\t\t} & Record<string, unknown>;\n\n\t\treturn this.normalizeResult(data.result ?? data);\n\t}\n\n\tprivate async transcribeViaGateway(\n\t\taudioPayload: Record<string, unknown>,\n\t\toptions: Record<string, unknown>,\n\t): Promise<TranscriptionResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/audio/transcriptions\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\t...audioPayload,\n\t\t\t\t...options,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI transcription gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst data = (await response.json()) as Record<string, unknown>;\n\t\treturn this.normalizeResult(data);\n\t}\n\n\t/**\n\t * Normalize Workers AI transcription results into the standard\n\t * TanStack AI TranscriptionResult shape.\n\t *\n\t * Handles three response formats:\n\t * - Whisper: `{ text, words?, vtt? }`\n\t * - Whisper v3-turbo: `{ text, segments?, transcription_info? }`\n\t * - Deepgram Nova-3: `{ results: { channels: [{ alternatives: [{ transcript, words }] }] } }`\n\t */\n\tprivate normalizeResult(raw: Record<string, unknown>): TranscriptionResult {\n\t\t// Deepgram Nova-3 format: { results: { channels: [{ alternatives: [{ transcript, words }] }] } }\n\t\tconst results = raw.results as Record<string, unknown> | undefined;\n\t\tif (results?.channels) {\n\t\t\tconst channels = results.channels as Array<{\n\t\t\t\talternatives?: Array<{\n\t\t\t\t\ttranscript?: string;\n\t\t\t\t\tconfidence?: number;\n\t\t\t\t\twords?: Array<{ word: string; start: number; end: number; confidence: number }>;\n\t\t\t\t}>;\n\t\t\t}>;\n\t\t\tconst alt = channels?.[0]?.alternatives?.[0];\n\t\t\tconst text = alt?.transcript ?? \"\";\n\t\t\tconst result: TranscriptionResult = {\n\t\t\t\tid: this.generateId(),\n\t\t\t\tmodel: this.model,\n\t\t\t\ttext,\n\t\t\t};\n\t\t\tif (alt?.words && Array.isArray(alt.words)) {\n\t\t\t\tresult.words = alt.words.map((w) => ({\n\t\t\t\t\tword: w.word ?? \"\",\n\t\t\t\t\tstart: w.start ?? 0,\n\t\t\t\t\tend: w.end ?? 0,\n\t\t\t\t}));\n\t\t\t}\n\t\t\treturn result;\n\t\t}\n\n\t\t// Whisper format: { text, words?, vtt? }\n\t\t// Whisper v3-turbo format: { text, segments?, transcription_info? }\n\t\tconst result: TranscriptionResult = {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\ttext: (raw.text as string) ?? \"\",\n\t\t};\n\n\t\t// Language from transcription_info (whisper-large-v3-turbo)\n\t\tconst transcriptionInfo = raw.transcription_info as Record<string, unknown> | undefined;\n\t\tif (transcriptionInfo?.language) {\n\t\t\tresult.language = transcriptionInfo.language as string;\n\t\t}\n\n\t\t// Duration\n\t\tif (transcriptionInfo?.duration != null) {\n\t\t\tresult.duration = transcriptionInfo.duration as number;\n\t\t}\n\n\t\t// Segments (whisper-large-v3-turbo returns these)\n\t\tif (raw.segments && Array.isArray(raw.segments)) {\n\t\t\tresult.segments = raw.segments.map((seg: Record<string, unknown>, idx: number) => ({\n\t\t\t\tid: idx,\n\t\t\t\ttext: (seg.text as string) ?? \"\",\n\t\t\t\tstart: (seg.start as number) ?? 0,\n\t\t\t\tend: (seg.end as number) ?? 0,\n\t\t\t}));\n\t\t}\n\n\t\t// Words — basic whisper returns top-level words[], v3-turbo nests them in segments\n\t\tif (raw.words && Array.isArray(raw.words)) {\n\t\t\tresult.words = raw.words.map((w: Record<string, unknown>) => ({\n\t\t\t\tword: (w.word as string) ?? \"\",\n\t\t\t\tstart: (w.start as number) ?? 0,\n\t\t\t\tend: (w.end as number) ?? 0,\n\t\t\t}));\n\t\t}\n\n\t\treturn result;\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI transcription adapter for speech-to-text.\n *\n * Works with TanStack AI's `generateTranscription()` activity function:\n * ```ts\n * import { generateTranscription } from \"@tanstack/ai\";\n * import { createWorkersAiTranscription } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiTranscription(\n * \"@cf/openai/whisper-large-v3-turbo\",\n * { binding: env.AI },\n * );\n *\n * const result = await generateTranscription({ adapter, audio: audioData });\n * // result.text — the transcribed text\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiTranscription(\n\tmodel: WorkersAiTranscriptionModel,\n\tconfig: WorkersAiAdapterConfig,\n) {\n\treturn new WorkersAiTranscriptionAdapter(config, model);\n}\n\n// ---------------------------------------------------------------------------\n// Utilities\n// ---------------------------------------------------------------------------\n\n/**\n * Normalize various audio input formats into a number[] (raw bytes).\n *\n * Note: `File extends Blob`, so `File` instances are handled by the\n * `instanceof Blob` branch. `Blob.arrayBuffer()` always reads the full\n * contents regardless of any prior reads — there's no cursor to worry about.\n */\nasync function normalizeAudioToBytes(audio: string | File | Blob | ArrayBuffer): Promise<number[]> {\n\tif (audio instanceof ArrayBuffer) {\n\t\treturn Array.from(new Uint8Array(audio));\n\t}\n\n\tif (audio instanceof Blob) {\n\t\t// This also handles `File` (which extends Blob)\n\t\tconst buffer = await audio.arrayBuffer();\n\t\treturn Array.from(new Uint8Array(buffer));\n\t}\n\n\tif (typeof audio === \"string\") {\n\t\t// Assume base64 string — decode to bytes\n\t\tconst binary = atob(audio);\n\t\tconst bytes = new Uint8Array(binary.length);\n\t\tfor (let i = 0; i < binary.length; i++) {\n\t\t\tbytes[i] = binary.charCodeAt(i);\n\t\t}\n\t\treturn Array.from(bytes);\n\t}\n\n\tthrow new Error(\"Unsupported audio format. Expected string, File, Blob, or ArrayBuffer.\");\n}\n\n/**\n * Detect the MIME type of the audio input for models that require it\n * (e.g., Deepgram Nova-3).\n *\n * - `File` / `Blob`: use the `.type` property (e.g., \"audio/wav\")\n * - `ArrayBuffer` / `string`: defaults to \"audio/wav\"\n */\nfunction detectAudioContentType(audio: string | File | Blob | ArrayBuffer): string {\n\t// File and Blob carry their own MIME type\n\tif (audio instanceof Blob && audio.type) {\n\t\treturn audio.type;\n\t}\n\n\t// For raw bytes, default to audio/wav — this is the most common\n\t// format for transcription inputs and what the E2E tests use.\n\treturn \"audio/wav\";\n}\n"]}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import {
|
|
2
2
|
createGatewayFetch
|
|
3
|
-
} from "./chunk-
|
|
3
|
+
} from "./chunk-NGOJB274.js";
|
|
4
4
|
|
|
5
5
|
// src/adapters/grok.ts
|
|
6
6
|
import {
|
|
@@ -45,4 +45,4 @@ export {
|
|
|
45
45
|
createGrokImage,
|
|
46
46
|
createGrokSummarize
|
|
47
47
|
};
|
|
48
|
-
//# sourceMappingURL=chunk-
|
|
48
|
+
//# sourceMappingURL=chunk-73UI5P36.js.map
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var _chunk3L6IRVAVcjs = require('./chunk-3L6IRVAV.cjs');
|
|
4
4
|
|
|
5
5
|
// src/adapters/grok.ts
|
|
6
6
|
|
|
@@ -14,7 +14,7 @@ function createGrokChat(model, config) {
|
|
|
14
14
|
return new (0, _aigrok.GrokTextAdapter)(
|
|
15
15
|
{
|
|
16
16
|
apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
|
|
17
|
-
fetch:
|
|
17
|
+
fetch: _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "grok", config)
|
|
18
18
|
},
|
|
19
19
|
model
|
|
20
20
|
);
|
|
@@ -23,7 +23,7 @@ function createGrokImage(model, config) {
|
|
|
23
23
|
return new (0, _aigrok.GrokImageAdapter)(
|
|
24
24
|
{
|
|
25
25
|
apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
|
|
26
|
-
fetch:
|
|
26
|
+
fetch: _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "grok", config)
|
|
27
27
|
},
|
|
28
28
|
model
|
|
29
29
|
);
|
|
@@ -32,7 +32,7 @@ function createGrokSummarize(model, config) {
|
|
|
32
32
|
return new (0, _aigrok.GrokSummarizeAdapter)(
|
|
33
33
|
{
|
|
34
34
|
apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
|
|
35
|
-
fetch:
|
|
35
|
+
fetch: _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "grok", config)
|
|
36
36
|
},
|
|
37
37
|
model
|
|
38
38
|
);
|
|
@@ -45,4 +45,4 @@ function createGrokSummarize(model, config) {
|
|
|
45
45
|
|
|
46
46
|
|
|
47
47
|
exports.GROK_CHAT_MODELS = _aigrok.GROK_CHAT_MODELS; exports.GROK_IMAGE_MODELS = _aigrok.GROK_IMAGE_MODELS; exports.createGrokChat = createGrokChat; exports.createGrokImage = createGrokImage; exports.createGrokSummarize = createGrokSummarize;
|
|
48
|
-
//# sourceMappingURL=chunk-
|
|
48
|
+
//# sourceMappingURL=chunk-CEQA4ZVD.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-CEQA4ZVD.cjs","../src/adapters/grok.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA;AACC;AACA;AACA;AACA;AACA;AAAA,2CAIM;AAYA,SAAS,cAAA,CAAe,KAAA,EAAsB,MAAA,EAA2B;AAC/E,EAAA,OAAO,IAAI,4BAAA;AAAA,IACV;AAAA,MACC,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,MACzB,KAAA,EAAO,kDAAA,MAAmB,EAAQ,MAAM;AAAA,IACzC,CAAA;AAAA,IACA;AAAA,EACD,CAAA;AACD;AAMO,SAAS,eAAA,CAAgB,KAAA,EAAuB,MAAA,EAA2B;AACjF,EAAA,OAAO,IAAI,6BAAA;AAAA,IACV;AAAA,MACC,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,MACzB,KAAA,EAAO,kDAAA,MAAmB,EAAQ,MAAM;AAAA,IACzC,CAAA;AAAA,IACA;AAAA,EACD,CAAA;AACD;AAMO,SAAS,mBAAA,CAAoB,KAAA,EAA2B,MAAA,EAA2B;AACzF,EAAA,OAAO,IAAI,iCAAA;AAAA,IACV;AAAA,MACC,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,MACzB,KAAA,EAAO,kDAAA,MAAmB,EAAQ,MAAM;AAAA,IACzC,CAAA;AAAA,IACA;AAAA,EACD,CAAA;AACD;ADlBA;AACA;AACE;AACA;AACA;AACA;AACA;AACF,kPAAC","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-CEQA4ZVD.cjs","sourcesContent":[null,"import {\n\tGrokTextAdapter,\n\tGrokImageAdapter,\n\tGrokSummarizeAdapter,\n\tGROK_CHAT_MODELS,\n\tGROK_IMAGE_MODELS,\n\ttype GrokChatModel,\n\ttype GrokImageModel,\n\ttype GrokSummarizeModel,\n} from \"@tanstack/ai-grok\";\nimport { createGatewayFetch, type AiGatewayAdapterConfig } from \"../utils/create-fetcher\";\n\nexport type GrokGatewayConfig = AiGatewayAdapterConfig;\n\n/**\n * Creates a Grok chat adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * Since GrokTextConfig extends the OpenAI SDK's ClientOptions,\n * we can inject the gateway fetch directly — no subclassing needed.\n */\nexport function createGrokChat(model: GrokChatModel, config: GrokGatewayConfig) {\n\treturn new GrokTextAdapter(\n\t\t{\n\t\t\tapiKey: config.apiKey ?? \"unused\",\n\t\t\tfetch: createGatewayFetch(\"grok\", config),\n\t\t},\n\t\tmodel,\n\t);\n}\n\n/**\n * Creates a Grok image adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n */\nexport function createGrokImage(model: GrokImageModel, config: GrokGatewayConfig) {\n\treturn new GrokImageAdapter(\n\t\t{\n\t\t\tapiKey: config.apiKey ?? \"unused\",\n\t\t\tfetch: createGatewayFetch(\"grok\", config),\n\t\t},\n\t\tmodel,\n\t);\n}\n\n/**\n * Creates a Grok summarize adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n */\nexport function createGrokSummarize(model: GrokSummarizeModel, config: GrokGatewayConfig) {\n\treturn new GrokSummarizeAdapter(\n\t\t{\n\t\t\tapiKey: config.apiKey ?? \"unused\",\n\t\t\tfetch: createGatewayFetch(\"grok\", config),\n\t\t},\n\t\tmodel,\n\t);\n}\n\nexport {\n\tGROK_CHAT_MODELS,\n\tGROK_IMAGE_MODELS,\n\ttype GrokChatModel,\n\ttype GrokImageModel,\n\ttype GrokSummarizeModel,\n};\n"]}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var _chunk3L6IRVAVcjs = require('./chunk-3L6IRVAV.cjs');
|
|
4
4
|
|
|
5
5
|
// src/adapters/openai.ts
|
|
6
6
|
|
|
@@ -19,7 +19,7 @@ var _aiopenai = require('@tanstack/ai-openai');
|
|
|
19
19
|
function buildOpenAiConfig(provider, config) {
|
|
20
20
|
return {
|
|
21
21
|
apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
|
|
22
|
-
fetch:
|
|
22
|
+
fetch: _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, provider, config)
|
|
23
23
|
};
|
|
24
24
|
}
|
|
25
25
|
function createOpenAiChat(model, config) {
|
|
@@ -54,4 +54,4 @@ function createOpenAiVideo(model, config) {
|
|
|
54
54
|
|
|
55
55
|
|
|
56
56
|
exports.OPENAI_CHAT_MODELS = _aiopenai.OPENAI_CHAT_MODELS; exports.OPENAI_IMAGE_MODELS = _aiopenai.OPENAI_IMAGE_MODELS; exports.OPENAI_TRANSCRIPTION_MODELS = _aiopenai.OPENAI_TRANSCRIPTION_MODELS; exports.OPENAI_TTS_MODELS = _aiopenai.OPENAI_TTS_MODELS; exports.OPENAI_VIDEO_MODELS = _aiopenai.OPENAI_VIDEO_MODELS; exports.createOpenAiChat = createOpenAiChat; exports.createOpenAiSummarize = createOpenAiSummarize; exports.createOpenAiImage = createOpenAiImage; exports.createOpenAiTranscription = createOpenAiTranscription; exports.createOpenAiTts = createOpenAiTts; exports.createOpenAiVideo = createOpenAiVideo;
|
|
57
|
-
//# sourceMappingURL=chunk-
|
|
57
|
+
//# sourceMappingURL=chunk-GT3TZXZX.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-GT3TZXZX.cjs","../src/adapters/openai.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA;AACC;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AAAA,+CAMM;AASP,SAAS,iBAAA,CAAkB,QAAA,EAAkB,MAAA,EAA6B;AACzE,EAAA,OAAO;AAAA,IACN,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,IACzB,KAAA,EAAO,kDAAA,QAAmB,EAAU,MAAM;AAAA,EAC3C,CAAA;AACD;AAQO,SAAS,gBAAA,CAAiB,KAAA,EAAwB,MAAA,EAA6B;AACrF,EAAA,OAAO,IAAI,gCAAA,CAAkB,iBAAA,CAAkB,QAAA,EAAU,MAAM,CAAA,EAAG,KAAK,CAAA;AACxE;AAOO,SAAS,qBAAA,CAAsB,KAAA,EAAwB,MAAA,EAA6B;AAC1F,EAAA,OAAO,IAAI,qCAAA,CAAuB,iBAAA,CAAkB,QAAA,EAAU,MAAM,CAAA,EAAG,KAAK,CAAA;AAC7E;AAOO,SAAS,iBAAA,CAAkB,KAAA,EAAyB,MAAA,EAA6B;AACvF,EAAA,OAAO,IAAI,iCAAA,CAAmB,iBAAA,CAAkB,QAAA,EAAU,MAAM,CAAA,EAAG,KAAK,CAAA;AACzE;AAOO,SAAS,yBAAA,CACf,KAAA,EACA,MAAA,EACC;AACD,EAAA,OAAO,IAAI,yCAAA,CAA2B,iBAAA,CAAkB,QAAA,EAAU,MAAM,CAAA,EAAG,KAAK,CAAA;AACjF;AAOO,SAAS,eAAA,CAAgB,KAAA,EAAuB,MAAA,EAA6B;AACnF,EAAA,OAAO,IAAI,+BAAA,CAAiB,iBAAA,CAAkB,QAAA,EAAU,MAAM,CAAA,EAAG,KAAK,CAAA;AACvE;AAOO,SAAS,iBAAA,CAAkB,KAAA,EAAyB,MAAA,EAA6B;AACvF,EAAA,OAAO,IAAI,iCAAA,CAAmB,iBAAA,CAAkB,QAAA,EAAU,MAAM,CAAA,EAAG,KAAK,CAAA;AACzE;AD/CA;AACA;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,smBAAC","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-GT3TZXZX.cjs","sourcesContent":[null,"import {\n\tOpenAIImageAdapter,\n\tOpenAISummarizeAdapter,\n\tOpenAITextAdapter,\n\tOpenAITranscriptionAdapter,\n\tOpenAITTSAdapter,\n\tOpenAIVideoAdapter,\n\tOPENAI_CHAT_MODELS,\n\tOPENAI_IMAGE_MODELS,\n\tOPENAI_TRANSCRIPTION_MODELS,\n\tOPENAI_TTS_MODELS,\n\tOPENAI_VIDEO_MODELS,\n\ttype OpenAIChatModel,\n\ttype OpenAIImageModel,\n\ttype OpenAITranscriptionModel,\n\ttype OpenAITTSModel,\n\ttype OpenAIVideoModel,\n} from \"@tanstack/ai-openai\";\nimport { createGatewayFetch, type AiGatewayAdapterConfig } from \"../utils/create-fetcher\";\n\nexport type OpenAiGatewayConfig = AiGatewayAdapterConfig;\n\n/**\n * Builds an OpenAI-compatible config that injects the gateway fetch.\n * OpenAITextConfig extends OpenAI SDK's ClientOptions, which supports a `fetch` parameter.\n */\nfunction buildOpenAiConfig(provider: string, config: OpenAiGatewayConfig) {\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\tfetch: createGatewayFetch(provider, config),\n\t};\n}\n\n/**\n * Creates an OpenAI chat adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n * @param model The OpenAI model to use\n * @param config Configuration options\n */\nexport function createOpenAiChat(model: OpenAIChatModel, config: OpenAiGatewayConfig) {\n\treturn new OpenAITextAdapter(buildOpenAiConfig(\"openai\", config), model);\n}\n\n/**\n * Creates an OpenAI summarize adapter which uses Cloudflare AI Gateway.\n * @param model The OpenAI model to use\n * @param config Configuration options\n */\nexport function createOpenAiSummarize(model: OpenAIChatModel, config: OpenAiGatewayConfig) {\n\treturn new OpenAISummarizeAdapter(buildOpenAiConfig(\"openai\", config), model);\n}\n\n/**\n * Creates an OpenAI image adapter which uses Cloudflare AI Gateway.\n * @param model The OpenAI image model to use\n * @param config Configuration options\n */\nexport function createOpenAiImage(model: OpenAIImageModel, config: OpenAiGatewayConfig) {\n\treturn new OpenAIImageAdapter(buildOpenAiConfig(\"openai\", config), model);\n}\n\n/**\n * Creates an OpenAI transcription adapter which uses Cloudflare AI Gateway.\n * @param model The OpenAI transcription model to use\n * @param config Configuration options\n */\nexport function createOpenAiTranscription(\n\tmodel: OpenAITranscriptionModel,\n\tconfig: OpenAiGatewayConfig,\n) {\n\treturn new OpenAITranscriptionAdapter(buildOpenAiConfig(\"openai\", config), model);\n}\n\n/**\n * Creates an OpenAI TTS adapter which uses Cloudflare AI Gateway.\n * @param model The OpenAI TTS model to use\n * @param config Configuration options\n */\nexport function createOpenAiTts(model: OpenAITTSModel, config: OpenAiGatewayConfig) {\n\treturn new OpenAITTSAdapter(buildOpenAiConfig(\"openai\", config), model);\n}\n\n/**\n * Creates an OpenAI video adapter which uses Cloudflare AI Gateway.\n * @param model The OpenAI video model to use\n * @param config Configuration options\n */\nexport function createOpenAiVideo(model: OpenAIVideoModel, config: OpenAiGatewayConfig) {\n\treturn new OpenAIVideoAdapter(buildOpenAiConfig(\"openai\", config), model);\n}\n\nexport {\n\tOPENAI_CHAT_MODELS,\n\tOPENAI_IMAGE_MODELS,\n\tOPENAI_TRANSCRIPTION_MODELS,\n\tOPENAI_TTS_MODELS,\n\tOPENAI_VIDEO_MODELS,\n\ttype OpenAIChatModel,\n\ttype OpenAIImageModel,\n\ttype OpenAITranscriptionModel,\n\ttype OpenAITTSModel,\n\ttype OpenAIVideoModel,\n};\n"]}
|
|
@@ -6,7 +6,7 @@ var _chunk7HSUHP63cjs = require('./chunk-7HSUHP63.cjs');
|
|
|
6
6
|
|
|
7
7
|
|
|
8
8
|
|
|
9
|
-
var
|
|
9
|
+
var _chunk3L6IRVAVcjs = require('./chunk-3L6IRVAV.cjs');
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
|
|
@@ -18,17 +18,17 @@ var WorkersAiSummarizeAdapter = class extends _adapters.BaseSummarizeAdapter {
|
|
|
18
18
|
super({}, model);
|
|
19
19
|
_chunk4DE2IREAcjs.__publicField.call(void 0, this, "name", "workers-ai-summarize");
|
|
20
20
|
_chunk4DE2IREAcjs.__publicField.call(void 0, this, "adapterConfig");
|
|
21
|
-
|
|
21
|
+
_chunk3L6IRVAVcjs.validateWorkersAiConfig.call(void 0, config);
|
|
22
22
|
this.adapterConfig = config;
|
|
23
23
|
}
|
|
24
24
|
async summarize(options) {
|
|
25
25
|
const { text, maxLength } = options;
|
|
26
26
|
const payload = { input_text: text };
|
|
27
27
|
if (maxLength != null) payload.max_length = maxLength;
|
|
28
|
-
if (
|
|
28
|
+
if (_chunk3L6IRVAVcjs.isDirectBindingConfig.call(void 0, this.adapterConfig)) {
|
|
29
29
|
return this.summarizeViaBinding(payload);
|
|
30
30
|
}
|
|
31
|
-
if (
|
|
31
|
+
if (_chunk3L6IRVAVcjs.isDirectCredentialsConfig.call(void 0, this.adapterConfig)) {
|
|
32
32
|
return this.summarizeViaRest(payload);
|
|
33
33
|
}
|
|
34
34
|
return this.summarizeViaGateway(payload);
|
|
@@ -48,7 +48,7 @@ var WorkersAiSummarizeAdapter = class extends _adapters.BaseSummarizeAdapter {
|
|
|
48
48
|
}
|
|
49
49
|
async summarizeViaGateway(payload) {
|
|
50
50
|
const gatewayConfig = this.adapterConfig;
|
|
51
|
-
const gatewayFetch =
|
|
51
|
+
const gatewayFetch = _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "workers-ai", gatewayConfig);
|
|
52
52
|
const response = await gatewayFetch("https://api.cloudflare.com/v1/ai/summarization", {
|
|
53
53
|
method: "POST",
|
|
54
54
|
body: JSON.stringify({
|
|
@@ -83,4 +83,4 @@ function createWorkersAiSummarize(model, config) {
|
|
|
83
83
|
|
|
84
84
|
|
|
85
85
|
exports.WorkersAiSummarizeAdapter = WorkersAiSummarizeAdapter; exports.createWorkersAiSummarize = createWorkersAiSummarize;
|
|
86
|
-
//# sourceMappingURL=chunk-
|
|
86
|
+
//# sourceMappingURL=chunk-HPMTYA35.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-HPMTYA35.cjs","../src/adapters/workers-ai-summarize.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACbA,iDAAqC;AA2B9B,IAAM,0BAAA,EAAN,MAAA,QAAwC,+BAA8C;AAAA,EAI5F,WAAA,CAAY,MAAA,EAAgC,KAAA,EAAgC;AAC3E,IAAA,KAAA,CAAM,CAAC,CAAA,EAAG,KAAK,CAAA;AAJhB,IAAA,6CAAA,IAAA,EAAS,MAAA,EAAO,sBAAA,CAAA;AAChB,IAAA,6CAAA,IAAA,EAAQ,eAAA,CAAA;AAIP,IAAA,uDAAA,MAA8B,CAAA;AAC9B,IAAA,IAAA,CAAK,cAAA,EAAgB,MAAA;AAAA,EACtB;AAAA,EAEA,MAAM,SAAA,CAAU,OAAA,EAA6D;AAC5E,IAAA,MAAM,EAAE,IAAA,EAAM,UAAU,EAAA,EAAI,OAAA;AAE5B,IAAA,MAAM,QAAA,EAAmC,EAAE,UAAA,EAAY,KAAK,CAAA;AAC5D,IAAA,GAAA,CAAI,UAAA,GAAa,IAAA,EAAM,OAAA,CAAQ,WAAA,EAAa,SAAA;AAE5C,IAAA,GAAA,CAAI,qDAAA,IAAsB,CAAK,aAAa,CAAA,EAAG;AAC9C,MAAA,OAAO,IAAA,CAAK,mBAAA,CAAoB,OAAO,CAAA;AAAA,IACxC;AAEA,IAAA,GAAA,CAAI,yDAAA,IAA0B,CAAK,aAAa,CAAA,EAAG;AAClD,MAAA,OAAO,IAAA,CAAK,gBAAA,CAAiB,OAAO,CAAA;AAAA,IACrC;AAEA,IAAA,OAAO,IAAA,CAAK,mBAAA,CAAoB,OAAO,CAAA;AAAA,EACxC;AAAA,EAEA,MAAc,mBAAA,CACb,OAAA,EAC+B;AAC/B,IAAA,MAAM,GAAA,EAAM,IAAA,CAAK,aAAA,CAA+C,OAAA;AAChE,IAAA,MAAM,OAAA,EAAU,MAAM,EAAA,CAAG,GAAA,CAAI,IAAA,CAAK,KAAA,EAAO,OAAO,CAAA;AAChD,IAAA,OAAO,IAAA,CAAK,UAAA,kBAAY,MAAA,CAAO,OAAA,UAAsB,IAAE,CAAA;AAAA,EACxD;AAAA,EAEA,MAAc,gBAAA,CAAiB,OAAA,EAAgE;AAC9F,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AACpB,IAAA,MAAM,SAAA,EAAW,MAAM,kDAAA,MAAmB,EAAQ,IAAA,CAAK,KAAA,EAAO,OAAA,EAAS;AAAA,MACtE,KAAA,EAAO;AAAA,IACR,CAAC,CAAA;AAED,IAAA,MAAM,KAAA,EAAQ,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AAClC,IAAA,OAAO,IAAA,CAAK,UAAA,kCAAW,IAAA,mBAAK,MAAA,6BAAQ,SAAA,UAAW,IAAE,CAAA;AAAA,EAClD;AAAA,EAEA,MAAc,mBAAA,CACb,OAAA,EAC+B;AAC/B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,aAAA;AAC3B,IAAA,MAAM,aAAA,EAAe,kDAAA,YAAmB,EAAc,aAAa,CAAA;AAKnE,IAAA,MAAM,SAAA,EAAW,MAAM,YAAA,CAAa,gDAAA,EAAkD;AAAA,MACrF,MAAA,EAAQ,MAAA;AAAA,MACR,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU;AAAA,QACpB,KAAA,EAAO,IAAA,CAAK,KAAA;AAAA,QACZ,GAAG;AAAA,MACJ,CAAC;AAAA,IACF,CAAC,CAAA;AAED,IAAA,GAAA,CAAI,CAAC,QAAA,CAAS,EAAA,EAAI;AACjB,MAAA,MAAM,UAAA,EAAY,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AACtC,MAAA,MAAM,IAAI,KAAA;AAAA,QACT,CAAA,6CAAA,EAAgD,QAAA,CAAS,MAAM,CAAA,GAAA,EAAM,SAAS,CAAA;AAAA,MAAA;AAC/E,IAAA;AAGD,IAAA;AACA,IAAA;AAAiE,EAAA;AAClE,EAAA;AAGC,IAAA;AAAO,MAAA;AACc,MAAA;AACR,MAAA;AACZ;AAAA,MAAA;AAE8D,IAAA;AAC/D,EAAA;AAEF;AAyBO;AAIN,EAAA;AACD;AD3DA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-HPMTYA35.cjs","sourcesContent":[null,"import { BaseSummarizeAdapter } from \"@tanstack/ai/adapters\";\nimport type { SummarizationOptions, SummarizationResult } from \"@tanstack/ai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype WorkersAiDirectCredentialsConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch } from \"../utils/workers-ai-rest\";\n\n// ---------------------------------------------------------------------------\n// Model types\n// ---------------------------------------------------------------------------\n\n/**\n * Workers AI models that support summarization.\n */\nexport type WorkersAiSummarizeModel = \"@cf/facebook/bart-large-cnn\" | (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiSummarizeAdapter\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiSummarizeAdapter extends BaseSummarizeAdapter<WorkersAiSummarizeModel> {\n\treadonly name = \"workers-ai-summarize\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiSummarizeModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n\t\tconst { text, maxLength } = options;\n\n\t\tconst payload: Record<string, unknown> = { input_text: text };\n\t\tif (maxLength != null) payload.max_length = maxLength;\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.summarizeViaBinding(payload);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\treturn this.summarizeViaRest(payload);\n\t\t}\n\n\t\treturn this.summarizeViaGateway(payload);\n\t}\n\n\tprivate async summarizeViaBinding(\n\t\tpayload: Record<string, unknown>,\n\t): Promise<SummarizationResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = (await ai.run(this.model, payload)) as Record<string, unknown>;\n\t\treturn this.wrapResult((result.summary as string) ?? \"\");\n\t}\n\n\tprivate async summarizeViaRest(payload: Record<string, unknown>): Promise<SummarizationResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst response = await workersAiRestFetch(config, this.model, payload, {\n\t\t\tlabel: \"Workers AI summarize\",\n\t\t});\n\n\t\tconst data = (await response.json()) as { result?: { summary?: string } };\n\t\treturn this.wrapResult(data.result?.summary ?? \"\");\n\t}\n\n\tprivate async summarizeViaGateway(\n\t\tpayload: Record<string, unknown>,\n\t): Promise<SummarizationResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/ai/summarization\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\t...payload,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI summarize gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst data = (await response.json()) as { result?: { summary?: string }; summary?: string };\n\t\treturn this.wrapResult(data.result?.summary ?? data.summary ?? \"\");\n\t}\n\n\tprivate wrapResult(summary: string): SummarizationResult {\n\t\treturn {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\tsummary,\n\t\t\t// BART-large-CNN doesn't return token usage\n\t\t\tusage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },\n\t\t};\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI summarization adapter.\n *\n * Works with TanStack AI's `summarize()` activity function:\n * ```ts\n * import { summarize } from \"@tanstack/ai\";\n * import { createWorkersAiSummarize } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiSummarize(\"@cf/facebook/bart-large-cnn\", {\n * binding: env.AI,\n * });\n *\n * const result = await summarize({ adapter, text: \"Long article here...\" });\n * // result.summary\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiSummarize(\n\tmodel: WorkersAiSummarizeModel,\n\tconfig: WorkersAiAdapterConfig,\n) {\n\treturn new WorkersAiSummarizeAdapter(config, model);\n}\n"]}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
|
|
5
5
|
|
|
6
6
|
|
|
7
|
-
var
|
|
7
|
+
var _chunk3L6IRVAVcjs = require('./chunk-3L6IRVAV.cjs');
|
|
8
8
|
|
|
9
9
|
|
|
10
10
|
var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
|
|
@@ -15,18 +15,18 @@ var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
|
|
|
15
15
|
var _adapters = require('@tanstack/ai/adapters');
|
|
16
16
|
var _openai = require('openai'); var _openai2 = _interopRequireDefault(_openai);
|
|
17
17
|
function buildWorkersAiClient(config) {
|
|
18
|
-
|
|
18
|
+
_chunk3L6IRVAVcjs.validateWorkersAiConfig.call(void 0, config);
|
|
19
19
|
const sessionHeaders = config.sessionAffinity ? { "x-session-affinity": config.sessionAffinity } : void 0;
|
|
20
|
-
if (
|
|
20
|
+
if (_chunk3L6IRVAVcjs.isDirectBindingConfig.call(void 0, config)) {
|
|
21
21
|
return new (0, _openai2.default)({
|
|
22
22
|
apiKey: "unused",
|
|
23
|
-
fetch:
|
|
23
|
+
fetch: _chunk3L6IRVAVcjs.createWorkersAiBindingFetch.call(void 0,
|
|
24
24
|
config.binding,
|
|
25
25
|
sessionHeaders ? { extraHeaders: sessionHeaders } : void 0
|
|
26
26
|
)
|
|
27
27
|
});
|
|
28
28
|
}
|
|
29
|
-
if (
|
|
29
|
+
if (_chunk3L6IRVAVcjs.isDirectCredentialsConfig.call(void 0, config)) {
|
|
30
30
|
return new (0, _openai2.default)({
|
|
31
31
|
baseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,
|
|
32
32
|
apiKey: config.apiKey,
|
|
@@ -35,7 +35,7 @@ function buildWorkersAiClient(config) {
|
|
|
35
35
|
}
|
|
36
36
|
const gatewayConfig = config;
|
|
37
37
|
return new (0, _openai2.default)({
|
|
38
|
-
fetch:
|
|
38
|
+
fetch: _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "workers-ai", gatewayConfig, sessionHeaders),
|
|
39
39
|
apiKey: _nullishCoalesce(gatewayConfig.apiKey, () => ( "unused"))
|
|
40
40
|
});
|
|
41
41
|
}
|
|
@@ -518,4 +518,4 @@ function createWorkersAiChat(model, config) {
|
|
|
518
518
|
|
|
519
519
|
|
|
520
520
|
exports.WorkersAiTextAdapter = WorkersAiTextAdapter; exports.createWorkersAiChat = createWorkersAiChat;
|
|
521
|
-
//# sourceMappingURL=chunk-
|
|
521
|
+
//# sourceMappingURL=chunk-IQAGFHRT.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-DTGQYEJ2.cjs","../src/adapters/workers-ai.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACTA;AACC;AAAA,iDAGM;AACP,gFAAmB;AAyBnB,SAAS,oBAAA,CAAqB,MAAA,EAAwC;AACrE,EAAA,uDAAA,MAA8B,CAAA;AAE9B,EAAA,MAAM,eAAA,EAAqD,MAAA,CAAO,gBAAA,EAC/D,EAAE,oBAAA,EAAsB,MAAA,CAAO,gBAAgB,EAAA,EAC/C,KAAA,CAAA;AAEH,EAAA,GAAA,CAAI,qDAAA,MAA4B,CAAA,EAAG;AAElC,IAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,MACjB,MAAA,EAAQ,QAAA;AAAA,MACR,KAAA,EAAO,2DAAA;AAAA,QACN,MAAA,CAAO,OAAA;AAAA,QACP,eAAA,EAAiB,EAAE,YAAA,EAAc,eAAe,EAAA,EAAI,KAAA;AAAA,MACrD;AAAA,IACD,CAAC,CAAA;AAAA,EACF;AAEA,EAAA,GAAA,CAAI,yDAAA,MAAgC,CAAA,EAAG;AAEtC,IAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,MACjB,OAAA,EAAS,CAAA,8CAAA,EAAiD,MAAA,CAAO,SAAS,CAAA,MAAA,CAAA;AAAA,MAC1E,MAAA,EAAQ,MAAA,CAAO,MAAA;AAAA,MACf,cAAA,EAAgB;AAAA,IACjB,CAAC,CAAA;AAAA,EACF;AAGA,EAAA,MAAM,cAAA,EAAgB,MAAA;AACtB,EAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,IACjB,KAAA,EAAO,kDAAA,YAAmB,EAAc,aAAA,EAAe,cAAc,CAAA;AAAA,IACrE,MAAA,mBAAQ,aAAA,CAAc,MAAA,UAAU;AAAA,EACjC,CAAC,CAAA;AACF;AAMA,SAAS,kBAAA,CAAmB,OAAA,EAA0C;AACrE,EAAA,GAAA,CAAI,QAAA,IAAY,IAAA,EAAM,OAAO,EAAA;AAC7B,EAAA,GAAA,CAAI,OAAO,QAAA,IAAY,QAAA,EAAU,OAAO,OAAA;AACxC,EAAA,OAAO,OAAA,CAAQ,OAAA,CAAQ,CAAC,CAAA,EAAA,GAAO,CAAA,CAAE,KAAA,IAAS,OAAA,EAAS,CAAC,CAAA,CAAE,OAAO,EAAA,EAAI,CAAC,CAAE,CAAA,CAAE,IAAA,CAAK,EAAE,CAAA;AAC9E;AAQA,SAAS,kBAAA,CAAmB,IAAA,EAAsE;AACjG,EAAA,OAAA,CAAQ,IAAA,CAAK,IAAA,EAAM;AAAA,IAClB,KAAK,MAAA;AACJ,MAAA,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS;AACjB,QAAA,OAAO,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,IAAA,CAAK,QAAQ,CAAA;AAAA,MAC3C;AACA,MAAA,OAAO,KAAA,CAAA;AAAA,IACR,KAAK,OAAA,EAAS;AACb,MAAA,IAAI,GAAA;AACJ,MAAA,GAAA,CAAI,IAAA,CAAK,MAAA,CAAO,KAAA,IAAS,MAAA,EAAQ;AAChC,QAAA,IAAA,EAAM,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,UAAA,CAAW,OAAO,EAAA,EACvC,IAAA,CAAK,MAAA,CAAO,MAAA,EACZ,CAAA,KAAA,EAAQ,IAAA,CAAK,MAAA,CAAO,QAAQ,CAAA,QAAA,EAAW,IAAA,CAAK,MAAA,CAAO,KAAK,CAAA,CAAA;AACrD,MAAA;AACY,QAAA;AACnB,MAAA;AAC+C,MAAA;AAChD,IAAA;AACA,IAAA;AAES,MAAA;AAC8D,QAAA;AACtE,MAAA;AACO,MAAA;AACT,EAAA;AACD;AAWoD;AACtB,EAAA;AACW,EAAA;AAEgB,EAAA;AACxC,EAAA;AAC8D,IAAA;AAC9E,EAAA;AAEwD,EAAA;AAC5B,EAAA;AACc,IAAA;AAC1B,IAAA;AACM,MAAA;AACrB,IAAA;AACD,EAAA;AACO,EAAA;AACR;AAM4C;AACU,EAAA;AACa,EAAA;AAEnB,EAAA;AAC1B,IAAA;AACb,MAAA;AAC0B,MAAA;AAChC,IAAA;AACF,EAAA;AAEgC,EAAA;AACF,IAAA;AACR,MAAA;AACb,QAAA;AACmC,QAAA;AACzC,MAAA;AACuC,IAAA;AACkC,MAAA;AACnE,QAAA;AACqC,QAAA;AAC5C,MAAA;AACuE,MAAA;AACT,QAAA;AACrD,UAAA;AACD,UAAA;AACI,UAAA;AACS,YAAA;AACK,YAAA;AACxB,UAAA;AACC,QAAA;AACH,MAAA;AACoC,MAAA;AACe,IAAA;AAC/C,MAAA;AACqC,MAAA;AACpC,QAAA;AACuB,UAAA;AACJ,UAAA;AACf,QAAA;AACqC,UAAA;AAC7C,QAAA;AACM,MAAA;AACsC,QAAA;AAC7C,MAAA;AACoB,MAAA;AACb,QAAA;AACqE,QAAA;AAClE,QAAA;AACT,MAAA;AACF,IAAA;AACD,EAAA;AAEO,EAAA;AACR;AAIgD;AAC5B,EAAA;AACS,EAAA;AACrB,IAAA;AACI,IAAA;AACE,MAAA;AACO,MAAA;AACD,MAAA;AAClB,IAAA;AACC,EAAA;AACH;AAMiD;AACsB,EAAA;AACvE;AAeE;AAK0D,EAAA;AACzB,IAAA;AAL3B,IAAA;AAEC,IAAA;AAIkC,IAAA;AAC1C,EAAA;AAEyE,EAAA;AACT,IAAA;AAEG,IAAA;AACxB,IAAA;AAEf,IAAA;AACF,IAAA;AACI,IAAA;AACF,IAAA;AACM,IAAA;AACR,IAAA;AACG,IAAA;AACD,IAAA;AACD,IAAA;AACI,IAAA;AAI5B,IAAA;AAEE,IAAA;AACC,MAAA;AACA,MAAA;AACgD,QAAA;AAC7B,UAAA;AACX,UAAA;AACH,UAAA;AACP,UAAA;AACQ,UAAA;AAC8B,UAAA;AACtC,QAAA;AAC6B,MAAA;AAGtB,QAAA;AACP,UAAA;AACqD,UAAA;AACtD,QAAA;AACkE,QAAA;AAC5C,UAAA;AACX,UAAA;AACH,UAAA;AACP,UAAA;AACA,QAAA;AAEK,QAAA;AACC,UAAA;AACN,UAAA;AAC8C,UAAA;AAC9C,UAAA;AACD,QAAA;AAEwC,QAAA;AACtB,QAAA;AACX,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACM,YAAA;AACP,UAAA;AACM,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACW,YAAA;AACE,YAAA;AACd,UAAA;AACM,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACD,UAAA;AACD,QAAA;AAEqB,QAAA;AACa,UAAA;AACJ,YAAA;AACd,YAAA;AACc,YAAA;AACxB,YAAA;AACsD,cAAA;AAClD,YAAA;AACQ,cAAA;AAChB,YAAA;AACM,YAAA;AACC,cAAA;AACS,cAAA;AACF,cAAA;AACiC,cAAA;AAC9C,cAAA;AACO,cAAA;AACR,YAAA;AACM,YAAA;AACC,cAAA;AACS,cAAA;AACF,cAAA;AACiC,cAAA;AAC9C,cAAA;AACO,cAAA;AACR,YAAA;AACD,UAAA;AACD,QAAA;AAEiD,QAAA;AAC3C,QAAA;AACC,UAAA;AACN,UAAA;AAC8C,UAAA;AAC9C,UAAA;AAEG,UAAA;AACoC,YAAA;AACI,YAAA;AACL,YAAA;AAEnC,UAAA;AAI4D,UAAA;AAChE,QAAA;AACA,QAAA;AACD,MAAA;AAEkC,MAAA;AACiB,QAAA;AACpB,QAAA;AACjB,QAAA;AAGc,QAAA;AACH,UAAA;AACjB,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AACD,UAAA;AACD,QAAA;AAEqB,QAAA;AAKe,QAAA;AACd,QAAA;AAEO,UAAA;AACH,YAAA;AAClB,YAAA;AACC,cAAA;AACN,cAAA;AACU,cAAA;AAC0B,cAAA;AACpC,cAAA;AACD,YAAA;AACD,UAAA;AACwB,UAAA;AAOlB,UAAA;AACC,YAAA;AACN,YAAA;AACO,YAAA;AACE,YAAA;AAC2B,YAAA;AACpC,YAAA;AACD,UAAA;AACD,QAAA;AAGmB,QAAA;AACe,UAAA;AACH,YAAA;AACvB,YAAA;AACC,cAAA;AACN,cAAA;AACoC,cAAA;AACpC,cAAA;AACM,cAAA;AACP,YAAA;AACD,UAAA;AAE4B,UAAA;AACtB,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AACa,YAAA;AACJ,YAAA;AACV,UAAA;AACD,QAAA;AAGsB,QAAA;AACyB,UAAA;AACjB,YAAA;AAES,YAAA;AAIL,cAAA;AACA,gBAAA;AACQ,gBAAA;AAC3B,gBAAA;AACF,gBAAA;AACT,cAAA;AACF,YAAA;AAE8C,YAAA;AAKZ,YAAA;AACM,cAAA;AACxC,YAAA;AACuC,YAAA;AACO,cAAA;AAC9C,YAAA;AAGuD,YAAA;AACnC,cAAA;AACb,cAAA;AACC,gBAAA;AACe,gBAAA;AACF,gBAAA;AACiB,gBAAA;AACpC,gBAAA;AACA,gBAAA;AACD,cAAA;AACD,YAAA;AAG2D,YAAA;AACpD,cAAA;AACC,gBAAA;AACe,gBAAA;AACe,gBAAA;AACpC,gBAAA;AAC8B,gBAAA;AAC/B,cAAA;AACD,YAAA;AACD,UAAA;AACD,QAAA;AAG0B,QAAA;AACC,UAAA;AAGiD,UAAA;AAC1B,YAAA;AACnB,cAAA;AACxB,cAAA;AAGC,gBAAA;AACG,cAAA;AACQ,gBAAA;AAChB,cAAA;AACM,cAAA;AACC,gBAAA;AACe,gBAAA;AACF,gBAAA;AACiB,gBAAA;AACpC,gBAAA;AACO,gBAAA;AACR,cAAA;AACD,YAAA;AACD,UAAA;AAKC,UAAA;AAK+B,UAAA;AACzB,YAAA;AACC,cAAA;AACN,cAAA;AACoC,cAAA;AACpC,cAAA;AACD,YAAA;AACD,UAAA;AAGM,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AAEG,YAAA;AAC0B,cAAA;AACI,cAAA;AACL,cAAA;AAEzB,YAAA;AACW,YAAA;AACf,UAAA;AACD,QAAA;AACD,MAAA;AAKsD,MAAA;AAC7C,QAAA;AACP,UAAA;AACD,QAAA;AAGgD,QAAA;AACzB,UAAA;AACO,YAAA;AACxB,YAAA;AACkE,cAAA;AAC9D,YAAA;AACQ,cAAA;AAChB,YAAA;AACM,YAAA;AACC,cAAA;AACe,cAAA;AACF,cAAA;AACE,cAAA;AACrB,cAAA;AACO,cAAA;AACR,YAAA;AACD,UAAA;AACD,QAAA;AAGgC,QAAA;AACzB,UAAA;AACC,YAAA;AACN,YAAA;AACqB,YAAA;AACrB,YAAA;AACD,UAAA;AACD,QAAA;AAEM,QAAA;AACC,UAAA;AACN,UAAA;AACqB,UAAA;AACrB,UAAA;AACc,UAAA;AACf,QAAA;AACD,MAAA;AACe,IAAA;AACsD,MAAA;AAEC,MAAA;AAC3C,MAAA;AACpB,QAAA;AACC,UAAA;AACN,UAAA;AACqB,UAAA;AACrB,UAAA;AACD,QAAA;AACD,MAAA;AACM,MAAA;AACC,QAAA;AACN,QAAA;AACqB,QAAA;AACrB,QAAA;AACO,QAAA;AACc,UAAA;AACpB,UAAA;AACD,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAI4C,EAAA;AACL,IAAA;AACkB,IAAA;AAEY,IAAA;AAC9C,MAAA;AACrB,IAAA;AAE0D,IAAA;AACrC,MAAA;AACX,MAAA;AACV,MAAA;AACQ,MAAA;AACS,MAAA;AACV,QAAA;AACO,QAAA;AACN,UAAA;AACE,UAAA;AACA,UAAA;AACT,QAAA;AACD,MAAA;AACA,IAAA;AAEkC,IAAA;AAEtB,IAAA;AACF,MAAA;AACoE,QAAA;AAC9E,MAAA;AACD,IAAA;AAE8C,IAAA;AAI1C,IAAA;AACA,IAAA;AAEgC,IAAA;AACzB,MAAA;AACN,MAAA;AACsB,QAAA;AAClB,MAAA;AACA,QAAA;AACR,MAAA;AACM,IAAA;AAEC,MAAA;AAC4B,MAAA;AACpC,IAAA;AAEuB,IAAA;AACxB,EAAA;AACD;AAM+F;AACjD,EAAA;AAC9C;ADrLmI;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-DTGQYEJ2.cjs","sourcesContent":[null,"import type { AiModels, BaseAiTextGeneration } from \"@cloudflare/workers-types\";\nimport type { ContentPart, ModelMessage, StreamChunk, TextOptions } from \"@tanstack/ai\";\nimport {\n\tBaseTextAdapter,\n\ttype StructuredOutputOptions,\n\ttype StructuredOutputResult,\n} from \"@tanstack/ai/adapters\";\nimport OpenAI from \"openai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tcreateWorkersAiBindingFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model types derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiTextModel =\n\t| {\n\t\t\t[K in keyof AiModels]: AiModels[K] extends BaseAiTextGeneration ? K : never;\n\t }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// Helpers: build the right OpenAI client depending on config mode\n// ---------------------------------------------------------------------------\n\nfunction buildWorkersAiClient(config: WorkersAiAdapterConfig): OpenAI {\n\tvalidateWorkersAiConfig(config);\n\n\tconst sessionHeaders: Record<string, string> | undefined = config.sessionAffinity\n\t\t? { \"x-session-affinity\": config.sessionAffinity }\n\t\t: undefined;\n\n\tif (isDirectBindingConfig(config)) {\n\t\t// Plain binding mode: shim translates OpenAI fetch calls to env.AI.run()\n\t\treturn new OpenAI({\n\t\t\tapiKey: \"unused\",\n\t\t\tfetch: createWorkersAiBindingFetch(\n\t\t\t\tconfig.binding,\n\t\t\t\tsessionHeaders ? { extraHeaders: sessionHeaders } : undefined,\n\t\t\t),\n\t\t});\n\t}\n\n\tif (isDirectCredentialsConfig(config)) {\n\t\t// Plain REST mode: point OpenAI SDK at Workers AI's OpenAI-compatible endpoint\n\t\treturn new OpenAI({\n\t\t\tbaseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,\n\t\t\tapiKey: config.apiKey,\n\t\t\tdefaultHeaders: sessionHeaders,\n\t\t});\n\t}\n\n\t// Gateway mode (existing): use createGatewayFetch\n\tconst gatewayConfig = config as AiGatewayAdapterConfig;\n\treturn new OpenAI({\n\t\tfetch: createGatewayFetch(\"workers-ai\", gatewayConfig, sessionHeaders),\n\t\tapiKey: gatewayConfig.apiKey ?? \"unused\",\n\t});\n}\n\n// ---------------------------------------------------------------------------\n// Shared message-building helpers\n// ---------------------------------------------------------------------------\n\nfunction extractTextContent(content: ModelMessage[\"content\"]): string {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n}\n\n/**\n * Convert a single TanStack AI {@link ContentPart} to the OpenAI Chat\n * Completions multi-modal format.\n *\n * TODO: handle other content types (audio, video, document)\n */\nfunction convertContentPart(part: ContentPart): OpenAI.Chat.ChatCompletionContentPart | undefined {\n\tswitch (part.type) {\n\t\tcase \"text\":\n\t\t\tif (part.content) {\n\t\t\t\treturn { type: \"text\", text: part.content };\n\t\t\t}\n\t\t\treturn undefined;\n\t\tcase \"image\": {\n\t\t\tlet url: string;\n\t\t\tif (part.source.type === \"data\") {\n\t\t\t\turl = part.source.value.startsWith(\"data:\")\n\t\t\t\t\t? part.source.value\n\t\t\t\t\t: `data:${part.source.mimeType};base64,${part.source.value}`;\n\t\t\t} else {\n\t\t\t\turl = part.source.value;\n\t\t\t}\n\t\t\treturn { type: \"image_url\", image_url: { url } };\n\t\t}\n\t\tdefault:\n\t\t\t// audio, video, document — not supported for now\n\t\t\tconsole.warn(\n\t\t\t\t`[@cloudflare/tanstack-ai] Unsupported content part type \"${part.type}\" — skipping`,\n\t\t\t);\n\t\t\treturn undefined;\n\t}\n}\n\n/**\n * Build OpenAI-compatible user message content from TanStack AI content.\n *\n * If the content has only text parts, returns a plain string.\n * If it includes image parts, returns an array of content parts in\n * OpenAI's multi-modal format (text + image_url).\n */\nfunction buildUserContent(\n\tcontent: ModelMessage[\"content\"],\n): string | OpenAI.Chat.ChatCompletionContentPart[] {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\n\tconst hasImages = content.some((p) => p.type === \"image\");\n\tif (!hasImages) {\n\t\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n\t}\n\n\tconst parts: OpenAI.Chat.ChatCompletionContentPart[] = [];\n\tfor (const part of content) {\n\t\tconst converted = convertContentPart(part);\n\t\tif (converted) {\n\t\t\tparts.push(converted);\n\t\t}\n\t}\n\treturn parts;\n}\n\nfunction buildOpenAIMessages(\n\tsystemPrompts: string[] | undefined,\n\tmessages: ModelMessage[],\n\toptions?: { includeToolMessages?: boolean },\n): OpenAI.Chat.ChatCompletionMessageParam[] {\n\tconst includeTools = options?.includeToolMessages ?? true;\n\tconst openAIMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [];\n\n\tif (systemPrompts && systemPrompts.length > 0) {\n\t\topenAIMessages.push({\n\t\t\trole: \"system\",\n\t\t\tcontent: systemPrompts.join(\"\\n\"),\n\t\t});\n\t}\n\n\tfor (const message of messages) {\n\t\tif (message.role === \"user\") {\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"user\",\n\t\t\t\tcontent: buildUserContent(message.content),\n\t\t\t});\n\t\t} else if (message.role === \"assistant\") {\n\t\t\tconst assistantMessage: OpenAI.Chat.ChatCompletionAssistantMessageParam = {\n\t\t\t\trole: \"assistant\",\n\t\t\t\tcontent: extractTextContent(message.content),\n\t\t\t};\n\t\t\tif (includeTools && message.toolCalls && message.toolCalls.length > 0) {\n\t\t\t\tassistantMessage.tool_calls = message.toolCalls.map((tc) => ({\n\t\t\t\t\tid: tc.id,\n\t\t\t\t\ttype: \"function\" as const,\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: tc.function.name,\n\t\t\t\t\t\targuments: tc.function.arguments,\n\t\t\t\t\t},\n\t\t\t\t}));\n\t\t\t}\n\t\t\topenAIMessages.push(assistantMessage);\n\t\t} else if (includeTools && message.role === \"tool\") {\n\t\t\tlet toolContent: string;\n\t\t\tif (typeof message.content === \"string\") {\n\t\t\t\ttry {\n\t\t\t\t\tJSON.parse(message.content);\n\t\t\t\t\ttoolContent = message.content;\n\t\t\t\t} catch {\n\t\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t}\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"tool\",\n\t\t\t\ttool_call_id: message.toolCallId || `tool_${crypto.randomUUID().slice(0, 8)}`,\n\t\t\t\tcontent: toolContent,\n\t\t\t});\n\t\t}\n\t}\n\n\treturn openAIMessages;\n}\n\nfunction buildOpenAITools(\n\ttools: Array<{ name: string; description: string; inputSchema?: unknown }> | undefined,\n): OpenAI.Chat.ChatCompletionTool[] | undefined {\n\tif (!tools) return undefined;\n\treturn tools.map((tool) => ({\n\t\ttype: \"function\" as const,\n\t\tfunction: {\n\t\t\tname: tool.name,\n\t\t\tdescription: tool.description,\n\t\t\tparameters: tool.inputSchema as Record<string, unknown>,\n\t\t},\n\t}));\n}\n\n// ---------------------------------------------------------------------------\n// ID generation\n// ---------------------------------------------------------------------------\n\nfunction generateId(prefix = \"chatcmpl\"): string {\n\treturn `${prefix}-${crypto.randomUUID().replace(/-/g, \"\").slice(0, 16)}`;\n}\n\n// ---------------------------------------------------------------------------\n// WorkersAiTextAdapter: chat / structured output via OpenAI Chat Completions\n// ---------------------------------------------------------------------------\n\n// TODO: Replace `any` generic params with proper types once BaseTextAdapter's\n// provider-options generics stabilize. Workers AI doesn't have provider-specific\n// options in the TanStack sense, so `any` is pragmatic for now.\nexport class WorkersAiTextAdapter<TModel extends WorkersAiTextModel> extends BaseTextAdapter<\n\tTModel,\n\t// eslint-disable-next-line @typescript-eslint/no-explicit-any -- BaseTextAdapter generic params are opaque\n\tany,\n\tany,\n\tany\n> {\n\tname = \"workers-ai\" as const;\n\n\tprivate client: OpenAI;\n\n\tconstructor(model: TModel, config: WorkersAiAdapterConfig) {\n\t\tsuper({ apiKey: \"unused\" }, model);\n\t\tthis.client = buildWorkersAiClient(config);\n\t}\n\n\tasync *chatStream(options: TextOptions<any>): AsyncIterable<StreamChunk> {\n\t\tconst { systemPrompts, messages, tools, temperature, model } = options;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages);\n\t\tconst openAITools = buildOpenAITools(tools);\n\n\t\tconst timestamp = Date.now();\n\t\tconst runId = generateId();\n\t\tconst messageId = generateId();\n\t\tlet hasEmittedRunStarted = false;\n\t\tlet hasEmittedTextMessageStart = false;\n\t\tlet accumulatedContent = \"\";\n\t\tlet hasEmittedStepStarted = false;\n\t\tlet accumulatedReasoning = \"\";\n\t\tconst stepId = generateId();\n\t\tlet hasReceivedFinishReason = false;\n\t\tconst toolCallsInProgress = new Map<\n\t\t\tnumber,\n\t\t\t{ id: string; name: string; arguments: string; started: boolean }\n\t\t>();\n\n\t\ttry {\n\t\t\tlet stream: AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>;\n\t\t\ttry {\n\t\t\t\tstream = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t\tstream: true,\n\t\t\t\t\tstream_options: { include_usage: true },\n\t\t\t\t});\n\t\t\t} catch (streamError: unknown) {\n\t\t\t\t// Some models (e.g. GPT-OSS) don't support streaming via the REST API.\n\t\t\t\t// Fall back to a non-streaming call and yield the result as a single chunk.\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Streaming failed, falling back to non-streaming:\",\n\t\t\t\t\tstreamError instanceof Error ? streamError.message : streamError,\n\t\t\t\t);\n\t\t\t\tconst nonStreamResult = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t});\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\n\t\t\t\tconst msg = nonStreamResult.choices[0]?.message;\n\t\t\t\tif (msg?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: msg.content,\n\t\t\t\t\t\tcontent: msg.content,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tif (msg?.tool_calls) {\n\t\t\t\t\tfor (const tc of msg.tool_calls) {\n\t\t\t\t\t\tif (tc.type !== \"function\") continue;\n\t\t\t\t\t\tconst fn = tc.function;\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = fn.arguments ? JSON.parse(fn.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tconst finishReason = nonStreamResult.choices[0]?.finish_reason;\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tusage: nonStreamResult.usage\n\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\tpromptTokens: nonStreamResult.usage.prompt_tokens,\n\t\t\t\t\t\t\t\tcompletionTokens: nonStreamResult.usage.completion_tokens,\n\t\t\t\t\t\t\t\ttotalTokens: nonStreamResult.usage.total_tokens,\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t: undefined,\n\t\t\t\t\tfinishReason:\n\t\t\t\t\t\tfinishReason === \"tool_calls\" || finishReason === \"function_call\"\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: ((finishReason as \"stop\" | \"length\" | \"content_filter\") ?? \"stop\"),\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tfor await (const chunk of stream) {\n\t\t\t\tif (!chunk.choices || chunk.choices.length === 0) continue;\n\t\t\t\tconst choice = chunk.choices[0];\n\t\t\t\tif (!choice) continue;\n\n\t\t\t\t// Emit RUN_STARTED on first chunk\n\t\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\t\thasEmittedRunStarted = true;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tconst delta = choice.delta;\n\n\t\t\t\t// Reasoning content (used by models like QwQ, DeepSeek R1, Kimi K2.5)\n\t\t\t\t// The OpenAI SDK doesn't type this field, but models send it as an extension.\n\t\t\t\tconst reasoningContent = ((delta as Record<string, unknown>).reasoning_content ??\n\t\t\t\t\t(delta as Record<string, unknown>).reasoning) as string | undefined;\n\t\t\t\tif (reasoningContent) {\n\t\t\t\t\t// RUN_STARTED is already guaranteed by the guard above\n\t\t\t\t\tif (!hasEmittedStepStarted) {\n\t\t\t\t\t\thasEmittedStepStarted = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"STEP_STARTED\",\n\t\t\t\t\t\t\tstepId,\n\t\t\t\t\t\t\tstepType: \"thinking\",\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t\taccumulatedReasoning += reasoningContent;\n\t\t\t\t\t// TODO: TanStack AI's StreamProcessor currently treats STEP_FINISHED as an\n\t\t\t\t\t// incremental reasoning event (with `delta` + accumulated `content`), so we\n\t\t\t\t\t// emit one per token. If TanStack AI adds a dedicated STEP_CONTENT event\n\t\t\t\t\t// type, this should be updated to emit STEP_CONTENT per token and a single\n\t\t\t\t\t// STEP_FINISHED when reasoning ends (i.e. when the first non-reasoning\n\t\t\t\t\t// content or finish_reason arrives).\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"STEP_FINISHED\",\n\t\t\t\t\t\tstepId,\n\t\t\t\t\t\tdelta: reasoningContent,\n\t\t\t\t\t\tcontent: accumulatedReasoning,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Text content\n\t\t\t\tif (delta.content) {\n\t\t\t\t\tif (!hasEmittedTextMessageStart) {\n\t\t\t\t\t\thasEmittedTextMessageStart = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\taccumulatedContent += delta.content;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: delta.content,\n\t\t\t\t\t\tcontent: accumulatedContent,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Tool calls\n\t\t\t\tif (delta.tool_calls) {\n\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\tif (!toolCallsInProgress.has(index)) {\n\t\t\t\t\t\t\t// Always generate a unique ID per tool call index.\n\t\t\t\t\t\t\t// The backend may send the same ID for multiple tool calls,\n\t\t\t\t\t\t\t// so we cannot trust toolCallDelta.id to be unique.\n\t\t\t\t\t\t\ttoolCallsInProgress.set(index, {\n\t\t\t\t\t\t\t\tid: generateId(\"chatcmpl-tool\"),\n\t\t\t\t\t\t\t\tname: toolCallDelta.function?.name || \"\",\n\t\t\t\t\t\t\t\targuments: \"\",\n\t\t\t\t\t\t\t\tstarted: false,\n\t\t\t\t\t\t\t});\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tconst toolCall = toolCallsInProgress.get(index)!;\n\n\t\t\t\t\t\t// Only update name if provided (ID is already set at creation time\n\t\t\t\t\t\t// and should not be overwritten by subsequent chunks that may have\n\t\t\t\t\t\t// duplicate/shared IDs from the backend)\n\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\ttoolCall.name = toolCallDelta.function.name;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\ttoolCall.arguments += toolCallDelta.function.arguments;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Emit TOOL_CALL_START once we have id and name\n\t\t\t\t\t\tif (toolCall.id && toolCall.name && !toolCall.started) {\n\t\t\t\t\t\t\ttoolCall.started = true;\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tindex,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Stream tool call arguments\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments && toolCall.started) {\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_ARGS\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tdelta: toolCallDelta.function.arguments,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Finish\n\t\t\t\tif (choice.finish_reason) {\n\t\t\t\t\thasReceivedFinishReason = true;\n\n\t\t\t\t\t// End tool calls\n\t\t\t\t\tif (choice.finish_reason === \"tool_calls\" || toolCallsInProgress.size > 0) {\n\t\t\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\t\tparsedInput = toolCall.arguments\n\t\t\t\t\t\t\t\t\t? JSON.parse(toolCall.arguments)\n\t\t\t\t\t\t\t\t\t: {};\n\t\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\n\t\t\t\t\tconst computedFinishReason =\n\t\t\t\t\t\tchoice.finish_reason === \"tool_calls\" ||\n\t\t\t\t\t\tchoice.finish_reason === \"function_call\" ||\n\t\t\t\t\t\ttoolCallsInProgress.size > 0\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: (choice.finish_reason as \"stop\" | \"length\" | \"content_filter\");\n\n\t\t\t\t\t// End text message if started\n\t\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Emit RUN_FINISHED\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tusage: chunk.usage\n\t\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\t\tpromptTokens: chunk.usage.prompt_tokens,\n\t\t\t\t\t\t\t\t\tcompletionTokens: chunk.usage.completion_tokens,\n\t\t\t\t\t\t\t\t\ttotalTokens: chunk.usage.total_tokens,\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t: undefined,\n\t\t\t\t\t\tfinishReason: computedFinishReason,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Premature stream termination: the stream ended without a finish_reason.\n\t\t\t// This can happen when Workers AI truncates a response or the connection drops.\n\t\t\t// Emit proper closing events so the consumer doesn't hang.\n\t\t\tif (hasEmittedRunStarted && !hasReceivedFinishReason) {\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Stream ended without finish_reason — possible truncation or connection drop\",\n\t\t\t\t);\n\n\t\t\t\t// Close any open tool calls\n\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\tif (toolCall.started) {\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Close text message if open\n\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tfinishReason: \"stop\",\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tconst code =\n\t\t\t\terror instanceof Error ? (error as Error & { code?: string }).code : undefined;\n\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t\tyield {\n\t\t\t\ttype: \"RUN_ERROR\",\n\t\t\t\trunId,\n\t\t\t\tmodel: model ?? this.model,\n\t\t\t\ttimestamp,\n\t\t\t\terror: {\n\t\t\t\t\tmessage: message || \"Unknown error\",\n\t\t\t\t\tcode,\n\t\t\t\t},\n\t\t\t} satisfies StreamChunk;\n\t\t}\n\t}\n\n\tasync structuredOutput(\n\t\toptions: StructuredOutputOptions<any>,\n\t): Promise<StructuredOutputResult<unknown>> {\n\t\tconst { outputSchema, chatOptions } = options;\n\t\tconst { systemPrompts, messages, temperature, model } = chatOptions;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages, {\n\t\t\tincludeToolMessages: false,\n\t\t});\n\n\t\tconst response = await this.client.chat.completions.create({\n\t\t\tmodel: model ?? this.model,\n\t\t\tmessages: openAIMessages,\n\t\t\ttemperature,\n\t\t\tstream: false,\n\t\t\tresponse_format: {\n\t\t\t\ttype: \"json_schema\",\n\t\t\t\tjson_schema: {\n\t\t\t\t\tname: \"structured_output\",\n\t\t\t\t\tstrict: true,\n\t\t\t\t\tschema: outputSchema,\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst choice = response.choices?.[0];\n\n\t\tif (!choice) {\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI structured output returned no choices: ${JSON.stringify(response)}`,\n\t\t\t);\n\t\t}\n\n\t\tconst rawContent = choice.message?.content ?? \"\";\n\n\t\t// Workers AI REST endpoint may return `content` as an already-parsed object\n\t\t// when using json_schema response format, so normalise both cases.\n\t\tlet data: unknown;\n\t\tlet rawText: string;\n\n\t\tif (typeof rawContent === \"string\") {\n\t\t\trawText = rawContent;\n\t\t\ttry {\n\t\t\t\tdata = JSON.parse(rawText);\n\t\t\t} catch {\n\t\t\t\tdata = rawText;\n\t\t\t}\n\t\t} else {\n\t\t\t// Already an object — stringify for rawText, use directly for data\n\t\t\tdata = rawContent;\n\t\t\trawText = JSON.stringify(rawContent);\n\t\t}\n\n\t\treturn { data, rawText };\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\nexport function createWorkersAiChat(model: WorkersAiTextModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiTextAdapter(model, config);\n}\n"]}
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-IQAGFHRT.cjs","../src/adapters/workers-ai.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACTA;AACC;AAAA,iDAGM;AACP,gFAAmB;AAyBnB,SAAS,oBAAA,CAAqB,MAAA,EAAwC;AACrE,EAAA,uDAAA,MAA8B,CAAA;AAE9B,EAAA,MAAM,eAAA,EAAqD,MAAA,CAAO,gBAAA,EAC/D,EAAE,oBAAA,EAAsB,MAAA,CAAO,gBAAgB,EAAA,EAC/C,KAAA,CAAA;AAEH,EAAA,GAAA,CAAI,qDAAA,MAA4B,CAAA,EAAG;AAElC,IAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,MACjB,MAAA,EAAQ,QAAA;AAAA,MACR,KAAA,EAAO,2DAAA;AAAA,QACN,MAAA,CAAO,OAAA;AAAA,QACP,eAAA,EAAiB,EAAE,YAAA,EAAc,eAAe,EAAA,EAAI,KAAA;AAAA,MACrD;AAAA,IACD,CAAC,CAAA;AAAA,EACF;AAEA,EAAA,GAAA,CAAI,yDAAA,MAAgC,CAAA,EAAG;AAEtC,IAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,MACjB,OAAA,EAAS,CAAA,8CAAA,EAAiD,MAAA,CAAO,SAAS,CAAA,MAAA,CAAA;AAAA,MAC1E,MAAA,EAAQ,MAAA,CAAO,MAAA;AAAA,MACf,cAAA,EAAgB;AAAA,IACjB,CAAC,CAAA;AAAA,EACF;AAGA,EAAA,MAAM,cAAA,EAAgB,MAAA;AACtB,EAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,IACjB,KAAA,EAAO,kDAAA,YAAmB,EAAc,aAAA,EAAe,cAAc,CAAA;AAAA,IACrE,MAAA,mBAAQ,aAAA,CAAc,MAAA,UAAU;AAAA,EACjC,CAAC,CAAA;AACF;AAMA,SAAS,kBAAA,CAAmB,OAAA,EAA0C;AACrE,EAAA,GAAA,CAAI,QAAA,IAAY,IAAA,EAAM,OAAO,EAAA;AAC7B,EAAA,GAAA,CAAI,OAAO,QAAA,IAAY,QAAA,EAAU,OAAO,OAAA;AACxC,EAAA,OAAO,OAAA,CAAQ,OAAA,CAAQ,CAAC,CAAA,EAAA,GAAO,CAAA,CAAE,KAAA,IAAS,OAAA,EAAS,CAAC,CAAA,CAAE,OAAO,EAAA,EAAI,CAAC,CAAE,CAAA,CAAE,IAAA,CAAK,EAAE,CAAA;AAC9E;AAQA,SAAS,kBAAA,CAAmB,IAAA,EAAsE;AACjG,EAAA,OAAA,CAAQ,IAAA,CAAK,IAAA,EAAM;AAAA,IAClB,KAAK,MAAA;AACJ,MAAA,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS;AACjB,QAAA,OAAO,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,IAAA,CAAK,QAAQ,CAAA;AAAA,MAC3C;AACA,MAAA,OAAO,KAAA,CAAA;AAAA,IACR,KAAK,OAAA,EAAS;AACb,MAAA,IAAI,GAAA;AACJ,MAAA,GAAA,CAAI,IAAA,CAAK,MAAA,CAAO,KAAA,IAAS,MAAA,EAAQ;AAChC,QAAA,IAAA,EAAM,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,UAAA,CAAW,OAAO,EAAA,EACvC,IAAA,CAAK,MAAA,CAAO,MAAA,EACZ,CAAA,KAAA,EAAQ,IAAA,CAAK,MAAA,CAAO,QAAQ,CAAA,QAAA,EAAW,IAAA,CAAK,MAAA,CAAO,KAAK,CAAA,CAAA;AACrD,MAAA;AACY,QAAA;AACnB,MAAA;AAC+C,MAAA;AAChD,IAAA;AACA,IAAA;AAES,MAAA;AAC8D,QAAA;AACtE,MAAA;AACO,MAAA;AACT,EAAA;AACD;AAWoD;AACtB,EAAA;AACW,EAAA;AAEgB,EAAA;AACxC,EAAA;AAC8D,IAAA;AAC9E,EAAA;AAEwD,EAAA;AAC5B,EAAA;AACc,IAAA;AAC1B,IAAA;AACM,MAAA;AACrB,IAAA;AACD,EAAA;AACO,EAAA;AACR;AAM4C;AACU,EAAA;AACa,EAAA;AAEnB,EAAA;AAC1B,IAAA;AACb,MAAA;AAC0B,MAAA;AAChC,IAAA;AACF,EAAA;AAEgC,EAAA;AACF,IAAA;AACR,MAAA;AACb,QAAA;AACmC,QAAA;AACzC,MAAA;AACuC,IAAA;AACkC,MAAA;AACnE,QAAA;AACqC,QAAA;AAC5C,MAAA;AACuE,MAAA;AACT,QAAA;AACrD,UAAA;AACD,UAAA;AACI,UAAA;AACS,YAAA;AACK,YAAA;AACxB,UAAA;AACC,QAAA;AACH,MAAA;AACoC,MAAA;AACe,IAAA;AAC/C,MAAA;AACqC,MAAA;AACpC,QAAA;AACuB,UAAA;AACJ,UAAA;AACf,QAAA;AACqC,UAAA;AAC7C,QAAA;AACM,MAAA;AACsC,QAAA;AAC7C,MAAA;AACoB,MAAA;AACb,QAAA;AACqE,QAAA;AAClE,QAAA;AACT,MAAA;AACF,IAAA;AACD,EAAA;AAEO,EAAA;AACR;AAIgD;AAC5B,EAAA;AACS,EAAA;AACrB,IAAA;AACI,IAAA;AACE,MAAA;AACO,MAAA;AACD,MAAA;AAClB,IAAA;AACC,EAAA;AACH;AAMiD;AACsB,EAAA;AACvE;AAeE;AAK0D,EAAA;AACzB,IAAA;AAL3B,IAAA;AAEC,IAAA;AAIkC,IAAA;AAC1C,EAAA;AAEyE,EAAA;AACT,IAAA;AAEG,IAAA;AACxB,IAAA;AAEf,IAAA;AACF,IAAA;AACI,IAAA;AACF,IAAA;AACM,IAAA;AACR,IAAA;AACG,IAAA;AACD,IAAA;AACD,IAAA;AACI,IAAA;AAI5B,IAAA;AAEE,IAAA;AACC,MAAA;AACA,MAAA;AACgD,QAAA;AAC7B,UAAA;AACX,UAAA;AACH,UAAA;AACP,UAAA;AACQ,UAAA;AAC8B,UAAA;AACtC,QAAA;AAC6B,MAAA;AAGtB,QAAA;AACP,UAAA;AACqD,UAAA;AACtD,QAAA;AACkE,QAAA;AAC5C,UAAA;AACX,UAAA;AACH,UAAA;AACP,UAAA;AACA,QAAA;AAEK,QAAA;AACC,UAAA;AACN,UAAA;AAC8C,UAAA;AAC9C,UAAA;AACD,QAAA;AAEwC,QAAA;AACtB,QAAA;AACX,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACM,YAAA;AACP,UAAA;AACM,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACW,YAAA;AACE,YAAA;AACd,UAAA;AACM,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACD,UAAA;AACD,QAAA;AAEqB,QAAA;AACa,UAAA;AACJ,YAAA;AACd,YAAA;AACc,YAAA;AACxB,YAAA;AACsD,cAAA;AAClD,YAAA;AACQ,cAAA;AAChB,YAAA;AACM,YAAA;AACC,cAAA;AACS,cAAA;AACF,cAAA;AACiC,cAAA;AAC9C,cAAA;AACO,cAAA;AACR,YAAA;AACM,YAAA;AACC,cAAA;AACS,cAAA;AACF,cAAA;AACiC,cAAA;AAC9C,cAAA;AACO,cAAA;AACR,YAAA;AACD,UAAA;AACD,QAAA;AAEiD,QAAA;AAC3C,QAAA;AACC,UAAA;AACN,UAAA;AAC8C,UAAA;AAC9C,UAAA;AAEG,UAAA;AACoC,YAAA;AACI,YAAA;AACL,YAAA;AAEnC,UAAA;AAI4D,UAAA;AAChE,QAAA;AACA,QAAA;AACD,MAAA;AAEkC,MAAA;AACiB,QAAA;AACpB,QAAA;AACjB,QAAA;AAGc,QAAA;AACH,UAAA;AACjB,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AACD,UAAA;AACD,QAAA;AAEqB,QAAA;AAKe,QAAA;AACd,QAAA;AAEO,UAAA;AACH,YAAA;AAClB,YAAA;AACC,cAAA;AACN,cAAA;AACU,cAAA;AAC0B,cAAA;AACpC,cAAA;AACD,YAAA;AACD,UAAA;AACwB,UAAA;AAOlB,UAAA;AACC,YAAA;AACN,YAAA;AACO,YAAA;AACE,YAAA;AAC2B,YAAA;AACpC,YAAA;AACD,UAAA;AACD,QAAA;AAGmB,QAAA;AACe,UAAA;AACH,YAAA;AACvB,YAAA;AACC,cAAA;AACN,cAAA;AACoC,cAAA;AACpC,cAAA;AACM,cAAA;AACP,YAAA;AACD,UAAA;AAE4B,UAAA;AACtB,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AACa,YAAA;AACJ,YAAA;AACV,UAAA;AACD,QAAA;AAGsB,QAAA;AACyB,UAAA;AACjB,YAAA;AAES,YAAA;AAIL,cAAA;AACA,gBAAA;AACQ,gBAAA;AAC3B,gBAAA;AACF,gBAAA;AACT,cAAA;AACF,YAAA;AAE8C,YAAA;AAKZ,YAAA;AACM,cAAA;AACxC,YAAA;AACuC,YAAA;AACO,cAAA;AAC9C,YAAA;AAGuD,YAAA;AACnC,cAAA;AACb,cAAA;AACC,gBAAA;AACe,gBAAA;AACF,gBAAA;AACiB,gBAAA;AACpC,gBAAA;AACA,gBAAA;AACD,cAAA;AACD,YAAA;AAG2D,YAAA;AACpD,cAAA;AACC,gBAAA;AACe,gBAAA;AACe,gBAAA;AACpC,gBAAA;AAC8B,gBAAA;AAC/B,cAAA;AACD,YAAA;AACD,UAAA;AACD,QAAA;AAG0B,QAAA;AACC,UAAA;AAGiD,UAAA;AAC1B,YAAA;AACnB,cAAA;AACxB,cAAA;AAGC,gBAAA;AACG,cAAA;AACQ,gBAAA;AAChB,cAAA;AACM,cAAA;AACC,gBAAA;AACe,gBAAA;AACF,gBAAA;AACiB,gBAAA;AACpC,gBAAA;AACO,gBAAA;AACR,cAAA;AACD,YAAA;AACD,UAAA;AAKC,UAAA;AAK+B,UAAA;AACzB,YAAA;AACC,cAAA;AACN,cAAA;AACoC,cAAA;AACpC,cAAA;AACD,YAAA;AACD,UAAA;AAGM,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AAEG,YAAA;AAC0B,cAAA;AACI,cAAA;AACL,cAAA;AAEzB,YAAA;AACW,YAAA;AACf,UAAA;AACD,QAAA;AACD,MAAA;AAKsD,MAAA;AAC7C,QAAA;AACP,UAAA;AACD,QAAA;AAGgD,QAAA;AACzB,UAAA;AACO,YAAA;AACxB,YAAA;AACkE,cAAA;AAC9D,YAAA;AACQ,cAAA;AAChB,YAAA;AACM,YAAA;AACC,cAAA;AACe,cAAA;AACF,cAAA;AACE,cAAA;AACrB,cAAA;AACO,cAAA;AACR,YAAA;AACD,UAAA;AACD,QAAA;AAGgC,QAAA;AACzB,UAAA;AACC,YAAA;AACN,YAAA;AACqB,YAAA;AACrB,YAAA;AACD,UAAA;AACD,QAAA;AAEM,QAAA;AACC,UAAA;AACN,UAAA;AACqB,UAAA;AACrB,UAAA;AACc,UAAA;AACf,QAAA;AACD,MAAA;AACe,IAAA;AACsD,MAAA;AAEC,MAAA;AAC3C,MAAA;AACpB,QAAA;AACC,UAAA;AACN,UAAA;AACqB,UAAA;AACrB,UAAA;AACD,QAAA;AACD,MAAA;AACM,MAAA;AACC,QAAA;AACN,QAAA;AACqB,QAAA;AACrB,QAAA;AACO,QAAA;AACc,UAAA;AACpB,UAAA;AACD,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAI4C,EAAA;AACL,IAAA;AACkB,IAAA;AAEY,IAAA;AAC9C,MAAA;AACrB,IAAA;AAE0D,IAAA;AACrC,MAAA;AACX,MAAA;AACV,MAAA;AACQ,MAAA;AACS,MAAA;AACV,QAAA;AACO,QAAA;AACN,UAAA;AACE,UAAA;AACA,UAAA;AACT,QAAA;AACD,MAAA;AACA,IAAA;AAEkC,IAAA;AAEtB,IAAA;AACF,MAAA;AACoE,QAAA;AAC9E,MAAA;AACD,IAAA;AAE8C,IAAA;AAI1C,IAAA;AACA,IAAA;AAEgC,IAAA;AACzB,MAAA;AACN,MAAA;AACsB,QAAA;AAClB,MAAA;AACA,QAAA;AACR,MAAA;AACM,IAAA;AAEC,MAAA;AAC4B,MAAA;AACpC,IAAA;AAEuB,IAAA;AACxB,EAAA;AACD;AAM+F;AACjD,EAAA;AAC9C;ADrLmI;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-IQAGFHRT.cjs","sourcesContent":[null,"import type { AiModels, BaseAiTextGeneration } from \"@cloudflare/workers-types\";\nimport type { ContentPart, ModelMessage, StreamChunk, TextOptions } from \"@tanstack/ai\";\nimport {\n\tBaseTextAdapter,\n\ttype StructuredOutputOptions,\n\ttype StructuredOutputResult,\n} from \"@tanstack/ai/adapters\";\nimport OpenAI from \"openai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tcreateWorkersAiBindingFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model types derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiTextModel =\n\t| {\n\t\t\t[K in keyof AiModels]: AiModels[K] extends BaseAiTextGeneration ? K : never;\n\t }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// Helpers: build the right OpenAI client depending on config mode\n// ---------------------------------------------------------------------------\n\nfunction buildWorkersAiClient(config: WorkersAiAdapterConfig): OpenAI {\n\tvalidateWorkersAiConfig(config);\n\n\tconst sessionHeaders: Record<string, string> | undefined = config.sessionAffinity\n\t\t? { \"x-session-affinity\": config.sessionAffinity }\n\t\t: undefined;\n\n\tif (isDirectBindingConfig(config)) {\n\t\t// Plain binding mode: shim translates OpenAI fetch calls to env.AI.run()\n\t\treturn new OpenAI({\n\t\t\tapiKey: \"unused\",\n\t\t\tfetch: createWorkersAiBindingFetch(\n\t\t\t\tconfig.binding,\n\t\t\t\tsessionHeaders ? { extraHeaders: sessionHeaders } : undefined,\n\t\t\t),\n\t\t});\n\t}\n\n\tif (isDirectCredentialsConfig(config)) {\n\t\t// Plain REST mode: point OpenAI SDK at Workers AI's OpenAI-compatible endpoint\n\t\treturn new OpenAI({\n\t\t\tbaseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,\n\t\t\tapiKey: config.apiKey,\n\t\t\tdefaultHeaders: sessionHeaders,\n\t\t});\n\t}\n\n\t// Gateway mode (existing): use createGatewayFetch\n\tconst gatewayConfig = config as AiGatewayAdapterConfig;\n\treturn new OpenAI({\n\t\tfetch: createGatewayFetch(\"workers-ai\", gatewayConfig, sessionHeaders),\n\t\tapiKey: gatewayConfig.apiKey ?? \"unused\",\n\t});\n}\n\n// ---------------------------------------------------------------------------\n// Shared message-building helpers\n// ---------------------------------------------------------------------------\n\nfunction extractTextContent(content: ModelMessage[\"content\"]): string {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n}\n\n/**\n * Convert a single TanStack AI {@link ContentPart} to the OpenAI Chat\n * Completions multi-modal format.\n *\n * TODO: handle other content types (audio, video, document)\n */\nfunction convertContentPart(part: ContentPart): OpenAI.Chat.ChatCompletionContentPart | undefined {\n\tswitch (part.type) {\n\t\tcase \"text\":\n\t\t\tif (part.content) {\n\t\t\t\treturn { type: \"text\", text: part.content };\n\t\t\t}\n\t\t\treturn undefined;\n\t\tcase \"image\": {\n\t\t\tlet url: string;\n\t\t\tif (part.source.type === \"data\") {\n\t\t\t\turl = part.source.value.startsWith(\"data:\")\n\t\t\t\t\t? part.source.value\n\t\t\t\t\t: `data:${part.source.mimeType};base64,${part.source.value}`;\n\t\t\t} else {\n\t\t\t\turl = part.source.value;\n\t\t\t}\n\t\t\treturn { type: \"image_url\", image_url: { url } };\n\t\t}\n\t\tdefault:\n\t\t\t// audio, video, document — not supported for now\n\t\t\tconsole.warn(\n\t\t\t\t`[@cloudflare/tanstack-ai] Unsupported content part type \"${part.type}\" — skipping`,\n\t\t\t);\n\t\t\treturn undefined;\n\t}\n}\n\n/**\n * Build OpenAI-compatible user message content from TanStack AI content.\n *\n * If the content has only text parts, returns a plain string.\n * If it includes image parts, returns an array of content parts in\n * OpenAI's multi-modal format (text + image_url).\n */\nfunction buildUserContent(\n\tcontent: ModelMessage[\"content\"],\n): string | OpenAI.Chat.ChatCompletionContentPart[] {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\n\tconst hasImages = content.some((p) => p.type === \"image\");\n\tif (!hasImages) {\n\t\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n\t}\n\n\tconst parts: OpenAI.Chat.ChatCompletionContentPart[] = [];\n\tfor (const part of content) {\n\t\tconst converted = convertContentPart(part);\n\t\tif (converted) {\n\t\t\tparts.push(converted);\n\t\t}\n\t}\n\treturn parts;\n}\n\nfunction buildOpenAIMessages(\n\tsystemPrompts: string[] | undefined,\n\tmessages: ModelMessage[],\n\toptions?: { includeToolMessages?: boolean },\n): OpenAI.Chat.ChatCompletionMessageParam[] {\n\tconst includeTools = options?.includeToolMessages ?? true;\n\tconst openAIMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [];\n\n\tif (systemPrompts && systemPrompts.length > 0) {\n\t\topenAIMessages.push({\n\t\t\trole: \"system\",\n\t\t\tcontent: systemPrompts.join(\"\\n\"),\n\t\t});\n\t}\n\n\tfor (const message of messages) {\n\t\tif (message.role === \"user\") {\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"user\",\n\t\t\t\tcontent: buildUserContent(message.content),\n\t\t\t});\n\t\t} else if (message.role === \"assistant\") {\n\t\t\tconst assistantMessage: OpenAI.Chat.ChatCompletionAssistantMessageParam = {\n\t\t\t\trole: \"assistant\",\n\t\t\t\tcontent: extractTextContent(message.content),\n\t\t\t};\n\t\t\tif (includeTools && message.toolCalls && message.toolCalls.length > 0) {\n\t\t\t\tassistantMessage.tool_calls = message.toolCalls.map((tc) => ({\n\t\t\t\t\tid: tc.id,\n\t\t\t\t\ttype: \"function\" as const,\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: tc.function.name,\n\t\t\t\t\t\targuments: tc.function.arguments,\n\t\t\t\t\t},\n\t\t\t\t}));\n\t\t\t}\n\t\t\topenAIMessages.push(assistantMessage);\n\t\t} else if (includeTools && message.role === \"tool\") {\n\t\t\tlet toolContent: string;\n\t\t\tif (typeof message.content === \"string\") {\n\t\t\t\ttry {\n\t\t\t\t\tJSON.parse(message.content);\n\t\t\t\t\ttoolContent = message.content;\n\t\t\t\t} catch {\n\t\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t}\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"tool\",\n\t\t\t\ttool_call_id: message.toolCallId || `tool_${crypto.randomUUID().slice(0, 8)}`,\n\t\t\t\tcontent: toolContent,\n\t\t\t});\n\t\t}\n\t}\n\n\treturn openAIMessages;\n}\n\nfunction buildOpenAITools(\n\ttools: Array<{ name: string; description: string; inputSchema?: unknown }> | undefined,\n): OpenAI.Chat.ChatCompletionTool[] | undefined {\n\tif (!tools) return undefined;\n\treturn tools.map((tool) => ({\n\t\ttype: \"function\" as const,\n\t\tfunction: {\n\t\t\tname: tool.name,\n\t\t\tdescription: tool.description,\n\t\t\tparameters: tool.inputSchema as Record<string, unknown>,\n\t\t},\n\t}));\n}\n\n// ---------------------------------------------------------------------------\n// ID generation\n// ---------------------------------------------------------------------------\n\nfunction generateId(prefix = \"chatcmpl\"): string {\n\treturn `${prefix}-${crypto.randomUUID().replace(/-/g, \"\").slice(0, 16)}`;\n}\n\n// ---------------------------------------------------------------------------\n// WorkersAiTextAdapter: chat / structured output via OpenAI Chat Completions\n// ---------------------------------------------------------------------------\n\n// TODO: Replace `any` generic params with proper types once BaseTextAdapter's\n// provider-options generics stabilize. Workers AI doesn't have provider-specific\n// options in the TanStack sense, so `any` is pragmatic for now.\nexport class WorkersAiTextAdapter<TModel extends WorkersAiTextModel> extends BaseTextAdapter<\n\tTModel,\n\t// eslint-disable-next-line @typescript-eslint/no-explicit-any -- BaseTextAdapter generic params are opaque\n\tany,\n\tany,\n\tany\n> {\n\tname = \"workers-ai\" as const;\n\n\tprivate client: OpenAI;\n\n\tconstructor(model: TModel, config: WorkersAiAdapterConfig) {\n\t\tsuper({ apiKey: \"unused\" }, model);\n\t\tthis.client = buildWorkersAiClient(config);\n\t}\n\n\tasync *chatStream(options: TextOptions<any>): AsyncIterable<StreamChunk> {\n\t\tconst { systemPrompts, messages, tools, temperature, model } = options;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages);\n\t\tconst openAITools = buildOpenAITools(tools);\n\n\t\tconst timestamp = Date.now();\n\t\tconst runId = generateId();\n\t\tconst messageId = generateId();\n\t\tlet hasEmittedRunStarted = false;\n\t\tlet hasEmittedTextMessageStart = false;\n\t\tlet accumulatedContent = \"\";\n\t\tlet hasEmittedStepStarted = false;\n\t\tlet accumulatedReasoning = \"\";\n\t\tconst stepId = generateId();\n\t\tlet hasReceivedFinishReason = false;\n\t\tconst toolCallsInProgress = new Map<\n\t\t\tnumber,\n\t\t\t{ id: string; name: string; arguments: string; started: boolean }\n\t\t>();\n\n\t\ttry {\n\t\t\tlet stream: AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>;\n\t\t\ttry {\n\t\t\t\tstream = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t\tstream: true,\n\t\t\t\t\tstream_options: { include_usage: true },\n\t\t\t\t});\n\t\t\t} catch (streamError: unknown) {\n\t\t\t\t// Some models (e.g. GPT-OSS) don't support streaming via the REST API.\n\t\t\t\t// Fall back to a non-streaming call and yield the result as a single chunk.\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Streaming failed, falling back to non-streaming:\",\n\t\t\t\t\tstreamError instanceof Error ? streamError.message : streamError,\n\t\t\t\t);\n\t\t\t\tconst nonStreamResult = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t});\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\n\t\t\t\tconst msg = nonStreamResult.choices[0]?.message;\n\t\t\t\tif (msg?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: msg.content,\n\t\t\t\t\t\tcontent: msg.content,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tif (msg?.tool_calls) {\n\t\t\t\t\tfor (const tc of msg.tool_calls) {\n\t\t\t\t\t\tif (tc.type !== \"function\") continue;\n\t\t\t\t\t\tconst fn = tc.function;\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = fn.arguments ? JSON.parse(fn.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tconst finishReason = nonStreamResult.choices[0]?.finish_reason;\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tusage: nonStreamResult.usage\n\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\tpromptTokens: nonStreamResult.usage.prompt_tokens,\n\t\t\t\t\t\t\t\tcompletionTokens: nonStreamResult.usage.completion_tokens,\n\t\t\t\t\t\t\t\ttotalTokens: nonStreamResult.usage.total_tokens,\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t: undefined,\n\t\t\t\t\tfinishReason:\n\t\t\t\t\t\tfinishReason === \"tool_calls\" || finishReason === \"function_call\"\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: ((finishReason as \"stop\" | \"length\" | \"content_filter\") ?? \"stop\"),\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tfor await (const chunk of stream) {\n\t\t\t\tif (!chunk.choices || chunk.choices.length === 0) continue;\n\t\t\t\tconst choice = chunk.choices[0];\n\t\t\t\tif (!choice) continue;\n\n\t\t\t\t// Emit RUN_STARTED on first chunk\n\t\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\t\thasEmittedRunStarted = true;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tconst delta = choice.delta;\n\n\t\t\t\t// Reasoning content (used by models like QwQ, DeepSeek R1, Kimi K2.5)\n\t\t\t\t// The OpenAI SDK doesn't type this field, but models send it as an extension.\n\t\t\t\tconst reasoningContent = ((delta as Record<string, unknown>).reasoning_content ??\n\t\t\t\t\t(delta as Record<string, unknown>).reasoning) as string | undefined;\n\t\t\t\tif (reasoningContent) {\n\t\t\t\t\t// RUN_STARTED is already guaranteed by the guard above\n\t\t\t\t\tif (!hasEmittedStepStarted) {\n\t\t\t\t\t\thasEmittedStepStarted = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"STEP_STARTED\",\n\t\t\t\t\t\t\tstepId,\n\t\t\t\t\t\t\tstepType: \"thinking\",\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t\taccumulatedReasoning += reasoningContent;\n\t\t\t\t\t// TODO: TanStack AI's StreamProcessor currently treats STEP_FINISHED as an\n\t\t\t\t\t// incremental reasoning event (with `delta` + accumulated `content`), so we\n\t\t\t\t\t// emit one per token. If TanStack AI adds a dedicated STEP_CONTENT event\n\t\t\t\t\t// type, this should be updated to emit STEP_CONTENT per token and a single\n\t\t\t\t\t// STEP_FINISHED when reasoning ends (i.e. when the first non-reasoning\n\t\t\t\t\t// content or finish_reason arrives).\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"STEP_FINISHED\",\n\t\t\t\t\t\tstepId,\n\t\t\t\t\t\tdelta: reasoningContent,\n\t\t\t\t\t\tcontent: accumulatedReasoning,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Text content\n\t\t\t\tif (delta.content) {\n\t\t\t\t\tif (!hasEmittedTextMessageStart) {\n\t\t\t\t\t\thasEmittedTextMessageStart = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\taccumulatedContent += delta.content;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: delta.content,\n\t\t\t\t\t\tcontent: accumulatedContent,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Tool calls\n\t\t\t\tif (delta.tool_calls) {\n\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\tif (!toolCallsInProgress.has(index)) {\n\t\t\t\t\t\t\t// Always generate a unique ID per tool call index.\n\t\t\t\t\t\t\t// The backend may send the same ID for multiple tool calls,\n\t\t\t\t\t\t\t// so we cannot trust toolCallDelta.id to be unique.\n\t\t\t\t\t\t\ttoolCallsInProgress.set(index, {\n\t\t\t\t\t\t\t\tid: generateId(\"chatcmpl-tool\"),\n\t\t\t\t\t\t\t\tname: toolCallDelta.function?.name || \"\",\n\t\t\t\t\t\t\t\targuments: \"\",\n\t\t\t\t\t\t\t\tstarted: false,\n\t\t\t\t\t\t\t});\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tconst toolCall = toolCallsInProgress.get(index)!;\n\n\t\t\t\t\t\t// Only update name if provided (ID is already set at creation time\n\t\t\t\t\t\t// and should not be overwritten by subsequent chunks that may have\n\t\t\t\t\t\t// duplicate/shared IDs from the backend)\n\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\ttoolCall.name = toolCallDelta.function.name;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\ttoolCall.arguments += toolCallDelta.function.arguments;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Emit TOOL_CALL_START once we have id and name\n\t\t\t\t\t\tif (toolCall.id && toolCall.name && !toolCall.started) {\n\t\t\t\t\t\t\ttoolCall.started = true;\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tindex,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Stream tool call arguments\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments && toolCall.started) {\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_ARGS\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tdelta: toolCallDelta.function.arguments,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Finish\n\t\t\t\tif (choice.finish_reason) {\n\t\t\t\t\thasReceivedFinishReason = true;\n\n\t\t\t\t\t// End tool calls\n\t\t\t\t\tif (choice.finish_reason === \"tool_calls\" || toolCallsInProgress.size > 0) {\n\t\t\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\t\tparsedInput = toolCall.arguments\n\t\t\t\t\t\t\t\t\t? JSON.parse(toolCall.arguments)\n\t\t\t\t\t\t\t\t\t: {};\n\t\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\n\t\t\t\t\tconst computedFinishReason =\n\t\t\t\t\t\tchoice.finish_reason === \"tool_calls\" ||\n\t\t\t\t\t\tchoice.finish_reason === \"function_call\" ||\n\t\t\t\t\t\ttoolCallsInProgress.size > 0\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: (choice.finish_reason as \"stop\" | \"length\" | \"content_filter\");\n\n\t\t\t\t\t// End text message if started\n\t\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Emit RUN_FINISHED\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tusage: chunk.usage\n\t\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\t\tpromptTokens: chunk.usage.prompt_tokens,\n\t\t\t\t\t\t\t\t\tcompletionTokens: chunk.usage.completion_tokens,\n\t\t\t\t\t\t\t\t\ttotalTokens: chunk.usage.total_tokens,\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t: undefined,\n\t\t\t\t\t\tfinishReason: computedFinishReason,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Premature stream termination: the stream ended without a finish_reason.\n\t\t\t// This can happen when Workers AI truncates a response or the connection drops.\n\t\t\t// Emit proper closing events so the consumer doesn't hang.\n\t\t\tif (hasEmittedRunStarted && !hasReceivedFinishReason) {\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Stream ended without finish_reason — possible truncation or connection drop\",\n\t\t\t\t);\n\n\t\t\t\t// Close any open tool calls\n\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\tif (toolCall.started) {\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Close text message if open\n\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tfinishReason: \"stop\",\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tconst code =\n\t\t\t\terror instanceof Error ? (error as Error & { code?: string }).code : undefined;\n\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t\tyield {\n\t\t\t\ttype: \"RUN_ERROR\",\n\t\t\t\trunId,\n\t\t\t\tmodel: model ?? this.model,\n\t\t\t\ttimestamp,\n\t\t\t\terror: {\n\t\t\t\t\tmessage: message || \"Unknown error\",\n\t\t\t\t\tcode,\n\t\t\t\t},\n\t\t\t} satisfies StreamChunk;\n\t\t}\n\t}\n\n\tasync structuredOutput(\n\t\toptions: StructuredOutputOptions<any>,\n\t): Promise<StructuredOutputResult<unknown>> {\n\t\tconst { outputSchema, chatOptions } = options;\n\t\tconst { systemPrompts, messages, temperature, model } = chatOptions;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages, {\n\t\t\tincludeToolMessages: false,\n\t\t});\n\n\t\tconst response = await this.client.chat.completions.create({\n\t\t\tmodel: model ?? this.model,\n\t\t\tmessages: openAIMessages,\n\t\t\ttemperature,\n\t\t\tstream: false,\n\t\t\tresponse_format: {\n\t\t\t\ttype: \"json_schema\",\n\t\t\t\tjson_schema: {\n\t\t\t\t\tname: \"structured_output\",\n\t\t\t\t\tstrict: true,\n\t\t\t\t\tschema: outputSchema,\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst choice = response.choices?.[0];\n\n\t\tif (!choice) {\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI structured output returned no choices: ${JSON.stringify(response)}`,\n\t\t\t);\n\t\t}\n\n\t\tconst rawContent = choice.message?.content ?? \"\";\n\n\t\t// Workers AI REST endpoint may return `content` as an already-parsed object\n\t\t// when using json_schema response format, so normalise both cases.\n\t\tlet data: unknown;\n\t\tlet rawText: string;\n\n\t\tif (typeof rawContent === \"string\") {\n\t\t\trawText = rawContent;\n\t\t\ttry {\n\t\t\t\tdata = JSON.parse(rawText);\n\t\t\t} catch {\n\t\t\t\tdata = rawText;\n\t\t\t}\n\t\t} else {\n\t\t\t// Already an object — stringify for rawText, use directly for data\n\t\t\tdata = rawContent;\n\t\t\trawText = JSON.stringify(rawContent);\n\t\t}\n\n\t\treturn { data, rawText };\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\nexport function createWorkersAiChat(model: WorkersAiTextModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiTextAdapter(model, config);\n}\n"]}
|