@cloudflare/tanstack-ai 0.1.6 → 0.1.8

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 (176) hide show
  1. package/README.md +35 -6
  2. package/dist/adapters/anthropic.cjs +32 -11
  3. package/dist/adapters/anthropic.cjs.map +1 -1
  4. package/dist/adapters/anthropic.d.cts +23 -5
  5. package/dist/adapters/anthropic.d.mts +23 -0
  6. package/dist/adapters/anthropic.mjs +30 -0
  7. package/dist/adapters/anthropic.mjs.map +1 -0
  8. package/dist/adapters/gemini.cjs +83 -20
  9. package/dist/adapters/gemini.cjs.map +1 -1
  10. package/dist/adapters/gemini.d.cts +52 -14
  11. package/dist/adapters/gemini.d.mts +52 -0
  12. package/dist/adapters/gemini.mjs +76 -0
  13. package/dist/adapters/gemini.mjs.map +1 -0
  14. package/dist/adapters/grok.cjs +44 -15
  15. package/dist/adapters/grok.cjs.map +1 -1
  16. package/dist/adapters/grok.d.cts +26 -9
  17. package/dist/adapters/grok.d.mts +26 -0
  18. package/dist/adapters/grok.mjs +40 -0
  19. package/dist/adapters/grok.mjs.map +1 -0
  20. package/dist/adapters/openai.cjs +76 -27
  21. package/dist/adapters/openai.cjs.map +1 -1
  22. package/dist/adapters/openai.d.cts +45 -17
  23. package/dist/adapters/openai.d.mts +45 -0
  24. package/dist/adapters/openai.mjs +66 -0
  25. package/dist/adapters/openai.mjs.map +1 -0
  26. package/dist/adapters/openrouter.cjs +83 -11
  27. package/dist/adapters/openrouter.cjs.map +1 -1
  28. package/dist/adapters/openrouter.d.cts +39 -7
  29. package/dist/adapters/openrouter.d.mts +39 -0
  30. package/dist/adapters/openrouter.mjs +81 -0
  31. package/dist/adapters/openrouter.mjs.map +1 -0
  32. package/dist/adapters/workers-ai-image.cjs +97 -11
  33. package/dist/adapters/workers-ai-image.cjs.map +1 -1
  34. package/dist/adapters/workers-ai-image.d.cts +41 -3
  35. package/dist/adapters/workers-ai-image.d.mts +41 -0
  36. package/dist/adapters/workers-ai-image.mjs +96 -0
  37. package/dist/adapters/workers-ai-image.mjs.map +1 -0
  38. package/dist/adapters/workers-ai-summarize.cjs +86 -10
  39. package/dist/adapters/workers-ai-summarize.cjs.map +1 -1
  40. package/dist/adapters/workers-ai-summarize.d.cts +42 -3
  41. package/dist/adapters/workers-ai-summarize.d.mts +42 -0
  42. package/dist/adapters/workers-ai-summarize.mjs +85 -0
  43. package/dist/adapters/workers-ai-summarize.mjs.map +1 -0
  44. package/dist/adapters/workers-ai-transcription.cjs +200 -11
  45. package/dist/adapters/workers-ai-transcription.cjs.map +1 -1
  46. package/dist/adapters/workers-ai-transcription.d.cts +74 -3
  47. package/dist/adapters/workers-ai-transcription.d.mts +74 -0
  48. package/dist/adapters/workers-ai-transcription.mjs +199 -0
  49. package/dist/adapters/workers-ai-transcription.mjs.map +1 -0
  50. package/dist/adapters/workers-ai-tts.cjs +112 -11
  51. package/dist/adapters/workers-ai-tts.cjs.map +1 -1
  52. package/dist/adapters/workers-ai-tts.d.cts +54 -3
  53. package/dist/adapters/workers-ai-tts.d.mts +54 -0
  54. package/dist/adapters/workers-ai-tts.mjs +111 -0
  55. package/dist/adapters/workers-ai-tts.mjs.map +1 -0
  56. package/dist/adapters/workers-ai.cjs +4 -11
  57. package/dist/adapters/workers-ai.d.cts +45 -3
  58. package/dist/adapters/workers-ai.d.mts +45 -0
  59. package/dist/adapters/workers-ai.mjs +481 -0
  60. package/dist/adapters/workers-ai.mjs.map +1 -0
  61. package/dist/binary-C9FAYwZj.cjs +70 -0
  62. package/dist/binary-C9FAYwZj.cjs.map +1 -0
  63. package/dist/binary-p4H_N_3M.mjs +59 -0
  64. package/dist/binary-p4H_N_3M.mjs.map +1 -0
  65. package/dist/create-fetcher-6p6heb85.d.mts +93 -0
  66. package/dist/create-fetcher-By-hTiT9.cjs +312 -0
  67. package/dist/create-fetcher-By-hTiT9.cjs.map +1 -0
  68. package/dist/create-fetcher-CeUOJgrh.mjs +283 -0
  69. package/dist/create-fetcher-CeUOJgrh.mjs.map +1 -0
  70. package/dist/create-fetcher-vAQ8WW-p.d.cts +93 -0
  71. package/dist/defineProperty-CbyrzcbA.mjs +39 -0
  72. package/dist/defineProperty-DQoAg20E.cjs +44 -0
  73. package/dist/index.cjs +50 -97
  74. package/dist/index.d.cts +12 -64
  75. package/dist/index.d.mts +12 -0
  76. package/dist/index.mjs +11 -0
  77. package/dist/workers-ai-BOZ5iyhY.cjs +521 -0
  78. package/dist/workers-ai-BOZ5iyhY.cjs.map +1 -0
  79. package/dist/workers-ai-rest-CkNCtBwv.cjs +85 -0
  80. package/dist/workers-ai-rest-CkNCtBwv.cjs.map +1 -0
  81. package/dist/workers-ai-rest-GKy2r7eG.mjs +74 -0
  82. package/dist/workers-ai-rest-GKy2r7eG.mjs.map +1 -0
  83. package/package.json +73 -40
  84. package/dist/_tsup-dts-rollup.d.cts +0 -837
  85. package/dist/_tsup-dts-rollup.d.ts +0 -837
  86. package/dist/adapters/anthropic.d.ts +0 -5
  87. package/dist/adapters/anthropic.js +0 -13
  88. package/dist/adapters/anthropic.js.map +0 -1
  89. package/dist/adapters/gemini.d.ts +0 -14
  90. package/dist/adapters/gemini.js +0 -22
  91. package/dist/adapters/gemini.js.map +0 -1
  92. package/dist/adapters/grok.d.ts +0 -9
  93. package/dist/adapters/grok.js +0 -17
  94. package/dist/adapters/grok.js.map +0 -1
  95. package/dist/adapters/openai.d.ts +0 -17
  96. package/dist/adapters/openai.js +0 -29
  97. package/dist/adapters/openai.js.map +0 -1
  98. package/dist/adapters/openrouter.d.ts +0 -7
  99. package/dist/adapters/openrouter.js +0 -13
  100. package/dist/adapters/openrouter.js.map +0 -1
  101. package/dist/adapters/workers-ai-image.d.ts +0 -3
  102. package/dist/adapters/workers-ai-image.js +0 -13
  103. package/dist/adapters/workers-ai-image.js.map +0 -1
  104. package/dist/adapters/workers-ai-summarize.d.ts +0 -3
  105. package/dist/adapters/workers-ai-summarize.js +0 -12
  106. package/dist/adapters/workers-ai-summarize.js.map +0 -1
  107. package/dist/adapters/workers-ai-transcription.d.ts +0 -3
  108. package/dist/adapters/workers-ai-transcription.js +0 -13
  109. package/dist/adapters/workers-ai-transcription.js.map +0 -1
  110. package/dist/adapters/workers-ai-tts.d.ts +0 -3
  111. package/dist/adapters/workers-ai-tts.js +0 -13
  112. package/dist/adapters/workers-ai-tts.js.map +0 -1
  113. package/dist/adapters/workers-ai.cjs.map +0 -1
  114. package/dist/adapters/workers-ai.d.ts +0 -3
  115. package/dist/adapters/workers-ai.js +0 -11
  116. package/dist/adapters/workers-ai.js.map +0 -1
  117. package/dist/chunk-2VII5BK2.js +0 -42
  118. package/dist/chunk-2VII5BK2.js.map +0 -1
  119. package/dist/chunk-3VQDXJLW.cjs +0 -46
  120. package/dist/chunk-3VQDXJLW.cjs.map +0 -1
  121. package/dist/chunk-3WK5ZD57.js +0 -57
  122. package/dist/chunk-3WK5ZD57.js.map +0 -1
  123. package/dist/chunk-4DE2IREA.cjs +0 -8
  124. package/dist/chunk-4DE2IREA.cjs.map +0 -1
  125. package/dist/chunk-53A2QIXV.js +0 -98
  126. package/dist/chunk-53A2QIXV.js.map +0 -1
  127. package/dist/chunk-6MEX2ZSM.cjs +0 -48
  128. package/dist/chunk-6MEX2ZSM.cjs.map +0 -1
  129. package/dist/chunk-7HSUHP63.cjs +0 -42
  130. package/dist/chunk-7HSUHP63.cjs.map +0 -1
  131. package/dist/chunk-AIAJANLC.js +0 -327
  132. package/dist/chunk-AIAJANLC.js.map +0 -1
  133. package/dist/chunk-CB3PSZY2.cjs +0 -218
  134. package/dist/chunk-CB3PSZY2.cjs.map +0 -1
  135. package/dist/chunk-FQE2UKUU.cjs +0 -57
  136. package/dist/chunk-FQE2UKUU.cjs.map +0 -1
  137. package/dist/chunk-GL3HHR4M.js +0 -523
  138. package/dist/chunk-GL3HHR4M.js.map +0 -1
  139. package/dist/chunk-J5DSSZTO.js +0 -71
  140. package/dist/chunk-J5DSSZTO.js.map +0 -1
  141. package/dist/chunk-JRFPCMSE.js +0 -57
  142. package/dist/chunk-JRFPCMSE.js.map +0 -1
  143. package/dist/chunk-KWYOUT2Y.js +0 -111
  144. package/dist/chunk-KWYOUT2Y.js.map +0 -1
  145. package/dist/chunk-MJ45L566.cjs +0 -57
  146. package/dist/chunk-MJ45L566.cjs.map +0 -1
  147. package/dist/chunk-N6U27RQW.cjs +0 -111
  148. package/dist/chunk-N6U27RQW.cjs.map +0 -1
  149. package/dist/chunk-NH3FF5JC.js +0 -86
  150. package/dist/chunk-NH3FF5JC.js.map +0 -1
  151. package/dist/chunk-OFUYHQZE.cjs +0 -31
  152. package/dist/chunk-OFUYHQZE.cjs.map +0 -1
  153. package/dist/chunk-OUEAAP25.cjs +0 -523
  154. package/dist/chunk-OUEAAP25.cjs.map +0 -1
  155. package/dist/chunk-Q33U5I5S.js +0 -31
  156. package/dist/chunk-Q33U5I5S.js.map +0 -1
  157. package/dist/chunk-V4LKHWJA.cjs +0 -71
  158. package/dist/chunk-V4LKHWJA.cjs.map +0 -1
  159. package/dist/chunk-V6TY7KAL.js +0 -8
  160. package/dist/chunk-V6TY7KAL.js.map +0 -1
  161. package/dist/chunk-WBTQKWFN.cjs +0 -327
  162. package/dist/chunk-WBTQKWFN.cjs.map +0 -1
  163. package/dist/chunk-XU7YEPML.js +0 -46
  164. package/dist/chunk-XU7YEPML.js.map +0 -1
  165. package/dist/chunk-YCGNMVQZ.js +0 -218
  166. package/dist/chunk-YCGNMVQZ.js.map +0 -1
  167. package/dist/chunk-Z5ZC5ZOQ.js +0 -48
  168. package/dist/chunk-Z5ZC5ZOQ.js.map +0 -1
  169. package/dist/chunk-ZFB72L57.cjs +0 -86
  170. package/dist/chunk-ZFB72L57.cjs.map +0 -1
  171. package/dist/chunk-ZGLSJM26.cjs +0 -98
  172. package/dist/chunk-ZGLSJM26.cjs.map +0 -1
  173. package/dist/index.cjs.map +0 -1
  174. package/dist/index.d.ts +0 -64
  175. package/dist/index.js +0 -97
  176. package/dist/index.js.map +0 -1
