@cloudflare/tanstack-ai 0.0.0 → 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (127) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +282 -0
  3. package/dist/_tsup-dts-rollup.d.cts +820 -0
  4. package/dist/_tsup-dts-rollup.d.ts +820 -0
  5. package/dist/adapters/anthropic.cjs +13 -0
  6. package/dist/adapters/anthropic.cjs.map +1 -0
  7. package/dist/adapters/anthropic.d.cts +5 -0
  8. package/dist/adapters/anthropic.d.ts +5 -0
  9. package/dist/adapters/anthropic.js +13 -0
  10. package/dist/adapters/anthropic.js.map +1 -0
  11. package/dist/adapters/gemini.cjs +22 -0
  12. package/dist/adapters/gemini.cjs.map +1 -0
  13. package/dist/adapters/gemini.d.cts +14 -0
  14. package/dist/adapters/gemini.d.ts +14 -0
  15. package/dist/adapters/gemini.js +22 -0
  16. package/dist/adapters/gemini.js.map +1 -0
  17. package/dist/adapters/grok.cjs +17 -0
  18. package/dist/adapters/grok.cjs.map +1 -0
  19. package/dist/adapters/grok.d.cts +9 -0
  20. package/dist/adapters/grok.d.ts +9 -0
  21. package/dist/adapters/grok.js +17 -0
  22. package/dist/adapters/grok.js.map +1 -0
  23. package/dist/adapters/openai.cjs +29 -0
  24. package/dist/adapters/openai.cjs.map +1 -0
  25. package/dist/adapters/openai.d.cts +17 -0
  26. package/dist/adapters/openai.d.ts +17 -0
  27. package/dist/adapters/openai.js +29 -0
  28. package/dist/adapters/openai.js.map +1 -0
  29. package/dist/adapters/openrouter.cjs +13 -0
  30. package/dist/adapters/openrouter.cjs.map +1 -0
  31. package/dist/adapters/openrouter.d.cts +7 -0
  32. package/dist/adapters/openrouter.d.ts +7 -0
  33. package/dist/adapters/openrouter.js +13 -0
  34. package/dist/adapters/openrouter.js.map +1 -0
  35. package/dist/adapters/workers-ai-image.cjs +13 -0
  36. package/dist/adapters/workers-ai-image.cjs.map +1 -0
  37. package/dist/adapters/workers-ai-image.d.cts +3 -0
  38. package/dist/adapters/workers-ai-image.d.ts +3 -0
  39. package/dist/adapters/workers-ai-image.js +13 -0
  40. package/dist/adapters/workers-ai-image.js.map +1 -0
  41. package/dist/adapters/workers-ai-summarize.cjs +12 -0
  42. package/dist/adapters/workers-ai-summarize.cjs.map +1 -0
  43. package/dist/adapters/workers-ai-summarize.d.cts +3 -0
  44. package/dist/adapters/workers-ai-summarize.d.ts +3 -0
  45. package/dist/adapters/workers-ai-summarize.js +12 -0
  46. package/dist/adapters/workers-ai-summarize.js.map +1 -0
  47. package/dist/adapters/workers-ai-transcription.cjs +13 -0
  48. package/dist/adapters/workers-ai-transcription.cjs.map +1 -0
  49. package/dist/adapters/workers-ai-transcription.d.cts +3 -0
  50. package/dist/adapters/workers-ai-transcription.d.ts +3 -0
  51. package/dist/adapters/workers-ai-transcription.js +13 -0
  52. package/dist/adapters/workers-ai-transcription.js.map +1 -0
  53. package/dist/adapters/workers-ai-tts.cjs +13 -0
  54. package/dist/adapters/workers-ai-tts.cjs.map +1 -0
  55. package/dist/adapters/workers-ai-tts.d.cts +3 -0
  56. package/dist/adapters/workers-ai-tts.d.ts +3 -0
  57. package/dist/adapters/workers-ai-tts.js +13 -0
  58. package/dist/adapters/workers-ai-tts.js.map +1 -0
  59. package/dist/adapters/workers-ai.cjs +11 -0
  60. package/dist/adapters/workers-ai.cjs.map +1 -0
  61. package/dist/adapters/workers-ai.d.cts +3 -0
  62. package/dist/adapters/workers-ai.d.ts +3 -0
  63. package/dist/adapters/workers-ai.js +11 -0
  64. package/dist/adapters/workers-ai.js.map +1 -0
  65. package/dist/chunk-2AJ6LV2D.js +84 -0
  66. package/dist/chunk-2AJ6LV2D.js.map +1 -0
  67. package/dist/chunk-2VII5BK2.js +42 -0
  68. package/dist/chunk-2VII5BK2.js.map +1 -0
  69. package/dist/chunk-3VQDXJLW.cjs +46 -0
  70. package/dist/chunk-3VQDXJLW.cjs.map +1 -0
  71. package/dist/chunk-4DE2IREA.cjs +8 -0
  72. package/dist/chunk-4DE2IREA.cjs.map +1 -0
  73. package/dist/chunk-7AEFXYJG.js +65 -0
  74. package/dist/chunk-7AEFXYJG.js.map +1 -0
  75. package/dist/chunk-7HSUHP63.cjs +42 -0
  76. package/dist/chunk-7HSUHP63.cjs.map +1 -0
  77. package/dist/chunk-7T4CVHKD.cjs +84 -0
  78. package/dist/chunk-7T4CVHKD.cjs.map +1 -0
  79. package/dist/chunk-BD4CRW3Q.js +389 -0
  80. package/dist/chunk-BD4CRW3Q.js.map +1 -0
  81. package/dist/chunk-BPWGWJJV.cjs +389 -0
  82. package/dist/chunk-BPWGWJJV.cjs.map +1 -0
  83. package/dist/chunk-F5YJMXZR.js +315 -0
  84. package/dist/chunk-F5YJMXZR.js.map +1 -0
  85. package/dist/chunk-GOU66I5T.cjs +315 -0
  86. package/dist/chunk-GOU66I5T.cjs.map +1 -0
  87. package/dist/chunk-IWZJCLOE.cjs +31 -0
  88. package/dist/chunk-IWZJCLOE.cjs.map +1 -0
  89. package/dist/chunk-LBYDBPHY.cjs +109 -0
  90. package/dist/chunk-LBYDBPHY.cjs.map +1 -0
  91. package/dist/chunk-LIUHRGEK.cjs +96 -0
  92. package/dist/chunk-LIUHRGEK.cjs.map +1 -0
  93. package/dist/chunk-M6NETFDR.cjs +48 -0
  94. package/dist/chunk-M6NETFDR.cjs.map +1 -0
  95. package/dist/chunk-O2C4CR57.cjs +54 -0
  96. package/dist/chunk-O2C4CR57.cjs.map +1 -0
  97. package/dist/chunk-PLTFCUMO.js +216 -0
  98. package/dist/chunk-PLTFCUMO.js.map +1 -0
  99. package/dist/chunk-QRHKPL75.js +54 -0
  100. package/dist/chunk-QRHKPL75.js.map +1 -0
  101. package/dist/chunk-RGDUK5KX.cjs +65 -0
  102. package/dist/chunk-RGDUK5KX.cjs.map +1 -0
  103. package/dist/chunk-RQT7M6MU.js +96 -0
  104. package/dist/chunk-RQT7M6MU.js.map +1 -0
  105. package/dist/chunk-U5YJQYLZ.cjs +57 -0
  106. package/dist/chunk-U5YJQYLZ.cjs.map +1 -0
  107. package/dist/chunk-V6TY7KAL.js +8 -0
  108. package/dist/chunk-V6TY7KAL.js.map +1 -0
  109. package/dist/chunk-VTDJEUFS.js +57 -0
  110. package/dist/chunk-VTDJEUFS.js.map +1 -0
  111. package/dist/chunk-XCNU7EEC.js +48 -0
  112. package/dist/chunk-XCNU7EEC.js.map +1 -0
  113. package/dist/chunk-XI2BOYEI.js +109 -0
  114. package/dist/chunk-XI2BOYEI.js.map +1 -0
  115. package/dist/chunk-XKSFDPDY.cjs +216 -0
  116. package/dist/chunk-XKSFDPDY.cjs.map +1 -0
  117. package/dist/chunk-XU7YEPML.js +46 -0
  118. package/dist/chunk-XU7YEPML.js.map +1 -0
  119. package/dist/chunk-YIA5B3QT.js +31 -0
  120. package/dist/chunk-YIA5B3QT.js.map +1 -0
  121. package/dist/index.cjs +97 -0
  122. package/dist/index.cjs.map +1 -0
  123. package/dist/index.d.cts +64 -0
  124. package/dist/index.d.ts +64 -0
  125. package/dist/index.js +97 -0
  126. package/dist/index.js.map +1 -0
  127. package/package.json +119 -10
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-LIUHRGEK.cjs","../src/adapters/workers-ai-image.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;AChBA,iDAAiC;AA4B1B,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,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-LIUHRGEK.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} 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;\n}[keyof AiModels];\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\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"]}
@@ -0,0 +1,48 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
2
+
3
+ var _chunkGOU66I5Tcjs = require('./chunk-GOU66I5T.cjs');
4
+
5
+ // src/adapters/grok.ts
6
+
7
+
8
+
9
+
10
+
11
+
12
+ var _aigrok = require('@tanstack/ai-grok');
13
+ function createGrokChat(model, config) {
14
+ return new (0, _aigrok.GrokTextAdapter)(
15
+ {
16
+ apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
17
+ fetch: _chunkGOU66I5Tcjs.createGatewayFetch.call(void 0, "grok", config)
18
+ },
19
+ model
20
+ );
21
+ }
22
+ function createGrokImage(model, config) {
23
+ return new (0, _aigrok.GrokImageAdapter)(
24
+ {
25
+ apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
26
+ fetch: _chunkGOU66I5Tcjs.createGatewayFetch.call(void 0, "grok", config)
27
+ },
28
+ model
29
+ );
30
+ }
31
+ function createGrokSummarize(model, config) {
32
+ return new (0, _aigrok.GrokSummarizeAdapter)(
33
+ {
34
+ apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
35
+ fetch: _chunkGOU66I5Tcjs.createGatewayFetch.call(void 0, "grok", config)
36
+ },
37
+ model
38
+ );
39
+ }
40
+
41
+
42
+
43
+
44
+
45
+
46
+
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-M6NETFDR.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-M6NETFDR.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-M6NETFDR.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"]}
@@ -0,0 +1,54 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
2
+
3
+ var _chunkGOU66I5Tcjs = require('./chunk-GOU66I5T.cjs');
4
+
5
+ // src/adapters/openrouter.ts
6
+
7
+
8
+
9
+
10
+ var _aiopenrouter = require('@tanstack/ai-openrouter');
11
+ var _sdk = require('@openrouter/sdk');
12
+ function buildOpenRouterConfig(config) {
13
+ const httpClient = new (0, _sdk.HTTPClient)({
14
+ fetcher: _chunkGOU66I5Tcjs.createGatewayFetch.call(void 0, "openrouter", config)
15
+ });
16
+ return {
17
+ apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
18
+ httpClient
19
+ };
20
+ }
21
+ function buildOpenRouterImageConfig(config) {
22
+ const httpClient = new (0, _sdk.HTTPClient)({
23
+ fetcher: _chunkGOU66I5Tcjs.createGatewayFetch.call(void 0, "openrouter", config)
24
+ });
25
+ return {
26
+ apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
27
+ httpClient
28
+ };
29
+ }
30
+ function buildOpenRouterSummarizeConfig(config) {
31
+ const httpClient = new (0, _sdk.HTTPClient)({
32
+ fetcher: _chunkGOU66I5Tcjs.createGatewayFetch.call(void 0, "openrouter", config)
33
+ });
34
+ return {
35
+ apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
36
+ httpClient
37
+ };
38
+ }
39
+ function createOpenRouterChat(model, config) {
40
+ return new (0, _aiopenrouter.OpenRouterTextAdapter)(buildOpenRouterConfig(config), model);
41
+ }
42
+ function createOpenRouterImage(model, config) {
43
+ return new (0, _aiopenrouter.OpenRouterImageAdapter)(buildOpenRouterImageConfig(config), model);
44
+ }
45
+ function createOpenRouterSummarize(model, config) {
46
+ return new (0, _aiopenrouter.OpenRouterSummarizeAdapter)(buildOpenRouterSummarizeConfig(config), model);
47
+ }
48
+
49
+
50
+
51
+
52
+
53
+ exports.createOpenRouterChat = createOpenRouterChat; exports.createOpenRouterImage = createOpenRouterImage; exports.createOpenRouterSummarize = createOpenRouterSummarize;
54
+ //# sourceMappingURL=chunk-O2C4CR57.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-O2C4CR57.cjs","../src/adapters/openrouter.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA;AACC;AACA;AACA;AAAA,uDAIM;AACP,sCAA2B;AAY3B,SAAS,qBAAA,CAAsB,MAAA,EAAmD;AACjF,EAAA,MAAM,WAAA,EAAa,IAAI,oBAAA,CAAW;AAAA,IACjC,OAAA,EAAS,kDAAA,YAAmB,EAAc,MAAM;AAAA,EACjD,CAAC,CAAA;AACD,EAAA,OAAO;AAAA,IACN,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,IACzB;AAAA,EACD,CAAA;AACD;AAWA,SAAS,0BAAA,CAA2B,MAAA,EAAwD;AAC3F,EAAA,MAAM,WAAA,EAAa,IAAI,oBAAA,CAAW;AAAA,IACjC,OAAA,EAAS,kDAAA,YAAmB,EAAc,MAAM;AAAA,EACjD,CAAC,CAAA;AACD,EAAA,OAAO;AAAA,IACN,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,IACzB;AAAA,EACD,CAAA;AACD;AASA,SAAS,8BAAA,CACR,MAAA,EAC4B;AAC5B,EAAA,MAAM,WAAA,EAAa,IAAI,oBAAA,CAAW;AAAA,IACjC,OAAA,EAAS,kDAAA,YAAmB,EAAc,MAAM;AAAA,EACjD,CAAC,CAAA;AACD,EAAA,OAAO;AAAA,IACN,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,IACzB;AAAA,EACD,CAAA;AACD;AA0BO,SAAS,oBAAA,CACf,KAAA,EACA,MAAA,EACiB;AAEjB,EAAA,OAAO,IAAI,wCAAA,CAAsB,qBAAA,CAAsB,MAAM,CAAA,EAAG,KAAY,CAAA;AAC7E;AASO,SAAS,qBAAA,CACf,KAAA,EACA,MAAA,EACC;AAED,EAAA,OAAO,IAAI,yCAAA,CAAuB,0BAAA,CAA2B,MAAM,CAAA,EAAG,KAAY,CAAA;AACnF;AASO,SAAS,yBAAA,CACf,KAAA,EACA,MAAA,EACC;AAED,EAAA,OAAO,IAAI,6CAAA,CAA2B,8BAAA,CAA+B,MAAM,CAAA,EAAG,KAAY,CAAA;AAC3F;ADjFA;AACA;AACE;AACA;AACA;AACF,0KAAC","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-O2C4CR57.cjs","sourcesContent":[null,"import {\n\tOpenRouterTextAdapter,\n\tOpenRouterImageAdapter,\n\tOpenRouterSummarizeAdapter,\n\ttype OpenRouterConfig,\n\ttype OpenRouterImageConfig,\n\ttype OpenRouterSummarizeConfig,\n} from \"@tanstack/ai-openrouter\";\nimport { HTTPClient } from \"@openrouter/sdk\";\nimport { createGatewayFetch, type AiGatewayAdapterConfig } from \"../utils/create-fetcher\";\nimport type { AnyTextAdapter } from \"@tanstack/ai\";\n\nexport type OpenRouterGatewayConfig = AiGatewayAdapterConfig;\n\n/**\n * Build OpenRouter config that routes requests through AI Gateway.\n *\n * The OpenRouter SDK accepts an `httpClient` with a custom `fetcher`,\n * which we use to inject the AI Gateway fetch.\n */\nfunction buildOpenRouterConfig(config: OpenRouterGatewayConfig): OpenRouterConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t};\n}\n\n/**\n * Build OpenRouter image config that routes requests through AI Gateway.\n *\n * `OpenRouterImageConfig` extends `OpenRouterClientConfig` which declares\n * `baseURL` and `apiKey` but not `httpClient`. However, the image adapter\n * internally creates an `OpenRouter` SDK instance (which does accept\n * `httpClient` via `SDKOptions`). The double-cast is needed because the\n * declared config type is narrower than what the SDK constructor accepts.\n */\nfunction buildOpenRouterImageConfig(config: OpenRouterGatewayConfig): OpenRouterImageConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t} as unknown as OpenRouterImageConfig;\n}\n\n/**\n * Build OpenRouter summarize config.\n *\n * `OpenRouterSummarizeConfig` extends `OpenRouterConfig` with optional\n * `temperature` and `maxTokens` fields. Since those are optional, a plain\n * `OpenRouterConfig` is structurally compatible — no cast needed.\n */\nfunction buildOpenRouterSummarizeConfig(\n\tconfig: OpenRouterGatewayConfig,\n): OpenRouterSummarizeConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// Chat model type — OpenRouter supports many models; we use a loose string type\n// ---------------------------------------------------------------------------\n\n/** OpenRouter chat model identifier. Accepts any string since OpenRouter proxies hundreds of models. */\nexport type OpenRouterChatModel = string;\n\n/** OpenRouter image model identifier. */\nexport type OpenRouterImageModel = string;\n\n/** OpenRouter summarize model identifier (same as chat models). */\nexport type OpenRouterSummarizeModel = string;\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\n/**\n * Creates an OpenRouter chat adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use (e.g. \"openai/gpt-4o\", \"anthropic/claude-sonnet-4-5\")\n * @param config Configuration options\n */\nexport function createOpenRouterChat(\n\tmodel: OpenRouterChatModel,\n\tconfig: OpenRouterGatewayConfig,\n): AnyTextAdapter {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterTextAdapter(buildOpenRouterConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter image adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The image model to use\n * @param config Configuration options\n */\nexport function createOpenRouterImage(\n\tmodel: OpenRouterImageModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterImageAdapter(buildOpenRouterImageConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter summarize adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use for summarization\n * @param config Configuration options\n */\nexport function createOpenRouterSummarize(\n\tmodel: OpenRouterSummarizeModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterSummarizeAdapter(buildOpenRouterSummarizeConfig(config), model as any);\n}\n"]}
@@ -0,0 +1,216 @@
1
+ import {
2
+ uint8ArrayToBase64
3
+ } from "./chunk-XU7YEPML.js";
4
+ import {
5
+ workersAiRestFetch,
6
+ workersAiRestFetchBinary
7
+ } from "./chunk-2VII5BK2.js";
8
+ import {
9
+ createGatewayFetch,
10
+ isDirectBindingConfig,
11
+ isDirectCredentialsConfig
12
+ } from "./chunk-F5YJMXZR.js";
13
+ import {
14
+ __publicField
15
+ } from "./chunk-V6TY7KAL.js";
16
+
17
+ // src/adapters/workers-ai-transcription.ts
18
+ import { BaseTranscriptionAdapter } from "@tanstack/ai/adapters";
19
+ var WorkersAiTranscriptionAdapter = class extends BaseTranscriptionAdapter {
20
+ constructor(config, model) {
21
+ super({}, model);
22
+ __publicField(this, "name", "workers-ai-transcription");
23
+ __publicField(this, "adapterConfig");
24
+ this.adapterConfig = config;
25
+ }
26
+ async transcribe(options) {
27
+ const { audio, language, prompt, modelOptions } = options;
28
+ const audioBytes = await normalizeAudioToBytes(audio);
29
+ const extra = { ...modelOptions };
30
+ if (language) extra.language = language;
31
+ if (prompt) extra.initial_prompt = prompt;
32
+ const audioPayload = this.buildAudioPayload(audioBytes, audio);
33
+ if (isDirectBindingConfig(this.adapterConfig)) {
34
+ return this.transcribeViaBinding(audioPayload, extra);
35
+ }
36
+ if (isDirectCredentialsConfig(this.adapterConfig)) {
37
+ if (this.model === "@cf/deepgram/nova-3") {
38
+ return this.transcribeViaRestBinary(audioBytes, audio, extra);
39
+ }
40
+ return this.transcribeViaRest(audioPayload, extra);
41
+ }
42
+ return this.transcribeViaGateway(audioPayload, extra);
43
+ }
44
+ /**
45
+ * Build the audio field for the request payload, handling model-specific formats.
46
+ *
47
+ * - `@cf/deepgram/nova-3` requires `{ body: base64, contentType: "audio/..." }`
48
+ * - `@cf/openai/whisper-large-v3-turbo` REST/gateway accepts a base64 string
49
+ * - Other Whisper models accept `number[]` (binding) or base64 (REST)
50
+ */
51
+ buildAudioPayload(audioBytes, originalAudio) {
52
+ if (this.model === "@cf/deepgram/nova-3") {
53
+ const b64 = uint8ArrayToBase64(new Uint8Array(audioBytes));
54
+ const contentType = detectAudioContentType(originalAudio);
55
+ return { audio: { body: b64, contentType } };
56
+ }
57
+ if (this.model === "@cf/openai/whisper-large-v3-turbo") {
58
+ return { audio: uint8ArrayToBase64(new Uint8Array(audioBytes)) };
59
+ }
60
+ return { audio: audioBytes };
61
+ }
62
+ async transcribeViaBinding(audioPayload, options) {
63
+ const ai = this.adapterConfig.binding;
64
+ const result = await ai.run(this.model, {
65
+ ...audioPayload,
66
+ ...options
67
+ });
68
+ return this.normalizeResult(result);
69
+ }
70
+ async transcribeViaRest(audioPayload, options) {
71
+ const config = this.adapterConfig;
72
+ const response = await workersAiRestFetch(
73
+ config,
74
+ this.model,
75
+ { ...audioPayload, ...options },
76
+ {
77
+ label: "Workers AI transcription",
78
+ signal: options.signal
79
+ }
80
+ );
81
+ const data = await response.json();
82
+ return this.normalizeResult(data.result ?? data);
83
+ }
84
+ /**
85
+ * Transcribe via REST using raw binary audio.
86
+ * Required for models like Deepgram Nova-3 that expect raw audio bytes
87
+ * with a Content-Type header (e.g. "audio/wav") instead of JSON.
88
+ */
89
+ async transcribeViaRestBinary(audioBytes, originalAudio, options) {
90
+ const config = this.adapterConfig;
91
+ const contentType = detectAudioContentType(originalAudio);
92
+ const response = await workersAiRestFetchBinary(
93
+ config,
94
+ this.model,
95
+ new Uint8Array(audioBytes),
96
+ contentType,
97
+ {
98
+ label: "Workers AI transcription",
99
+ signal: options.signal
100
+ }
101
+ );
102
+ const data = await response.json();
103
+ return this.normalizeResult(data.result ?? data);
104
+ }
105
+ async transcribeViaGateway(audioPayload, options) {
106
+ const gatewayConfig = this.adapterConfig;
107
+ const gatewayFetch = createGatewayFetch("workers-ai", gatewayConfig);
108
+ const response = await gatewayFetch("https://api.cloudflare.com/v1/audio/transcriptions", {
109
+ method: "POST",
110
+ body: JSON.stringify({
111
+ model: this.model,
112
+ ...audioPayload,
113
+ ...options
114
+ })
115
+ });
116
+ if (!response.ok) {
117
+ const errorText = await response.text();
118
+ throw new Error(
119
+ `Workers AI transcription gateway request failed (${response.status}): ${errorText}`
120
+ );
121
+ }
122
+ const data = await response.json();
123
+ return this.normalizeResult(data);
124
+ }
125
+ /**
126
+ * Normalize Workers AI transcription results into the standard
127
+ * TanStack AI TranscriptionResult shape.
128
+ *
129
+ * Handles three response formats:
130
+ * - Whisper: `{ text, words?, vtt? }`
131
+ * - Whisper v3-turbo: `{ text, segments?, transcription_info? }`
132
+ * - Deepgram Nova-3: `{ results: { channels: [{ alternatives: [{ transcript, words }] }] } }`
133
+ */
134
+ normalizeResult(raw) {
135
+ const results = raw.results;
136
+ if (results?.channels) {
137
+ const channels = results.channels;
138
+ const alt = channels?.[0]?.alternatives?.[0];
139
+ const text = alt?.transcript ?? "";
140
+ const result2 = {
141
+ id: this.generateId(),
142
+ model: this.model,
143
+ text
144
+ };
145
+ if (alt?.words && Array.isArray(alt.words)) {
146
+ result2.words = alt.words.map((w) => ({
147
+ word: w.word ?? "",
148
+ start: w.start ?? 0,
149
+ end: w.end ?? 0
150
+ }));
151
+ }
152
+ return result2;
153
+ }
154
+ const result = {
155
+ id: this.generateId(),
156
+ model: this.model,
157
+ text: raw.text ?? ""
158
+ };
159
+ const transcriptionInfo = raw.transcription_info;
160
+ if (transcriptionInfo?.language) {
161
+ result.language = transcriptionInfo.language;
162
+ }
163
+ if (transcriptionInfo?.duration != null) {
164
+ result.duration = transcriptionInfo.duration;
165
+ }
166
+ if (raw.segments && Array.isArray(raw.segments)) {
167
+ result.segments = raw.segments.map((seg, idx) => ({
168
+ id: idx,
169
+ text: seg.text ?? "",
170
+ start: seg.start ?? 0,
171
+ end: seg.end ?? 0
172
+ }));
173
+ }
174
+ if (raw.words && Array.isArray(raw.words)) {
175
+ result.words = raw.words.map((w) => ({
176
+ word: w.word ?? "",
177
+ start: w.start ?? 0,
178
+ end: w.end ?? 0
179
+ }));
180
+ }
181
+ return result;
182
+ }
183
+ };
184
+ function createWorkersAiTranscription(model, config) {
185
+ return new WorkersAiTranscriptionAdapter(config, model);
186
+ }
187
+ async function normalizeAudioToBytes(audio) {
188
+ if (audio instanceof ArrayBuffer) {
189
+ return Array.from(new Uint8Array(audio));
190
+ }
191
+ if (audio instanceof Blob) {
192
+ const buffer = await audio.arrayBuffer();
193
+ return Array.from(new Uint8Array(buffer));
194
+ }
195
+ if (typeof audio === "string") {
196
+ const binary = atob(audio);
197
+ const bytes = new Uint8Array(binary.length);
198
+ for (let i = 0; i < binary.length; i++) {
199
+ bytes[i] = binary.charCodeAt(i);
200
+ }
201
+ return Array.from(bytes);
202
+ }
203
+ throw new Error("Unsupported audio format. Expected string, File, Blob, or ArrayBuffer.");
204
+ }
205
+ function detectAudioContentType(audio) {
206
+ if (audio instanceof Blob && audio.type) {
207
+ return audio.type;
208
+ }
209
+ return "audio/wav";
210
+ }
211
+
212
+ export {
213
+ WorkersAiTranscriptionAdapter,
214
+ createWorkersAiTranscription
215
+ };
216
+ //# sourceMappingURL=chunk-PLTFCUMO.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/adapters/workers-ai-transcription.ts"],"sourcesContent":["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} 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\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\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"],"mappings":";;;;;;;;;;;;;;;;;AAAA,SAAS,gCAAgC;AAuClC,IAAM,gCAAN,cAA4C,yBAAsD;AAAA,EAIxG,YAAY,QAAgC,OAAoC;AAC/E,UAAM,CAAC,GAAG,KAAK;AAJhB,wBAAS,QAAO;AAChB,wBAAQ;AAIP,SAAK,gBAAgB;AAAA,EACtB;AAAA,EAEA,MAAM,WAAW,SAA6D;AAC7E,UAAM,EAAE,OAAO,UAAU,QAAQ,aAAa,IAAI;AAGlD,UAAM,aAAa,MAAM,sBAAsB,KAAK;AAEpD,UAAM,QAAiC,EAAE,GAAG,aAAa;AACzD,QAAI,SAAU,OAAM,WAAW;AAC/B,QAAI,OAAQ,OAAM,iBAAiB;AAOnC,UAAM,eAAe,KAAK,kBAAkB,YAAY,KAAK;AAE7D,QAAI,sBAAsB,KAAK,aAAa,GAAG;AAC9C,aAAO,KAAK,qBAAqB,cAAc,KAAK;AAAA,IACrD;AAEA,QAAI,0BAA0B,KAAK,aAAa,GAAG;AAElD,UAAI,KAAK,UAAU,uBAAuB;AACzC,eAAO,KAAK,wBAAwB,YAAY,OAAO,KAAK;AAAA,MAC7D;AACA,aAAO,KAAK,kBAAkB,cAAc,KAAK;AAAA,IAClD;AAEA,WAAO,KAAK,qBAAqB,cAAc,KAAK;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASQ,kBACP,YACA,eAC0B;AAC1B,QAAI,KAAK,UAAU,uBAAuB;AACzC,YAAM,MAAM,mBAAmB,IAAI,WAAW,UAAU,CAAC;AACzD,YAAM,cAAc,uBAAuB,aAAa;AACxD,aAAO,EAAE,OAAO,EAAE,MAAM,KAAK,YAAY,EAAE;AAAA,IAC5C;AAEA,QAAI,KAAK,UAAU,qCAAqC;AACvD,aAAO,EAAE,OAAO,mBAAmB,IAAI,WAAW,UAAU,CAAC,EAAE;AAAA,IAChE;AAEA,WAAO,EAAE,OAAO,WAAW;AAAA,EAC5B;AAAA,EAEA,MAAc,qBACb,cACA,SAC+B;AAC/B,UAAM,KAAM,KAAK,cAA+C;AAChE,UAAM,SAAU,MAAM,GAAG,IAAI,KAAK,OAAO;AAAA,MACxC,GAAG;AAAA,MACH,GAAG;AAAA,IACJ,CAAC;AACD,WAAO,KAAK,gBAAgB,MAAM;AAAA,EACnC;AAAA,EAEA,MAAc,kBACb,cACA,SAC+B;AAC/B,UAAM,SAAS,KAAK;AAEpB,UAAM,WAAW,MAAM;AAAA,MACtB;AAAA,MACA,KAAK;AAAA,MACL,EAAE,GAAG,cAAc,GAAG,QAAQ;AAAA,MAC9B;AAAA,QACC,OAAO;AAAA,QACP,QAAS,QAAqC;AAAA,MAC/C;AAAA,IACD;AAEA,UAAM,OAAQ,MAAM,SAAS,KAAK;AAMlC,WAAO,KAAK,gBAAgB,KAAK,UAAU,IAAI;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAc,wBACb,YACA,eACA,SAC+B;AAC/B,UAAM,SAAS,KAAK;AACpB,UAAM,cAAc,uBAAuB,aAAa;AAExD,UAAM,WAAW,MAAM;AAAA,MACtB;AAAA,MACA,KAAK;AAAA,MACL,IAAI,WAAW,UAAU;AAAA,MACzB;AAAA,MACA;AAAA,QACC,OAAO;AAAA,QACP,QAAS,QAAqC;AAAA,MAC/C;AAAA,IACD;AAEA,UAAM,OAAQ,MAAM,SAAS,KAAK;AAIlC,WAAO,KAAK,gBAAgB,KAAK,UAAU,IAAI;AAAA,EAChD;AAAA,EAEA,MAAc,qBACb,cACA,SAC+B;AAC/B,UAAM,gBAAgB,KAAK;AAC3B,UAAM,eAAe,mBAAmB,cAAc,aAAa;AAKnE,UAAM,WAAW,MAAM,aAAa,sDAAsD;AAAA,MACzF,QAAQ;AAAA,MACR,MAAM,KAAK,UAAU;AAAA,QACpB,OAAO,KAAK;AAAA,QACZ,GAAG;AAAA,QACH,GAAG;AAAA,MACJ,CAAC;AAAA,IACF,CAAC;AAED,QAAI,CAAC,SAAS,IAAI;AACjB,YAAM,YAAY,MAAM,SAAS,KAAK;AACtC,YAAM,IAAI;AAAA,QACT,oDAAoD,SAAS,MAAM,MAAM,SAAS;AAAA,MACnF;AAAA,IACD;AAEA,UAAM,OAAQ,MAAM,SAAS,KAAK;AAClC,WAAO,KAAK,gBAAgB,IAAI;AAAA,EACjC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWQ,gBAAgB,KAAmD;AAE1E,UAAM,UAAU,IAAI;AACpB,QAAI,SAAS,UAAU;AACtB,YAAM,WAAW,QAAQ;AAOzB,YAAM,MAAM,WAAW,CAAC,GAAG,eAAe,CAAC;AAC3C,YAAM,OAAO,KAAK,cAAc;AAChC,YAAMA,UAA8B;AAAA,QACnC,IAAI,KAAK,WAAW;AAAA,QACpB,OAAO,KAAK;AAAA,QACZ;AAAA,MACD;AACA,UAAI,KAAK,SAAS,MAAM,QAAQ,IAAI,KAAK,GAAG;AAC3C,QAAAA,QAAO,QAAQ,IAAI,MAAM,IAAI,CAAC,OAAO;AAAA,UACpC,MAAM,EAAE,QAAQ;AAAA,UAChB,OAAO,EAAE,SAAS;AAAA,UAClB,KAAK,EAAE,OAAO;AAAA,QACf,EAAE;AAAA,MACH;AACA,aAAOA;AAAA,IACR;AAIA,UAAM,SAA8B;AAAA,MACnC,IAAI,KAAK,WAAW;AAAA,MACpB,OAAO,KAAK;AAAA,MACZ,MAAO,IAAI,QAAmB;AAAA,IAC/B;AAGA,UAAM,oBAAoB,IAAI;AAC9B,QAAI,mBAAmB,UAAU;AAChC,aAAO,WAAW,kBAAkB;AAAA,IACrC;AAGA,QAAI,mBAAmB,YAAY,MAAM;AACxC,aAAO,WAAW,kBAAkB;AAAA,IACrC;AAGA,QAAI,IAAI,YAAY,MAAM,QAAQ,IAAI,QAAQ,GAAG;AAChD,aAAO,WAAW,IAAI,SAAS,IAAI,CAAC,KAA8B,SAAiB;AAAA,QAClF,IAAI;AAAA,QACJ,MAAO,IAAI,QAAmB;AAAA,QAC9B,OAAQ,IAAI,SAAoB;AAAA,QAChC,KAAM,IAAI,OAAkB;AAAA,MAC7B,EAAE;AAAA,IACH;AAGA,QAAI,IAAI,SAAS,MAAM,QAAQ,IAAI,KAAK,GAAG;AAC1C,aAAO,QAAQ,IAAI,MAAM,IAAI,CAAC,OAAgC;AAAA,QAC7D,MAAO,EAAE,QAAmB;AAAA,QAC5B,OAAQ,EAAE,SAAoB;AAAA,QAC9B,KAAM,EAAE,OAAkB;AAAA,MAC3B,EAAE;AAAA,IACH;AAEA,WAAO;AAAA,EACR;AACD;AA0BO,SAAS,6BACf,OACA,QACC;AACD,SAAO,IAAI,8BAA8B,QAAQ,KAAK;AACvD;AAaA,eAAe,sBAAsB,OAA8D;AAClG,MAAI,iBAAiB,aAAa;AACjC,WAAO,MAAM,KAAK,IAAI,WAAW,KAAK,CAAC;AAAA,EACxC;AAEA,MAAI,iBAAiB,MAAM;AAE1B,UAAM,SAAS,MAAM,MAAM,YAAY;AACvC,WAAO,MAAM,KAAK,IAAI,WAAW,MAAM,CAAC;AAAA,EACzC;AAEA,MAAI,OAAO,UAAU,UAAU;AAE9B,UAAM,SAAS,KAAK,KAAK;AACzB,UAAM,QAAQ,IAAI,WAAW,OAAO,MAAM;AAC1C,aAAS,IAAI,GAAG,IAAI,OAAO,QAAQ,KAAK;AACvC,YAAM,CAAC,IAAI,OAAO,WAAW,CAAC;AAAA,IAC/B;AACA,WAAO,MAAM,KAAK,KAAK;AAAA,EACxB;AAEA,QAAM,IAAI,MAAM,wEAAwE;AACzF;AASA,SAAS,uBAAuB,OAAmD;AAElF,MAAI,iBAAiB,QAAQ,MAAM,MAAM;AACxC,WAAO,MAAM;AAAA,EACd;AAIA,SAAO;AACR;","names":["result"]}
@@ -0,0 +1,54 @@
1
+ import {
2
+ createGatewayFetch
3
+ } from "./chunk-F5YJMXZR.js";
4
+
5
+ // src/adapters/openrouter.ts
6
+ import {
7
+ OpenRouterTextAdapter,
8
+ OpenRouterImageAdapter,
9
+ OpenRouterSummarizeAdapter
10
+ } from "@tanstack/ai-openrouter";
11
+ import { HTTPClient } from "@openrouter/sdk";
12
+ function buildOpenRouterConfig(config) {
13
+ const httpClient = new HTTPClient({
14
+ fetcher: createGatewayFetch("openrouter", config)
15
+ });
16
+ return {
17
+ apiKey: config.apiKey ?? "unused",
18
+ httpClient
19
+ };
20
+ }
21
+ function buildOpenRouterImageConfig(config) {
22
+ const httpClient = new HTTPClient({
23
+ fetcher: createGatewayFetch("openrouter", config)
24
+ });
25
+ return {
26
+ apiKey: config.apiKey ?? "unused",
27
+ httpClient
28
+ };
29
+ }
30
+ function buildOpenRouterSummarizeConfig(config) {
31
+ const httpClient = new HTTPClient({
32
+ fetcher: createGatewayFetch("openrouter", config)
33
+ });
34
+ return {
35
+ apiKey: config.apiKey ?? "unused",
36
+ httpClient
37
+ };
38
+ }
39
+ function createOpenRouterChat(model, config) {
40
+ return new OpenRouterTextAdapter(buildOpenRouterConfig(config), model);
41
+ }
42
+ function createOpenRouterImage(model, config) {
43
+ return new OpenRouterImageAdapter(buildOpenRouterImageConfig(config), model);
44
+ }
45
+ function createOpenRouterSummarize(model, config) {
46
+ return new OpenRouterSummarizeAdapter(buildOpenRouterSummarizeConfig(config), model);
47
+ }
48
+
49
+ export {
50
+ createOpenRouterChat,
51
+ createOpenRouterImage,
52
+ createOpenRouterSummarize
53
+ };
54
+ //# sourceMappingURL=chunk-QRHKPL75.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/adapters/openrouter.ts"],"sourcesContent":["import {\n\tOpenRouterTextAdapter,\n\tOpenRouterImageAdapter,\n\tOpenRouterSummarizeAdapter,\n\ttype OpenRouterConfig,\n\ttype OpenRouterImageConfig,\n\ttype OpenRouterSummarizeConfig,\n} from \"@tanstack/ai-openrouter\";\nimport { HTTPClient } from \"@openrouter/sdk\";\nimport { createGatewayFetch, type AiGatewayAdapterConfig } from \"../utils/create-fetcher\";\nimport type { AnyTextAdapter } from \"@tanstack/ai\";\n\nexport type OpenRouterGatewayConfig = AiGatewayAdapterConfig;\n\n/**\n * Build OpenRouter config that routes requests through AI Gateway.\n *\n * The OpenRouter SDK accepts an `httpClient` with a custom `fetcher`,\n * which we use to inject the AI Gateway fetch.\n */\nfunction buildOpenRouterConfig(config: OpenRouterGatewayConfig): OpenRouterConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t};\n}\n\n/**\n * Build OpenRouter image config that routes requests through AI Gateway.\n *\n * `OpenRouterImageConfig` extends `OpenRouterClientConfig` which declares\n * `baseURL` and `apiKey` but not `httpClient`. However, the image adapter\n * internally creates an `OpenRouter` SDK instance (which does accept\n * `httpClient` via `SDKOptions`). The double-cast is needed because the\n * declared config type is narrower than what the SDK constructor accepts.\n */\nfunction buildOpenRouterImageConfig(config: OpenRouterGatewayConfig): OpenRouterImageConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t} as unknown as OpenRouterImageConfig;\n}\n\n/**\n * Build OpenRouter summarize config.\n *\n * `OpenRouterSummarizeConfig` extends `OpenRouterConfig` with optional\n * `temperature` and `maxTokens` fields. Since those are optional, a plain\n * `OpenRouterConfig` is structurally compatible — no cast needed.\n */\nfunction buildOpenRouterSummarizeConfig(\n\tconfig: OpenRouterGatewayConfig,\n): OpenRouterSummarizeConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// Chat model type — OpenRouter supports many models; we use a loose string type\n// ---------------------------------------------------------------------------\n\n/** OpenRouter chat model identifier. Accepts any string since OpenRouter proxies hundreds of models. */\nexport type OpenRouterChatModel = string;\n\n/** OpenRouter image model identifier. */\nexport type OpenRouterImageModel = string;\n\n/** OpenRouter summarize model identifier (same as chat models). */\nexport type OpenRouterSummarizeModel = string;\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\n/**\n * Creates an OpenRouter chat adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use (e.g. \"openai/gpt-4o\", \"anthropic/claude-sonnet-4-5\")\n * @param config Configuration options\n */\nexport function createOpenRouterChat(\n\tmodel: OpenRouterChatModel,\n\tconfig: OpenRouterGatewayConfig,\n): AnyTextAdapter {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterTextAdapter(buildOpenRouterConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter image adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The image model to use\n * @param config Configuration options\n */\nexport function createOpenRouterImage(\n\tmodel: OpenRouterImageModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterImageAdapter(buildOpenRouterImageConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter summarize adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use for summarization\n * @param config Configuration options\n */\nexport function createOpenRouterSummarize(\n\tmodel: OpenRouterSummarizeModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterSummarizeAdapter(buildOpenRouterSummarizeConfig(config), model as any);\n}\n"],"mappings":";;;;;AAAA;AAAA,EACC;AAAA,EACA;AAAA,EACA;AAAA,OAIM;AACP,SAAS,kBAAkB;AAY3B,SAAS,sBAAsB,QAAmD;AACjF,QAAM,aAAa,IAAI,WAAW;AAAA,IACjC,SAAS,mBAAmB,cAAc,MAAM;AAAA,EACjD,CAAC;AACD,SAAO;AAAA,IACN,QAAQ,OAAO,UAAU;AAAA,IACzB;AAAA,EACD;AACD;AAWA,SAAS,2BAA2B,QAAwD;AAC3F,QAAM,aAAa,IAAI,WAAW;AAAA,IACjC,SAAS,mBAAmB,cAAc,MAAM;AAAA,EACjD,CAAC;AACD,SAAO;AAAA,IACN,QAAQ,OAAO,UAAU;AAAA,IACzB;AAAA,EACD;AACD;AASA,SAAS,+BACR,QAC4B;AAC5B,QAAM,aAAa,IAAI,WAAW;AAAA,IACjC,SAAS,mBAAmB,cAAc,MAAM;AAAA,EACjD,CAAC;AACD,SAAO;AAAA,IACN,QAAQ,OAAO,UAAU;AAAA,IACzB;AAAA,EACD;AACD;AA0BO,SAAS,qBACf,OACA,QACiB;AAEjB,SAAO,IAAI,sBAAsB,sBAAsB,MAAM,GAAG,KAAY;AAC7E;AASO,SAAS,sBACf,OACA,QACC;AAED,SAAO,IAAI,uBAAuB,2BAA2B,MAAM,GAAG,KAAY;AACnF;AASO,SAAS,0BACf,OACA,QACC;AAED,SAAO,IAAI,2BAA2B,+BAA+B,MAAM,GAAG,KAAY;AAC3F;","names":[]}
@@ -0,0 +1,65 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }// src/adapters/gemini.ts
2
+
3
+
4
+
5
+
6
+
7
+
8
+
9
+
10
+
11
+ var _aigemini = require('@tanstack/ai-gemini');
12
+ function buildGeminiGatewayConfig(config) {
13
+ if ("binding" in config) {
14
+ throw new Error(
15
+ "Gemini adapters do not support binding config. The Google GenAI SDK does not support a custom fetch function \u2014 only credential-based config ({ accountId, gatewayId }) is supported. See https://github.com/googleapis/js-genai/issues/999"
16
+ );
17
+ }
18
+ const headers = {};
19
+ if (config.cfApiKey) {
20
+ headers["cf-aig-authorization"] = `Bearer ${config.cfApiKey}`;
21
+ }
22
+ if (config.skipCache) {
23
+ headers["cf-aig-skip-cache"] = "true";
24
+ }
25
+ if (typeof config.cacheTtl === "number") {
26
+ headers["cf-aig-cache-ttl"] = String(config.cacheTtl);
27
+ }
28
+ if (typeof config.customCacheKey === "string") {
29
+ headers["cf-aig-cache-key"] = config.customCacheKey;
30
+ }
31
+ if (typeof config.metadata === "object") {
32
+ headers["cf-aig-metadata"] = JSON.stringify(config.metadata);
33
+ }
34
+ return {
35
+ apiKey: _nullishCoalesce(config.apiKey, () => ( "unused")),
36
+ httpOptions: {
37
+ baseUrl: `https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}/google-ai-studio`,
38
+ headers: Object.keys(headers).length > 0 ? headers : void 0
39
+ }
40
+ };
41
+ }
42
+ function createGeminiChat(model, config) {
43
+ return new (0, _aigemini.GeminiTextAdapter)(buildGeminiGatewayConfig(config), model);
44
+ }
45
+ function createGeminiImage(model, config) {
46
+ return new (0, _aigemini.GeminiImageAdapter)(buildGeminiGatewayConfig(config), model);
47
+ }
48
+ function createGeminiSummarize(model, config) {
49
+ return new (0, _aigemini.GeminiSummarizeAdapter)(buildGeminiGatewayConfig(config), model);
50
+ }
51
+ function createGeminiTts(model, config) {
52
+ return new (0, _aigemini.GeminiTTSAdapter)(buildGeminiGatewayConfig(config), model);
53
+ }
54
+
55
+
56
+
57
+
58
+
59
+
60
+
61
+
62
+
63
+
64
+ exports.GeminiTextModels = _aigemini.GeminiTextModels; exports.GeminiImageModels = _aigemini.GeminiImageModels; exports.GeminiSummarizeModels = _aigemini.GeminiSummarizeModels; exports.GeminiTTSModels = _aigemini.GeminiTTSModels; exports.createGeminiChat = createGeminiChat; exports.createGeminiImage = createGeminiImage; exports.createGeminiSummarize = createGeminiSummarize; exports.createGeminiTts = createGeminiTts;
65
+ //# sourceMappingURL=chunk-RGDUK5KX.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-RGDUK5KX.cjs","../src/adapters/gemini.ts"],"names":[],"mappings":"AAAA;ACAA;AACC;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AAAA,+CAIM;AAwBP,SAAS,wBAAA,CAAyB,MAAA,EAA6B;AAM9D,EAAA,GAAA,CAAI,UAAA,GAAa,MAAA,EAAQ;AACxB,IAAA,MAAM,IAAI,KAAA;AAAA,MACT;AAAA,IAID,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,QAAA,EAAkC,CAAC,CAAA;AACzC,EAAA,GAAA,CAAI,MAAA,CAAO,QAAA,EAAU;AACpB,IAAA,OAAA,CAAQ,sBAAsB,EAAA,EAAI,CAAA,OAAA,EAAU,MAAA,CAAO,QAAQ,CAAA,CAAA;AAC5D,EAAA;AACsB,EAAA;AACU,IAAA;AAChC,EAAA;AACyC,EAAA;AACY,IAAA;AACrD,EAAA;AAC+C,EAAA;AACT,IAAA;AACtC,EAAA;AACyC,EAAA;AACmB,IAAA;AAC5D,EAAA;AAEO,EAAA;AACmB,IAAA;AACZ,IAAA;AAC4C,MAAA;AACH,MAAA;AACtD,IAAA;AACD,EAAA;AACD;AAekB;AAC2C,EAAA;AAC7D;AASwF;AAC1B,EAAA;AAC9D;AASgG;AACpC,EAAA;AAC5D;AAWoF;AACrB,EAAA;AAC/D;AD1EgE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-RGDUK5KX.cjs","sourcesContent":[null,"import {\n\tGeminiTextAdapter,\n\tGeminiImageAdapter,\n\tGeminiSummarizeAdapter,\n\tGeminiTTSAdapter,\n\tGeminiTextModels,\n\tGeminiImageModels,\n\tGeminiSummarizeModels,\n\tGeminiTTSModels,\n\ttype GeminiTextModel,\n\ttype GeminiImageModel,\n\ttype GeminiSummarizeModel,\n} from \"@tanstack/ai-gemini\";\n\n/** Derived from GeminiTTSModels since @tanstack/ai-gemini doesn't export a GeminiTTSModel type. */\nexport type GeminiTTSModel = (typeof GeminiTTSModels)[number];\nimport type { AnyTextAdapter } from \"@tanstack/ai\";\nimport type { AiGatewayCredentialsConfig, AiGatewayConfig } from \"../utils/create-fetcher\";\n\n/**\n * Gemini-specific gateway config (credentials only, no binding support).\n * Includes cache control options from AiGatewayConfig.\n * See {@link https://github.com/googleapis/js-genai/issues/999 | googleapis/js-genai#999}.\n */\nexport type GeminiGatewayConfig = AiGatewayCredentialsConfig & AiGatewayConfig;\n\n/**\n * Build Gemini client config that routes through AI Gateway.\n * Since GeminiClientConfig extends GoogleGenAIOptions, we can inject\n * httpOptions.baseUrl directly — no subclassing needed.\n *\n * The Google GenAI SDK doesn't support a custom `fetch` override,\n * so we set the baseUrl to the AI Gateway endpoint for Google AI Studio.\n *\n * Tracking issue: https://github.com/googleapis/js-genai/issues/999\n */\nfunction buildGeminiGatewayConfig(config: GeminiGatewayConfig) {\n\t// Runtime guard: catch binding configs that bypass TypeScript (JS callers, `as any`, etc.)\n\t// We integrate with the Gemini SDK via `httpOptions` (baseUrl + headers), which allows\n\t// gateway routing and cache control but not request interception. A binding config\n\t// requires a custom `fetch` to route through the AI Gateway binding, and the Google\n\t// GenAI SDK doesn't support that yet.\n\tif (\"binding\" in config) {\n\t\tthrow new Error(\n\t\t\t\"Gemini adapters do not support binding config. \" +\n\t\t\t\t\"The Google GenAI SDK does not support a custom fetch function — \" +\n\t\t\t\t\"only credential-based config ({ accountId, gatewayId }) is supported. \" +\n\t\t\t\t\"See https://github.com/googleapis/js-genai/issues/999\",\n\t\t);\n\t}\n\n\tconst headers: Record<string, string> = {};\n\tif (config.cfApiKey) {\n\t\theaders[\"cf-aig-authorization\"] = `Bearer ${config.cfApiKey}`;\n\t}\n\tif (config.skipCache) {\n\t\theaders[\"cf-aig-skip-cache\"] = \"true\";\n\t}\n\tif (typeof config.cacheTtl === \"number\") {\n\t\theaders[\"cf-aig-cache-ttl\"] = String(config.cacheTtl);\n\t}\n\tif (typeof config.customCacheKey === \"string\") {\n\t\theaders[\"cf-aig-cache-key\"] = config.customCacheKey;\n\t}\n\tif (typeof config.metadata === \"object\") {\n\t\theaders[\"cf-aig-metadata\"] = JSON.stringify(config.metadata);\n\t}\n\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpOptions: {\n\t\t\tbaseUrl: `https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}/google-ai-studio`,\n\t\t\theaders: Object.keys(headers).length > 0 ? headers : undefined,\n\t\t},\n\t};\n}\n\n/** Alias for consistency with other providers (AnthropicChatModel, GrokChatModel, etc.) */\nexport type GeminiChatModel = GeminiTextModel;\n\n/**\n * Creates a Gemini adapter which uses Cloudflare AI Gateway.\n * Does not support the AI binding (Google GenAI SDK lacks custom fetch support).\n * See {@link https://github.com/googleapis/js-genai/issues/999 | googleapis/js-genai#999}.\n * @param model The Gemini model to use\n * @param config Configuration options (credentials only)\n */\nexport function createGeminiChat(\n\tmodel: GeminiChatModel,\n\tconfig: GeminiGatewayConfig,\n): AnyTextAdapter {\n\treturn new GeminiTextAdapter(buildGeminiGatewayConfig(config), model);\n}\n\n/**\n * Creates a Gemini Image adapter which uses Cloudflare AI Gateway.\n * Does not support the AI binding (Google GenAI SDK lacks custom fetch support).\n * See {@link https://github.com/googleapis/js-genai/issues/999 | googleapis/js-genai#999}.\n * @param model The Gemini model to use\n * @param config Configuration options (credentials only)\n */\nexport function createGeminiImage(model: GeminiImageModel, config: GeminiGatewayConfig) {\n\treturn new GeminiImageAdapter(buildGeminiGatewayConfig(config), model);\n}\n\n/**\n * Creates a Gemini Summarize adapter which uses Cloudflare AI Gateway.\n * Does not support the AI binding (Google GenAI SDK lacks custom fetch support).\n * See {@link https://github.com/googleapis/js-genai/issues/999 | googleapis/js-genai#999}.\n * @param model The Gemini model to use\n * @param config Configuration options (credentials only)\n */\nexport function createGeminiSummarize(model: GeminiSummarizeModel, config: GeminiGatewayConfig) {\n\treturn new GeminiSummarizeAdapter(buildGeminiGatewayConfig(config), model);\n}\n\n/**\n * Creates a Gemini TTS adapter which uses Cloudflare AI Gateway.\n * Does not support the AI binding (Google GenAI SDK lacks custom fetch support).\n * See {@link https://github.com/googleapis/js-genai/issues/999 | googleapis/js-genai#999}.\n *\n * @experimental Gemini TTS is an experimental feature and may change.\n * @param model The Gemini TTS model to use\n * @param config Configuration options (credentials only)\n */\nexport function createGeminiTts(model: GeminiTTSModel, config: GeminiGatewayConfig) {\n\treturn new GeminiTTSAdapter(buildGeminiGatewayConfig(config), model);\n}\n\nexport {\n\tGeminiTextModels,\n\tGeminiImageModels,\n\tGeminiSummarizeModels,\n\tGeminiTTSModels,\n\ttype GeminiTextModel,\n\ttype GeminiImageModel,\n\ttype GeminiSummarizeModel,\n};\n"]}
@@ -0,0 +1,96 @@
1
+ import {
2
+ binaryToBase64,
3
+ uint8ArrayToBase64
4
+ } from "./chunk-XU7YEPML.js";
5
+ import {
6
+ workersAiRestFetch
7
+ } from "./chunk-2VII5BK2.js";
8
+ import {
9
+ createGatewayFetch,
10
+ isDirectBindingConfig,
11
+ isDirectCredentialsConfig
12
+ } from "./chunk-F5YJMXZR.js";
13
+ import {
14
+ __publicField
15
+ } from "./chunk-V6TY7KAL.js";
16
+
17
+ // src/adapters/workers-ai-image.ts
18
+ import { BaseImageAdapter } from "@tanstack/ai/adapters";
19
+ var WorkersAiImageAdapter = class extends BaseImageAdapter {
20
+ constructor(config, model) {
21
+ super({}, model);
22
+ __publicField(this, "name", "workers-ai-image");
23
+ __publicField(this, "adapterConfig");
24
+ this.adapterConfig = config;
25
+ }
26
+ async generateImages(options) {
27
+ const { prompt, size, modelOptions } = options;
28
+ const extra = { ...modelOptions };
29
+ if (size) {
30
+ const [w, h] = size.split("x");
31
+ if (w) extra.width = Number(w);
32
+ if (h) extra.height = Number(h);
33
+ }
34
+ if (isDirectBindingConfig(this.adapterConfig)) {
35
+ return this.generateViaBinding(prompt, extra);
36
+ }
37
+ if (isDirectCredentialsConfig(this.adapterConfig)) {
38
+ return this.generateViaRest(prompt, extra);
39
+ }
40
+ return this.generateViaGateway(prompt, extra);
41
+ }
42
+ async generateViaBinding(prompt, options) {
43
+ const ai = this.adapterConfig.binding;
44
+ const result = await ai.run(this.model, { prompt, ...options });
45
+ const b64 = await binaryToBase64(result, "image");
46
+ return this.wrapResult(b64);
47
+ }
48
+ async generateViaRest(prompt, options) {
49
+ const config = this.adapterConfig;
50
+ const response = await workersAiRestFetch(
51
+ config,
52
+ this.model,
53
+ { prompt, ...options },
54
+ { label: "Workers AI image", signal: options.signal }
55
+ );
56
+ const buffer = await response.arrayBuffer();
57
+ return this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));
58
+ }
59
+ async generateViaGateway(prompt, options) {
60
+ const gatewayConfig = this.adapterConfig;
61
+ const gatewayFetch = createGatewayFetch("workers-ai", gatewayConfig);
62
+ const response = await gatewayFetch("https://api.cloudflare.com/v1/images/generations", {
63
+ method: "POST",
64
+ body: JSON.stringify({
65
+ model: this.model,
66
+ prompt,
67
+ ...options
68
+ })
69
+ });
70
+ if (!response.ok) {
71
+ const errorText = await response.text();
72
+ throw new Error(
73
+ `Workers AI image gateway request failed (${response.status}): ${errorText}`
74
+ );
75
+ }
76
+ const buffer = await response.arrayBuffer();
77
+ return this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));
78
+ }
79
+ /** Wrap a base64 image string into the standard ImageGenerationResult. */
80
+ wrapResult(b64) {
81
+ return {
82
+ id: this.generateId(),
83
+ model: this.model,
84
+ images: [{ b64Json: b64 }]
85
+ };
86
+ }
87
+ };
88
+ function createWorkersAiImage(model, config) {
89
+ return new WorkersAiImageAdapter(config, model);
90
+ }
91
+
92
+ export {
93
+ WorkersAiImageAdapter,
94
+ createWorkersAiImage
95
+ };
96
+ //# sourceMappingURL=chunk-RQT7M6MU.js.map