@cloudflare/tanstack-ai 0.1.2 → 0.1.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +9 -0
- package/dist/_tsup-dts-rollup.d.cts +13 -4
- package/dist/_tsup-dts-rollup.d.ts +13 -4
- package/dist/adapters/anthropic.cjs +3 -3
- package/dist/adapters/anthropic.js +2 -2
- package/dist/adapters/grok.cjs +3 -3
- package/dist/adapters/grok.js +2 -2
- package/dist/adapters/openai.cjs +3 -3
- package/dist/adapters/openai.js +2 -2
- package/dist/adapters/openrouter.cjs +3 -3
- package/dist/adapters/openrouter.js +2 -2
- package/dist/adapters/workers-ai-image.cjs +3 -3
- package/dist/adapters/workers-ai-image.js +2 -2
- package/dist/adapters/workers-ai-summarize.cjs +3 -3
- package/dist/adapters/workers-ai-summarize.js +2 -2
- package/dist/adapters/workers-ai-transcription.cjs +3 -3
- package/dist/adapters/workers-ai-transcription.js +2 -2
- package/dist/adapters/workers-ai-tts.cjs +3 -3
- package/dist/adapters/workers-ai-tts.js +2 -2
- package/dist/adapters/workers-ai.cjs +3 -3
- package/dist/adapters/workers-ai.js +2 -2
- package/dist/{chunk-AYTKQRE4.cjs → chunk-45DEJBFY.cjs} +6 -6
- package/dist/{chunk-AYTKQRE4.cjs.map → chunk-45DEJBFY.cjs.map} +1 -1
- package/dist/{chunk-JK5ZFOZZ.js → chunk-4HOZDNW6.js} +2 -2
- package/dist/{chunk-IMURYJRF.cjs → chunk-53RO5I22.cjs} +33 -36
- package/dist/chunk-53RO5I22.cjs.map +1 -0
- package/dist/{chunk-T5KPTTLK.js → chunk-627RVSSB.js} +2 -2
- package/dist/{chunk-O7MBRPCW.js → chunk-727MRLUF.js} +26 -29
- package/dist/chunk-727MRLUF.js.map +1 -0
- package/dist/{chunk-7KZQWFUE.cjs → chunk-7YINQQCY.cjs} +3 -3
- package/dist/chunk-7YINQQCY.cjs.map +1 -0
- package/dist/{chunk-DSBHOQQ4.js → chunk-AJMCHPPR.js} +2 -2
- package/dist/chunk-AJMCHPPR.js.map +1 -0
- package/dist/{chunk-ARLHMZ5B.cjs → chunk-CPOXZIMA.cjs} +6 -6
- package/dist/{chunk-ARLHMZ5B.cjs.map → chunk-CPOXZIMA.cjs.map} +1 -1
- package/dist/{chunk-7SKFZ4SU.cjs → chunk-DQGO754X.cjs} +5 -5
- package/dist/{chunk-7SKFZ4SU.cjs.map → chunk-DQGO754X.cjs.map} +1 -1
- package/dist/{chunk-LG2I3ANT.cjs → chunk-DTGQYEJ2.cjs} +49 -33
- package/dist/chunk-DTGQYEJ2.cjs.map +1 -0
- package/dist/{chunk-2QN5XFI2.js → chunk-GST6AT3R.js} +2 -2
- package/dist/{chunk-3KUHT3K6.js → chunk-HDDZEGLC.js} +46 -30
- package/dist/chunk-HDDZEGLC.js.map +1 -0
- package/dist/{chunk-Y7W7NDVS.js → chunk-HFZK4C5B.js} +2 -2
- package/dist/{chunk-ZHTFBZXJ.cjs → chunk-IL6PTKJE.cjs} +6 -6
- package/dist/{chunk-ZHTFBZXJ.cjs.map → chunk-IL6PTKJE.cjs.map} +1 -1
- package/dist/{chunk-ICHJ2G63.cjs → chunk-MPVUETJY.cjs} +3 -3
- package/dist/{chunk-ICHJ2G63.cjs.map → chunk-MPVUETJY.cjs.map} +1 -1
- package/dist/{chunk-OKU3GB5P.js → chunk-PMPGIUVA.js} +2 -2
- package/dist/{chunk-UWBO33IG.cjs → chunk-QUS7INYA.cjs} +5 -5
- package/dist/{chunk-UWBO33IG.cjs.map → chunk-QUS7INYA.cjs.map} +1 -1
- package/dist/{chunk-5A2F2QCL.js → chunk-RKBJLYVO.js} +2 -2
- package/dist/{chunk-DZZVI4W5.js → chunk-XCLZJGUI.js} +2 -2
- package/dist/{chunk-IDPLTBQX.cjs → chunk-Y2PUBTP5.cjs} +6 -6
- package/dist/{chunk-IDPLTBQX.cjs.map → chunk-Y2PUBTP5.cjs.map} +1 -1
- package/dist/index.cjs +11 -11
- package/dist/index.js +10 -10
- package/package.json +11 -10
- package/dist/chunk-3KUHT3K6.js.map +0 -1
- package/dist/chunk-7KZQWFUE.cjs.map +0 -1
- package/dist/chunk-DSBHOQQ4.js.map +0 -1
- package/dist/chunk-IMURYJRF.cjs.map +0 -1
- package/dist/chunk-LG2I3ANT.cjs.map +0 -1
- package/dist/chunk-O7MBRPCW.js.map +0 -1
- /package/dist/{chunk-JK5ZFOZZ.js.map → chunk-4HOZDNW6.js.map} +0 -0
- /package/dist/{chunk-T5KPTTLK.js.map → chunk-627RVSSB.js.map} +0 -0
- /package/dist/{chunk-2QN5XFI2.js.map → chunk-GST6AT3R.js.map} +0 -0
- /package/dist/{chunk-Y7W7NDVS.js.map → chunk-HFZK4C5B.js.map} +0 -0
- /package/dist/{chunk-OKU3GB5P.js.map → chunk-PMPGIUVA.js.map} +0 -0
- /package/dist/{chunk-5A2F2QCL.js.map → chunk-RKBJLYVO.js.map} +0 -0
- /package/dist/{chunk-DZZVI4W5.js.map → chunk-XCLZJGUI.js.map} +0 -0
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-DQGO754X.cjs","../src/adapters/openrouter.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA;AACC;AACA;AACA;AAAA,uDAIM;AACP,sCAA2B;AAY3B,SAAS,qBAAA,CAAsB,MAAA,EAAmD;AACjF,EAAA,MAAM,WAAA,EAAa,IAAI,oBAAA,CAAW;AAAA,IACjC,OAAA,EAAS,kDAAA,YAAmB,EAAc,MAAM;AAAA,EACjD,CAAC,CAAA;AACD,EAAA,OAAO;AAAA,IACN,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA;AAAA;AAAA;AAAA,IAIzB;AAAA,EACD,CAAA;AACD;AAWA,SAAS,0BAAA,CAA2B,MAAA,EAAwD;AAC3F,EAAA,MAAM,WAAA,EAAa,IAAI,oBAAA,CAAW;AAAA,IACjC,OAAA,EAAS,kDAAA,YAAmB,EAAc,MAAM;AAAA,EACjD,CAAC,CAAA;AACD,EAAA,OAAO;AAAA,IACN,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,IACzB;AAAA,EACD,CAAA;AACD;AASA,SAAS,8BAAA,CACR,MAAA,EAC4B;AAC5B,EAAA,MAAM,WAAA,EAAa,IAAI,oBAAA,CAAW;AAAA,IACjC,OAAA,EAAS,kDAAA,YAAmB,EAAc,MAAM;AAAA,EACjD,CAAC,CAAA;AACD,EAAA,OAAO;AAAA,IACN,MAAA,mBAAQ,MAAA,CAAO,MAAA,UAAU,UAAA;AAAA,IACzB;AAAA,EACD,CAAA;AACD;AA0BO,SAAS,oBAAA,CACf,KAAA,EACA,MAAA,EACiB;AAEjB,EAAA,OAAO,IAAI,wCAAA,CAAsB,qBAAA,CAAsB,MAAM,CAAA,EAAG,KAAY,CAAA;AAC7E;AASO,SAAS,qBAAA,CACf,KAAA,EACA,MAAA,EACC;AAED,EAAA,OAAO,IAAI,yCAAA,CAAuB,0BAAA,CAA2B,MAAM,CAAA,EAAG,KAAY,CAAA;AACnF;AASO,SAAS,yBAAA,CACf,KAAA,EACA,MAAA,EACC;AAED,EAAA,OAAO,IAAI,6CAAA,CAA2B,8BAAA,CAA+B,MAAM,CAAA,EAAG,KAAY,CAAA;AAC3F;ADjFA;AACA;AACE;AACA;AACA;AACF,0KAAC","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-DQGO754X.cjs","sourcesContent":[null,"import {\n\tOpenRouterTextAdapter,\n\tOpenRouterImageAdapter,\n\tOpenRouterSummarizeAdapter,\n\ttype OpenRouterConfig,\n\ttype OpenRouterImageConfig,\n\ttype OpenRouterSummarizeConfig,\n} from \"@tanstack/ai-openrouter\";\nimport { HTTPClient } from \"@openrouter/sdk\";\nimport { createGatewayFetch, type AiGatewayAdapterConfig } from \"../utils/create-fetcher\";\nimport type { AnyTextAdapter } from \"@tanstack/ai\";\n\nexport type OpenRouterGatewayConfig = AiGatewayAdapterConfig;\n\n/**\n * Build OpenRouter config that routes requests through AI Gateway.\n *\n * The OpenRouter SDK accepts an `httpClient` with a custom `fetcher`,\n * which we use to inject the AI Gateway fetch.\n */\nfunction buildOpenRouterConfig(config: OpenRouterGatewayConfig): OpenRouterConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\t// Cast needed: the installed @openrouter/sdk version may differ from the\n\t\t// version @tanstack/ai-openrouter was built against. The HTTPClient interface\n\t\t// is structurally compatible but TypeScript sees them as separate declarations.\n\t\thttpClient: httpClient as unknown as OpenRouterConfig[\"httpClient\"],\n\t};\n}\n\n/**\n * Build OpenRouter image config that routes requests through AI Gateway.\n *\n * `OpenRouterImageConfig` extends `OpenRouterClientConfig` which declares\n * `baseURL` and `apiKey` but not `httpClient`. However, the image adapter\n * internally creates an `OpenRouter` SDK instance (which does accept\n * `httpClient` via `SDKOptions`). The double-cast is needed because the\n * declared config type is narrower than what the SDK constructor accepts.\n */\nfunction buildOpenRouterImageConfig(config: OpenRouterGatewayConfig): OpenRouterImageConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t} as unknown as OpenRouterImageConfig;\n}\n\n/**\n * Build OpenRouter summarize config.\n *\n * `OpenRouterSummarizeConfig` extends `OpenRouterConfig` with optional\n * `temperature` and `maxTokens` fields. Since those are optional, a plain\n * `OpenRouterConfig` is structurally compatible — no cast needed.\n */\nfunction buildOpenRouterSummarizeConfig(\n\tconfig: OpenRouterGatewayConfig,\n): OpenRouterSummarizeConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient: httpClient as unknown as OpenRouterSummarizeConfig[\"httpClient\"],\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// Chat model type — OpenRouter supports many models; we use a loose string type\n// ---------------------------------------------------------------------------\n\n/** OpenRouter chat model identifier. Accepts any string since OpenRouter proxies hundreds of models. */\nexport type OpenRouterChatModel = string;\n\n/** OpenRouter image model identifier. */\nexport type OpenRouterImageModel = string;\n\n/** OpenRouter summarize model identifier (same as chat models). */\nexport type OpenRouterSummarizeModel = string;\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\n/**\n * Creates an OpenRouter chat adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use (e.g. \"openai/gpt-4o\", \"anthropic/claude-sonnet-4-5\")\n * @param config Configuration options\n */\nexport function createOpenRouterChat(\n\tmodel: OpenRouterChatModel,\n\tconfig: OpenRouterGatewayConfig,\n): AnyTextAdapter {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterTextAdapter(buildOpenRouterConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter image adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The image model to use\n * @param config Configuration options\n */\nexport function createOpenRouterImage(\n\tmodel: OpenRouterImageModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterImageAdapter(buildOpenRouterImageConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter summarize adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use for summarization\n * @param config Configuration options\n */\nexport function createOpenRouterSummarize(\n\tmodel: OpenRouterSummarizeModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterSummarizeAdapter(buildOpenRouterSummarizeConfig(config), model as any);\n}\n"]}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
|
|
5
5
|
|
|
6
6
|
|
|
7
|
-
var
|
|
7
|
+
var _chunk53RO5I22cjs = require('./chunk-53RO5I22.cjs');
|
|
8
8
|
|
|
9
9
|
|
|
10
10
|
var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
|
|
@@ -15,51 +15,70 @@ var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
|
|
|
15
15
|
var _adapters = require('@tanstack/ai/adapters');
|
|
16
16
|
var _openai = require('openai'); var _openai2 = _interopRequireDefault(_openai);
|
|
17
17
|
function buildWorkersAiClient(config) {
|
|
18
|
-
|
|
19
|
-
|
|
18
|
+
_chunk53RO5I22cjs.validateWorkersAiConfig.call(void 0, config);
|
|
19
|
+
const sessionHeaders = config.sessionAffinity ? { "x-session-affinity": config.sessionAffinity } : void 0;
|
|
20
|
+
if (_chunk53RO5I22cjs.isDirectBindingConfig.call(void 0, config)) {
|
|
20
21
|
return new (0, _openai2.default)({
|
|
21
22
|
apiKey: "unused",
|
|
22
|
-
fetch:
|
|
23
|
+
fetch: _chunk53RO5I22cjs.createWorkersAiBindingFetch.call(void 0,
|
|
24
|
+
config.binding,
|
|
25
|
+
sessionHeaders ? { extraHeaders: sessionHeaders } : void 0
|
|
26
|
+
)
|
|
23
27
|
});
|
|
24
28
|
}
|
|
25
|
-
if (
|
|
29
|
+
if (_chunk53RO5I22cjs.isDirectCredentialsConfig.call(void 0, config)) {
|
|
26
30
|
return new (0, _openai2.default)({
|
|
27
31
|
baseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,
|
|
28
|
-
apiKey: config.apiKey
|
|
32
|
+
apiKey: config.apiKey,
|
|
33
|
+
defaultHeaders: sessionHeaders
|
|
29
34
|
});
|
|
30
35
|
}
|
|
31
36
|
const gatewayConfig = config;
|
|
32
37
|
return new (0, _openai2.default)({
|
|
33
|
-
fetch:
|
|
38
|
+
fetch: _chunk53RO5I22cjs.createGatewayFetch.call(void 0, "workers-ai", gatewayConfig, sessionHeaders),
|
|
34
39
|
apiKey: _nullishCoalesce(gatewayConfig.apiKey, () => ( "unused"))
|
|
35
40
|
});
|
|
36
41
|
}
|
|
37
42
|
function extractTextContent(content) {
|
|
38
43
|
if (content === null) return "";
|
|
39
44
|
if (typeof content === "string") return content;
|
|
40
|
-
return content.
|
|
45
|
+
return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
|
|
46
|
+
}
|
|
47
|
+
function convertContentPart(part) {
|
|
48
|
+
switch (part.type) {
|
|
49
|
+
case "text":
|
|
50
|
+
if (part.content) {
|
|
51
|
+
return { type: "text", text: part.content };
|
|
52
|
+
}
|
|
53
|
+
return void 0;
|
|
54
|
+
case "image": {
|
|
55
|
+
let url;
|
|
56
|
+
if (part.source.type === "data") {
|
|
57
|
+
url = part.source.value.startsWith("data:") ? part.source.value : `data:${part.source.mimeType};base64,${part.source.value}`;
|
|
58
|
+
} else {
|
|
59
|
+
url = part.source.value;
|
|
60
|
+
}
|
|
61
|
+
return { type: "image_url", image_url: { url } };
|
|
62
|
+
}
|
|
63
|
+
default:
|
|
64
|
+
console.warn(
|
|
65
|
+
`[@cloudflare/tanstack-ai] Unsupported content part type "${part.type}" \u2014 skipping`
|
|
66
|
+
);
|
|
67
|
+
return void 0;
|
|
68
|
+
}
|
|
41
69
|
}
|
|
42
70
|
function buildUserContent(content) {
|
|
43
71
|
if (content === null) return "";
|
|
44
72
|
if (typeof content === "string") return content;
|
|
45
|
-
const hasImages = content.some((p) => p.type === "
|
|
73
|
+
const hasImages = content.some((p) => p.type === "image");
|
|
46
74
|
if (!hasImages) {
|
|
47
|
-
return content.
|
|
75
|
+
return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
|
|
48
76
|
}
|
|
49
77
|
const parts = [];
|
|
50
78
|
for (const part of content) {
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
parts.push({
|
|
55
|
-
type: "image_url",
|
|
56
|
-
image_url: { url: part.content }
|
|
57
|
-
});
|
|
58
|
-
} else if (part.type === "image_url" && part.image_url) {
|
|
59
|
-
parts.push({
|
|
60
|
-
type: "image_url",
|
|
61
|
-
image_url: part.image_url
|
|
62
|
-
});
|
|
79
|
+
const converted = convertContentPart(part);
|
|
80
|
+
if (converted) {
|
|
81
|
+
parts.push(converted);
|
|
63
82
|
}
|
|
64
83
|
}
|
|
65
84
|
return parts;
|
|
@@ -127,8 +146,8 @@ function buildOpenAITools(tools) {
|
|
|
127
146
|
}
|
|
128
147
|
}));
|
|
129
148
|
}
|
|
130
|
-
function generateId(prefix) {
|
|
131
|
-
return `${prefix}-${crypto.randomUUID()}`;
|
|
149
|
+
function generateId(prefix = "chatcmpl") {
|
|
150
|
+
return `${prefix}-${crypto.randomUUID().replace(/-/g, "").slice(0, 16)}`;
|
|
132
151
|
}
|
|
133
152
|
var WorkersAiTextAdapter = class extends _adapters.BaseTextAdapter {
|
|
134
153
|
constructor(model, config) {
|
|
@@ -142,14 +161,14 @@ var WorkersAiTextAdapter = class extends _adapters.BaseTextAdapter {
|
|
|
142
161
|
const openAIMessages = buildOpenAIMessages(systemPrompts, messages);
|
|
143
162
|
const openAITools = buildOpenAITools(tools);
|
|
144
163
|
const timestamp = Date.now();
|
|
145
|
-
const runId = generateId(
|
|
146
|
-
const messageId = generateId(
|
|
164
|
+
const runId = generateId();
|
|
165
|
+
const messageId = generateId();
|
|
147
166
|
let hasEmittedRunStarted = false;
|
|
148
167
|
let hasEmittedTextMessageStart = false;
|
|
149
168
|
let accumulatedContent = "";
|
|
150
169
|
let hasEmittedStepStarted = false;
|
|
151
170
|
let accumulatedReasoning = "";
|
|
152
|
-
const stepId = generateId(
|
|
171
|
+
const stepId = generateId();
|
|
153
172
|
let hasReceivedFinishReason = false;
|
|
154
173
|
const toolCallsInProgress = /* @__PURE__ */ new Map();
|
|
155
174
|
try {
|
|
@@ -261,7 +280,7 @@ var WorkersAiTextAdapter = class extends _adapters.BaseTextAdapter {
|
|
|
261
280
|
};
|
|
262
281
|
}
|
|
263
282
|
const delta = choice.delta;
|
|
264
|
-
const reasoningContent = delta.reasoning_content;
|
|
283
|
+
const reasoningContent = _nullishCoalesce(delta.reasoning_content, () => ( delta.reasoning));
|
|
265
284
|
if (reasoningContent) {
|
|
266
285
|
if (!hasEmittedStepStarted) {
|
|
267
286
|
hasEmittedStepStarted = true;
|
|
@@ -309,16 +328,13 @@ var WorkersAiTextAdapter = class extends _adapters.BaseTextAdapter {
|
|
|
309
328
|
const index = toolCallDelta.index;
|
|
310
329
|
if (!toolCallsInProgress.has(index)) {
|
|
311
330
|
toolCallsInProgress.set(index, {
|
|
312
|
-
id:
|
|
331
|
+
id: generateId("chatcmpl-tool"),
|
|
313
332
|
name: _optionalChain([toolCallDelta, 'access', _10 => _10.function, 'optionalAccess', _11 => _11.name]) || "",
|
|
314
333
|
arguments: "",
|
|
315
334
|
started: false
|
|
316
335
|
});
|
|
317
336
|
}
|
|
318
337
|
const toolCall = toolCallsInProgress.get(index);
|
|
319
|
-
if (toolCallDelta.id) {
|
|
320
|
-
toolCall.id = toolCallDelta.id;
|
|
321
|
-
}
|
|
322
338
|
if (_optionalChain([toolCallDelta, 'access', _12 => _12.function, 'optionalAccess', _13 => _13.name])) {
|
|
323
339
|
toolCall.name = toolCallDelta.function.name;
|
|
324
340
|
}
|
|
@@ -502,4 +518,4 @@ function createWorkersAiChat(model, config) {
|
|
|
502
518
|
|
|
503
519
|
|
|
504
520
|
exports.WorkersAiTextAdapter = WorkersAiTextAdapter; exports.createWorkersAiChat = createWorkersAiChat;
|
|
505
|
-
//# sourceMappingURL=chunk-
|
|
521
|
+
//# sourceMappingURL=chunk-DTGQYEJ2.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-DTGQYEJ2.cjs","../src/adapters/workers-ai.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACTA;AACC;AAAA,iDAGM;AACP,gFAAmB;AAyBnB,SAAS,oBAAA,CAAqB,MAAA,EAAwC;AACrE,EAAA,uDAAA,MAA8B,CAAA;AAE9B,EAAA,MAAM,eAAA,EAAqD,MAAA,CAAO,gBAAA,EAC/D,EAAE,oBAAA,EAAsB,MAAA,CAAO,gBAAgB,EAAA,EAC/C,KAAA,CAAA;AAEH,EAAA,GAAA,CAAI,qDAAA,MAA4B,CAAA,EAAG;AAElC,IAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,MACjB,MAAA,EAAQ,QAAA;AAAA,MACR,KAAA,EAAO,2DAAA;AAAA,QACN,MAAA,CAAO,OAAA;AAAA,QACP,eAAA,EAAiB,EAAE,YAAA,EAAc,eAAe,EAAA,EAAI,KAAA;AAAA,MACrD;AAAA,IACD,CAAC,CAAA;AAAA,EACF;AAEA,EAAA,GAAA,CAAI,yDAAA,MAAgC,CAAA,EAAG;AAEtC,IAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,MACjB,OAAA,EAAS,CAAA,8CAAA,EAAiD,MAAA,CAAO,SAAS,CAAA,MAAA,CAAA;AAAA,MAC1E,MAAA,EAAQ,MAAA,CAAO,MAAA;AAAA,MACf,cAAA,EAAgB;AAAA,IACjB,CAAC,CAAA;AAAA,EACF;AAGA,EAAA,MAAM,cAAA,EAAgB,MAAA;AACtB,EAAA,OAAO,IAAI,qBAAA,CAAO;AAAA,IACjB,KAAA,EAAO,kDAAA,YAAmB,EAAc,aAAA,EAAe,cAAc,CAAA;AAAA,IACrE,MAAA,mBAAQ,aAAA,CAAc,MAAA,UAAU;AAAA,EACjC,CAAC,CAAA;AACF;AAMA,SAAS,kBAAA,CAAmB,OAAA,EAA0C;AACrE,EAAA,GAAA,CAAI,QAAA,IAAY,IAAA,EAAM,OAAO,EAAA;AAC7B,EAAA,GAAA,CAAI,OAAO,QAAA,IAAY,QAAA,EAAU,OAAO,OAAA;AACxC,EAAA,OAAO,OAAA,CAAQ,OAAA,CAAQ,CAAC,CAAA,EAAA,GAAO,CAAA,CAAE,KAAA,IAAS,OAAA,EAAS,CAAC,CAAA,CAAE,OAAO,EAAA,EAAI,CAAC,CAAE,CAAA,CAAE,IAAA,CAAK,EAAE,CAAA;AAC9E;AAQA,SAAS,kBAAA,CAAmB,IAAA,EAAsE;AACjG,EAAA,OAAA,CAAQ,IAAA,CAAK,IAAA,EAAM;AAAA,IAClB,KAAK,MAAA;AACJ,MAAA,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS;AACjB,QAAA,OAAO,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,IAAA,CAAK,QAAQ,CAAA;AAAA,MAC3C;AACA,MAAA,OAAO,KAAA,CAAA;AAAA,IACR,KAAK,OAAA,EAAS;AACb,MAAA,IAAI,GAAA;AACJ,MAAA,GAAA,CAAI,IAAA,CAAK,MAAA,CAAO,KAAA,IAAS,MAAA,EAAQ;AAChC,QAAA,IAAA,EAAM,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,UAAA,CAAW,OAAO,EAAA,EACvC,IAAA,CAAK,MAAA,CAAO,MAAA,EACZ,CAAA,KAAA,EAAQ,IAAA,CAAK,MAAA,CAAO,QAAQ,CAAA,QAAA,EAAW,IAAA,CAAK,MAAA,CAAO,KAAK,CAAA,CAAA;AACrD,MAAA;AACY,QAAA;AACnB,MAAA;AAC+C,MAAA;AAChD,IAAA;AACA,IAAA;AAES,MAAA;AAC8D,QAAA;AACtE,MAAA;AACO,MAAA;AACT,EAAA;AACD;AAWoD;AACtB,EAAA;AACW,EAAA;AAEgB,EAAA;AACxC,EAAA;AAC8D,IAAA;AAC9E,EAAA;AAEwD,EAAA;AAC5B,EAAA;AACc,IAAA;AAC1B,IAAA;AACM,MAAA;AACrB,IAAA;AACD,EAAA;AACO,EAAA;AACR;AAM4C;AACU,EAAA;AACa,EAAA;AAEnB,EAAA;AAC1B,IAAA;AACb,MAAA;AAC0B,MAAA;AAChC,IAAA;AACF,EAAA;AAEgC,EAAA;AACF,IAAA;AACR,MAAA;AACb,QAAA;AACmC,QAAA;AACzC,MAAA;AACuC,IAAA;AACkC,MAAA;AACnE,QAAA;AACqC,QAAA;AAC5C,MAAA;AACuE,MAAA;AACT,QAAA;AACrD,UAAA;AACD,UAAA;AACI,UAAA;AACS,YAAA;AACK,YAAA;AACxB,UAAA;AACC,QAAA;AACH,MAAA;AACoC,MAAA;AACe,IAAA;AAC/C,MAAA;AACqC,MAAA;AACpC,QAAA;AACuB,UAAA;AACJ,UAAA;AACf,QAAA;AACqC,UAAA;AAC7C,QAAA;AACM,MAAA;AACsC,QAAA;AAC7C,MAAA;AACoB,MAAA;AACb,QAAA;AACqE,QAAA;AAClE,QAAA;AACT,MAAA;AACF,IAAA;AACD,EAAA;AAEO,EAAA;AACR;AAIgD;AAC5B,EAAA;AACS,EAAA;AACrB,IAAA;AACI,IAAA;AACE,MAAA;AACO,MAAA;AACD,MAAA;AAClB,IAAA;AACC,EAAA;AACH;AAMiD;AACsB,EAAA;AACvE;AAeE;AAK0D,EAAA;AACzB,IAAA;AAL3B,IAAA;AAEC,IAAA;AAIkC,IAAA;AAC1C,EAAA;AAEyE,EAAA;AACT,IAAA;AAEG,IAAA;AACxB,IAAA;AAEf,IAAA;AACF,IAAA;AACI,IAAA;AACF,IAAA;AACM,IAAA;AACR,IAAA;AACG,IAAA;AACD,IAAA;AACD,IAAA;AACI,IAAA;AAI5B,IAAA;AAEE,IAAA;AACC,MAAA;AACA,MAAA;AACgD,QAAA;AAC7B,UAAA;AACX,UAAA;AACH,UAAA;AACP,UAAA;AACQ,UAAA;AAC8B,UAAA;AACtC,QAAA;AAC6B,MAAA;AAGtB,QAAA;AACP,UAAA;AACqD,UAAA;AACtD,QAAA;AACkE,QAAA;AAC5C,UAAA;AACX,UAAA;AACH,UAAA;AACP,UAAA;AACA,QAAA;AAEK,QAAA;AACC,UAAA;AACN,UAAA;AAC8C,UAAA;AAC9C,UAAA;AACD,QAAA;AAEwC,QAAA;AACtB,QAAA;AACX,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACM,YAAA;AACP,UAAA;AACM,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACW,YAAA;AACE,YAAA;AACd,UAAA;AACM,UAAA;AACC,YAAA;AACN,YAAA;AAC8C,YAAA;AAC9C,YAAA;AACD,UAAA;AACD,QAAA;AAEqB,QAAA;AACa,UAAA;AACJ,YAAA;AACd,YAAA;AACc,YAAA;AACxB,YAAA;AACsD,cAAA;AAClD,YAAA;AACQ,cAAA;AAChB,YAAA;AACM,YAAA;AACC,cAAA;AACS,cAAA;AACF,cAAA;AACiC,cAAA;AAC9C,cAAA;AACO,cAAA;AACR,YAAA;AACM,YAAA;AACC,cAAA;AACS,cAAA;AACF,cAAA;AACiC,cAAA;AAC9C,cAAA;AACO,cAAA;AACR,YAAA;AACD,UAAA;AACD,QAAA;AAEiD,QAAA;AAC3C,QAAA;AACC,UAAA;AACN,UAAA;AAC8C,UAAA;AAC9C,UAAA;AAEG,UAAA;AACoC,YAAA;AACI,YAAA;AACL,YAAA;AAEnC,UAAA;AAI4D,UAAA;AAChE,QAAA;AACA,QAAA;AACD,MAAA;AAEkC,MAAA;AACiB,QAAA;AACpB,QAAA;AACjB,QAAA;AAGc,QAAA;AACH,UAAA;AACjB,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AACD,UAAA;AACD,QAAA;AAEqB,QAAA;AAKe,QAAA;AACd,QAAA;AAEO,UAAA;AACH,YAAA;AAClB,YAAA;AACC,cAAA;AACN,cAAA;AACU,cAAA;AAC0B,cAAA;AACpC,cAAA;AACD,YAAA;AACD,UAAA;AACwB,UAAA;AAOlB,UAAA;AACC,YAAA;AACN,YAAA;AACO,YAAA;AACE,YAAA;AAC2B,YAAA;AACpC,YAAA;AACD,UAAA;AACD,QAAA;AAGmB,QAAA;AACe,UAAA;AACH,YAAA;AACvB,YAAA;AACC,cAAA;AACN,cAAA;AACoC,cAAA;AACpC,cAAA;AACM,cAAA;AACP,YAAA;AACD,UAAA;AAE4B,UAAA;AACtB,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AACa,YAAA;AACJ,YAAA;AACV,UAAA;AACD,QAAA;AAGsB,QAAA;AACyB,UAAA;AACjB,YAAA;AAES,YAAA;AAIL,cAAA;AACA,gBAAA;AACQ,gBAAA;AAC3B,gBAAA;AACF,gBAAA;AACT,cAAA;AACF,YAAA;AAE8C,YAAA;AAKZ,YAAA;AACM,cAAA;AACxC,YAAA;AACuC,YAAA;AACO,cAAA;AAC9C,YAAA;AAGuD,YAAA;AACnC,cAAA;AACb,cAAA;AACC,gBAAA;AACe,gBAAA;AACF,gBAAA;AACiB,gBAAA;AACpC,gBAAA;AACA,gBAAA;AACD,cAAA;AACD,YAAA;AAG2D,YAAA;AACpD,cAAA;AACC,gBAAA;AACe,gBAAA;AACe,gBAAA;AACpC,gBAAA;AAC8B,gBAAA;AAC/B,cAAA;AACD,YAAA;AACD,UAAA;AACD,QAAA;AAG0B,QAAA;AACC,UAAA;AAGiD,UAAA;AAC1B,YAAA;AACnB,cAAA;AACxB,cAAA;AAGC,gBAAA;AACG,cAAA;AACQ,gBAAA;AAChB,cAAA;AACM,cAAA;AACC,gBAAA;AACe,gBAAA;AACF,gBAAA;AACiB,gBAAA;AACpC,gBAAA;AACO,gBAAA;AACR,cAAA;AACD,YAAA;AACD,UAAA;AAKC,UAAA;AAK+B,UAAA;AACzB,YAAA;AACC,cAAA;AACN,cAAA;AACoC,cAAA;AACpC,cAAA;AACD,YAAA;AACD,UAAA;AAGM,UAAA;AACC,YAAA;AACN,YAAA;AACoC,YAAA;AACpC,YAAA;AAEG,YAAA;AAC0B,cAAA;AACI,cAAA;AACL,cAAA;AAEzB,YAAA;AACW,YAAA;AACf,UAAA;AACD,QAAA;AACD,MAAA;AAKsD,MAAA;AAC7C,QAAA;AACP,UAAA;AACD,QAAA;AAGgD,QAAA;AACzB,UAAA;AACO,YAAA;AACxB,YAAA;AACkE,cAAA;AAC9D,YAAA;AACQ,cAAA;AAChB,YAAA;AACM,YAAA;AACC,cAAA;AACe,cAAA;AACF,cAAA;AACE,cAAA;AACrB,cAAA;AACO,cAAA;AACR,YAAA;AACD,UAAA;AACD,QAAA;AAGgC,QAAA;AACzB,UAAA;AACC,YAAA;AACN,YAAA;AACqB,YAAA;AACrB,YAAA;AACD,UAAA;AACD,QAAA;AAEM,QAAA;AACC,UAAA;AACN,UAAA;AACqB,UAAA;AACrB,UAAA;AACc,UAAA;AACf,QAAA;AACD,MAAA;AACe,IAAA;AACsD,MAAA;AAEC,MAAA;AAC3C,MAAA;AACpB,QAAA;AACC,UAAA;AACN,UAAA;AACqB,UAAA;AACrB,UAAA;AACD,QAAA;AACD,MAAA;AACM,MAAA;AACC,QAAA;AACN,QAAA;AACqB,QAAA;AACrB,QAAA;AACO,QAAA;AACc,UAAA;AACpB,UAAA;AACD,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAI4C,EAAA;AACL,IAAA;AACkB,IAAA;AAEY,IAAA;AAC9C,MAAA;AACrB,IAAA;AAE0D,IAAA;AACrC,MAAA;AACX,MAAA;AACV,MAAA;AACQ,MAAA;AACS,MAAA;AACV,QAAA;AACO,QAAA;AACN,UAAA;AACE,UAAA;AACA,UAAA;AACT,QAAA;AACD,MAAA;AACA,IAAA;AAEkC,IAAA;AAEtB,IAAA;AACF,MAAA;AACoE,QAAA;AAC9E,MAAA;AACD,IAAA;AAE8C,IAAA;AAI1C,IAAA;AACA,IAAA;AAEgC,IAAA;AACzB,MAAA;AACN,MAAA;AACsB,QAAA;AAClB,MAAA;AACA,QAAA;AACR,MAAA;AACM,IAAA;AAEC,MAAA;AAC4B,MAAA;AACpC,IAAA;AAEuB,IAAA;AACxB,EAAA;AACD;AAM+F;AACjD,EAAA;AAC9C;ADrLmI;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-DTGQYEJ2.cjs","sourcesContent":[null,"import type { AiModels, BaseAiTextGeneration } from \"@cloudflare/workers-types\";\nimport type { ContentPart, ModelMessage, StreamChunk, TextOptions } from \"@tanstack/ai\";\nimport {\n\tBaseTextAdapter,\n\ttype StructuredOutputOptions,\n\ttype StructuredOutputResult,\n} from \"@tanstack/ai/adapters\";\nimport OpenAI from \"openai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tcreateWorkersAiBindingFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model types derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiTextModel =\n\t| {\n\t\t\t[K in keyof AiModels]: AiModels[K] extends BaseAiTextGeneration ? K : never;\n\t }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// Helpers: build the right OpenAI client depending on config mode\n// ---------------------------------------------------------------------------\n\nfunction buildWorkersAiClient(config: WorkersAiAdapterConfig): OpenAI {\n\tvalidateWorkersAiConfig(config);\n\n\tconst sessionHeaders: Record<string, string> | undefined = config.sessionAffinity\n\t\t? { \"x-session-affinity\": config.sessionAffinity }\n\t\t: undefined;\n\n\tif (isDirectBindingConfig(config)) {\n\t\t// Plain binding mode: shim translates OpenAI fetch calls to env.AI.run()\n\t\treturn new OpenAI({\n\t\t\tapiKey: \"unused\",\n\t\t\tfetch: createWorkersAiBindingFetch(\n\t\t\t\tconfig.binding,\n\t\t\t\tsessionHeaders ? { extraHeaders: sessionHeaders } : undefined,\n\t\t\t),\n\t\t});\n\t}\n\n\tif (isDirectCredentialsConfig(config)) {\n\t\t// Plain REST mode: point OpenAI SDK at Workers AI's OpenAI-compatible endpoint\n\t\treturn new OpenAI({\n\t\t\tbaseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,\n\t\t\tapiKey: config.apiKey,\n\t\t\tdefaultHeaders: sessionHeaders,\n\t\t});\n\t}\n\n\t// Gateway mode (existing): use createGatewayFetch\n\tconst gatewayConfig = config as AiGatewayAdapterConfig;\n\treturn new OpenAI({\n\t\tfetch: createGatewayFetch(\"workers-ai\", gatewayConfig, sessionHeaders),\n\t\tapiKey: gatewayConfig.apiKey ?? \"unused\",\n\t});\n}\n\n// ---------------------------------------------------------------------------\n// Shared message-building helpers\n// ---------------------------------------------------------------------------\n\nfunction extractTextContent(content: ModelMessage[\"content\"]): string {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n}\n\n/**\n * Convert a single TanStack AI {@link ContentPart} to the OpenAI Chat\n * Completions multi-modal format.\n *\n * TODO: handle other content types (audio, video, document)\n */\nfunction convertContentPart(part: ContentPart): OpenAI.Chat.ChatCompletionContentPart | undefined {\n\tswitch (part.type) {\n\t\tcase \"text\":\n\t\t\tif (part.content) {\n\t\t\t\treturn { type: \"text\", text: part.content };\n\t\t\t}\n\t\t\treturn undefined;\n\t\tcase \"image\": {\n\t\t\tlet url: string;\n\t\t\tif (part.source.type === \"data\") {\n\t\t\t\turl = part.source.value.startsWith(\"data:\")\n\t\t\t\t\t? part.source.value\n\t\t\t\t\t: `data:${part.source.mimeType};base64,${part.source.value}`;\n\t\t\t} else {\n\t\t\t\turl = part.source.value;\n\t\t\t}\n\t\t\treturn { type: \"image_url\", image_url: { url } };\n\t\t}\n\t\tdefault:\n\t\t\t// audio, video, document — not supported for now\n\t\t\tconsole.warn(\n\t\t\t\t`[@cloudflare/tanstack-ai] Unsupported content part type \"${part.type}\" — skipping`,\n\t\t\t);\n\t\t\treturn undefined;\n\t}\n}\n\n/**\n * Build OpenAI-compatible user message content from TanStack AI content.\n *\n * If the content has only text parts, returns a plain string.\n * If it includes image parts, returns an array of content parts in\n * OpenAI's multi-modal format (text + image_url).\n */\nfunction buildUserContent(\n\tcontent: ModelMessage[\"content\"],\n): string | OpenAI.Chat.ChatCompletionContentPart[] {\n\tif (content === null) return \"\";\n\tif (typeof content === \"string\") return content;\n\n\tconst hasImages = content.some((p) => p.type === \"image\");\n\tif (!hasImages) {\n\t\treturn content.flatMap((p) => (p.type === \"text\" ? [p.content] : [])).join(\"\");\n\t}\n\n\tconst parts: OpenAI.Chat.ChatCompletionContentPart[] = [];\n\tfor (const part of content) {\n\t\tconst converted = convertContentPart(part);\n\t\tif (converted) {\n\t\t\tparts.push(converted);\n\t\t}\n\t}\n\treturn parts;\n}\n\nfunction buildOpenAIMessages(\n\tsystemPrompts: string[] | undefined,\n\tmessages: ModelMessage[],\n\toptions?: { includeToolMessages?: boolean },\n): OpenAI.Chat.ChatCompletionMessageParam[] {\n\tconst includeTools = options?.includeToolMessages ?? true;\n\tconst openAIMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [];\n\n\tif (systemPrompts && systemPrompts.length > 0) {\n\t\topenAIMessages.push({\n\t\t\trole: \"system\",\n\t\t\tcontent: systemPrompts.join(\"\\n\"),\n\t\t});\n\t}\n\n\tfor (const message of messages) {\n\t\tif (message.role === \"user\") {\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"user\",\n\t\t\t\tcontent: buildUserContent(message.content),\n\t\t\t});\n\t\t} else if (message.role === \"assistant\") {\n\t\t\tconst assistantMessage: OpenAI.Chat.ChatCompletionAssistantMessageParam = {\n\t\t\t\trole: \"assistant\",\n\t\t\t\tcontent: extractTextContent(message.content),\n\t\t\t};\n\t\t\tif (includeTools && message.toolCalls && message.toolCalls.length > 0) {\n\t\t\t\tassistantMessage.tool_calls = message.toolCalls.map((tc) => ({\n\t\t\t\t\tid: tc.id,\n\t\t\t\t\ttype: \"function\" as const,\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: tc.function.name,\n\t\t\t\t\t\targuments: tc.function.arguments,\n\t\t\t\t\t},\n\t\t\t\t}));\n\t\t\t}\n\t\t\topenAIMessages.push(assistantMessage);\n\t\t} else if (includeTools && message.role === \"tool\") {\n\t\t\tlet toolContent: string;\n\t\t\tif (typeof message.content === \"string\") {\n\t\t\t\ttry {\n\t\t\t\t\tJSON.parse(message.content);\n\t\t\t\t\ttoolContent = message.content;\n\t\t\t\t} catch {\n\t\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttoolContent = JSON.stringify(message.content);\n\t\t\t}\n\t\t\topenAIMessages.push({\n\t\t\t\trole: \"tool\",\n\t\t\t\ttool_call_id: message.toolCallId || `tool_${crypto.randomUUID().slice(0, 8)}`,\n\t\t\t\tcontent: toolContent,\n\t\t\t});\n\t\t}\n\t}\n\n\treturn openAIMessages;\n}\n\nfunction buildOpenAITools(\n\ttools: Array<{ name: string; description: string; inputSchema?: unknown }> | undefined,\n): OpenAI.Chat.ChatCompletionTool[] | undefined {\n\tif (!tools) return undefined;\n\treturn tools.map((tool) => ({\n\t\ttype: \"function\" as const,\n\t\tfunction: {\n\t\t\tname: tool.name,\n\t\t\tdescription: tool.description,\n\t\t\tparameters: tool.inputSchema as Record<string, unknown>,\n\t\t},\n\t}));\n}\n\n// ---------------------------------------------------------------------------\n// ID generation\n// ---------------------------------------------------------------------------\n\nfunction generateId(prefix = \"chatcmpl\"): string {\n\treturn `${prefix}-${crypto.randomUUID().replace(/-/g, \"\").slice(0, 16)}`;\n}\n\n// ---------------------------------------------------------------------------\n// WorkersAiTextAdapter: chat / structured output via OpenAI Chat Completions\n// ---------------------------------------------------------------------------\n\n// TODO: Replace `any` generic params with proper types once BaseTextAdapter's\n// provider-options generics stabilize. Workers AI doesn't have provider-specific\n// options in the TanStack sense, so `any` is pragmatic for now.\nexport class WorkersAiTextAdapter<TModel extends WorkersAiTextModel> extends BaseTextAdapter<\n\tTModel,\n\t// eslint-disable-next-line @typescript-eslint/no-explicit-any -- BaseTextAdapter generic params are opaque\n\tany,\n\tany,\n\tany\n> {\n\tname = \"workers-ai\" as const;\n\n\tprivate client: OpenAI;\n\n\tconstructor(model: TModel, config: WorkersAiAdapterConfig) {\n\t\tsuper({ apiKey: \"unused\" }, model);\n\t\tthis.client = buildWorkersAiClient(config);\n\t}\n\n\tasync *chatStream(options: TextOptions<any>): AsyncIterable<StreamChunk> {\n\t\tconst { systemPrompts, messages, tools, temperature, model } = options;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages);\n\t\tconst openAITools = buildOpenAITools(tools);\n\n\t\tconst timestamp = Date.now();\n\t\tconst runId = generateId();\n\t\tconst messageId = generateId();\n\t\tlet hasEmittedRunStarted = false;\n\t\tlet hasEmittedTextMessageStart = false;\n\t\tlet accumulatedContent = \"\";\n\t\tlet hasEmittedStepStarted = false;\n\t\tlet accumulatedReasoning = \"\";\n\t\tconst stepId = generateId();\n\t\tlet hasReceivedFinishReason = false;\n\t\tconst toolCallsInProgress = new Map<\n\t\t\tnumber,\n\t\t\t{ id: string; name: string; arguments: string; started: boolean }\n\t\t>();\n\n\t\ttry {\n\t\t\tlet stream: AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>;\n\t\t\ttry {\n\t\t\t\tstream = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t\tstream: true,\n\t\t\t\t\tstream_options: { include_usage: true },\n\t\t\t\t});\n\t\t\t} catch (streamError: unknown) {\n\t\t\t\t// Some models (e.g. GPT-OSS) don't support streaming via the REST API.\n\t\t\t\t// Fall back to a non-streaming call and yield the result as a single chunk.\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Streaming failed, falling back to non-streaming:\",\n\t\t\t\t\tstreamError instanceof Error ? streamError.message : streamError,\n\t\t\t\t);\n\t\t\t\tconst nonStreamResult = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t});\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\n\t\t\t\tconst msg = nonStreamResult.choices[0]?.message;\n\t\t\t\tif (msg?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: msg.content,\n\t\t\t\t\t\tcontent: msg.content,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tif (msg?.tool_calls) {\n\t\t\t\t\tfor (const tc of msg.tool_calls) {\n\t\t\t\t\t\tif (tc.type !== \"function\") continue;\n\t\t\t\t\t\tconst fn = tc.function;\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = fn.arguments ? JSON.parse(fn.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tconst finishReason = nonStreamResult.choices[0]?.finish_reason;\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tusage: nonStreamResult.usage\n\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\tpromptTokens: nonStreamResult.usage.prompt_tokens,\n\t\t\t\t\t\t\t\tcompletionTokens: nonStreamResult.usage.completion_tokens,\n\t\t\t\t\t\t\t\ttotalTokens: nonStreamResult.usage.total_tokens,\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t: undefined,\n\t\t\t\t\tfinishReason:\n\t\t\t\t\t\tfinishReason === \"tool_calls\" || finishReason === \"function_call\"\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: ((finishReason as \"stop\" | \"length\" | \"content_filter\") ?? \"stop\"),\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tfor await (const chunk of stream) {\n\t\t\t\tif (!chunk.choices || chunk.choices.length === 0) continue;\n\t\t\t\tconst choice = chunk.choices[0];\n\t\t\t\tif (!choice) continue;\n\n\t\t\t\t// Emit RUN_STARTED on first chunk\n\t\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\t\thasEmittedRunStarted = true;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tconst delta = choice.delta;\n\n\t\t\t\t// Reasoning content (used by models like QwQ, DeepSeek R1, Kimi K2.5)\n\t\t\t\t// The OpenAI SDK doesn't type this field, but models send it as an extension.\n\t\t\t\tconst reasoningContent = ((delta as Record<string, unknown>).reasoning_content ??\n\t\t\t\t\t(delta as Record<string, unknown>).reasoning) as string | undefined;\n\t\t\t\tif (reasoningContent) {\n\t\t\t\t\t// RUN_STARTED is already guaranteed by the guard above\n\t\t\t\t\tif (!hasEmittedStepStarted) {\n\t\t\t\t\t\thasEmittedStepStarted = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"STEP_STARTED\",\n\t\t\t\t\t\t\tstepId,\n\t\t\t\t\t\t\tstepType: \"thinking\",\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t\taccumulatedReasoning += reasoningContent;\n\t\t\t\t\t// TODO: TanStack AI's StreamProcessor currently treats STEP_FINISHED as an\n\t\t\t\t\t// incremental reasoning event (with `delta` + accumulated `content`), so we\n\t\t\t\t\t// emit one per token. If TanStack AI adds a dedicated STEP_CONTENT event\n\t\t\t\t\t// type, this should be updated to emit STEP_CONTENT per token and a single\n\t\t\t\t\t// STEP_FINISHED when reasoning ends (i.e. when the first non-reasoning\n\t\t\t\t\t// content or finish_reason arrives).\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"STEP_FINISHED\",\n\t\t\t\t\t\tstepId,\n\t\t\t\t\t\tdelta: reasoningContent,\n\t\t\t\t\t\tcontent: accumulatedReasoning,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Text content\n\t\t\t\tif (delta.content) {\n\t\t\t\t\tif (!hasEmittedTextMessageStart) {\n\t\t\t\t\t\thasEmittedTextMessageStart = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\taccumulatedContent += delta.content;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: delta.content,\n\t\t\t\t\t\tcontent: accumulatedContent,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Tool calls\n\t\t\t\tif (delta.tool_calls) {\n\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\tif (!toolCallsInProgress.has(index)) {\n\t\t\t\t\t\t\t// Always generate a unique ID per tool call index.\n\t\t\t\t\t\t\t// The backend may send the same ID for multiple tool calls,\n\t\t\t\t\t\t\t// so we cannot trust toolCallDelta.id to be unique.\n\t\t\t\t\t\t\ttoolCallsInProgress.set(index, {\n\t\t\t\t\t\t\t\tid: generateId(\"chatcmpl-tool\"),\n\t\t\t\t\t\t\t\tname: toolCallDelta.function?.name || \"\",\n\t\t\t\t\t\t\t\targuments: \"\",\n\t\t\t\t\t\t\t\tstarted: false,\n\t\t\t\t\t\t\t});\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tconst toolCall = toolCallsInProgress.get(index)!;\n\n\t\t\t\t\t\t// Only update name if provided (ID is already set at creation time\n\t\t\t\t\t\t// and should not be overwritten by subsequent chunks that may have\n\t\t\t\t\t\t// duplicate/shared IDs from the backend)\n\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\ttoolCall.name = toolCallDelta.function.name;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\ttoolCall.arguments += toolCallDelta.function.arguments;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Emit TOOL_CALL_START once we have id and name\n\t\t\t\t\t\tif (toolCall.id && toolCall.name && !toolCall.started) {\n\t\t\t\t\t\t\ttoolCall.started = true;\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tindex,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Stream tool call arguments\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments && toolCall.started) {\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_ARGS\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tdelta: toolCallDelta.function.arguments,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Finish\n\t\t\t\tif (choice.finish_reason) {\n\t\t\t\t\thasReceivedFinishReason = true;\n\n\t\t\t\t\t// End tool calls\n\t\t\t\t\tif (choice.finish_reason === \"tool_calls\" || toolCallsInProgress.size > 0) {\n\t\t\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\t\tparsedInput = toolCall.arguments\n\t\t\t\t\t\t\t\t\t? JSON.parse(toolCall.arguments)\n\t\t\t\t\t\t\t\t\t: {};\n\t\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\n\t\t\t\t\tconst computedFinishReason =\n\t\t\t\t\t\tchoice.finish_reason === \"tool_calls\" ||\n\t\t\t\t\t\tchoice.finish_reason === \"function_call\" ||\n\t\t\t\t\t\ttoolCallsInProgress.size > 0\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: (choice.finish_reason as \"stop\" | \"length\" | \"content_filter\");\n\n\t\t\t\t\t// End text message if started\n\t\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Emit RUN_FINISHED\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tusage: chunk.usage\n\t\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\t\tpromptTokens: chunk.usage.prompt_tokens,\n\t\t\t\t\t\t\t\t\tcompletionTokens: chunk.usage.completion_tokens,\n\t\t\t\t\t\t\t\t\ttotalTokens: chunk.usage.total_tokens,\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t: undefined,\n\t\t\t\t\t\tfinishReason: computedFinishReason,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Premature stream termination: the stream ended without a finish_reason.\n\t\t\t// This can happen when Workers AI truncates a response or the connection drops.\n\t\t\t// Emit proper closing events so the consumer doesn't hang.\n\t\t\tif (hasEmittedRunStarted && !hasReceivedFinishReason) {\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Stream ended without finish_reason — possible truncation or connection drop\",\n\t\t\t\t);\n\n\t\t\t\t// Close any open tool calls\n\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\tif (toolCall.started) {\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Close text message if open\n\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tfinishReason: \"stop\",\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tconst code =\n\t\t\t\terror instanceof Error ? (error as Error & { code?: string }).code : undefined;\n\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t\tyield {\n\t\t\t\ttype: \"RUN_ERROR\",\n\t\t\t\trunId,\n\t\t\t\tmodel: model ?? this.model,\n\t\t\t\ttimestamp,\n\t\t\t\terror: {\n\t\t\t\t\tmessage: message || \"Unknown error\",\n\t\t\t\t\tcode,\n\t\t\t\t},\n\t\t\t} satisfies StreamChunk;\n\t\t}\n\t}\n\n\tasync structuredOutput(\n\t\toptions: StructuredOutputOptions<any>,\n\t): Promise<StructuredOutputResult<unknown>> {\n\t\tconst { outputSchema, chatOptions } = options;\n\t\tconst { systemPrompts, messages, temperature, model } = chatOptions;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages, {\n\t\t\tincludeToolMessages: false,\n\t\t});\n\n\t\tconst response = await this.client.chat.completions.create({\n\t\t\tmodel: model ?? this.model,\n\t\t\tmessages: openAIMessages,\n\t\t\ttemperature,\n\t\t\tstream: false,\n\t\t\tresponse_format: {\n\t\t\t\ttype: \"json_schema\",\n\t\t\t\tjson_schema: {\n\t\t\t\t\tname: \"structured_output\",\n\t\t\t\t\tstrict: true,\n\t\t\t\t\tschema: outputSchema,\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst choice = response.choices?.[0];\n\n\t\tif (!choice) {\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI structured output returned no choices: ${JSON.stringify(response)}`,\n\t\t\t);\n\t\t}\n\n\t\tconst rawContent = choice.message?.content ?? \"\";\n\n\t\t// Workers AI REST endpoint may return `content` as an already-parsed object\n\t\t// when using json_schema response format, so normalise both cases.\n\t\tlet data: unknown;\n\t\tlet rawText: string;\n\n\t\tif (typeof rawContent === \"string\") {\n\t\t\trawText = rawContent;\n\t\t\ttry {\n\t\t\t\tdata = JSON.parse(rawText);\n\t\t\t} catch {\n\t\t\t\tdata = rawText;\n\t\t\t}\n\t\t} else {\n\t\t\t// Already an object — stringify for rawText, use directly for data\n\t\t\tdata = rawContent;\n\t\t\trawText = JSON.stringify(rawContent);\n\t\t}\n\n\t\treturn { data, rawText };\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\nexport function createWorkersAiChat(model: WorkersAiTextModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiTextAdapter(model, config);\n}\n"]}
|
|
@@ -10,7 +10,7 @@ import {
|
|
|
10
10
|
isDirectBindingConfig,
|
|
11
11
|
isDirectCredentialsConfig,
|
|
12
12
|
validateWorkersAiConfig
|
|
13
|
-
} from "./chunk-
|
|
13
|
+
} from "./chunk-727MRLUF.js";
|
|
14
14
|
import {
|
|
15
15
|
__publicField
|
|
16
16
|
} from "./chunk-V6TY7KAL.js";
|
|
@@ -108,4 +108,4 @@ export {
|
|
|
108
108
|
WorkersAiTTSAdapter,
|
|
109
109
|
createWorkersAiTts
|
|
110
110
|
};
|
|
111
|
-
//# sourceMappingURL=chunk-
|
|
111
|
+
//# sourceMappingURL=chunk-GST6AT3R.js.map
|
|
@@ -4,7 +4,7 @@ import {
|
|
|
4
4
|
isDirectBindingConfig,
|
|
5
5
|
isDirectCredentialsConfig,
|
|
6
6
|
validateWorkersAiConfig
|
|
7
|
-
} from "./chunk-
|
|
7
|
+
} from "./chunk-727MRLUF.js";
|
|
8
8
|
import {
|
|
9
9
|
__publicField
|
|
10
10
|
} from "./chunk-V6TY7KAL.js";
|
|
@@ -16,50 +16,69 @@ import {
|
|
|
16
16
|
import OpenAI from "openai";
|
|
17
17
|
function buildWorkersAiClient(config) {
|
|
18
18
|
validateWorkersAiConfig(config);
|
|
19
|
+
const sessionHeaders = config.sessionAffinity ? { "x-session-affinity": config.sessionAffinity } : void 0;
|
|
19
20
|
if (isDirectBindingConfig(config)) {
|
|
20
21
|
return new OpenAI({
|
|
21
22
|
apiKey: "unused",
|
|
22
|
-
fetch: createWorkersAiBindingFetch(
|
|
23
|
+
fetch: createWorkersAiBindingFetch(
|
|
24
|
+
config.binding,
|
|
25
|
+
sessionHeaders ? { extraHeaders: sessionHeaders } : void 0
|
|
26
|
+
)
|
|
23
27
|
});
|
|
24
28
|
}
|
|
25
29
|
if (isDirectCredentialsConfig(config)) {
|
|
26
30
|
return new OpenAI({
|
|
27
31
|
baseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,
|
|
28
|
-
apiKey: config.apiKey
|
|
32
|
+
apiKey: config.apiKey,
|
|
33
|
+
defaultHeaders: sessionHeaders
|
|
29
34
|
});
|
|
30
35
|
}
|
|
31
36
|
const gatewayConfig = config;
|
|
32
37
|
return new OpenAI({
|
|
33
|
-
fetch: createGatewayFetch("workers-ai", gatewayConfig),
|
|
38
|
+
fetch: createGatewayFetch("workers-ai", gatewayConfig, sessionHeaders),
|
|
34
39
|
apiKey: gatewayConfig.apiKey ?? "unused"
|
|
35
40
|
});
|
|
36
41
|
}
|
|
37
42
|
function extractTextContent(content) {
|
|
38
43
|
if (content === null) return "";
|
|
39
44
|
if (typeof content === "string") return content;
|
|
40
|
-
return content.
|
|
45
|
+
return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
|
|
46
|
+
}
|
|
47
|
+
function convertContentPart(part) {
|
|
48
|
+
switch (part.type) {
|
|
49
|
+
case "text":
|
|
50
|
+
if (part.content) {
|
|
51
|
+
return { type: "text", text: part.content };
|
|
52
|
+
}
|
|
53
|
+
return void 0;
|
|
54
|
+
case "image": {
|
|
55
|
+
let url;
|
|
56
|
+
if (part.source.type === "data") {
|
|
57
|
+
url = part.source.value.startsWith("data:") ? part.source.value : `data:${part.source.mimeType};base64,${part.source.value}`;
|
|
58
|
+
} else {
|
|
59
|
+
url = part.source.value;
|
|
60
|
+
}
|
|
61
|
+
return { type: "image_url", image_url: { url } };
|
|
62
|
+
}
|
|
63
|
+
default:
|
|
64
|
+
console.warn(
|
|
65
|
+
`[@cloudflare/tanstack-ai] Unsupported content part type "${part.type}" \u2014 skipping`
|
|
66
|
+
);
|
|
67
|
+
return void 0;
|
|
68
|
+
}
|
|
41
69
|
}
|
|
42
70
|
function buildUserContent(content) {
|
|
43
71
|
if (content === null) return "";
|
|
44
72
|
if (typeof content === "string") return content;
|
|
45
|
-
const hasImages = content.some((p) => p.type === "
|
|
73
|
+
const hasImages = content.some((p) => p.type === "image");
|
|
46
74
|
if (!hasImages) {
|
|
47
|
-
return content.
|
|
75
|
+
return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
|
|
48
76
|
}
|
|
49
77
|
const parts = [];
|
|
50
78
|
for (const part of content) {
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
parts.push({
|
|
55
|
-
type: "image_url",
|
|
56
|
-
image_url: { url: part.content }
|
|
57
|
-
});
|
|
58
|
-
} else if (part.type === "image_url" && part.image_url) {
|
|
59
|
-
parts.push({
|
|
60
|
-
type: "image_url",
|
|
61
|
-
image_url: part.image_url
|
|
62
|
-
});
|
|
79
|
+
const converted = convertContentPart(part);
|
|
80
|
+
if (converted) {
|
|
81
|
+
parts.push(converted);
|
|
63
82
|
}
|
|
64
83
|
}
|
|
65
84
|
return parts;
|
|
@@ -127,8 +146,8 @@ function buildOpenAITools(tools) {
|
|
|
127
146
|
}
|
|
128
147
|
}));
|
|
129
148
|
}
|
|
130
|
-
function generateId(prefix) {
|
|
131
|
-
return `${prefix}-${crypto.randomUUID()}`;
|
|
149
|
+
function generateId(prefix = "chatcmpl") {
|
|
150
|
+
return `${prefix}-${crypto.randomUUID().replace(/-/g, "").slice(0, 16)}`;
|
|
132
151
|
}
|
|
133
152
|
var WorkersAiTextAdapter = class extends BaseTextAdapter {
|
|
134
153
|
constructor(model, config) {
|
|
@@ -142,14 +161,14 @@ var WorkersAiTextAdapter = class extends BaseTextAdapter {
|
|
|
142
161
|
const openAIMessages = buildOpenAIMessages(systemPrompts, messages);
|
|
143
162
|
const openAITools = buildOpenAITools(tools);
|
|
144
163
|
const timestamp = Date.now();
|
|
145
|
-
const runId = generateId(
|
|
146
|
-
const messageId = generateId(
|
|
164
|
+
const runId = generateId();
|
|
165
|
+
const messageId = generateId();
|
|
147
166
|
let hasEmittedRunStarted = false;
|
|
148
167
|
let hasEmittedTextMessageStart = false;
|
|
149
168
|
let accumulatedContent = "";
|
|
150
169
|
let hasEmittedStepStarted = false;
|
|
151
170
|
let accumulatedReasoning = "";
|
|
152
|
-
const stepId = generateId(
|
|
171
|
+
const stepId = generateId();
|
|
153
172
|
let hasReceivedFinishReason = false;
|
|
154
173
|
const toolCallsInProgress = /* @__PURE__ */ new Map();
|
|
155
174
|
try {
|
|
@@ -261,7 +280,7 @@ var WorkersAiTextAdapter = class extends BaseTextAdapter {
|
|
|
261
280
|
};
|
|
262
281
|
}
|
|
263
282
|
const delta = choice.delta;
|
|
264
|
-
const reasoningContent = delta.reasoning_content;
|
|
283
|
+
const reasoningContent = delta.reasoning_content ?? delta.reasoning;
|
|
265
284
|
if (reasoningContent) {
|
|
266
285
|
if (!hasEmittedStepStarted) {
|
|
267
286
|
hasEmittedStepStarted = true;
|
|
@@ -309,16 +328,13 @@ var WorkersAiTextAdapter = class extends BaseTextAdapter {
|
|
|
309
328
|
const index = toolCallDelta.index;
|
|
310
329
|
if (!toolCallsInProgress.has(index)) {
|
|
311
330
|
toolCallsInProgress.set(index, {
|
|
312
|
-
id:
|
|
331
|
+
id: generateId("chatcmpl-tool"),
|
|
313
332
|
name: toolCallDelta.function?.name || "",
|
|
314
333
|
arguments: "",
|
|
315
334
|
started: false
|
|
316
335
|
});
|
|
317
336
|
}
|
|
318
337
|
const toolCall = toolCallsInProgress.get(index);
|
|
319
|
-
if (toolCallDelta.id) {
|
|
320
|
-
toolCall.id = toolCallDelta.id;
|
|
321
|
-
}
|
|
322
338
|
if (toolCallDelta.function?.name) {
|
|
323
339
|
toolCall.name = toolCallDelta.function.name;
|
|
324
340
|
}
|
|
@@ -502,4 +518,4 @@ export {
|
|
|
502
518
|
WorkersAiTextAdapter,
|
|
503
519
|
createWorkersAiChat
|
|
504
520
|
};
|
|
505
|
-
//# sourceMappingURL=chunk-
|
|
521
|
+
//# sourceMappingURL=chunk-HDDZEGLC.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"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, model } = options;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages);\n\t\tconst openAITools = buildOpenAITools(tools);\n\n\t\tconst timestamp = Date.now();\n\t\tconst runId = generateId();\n\t\tconst messageId = generateId();\n\t\tlet hasEmittedRunStarted = false;\n\t\tlet hasEmittedTextMessageStart = false;\n\t\tlet accumulatedContent = \"\";\n\t\tlet hasEmittedStepStarted = false;\n\t\tlet accumulatedReasoning = \"\";\n\t\tconst stepId = generateId();\n\t\tlet hasReceivedFinishReason = false;\n\t\tconst toolCallsInProgress = new Map<\n\t\t\tnumber,\n\t\t\t{ id: string; name: string; arguments: string; started: boolean }\n\t\t>();\n\n\t\ttry {\n\t\t\tlet stream: AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>;\n\t\t\ttry {\n\t\t\t\tstream = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t\tstream: true,\n\t\t\t\t\tstream_options: { include_usage: true },\n\t\t\t\t});\n\t\t\t} catch (streamError: unknown) {\n\t\t\t\t// Some models (e.g. GPT-OSS) don't support streaming via the REST API.\n\t\t\t\t// Fall back to a non-streaming call and yield the result as a single chunk.\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Streaming failed, falling back to non-streaming:\",\n\t\t\t\t\tstreamError instanceof Error ? streamError.message : streamError,\n\t\t\t\t);\n\t\t\t\tconst nonStreamResult = await this.client.chat.completions.create({\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\tmessages: openAIMessages,\n\t\t\t\t\ttools: openAITools,\n\t\t\t\t\ttemperature,\n\t\t\t\t});\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\n\t\t\t\tconst msg = nonStreamResult.choices[0]?.message;\n\t\t\t\tif (msg?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: msg.content,\n\t\t\t\t\t\tcontent: msg.content,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tif (msg?.tool_calls) {\n\t\t\t\t\tfor (const tc of msg.tool_calls) {\n\t\t\t\t\t\tif (tc.type !== \"function\") continue;\n\t\t\t\t\t\tconst fn = tc.function;\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = fn.arguments ? JSON.parse(fn.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: tc.id,\n\t\t\t\t\t\t\ttoolName: fn.name,\n\t\t\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tconst finishReason = nonStreamResult.choices[0]?.finish_reason;\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: nonStreamResult.model || model || this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tusage: nonStreamResult.usage\n\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\tpromptTokens: nonStreamResult.usage.prompt_tokens,\n\t\t\t\t\t\t\t\tcompletionTokens: nonStreamResult.usage.completion_tokens,\n\t\t\t\t\t\t\t\ttotalTokens: nonStreamResult.usage.total_tokens,\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t: undefined,\n\t\t\t\t\tfinishReason:\n\t\t\t\t\t\tfinishReason === \"tool_calls\" || finishReason === \"function_call\"\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: ((finishReason as \"stop\" | \"length\" | \"content_filter\") ?? \"stop\"),\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tfor await (const chunk of stream) {\n\t\t\t\tif (!chunk.choices || chunk.choices.length === 0) continue;\n\t\t\t\tconst choice = chunk.choices[0];\n\t\t\t\tif (!choice) continue;\n\n\t\t\t\t// Emit RUN_STARTED on first chunk\n\t\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\t\thasEmittedRunStarted = true;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tconst delta = choice.delta;\n\n\t\t\t\t// Reasoning content (used by models like QwQ, DeepSeek R1, Kimi K2.5)\n\t\t\t\t// The OpenAI SDK doesn't type this field, but models send it as an extension.\n\t\t\t\tconst reasoningContent = ((delta as Record<string, unknown>).reasoning_content ??\n\t\t\t\t\t(delta as Record<string, unknown>).reasoning) as string | undefined;\n\t\t\t\tif (reasoningContent) {\n\t\t\t\t\t// RUN_STARTED is already guaranteed by the guard above\n\t\t\t\t\tif (!hasEmittedStepStarted) {\n\t\t\t\t\t\thasEmittedStepStarted = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"STEP_STARTED\",\n\t\t\t\t\t\t\tstepId,\n\t\t\t\t\t\t\tstepType: \"thinking\",\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t\taccumulatedReasoning += reasoningContent;\n\t\t\t\t\t// TODO: TanStack AI's StreamProcessor currently treats STEP_FINISHED as an\n\t\t\t\t\t// incremental reasoning event (with `delta` + accumulated `content`), so we\n\t\t\t\t\t// emit one per token. If TanStack AI adds a dedicated STEP_CONTENT event\n\t\t\t\t\t// type, this should be updated to emit STEP_CONTENT per token and a single\n\t\t\t\t\t// STEP_FINISHED when reasoning ends (i.e. when the first non-reasoning\n\t\t\t\t\t// content or finish_reason arrives).\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"STEP_FINISHED\",\n\t\t\t\t\t\tstepId,\n\t\t\t\t\t\tdelta: reasoningContent,\n\t\t\t\t\t\tcontent: accumulatedReasoning,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Text content\n\t\t\t\tif (delta.content) {\n\t\t\t\t\tif (!hasEmittedTextMessageStart) {\n\t\t\t\t\t\thasEmittedTextMessageStart = true;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_START\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\trole: \"assistant\",\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\taccumulatedContent += delta.content;\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_CONTENT\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tdelta: delta.content,\n\t\t\t\t\t\tcontent: accumulatedContent,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\t// Tool calls\n\t\t\t\tif (delta.tool_calls) {\n\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\tif (!toolCallsInProgress.has(index)) {\n\t\t\t\t\t\t\t// Always generate a unique ID per tool call index.\n\t\t\t\t\t\t\t// The backend may send the same ID for multiple tool calls,\n\t\t\t\t\t\t\t// so we cannot trust toolCallDelta.id to be unique.\n\t\t\t\t\t\t\ttoolCallsInProgress.set(index, {\n\t\t\t\t\t\t\t\tid: generateId(\"chatcmpl-tool\"),\n\t\t\t\t\t\t\t\tname: toolCallDelta.function?.name || \"\",\n\t\t\t\t\t\t\t\targuments: \"\",\n\t\t\t\t\t\t\t\tstarted: false,\n\t\t\t\t\t\t\t});\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tconst toolCall = toolCallsInProgress.get(index)!;\n\n\t\t\t\t\t\t// Only update name if provided (ID is already set at creation time\n\t\t\t\t\t\t// and should not be overwritten by subsequent chunks that may have\n\t\t\t\t\t\t// duplicate/shared IDs from the backend)\n\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\ttoolCall.name = toolCallDelta.function.name;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\ttoolCall.arguments += toolCallDelta.function.arguments;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Emit TOOL_CALL_START once we have id and name\n\t\t\t\t\t\tif (toolCall.id && toolCall.name && !toolCall.started) {\n\t\t\t\t\t\t\ttoolCall.started = true;\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_START\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tindex,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Stream tool call arguments\n\t\t\t\t\t\tif (toolCallDelta.function?.arguments && toolCall.started) {\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_ARGS\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tdelta: toolCallDelta.function.arguments,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Finish\n\t\t\t\tif (choice.finish_reason) {\n\t\t\t\t\thasReceivedFinishReason = true;\n\n\t\t\t\t\t// End tool calls\n\t\t\t\t\tif (choice.finish_reason === \"tool_calls\" || toolCallsInProgress.size > 0) {\n\t\t\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\t\tparsedInput = toolCall.arguments\n\t\t\t\t\t\t\t\t\t? JSON.parse(toolCall.arguments)\n\t\t\t\t\t\t\t\t\t: {};\n\t\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\n\t\t\t\t\tconst computedFinishReason =\n\t\t\t\t\t\tchoice.finish_reason === \"tool_calls\" ||\n\t\t\t\t\t\tchoice.finish_reason === \"function_call\" ||\n\t\t\t\t\t\ttoolCallsInProgress.size > 0\n\t\t\t\t\t\t\t? \"tool_calls\"\n\t\t\t\t\t\t\t: (choice.finish_reason as \"stop\" | \"length\" | \"content_filter\");\n\n\t\t\t\t\t// End text message if started\n\t\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\n\t\t\t\t\t// Emit RUN_FINISHED\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\t\trunId,\n\t\t\t\t\t\tmodel: chunk.model || model || this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\tusage: chunk.usage\n\t\t\t\t\t\t\t? {\n\t\t\t\t\t\t\t\t\tpromptTokens: chunk.usage.prompt_tokens,\n\t\t\t\t\t\t\t\t\tcompletionTokens: chunk.usage.completion_tokens,\n\t\t\t\t\t\t\t\t\ttotalTokens: chunk.usage.total_tokens,\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t: undefined,\n\t\t\t\t\t\tfinishReason: computedFinishReason,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Premature stream termination: the stream ended without a finish_reason.\n\t\t\t// This can happen when Workers AI truncates a response or the connection drops.\n\t\t\t// Emit proper closing events so the consumer doesn't hang.\n\t\t\tif (hasEmittedRunStarted && !hasReceivedFinishReason) {\n\t\t\t\tconsole.warn(\n\t\t\t\t\t\"[tanstack-ai] Stream ended without finish_reason — possible truncation or connection drop\",\n\t\t\t\t);\n\n\t\t\t\t// Close any open tool calls\n\t\t\t\tfor (const [, toolCall] of toolCallsInProgress) {\n\t\t\t\t\tif (toolCall.started) {\n\t\t\t\t\t\tlet parsedInput: unknown = {};\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tparsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tparsedInput = {};\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\ttype: \"TOOL_CALL_END\",\n\t\t\t\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\t\t\t\ttoolName: toolCall.name,\n\t\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t\t\tinput: parsedInput,\n\t\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Close text message if open\n\t\t\t\tif (hasEmittedTextMessageStart) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\ttype: \"TEXT_MESSAGE_END\",\n\t\t\t\t\t\tmessageId,\n\t\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\t\ttimestamp,\n\t\t\t\t\t} satisfies StreamChunk;\n\t\t\t\t}\n\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_FINISHED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t\tfinishReason: \"stop\",\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tconst code =\n\t\t\t\terror instanceof Error ? (error as Error & { code?: string }).code : undefined;\n\t\t\tif (!hasEmittedRunStarted) {\n\t\t\t\tyield {\n\t\t\t\t\ttype: \"RUN_STARTED\",\n\t\t\t\t\trunId,\n\t\t\t\t\tmodel: model ?? this.model,\n\t\t\t\t\ttimestamp,\n\t\t\t\t} satisfies StreamChunk;\n\t\t\t}\n\t\t\tyield {\n\t\t\t\ttype: \"RUN_ERROR\",\n\t\t\t\trunId,\n\t\t\t\tmodel: model ?? this.model,\n\t\t\t\ttimestamp,\n\t\t\t\terror: {\n\t\t\t\t\tmessage: message || \"Unknown error\",\n\t\t\t\t\tcode,\n\t\t\t\t},\n\t\t\t} satisfies StreamChunk;\n\t\t}\n\t}\n\n\tasync structuredOutput(\n\t\toptions: StructuredOutputOptions<any>,\n\t): Promise<StructuredOutputResult<unknown>> {\n\t\tconst { outputSchema, chatOptions } = options;\n\t\tconst { systemPrompts, messages, temperature, model } = chatOptions;\n\n\t\tconst openAIMessages = buildOpenAIMessages(systemPrompts, messages, {\n\t\t\tincludeToolMessages: false,\n\t\t});\n\n\t\tconst response = await this.client.chat.completions.create({\n\t\t\tmodel: model ?? this.model,\n\t\t\tmessages: openAIMessages,\n\t\t\ttemperature,\n\t\t\tstream: false,\n\t\t\tresponse_format: {\n\t\t\t\ttype: \"json_schema\",\n\t\t\t\tjson_schema: {\n\t\t\t\t\tname: \"structured_output\",\n\t\t\t\t\tstrict: true,\n\t\t\t\t\tschema: outputSchema,\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst choice = response.choices?.[0];\n\n\t\tif (!choice) {\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI structured output returned no choices: ${JSON.stringify(response)}`,\n\t\t\t);\n\t\t}\n\n\t\tconst rawContent = choice.message?.content ?? \"\";\n\n\t\t// Workers AI REST endpoint may return `content` as an already-parsed object\n\t\t// when using json_schema response format, so normalise both cases.\n\t\tlet data: unknown;\n\t\tlet rawText: string;\n\n\t\tif (typeof rawContent === \"string\") {\n\t\t\trawText = rawContent;\n\t\t\ttry {\n\t\t\t\tdata = JSON.parse(rawText);\n\t\t\t} catch {\n\t\t\t\tdata = rawText;\n\t\t\t}\n\t\t} else {\n\t\t\t// Already an object — stringify for rawText, use directly for data\n\t\t\tdata = rawContent;\n\t\t\trawText = JSON.stringify(rawContent);\n\t\t}\n\n\t\treturn { data, rawText };\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\nexport function createWorkersAiChat(model: WorkersAiTextModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiTextAdapter(model, config);\n}\n"],"mappings":";;;;;;;;;;;;AAEA;AAAA,EACC;AAAA,OAGM;AACP,OAAO,YAAY;AAyBnB,SAAS,qBAAqB,QAAwC;AACrE,0BAAwB,MAAM;AAE9B,QAAM,iBAAqD,OAAO,kBAC/D,EAAE,sBAAsB,OAAO,gBAAgB,IAC/C;AAEH,MAAI,sBAAsB,MAAM,GAAG;AAElC,WAAO,IAAI,OAAO;AAAA,MACjB,QAAQ;AAAA,MACR,OAAO;AAAA,QACN,OAAO;AAAA,QACP,iBAAiB,EAAE,cAAc,eAAe,IAAI;AAAA,MACrD;AAAA,IACD,CAAC;AAAA,EACF;AAEA,MAAI,0BAA0B,MAAM,GAAG;AAEtC,WAAO,IAAI,OAAO;AAAA,MACjB,SAAS,iDAAiD,OAAO,SAAS;AAAA,MAC1E,QAAQ,OAAO;AAAA,MACf,gBAAgB;AAAA,IACjB,CAAC;AAAA,EACF;AAGA,QAAM,gBAAgB;AACtB,SAAO,IAAI,OAAO;AAAA,IACjB,OAAO,mBAAmB,cAAc,eAAe,cAAc;AAAA,IACrE,QAAQ,cAAc,UAAU;AAAA,EACjC,CAAC;AACF;AAMA,SAAS,mBAAmB,SAA0C;AACrE,MAAI,YAAY,KAAM,QAAO;AAC7B,MAAI,OAAO,YAAY,SAAU,QAAO;AACxC,SAAO,QAAQ,QAAQ,CAAC,MAAO,EAAE,SAAS,SAAS,CAAC,EAAE,OAAO,IAAI,CAAC,CAAE,EAAE,KAAK,EAAE;AAC9E;AAQA,SAAS,mBAAmB,MAAsE;AACjG,UAAQ,KAAK,MAAM;AAAA,IAClB,KAAK;AACJ,UAAI,KAAK,SAAS;AACjB,eAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,QAAQ;AAAA,MAC3C;AACA,aAAO;AAAA,IACR,KAAK,SAAS;AACb,UAAI;AACJ,UAAI,KAAK,OAAO,SAAS,QAAQ;AAChC,cAAM,KAAK,OAAO,MAAM,WAAW,OAAO,IACvC,KAAK,OAAO,QACZ,QAAQ,KAAK,OAAO,QAAQ,WAAW,KAAK,OAAO,KAAK;AAAA,MAC5D,OAAO;AACN,cAAM,KAAK,OAAO;AAAA,MACnB;AACA,aAAO,EAAE,MAAM,aAAa,WAAW,EAAE,IAAI,EAAE;AAAA,IAChD;AAAA,IACA;AAEC,cAAQ;AAAA,QACP,4DAA4D,KAAK,IAAI;AAAA,MACtE;AACA,aAAO;AAAA,EACT;AACD;AASA,SAAS,iBACR,SACmD;AACnD,MAAI,YAAY,KAAM,QAAO;AAC7B,MAAI,OAAO,YAAY,SAAU,QAAO;AAExC,QAAM,YAAY,QAAQ,KAAK,CAAC,MAAM,EAAE,SAAS,OAAO;AACxD,MAAI,CAAC,WAAW;AACf,WAAO,QAAQ,QAAQ,CAAC,MAAO,EAAE,SAAS,SAAS,CAAC,EAAE,OAAO,IAAI,CAAC,CAAE,EAAE,KAAK,EAAE;AAAA,EAC9E;AAEA,QAAM,QAAiD,CAAC;AACxD,aAAW,QAAQ,SAAS;AAC3B,UAAM,YAAY,mBAAmB,IAAI;AACzC,QAAI,WAAW;AACd,YAAM,KAAK,SAAS;AAAA,IACrB;AAAA,EACD;AACA,SAAO;AACR;AAEA,SAAS,oBACR,eACA,UACA,SAC2C;AAC3C,QAAM,eAAe,SAAS,uBAAuB;AACrD,QAAM,iBAA2D,CAAC;AAElE,MAAI,iBAAiB,cAAc,SAAS,GAAG;AAC9C,mBAAe,KAAK;AAAA,MACnB,MAAM;AAAA,MACN,SAAS,cAAc,KAAK,IAAI;AAAA,IACjC,CAAC;AAAA,EACF;AAEA,aAAW,WAAW,UAAU;AAC/B,QAAI,QAAQ,SAAS,QAAQ;AAC5B,qBAAe,KAAK;AAAA,QACnB,MAAM;AAAA,QACN,SAAS,iBAAiB,QAAQ,OAAO;AAAA,MAC1C,CAAC;AAAA,IACF,WAAW,QAAQ,SAAS,aAAa;AACxC,YAAM,mBAAoE;AAAA,QACzE,MAAM;AAAA,QACN,SAAS,mBAAmB,QAAQ,OAAO;AAAA,MAC5C;AACA,UAAI,gBAAgB,QAAQ,aAAa,QAAQ,UAAU,SAAS,GAAG;AACtE,yBAAiB,aAAa,QAAQ,UAAU,IAAI,CAAC,QAAQ;AAAA,UAC5D,IAAI,GAAG;AAAA,UACP,MAAM;AAAA,UACN,UAAU;AAAA,YACT,MAAM,GAAG,SAAS;AAAA,YAClB,WAAW,GAAG,SAAS;AAAA,UACxB;AAAA,QACD,EAAE;AAAA,MACH;AACA,qBAAe,KAAK,gBAAgB;AAAA,IACrC,WAAW,gBAAgB,QAAQ,SAAS,QAAQ;AACnD,UAAI;AACJ,UAAI,OAAO,QAAQ,YAAY,UAAU;AACxC,YAAI;AACH,eAAK,MAAM,QAAQ,OAAO;AAC1B,wBAAc,QAAQ;AAAA,QACvB,QAAQ;AACP,wBAAc,KAAK,UAAU,QAAQ,OAAO;AAAA,QAC7C;AAAA,MACD,OAAO;AACN,sBAAc,KAAK,UAAU,QAAQ,OAAO;AAAA,MAC7C;AACA,qBAAe,KAAK;AAAA,QACnB,MAAM;AAAA,QACN,cAAc,QAAQ,cAAc,QAAQ,OAAO,WAAW,EAAE,MAAM,GAAG,CAAC,CAAC;AAAA,QAC3E,SAAS;AAAA,MACV,CAAC;AAAA,IACF;AAAA,EACD;AAEA,SAAO;AACR;AAEA,SAAS,iBACR,OAC+C;AAC/C,MAAI,CAAC,MAAO,QAAO;AACnB,SAAO,MAAM,IAAI,CAAC,UAAU;AAAA,IAC3B,MAAM;AAAA,IACN,UAAU;AAAA,MACT,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,YAAY,KAAK;AAAA,IAClB;AAAA,EACD,EAAE;AACH;AAMA,SAAS,WAAW,SAAS,YAAoB;AAChD,SAAO,GAAG,MAAM,IAAI,OAAO,WAAW,EAAE,QAAQ,MAAM,EAAE,EAAE,MAAM,GAAG,EAAE,CAAC;AACvE;AASO,IAAM,uBAAN,cAAsE,gBAM3E;AAAA,EAKD,YAAY,OAAe,QAAgC;AAC1D,UAAM,EAAE,QAAQ,SAAS,GAAG,KAAK;AALlC,gCAAO;AAEP,wBAAQ;AAIP,SAAK,SAAS,qBAAqB,MAAM;AAAA,EAC1C;AAAA,EAEA,OAAO,WAAW,SAAuD;AACxE,UAAM,EAAE,eAAe,UAAU,OAAO,aAAa,MAAM,IAAI;AAE/D,UAAM,iBAAiB,oBAAoB,eAAe,QAAQ;AAClE,UAAM,cAAc,iBAAiB,KAAK;AAE1C,UAAM,YAAY,KAAK,IAAI;AAC3B,UAAM,QAAQ,WAAW;AACzB,UAAM,YAAY,WAAW;AAC7B,QAAI,uBAAuB;AAC3B,QAAI,6BAA6B;AACjC,QAAI,qBAAqB;AACzB,QAAI,wBAAwB;AAC5B,QAAI,uBAAuB;AAC3B,UAAM,SAAS,WAAW;AAC1B,QAAI,0BAA0B;AAC9B,UAAM,sBAAsB,oBAAI,IAG9B;AAEF,QAAI;AACH,UAAI;AACJ,UAAI;AACH,iBAAS,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,UAClD,OAAO,SAAS,KAAK;AAAA,UACrB,UAAU;AAAA,UACV,OAAO;AAAA,UACP;AAAA,UACA,QAAQ;AAAA,UACR,gBAAgB,EAAE,eAAe,KAAK;AAAA,QACvC,CAAC;AAAA,MACF,SAAS,aAAsB;AAG9B,gBAAQ;AAAA,UACP;AAAA,UACA,uBAAuB,QAAQ,YAAY,UAAU;AAAA,QACtD;AACA,cAAM,kBAAkB,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,UACjE,OAAO,SAAS,KAAK;AAAA,UACrB,UAAU;AAAA,UACV,OAAO;AAAA,UACP;AAAA,QACD,CAAC;AAED,cAAM;AAAA,UACL,MAAM;AAAA,UACN;AAAA,UACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;AAAA,UAC9C;AAAA,QACD;AAEA,cAAM,MAAM,gBAAgB,QAAQ,CAAC,GAAG;AACxC,YAAI,KAAK,SAAS;AACjB,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;AAAA,YAC9C;AAAA,YACA,MAAM;AAAA,UACP;AACA,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;AAAA,YAC9C;AAAA,YACA,OAAO,IAAI;AAAA,YACX,SAAS,IAAI;AAAA,UACd;AACA,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;AAAA,YAC9C;AAAA,UACD;AAAA,QACD;AAEA,YAAI,KAAK,YAAY;AACpB,qBAAW,MAAM,IAAI,YAAY;AAChC,gBAAI,GAAG,SAAS,WAAY;AAC5B,kBAAM,KAAK,GAAG;AACd,gBAAI,cAAuB,CAAC;AAC5B,gBAAI;AACH,4BAAc,GAAG,YAAY,KAAK,MAAM,GAAG,SAAS,IAAI,CAAC;AAAA,YAC1D,QAAQ;AACP,4BAAc,CAAC;AAAA,YAChB;AACA,kBAAM;AAAA,cACL,MAAM;AAAA,cACN,YAAY,GAAG;AAAA,cACf,UAAU,GAAG;AAAA,cACb,OAAO,gBAAgB,SAAS,SAAS,KAAK;AAAA,cAC9C;AAAA,cACA,OAAO;AAAA,YACR;AACA,kBAAM;AAAA,cACL,MAAM;AAAA,cACN,YAAY,GAAG;AAAA,cACf,UAAU,GAAG;AAAA,cACb,OAAO,gBAAgB,SAAS,SAAS,KAAK;AAAA,cAC9C;AAAA,cACA,OAAO;AAAA,YACR;AAAA,UACD;AAAA,QACD;AAEA,cAAM,eAAe,gBAAgB,QAAQ,CAAC,GAAG;AACjD,cAAM;AAAA,UACL,MAAM;AAAA,UACN;AAAA,UACA,OAAO,gBAAgB,SAAS,SAAS,KAAK;AAAA,UAC9C;AAAA,UACA,OAAO,gBAAgB,QACpB;AAAA,YACA,cAAc,gBAAgB,MAAM;AAAA,YACpC,kBAAkB,gBAAgB,MAAM;AAAA,YACxC,aAAa,gBAAgB,MAAM;AAAA,UACpC,IACC;AAAA,UACH,cACC,iBAAiB,gBAAgB,iBAAiB,kBAC/C,eACE,gBAAyD;AAAA,QAChE;AACA;AAAA,MACD;AAEA,uBAAiB,SAAS,QAAQ;AACjC,YAAI,CAAC,MAAM,WAAW,MAAM,QAAQ,WAAW,EAAG;AAClD,cAAM,SAAS,MAAM,QAAQ,CAAC;AAC9B,YAAI,CAAC,OAAQ;AAGb,YAAI,CAAC,sBAAsB;AAC1B,iCAAuB;AACvB,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,YACpC;AAAA,UACD;AAAA,QACD;AAEA,cAAM,QAAQ,OAAO;AAIrB,cAAM,mBAAqB,MAAkC,qBAC3D,MAAkC;AACpC,YAAI,kBAAkB;AAErB,cAAI,CAAC,uBAAuB;AAC3B,oCAAwB;AACxB,kBAAM;AAAA,cACL,MAAM;AAAA,cACN;AAAA,cACA,UAAU;AAAA,cACV,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,cACpC;AAAA,YACD;AAAA,UACD;AACA,kCAAwB;AAOxB,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO;AAAA,YACP,SAAS;AAAA,YACT,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,YACpC;AAAA,UACD;AAAA,QACD;AAGA,YAAI,MAAM,SAAS;AAClB,cAAI,CAAC,4BAA4B;AAChC,yCAA6B;AAC7B,kBAAM;AAAA,cACL,MAAM;AAAA,cACN;AAAA,cACA,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,cACpC;AAAA,cACA,MAAM;AAAA,YACP;AAAA,UACD;AAEA,gCAAsB,MAAM;AAC5B,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,YACpC;AAAA,YACA,OAAO,MAAM;AAAA,YACb,SAAS;AAAA,UACV;AAAA,QACD;AAGA,YAAI,MAAM,YAAY;AACrB,qBAAW,iBAAiB,MAAM,YAAY;AAC7C,kBAAM,QAAQ,cAAc;AAE5B,gBAAI,CAAC,oBAAoB,IAAI,KAAK,GAAG;AAIpC,kCAAoB,IAAI,OAAO;AAAA,gBAC9B,IAAI,WAAW,eAAe;AAAA,gBAC9B,MAAM,cAAc,UAAU,QAAQ;AAAA,gBACtC,WAAW;AAAA,gBACX,SAAS;AAAA,cACV,CAAC;AAAA,YACF;AAEA,kBAAM,WAAW,oBAAoB,IAAI,KAAK;AAK9C,gBAAI,cAAc,UAAU,MAAM;AACjC,uBAAS,OAAO,cAAc,SAAS;AAAA,YACxC;AACA,gBAAI,cAAc,UAAU,WAAW;AACtC,uBAAS,aAAa,cAAc,SAAS;AAAA,YAC9C;AAGA,gBAAI,SAAS,MAAM,SAAS,QAAQ,CAAC,SAAS,SAAS;AACtD,uBAAS,UAAU;AACnB,oBAAM;AAAA,gBACL,MAAM;AAAA,gBACN,YAAY,SAAS;AAAA,gBACrB,UAAU,SAAS;AAAA,gBACnB,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,gBACpC;AAAA,gBACA;AAAA,cACD;AAAA,YACD;AAGA,gBAAI,cAAc,UAAU,aAAa,SAAS,SAAS;AAC1D,oBAAM;AAAA,gBACL,MAAM;AAAA,gBACN,YAAY,SAAS;AAAA,gBACrB,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,gBACpC;AAAA,gBACA,OAAO,cAAc,SAAS;AAAA,cAC/B;AAAA,YACD;AAAA,UACD;AAAA,QACD;AAGA,YAAI,OAAO,eAAe;AACzB,oCAA0B;AAG1B,cAAI,OAAO,kBAAkB,gBAAgB,oBAAoB,OAAO,GAAG;AAC1E,uBAAW,CAAC,EAAE,QAAQ,KAAK,qBAAqB;AAC/C,kBAAI,cAAuB,CAAC;AAC5B,kBAAI;AACH,8BAAc,SAAS,YACpB,KAAK,MAAM,SAAS,SAAS,IAC7B,CAAC;AAAA,cACL,QAAQ;AACP,8BAAc,CAAC;AAAA,cAChB;AACA,oBAAM;AAAA,gBACL,MAAM;AAAA,gBACN,YAAY,SAAS;AAAA,gBACrB,UAAU,SAAS;AAAA,gBACnB,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,gBACpC;AAAA,gBACA,OAAO;AAAA,cACR;AAAA,YACD;AAAA,UACD;AAEA,gBAAM,uBACL,OAAO,kBAAkB,gBACzB,OAAO,kBAAkB,mBACzB,oBAAoB,OAAO,IACxB,eACC,OAAO;AAGZ,cAAI,4BAA4B;AAC/B,kBAAM;AAAA,cACL,MAAM;AAAA,cACN;AAAA,cACA,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,cACpC;AAAA,YACD;AAAA,UACD;AAGA,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO,MAAM,SAAS,SAAS,KAAK;AAAA,YACpC;AAAA,YACA,OAAO,MAAM,QACV;AAAA,cACA,cAAc,MAAM,MAAM;AAAA,cAC1B,kBAAkB,MAAM,MAAM;AAAA,cAC9B,aAAa,MAAM,MAAM;AAAA,YAC1B,IACC;AAAA,YACH,cAAc;AAAA,UACf;AAAA,QACD;AAAA,MACD;AAKA,UAAI,wBAAwB,CAAC,yBAAyB;AACrD,gBAAQ;AAAA,UACP;AAAA,QACD;AAGA,mBAAW,CAAC,EAAE,QAAQ,KAAK,qBAAqB;AAC/C,cAAI,SAAS,SAAS;AACrB,gBAAI,cAAuB,CAAC;AAC5B,gBAAI;AACH,4BAAc,SAAS,YAAY,KAAK,MAAM,SAAS,SAAS,IAAI,CAAC;AAAA,YACtE,QAAQ;AACP,4BAAc,CAAC;AAAA,YAChB;AACA,kBAAM;AAAA,cACL,MAAM;AAAA,cACN,YAAY,SAAS;AAAA,cACrB,UAAU,SAAS;AAAA,cACnB,OAAO,SAAS,KAAK;AAAA,cACrB;AAAA,cACA,OAAO;AAAA,YACR;AAAA,UACD;AAAA,QACD;AAGA,YAAI,4BAA4B;AAC/B,gBAAM;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,OAAO,SAAS,KAAK;AAAA,YACrB;AAAA,UACD;AAAA,QACD;AAEA,cAAM;AAAA,UACL,MAAM;AAAA,UACN;AAAA,UACA,OAAO,SAAS,KAAK;AAAA,UACrB;AAAA,UACA,cAAc;AAAA,QACf;AAAA,MACD;AAAA,IACD,SAAS,OAAO;AACf,YAAM,UAAU,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AACrE,YAAM,OACL,iBAAiB,QAAS,MAAoC,OAAO;AACtE,UAAI,CAAC,sBAAsB;AAC1B,cAAM;AAAA,UACL,MAAM;AAAA,UACN;AAAA,UACA,OAAO,SAAS,KAAK;AAAA,UACrB;AAAA,QACD;AAAA,MACD;AACA,YAAM;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,OAAO,SAAS,KAAK;AAAA,QACrB;AAAA,QACA,OAAO;AAAA,UACN,SAAS,WAAW;AAAA,UACpB;AAAA,QACD;AAAA,MACD;AAAA,IACD;AAAA,EACD;AAAA,EAEA,MAAM,iBACL,SAC2C;AAC3C,UAAM,EAAE,cAAc,YAAY,IAAI;AACtC,UAAM,EAAE,eAAe,UAAU,aAAa,MAAM,IAAI;AAExD,UAAM,iBAAiB,oBAAoB,eAAe,UAAU;AAAA,MACnE,qBAAqB;AAAA,IACtB,CAAC;AAED,UAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,MAC1D,OAAO,SAAS,KAAK;AAAA,MACrB,UAAU;AAAA,MACV;AAAA,MACA,QAAQ;AAAA,MACR,iBAAiB;AAAA,QAChB,MAAM;AAAA,QACN,aAAa;AAAA,UACZ,MAAM;AAAA,UACN,QAAQ;AAAA,UACR,QAAQ;AAAA,QACT;AAAA,MACD;AAAA,IACD,CAAC;AAED,UAAM,SAAS,SAAS,UAAU,CAAC;AAEnC,QAAI,CAAC,QAAQ;AACZ,YAAM,IAAI;AAAA,QACT,qDAAqD,KAAK,UAAU,QAAQ,CAAC;AAAA,MAC9E;AAAA,IACD;AAEA,UAAM,aAAa,OAAO,SAAS,WAAW;AAI9C,QAAI;AACJ,QAAI;AAEJ,QAAI,OAAO,eAAe,UAAU;AACnC,gBAAU;AACV,UAAI;AACH,eAAO,KAAK,MAAM,OAAO;AAAA,MAC1B,QAAQ;AACP,eAAO;AAAA,MACR;AAAA,IACD,OAAO;AAEN,aAAO;AACP,gBAAU,KAAK,UAAU,UAAU;AAAA,IACpC;AAEA,WAAO,EAAE,MAAM,QAAQ;AAAA,EACxB;AACD;AAMO,SAAS,oBAAoB,OAA2B,QAAgC;AAC9F,SAAO,IAAI,qBAAqB,OAAO,MAAM;AAC9C;","names":[]}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import {
|
|
2
2
|
createGatewayFetch
|
|
3
|
-
} from "./chunk-
|
|
3
|
+
} from "./chunk-727MRLUF.js";
|
|
4
4
|
|
|
5
5
|
// src/adapters/grok.ts
|
|
6
6
|
import {
|
|
@@ -45,4 +45,4 @@ export {
|
|
|
45
45
|
createGrokImage,
|
|
46
46
|
createGrokSummarize
|
|
47
47
|
};
|
|
48
|
-
//# sourceMappingURL=chunk-
|
|
48
|
+
//# sourceMappingURL=chunk-HFZK4C5B.js.map
|
|
@@ -10,7 +10,7 @@ var _chunk7HSUHP63cjs = require('./chunk-7HSUHP63.cjs');
|
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
|
|
13
|
-
var
|
|
13
|
+
var _chunk53RO5I22cjs = require('./chunk-53RO5I22.cjs');
|
|
14
14
|
|
|
15
15
|
|
|
16
16
|
var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
|
|
@@ -22,7 +22,7 @@ var WorkersAiImageAdapter = class extends _adapters.BaseImageAdapter {
|
|
|
22
22
|
super({}, model);
|
|
23
23
|
_chunk4DE2IREAcjs.__publicField.call(void 0, this, "name", "workers-ai-image");
|
|
24
24
|
_chunk4DE2IREAcjs.__publicField.call(void 0, this, "adapterConfig");
|
|
25
|
-
|
|
25
|
+
_chunk53RO5I22cjs.validateWorkersAiConfig.call(void 0, config);
|
|
26
26
|
this.adapterConfig = config;
|
|
27
27
|
}
|
|
28
28
|
async generateImages(options) {
|
|
@@ -33,10 +33,10 @@ var WorkersAiImageAdapter = class extends _adapters.BaseImageAdapter {
|
|
|
33
33
|
if (w) extra.width = Number(w);
|
|
34
34
|
if (h) extra.height = Number(h);
|
|
35
35
|
}
|
|
36
|
-
if (
|
|
36
|
+
if (_chunk53RO5I22cjs.isDirectBindingConfig.call(void 0, this.adapterConfig)) {
|
|
37
37
|
return this.generateViaBinding(prompt, extra);
|
|
38
38
|
}
|
|
39
|
-
if (
|
|
39
|
+
if (_chunk53RO5I22cjs.isDirectCredentialsConfig.call(void 0, this.adapterConfig)) {
|
|
40
40
|
return this.generateViaRest(prompt, extra);
|
|
41
41
|
}
|
|
42
42
|
return this.generateViaGateway(prompt, extra);
|
|
@@ -60,7 +60,7 @@ var WorkersAiImageAdapter = class extends _adapters.BaseImageAdapter {
|
|
|
60
60
|
}
|
|
61
61
|
async generateViaGateway(prompt, options) {
|
|
62
62
|
const gatewayConfig = this.adapterConfig;
|
|
63
|
-
const gatewayFetch =
|
|
63
|
+
const gatewayFetch = _chunk53RO5I22cjs.createGatewayFetch.call(void 0, "workers-ai", gatewayConfig);
|
|
64
64
|
const response = await gatewayFetch("https://api.cloudflare.com/v1/images/generations", {
|
|
65
65
|
method: "POST",
|
|
66
66
|
body: JSON.stringify({
|
|
@@ -95,4 +95,4 @@ function createWorkersAiImage(model, config) {
|
|
|
95
95
|
|
|
96
96
|
|
|
97
97
|
exports.WorkersAiImageAdapter = WorkersAiImageAdapter; exports.createWorkersAiImage = createWorkersAiImage;
|
|
98
|
-
//# sourceMappingURL=chunk-
|
|
98
|
+
//# sourceMappingURL=chunk-IL6PTKJE.cjs.map
|