@cloudflare/tanstack-ai 0.1.6 → 0.1.7

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 +6 -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 +18 -3
  58. package/dist/adapters/workers-ai.d.mts +18 -0
  59. package/dist/adapters/workers-ai.mjs +470 -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-5iL34e6H.cjs +310 -0
  66. package/dist/create-fetcher-5iL34e6H.cjs.map +1 -0
  67. package/dist/create-fetcher-6p6heb85.d.mts +93 -0
  68. package/dist/create-fetcher-DY7wfYYy.mjs +281 -0
  69. package/dist/create-fetcher-DY7wfYYy.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-Bm7Up4or.cjs +510 -0
  78. package/dist/workers-ai-Bm7Up4or.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
@@ -0,0 +1,510 @@
1
+ //#region \0rolldown/runtime.js
2
+ var __create = Object.create;
3
+ var __defProp = Object.defineProperty;
4
+ var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
5
+ var __getOwnPropNames = Object.getOwnPropertyNames;
6
+ var __getProtoOf = Object.getPrototypeOf;
7
+ var __hasOwnProp = Object.prototype.hasOwnProperty;
8
+ var __copyProps = (to, from, except, desc) => {
9
+ if (from && typeof from === "object" || typeof from === "function") for (var keys = __getOwnPropNames(from), i = 0, n = keys.length, key; i < n; i++) {
10
+ key = keys[i];
11
+ if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, {
12
+ get: ((k) => from[k]).bind(null, key),
13
+ enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
14
+ });
15
+ }
16
+ return to;
17
+ };
18
+ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", {
19
+ value: mod,
20
+ enumerable: true
21
+ }) : target, mod));
22
+ //#endregion
23
+ const require_create_fetcher = require("./create-fetcher-5iL34e6H.cjs");
24
+ const require_defineProperty = require("./defineProperty-DQoAg20E.cjs");
25
+ let _tanstack_ai_adapters = require("@tanstack/ai/adapters");
26
+ let openai = require("openai");
27
+ openai = __toESM(openai);
28
+ //#region src/adapters/workers-ai.ts
29
+ function buildWorkersAiClient(config) {
30
+ require_create_fetcher.validateWorkersAiConfig(config);
31
+ const sessionHeaders = config.sessionAffinity ? { "x-session-affinity": config.sessionAffinity } : void 0;
32
+ if (require_create_fetcher.isDirectBindingConfig(config)) return new openai.default({
33
+ apiKey: "unused",
34
+ fetch: require_create_fetcher.createWorkersAiBindingFetch(config.binding, sessionHeaders ? { extraHeaders: sessionHeaders } : void 0)
35
+ });
36
+ if (require_create_fetcher.isDirectCredentialsConfig(config)) return new openai.default({
37
+ baseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,
38
+ apiKey: config.apiKey,
39
+ defaultHeaders: sessionHeaders
40
+ });
41
+ const gatewayConfig = config;
42
+ return new openai.default({
43
+ fetch: require_create_fetcher.createGatewayFetch("workers-ai", gatewayConfig, sessionHeaders),
44
+ apiKey: gatewayConfig.apiKey ?? "unused"
45
+ });
46
+ }
47
+ function extractTextContent(content) {
48
+ if (content === null) return "";
49
+ if (typeof content === "string") return content;
50
+ return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
51
+ }
52
+ /**
53
+ * Convert a single TanStack AI {@link ContentPart} to the OpenAI Chat
54
+ * Completions multi-modal format.
55
+ *
56
+ * TODO: handle other content types (audio, video, document)
57
+ */
58
+ function convertContentPart(part) {
59
+ switch (part.type) {
60
+ case "text":
61
+ if (part.content) return {
62
+ type: "text",
63
+ text: part.content
64
+ };
65
+ return;
66
+ case "image": {
67
+ let url;
68
+ if (part.source.type === "data") url = part.source.value.startsWith("data:") ? part.source.value : `data:${part.source.mimeType};base64,${part.source.value}`;
69
+ else url = part.source.value;
70
+ return {
71
+ type: "image_url",
72
+ image_url: { url }
73
+ };
74
+ }
75
+ default:
76
+ console.warn(`[@cloudflare/tanstack-ai] Unsupported content part type "${part.type}" — skipping`);
77
+ return;
78
+ }
79
+ }
80
+ /**
81
+ * Build OpenAI-compatible user message content from TanStack AI content.
82
+ *
83
+ * If the content has only text parts, returns a plain string.
84
+ * If it includes image parts, returns an array of content parts in
85
+ * OpenAI's multi-modal format (text + image_url).
86
+ */
87
+ function buildUserContent(content) {
88
+ if (content === null) return "";
89
+ if (typeof content === "string") return content;
90
+ if (!content.some((p) => p.type === "image")) return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
91
+ const parts = [];
92
+ for (const part of content) {
93
+ const converted = convertContentPart(part);
94
+ if (converted) parts.push(converted);
95
+ }
96
+ return parts;
97
+ }
98
+ function buildOpenAIMessages(systemPrompts, messages, options) {
99
+ const includeTools = options?.includeToolMessages ?? true;
100
+ const openAIMessages = [];
101
+ if (systemPrompts && systemPrompts.length > 0) openAIMessages.push({
102
+ role: "system",
103
+ content: systemPrompts.join("\n")
104
+ });
105
+ for (const message of messages) if (message.role === "user") openAIMessages.push({
106
+ role: "user",
107
+ content: buildUserContent(message.content)
108
+ });
109
+ else if (message.role === "assistant") {
110
+ const assistantMessage = {
111
+ role: "assistant",
112
+ content: extractTextContent(message.content)
113
+ };
114
+ if (includeTools && message.toolCalls && message.toolCalls.length > 0) assistantMessage.tool_calls = message.toolCalls.map((tc) => ({
115
+ id: tc.id,
116
+ type: "function",
117
+ function: {
118
+ name: tc.function.name,
119
+ arguments: tc.function.arguments
120
+ }
121
+ }));
122
+ openAIMessages.push(assistantMessage);
123
+ } else if (includeTools && message.role === "tool") {
124
+ let toolContent;
125
+ if (typeof message.content === "string") try {
126
+ JSON.parse(message.content);
127
+ toolContent = message.content;
128
+ } catch {
129
+ toolContent = JSON.stringify(message.content);
130
+ }
131
+ else toolContent = JSON.stringify(message.content);
132
+ openAIMessages.push({
133
+ role: "tool",
134
+ tool_call_id: message.toolCallId || `tool_${crypto.randomUUID().slice(0, 8)}`,
135
+ content: toolContent
136
+ });
137
+ }
138
+ return openAIMessages;
139
+ }
140
+ function buildOpenAITools(tools) {
141
+ if (!tools) return void 0;
142
+ return tools.map((tool) => ({
143
+ type: "function",
144
+ function: {
145
+ name: tool.name,
146
+ description: tool.description,
147
+ parameters: tool.inputSchema
148
+ }
149
+ }));
150
+ }
151
+ function generateId(prefix = "chatcmpl") {
152
+ return `${prefix}-${crypto.randomUUID().replace(/-/g, "").slice(0, 16)}`;
153
+ }
154
+ var WorkersAiTextAdapter = class extends _tanstack_ai_adapters.BaseTextAdapter {
155
+ constructor(model, config) {
156
+ super({ apiKey: "unused" }, model);
157
+ require_defineProperty._defineProperty(this, "name", "workers-ai");
158
+ require_defineProperty._defineProperty(this, "client", void 0);
159
+ this.client = buildWorkersAiClient(config);
160
+ }
161
+ async *chatStream(options) {
162
+ const { systemPrompts, messages, tools, temperature, maxTokens, model } = options;
163
+ const openAIMessages = buildOpenAIMessages(systemPrompts, messages);
164
+ const openAITools = buildOpenAITools(tools);
165
+ const timestamp = Date.now();
166
+ const runId = generateId();
167
+ const messageId = generateId();
168
+ let hasEmittedRunStarted = false;
169
+ let hasEmittedTextMessageStart = false;
170
+ let accumulatedContent = "";
171
+ let hasEmittedStepStarted = false;
172
+ let accumulatedReasoning = "";
173
+ const stepId = generateId();
174
+ let hasReceivedFinishReason = false;
175
+ const toolCallsInProgress = /* @__PURE__ */ new Map();
176
+ try {
177
+ let stream;
178
+ try {
179
+ stream = await this.client.chat.completions.create({
180
+ model: model ?? this.model,
181
+ messages: openAIMessages,
182
+ tools: openAITools,
183
+ temperature,
184
+ max_tokens: maxTokens,
185
+ stream: true,
186
+ stream_options: { include_usage: true }
187
+ });
188
+ } catch (streamError) {
189
+ console.warn("[tanstack-ai] Streaming failed, falling back to non-streaming:", streamError instanceof Error ? streamError.message : streamError);
190
+ const nonStreamResult = await this.client.chat.completions.create({
191
+ model: model ?? this.model,
192
+ messages: openAIMessages,
193
+ tools: openAITools,
194
+ temperature,
195
+ max_tokens: maxTokens
196
+ });
197
+ yield {
198
+ type: "RUN_STARTED",
199
+ runId,
200
+ model: nonStreamResult.model || model || this.model,
201
+ timestamp
202
+ };
203
+ const msg = nonStreamResult.choices[0]?.message;
204
+ if (msg?.content) {
205
+ yield {
206
+ type: "TEXT_MESSAGE_START",
207
+ messageId,
208
+ model: nonStreamResult.model || model || this.model,
209
+ timestamp,
210
+ role: "assistant"
211
+ };
212
+ yield {
213
+ type: "TEXT_MESSAGE_CONTENT",
214
+ messageId,
215
+ model: nonStreamResult.model || model || this.model,
216
+ timestamp,
217
+ delta: msg.content,
218
+ content: msg.content
219
+ };
220
+ yield {
221
+ type: "TEXT_MESSAGE_END",
222
+ messageId,
223
+ model: nonStreamResult.model || model || this.model,
224
+ timestamp
225
+ };
226
+ }
227
+ if (msg?.tool_calls) for (const tc of msg.tool_calls) {
228
+ if (tc.type !== "function") continue;
229
+ const fn = tc.function;
230
+ let parsedInput = {};
231
+ try {
232
+ parsedInput = fn.arguments ? JSON.parse(fn.arguments) : {};
233
+ } catch {
234
+ parsedInput = {};
235
+ }
236
+ yield {
237
+ type: "TOOL_CALL_START",
238
+ toolCallId: tc.id,
239
+ toolName: fn.name,
240
+ model: nonStreamResult.model || model || this.model,
241
+ timestamp,
242
+ index: 0
243
+ };
244
+ yield {
245
+ type: "TOOL_CALL_END",
246
+ toolCallId: tc.id,
247
+ toolName: fn.name,
248
+ model: nonStreamResult.model || model || this.model,
249
+ timestamp,
250
+ input: parsedInput
251
+ };
252
+ }
253
+ const finishReason = nonStreamResult.choices[0]?.finish_reason;
254
+ yield {
255
+ type: "RUN_FINISHED",
256
+ runId,
257
+ model: nonStreamResult.model || model || this.model,
258
+ timestamp,
259
+ usage: nonStreamResult.usage ? {
260
+ promptTokens: nonStreamResult.usage.prompt_tokens,
261
+ completionTokens: nonStreamResult.usage.completion_tokens,
262
+ totalTokens: nonStreamResult.usage.total_tokens
263
+ } : void 0,
264
+ finishReason: finishReason === "tool_calls" || finishReason === "function_call" ? "tool_calls" : finishReason ?? "stop"
265
+ };
266
+ return;
267
+ }
268
+ for await (const chunk of stream) {
269
+ if (!chunk.choices || chunk.choices.length === 0) continue;
270
+ const choice = chunk.choices[0];
271
+ if (!choice) continue;
272
+ if (!hasEmittedRunStarted) {
273
+ hasEmittedRunStarted = true;
274
+ yield {
275
+ type: "RUN_STARTED",
276
+ runId,
277
+ model: chunk.model || model || this.model,
278
+ timestamp
279
+ };
280
+ }
281
+ const delta = choice.delta;
282
+ const reasoningContent = delta.reasoning_content ?? delta.reasoning;
283
+ if (reasoningContent) {
284
+ if (!hasEmittedStepStarted) {
285
+ hasEmittedStepStarted = true;
286
+ yield {
287
+ type: "STEP_STARTED",
288
+ stepId,
289
+ stepType: "thinking",
290
+ model: chunk.model || model || this.model,
291
+ timestamp
292
+ };
293
+ }
294
+ accumulatedReasoning += reasoningContent;
295
+ yield {
296
+ type: "STEP_FINISHED",
297
+ stepId,
298
+ delta: reasoningContent,
299
+ content: accumulatedReasoning,
300
+ model: chunk.model || model || this.model,
301
+ timestamp
302
+ };
303
+ }
304
+ if (delta.content) {
305
+ if (!hasEmittedTextMessageStart) {
306
+ hasEmittedTextMessageStart = true;
307
+ yield {
308
+ type: "TEXT_MESSAGE_START",
309
+ messageId,
310
+ model: chunk.model || model || this.model,
311
+ timestamp,
312
+ role: "assistant"
313
+ };
314
+ }
315
+ accumulatedContent += delta.content;
316
+ yield {
317
+ type: "TEXT_MESSAGE_CONTENT",
318
+ messageId,
319
+ model: chunk.model || model || this.model,
320
+ timestamp,
321
+ delta: delta.content,
322
+ content: accumulatedContent
323
+ };
324
+ }
325
+ if (delta.tool_calls) for (const toolCallDelta of delta.tool_calls) {
326
+ const index = toolCallDelta.index;
327
+ if (!toolCallsInProgress.has(index)) toolCallsInProgress.set(index, {
328
+ id: generateId("chatcmpl-tool"),
329
+ name: toolCallDelta.function?.name || "",
330
+ arguments: "",
331
+ started: false
332
+ });
333
+ const toolCall = toolCallsInProgress.get(index);
334
+ if (toolCallDelta.function?.name) toolCall.name = toolCallDelta.function.name;
335
+ if (toolCallDelta.function?.arguments) toolCall.arguments += toolCallDelta.function.arguments;
336
+ if (toolCall.id && toolCall.name && !toolCall.started) {
337
+ toolCall.started = true;
338
+ yield {
339
+ type: "TOOL_CALL_START",
340
+ toolCallId: toolCall.id,
341
+ toolName: toolCall.name,
342
+ model: chunk.model || model || this.model,
343
+ timestamp,
344
+ index
345
+ };
346
+ }
347
+ if (toolCallDelta.function?.arguments && toolCall.started) yield {
348
+ type: "TOOL_CALL_ARGS",
349
+ toolCallId: toolCall.id,
350
+ model: chunk.model || model || this.model,
351
+ timestamp,
352
+ delta: toolCallDelta.function.arguments
353
+ };
354
+ }
355
+ if (choice.finish_reason) {
356
+ hasReceivedFinishReason = true;
357
+ if (choice.finish_reason === "tool_calls" || toolCallsInProgress.size > 0) for (const [, toolCall] of toolCallsInProgress) {
358
+ let parsedInput = {};
359
+ try {
360
+ parsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};
361
+ } catch {
362
+ parsedInput = {};
363
+ }
364
+ yield {
365
+ type: "TOOL_CALL_END",
366
+ toolCallId: toolCall.id,
367
+ toolName: toolCall.name,
368
+ model: chunk.model || model || this.model,
369
+ timestamp,
370
+ input: parsedInput
371
+ };
372
+ }
373
+ const computedFinishReason = choice.finish_reason === "tool_calls" || choice.finish_reason === "function_call" || toolCallsInProgress.size > 0 ? "tool_calls" : choice.finish_reason;
374
+ if (hasEmittedTextMessageStart) yield {
375
+ type: "TEXT_MESSAGE_END",
376
+ messageId,
377
+ model: chunk.model || model || this.model,
378
+ timestamp
379
+ };
380
+ yield {
381
+ type: "RUN_FINISHED",
382
+ runId,
383
+ model: chunk.model || model || this.model,
384
+ timestamp,
385
+ usage: chunk.usage ? {
386
+ promptTokens: chunk.usage.prompt_tokens,
387
+ completionTokens: chunk.usage.completion_tokens,
388
+ totalTokens: chunk.usage.total_tokens
389
+ } : void 0,
390
+ finishReason: computedFinishReason
391
+ };
392
+ }
393
+ }
394
+ if (hasEmittedRunStarted && !hasReceivedFinishReason) {
395
+ console.warn("[tanstack-ai] Stream ended without finish_reason — possible truncation or connection drop");
396
+ for (const [, toolCall] of toolCallsInProgress) if (toolCall.started) {
397
+ let parsedInput = {};
398
+ try {
399
+ parsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};
400
+ } catch {
401
+ parsedInput = {};
402
+ }
403
+ yield {
404
+ type: "TOOL_CALL_END",
405
+ toolCallId: toolCall.id,
406
+ toolName: toolCall.name,
407
+ model: model ?? this.model,
408
+ timestamp,
409
+ input: parsedInput
410
+ };
411
+ }
412
+ if (hasEmittedTextMessageStart) yield {
413
+ type: "TEXT_MESSAGE_END",
414
+ messageId,
415
+ model: model ?? this.model,
416
+ timestamp
417
+ };
418
+ yield {
419
+ type: "RUN_FINISHED",
420
+ runId,
421
+ model: model ?? this.model,
422
+ timestamp,
423
+ finishReason: "stop"
424
+ };
425
+ }
426
+ } catch (error) {
427
+ const message = error instanceof Error ? error.message : String(error);
428
+ const code = error instanceof Error ? error.code : void 0;
429
+ if (!hasEmittedRunStarted) yield {
430
+ type: "RUN_STARTED",
431
+ runId,
432
+ model: model ?? this.model,
433
+ timestamp
434
+ };
435
+ yield {
436
+ type: "RUN_ERROR",
437
+ runId,
438
+ model: model ?? this.model,
439
+ timestamp,
440
+ error: {
441
+ message: message || "Unknown error",
442
+ code
443
+ }
444
+ };
445
+ }
446
+ }
447
+ async structuredOutput(options) {
448
+ const { outputSchema, chatOptions } = options;
449
+ const { systemPrompts, messages, temperature, model } = chatOptions;
450
+ const openAIMessages = buildOpenAIMessages(systemPrompts, messages, { includeToolMessages: false });
451
+ const response = await this.client.chat.completions.create({
452
+ model: model ?? this.model,
453
+ messages: openAIMessages,
454
+ temperature,
455
+ stream: false,
456
+ response_format: {
457
+ type: "json_schema",
458
+ json_schema: {
459
+ name: "structured_output",
460
+ strict: true,
461
+ schema: outputSchema
462
+ }
463
+ }
464
+ });
465
+ const choice = response.choices?.[0];
466
+ if (!choice) throw new Error(`Workers AI structured output returned no choices: ${JSON.stringify(response)}`);
467
+ const rawContent = choice.message?.content ?? "";
468
+ let data;
469
+ let rawText;
470
+ if (typeof rawContent === "string") {
471
+ rawText = rawContent;
472
+ try {
473
+ data = JSON.parse(rawText);
474
+ } catch {
475
+ data = rawText;
476
+ }
477
+ } else {
478
+ data = rawContent;
479
+ rawText = JSON.stringify(rawContent);
480
+ }
481
+ return {
482
+ data,
483
+ rawText
484
+ };
485
+ }
486
+ };
487
+ function createWorkersAiChat(model, config) {
488
+ return new WorkersAiTextAdapter(model, config);
489
+ }
490
+ //#endregion
491
+ Object.defineProperty(exports, "WorkersAiTextAdapter", {
492
+ enumerable: true,
493
+ get: function() {
494
+ return WorkersAiTextAdapter;
495
+ }
496
+ });
497
+ Object.defineProperty(exports, "__toESM", {
498
+ enumerable: true,
499
+ get: function() {
500
+ return __toESM;
501
+ }
502
+ });
503
+ Object.defineProperty(exports, "createWorkersAiChat", {
504
+ enumerable: true,
505
+ get: function() {
506
+ return createWorkersAiChat;
507
+ }
508
+ });
509
+
510
+ //# sourceMappingURL=workers-ai-Bm7Up4or.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"workers-ai-Bm7Up4or.cjs","names":["isDirectBindingConfig","OpenAI","createWorkersAiBindingFetch","isDirectCredentialsConfig","createGatewayFetch","BaseTextAdapter"],"sources":["../src/adapters/workers-ai.ts"],"sourcesContent":["import type { AiModels, BaseAiTextGeneration } from \"@cloudflare/workers-types\";\nimport type { ContentPart, ModelMessage, StreamChunk, TextOptions } from \"@tanstack/ai\";\nimport {\n\tBaseTextAdapter,\n\ttype StructuredOutputOptions,\n\ttype StructuredOutputResult,\n} from \"@tanstack/ai/adapters\";\nimport OpenAI from \"openai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tcreateWorkersAiBindingFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model types derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiTextModel =\n\t| {\n\t\t\t[K in keyof AiModels]: AiModels[K] extends BaseAiTextGeneration ? K : never;\n\t }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// Helpers: build the right OpenAI client depending on config mode\n// ---------------------------------------------------------------------------\n\nfunction buildWorkersAiClient(config: WorkersAiAdapterConfig): OpenAI {\n\tvalidateWorkersAiConfig(config);\n\n\tconst sessionHeaders: Record<string, string> | undefined = config.sessionAffinity\n\t\t? { \"x-session-affinity\": config.sessionAffinity }\n\t\t: undefined;\n\n\tif (isDirectBindingConfig(config)) {\n\t\t// Plain binding mode: shim translates OpenAI fetch calls to env.AI.run()\n\t\treturn new OpenAI({\n\t\t\tapiKey: \"unused\",\n\t\t\tfetch: createWorkersAiBindingFetch(\n\t\t\t\tconfig.binding,\n\t\t\t\tsessionHeaders ? { extraHeaders: sessionHeaders } : undefined,\n\t\t\t),\n\t\t});\n\t}\n\n\tif (isDirectCredentialsConfig(config)) {\n\t\t// Plain REST mode: point OpenAI SDK at Workers AI's OpenAI-compatible endpoint\n\t\treturn new OpenAI({\n\t\t\tbaseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,\n\t\t\tapiKey: config.apiKey,\n\t\t\tdefaultHeaders: sessionHeaders,\n\t\t});\n\t}\n\n\t// Gateway mode (existing): use createGatewayFetch\n\tconst gatewayConfig = config as AiGatewayAdapterConfig;\n\treturn new OpenAI({\n\t\tfetch: createGatewayFetch(\"workers-ai\", gatewayConfig, sessionHeaders),\n\t\tapiKey: gatewayConfig.apiKey ?? \"unused\",\n\t});\n}\n\n// ---------------------------------------------------------------------------\n// Shared message-building helpers\n// ---------------------------------------------------------------------------\n\nfunction extractTextContent(content: ModelMessage[\"content\"]): string {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n}\n\n/**\n * Convert a single TanStack AI {@link ContentPart} to the OpenAI Chat\n * Completions multi-modal format.\n *\n * TODO: handle other content types (audio, video, document)\n */\nfunction convertContentPart(part: ContentPart): OpenAI.Chat.ChatCompletionContentPart | undefined {\n\tswitch (part.type) {\n\t\tcase \"text\":\n\t\t\tif (part.content) {\n\t\t\t\treturn { type: \"text\", text: part.content };\n\t\t\t}\n\t\t\treturn undefined;\n\t\tcase \"image\": {\n\t\t\tlet url: string;\n\t\t\tif (part.source.type === \"data\") {\n\t\t\t\turl = part.source.value.startsWith(\"data:\")\n\t\t\t\t\t? part.source.value\n\t\t\t\t\t: `data:${part.source.mimeType};base64,${part.source.value}`;\n\t\t\t} else {\n\t\t\t\turl = part.source.value;\n\t\t\t}\n\t\t\treturn { type: \"image_url\", image_url: { url } };\n\t\t}\n\t\tdefault:\n\t\t\t// audio, video, document — not supported for now\n\t\t\tconsole.warn(\n\t\t\t\t`[@cloudflare/tanstack-ai] Unsupported content part type \"${part.type}\" — skipping`,\n\t\t\t);\n\t\t\treturn undefined;\n\t}\n}\n\n/**\n * Build OpenAI-compatible user message content from TanStack AI content.\n *\n * If the content has only text parts, returns a plain string.\n * If it includes image parts, returns an array of content parts in\n * OpenAI's multi-modal format (text + image_url).\n */\nfunction buildUserContent(\n\tcontent: ModelMessage[\"content\"],\n): string | OpenAI.Chat.ChatCompletionContentPart[] {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\n\tconst hasImages = content.some((p) => p.type === \"image\");\n\tif (!hasImages) {\n\t\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n\t}\n\n\tconst parts: OpenAI.Chat.ChatCompletionContentPart[] = [];\n\tfor (const part of content) {\n\t\tconst converted = convertContentPart(part);\n\t\tif (converted) {\n\t\t\tparts.push(converted);\n\t\t}\n\t}\n\treturn parts;\n}\n\nfunction buildOpenAIMessages(\n\tsystemPrompts: string[] | undefined,\n\tmessages: ModelMessage[],\n\toptions?: { includeToolMessages?: boolean },\n): OpenAI.Chat.ChatCompletionMessageParam[] {\n\tconst includeTools = options?.includeToolMessages ?? true;\n\tconst openAIMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [];\n\n\tif (systemPrompts && systemPrompts.length > 0) {\n\t\topenAIMessages.push({\n\t\t\trole: \"system\",\n\t\t\tcontent: systemPrompts.join(\"\\n\"),\n\t\t});\n\t}\n\n\tfor (const message of messages) {\n\t\tif (message.role === \"user\") {\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"user\",\n\t\t\t\tcontent: buildUserContent(message.content),\n\t\t\t});\n\t\t} else if (message.role === \"assistant\") {\n\t\t\tconst assistantMessage: OpenAI.Chat.ChatCompletionAssistantMessageParam = {\n\t\t\t\trole: \"assistant\",\n\t\t\t\tcontent: extractTextContent(message.content),\n\t\t\t};\n\t\t\tif (includeTools && message.toolCalls && message.toolCalls.length > 0) {\n\t\t\t\tassistantMessage.tool_calls = message.toolCalls.map((tc) => ({\n\t\t\t\t\tid: tc.id,\n\t\t\t\t\ttype: \"function\" as const,\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: tc.function.name,\n\t\t\t\t\t\targuments: tc.function.arguments,\n\t\t\t\t\t},\n\t\t\t\t}));\n\t\t\t}\n\t\t\topenAIMessages.push(assistantMessage);\n\t\t} else if (includeTools && message.role === \"tool\") {\n\t\t\tlet toolContent: string;\n\t\t\tif (typeof message.content === \"string\") {\n\t\t\t\ttry {\n\t\t\t\t\tJSON.parse(message.content);\n\t\t\t\t\ttoolContent = message.content;\n\t\t\t\t} catch {\n\t\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t}\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"tool\",\n\t\t\t\ttool_call_id: message.toolCallId || `tool_${crypto.randomUUID().slice(0, 8)}`,\n\t\t\t\tcontent: toolContent,\n\t\t\t});\n\t\t}\n\t}\n\n\treturn openAIMessages;\n}\n\nfunction buildOpenAITools(\n\ttools: Array<{ name: string; description: string; inputSchema?: unknown }> | undefined,\n): OpenAI.Chat.ChatCompletionTool[] | undefined {\n\tif (!tools) return undefined;\n\treturn tools.map((tool) => ({\n\t\ttype: \"function\" as const,\n\t\tfunction: {\n\t\t\tname: tool.name,\n\t\t\tdescription: tool.description,\n\t\t\tparameters: tool.inputSchema as Record<string, unknown>,\n\t\t},\n\t}));\n}\n\n// ---------------------------------------------------------------------------\n// ID generation\n// ---------------------------------------------------------------------------\n\nfunction generateId(prefix = \"chatcmpl\"): string {\n\treturn `${prefix}-${crypto.randomUUID().replace(/-/g, \"\").slice(0, 16)}`;\n}\n\n// ---------------------------------------------------------------------------\n// WorkersAiTextAdapter: chat / structured output via OpenAI Chat Completions\n// ---------------------------------------------------------------------------\n\n// TODO: Replace `any` generic params with proper types once BaseTextAdapter's\n// provider-options generics stabilize. Workers AI doesn't have provider-specific\n// options in the TanStack sense, so `any` is pragmatic for now.\nexport class WorkersAiTextAdapter<TModel extends WorkersAiTextModel> extends BaseTextAdapter<\n\tTModel,\n\t// eslint-disable-next-line @typescript-eslint/no-explicit-any -- BaseTextAdapter generic params are opaque\n\tany,\n\tany,\n\tany\n> {\n\tname = \"workers-ai\" as const;\n\n\tprivate client: OpenAI;\n\n\tconstructor(model: TModel, config: WorkersAiAdapterConfig) {\n\t\tsuper({ apiKey: \"unused\" }, model);\n\t\tthis.client = buildWorkersAiClient(config);\n\t}\n\n\tasync *chatStream(options: TextOptions<any>): AsyncIterable<StreamChunk> {\n\t\tconst { systemPrompts, messages, tools, temperature, maxTokens, model } = options;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages);\n\t\tconst openAITools = buildOpenAITools(tools);\n\n\t\tconst timestamp = Date.now();\n\t\tconst runId = generateId();\n\t\tconst messageId = generateId();\n\t\tlet hasEmittedRunStarted = false;\n\t\tlet hasEmittedTextMessageStart = false;\n\t\tlet accumulatedContent = \"\";\n\t\tlet hasEmittedStepStarted = false;\n\t\tlet accumulatedReasoning = \"\";\n\t\tconst stepId = generateId();\n\t\tlet hasReceivedFinishReason = false;\n\t\tconst toolCallsInProgress = new Map<\n\t\t\tnumber,\n\t\t\t{ id: string; name: string; arguments: string; started: boolean }\n\t\t>();\n\n\t\ttry {\n\t\t\tlet stream: AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>;\n\t\t\ttry {\n\t\t\t\tstream = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t\tmax_tokens: maxTokens,\n\t\t\t\t\tstream: true,\n\t\t\t\t\tstream_options: { include_usage: true },\n\t\t\t\t});\n\t\t\t} catch (streamError: unknown) {\n\t\t\t\t// Some models (e.g. GPT-OSS) don't support streaming via the REST API.\n\t\t\t\t// Fall back to a non-streaming call and yield the result as a single chunk.\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Streaming failed, falling back to non-streaming:\",\n\t\t\t\t\tstreamError instanceof Error ? streamError.message : streamError,\n\t\t\t\t);\n\t\t\t\tconst nonStreamResult = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t\tmax_tokens: maxTokens,\n\t\t\t\t});\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\n\t\t\t\tconst msg = nonStreamResult.choices[0]?.message;\n\t\t\t\tif (msg?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: msg.content,\n\t\t\t\t\t\tcontent: msg.content,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tif (msg?.tool_calls) {\n\t\t\t\t\tfor (const tc of msg.tool_calls) {\n\t\t\t\t\t\tif (tc.type !== \"function\") continue;\n\t\t\t\t\t\tconst fn = tc.function;\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = fn.arguments ? JSON.parse(fn.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tconst finishReason = nonStreamResult.choices[0]?.finish_reason;\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tusage: nonStreamResult.usage\n\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\tpromptTokens: nonStreamResult.usage.prompt_tokens,\n\t\t\t\t\t\t\t\tcompletionTokens: nonStreamResult.usage.completion_tokens,\n\t\t\t\t\t\t\t\ttotalTokens: nonStreamResult.usage.total_tokens,\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t: undefined,\n\t\t\t\t\tfinishReason:\n\t\t\t\t\t\tfinishReason === \"tool_calls\" || finishReason === \"function_call\"\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: ((finishReason as \"stop\" | \"length\" | \"content_filter\") ?? \"stop\"),\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tfor await (const chunk of stream) {\n\t\t\t\tif (!chunk.choices || chunk.choices.length === 0) continue;\n\t\t\t\tconst choice = chunk.choices[0];\n\t\t\t\tif (!choice) continue;\n\n\t\t\t\t// Emit RUN_STARTED on first chunk\n\t\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\t\thasEmittedRunStarted = true;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tconst delta = choice.delta;\n\n\t\t\t\t// Reasoning content (used by models like QwQ, DeepSeek R1, Kimi K2.5)\n\t\t\t\t// The OpenAI SDK doesn't type this field, but models send it as an extension.\n\t\t\t\tconst reasoningContent = ((delta as Record<string, unknown>).reasoning_content ??\n\t\t\t\t\t(delta as Record<string, unknown>).reasoning) as string | undefined;\n\t\t\t\tif (reasoningContent) {\n\t\t\t\t\t// RUN_STARTED is already guaranteed by the guard above\n\t\t\t\t\tif (!hasEmittedStepStarted) {\n\t\t\t\t\t\thasEmittedStepStarted = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"STEP_STARTED\",\n\t\t\t\t\t\t\tstepId,\n\t\t\t\t\t\t\tstepType: \"thinking\",\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t\taccumulatedReasoning += reasoningContent;\n\t\t\t\t\t// TODO: TanStack AI's StreamProcessor currently treats STEP_FINISHED as an\n\t\t\t\t\t// incremental reasoning event (with `delta` + accumulated `content`), so we\n\t\t\t\t\t// emit one per token. If TanStack AI adds a dedicated STEP_CONTENT event\n\t\t\t\t\t// type, this should be updated to emit STEP_CONTENT per token and a single\n\t\t\t\t\t// STEP_FINISHED when reasoning ends (i.e. when the first non-reasoning\n\t\t\t\t\t// content or finish_reason arrives).\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"STEP_FINISHED\",\n\t\t\t\t\t\tstepId,\n\t\t\t\t\t\tdelta: reasoningContent,\n\t\t\t\t\t\tcontent: accumulatedReasoning,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Text content\n\t\t\t\tif (delta.content) {\n\t\t\t\t\tif (!hasEmittedTextMessageStart) {\n\t\t\t\t\t\thasEmittedTextMessageStart = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\taccumulatedContent += delta.content;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: delta.content,\n\t\t\t\t\t\tcontent: accumulatedContent,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Tool calls\n\t\t\t\tif (delta.tool_calls) {\n\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\tif (!toolCallsInProgress.has(index)) {\n\t\t\t\t\t\t\t// Always generate a unique ID per tool call index.\n\t\t\t\t\t\t\t// The backend may send the same ID for multiple tool calls,\n\t\t\t\t\t\t\t// so we cannot trust toolCallDelta.id to be unique.\n\t\t\t\t\t\t\ttoolCallsInProgress.set(index, {\n\t\t\t\t\t\t\t\tid: generateId(\"chatcmpl-tool\"),\n\t\t\t\t\t\t\t\tname: toolCallDelta.function?.name || \"\",\n\t\t\t\t\t\t\t\targuments: \"\",\n\t\t\t\t\t\t\t\tstarted: false,\n\t\t\t\t\t\t\t});\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tconst toolCall = toolCallsInProgress.get(index)!;\n\n\t\t\t\t\t\t// Only update name if provided (ID is already set at creation time\n\t\t\t\t\t\t// and should not be overwritten by subsequent chunks that may have\n\t\t\t\t\t\t// duplicate/shared IDs from the backend)\n\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\ttoolCall.name = toolCallDelta.function.name;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\ttoolCall.arguments += toolCallDelta.function.arguments;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Emit TOOL_CALL_START once we have id and name\n\t\t\t\t\t\tif (toolCall.id && toolCall.name && !toolCall.started) {\n\t\t\t\t\t\t\ttoolCall.started = true;\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tindex,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Stream tool call arguments\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments && toolCall.started) {\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_ARGS\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tdelta: toolCallDelta.function.arguments,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Finish\n\t\t\t\tif (choice.finish_reason) {\n\t\t\t\t\thasReceivedFinishReason = true;\n\n\t\t\t\t\t// End tool calls\n\t\t\t\t\tif (choice.finish_reason === \"tool_calls\" || toolCallsInProgress.size > 0) {\n\t\t\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\t\tparsedInput = toolCall.arguments\n\t\t\t\t\t\t\t\t\t? JSON.parse(toolCall.arguments)\n\t\t\t\t\t\t\t\t\t: {};\n\t\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\n\t\t\t\t\tconst computedFinishReason =\n\t\t\t\t\t\tchoice.finish_reason === \"tool_calls\" ||\n\t\t\t\t\t\tchoice.finish_reason === \"function_call\" ||\n\t\t\t\t\t\ttoolCallsInProgress.size > 0\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: (choice.finish_reason as \"stop\" | \"length\" | \"content_filter\");\n\n\t\t\t\t\t// End text message if started\n\t\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Emit RUN_FINISHED\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tusage: chunk.usage\n\t\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\t\tpromptTokens: chunk.usage.prompt_tokens,\n\t\t\t\t\t\t\t\t\tcompletionTokens: chunk.usage.completion_tokens,\n\t\t\t\t\t\t\t\t\ttotalTokens: chunk.usage.total_tokens,\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t: undefined,\n\t\t\t\t\t\tfinishReason: computedFinishReason,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Premature stream termination: the stream ended without a finish_reason.\n\t\t\t// This can happen when Workers AI truncates a response or the connection drops.\n\t\t\t// Emit proper closing events so the consumer doesn't hang.\n\t\t\tif (hasEmittedRunStarted && !hasReceivedFinishReason) {\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Stream ended without finish_reason — possible truncation or connection drop\",\n\t\t\t\t);\n\n\t\t\t\t// Close any open tool calls\n\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\tif (toolCall.started) {\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Close text message if open\n\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tfinishReason: \"stop\",\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tconst code =\n\t\t\t\terror instanceof Error ? (error as Error & { code?: string }).code : undefined;\n\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t\tyield {\n\t\t\t\ttype: \"RUN_ERROR\",\n\t\t\t\trunId,\n\t\t\t\tmodel: model ?? this.model,\n\t\t\t\ttimestamp,\n\t\t\t\terror: {\n\t\t\t\t\tmessage: message || \"Unknown error\",\n\t\t\t\t\tcode,\n\t\t\t\t},\n\t\t\t} satisfies StreamChunk;\n\t\t}\n\t}\n\n\tasync structuredOutput(\n\t\toptions: StructuredOutputOptions<any>,\n\t): Promise<StructuredOutputResult<unknown>> {\n\t\tconst { outputSchema, chatOptions } = options;\n\t\tconst { systemPrompts, messages, temperature, model } = chatOptions;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages, {\n\t\t\tincludeToolMessages: false,\n\t\t});\n\n\t\tconst response = await this.client.chat.completions.create({\n\t\t\tmodel: model ?? this.model,\n\t\t\tmessages: openAIMessages,\n\t\t\ttemperature,\n\t\t\tstream: false,\n\t\t\tresponse_format: {\n\t\t\t\ttype: \"json_schema\",\n\t\t\t\tjson_schema: {\n\t\t\t\t\tname: \"structured_output\",\n\t\t\t\t\tstrict: true,\n\t\t\t\t\tschema: outputSchema,\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst choice = response.choices?.[0];\n\n\t\tif (!choice) {\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI structured output returned no choices: ${JSON.stringify(response)}`,\n\t\t\t);\n\t\t}\n\n\t\tconst rawContent = choice.message?.content ?? \"\";\n\n\t\t// Workers AI REST endpoint may return `content` as an already-parsed object\n\t\t// when using json_schema response format, so normalise both cases.\n\t\tlet data: unknown;\n\t\tlet rawText: string;\n\n\t\tif (typeof rawContent === \"string\") {\n\t\t\trawText = rawContent;\n\t\t\ttry {\n\t\t\t\tdata = JSON.parse(rawText);\n\t\t\t} catch {\n\t\t\t\tdata = rawText;\n\t\t\t}\n\t\t} else {\n\t\t\t// Already an object — stringify for rawText, use directly for data\n\t\t\tdata = rawContent;\n\t\t\trawText = JSON.stringify(rawContent);\n\t\t}\n\n\t\treturn { data, rawText };\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\nexport function createWorkersAiChat(model: WorkersAiTextModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiTextAdapter(model, config);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgCA,SAAS,qBAAqB,QAAwC;AACrE,wBAAA,wBAAwB,OAAO;CAE/B,MAAM,iBAAqD,OAAO,kBAC/D,EAAE,sBAAsB,OAAO,iBAAiB,GAChD,KAAA;AAEH,KAAIA,uBAAAA,sBAAsB,OAAO,CAEhC,QAAO,IAAIC,OAAAA,QAAO;EACjB,QAAQ;EACR,OAAOC,uBAAAA,4BACN,OAAO,SACP,iBAAiB,EAAE,cAAc,gBAAgB,GAAG,KAAA,EACpD;EACD,CAAC;AAGH,KAAIC,uBAAAA,0BAA0B,OAAO,CAEpC,QAAO,IAAIF,OAAAA,QAAO;EACjB,SAAS,iDAAiD,OAAO,UAAU;EAC3E,QAAQ,OAAO;EACf,gBAAgB;EAChB,CAAC;CAIH,MAAM,gBAAgB;AACtB,QAAO,IAAIA,OAAAA,QAAO;EACjB,OAAOG,uBAAAA,mBAAmB,cAAc,eAAe,eAAe;EACtE,QAAQ,cAAc,UAAU;EAChC,CAAC;;AAOH,SAAS,mBAAmB,SAA0C;AACrE,KAAI,YAAY,KAAM,QAAO;AAC7B,KAAI,OAAO,YAAY,SAAU,QAAO;AACxC,QAAO,QAAQ,SAAS,MAAO,EAAE,SAAS,SAAS,CAAC,EAAE,QAAQ,GAAG,EAAE,CAAE,CAAC,KAAK,GAAG;;;;;;;;AAS/E,SAAS,mBAAmB,MAAsE;AACjG,SAAQ,KAAK,MAAb;EACC,KAAK;AACJ,OAAI,KAAK,QACR,QAAO;IAAE,MAAM;IAAQ,MAAM,KAAK;IAAS;AAE5C;EACD,KAAK,SAAS;GACb,IAAI;AACJ,OAAI,KAAK,OAAO,SAAS,OACxB,OAAM,KAAK,OAAO,MAAM,WAAW,QAAQ,GACxC,KAAK,OAAO,QACZ,QAAQ,KAAK,OAAO,SAAS,UAAU,KAAK,OAAO;OAEtD,OAAM,KAAK,OAAO;AAEnB,UAAO;IAAE,MAAM;IAAa,WAAW,EAAE,KAAK;IAAE;;EAEjD;AAEC,WAAQ,KACP,4DAA4D,KAAK,KAAK,cACtE;AACD;;;;;;;;;;AAWH,SAAS,iBACR,SACmD;AACnD,KAAI,YAAY,KAAM,QAAO;AAC7B,KAAI,OAAO,YAAY,SAAU,QAAO;AAGxC,KAAI,CADc,QAAQ,MAAM,MAAM,EAAE,SAAS,QAAQ,CAExD,QAAO,QAAQ,SAAS,MAAO,EAAE,SAAS,SAAS,CAAC,EAAE,QAAQ,GAAG,EAAE,CAAE,CAAC,KAAK,GAAG;CAG/E,MAAM,QAAiD,EAAE;AACzD,MAAK,MAAM,QAAQ,SAAS;EAC3B,MAAM,YAAY,mBAAmB,KAAK;AAC1C,MAAI,UACH,OAAM,KAAK,UAAU;;AAGvB,QAAO;;AAGR,SAAS,oBACR,eACA,UACA,SAC2C;CAC3C,MAAM,eAAe,SAAS,uBAAuB;CACrD,MAAM,iBAA2D,EAAE;AAEnE,KAAI,iBAAiB,cAAc,SAAS,EAC3C,gBAAe,KAAK;EACnB,MAAM;EACN,SAAS,cAAc,KAAK,KAAK;EACjC,CAAC;AAGH,MAAK,MAAM,WAAW,SACrB,KAAI,QAAQ,SAAS,OACpB,gBAAe,KAAK;EACnB,MAAM;EACN,SAAS,iBAAiB,QAAQ,QAAQ;EAC1C,CAAC;UACQ,QAAQ,SAAS,aAAa;EACxC,MAAM,mBAAoE;GACzE,MAAM;GACN,SAAS,mBAAmB,QAAQ,QAAQ;GAC5C;AACD,MAAI,gBAAgB,QAAQ,aAAa,QAAQ,UAAU,SAAS,EACnE,kBAAiB,aAAa,QAAQ,UAAU,KAAK,QAAQ;GAC5D,IAAI,GAAG;GACP,MAAM;GACN,UAAU;IACT,MAAM,GAAG,SAAS;IAClB,WAAW,GAAG,SAAS;IACvB;GACD,EAAE;AAEJ,iBAAe,KAAK,iBAAiB;YAC3B,gBAAgB,QAAQ,SAAS,QAAQ;EACnD,IAAI;AACJ,MAAI,OAAO,QAAQ,YAAY,SAC9B,KAAI;AACH,QAAK,MAAM,QAAQ,QAAQ;AAC3B,iBAAc,QAAQ;UACf;AACP,iBAAc,KAAK,UAAU,QAAQ,QAAQ;;MAG9C,eAAc,KAAK,UAAU,QAAQ,QAAQ;AAE9C,iBAAe,KAAK;GACnB,MAAM;GACN,cAAc,QAAQ,cAAc,QAAQ,OAAO,YAAY,CAAC,MAAM,GAAG,EAAE;GAC3E,SAAS;GACT,CAAC;;AAIJ,QAAO;;AAGR,SAAS,iBACR,OAC+C;AAC/C,KAAI,CAAC,MAAO,QAAO,KAAA;AACnB,QAAO,MAAM,KAAK,UAAU;EAC3B,MAAM;EACN,UAAU;GACT,MAAM,KAAK;GACX,aAAa,KAAK;GAClB,YAAY,KAAK;GACjB;EACD,EAAE;;AAOJ,SAAS,WAAW,SAAS,YAAoB;AAChD,QAAO,GAAG,OAAO,GAAG,OAAO,YAAY,CAAC,QAAQ,MAAM,GAAG,CAAC,MAAM,GAAG,GAAG;;AAUvE,IAAa,uBAAb,cAA6EC,sBAAAA,gBAM3E;CAKD,YAAY,OAAe,QAAgC;AAC1D,QAAM,EAAE,QAAQ,UAAU,EAAE,MAAM;+CALnC,QAAO,aAAsB;+CAErB,UAAA,KAAA,EAAe;AAItB,OAAK,SAAS,qBAAqB,OAAO;;CAG3C,OAAO,WAAW,SAAuD;EACxE,MAAM,EAAE,eAAe,UAAU,OAAO,aAAa,WAAW,UAAU;EAE1E,MAAM,iBAAiB,oBAAoB,eAAe,SAAS;EACnE,MAAM,cAAc,iBAAiB,MAAM;EAE3C,MAAM,YAAY,KAAK,KAAK;EAC5B,MAAM,QAAQ,YAAY;EAC1B,MAAM,YAAY,YAAY;EAC9B,IAAI,uBAAuB;EAC3B,IAAI,6BAA6B;EACjC,IAAI,qBAAqB;EACzB,IAAI,wBAAwB;EAC5B,IAAI,uBAAuB;EAC3B,MAAM,SAAS,YAAY;EAC3B,IAAI,0BAA0B;EAC9B,MAAM,sCAAsB,IAAI,KAG7B;AAEH,MAAI;GACH,IAAI;AACJ,OAAI;AACH,aAAS,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;KAClD,OAAO,SAAS,KAAK;KACrB,UAAU;KACV,OAAO;KACP;KACA,YAAY;KACZ,QAAQ;KACR,gBAAgB,EAAE,eAAe,MAAM;KACvC,CAAC;YACM,aAAsB;AAG9B,YAAQ,KACP,kEACA,uBAAuB,QAAQ,YAAY,UAAU,YACrD;IACD,MAAM,kBAAkB,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;KACjE,OAAO,SAAS,KAAK;KACrB,UAAU;KACV,OAAO;KACP;KACA,YAAY;KACZ,CAAC;AAEF,UAAM;KACL,MAAM;KACN;KACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;KAC9C;KACA;IAED,MAAM,MAAM,gBAAgB,QAAQ,IAAI;AACxC,QAAI,KAAK,SAAS;AACjB,WAAM;MACL,MAAM;MACN;MACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;MAC9C;MACA,MAAM;MACN;AACD,WAAM;MACL,MAAM;MACN;MACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;MAC9C;MACA,OAAO,IAAI;MACX,SAAS,IAAI;MACb;AACD,WAAM;MACL,MAAM;MACN;MACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;MAC9C;MACA;;AAGF,QAAI,KAAK,WACR,MAAK,MAAM,MAAM,IAAI,YAAY;AAChC,SAAI,GAAG,SAAS,WAAY;KAC5B,MAAM,KAAK,GAAG;KACd,IAAI,cAAuB,EAAE;AAC7B,SAAI;AACH,oBAAc,GAAG,YAAY,KAAK,MAAM,GAAG,UAAU,GAAG,EAAE;aACnD;AACP,oBAAc,EAAE;;AAEjB,WAAM;MACL,MAAM;MACN,YAAY,GAAG;MACf,UAAU,GAAG;MACb,OAAO,gBAAgB,SAAS,SAAS,KAAK;MAC9C;MACA,OAAO;MACP;AACD,WAAM;MACL,MAAM;MACN,YAAY,GAAG;MACf,UAAU,GAAG;MACb,OAAO,gBAAgB,SAAS,SAAS,KAAK;MAC9C;MACA,OAAO;MACP;;IAIH,MAAM,eAAe,gBAAgB,QAAQ,IAAI;AACjD,UAAM;KACL,MAAM;KACN;KACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;KAC9C;KACA,OAAO,gBAAgB,QACpB;MACA,cAAc,gBAAgB,MAAM;MACpC,kBAAkB,gBAAgB,MAAM;MACxC,aAAa,gBAAgB,MAAM;MACnC,GACA,KAAA;KACH,cACC,iBAAiB,gBAAgB,iBAAiB,kBAC/C,eACE,gBAAyD;KAC/D;AACD;;AAGD,cAAW,MAAM,SAAS,QAAQ;AACjC,QAAI,CAAC,MAAM,WAAW,MAAM,QAAQ,WAAW,EAAG;IAClD,MAAM,SAAS,MAAM,QAAQ;AAC7B,QAAI,CAAC,OAAQ;AAGb,QAAI,CAAC,sBAAsB;AAC1B,4BAAuB;AACvB,WAAM;MACL,MAAM;MACN;MACA,OAAO,MAAM,SAAS,SAAS,KAAK;MACpC;MACA;;IAGF,MAAM,QAAQ,OAAO;IAIrB,MAAM,mBAAqB,MAAkC,qBAC3D,MAAkC;AACpC,QAAI,kBAAkB;AAErB,SAAI,CAAC,uBAAuB;AAC3B,8BAAwB;AACxB,YAAM;OACL,MAAM;OACN;OACA,UAAU;OACV,OAAO,MAAM,SAAS,SAAS,KAAK;OACpC;OACA;;AAEF,6BAAwB;AAOxB,WAAM;MACL,MAAM;MACN;MACA,OAAO;MACP,SAAS;MACT,OAAO,MAAM,SAAS,SAAS,KAAK;MACpC;MACA;;AAIF,QAAI,MAAM,SAAS;AAClB,SAAI,CAAC,4BAA4B;AAChC,mCAA6B;AAC7B,YAAM;OACL,MAAM;OACN;OACA,OAAO,MAAM,SAAS,SAAS,KAAK;OACpC;OACA,MAAM;OACN;;AAGF,2BAAsB,MAAM;AAC5B,WAAM;MACL,MAAM;MACN;MACA,OAAO,MAAM,SAAS,SAAS,KAAK;MACpC;MACA,OAAO,MAAM;MACb,SAAS;MACT;;AAIF,QAAI,MAAM,WACT,MAAK,MAAM,iBAAiB,MAAM,YAAY;KAC7C,MAAM,QAAQ,cAAc;AAE5B,SAAI,CAAC,oBAAoB,IAAI,MAAM,CAIlC,qBAAoB,IAAI,OAAO;MAC9B,IAAI,WAAW,gBAAgB;MAC/B,MAAM,cAAc,UAAU,QAAQ;MACtC,WAAW;MACX,SAAS;MACT,CAAC;KAGH,MAAM,WAAW,oBAAoB,IAAI,MAAM;AAK/C,SAAI,cAAc,UAAU,KAC3B,UAAS,OAAO,cAAc,SAAS;AAExC,SAAI,cAAc,UAAU,UAC3B,UAAS,aAAa,cAAc,SAAS;AAI9C,SAAI,SAAS,MAAM,SAAS,QAAQ,CAAC,SAAS,SAAS;AACtD,eAAS,UAAU;AACnB,YAAM;OACL,MAAM;OACN,YAAY,SAAS;OACrB,UAAU,SAAS;OACnB,OAAO,MAAM,SAAS,SAAS,KAAK;OACpC;OACA;OACA;;AAIF,SAAI,cAAc,UAAU,aAAa,SAAS,QACjD,OAAM;MACL,MAAM;MACN,YAAY,SAAS;MACrB,OAAO,MAAM,SAAS,SAAS,KAAK;MACpC;MACA,OAAO,cAAc,SAAS;MAC9B;;AAMJ,QAAI,OAAO,eAAe;AACzB,+BAA0B;AAG1B,SAAI,OAAO,kBAAkB,gBAAgB,oBAAoB,OAAO,EACvE,MAAK,MAAM,GAAG,aAAa,qBAAqB;MAC/C,IAAI,cAAuB,EAAE;AAC7B,UAAI;AACH,qBAAc,SAAS,YACpB,KAAK,MAAM,SAAS,UAAU,GAC9B,EAAE;cACE;AACP,qBAAc,EAAE;;AAEjB,YAAM;OACL,MAAM;OACN,YAAY,SAAS;OACrB,UAAU,SAAS;OACnB,OAAO,MAAM,SAAS,SAAS,KAAK;OACpC;OACA,OAAO;OACP;;KAIH,MAAM,uBACL,OAAO,kBAAkB,gBACzB,OAAO,kBAAkB,mBACzB,oBAAoB,OAAO,IACxB,eACC,OAAO;AAGZ,SAAI,2BACH,OAAM;MACL,MAAM;MACN;MACA,OAAO,MAAM,SAAS,SAAS,KAAK;MACpC;MACA;AAIF,WAAM;MACL,MAAM;MACN;MACA,OAAO,MAAM,SAAS,SAAS,KAAK;MACpC;MACA,OAAO,MAAM,QACV;OACA,cAAc,MAAM,MAAM;OAC1B,kBAAkB,MAAM,MAAM;OAC9B,aAAa,MAAM,MAAM;OACzB,GACA,KAAA;MACH,cAAc;MACd;;;AAOH,OAAI,wBAAwB,CAAC,yBAAyB;AACrD,YAAQ,KACP,4FACA;AAGD,SAAK,MAAM,GAAG,aAAa,oBAC1B,KAAI,SAAS,SAAS;KACrB,IAAI,cAAuB,EAAE;AAC7B,SAAI;AACH,oBAAc,SAAS,YAAY,KAAK,MAAM,SAAS,UAAU,GAAG,EAAE;aAC/D;AACP,oBAAc,EAAE;;AAEjB,WAAM;MACL,MAAM;MACN,YAAY,SAAS;MACrB,UAAU,SAAS;MACnB,OAAO,SAAS,KAAK;MACrB;MACA,OAAO;MACP;;AAKH,QAAI,2BACH,OAAM;KACL,MAAM;KACN;KACA,OAAO,SAAS,KAAK;KACrB;KACA;AAGF,UAAM;KACL,MAAM;KACN;KACA,OAAO,SAAS,KAAK;KACrB;KACA,cAAc;KACd;;WAEM,OAAO;GACf,MAAM,UAAU,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;GACtE,MAAM,OACL,iBAAiB,QAAS,MAAoC,OAAO,KAAA;AACtE,OAAI,CAAC,qBACJ,OAAM;IACL,MAAM;IACN;IACA,OAAO,SAAS,KAAK;IACrB;IACA;AAEF,SAAM;IACL,MAAM;IACN;IACA,OAAO,SAAS,KAAK;IACrB;IACA,OAAO;KACN,SAAS,WAAW;KACpB;KACA;IACD;;;CAIH,MAAM,iBACL,SAC2C;EAC3C,MAAM,EAAE,cAAc,gBAAgB;EACtC,MAAM,EAAE,eAAe,UAAU,aAAa,UAAU;EAExD,MAAM,iBAAiB,oBAAoB,eAAe,UAAU,EACnE,qBAAqB,OACrB,CAAC;EAEF,MAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;GAC1D,OAAO,SAAS,KAAK;GACrB,UAAU;GACV;GACA,QAAQ;GACR,iBAAiB;IAChB,MAAM;IACN,aAAa;KACZ,MAAM;KACN,QAAQ;KACR,QAAQ;KACR;IACD;GACD,CAAC;EAEF,MAAM,SAAS,SAAS,UAAU;AAElC,MAAI,CAAC,OACJ,OAAM,IAAI,MACT,qDAAqD,KAAK,UAAU,SAAS,GAC7E;EAGF,MAAM,aAAa,OAAO,SAAS,WAAW;EAI9C,IAAI;EACJ,IAAI;AAEJ,MAAI,OAAO,eAAe,UAAU;AACnC,aAAU;AACV,OAAI;AACH,WAAO,KAAK,MAAM,QAAQ;WACnB;AACP,WAAO;;SAEF;AAEN,UAAO;AACP,aAAU,KAAK,UAAU,WAAW;;AAGrC,SAAO;GAAE;GAAM;GAAS;;;AAQ1B,SAAgB,oBAAoB,OAA2B,QAAgC;AAC9F,QAAO,IAAI,qBAAqB,OAAO,OAAO"}
@@ -0,0 +1,85 @@
1
+ //#region src/utils/workers-ai-rest.ts
2
+ /**
3
+ * Workers AI REST API base URL.
4
+ * All model endpoints follow the pattern: `${BASE_URL}/${accountId}/ai/run/${model}`
5
+ */
6
+ const WORKERS_AI_REST_BASE = "https://api.cloudflare.com/client/v4/accounts";
7
+ /**
8
+ * Make a REST API call to Workers AI.
9
+ *
10
+ * Handles the common pattern shared by all Workers AI adapters:
11
+ * - Build the URL from account ID and model name
12
+ * - Set Authorization and Content-Type headers
13
+ * - Check response.ok and throw a descriptive error on failure
14
+ *
15
+ * @param config Credentials config with accountId and apiKey
16
+ * @param model Workers AI model name (e.g. "@cf/stabilityai/stable-diffusion-xl-base-1.0")
17
+ * @param body JSON request body
18
+ * @param options Optional settings:
19
+ * - `label` — human-readable label for error messages (default: "Workers AI")
20
+ * - `signal` — AbortSignal for request cancellation / timeout
21
+ * @returns The raw Response object — caller is responsible for parsing
22
+ */
23
+ async function workersAiRestFetch(config, model, body, options) {
24
+ const response = await fetch(`${WORKERS_AI_REST_BASE}/${config.accountId}/ai/run/${model}`, {
25
+ method: "POST",
26
+ headers: {
27
+ Authorization: `Bearer ${config.apiKey}`,
28
+ "Content-Type": "application/json"
29
+ },
30
+ body: JSON.stringify(body),
31
+ signal: options?.signal
32
+ });
33
+ if (!response.ok) {
34
+ const errorText = await response.text();
35
+ const label = options?.label ?? "Workers AI";
36
+ throw new Error(`${label} request failed (${response.status}): ${errorText}`);
37
+ }
38
+ return response;
39
+ }
40
+ /**
41
+ * Make a binary REST API call to Workers AI.
42
+ *
43
+ * Some models (e.g. `@cf/deepgram/nova-3`) require raw binary audio with an
44
+ * appropriate `Content-Type` header instead of JSON. This function sends the
45
+ * audio bytes directly as the request body.
46
+ *
47
+ * @param config Credentials config with accountId and apiKey
48
+ * @param model Workers AI model name
49
+ * @param audioBytes Raw audio bytes
50
+ * @param contentType MIME type of the audio (e.g. "audio/wav")
51
+ * @param options Optional settings
52
+ * @returns The raw Response object
53
+ */
54
+ async function workersAiRestFetchBinary(config, model, audioBytes, contentType, options) {
55
+ const response = await fetch(`${WORKERS_AI_REST_BASE}/${config.accountId}/ai/run/${model}`, {
56
+ method: "POST",
57
+ headers: {
58
+ Authorization: `Bearer ${config.apiKey}`,
59
+ "Content-Type": contentType
60
+ },
61
+ body: audioBytes,
62
+ signal: options?.signal
63
+ });
64
+ if (!response.ok) {
65
+ const errorText = await response.text();
66
+ const label = options?.label ?? "Workers AI";
67
+ throw new Error(`${label} request failed (${response.status}): ${errorText}`);
68
+ }
69
+ return response;
70
+ }
71
+ //#endregion
72
+ Object.defineProperty(exports, "workersAiRestFetch", {
73
+ enumerable: true,
74
+ get: function() {
75
+ return workersAiRestFetch;
76
+ }
77
+ });
78
+ Object.defineProperty(exports, "workersAiRestFetchBinary", {
79
+ enumerable: true,
80
+ get: function() {
81
+ return workersAiRestFetchBinary;
82
+ }
83
+ });
84
+
85
+ //# sourceMappingURL=workers-ai-rest-CkNCtBwv.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"workers-ai-rest-CkNCtBwv.cjs","names":[],"sources":["../src/utils/workers-ai-rest.ts"],"sourcesContent":["import type { WorkersAiDirectCredentialsConfig } from \"./create-fetcher\";\n\n/**\n * Workers AI REST API base URL.\n * All model endpoints follow the pattern: `${BASE_URL}/${accountId}/ai/run/${model}`\n */\nconst WORKERS_AI_REST_BASE = \"https://api.cloudflare.com/client/v4/accounts\";\n\n/**\n * Make a REST API call to Workers AI.\n *\n * Handles the common pattern shared by all Workers AI adapters:\n * - Build the URL from account ID and model name\n * - Set Authorization and Content-Type headers\n * - Check response.ok and throw a descriptive error on failure\n *\n * @param config Credentials config with accountId and apiKey\n * @param model Workers AI model name (e.g. \"@cf/stabilityai/stable-diffusion-xl-base-1.0\")\n * @param body JSON request body\n * @param options Optional settings:\n * - `label` — human-readable label for error messages (default: \"Workers AI\")\n * - `signal` — AbortSignal for request cancellation / timeout\n * @returns The raw Response object — caller is responsible for parsing\n */\nexport async function workersAiRestFetch(\n\tconfig: WorkersAiDirectCredentialsConfig,\n\tmodel: string,\n\tbody: Record<string, unknown>,\n\toptions?: { label?: string; signal?: AbortSignal },\n): Promise<Response> {\n\tconst response = await fetch(`${WORKERS_AI_REST_BASE}/${config.accountId}/ai/run/${model}`, {\n\t\tmethod: \"POST\",\n\t\theaders: {\n\t\t\tAuthorization: `Bearer ${config.apiKey}`,\n\t\t\t\"Content-Type\": \"application/json\",\n\t\t},\n\t\tbody: JSON.stringify(body),\n\t\tsignal: options?.signal,\n\t});\n\n\tif (!response.ok) {\n\t\tconst errorText = await response.text();\n\t\tconst label = options?.label ?? \"Workers AI\";\n\t\tthrow new Error(`${label} request failed (${response.status}): ${errorText}`);\n\t}\n\n\treturn response;\n}\n\n/**\n * Make a binary REST API call to Workers AI.\n *\n * Some models (e.g. `@cf/deepgram/nova-3`) require raw binary audio with an\n * appropriate `Content-Type` header instead of JSON. This function sends the\n * audio bytes directly as the request body.\n *\n * @param config Credentials config with accountId and apiKey\n * @param model Workers AI model name\n * @param audioBytes Raw audio bytes\n * @param contentType MIME type of the audio (e.g. \"audio/wav\")\n * @param options Optional settings\n * @returns The raw Response object\n */\nexport async function workersAiRestFetchBinary(\n\tconfig: WorkersAiDirectCredentialsConfig,\n\tmodel: string,\n\taudioBytes: Uint8Array,\n\tcontentType: string,\n\toptions?: { label?: string; signal?: AbortSignal },\n): Promise<Response> {\n\tconst response = await fetch(`${WORKERS_AI_REST_BASE}/${config.accountId}/ai/run/${model}`, {\n\t\tmethod: \"POST\",\n\t\theaders: {\n\t\t\tAuthorization: `Bearer ${config.apiKey}`,\n\t\t\t\"Content-Type\": contentType,\n\t\t},\n\t\tbody: audioBytes,\n\t\tsignal: options?.signal,\n\t});\n\n\tif (!response.ok) {\n\t\tconst errorText = await response.text();\n\t\tconst label = options?.label ?? \"Workers AI\";\n\t\tthrow new Error(`${label} request failed (${response.status}): ${errorText}`);\n\t}\n\n\treturn response;\n}\n"],"mappings":";;;;;AAMA,MAAM,uBAAuB;;;;;;;;;;;;;;;;;AAkB7B,eAAsB,mBACrB,QACA,OACA,MACA,SACoB;CACpB,MAAM,WAAW,MAAM,MAAM,GAAG,qBAAqB,GAAG,OAAO,UAAU,UAAU,SAAS;EAC3F,QAAQ;EACR,SAAS;GACR,eAAe,UAAU,OAAO;GAChC,gBAAgB;GAChB;EACD,MAAM,KAAK,UAAU,KAAK;EAC1B,QAAQ,SAAS;EACjB,CAAC;AAEF,KAAI,CAAC,SAAS,IAAI;EACjB,MAAM,YAAY,MAAM,SAAS,MAAM;EACvC,MAAM,QAAQ,SAAS,SAAS;AAChC,QAAM,IAAI,MAAM,GAAG,MAAM,mBAAmB,SAAS,OAAO,KAAK,YAAY;;AAG9E,QAAO;;;;;;;;;;;;;;;;AAiBR,eAAsB,yBACrB,QACA,OACA,YACA,aACA,SACoB;CACpB,MAAM,WAAW,MAAM,MAAM,GAAG,qBAAqB,GAAG,OAAO,UAAU,UAAU,SAAS;EAC3F,QAAQ;EACR,SAAS;GACR,eAAe,UAAU,OAAO;GAChC,gBAAgB;GAChB;EACD,MAAM;EACN,QAAQ,SAAS;EACjB,CAAC;AAEF,KAAI,CAAC,SAAS,IAAI;EACjB,MAAM,YAAY,MAAM,SAAS,MAAM;EACvC,MAAM,QAAQ,SAAS,SAAS;AAChC,QAAM,IAAI,MAAM,GAAG,MAAM,mBAAmB,SAAS,OAAO,KAAK,YAAY;;AAG9E,QAAO"}