@@ -1,71 +0,0 @@
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.apiKey && 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
- const apiKey = _nullishCoalesce(config.apiKey, () => ( config.cfApiKey));
35
- if (!apiKey) {
36
- throw new Error(
37
- "If you want to use BYOK or unified billing, you need to pass the Cloudflare AI Gateway API key."
38
- );
39
- }
40
- return {
41
- apiKey,
42
- httpOptions: {
43
- baseUrl: `https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}/google-ai-studio`,
44
- headers: Object.keys(headers).length > 0 ? headers : void 0
45
- }
46
- };
47
- }
48
- function createGeminiChat(model, config) {
49
- return new (0, _aigemini.GeminiTextAdapter)(buildGeminiGatewayConfig(config), model);
50
- }
51
- function createGeminiImage(model, config) {
52
- return new (0, _aigemini.GeminiImageAdapter)(buildGeminiGatewayConfig(config), model);
53
- }
54
- function createGeminiSummarize(model, config) {
55
- return new (0, _aigemini.GeminiSummarizeAdapter)(buildGeminiGatewayConfig(config), model);
56
- }
57
- function createGeminiTts(model, config) {
58
- return new (0, _aigemini.GeminiTTSAdapter)(buildGeminiGatewayConfig(config), model);
59
- }
60
-
61
-
62
-
63
-
64
-
65
-
66
-
67
-
68
-
69
-
70
- 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;
71
- //# sourceMappingURL=chunk-V4LKHWJA.cjs.map
@@ -1 +0,0 @@
1
- {"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-V4LKHWJA.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;AAEzC,EAAA,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,MAAA,CAAO,QAAA,EAAU;AACrC,IAAA,OAAA,CAAQ,sBAAsB,EAAA,EAAI,CAAA,OAAA,EAAU,MAAA,CAAO,QAAQ,CAAA,CAAA;AAC5D,EAAA;AAEsB,EAAA;AACU,IAAA;AAChC,EAAA;AACyC,EAAA;AACY,IAAA;AACrD,EAAA;AAC+C,EAAA;AACT,IAAA;AACtC,EAAA;AACyC,EAAA;AACmB,IAAA;AAC5D,EAAA;AAEuC,EAAA;AAE1B,EAAA;AACF,IAAA;AACT,MAAA;AACD,IAAA;AACD,EAAA;AAEO,EAAA;AACN,IAAA;AACa,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;AD9EgE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-V4LKHWJA.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\n\tif (config.apiKey && config.cfApiKey) {\n\t\theaders[\"cf-aig-authorization\"] = `Bearer ${config.cfApiKey}`;\n\t}\n\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\tconst apiKey = config.apiKey ?? config.cfApiKey;\n\n\tif (!apiKey) {\n\t\tthrow new Error(\n\t\t\t\"If you want to use BYOK or unified billing, you need to pass the Cloudflare AI Gateway API key.\",\n\t\t);\n\t}\n\n\treturn {\n\t\tapiKey,\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"]}
@@ -1,8 +0,0 @@
1
- var __defProp = Object.defineProperty;
2
- var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
3
- var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
4
-
5
- export {
6
- __publicField
7
- };
8
- //# sourceMappingURL=chunk-V6TY7KAL.js.map
@@ -1 +0,0 @@
1
- {"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
@@ -1,327 +0,0 @@
1
- "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }// src/utils/create-fetcher.ts
2
- function isDirectBindingConfig(config) {
3
- return "binding" in config && typeof config.binding.gateway === "function";
4
- }
5
- function isDirectCredentialsConfig(config) {
6
- return "accountId" in config && "apiKey" in config && !("gatewayId" in config);
7
- }
8
- function isGatewayConfig(config) {
9
- if ("gatewayId" in config) return true;
10
- return "binding" in config && !isDirectBindingConfig(config);
11
- }
12
- function validateWorkersAiConfig(config) {
13
- if (!isDirectBindingConfig(config) && !isDirectCredentialsConfig(config) && !isGatewayConfig(config)) {
14
- throw new Error(
15
- "Invalid Workers AI configuration: you must provide either a binding (e.g. { binding: env.AI }), credentials ({ accountId, apiKey }), or a gateway configuration ({ binding: env.AI.gateway(id) } or { accountId, gatewayId })."
16
- );
17
- }
18
- }
19
- function createGatewayFetch(provider, config, headers = {}) {
20
- return (input, init) => {
21
- let query = {};
22
- const url = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
23
- const urlObj = new URL(url);
24
- const endpoint = urlObj.pathname.replace(/^\/v1\//, "").replace(/^\//, "") + urlObj.search;
25
- if (_optionalChain([init, 'optionalAccess', _ => _.body])) {
26
- try {
27
- query = JSON.parse(init.body);
28
- } catch (e2) {
29
- query = { _raw: init.body };
30
- }
31
- }
32
- const cacheHeaders = {};
33
- if ("skipCache" in config && config.skipCache) {
34
- cacheHeaders["cf-aig-skip-cache"] = "true";
35
- }
36
- if (typeof config.cacheTtl === "number") {
37
- cacheHeaders["cf-aig-cache-ttl"] = String(config.cacheTtl);
38
- }
39
- if (typeof config.customCacheKey === "string") {
40
- cacheHeaders["cf-aig-cache-key"] = config.customCacheKey;
41
- }
42
- if (typeof config.metadata === "object") {
43
- cacheHeaders["cf-aig-metadata"] = JSON.stringify(config.metadata);
44
- }
45
- const request = {
46
- provider,
47
- endpoint,
48
- headers: {
49
- ..._optionalChain([init, 'optionalAccess', _2 => _2.headers]),
50
- ...headers,
51
- ...cacheHeaders,
52
- "Content-Type": "application/json"
53
- },
54
- query
55
- };
56
- if (provider === "workers-ai") {
57
- if (!request.endpoint.startsWith("run/")) {
58
- request.endpoint = `run/${query.model}`;
59
- }
60
- delete query.model;
61
- delete query.instructions;
62
- }
63
- if (config.apiKey) {
64
- request.headers["authorization"] = `Bearer ${config.apiKey}`;
65
- }
66
- if ("binding" in config) {
67
- return config.binding.run(request, { signal: _nullishCoalesce(_optionalChain([init, 'optionalAccess', _3 => _3.signal]), () => ( void 0)) });
68
- }
69
- return fetch(
70
- `https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}`,
71
- {
72
- ...init,
73
- headers: {
74
- "Content-Type": "application/json",
75
- ...headers,
76
- ...cacheHeaders,
77
- ...config.cfApiKey ? { "cf-aig-authorization": `Bearer ${config.cfApiKey}` } : {}
78
- },
79
- body: JSON.stringify(request)
80
- }
81
- );
82
- };
83
- }
84
- function normalizeMessagesForBinding(messages) {
85
- return messages.map((msg) => {
86
- const normalized = { ...msg };
87
- if (normalized.content === null || normalized.content === void 0) {
88
- normalized.content = "";
89
- }
90
- return normalized;
91
- });
92
- }
93
- function createWorkersAiBindingFetch(binding, options) {
94
- return async (_input, init) => {
95
- if (!_optionalChain([init, 'optionalAccess', _4 => _4.body])) {
96
- return new Response("No body", { status: 400 });
97
- }
98
- let body;
99
- try {
100
- body = JSON.parse(init.body);
101
- } catch (e3) {
102
- return new Response("Invalid JSON body", { status: 400 });
103
- }
104
- const model = body.model;
105
- const stream = body.stream;
106
- const inputs = {};
107
- if (body.messages) {
108
- inputs.messages = normalizeMessagesForBinding(
109
- body.messages
110
- );
111
- }
112
- if (body.tools) inputs.tools = body.tools;
113
- if (typeof body.temperature === "number") inputs.temperature = body.temperature;
114
- if (typeof body.max_tokens === "number") inputs.max_tokens = body.max_tokens;
115
- if (body.response_format) inputs.response_format = body.response_format;
116
- if (stream) inputs.stream = true;
117
- const runOptions = {};
118
- if (_optionalChain([options, 'optionalAccess', _5 => _5.extraHeaders])) runOptions.extraHeaders = options.extraHeaders;
119
- if (_optionalChain([init, 'optionalAccess', _6 => _6.signal])) runOptions.signal = init.signal;
120
- const result = await binding.run(
121
- model,
122
- inputs,
123
- Object.keys(runOptions).length > 0 ? runOptions : void 0
124
- );
125
- if (stream && result instanceof ReadableStream) {
126
- const transformed = transformWorkersAiStream(
127
- result,
128
- model
129
- );
130
- return new Response(transformed, {
131
- headers: {
132
- "Content-Type": "text/event-stream",
133
- "Cache-Control": "no-cache"
134
- }
135
- });
136
- }
137
- const responseObj = typeof result === "object" && result !== null ? result : { response: String(result) };
138
- const responseText = typeof responseObj.response === "string" ? responseObj.response : typeof responseObj.response === "object" && responseObj.response !== null ? JSON.stringify(responseObj.response) : "";
139
- const message = {
140
- role: "assistant",
141
- content: responseText
142
- };
143
- let finishReason = "stop";
144
- if (Array.isArray(responseObj.tool_calls) && responseObj.tool_calls.length > 0) {
145
- finishReason = "tool_calls";
146
- message.tool_calls = responseObj.tool_calls.map(
147
- (tc) => ({
148
- id: tc.id || crypto.randomUUID(),
149
- type: "function",
150
- function: {
151
- name: _optionalChain([tc, 'access', _7 => _7.function, 'optionalAccess', _8 => _8.name]) || tc.name || "",
152
- arguments: typeof (_nullishCoalesce(_optionalChain([tc, 'access', _9 => _9.function, 'optionalAccess', _10 => _10.arguments]), () => ( tc.arguments))) === "string" ? _nullishCoalesce(_optionalChain([tc, 'access', _11 => _11.function, 'optionalAccess', _12 => _12.arguments]), () => ( tc.arguments)) : JSON.stringify(_nullishCoalesce(_nullishCoalesce(_optionalChain([tc, 'access', _13 => _13.function, 'optionalAccess', _14 => _14.arguments]), () => ( tc.arguments)), () => ( {})))
153
- }
154
- })
155
- );
156
- }
157
- const openAiResponse = {
158
- id: `workers-ai-${crypto.randomUUID()}`,
159
- object: "chat.completion",
160
- created: Math.floor(Date.now() / 1e3),
161
- model,
162
- choices: [{ index: 0, message, finish_reason: finishReason }]
163
- };
164
- return new Response(JSON.stringify(openAiResponse), {
165
- headers: { "Content-Type": "application/json" }
166
- });
167
- };
168
- }
169
- function transformWorkersAiStream(source, model) {
170
- const decoder = new TextDecoder();
171
- const encoder = new TextEncoder();
172
- const streamId = `workers-ai-${crypto.randomUUID()}`;
173
- const created = Math.floor(Date.now() / 1e3);
174
- let buffer = "";
175
- let hasToolCalls = false;
176
- let isOpenAiFormat = false;
177
- const toolCallState = /* @__PURE__ */ new Map();
178
- return source.pipeThrough(
179
- new TransformStream({
180
- transform(chunk, controller) {
181
- buffer += decoder.decode(chunk, { stream: true });
182
- const lines = buffer.split("\n");
183
- buffer = lines.pop() || "";
184
- for (const line of lines) {
185
- const trimmed = line.trim();
186
- if (!trimmed || !trimmed.startsWith("data: ")) continue;
187
- const data = trimmed.slice(6);
188
- if (data === "[DONE]") continue;
189
- try {
190
- const parsed = JSON.parse(data);
191
- if (parsed.choices !== void 0) {
192
- isOpenAiFormat = true;
193
- const choice = _optionalChain([parsed, 'access', _15 => _15.choices, 'optionalAccess', _16 => _16[0]]);
194
- if (_optionalChain([choice, 'optionalAccess', _17 => _17.delta, 'optionalAccess', _18 => _18.tool_calls])) {
195
- hasToolCalls = true;
196
- for (const tc of choice.delta.tool_calls) {
197
- const tcIndex = _nullishCoalesce(tc.index, () => ( 0));
198
- if (!toolCallState.has(tcIndex)) {
199
- const id = tc.id || `call${streamId}${tcIndex}`;
200
- toolCallState.set(tcIndex, {
201
- id,
202
- name: _optionalChain([tc, 'access', _19 => _19.function, 'optionalAccess', _20 => _20.name]) || ""
203
- });
204
- tc.id = id;
205
- } else {
206
- delete tc.id;
207
- }
208
- }
209
- }
210
- if (_optionalChain([choice, 'optionalAccess', _21 => _21.finish_reason]) === "tool_calls") {
211
- hasToolCalls = true;
212
- }
213
- controller.enqueue(
214
- encoder.encode(`data: ${JSON.stringify(parsed)}
215
-
216
- `)
217
- );
218
- continue;
219
- }
220
- if (parsed.response != null && parsed.response !== "") {
221
- const openAiChunk = {
222
- id: streamId,
223
- object: "chat.completion.chunk",
224
- created,
225
- model,
226
- choices: [
227
- {
228
- index: 0,
229
- delta: { content: parsed.response },
230
- finish_reason: null
231
- }
232
- ]
233
- };
234
- controller.enqueue(
235
- encoder.encode(`data: ${JSON.stringify(openAiChunk)}
236
-
237
- `)
238
- );
239
- }
240
- if (Array.isArray(parsed.tool_calls) && parsed.tool_calls.length > 0) {
241
- for (const tc of parsed.tool_calls) {
242
- const tcIndex = _nullishCoalesce(tc.index, () => ( 0));
243
- const tcName = _nullishCoalesce(_nullishCoalesce(_optionalChain([tc, 'access', _22 => _22.function, 'optionalAccess', _23 => _23.name]), () => ( tc.name)), () => ( null));
244
- const tcArgs = _nullishCoalesce(_nullishCoalesce(_optionalChain([tc, 'access', _24 => _24.function, 'optionalAccess', _25 => _25.arguments]), () => ( tc.arguments)), () => ( null));
245
- const tcId = _nullishCoalesce(tc.id, () => ( null));
246
- if (!tcId && !tcName && (!tcArgs || tcArgs === "")) continue;
247
- hasToolCalls = true;
248
- const toolCallDelta = {
249
- index: tcIndex
250
- };
251
- if (!toolCallState.has(tcIndex)) {
252
- const id = tcId || `call${streamId}${tcIndex}`;
253
- toolCallState.set(tcIndex, { id, name: tcName || "" });
254
- toolCallDelta.id = id;
255
- toolCallDelta.type = "function";
256
- toolCallDelta.function = {
257
- name: tcName || "",
258
- // Include arguments if they arrive in the same chunk
259
- arguments: tcArgs != null ? typeof tcArgs === "string" ? tcArgs : JSON.stringify(tcArgs) : ""
260
- };
261
- } else {
262
- if (tcArgs != null && tcArgs !== "") {
263
- toolCallDelta.function = {
264
- arguments: typeof tcArgs === "string" ? tcArgs : JSON.stringify(tcArgs)
265
- };
266
- } else {
267
- continue;
268
- }
269
- }
270
- const toolChunk = {
271
- id: streamId,
272
- object: "chat.completion.chunk",
273
- created,
274
- model,
275
- choices: [
276
- {
277
- index: 0,
278
- delta: { tool_calls: [toolCallDelta] },
279
- finish_reason: null
280
- }
281
- ]
282
- };
283
- controller.enqueue(
284
- encoder.encode(`data: ${JSON.stringify(toolChunk)}
285
-
286
- `)
287
- );
288
- }
289
- }
290
- } catch (e) {
291
- console.warn("[tanstack-ai] failed to parse SSE event:", data, e);
292
- }
293
- }
294
- },
295
- flush(controller) {
296
- if (!isOpenAiFormat) {
297
- const finalChunk = {
298
- id: streamId,
299
- object: "chat.completion.chunk",
300
- created,
301
- model,
302
- choices: [
303
- {
304
- index: 0,
305
- delta: {},
306
- finish_reason: hasToolCalls ? "tool_calls" : "stop"
307
- }
308
- ]
309
- };
310
- controller.enqueue(encoder.encode(`data: ${JSON.stringify(finalChunk)}
311
-
312
- `));
313
- }
314
- controller.enqueue(encoder.encode("data: [DONE]\n\n"));
315
- }
316
- })
317
- );
318
- }
319
-
320
-
321
-
322
-
323
-
324
-
325
-
326
- exports.isDirectBindingConfig = isDirectBindingConfig; exports.isDirectCredentialsConfig = isDirectCredentialsConfig; exports.validateWorkersAiConfig = validateWorkersAiConfig; exports.createGatewayFetch = createGatewayFetch; exports.createWorkersAiBindingFetch = createWorkersAiBindingFetch;
327
- //# sourceMappingURL=chunk-WBTQKWFN.cjs.map
@@ -1 +0,0 @@
1
- {"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-WBTQKWFN.cjs","../src/utils/create-fetcher.ts"],"names":[],"mappings":"AAAA;AC4HO,SAAS,qBAAA,CACf,MAAA,EACyC;AAGzC,EAAA,OACC,UAAA,GAAa,OAAA,GACb,OAAQ,MAAA,CAAO,OAAA,CAA+C,QAAA,IAAY,UAAA;AAE5E;AAGO,SAAS,yBAAA,CACf,MAAA,EAC6C;AAC7C,EAAA,OAAO,YAAA,GAAe,OAAA,GAAU,SAAA,GAAY,OAAA,GAAU,CAAA,CAAE,YAAA,GAAe,MAAA,CAAA;AACxE;AAGO,SAAS,eAAA,CAAgB,MAAA,EAAkE;AACjG,EAAA,GAAA,CAAI,YAAA,GAAe,MAAA,EAAQ,OAAO,IAAA;AAElC,EAAA,OAAO,UAAA,GAAa,OAAA,GAAU,CAAC,qBAAA,CAAsB,MAAM,CAAA;AAC5D;AAWO,SAAS,uBAAA,CAAwB,MAAA,EAAsC;AAC7E,EAAA,GAAA,CACC,CAAC,qBAAA,CAAsB,MAAM,EAAA,GAC7B,CAAC,yBAAA,CAA0B,MAAM,EAAA,GACjC,CAAC,eAAA,CAAgB,MAAM,CAAA,EACtB;AACD,IAAA,MAAM,IAAI,KAAA;AAAA,MACT;AAAA,IAGD,CAAA;AAAA,EACD;AACD;AAMO,SAAS,kBAAA,CACf,QAAA,EACA,MAAA,EACA,QAAA,EAAkC,CAAC,CAAA,EACpB;AACf,EAAA,OAAO,CAAC,KAAA,EAAO,IAAA,EAAA,GAAS;AACvB,IAAA,IAAI,MAAA,EAAiC,CAAC,CAAA;AAEtC,IAAA,MAAM,IAAA,EACL,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,MAAA,WAAiB,IAAA,EAAM,KAAA,CAAM,KAAA,EAAO,KAAA,CAAM,GAAA;AAC/E,IAAA,MAAM,OAAA,EAAS,IAAI,GAAA,CAAI,GAAG,CAAA;AAG1B,IAAA,MAAM,SAAA,EAAW,MAAA,CAAO,QAAA,CAAS,OAAA,CAAQ,SAAA,EAAW,EAAE,CAAA,CAAE,OAAA,CAAQ,KAAA,EAAO,EAAE,EAAA,EAAI,MAAA,CAAO,MAAA;AAEpF,IAAA,GAAA,iBAAI,IAAA,2BAAM,MAAA,EAAM;AACf,MAAA,IAAI;AACH,QAAA,MAAA,EAAQ,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,IAAc,CAAA;AAAA,MACvC,EAAA,WAAQ;AACP,QAAA,MAAA,EAAQ,EAAE,IAAA,EAAM,IAAA,CAAK,KAAK,CAAA;AAAA,MAC3B;AAAA,IACD;AAEA,IAAA,MAAM,aAAA,EAAuC,CAAC,CAAA;AAE9C,IAAA,GAAA,CAAI,YAAA,GAAe,OAAA,GAAU,MAAA,CAAO,SAAA,EAAW;AAC9C,MAAA,YAAA,CAAa,mBAAmB,EAAA,EAAI,MAAA;AAAA,IACrC;AAEA,IAAA,GAAA,CAAI,OAAO,MAAA,CAAO,SAAA,IAAa,QAAA,EAAU;AACxC,MAAA,YAAA,CAAa,kBAAkB,EAAA,EAAI,MAAA,CAAO,MAAA,CAAO,QAAQ,CAAA;AAAA,IAC1D;AAEA,IAAA,GAAA,CAAI,OAAO,MAAA,CAAO,eAAA,IAAmB,QAAA,EAAU;AAC9C,MAAA,YAAA,CAAa,kBAAkB,EAAA,EAAI,MAAA,CAAO,cAAA;AAAA,IAC3C;AAEA,IAAA,GAAA,CAAI,OAAO,MAAA,CAAO,SAAA,IAAa,QAAA,EAAU;AACxC,MAAA,YAAA,CAAa,iBAAiB,EAAA,EAAI,IAAA,CAAK,SAAA,CAAU,MAAA,CAAO,QAAQ,CAAA;AAAA,IACjE;AAEA,IAAA,MAAM,QAAA,EAKF;AAAA,MACH,QAAA;AAAA,MACA,QAAA;AAAA,MACA,OAAA,EAAS;AAAA,QACR,mBAAI,IAAA,6BAAM,SAAA;AAAA,QACV,GAAG,OAAA;AAAA,QACH,GAAG,YAAA;AAAA,QACH,cAAA,EAAgB;AAAA,MACjB,CAAA;AAAA,MACA;AAAA,IACD,CAAA;AAEA,IAAA,GAAA,CAAI,SAAA,IAAa,YAAA,EAAc;AAC9B,MAAA,GAAA,CAAI,CAAC,OAAA,CAAQ,QAAA,CAAS,UAAA,CAAW,MAAM,CAAA,EAAG;AACzC,QAAA,OAAA,CAAQ,SAAA,EAAW,CAAA,IAAA,EAAO,KAAA,CAAM,KAAK,CAAA,CAAA;AACtC,MAAA;AACa,MAAA;AACA,MAAA;AACd,IAAA;AAEmB,IAAA;AACiB,MAAA;AACpC,IAAA;AAEyB,IAAA;AAKP,MAAA;AAClB,IAAA;AAEO,IAAA;AACkC,MAAA;AACxC,MAAA;AACI,QAAA;AACM,QAAA;AACQ,UAAA;AACb,UAAA;AACA,UAAA;AAEE,UAAA;AAEN,QAAA;AAC4B,QAAA;AAC7B,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAgBC;AAE6B,EAAA;AACA,IAAA;AAGO,IAAA;AACb,MAAA;AACtB,IAAA;AAEO,IAAA;AACP,EAAA;AACF;AAaC;AAE+B,EAAA;AACb,IAAA;AACiB,MAAA;AAClC,IAAA;AAEI,IAAA;AACA,IAAA;AACkC,MAAA;AAC9B,IAAA;AACa,MAAA;AACrB,IAAA;AAEmB,IAAA;AACC,IAAA;AAGqB,IAAA;AACtB,IAAA;AACA,MAAA;AACZ,QAAA;AACN,MAAA;AACD,IAAA;AACoC,IAAA;AACM,IAAA;AACD,IAAA;AACR,IAAA;AACL,IAAA;AAEiB,IAAA;AACP,IAAA;AACA,IAAA;AAET,IAAA;AAC5B,MAAA;AACA,MAAA;AACqC,MAAA;AACtC,IAAA;AAEgC,IAAA;AAGX,MAAA;AACnB,QAAA;AACA,QAAA;AACD,MAAA;AACiC,MAAA;AACvB,QAAA;AACQ,UAAA;AACC,UAAA;AAClB,QAAA;AACA,MAAA;AACF,IAAA;AAUmB,IAAA;AAKC,IAAA;AAMqB,IAAA;AAClC,MAAA;AACG,MAAA;AACV,IAAA;AACmB,IAAA;AAGqB,IAAA;AACxB,MAAA;AACkB,MAAA;AAMzB,QAAA;AACyB,UAAA;AACzB,UAAA;AACI,UAAA;AACqB,YAAA;AAER,YAAA;AAGvB,UAAA;AACD,QAAA;AACD,MAAA;AACD,IAAA;AAEuB,IAAA;AACe,MAAA;AAC7B,MAAA;AAC6B,MAAA;AACrC,MAAA;AAC+B,MAAA;AAChC,IAAA;AAEmC,IAAA;AACP,MAAA;AAC3B,IAAA;AACF,EAAA;AACD;AAkBC;AAEgC,EAAA;AACA,EAAA;AAGM,EAAA;AACM,EAAA;AAC/B,EAAA;AACM,EAAA;AAIE,EAAA;AAGK,EAAA;AAEZ,EAAA;AAC+B,IAAA;AACd,MAAA;AACM,QAAA;AACH,QAAA;AACP,QAAA;AAEE,QAAA;AACC,UAAA;AACU,UAAA;AACR,UAAA;AAGL,UAAA;AAEnB,UAAA;AAC2B,YAAA;AAKI,YAAA;AAGhB,cAAA;AACe,cAAA;AACD,cAAA;AACf,gBAAA;AACe,gBAAA;AACD,kBAAA;AACL,kBAAA;AAEF,oBAAA;AACF,oBAAA;AACjB,sBAAA;AACmB,sBAAA;AACnB,oBAAA;AACO,oBAAA;AACF,kBAAA;AAGI,oBAAA;AACX,kBAAA;AACD,gBAAA;AACD,cAAA;AAC8B,cAAA;AACd,gBAAA;AAChB,cAAA;AACW,cAAA;AACmB,gBAAA;AAAiB;AAAM;AACrD,cAAA;AACA,cAAA;AACD,YAAA;AAK+B,YAAA;AACV,cAAA;AACf,gBAAA;AACI,gBAAA;AACR,gBAAA;AACA,gBAAA;AACS,gBAAA;AACR,kBAAA;AACQ,oBAAA;AACkB,oBAAA;AACV,oBAAA;AAChB,kBAAA;AACD,gBAAA;AACD,cAAA;AACW,cAAA;AACmB,gBAAA;AAAsB;AAAM;AAC1D,cAAA;AACD,YAAA;AAOyB,YAAA;AACA,cAAA;AACK,gBAAA;AAGA,gBAAA;AACA,gBAAA;AACN,gBAAA;AAGI,gBAAA;AAEX,gBAAA;AAGgC,gBAAA;AACvC,kBAAA;AACR,gBAAA;AAE8B,gBAAA;AAEH,kBAAA;AACC,kBAAA;AACR,kBAAA;AACE,kBAAA;AACI,kBAAA;AACR,oBAAA;AAAA;AAGL,oBAAA;AAKZ,kBAAA;AACM,gBAAA;AAEgB,kBAAA;AACI,oBAAA;AAEhB,sBAAA;AAGT,oBAAA;AACM,kBAAA;AACN,oBAAA;AACD,kBAAA;AACD,gBAAA;AAEkB,gBAAA;AACb,kBAAA;AACI,kBAAA;AACR,kBAAA;AACA,kBAAA;AACS,kBAAA;AACR,oBAAA;AACQ,sBAAA;AACe,sBAAA;AACP,sBAAA;AAChB,oBAAA;AACD,kBAAA;AACD,gBAAA;AACW,gBAAA;AACc,kBAAA;AAAyB;AAAM;AACxD,gBAAA;AACD,cAAA;AACD,YAAA;AACW,UAAA;AAEE,YAAA;AACd,UAAA;AACD,QAAA;AACD,MAAA;AACkB,MAAA;AACI,QAAA;AAED,UAAA;AACd,YAAA;AACI,YAAA;AACR,YAAA;AACA,YAAA;AACS,YAAA;AACR,cAAA;AACQ,gBAAA;AACC,gBAAA;AACsB,gBAAA;AAC/B,cAAA;AACD,YAAA;AACD,UAAA;AACkC,UAAA;AAAmC;AAAO;AAC7E,QAAA;AAGkC,QAAA;AACnC,MAAA;AACA,IAAA;AACF,EAAA;AACD;ADlU8C;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-WBTQKWFN.cjs","sourcesContent":[null,"// ---------------------------------------------------------------------------\n// AI Gateway types (for third-party providers + Workers AI through gateway)\n// ---------------------------------------------------------------------------\n\nexport interface CloudflareAiGateway {\n\trun(request: unknown): Promise<Response>;\n}\n\nexport interface AiGatewayBindingConfig {\n\t/**\n\t * The AI Gateway binding\n\t * @example\n\t * env.AI.gateway('my-gateway-id')\n\t */\n\tbinding: CloudflareAiGateway;\n\t/**\n\t * The Provider API Key if you want to manually pass it, ignore if using Unified Billing or BYOK.\n\t */\n\tapiKey?: string;\n}\n\nexport type AiGatewayCredentialsConfig = {\n\t/**\n\t * The Cloudflare account ID\n\t */\n\taccountId: string;\n\t/**\n\t * The AI Gateway ID\n\t */\n\tgatewayId: string;\n} & (\n\t| {\n\t\t\t/** Cloudflare API Key for AI Gateway */\n\t\t\tcfApiKey: string;\n\t\t\tapiKey?: string;\n\t }\n\t| {\n\t\t\t/** Provider API Key */\n\t\t\tapiKey: string;\n\t\t\t/** Cloudflare API Key for AI Gateway */\n\t\t\tcfApiKey?: string;\n\t }\n);\n\nexport interface AiGatewayConfig {\n\tskipCache?: boolean;\n\tcacheTtl?: number;\n\tcustomCacheKey?: string;\n\tmetadata?: Record<string, unknown>;\n}\n\nexport type AiGatewayAdapterConfig = (AiGatewayBindingConfig | AiGatewayCredentialsConfig) &\n\tAiGatewayConfig;\n\n// ---------------------------------------------------------------------------\n// Plain Workers AI types (direct binding or REST, no gateway)\n// ---------------------------------------------------------------------------\n\n/**\n * The Workers AI binding interface (env.AI).\n * Accepts a model name and inputs, returns results directly.\n * Includes `gateway()` which is present on `env.AI` but not on `env.AI.gateway(id)`,\n * enabling structural discrimination from `CloudflareAiGateway`.\n */\nexport interface WorkersAiBinding {\n\trun(\n\t\tmodel: string,\n\t\tinputs: Record<string, unknown>,\n\t\toptions?: Record<string, unknown>,\n\t): Promise<unknown>;\n\tgateway(gatewayId: string): CloudflareAiGateway;\n}\n\nexport interface WorkersAiDirectBindingConfig {\n\t/**\n\t * The Workers AI binding (env.AI).\n\t * @example\n\t * { binding: env.AI }\n\t */\n\tbinding: WorkersAiBinding;\n}\n\nexport interface WorkersAiDirectCredentialsConfig {\n\t/**\n\t * The Cloudflare account ID\n\t */\n\taccountId: string;\n\t/**\n\t * The Cloudflare API key for Workers AI\n\t */\n\tapiKey: string;\n}\n\n/**\n * Config for Workers AI adapters. Supports four modes:\n * - Plain binding: `{ binding: env.AI }`\n * - Plain REST: `{ accountId, apiKey }`\n * - AI Gateway binding: `{ binding: env.AI.gateway(id) }`\n * - AI Gateway REST: `{ accountId, gatewayId, ... }`\n *\n * The third union member intersects `AiGatewayAdapterConfig` with `{ apiKey?: string }`.\n * For the gateway binding variant, `AiGatewayBindingConfig` already includes `apiKey?`,\n * so the intersection is redundant there. For the gateway credentials variant, this\n * `apiKey` represents the Workers AI token (used in the `Authorization` header to the\n * upstream provider), distinct from `cfApiKey` (used in the `cf-aig-authorization`\n * header for authenticated gateways).\n */\nexport type WorkersAiAdapterConfig = (\n\t| WorkersAiDirectBindingConfig\n\t| WorkersAiDirectCredentialsConfig\n\t| (AiGatewayAdapterConfig & { apiKey?: string })\n) & {\n\t/**\n\t * Session affinity key for prefix-cache optimization.\n\t * Routes requests with the same key to the same backend replica.\n\t */\n\tsessionAffinity?: string;\n};\n\n// ---------------------------------------------------------------------------\n// Config detection helpers\n// ---------------------------------------------------------------------------\n\n/** Returns true if this is a plain Workers AI binding config (`{ binding: env.AI }`) */\nexport function isDirectBindingConfig(\n\tconfig: WorkersAiAdapterConfig,\n): config is WorkersAiDirectBindingConfig {\n\t// env.AI has a .gateway() method; env.AI.gateway(id) does not.\n\t// This distinguishes direct bindings from AI Gateway bindings.\n\treturn (\n\t\t\"binding\" in config &&\n\t\ttypeof (config.binding as unknown as Record<string, unknown>).gateway === \"function\"\n\t);\n}\n\n/** Returns true if this is a plain Workers AI REST config (accountId + apiKey, no gatewayId) */\nexport function isDirectCredentialsConfig(\n\tconfig: WorkersAiAdapterConfig,\n): config is WorkersAiDirectCredentialsConfig {\n\treturn \"accountId\" in config && \"apiKey\" in config && !(\"gatewayId\" in config);\n}\n\n/** Returns true if this is an AI Gateway config (has gateway binding or `gatewayId`) */\nexport function isGatewayConfig(config: WorkersAiAdapterConfig): config is AiGatewayAdapterConfig {\n\tif (\"gatewayId\" in config) return true;\n\t// Has `binding` but NOT a direct Workers AI binding (no .gateway method)\n\treturn \"binding\" in config && !isDirectBindingConfig(config);\n}\n\n// ---------------------------------------------------------------------------\n// Config validation\n// ---------------------------------------------------------------------------\n\n/**\n * Validates that a WorkersAiAdapterConfig contains a valid configuration.\n * Throws an error if neither a binding, credentials (accountId + apiKey),\n * nor a gateway configuration is provided.\n */\nexport function validateWorkersAiConfig(config: WorkersAiAdapterConfig): void {\n\tif (\n\t\t!isDirectBindingConfig(config) &&\n\t\t!isDirectCredentialsConfig(config) &&\n\t\t!isGatewayConfig(config)\n\t) {\n\t\tthrow new Error(\n\t\t\t\"Invalid Workers AI configuration: you must provide either a binding (e.g. { binding: env.AI }), \" +\n\t\t\t\t\"credentials ({ accountId, apiKey }), or a gateway configuration ({ binding: env.AI.gateway(id) } \" +\n\t\t\t\t\"or { accountId, gatewayId }).\",\n\t\t);\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// createGatewayFetch -- for routing through AI Gateway\n// ---------------------------------------------------------------------------\n\nexport function createGatewayFetch(\n\tprovider: string,\n\tconfig: AiGatewayAdapterConfig,\n\theaders: Record<string, string> = {},\n): typeof fetch {\n\treturn (input, init) => {\n\t\tlet query: Record<string, unknown> = {};\n\n\t\tconst url =\n\t\t\ttypeof input === \"string\" ? input : input instanceof URL ? input.href : input.url;\n\t\tconst urlObj = new URL(url);\n\n\t\t// Extract endpoint path (remove /v1/ prefix if present)\n\t\tconst endpoint = urlObj.pathname.replace(/^\\/v1\\//, \"\").replace(/^\\//, \"\") + urlObj.search;\n\n\t\tif (init?.body) {\n\t\t\ttry {\n\t\t\t\tquery = JSON.parse(init.body as string);\n\t\t\t} catch {\n\t\t\t\tquery = { _raw: init.body };\n\t\t\t}\n\t\t}\n\n\t\tconst cacheHeaders: Record<string, string> = {};\n\n\t\tif (\"skipCache\" in config && config.skipCache) {\n\t\t\tcacheHeaders[\"cf-aig-skip-cache\"] = \"true\";\n\t\t}\n\n\t\tif (typeof config.cacheTtl === \"number\") {\n\t\t\tcacheHeaders[\"cf-aig-cache-ttl\"] = String(config.cacheTtl);\n\t\t}\n\n\t\tif (typeof config.customCacheKey === \"string\") {\n\t\t\tcacheHeaders[\"cf-aig-cache-key\"] = config.customCacheKey;\n\t\t}\n\n\t\tif (typeof config.metadata === \"object\") {\n\t\t\tcacheHeaders[\"cf-aig-metadata\"] = JSON.stringify(config.metadata);\n\t\t}\n\n\t\tconst request: {\n\t\t\tprovider: string;\n\t\t\tendpoint: string;\n\t\t\theaders: Record<string, string>;\n\t\t\tquery: Record<string, unknown>;\n\t\t} = {\n\t\t\tprovider,\n\t\t\tendpoint,\n\t\t\theaders: {\n\t\t\t\t...(init?.headers as Record<string, string> | undefined),\n\t\t\t\t...headers,\n\t\t\t\t...cacheHeaders,\n\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t},\n\t\t\tquery,\n\t\t};\n\n\t\tif (provider === \"workers-ai\") {\n\t\t\tif (!request.endpoint.startsWith(\"run/\")) {\n\t\t\t\trequest.endpoint = `run/${query.model}`;\n\t\t\t}\n\t\t\tdelete query.model;\n\t\t\tdelete query.instructions;\n\t\t}\n\n\t\tif (config.apiKey) {\n\t\t\trequest.headers[\"authorization\"] = `Bearer ${config.apiKey}`;\n\t\t}\n\n\t\tif (\"binding\" in config) {\n\t\t\treturn (\n\t\t\t\tconfig.binding as {\n\t\t\t\t\trun(req: unknown, opts?: { signal?: AbortSignal }): Promise<Response>;\n\t\t\t\t}\n\t\t\t).run(request, { signal: init?.signal ?? undefined });\n\t\t}\n\n\t\treturn fetch(\n\t\t\t`https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}`,\n\t\t\t{\n\t\t\t\t...init,\n\t\t\t\theaders: {\n\t\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\t\t...headers,\n\t\t\t\t\t...cacheHeaders,\n\t\t\t\t\t...(config.cfApiKey\n\t\t\t\t\t\t? { \"cf-aig-authorization\": `Bearer ${config.cfApiKey}` }\n\t\t\t\t\t\t: {}),\n\t\t\t\t},\n\t\t\t\tbody: JSON.stringify(request),\n\t\t\t},\n\t\t);\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// createWorkersAiBindingFetch -- shim that makes env.AI look like an OpenAI endpoint\n// ---------------------------------------------------------------------------\n\n/**\n * Normalize messages before passing to Workers AI binding.\n *\n * The binding has strict schema validation that may differ from the OpenAI API:\n * - `content` must not be null\n *\n * Content arrays (with image_url parts) are passed through as-is since the\n * Workers AI binding accepts them at runtime for vision-capable models.\n */\nfunction normalizeMessagesForBinding(\n\tmessages: Record<string, unknown>[],\n): Record<string, unknown>[] {\n\treturn messages.map((msg) => {\n\t\tconst normalized = { ...msg };\n\n\t\t// content: null → content: \"\"\n\t\tif (normalized.content === null || normalized.content === undefined) {\n\t\t\tnormalized.content = \"\";\n\t\t}\n\n\t\treturn normalized;\n\t});\n}\n\n/**\n * Creates a fetch function that intercepts OpenAI SDK requests and translates them\n * to Workers AI binding calls (env.AI.run). This allows the WorkersAiTextAdapter\n * to use the OpenAI SDK against a plain Workers AI binding.\n *\n * NOTE: The `input` URL parameter is intentionally ignored. The model name and all\n * request parameters are extracted from the JSON body, matching Workers AI's\n * `binding.run(model, inputs)` calling convention.\n */\nexport function createWorkersAiBindingFetch(\n\tbinding: WorkersAiBinding,\n\toptions?: { extraHeaders?: Record<string, string> },\n): typeof fetch {\n\treturn async (_input, init) => {\n\t\tif (!init?.body) {\n\t\t\treturn new Response(\"No body\", { status: 400 });\n\t\t}\n\n\t\tlet body: Record<string, unknown>;\n\t\ttry {\n\t\t\tbody = JSON.parse(init.body as string);\n\t\t} catch {\n\t\t\treturn new Response(\"Invalid JSON body\", { status: 400 });\n\t\t}\n\n\t\tconst model = body.model as string;\n\t\tconst stream = body.stream as boolean | undefined;\n\n\t\t// Build Workers AI inputs from OpenAI format\n\t\tconst inputs: Record<string, unknown> = {};\n\t\tif (body.messages) {\n\t\t\tinputs.messages = normalizeMessagesForBinding(\n\t\t\t\tbody.messages as Record<string, unknown>[],\n\t\t\t);\n\t\t}\n\t\tif (body.tools) inputs.tools = body.tools;\n\t\tif (typeof body.temperature === \"number\") inputs.temperature = body.temperature;\n\t\tif (typeof body.max_tokens === \"number\") inputs.max_tokens = body.max_tokens;\n\t\tif (body.response_format) inputs.response_format = body.response_format;\n\t\tif (stream) inputs.stream = true;\n\n\t\tconst runOptions: Record<string, unknown> = {};\n\t\tif (options?.extraHeaders) runOptions.extraHeaders = options.extraHeaders;\n\t\tif (init?.signal) runOptions.signal = init.signal;\n\n\t\tconst result = await binding.run(\n\t\t\tmodel,\n\t\t\tinputs,\n\t\t\tObject.keys(runOptions).length > 0 ? runOptions : undefined,\n\t\t);\n\n\t\tif (stream && result instanceof ReadableStream) {\n\t\t\t// Workers AI returns an SSE stream with `data: {\"response\":\"chunk\"}` format.\n\t\t\t// Transform it to OpenAI-compatible SSE format.\n\t\t\tconst transformed = transformWorkersAiStream(\n\t\t\t\tresult as ReadableStream<Uint8Array>,\n\t\t\t\tmodel,\n\t\t\t);\n\t\t\treturn new Response(transformed, {\n\t\t\t\theaders: {\n\t\t\t\t\t\"Content-Type\": \"text/event-stream\",\n\t\t\t\t\t\"Cache-Control\": \"no-cache\",\n\t\t\t\t},\n\t\t\t});\n\t\t}\n\n\t\t// Graceful degradation: some models return a complete (non-streaming)\n\t\t// response even when `stream: true` is requested. Fall through to the\n\t\t// non-streaming wrapper which produces a valid OpenAI Chat Completion\n\t\t// response that the SDK can consume.\n\n\t\t// Non-streaming: Workers AI returns { response: \"text\", tool_calls?: [...] }\n\t\t// Wrap into OpenAI Chat Completion format.\n\t\tconst responseObj =\n\t\t\ttypeof result === \"object\" && result !== null\n\t\t\t\t? (result as Record<string, unknown>)\n\t\t\t\t: { response: String(result) };\n\n\t\tconst responseText =\n\t\t\ttypeof responseObj.response === \"string\"\n\t\t\t\t? responseObj.response\n\t\t\t\t: typeof responseObj.response === \"object\" && responseObj.response !== null\n\t\t\t\t\t? JSON.stringify(responseObj.response)\n\t\t\t\t\t: \"\";\n\n\t\tconst message: Record<string, unknown> = {\n\t\t\trole: \"assistant\",\n\t\t\tcontent: responseText,\n\t\t};\n\t\tlet finishReason = \"stop\";\n\n\t\t// Handle tool calls if present in Workers AI response\n\t\tif (Array.isArray(responseObj.tool_calls) && responseObj.tool_calls.length > 0) {\n\t\t\tfinishReason = \"tool_calls\";\n\t\t\tmessage.tool_calls = responseObj.tool_calls.map(\n\t\t\t\t(tc: {\n\t\t\t\t\tid?: string;\n\t\t\t\t\tname?: string;\n\t\t\t\t\targuments: unknown;\n\t\t\t\t\tfunction?: { name: string; arguments?: unknown };\n\t\t\t\t}) => ({\n\t\t\t\t\tid: tc.id || crypto.randomUUID(),\n\t\t\t\t\ttype: \"function\",\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: tc.function?.name || tc.name || \"\",\n\t\t\t\t\t\targuments:\n\t\t\t\t\t\t\ttypeof (tc.function?.arguments ?? tc.arguments) === \"string\"\n\t\t\t\t\t\t\t\t? ((tc.function?.arguments ?? tc.arguments) as string)\n\t\t\t\t\t\t\t\t: JSON.stringify(tc.function?.arguments ?? tc.arguments ?? {}),\n\t\t\t\t\t},\n\t\t\t\t}),\n\t\t\t);\n\t\t}\n\n\t\tconst openAiResponse = {\n\t\t\tid: `workers-ai-${crypto.randomUUID()}`,\n\t\t\tobject: \"chat.completion\",\n\t\t\tcreated: Math.floor(Date.now() / 1000),\n\t\t\tmodel,\n\t\t\tchoices: [{ index: 0, message, finish_reason: finishReason }],\n\t\t};\n\n\t\treturn new Response(JSON.stringify(openAiResponse), {\n\t\t\theaders: { \"Content-Type\": \"application/json\" },\n\t\t});\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// Stream transformer: Workers AI SSE -> OpenAI-compatible SSE\n// Uses TransformStream for proper backpressure.\n// ---------------------------------------------------------------------------\n\n/**\n * Transforms a Workers AI SSE stream (data: {\"response\":\"chunk\"}) into\n * an OpenAI-compatible SSE stream (data: {\"choices\":[{\"delta\":{\"content\":\"chunk\"}}]}).\n *\n * Workers AI binding streams tool calls in an OpenAI-like nested format:\n * { tool_calls: [{ id, type, index, function: { name, arguments } }] }\n * Arguments are streamed incrementally across multiple SSE chunks, so the\n * transformer must forward them as incremental deltas rather than a single blob.\n */\nfunction transformWorkersAiStream(\n\tsource: ReadableStream<Uint8Array>,\n\tmodel: string,\n): ReadableStream<Uint8Array> {\n\tconst decoder = new TextDecoder();\n\tconst encoder = new TextEncoder();\n\t// Generate a stable ID and timestamp for the entire stream, matching OpenAI's\n\t// convention where all chunks in a single response share the same id/created.\n\tconst streamId = `workers-ai-${crypto.randomUUID()}`;\n\tconst created = Math.floor(Date.now() / 1000);\n\tlet buffer = \"\";\n\tlet hasToolCalls = false;\n\t// When true, the source stream is already in OpenAI format (some models\n\t// like Qwen3, Kimi K2.5 stream OpenAI-compatible SSE through the binding).\n\t// In that case, flush() should only emit [DONE] and skip the finish chunk.\n\tlet isOpenAiFormat = false;\n\t// Track tool call state per index: store the generated/assigned ID so that\n\t// subsequent argument deltas use the same ID (matching the working streaming.ts pattern).\n\tconst toolCallState = new Map<number, { id: string; name: string }>();\n\n\treturn source.pipeThrough(\n\t\tnew TransformStream<Uint8Array, Uint8Array>({\n\t\t\ttransform(chunk, controller) {\n\t\t\t\tbuffer += decoder.decode(chunk, { stream: true });\n\t\t\t\tconst lines = buffer.split(\"\\n\");\n\t\t\t\tbuffer = lines.pop() || \"\";\n\n\t\t\t\tfor (const line of lines) {\n\t\t\t\t\tconst trimmed = line.trim();\n\t\t\t\t\tif (!trimmed || !trimmed.startsWith(\"data: \")) continue;\n\t\t\t\t\tconst data = trimmed.slice(6);\n\n\t\t\t\t\t// Swallow source [DONE]; we emit our own in flush()\n\t\t\t\t\tif (data === \"[DONE]\") continue;\n\n\t\t\t\t\ttry {\n\t\t\t\t\t\tconst parsed = JSON.parse(data);\n\n\t\t\t\t\t\t// Some models (Qwen3, Kimi K2.5) return OpenAI-compatible format\n\t\t\t\t\t\t// directly through the binding, with `choices[].delta.content` and\n\t\t\t\t\t\t// optional `reasoning_content`. Detect this and pass through as-is.\n\t\t\t\t\t\tif (parsed.choices !== undefined) {\n\t\t\t\t\t\t\t// Already OpenAI format — pass through but ensure each tool call\n\t\t\t\t\t\t\t// index gets a unique, stable ID across all chunks.\n\t\t\t\t\t\t\tisOpenAiFormat = true;\n\t\t\t\t\t\t\tconst choice = parsed.choices?.[0];\n\t\t\t\t\t\t\tif (choice?.delta?.tool_calls) {\n\t\t\t\t\t\t\t\thasToolCalls = true;\n\t\t\t\t\t\t\t\tfor (const tc of choice.delta.tool_calls) {\n\t\t\t\t\t\t\t\t\tconst tcIndex = tc.index ?? 0;\n\t\t\t\t\t\t\t\t\tif (!toolCallState.has(tcIndex)) {\n\t\t\t\t\t\t\t\t\t\t// First chunk for this index — generate/store unique ID\n\t\t\t\t\t\t\t\t\t\tconst id = tc.id || `call${streamId}${tcIndex}`;\n\t\t\t\t\t\t\t\t\t\ttoolCallState.set(tcIndex, {\n\t\t\t\t\t\t\t\t\t\t\tid,\n\t\t\t\t\t\t\t\t\t\t\tname: tc.function?.name || \"\",\n\t\t\t\t\t\t\t\t\t\t});\n\t\t\t\t\t\t\t\t\t\ttc.id = id;\n\t\t\t\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t\t\t\t// Subsequent chunk — reuse stored ID, remove id from delta\n\t\t\t\t\t\t\t\t\t\t// (OpenAI format only sends id in first chunk)\n\t\t\t\t\t\t\t\t\t\tdelete tc.id;\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tif (choice?.finish_reason === \"tool_calls\") {\n\t\t\t\t\t\t\t\thasToolCalls = true;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tcontroller.enqueue(\n\t\t\t\t\t\t\t\tencoder.encode(`data: ${JSON.stringify(parsed)}\\n\\n`),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\tcontinue;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// --- Workers AI native format handling below ---\n\n\t\t\t\t\t\t// Text content\n\t\t\t\t\t\tif (parsed.response != null && parsed.response !== \"\") {\n\t\t\t\t\t\t\tconst openAiChunk = {\n\t\t\t\t\t\t\t\tid: streamId,\n\t\t\t\t\t\t\t\tobject: \"chat.completion.chunk\",\n\t\t\t\t\t\t\t\tcreated,\n\t\t\t\t\t\t\t\tmodel,\n\t\t\t\t\t\t\t\tchoices: [\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t\t\t\t\tdelta: { content: parsed.response },\n\t\t\t\t\t\t\t\t\t\tfinish_reason: null,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\tcontroller.enqueue(\n\t\t\t\t\t\t\t\tencoder.encode(`data: ${JSON.stringify(openAiChunk)}\\n\\n`),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Tool calls — Workers AI binding streams these incrementally:\n\t\t\t\t\t\t// Chunk A: { id, type, index, function: { name } } — start\n\t\t\t\t\t\t// Chunk B: { index, function: { arguments: \"partial...\" } } — args delta\n\t\t\t\t\t\t// Chunk C: { index, function: { arguments: \"rest...\" } } — args delta\n\t\t\t\t\t\t// Chunk D: { id: null, type: null, index, function: { name: null, arguments: \"\" } } — finalize (skip)\n\t\t\t\t\t\tif (Array.isArray(parsed.tool_calls) && parsed.tool_calls.length > 0) {\n\t\t\t\t\t\t\tfor (const tc of parsed.tool_calls) {\n\t\t\t\t\t\t\t\tconst tcIndex = tc.index ?? 0;\n\n\t\t\t\t\t\t\t\t// Resolve name and arguments from either nested or flat format\n\t\t\t\t\t\t\t\tconst tcName = tc.function?.name ?? tc.name ?? null;\n\t\t\t\t\t\t\t\tconst tcArgs = tc.function?.arguments ?? tc.arguments ?? null;\n\t\t\t\t\t\t\t\tconst tcId = tc.id ?? null;\n\n\t\t\t\t\t\t\t\t// Skip finalization chunks where everything is null/empty\n\t\t\t\t\t\t\t\tif (!tcId && !tcName && (!tcArgs || tcArgs === \"\")) continue;\n\n\t\t\t\t\t\t\t\thasToolCalls = true;\n\n\t\t\t\t\t\t\t\t// Build the OpenAI-compatible tool_calls delta\n\t\t\t\t\t\t\t\tconst toolCallDelta: Record<string, unknown> = {\n\t\t\t\t\t\t\t\t\tindex: tcIndex,\n\t\t\t\t\t\t\t\t};\n\n\t\t\t\t\t\t\t\tif (!toolCallState.has(tcIndex)) {\n\t\t\t\t\t\t\t\t\t// First chunk for this tool call index — emit id, type, name.\n\t\t\t\t\t\t\t\t\tconst id = tcId || `call${streamId}${tcIndex}`;\n\t\t\t\t\t\t\t\t\ttoolCallState.set(tcIndex, { id, name: tcName || \"\" });\n\t\t\t\t\t\t\t\t\ttoolCallDelta.id = id;\n\t\t\t\t\t\t\t\t\ttoolCallDelta.type = \"function\";\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function = {\n\t\t\t\t\t\t\t\t\t\tname: tcName || \"\",\n\t\t\t\t\t\t\t\t\t\t// Include arguments if they arrive in the same chunk\n\t\t\t\t\t\t\t\t\t\targuments:\n\t\t\t\t\t\t\t\t\t\t\ttcArgs != null\n\t\t\t\t\t\t\t\t\t\t\t\t? typeof tcArgs === \"string\"\n\t\t\t\t\t\t\t\t\t\t\t\t\t? tcArgs\n\t\t\t\t\t\t\t\t\t\t\t\t\t: JSON.stringify(tcArgs)\n\t\t\t\t\t\t\t\t\t\t\t\t: \"\",\n\t\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t\t\t// Subsequent chunks — only include arguments delta\n\t\t\t\t\t\t\t\t\tif (tcArgs != null && tcArgs !== \"\") {\n\t\t\t\t\t\t\t\t\t\ttoolCallDelta.function = {\n\t\t\t\t\t\t\t\t\t\t\targuments:\n\t\t\t\t\t\t\t\t\t\t\t\ttypeof tcArgs === \"string\"\n\t\t\t\t\t\t\t\t\t\t\t\t\t? tcArgs\n\t\t\t\t\t\t\t\t\t\t\t\t\t: JSON.stringify(tcArgs),\n\t\t\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t\t\t\tcontinue; // Nothing useful to forward\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\t\tconst toolChunk = {\n\t\t\t\t\t\t\t\t\tid: streamId,\n\t\t\t\t\t\t\t\t\tobject: \"chat.completion.chunk\",\n\t\t\t\t\t\t\t\t\tcreated,\n\t\t\t\t\t\t\t\t\tmodel,\n\t\t\t\t\t\t\t\t\tchoices: [\n\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t\t\t\t\t\tdelta: { tool_calls: [toolCallDelta] },\n\t\t\t\t\t\t\t\t\t\t\tfinish_reason: null,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t\tcontroller.enqueue(\n\t\t\t\t\t\t\t\t\tencoder.encode(`data: ${JSON.stringify(toolChunk)}\\n\\n`),\n\t\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t} catch (e) {\n\t\t\t\t\t\t// Log malformed SSE events for debugging; don't break the stream.\n\t\t\t\t\t\tconsole.warn(\"[tanstack-ai] failed to parse SSE event:\", data, e);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t\tflush(controller) {\n\t\t\t\tif (!isOpenAiFormat) {\n\t\t\t\t\t// Workers AI native format: emit a finish chunk with stop/tool_calls\n\t\t\t\t\tconst finalChunk = {\n\t\t\t\t\t\tid: streamId,\n\t\t\t\t\t\tobject: \"chat.completion.chunk\",\n\t\t\t\t\t\tcreated,\n\t\t\t\t\t\tmodel,\n\t\t\t\t\t\tchoices: [\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t\t\tdelta: {},\n\t\t\t\t\t\t\t\tfinish_reason: hasToolCalls ? \"tool_calls\" : \"stop\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t],\n\t\t\t\t\t};\n\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${JSON.stringify(finalChunk)}\\n\\n`));\n\t\t\t\t}\n\t\t\t\t// OpenAI format already includes its own finish_reason in the stream.\n\t\t\t\t// Either way, emit a [DONE] sentinel.\n\t\t\t\tcontroller.enqueue(encoder.encode(\"data: [DONE]\\n\\n\"));\n\t\t\t},\n\t\t}),\n\t);\n}\n"]}
@@ -1,46 +0,0 @@
1
- // src/utils/binary.ts
2
- function uint8ArrayToBase64(bytes) {
3
- let binary = "";
4
- for (let i = 0; i < bytes.length; i++) {
5
- binary += String.fromCharCode(bytes[i]);
6
- }
7
- return btoa(binary);
8
- }
9
- async function collectStream(stream) {
10
- const reader = stream.getReader();
11
- const chunks = [];
12
- while (true) {
13
- const { done, value } = await reader.read();
14
- if (done) break;
15
- chunks.push(value);
16
- }
17
- const totalLength = chunks.reduce((acc, c) => acc + c.length, 0);
18
- const combined = new Uint8Array(totalLength);
19
- let offset = 0;
20
- for (const chunk of chunks) {
21
- combined.set(chunk, offset);
22
- offset += chunk.length;
23
- }
24
- return combined;
25
- }
26
- async function binaryToBase64(result, binaryKey = "image") {
27
- if (result instanceof ReadableStream) {
28
- return uint8ArrayToBase64(await collectStream(result));
29
- }
30
- if (result instanceof Uint8Array || result instanceof ArrayBuffer) {
31
- const bytes = result instanceof ArrayBuffer ? new Uint8Array(result) : result;
32
- return uint8ArrayToBase64(bytes);
33
- }
34
- if (typeof result === "object" && result !== null && binaryKey in result) {
35
- return result[binaryKey];
36
- }
37
- throw new Error(
38
- `Unexpected binary response format from Workers AI (expected Uint8Array, ArrayBuffer, ReadableStream, or { ${binaryKey}: string })`
39
- );
40
- }
41
-
42
- export {
43
- uint8ArrayToBase64,
44
- binaryToBase64
45
- };
46
- //# sourceMappingURL=chunk-XU7YEPML.js.map
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../src/utils/binary.ts"],"sourcesContent":["/**\n * Shared binary-data utilities for Workers AI adapters.\n *\n * Workers AI returns binary data (images, audio) in various formats —\n * Uint8Array, ArrayBuffer, ReadableStream, or JSON objects with a base64 field.\n * These helpers normalise everything into base64 strings for the TanStack AI\n * result types.\n */\n\n/**\n * Convert a Uint8Array to a base64 string.\n */\nexport function uint8ArrayToBase64(bytes: Uint8Array): string {\n\tlet binary = \"\";\n\tfor (let i = 0; i < bytes.length; i++) {\n\t\tbinary += String.fromCharCode(bytes[i]!);\n\t}\n\treturn btoa(binary);\n}\n\n/**\n * Collect all chunks from a ReadableStream<Uint8Array> into a single Uint8Array.\n */\nexport async function collectStream(stream: ReadableStream<Uint8Array>): Promise<Uint8Array> {\n\tconst reader = stream.getReader();\n\tconst chunks: Uint8Array[] = [];\n\twhile (true) {\n\t\tconst { done, value } = await reader.read();\n\t\tif (done) break;\n\t\tchunks.push(value);\n\t}\n\tconst totalLength = chunks.reduce((acc, c) => acc + c.length, 0);\n\tconst combined = new Uint8Array(totalLength);\n\tlet offset = 0;\n\tfor (const chunk of chunks) {\n\t\tcombined.set(chunk, offset);\n\t\toffset += chunk.length;\n\t}\n\treturn combined;\n}\n\n/**\n * Normalise a Workers AI binary response to a base64 string.\n *\n * Handles:\n * - `Uint8Array` / `ArrayBuffer` — raw bytes\n * - `ReadableStream<Uint8Array>` — streamed bytes\n * - `{ [binaryKey]: \"base64...\" }` — JSON wrapper (e.g. `{ image: \"...\" }`)\n *\n * @param result The raw value returned from Workers AI\n * @param binaryKey The JSON field to look for when the result is an object\n * (defaults to `\"image\"`; pass `\"audio\"` for TTS responses)\n */\nexport async function binaryToBase64(\n\tresult: unknown,\n\tbinaryKey: string = \"image\",\n): Promise<string> {\n\tif (result instanceof ReadableStream) {\n\t\treturn uint8ArrayToBase64(await collectStream(result));\n\t}\n\n\tif (result instanceof Uint8Array || result instanceof ArrayBuffer) {\n\t\tconst bytes = result instanceof ArrayBuffer ? new Uint8Array(result) : result;\n\t\treturn uint8ArrayToBase64(bytes);\n\t}\n\n\t// Some models return { image: \"base64...\" } or { audio: \"base64...\" }\n\tif (typeof result === \"object\" && result !== null && binaryKey in result) {\n\t\treturn (result as Record<string, string>)[binaryKey]!;\n\t}\n\n\tthrow new Error(\n\t\t`Unexpected binary response format from Workers AI (expected Uint8Array, ArrayBuffer, ReadableStream, or { ${binaryKey}: string })`,\n\t);\n}\n"],"mappings":";AAYO,SAAS,mBAAmB,OAA2B;AAC7D,MAAI,SAAS;AACb,WAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACtC,cAAU,OAAO,aAAa,MAAM,CAAC,CAAE;AAAA,EACxC;AACA,SAAO,KAAK,MAAM;AACnB;AAKA,eAAsB,cAAc,QAAyD;AAC5F,QAAM,SAAS,OAAO,UAAU;AAChC,QAAM,SAAuB,CAAC;AAC9B,SAAO,MAAM;AACZ,UAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,QAAI,KAAM;AACV,WAAO,KAAK,KAAK;AAAA,EAClB;AACA,QAAM,cAAc,OAAO,OAAO,CAAC,KAAK,MAAM,MAAM,EAAE,QAAQ,CAAC;AAC/D,QAAM,WAAW,IAAI,WAAW,WAAW;AAC3C,MAAI,SAAS;AACb,aAAW,SAAS,QAAQ;AAC3B,aAAS,IAAI,OAAO,MAAM;AAC1B,cAAU,MAAM;AAAA,EACjB;AACA,SAAO;AACR;AAcA,eAAsB,eACrB,QACA,YAAoB,SACF;AAClB,MAAI,kBAAkB,gBAAgB;AACrC,WAAO,mBAAmB,MAAM,cAAc,MAAM,CAAC;AAAA,EACtD;AAEA,MAAI,kBAAkB,cAAc,kBAAkB,aAAa;AAClE,UAAM,QAAQ,kBAAkB,cAAc,IAAI,WAAW,MAAM,IAAI;AACvE,WAAO,mBAAmB,KAAK;AAAA,EAChC;AAGA,MAAI,OAAO,WAAW,YAAY,WAAW,QAAQ,aAAa,QAAQ;AACzE,WAAQ,OAAkC,SAAS;AAAA,EACpD;AAEA,QAAM,IAAI;AAAA,IACT,6GAA6G,SAAS;AAAA,EACvH;AACD;","names":[]}