@cloudflare/tanstack-ai 0.1.5 → 0.1.7

package/dist/chunk-3L6IRVAV.cjs

@@ -1,327 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }// src/utils/create-fetcher.ts
-function isDirectBindingConfig(config) {
-  return "binding" in config && typeof config.binding.gateway === "function";
-}
-function isDirectCredentialsConfig(config) {
-  return "accountId" in config && "apiKey" in config && !("gatewayId" in config);
-}
-function isGatewayConfig(config) {
-  if ("gatewayId" in config) return true;
-  return "binding" in config && !isDirectBindingConfig(config);
-}
-function validateWorkersAiConfig(config) {
-  if (!isDirectBindingConfig(config) && !isDirectCredentialsConfig(config) && !isGatewayConfig(config)) {
-    throw new Error(
-      "Invalid Workers AI configuration: you must provide either a binding (e.g. { binding: env.AI }), credentials ({ accountId, apiKey }), or a gateway configuration ({ binding: env.AI.gateway(id) } or { accountId, gatewayId })."
-    );
-  }
-}
-function createGatewayFetch(provider, config, headers = {}) {
-  return (input, init) => {
-    let query = {};
-    const url = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
-    const urlObj = new URL(url);
-    const endpoint = urlObj.pathname.replace(/^\/v1\//, "").replace(/^\//, "") + urlObj.search;
-    if (_optionalChain([init, 'optionalAccess', _ => _.body])) {
-      try {
-        query = JSON.parse(init.body);
-      } catch (e2) {
-        query = { _raw: init.body };
-      }
-    }
-    const cacheHeaders = {};
-    if ("skipCache" in config && config.skipCache) {
-      cacheHeaders["cf-aig-skip-cache"] = "true";
-    }
-    if (typeof config.cacheTtl === "number") {
-      cacheHeaders["cf-aig-cache-ttl"] = String(config.cacheTtl);
-    }
-    if (typeof config.customCacheKey === "string") {
-      cacheHeaders["cf-aig-cache-key"] = config.customCacheKey;
-    }
-    if (typeof config.metadata === "object") {
-      cacheHeaders["cf-aig-metadata"] = JSON.stringify(config.metadata);
-    }
-    const request = {
-      provider,
-      endpoint,
-      headers: {
-        ..._optionalChain([init, 'optionalAccess', _2 => _2.headers]),
-        ...headers,
-        ...cacheHeaders,
-        "Content-Type": "application/json"
-      },
-      query
-    };
-    if (provider === "workers-ai") {
-      if (!request.endpoint.startsWith("run/")) {
-        request.endpoint = `run/${query.model}`;
-      }
-      delete query.model;
-      delete query.instructions;
-    }
-    if (config.apiKey) {
-      request.headers["authorization"] = `Bearer ${config.apiKey}`;
-    }
-    if ("binding" in config) {
-      return config.binding.run(request, { signal: _nullishCoalesce(_optionalChain([init, 'optionalAccess', _3 => _3.signal]), () => ( void 0)) });
-    }
-    return fetch(
-      `https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}`,
-      {
-        ...init,
-        headers: {
-          "Content-Type": "application/json",
-          ...headers,
-          ...cacheHeaders,
-          ...config.cfApiKey ? { "cf-aig-authorization": `Bearer ${config.cfApiKey}` } : {}
-        },
-        body: JSON.stringify(request)
-      }
-    );
-  };
-}
-function normalizeMessagesForBinding(messages) {
-  return messages.map((msg) => {
-    const normalized = { ...msg };
-    if (normalized.content === null || normalized.content === void 0) {
-      normalized.content = "";
-    }
-    return normalized;
-  });
-}
-function createWorkersAiBindingFetch(binding, options) {
-  return async (_input, init) => {
-    if (!_optionalChain([init, 'optionalAccess', _4 => _4.body])) {
-      return new Response("No body", { status: 400 });
-    }
-    let body;
-    try {
-      body = JSON.parse(init.body);
-    } catch (e3) {
-      return new Response("Invalid JSON body", { status: 400 });
-    }
-    const model = body.model;
-    const stream = body.stream;
-    const inputs = {};
-    if (body.messages) {
-      inputs.messages = normalizeMessagesForBinding(
-        body.messages
-      );
-    }
-    if (body.tools) inputs.tools = body.tools;
-    if (typeof body.temperature === "number") inputs.temperature = body.temperature;
-    if (typeof body.max_tokens === "number") inputs.max_tokens = body.max_tokens;
-    if (body.response_format) inputs.response_format = body.response_format;
-    if (stream) inputs.stream = true;
-    const runOptions = {};
-    if (_optionalChain([options, 'optionalAccess', _5 => _5.extraHeaders])) runOptions.extraHeaders = options.extraHeaders;
-    if (_optionalChain([init, 'optionalAccess', _6 => _6.signal])) runOptions.signal = init.signal;
-    const result = await binding.run(
-      model,
-      inputs,
-      Object.keys(runOptions).length > 0 ? runOptions : void 0
-    );
-    if (stream && result instanceof ReadableStream) {
-      const transformed = transformWorkersAiStream(
-        result,
-        model
-      );
-      return new Response(transformed, {
-        headers: {
-          "Content-Type": "text/event-stream",
-          "Cache-Control": "no-cache"
-        }
-      });
-    }
-    const responseObj = typeof result === "object" && result !== null ? result : { response: String(result) };
-    const responseText = typeof responseObj.response === "string" ? responseObj.response : "";
-    const message = {
-      role: "assistant",
-      content: responseText
-    };
-    let finishReason = "stop";
-    if (Array.isArray(responseObj.tool_calls) && responseObj.tool_calls.length > 0) {
-      finishReason = "tool_calls";
-      message.tool_calls = responseObj.tool_calls.map(
-        (tc) => ({
-          id: tc.id || crypto.randomUUID(),
-          type: "function",
-          function: {
-            name: _optionalChain([tc, 'access', _7 => _7.function, 'optionalAccess', _8 => _8.name]) || tc.name || "",
-            arguments: typeof (_nullishCoalesce(_optionalChain([tc, 'access', _9 => _9.function, 'optionalAccess', _10 => _10.arguments]), () => ( tc.arguments))) === "string" ? _nullishCoalesce(_optionalChain([tc, 'access', _11 => _11.function, 'optionalAccess', _12 => _12.arguments]), () => ( tc.arguments)) : JSON.stringify(_nullishCoalesce(_nullishCoalesce(_optionalChain([tc, 'access', _13 => _13.function, 'optionalAccess', _14 => _14.arguments]), () => ( tc.arguments)), () => ( {})))
-          }
-        })
-      );
-    }
-    const openAiResponse = {
-      id: `workers-ai-${crypto.randomUUID()}`,
-      object: "chat.completion",
-      created: Math.floor(Date.now() / 1e3),
-      model,
-      choices: [{ index: 0, message, finish_reason: finishReason }]
-    };
-    return new Response(JSON.stringify(openAiResponse), {
-      headers: { "Content-Type": "application/json" }
-    });
-  };
-}
-function transformWorkersAiStream(source, model) {
-  const decoder = new TextDecoder();
-  const encoder = new TextEncoder();
-  const streamId = `workers-ai-${crypto.randomUUID()}`;
-  const created = Math.floor(Date.now() / 1e3);
-  let buffer = "";
-  let hasToolCalls = false;
-  let isOpenAiFormat = false;
-  const toolCallState = /* @__PURE__ */ new Map();
-  return source.pipeThrough(
-    new TransformStream({
-      transform(chunk, controller) {
-        buffer += decoder.decode(chunk, { stream: true });
-        const lines = buffer.split("\n");
-        buffer = lines.pop() || "";
-        for (const line of lines) {
-          const trimmed = line.trim();
-          if (!trimmed || !trimmed.startsWith("data: ")) continue;
-          const data = trimmed.slice(6);
-          if (data === "[DONE]") continue;
-          try {
-            const parsed = JSON.parse(data);
-            if (parsed.choices !== void 0) {
-              isOpenAiFormat = true;
-              const choice = _optionalChain([parsed, 'access', _15 => _15.choices, 'optionalAccess', _16 => _16[0]]);
-              if (_optionalChain([choice, 'optionalAccess', _17 => _17.delta, 'optionalAccess', _18 => _18.tool_calls])) {
-                hasToolCalls = true;
-                for (const tc of choice.delta.tool_calls) {
-                  const tcIndex = _nullishCoalesce(tc.index, () => ( 0));
-                  if (!toolCallState.has(tcIndex)) {
-                    const id = tc.id || `call${streamId}${tcIndex}`;
-                    toolCallState.set(tcIndex, {
-                      id,
-                      name: _optionalChain([tc, 'access', _19 => _19.function, 'optionalAccess', _20 => _20.name]) || ""
-                    });
-                    tc.id = id;
-                  } else {
-                    delete tc.id;
-                  }
-                }
-              }
-              if (_optionalChain([choice, 'optionalAccess', _21 => _21.finish_reason]) === "tool_calls") {
-                hasToolCalls = true;
-              }
-              controller.enqueue(
-                encoder.encode(`data: ${JSON.stringify(parsed)}
-
-`)
-              );
-              continue;
-            }
-            if (parsed.response != null && parsed.response !== "") {
-              const openAiChunk = {
-                id: streamId,
-                object: "chat.completion.chunk",
-                created,
-                model,
-                choices: [
-                  {
-                    index: 0,
-                    delta: { content: parsed.response },
-                    finish_reason: null
-                  }
-                ]
-              };
-              controller.enqueue(
-                encoder.encode(`data: ${JSON.stringify(openAiChunk)}
-
-`)
-              );
-            }
-            if (Array.isArray(parsed.tool_calls) && parsed.tool_calls.length > 0) {
-              for (const tc of parsed.tool_calls) {
-                const tcIndex = _nullishCoalesce(tc.index, () => ( 0));
-                const tcName = _nullishCoalesce(_nullishCoalesce(_optionalChain([tc, 'access', _22 => _22.function, 'optionalAccess', _23 => _23.name]), () => ( tc.name)), () => ( null));
-                const tcArgs = _nullishCoalesce(_nullishCoalesce(_optionalChain([tc, 'access', _24 => _24.function, 'optionalAccess', _25 => _25.arguments]), () => ( tc.arguments)), () => ( null));
-                const tcId = _nullishCoalesce(tc.id, () => ( null));
-                if (!tcId && !tcName && (!tcArgs || tcArgs === "")) continue;
-                hasToolCalls = true;
-                const toolCallDelta = {
-                  index: tcIndex
-                };
-                if (!toolCallState.has(tcIndex)) {
-                  const id = tcId || `call${streamId}${tcIndex}`;
-                  toolCallState.set(tcIndex, { id, name: tcName || "" });
-                  toolCallDelta.id = id;
-                  toolCallDelta.type = "function";
-                  toolCallDelta.function = {
-                    name: tcName || "",
-                    // Include arguments if they arrive in the same chunk
-                    arguments: tcArgs != null ? typeof tcArgs === "string" ? tcArgs : JSON.stringify(tcArgs) : ""
-                  };
-                } else {
-                  if (tcArgs != null && tcArgs !== "") {
-                    toolCallDelta.function = {
-                      arguments: typeof tcArgs === "string" ? tcArgs : JSON.stringify(tcArgs)
-                    };
-                  } else {
-                    continue;
-                  }
-                }
-                const toolChunk = {
-                  id: streamId,
-                  object: "chat.completion.chunk",
-                  created,
-                  model,
-                  choices: [
-                    {
-                      index: 0,
-                      delta: { tool_calls: [toolCallDelta] },
-                      finish_reason: null
-                    }
-                  ]
-                };
-                controller.enqueue(
-                  encoder.encode(`data: ${JSON.stringify(toolChunk)}
-
-`)
-                );
-              }
-            }
-          } catch (e) {
-            console.warn("[tanstack-ai] failed to parse SSE event:", data, e);
-          }
-        }
-      },
-      flush(controller) {
-        if (!isOpenAiFormat) {
-          const finalChunk = {
-            id: streamId,
-            object: "chat.completion.chunk",
-            created,
-            model,
-            choices: [
-              {
-                index: 0,
-                delta: {},
-                finish_reason: hasToolCalls ? "tool_calls" : "stop"
-              }
-            ]
-          };
-          controller.enqueue(encoder.encode(`data: ${JSON.stringify(finalChunk)}
-
-`));
-        }
-        controller.enqueue(encoder.encode("data: [DONE]\n\n"));
-      }
-    })
-  );
-}
-
-
-
-
-
-
-
-exports.isDirectBindingConfig = isDirectBindingConfig; exports.isDirectCredentialsConfig = isDirectCredentialsConfig; exports.validateWorkersAiConfig = validateWorkersAiConfig; exports.createGatewayFetch = createGatewayFetch; exports.createWorkersAiBindingFetch = createWorkersAiBindingFetch;
-//# sourceMappingURL=chunk-3L6IRVAV.cjs.map

package/dist/chunk-3L6IRVAV.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-3L6IRVAV.cjs","../src/utils/create-fetcher.ts"],"names":[],"mappings":"AAAA;AC4HO,SAAS,qBAAA,CACf,MAAA,EACyC;AAGzC,EAAA,OACC,UAAA,GAAa,OAAA,GACb,OAAQ,MAAA,CAAO,OAAA,CAA+C,QAAA,IAAY,UAAA;AAE5E;AAGO,SAAS,yBAAA,CACf,MAAA,EAC6C;AAC7C,EAAA,OAAO,YAAA,GAAe,OAAA,GAAU,SAAA,GAAY,OAAA,GAAU,CAAA,CAAE,YAAA,GAAe,MAAA,CAAA;AACxE;AAGO,SAAS,eAAA,CAAgB,MAAA,EAAkE;AACjG,EAAA,GAAA,CAAI,YAAA,GAAe,MAAA,EAAQ,OAAO,IAAA;AAElC,EAAA,OAAO,UAAA,GAAa,OAAA,GAAU,CAAC,qBAAA,CAAsB,MAAM,CAAA;AAC5D;AAWO,SAAS,uBAAA,CAAwB,MAAA,EAAsC;AAC7E,EAAA,GAAA,CACC,CAAC,qBAAA,CAAsB,MAAM,EAAA,GAC7B,CAAC,yBAAA,CAA0B,MAAM,EAAA,GACjC,CAAC,eAAA,CAAgB,MAAM,CAAA,EACtB;AACD,IAAA,MAAM,IAAI,KAAA;AAAA,MACT;AAAA,IAGD,CAAA;AAAA,EACD;AACD;AAMO,SAAS,kBAAA,CACf,QAAA,EACA,MAAA,EACA,QAAA,EAAkC,CAAC,CAAA,EACpB;AACf,EAAA,OAAO,CAAC,KAAA,EAAO,IAAA,EAAA,GAAS;AACvB,IAAA,IAAI,MAAA,EAAiC,CAAC,CAAA;AAEtC,IAAA,MAAM,IAAA,EACL,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,MAAA,WAAiB,IAAA,EAAM,KAAA,CAAM,KAAA,EAAO,KAAA,CAAM,GAAA;AAC/E,IAAA,MAAM,OAAA,EAAS,IAAI,GAAA,CAAI,GAAG,CAAA;AAG1B,IAAA,MAAM,SAAA,EAAW,MAAA,CAAO,QAAA,CAAS,OAAA,CAAQ,SAAA,EAAW,EAAE,CAAA,CAAE,OAAA,CAAQ,KAAA,EAAO,EAAE,EAAA,EAAI,MAAA,CAAO,MAAA;AAEpF,IAAA,GAAA,iBAAI,IAAA,2BAAM,MAAA,EAAM;AACf,MAAA,IAAI;AACH,QAAA,MAAA,EAAQ,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,IAAc,CAAA;AAAA,MACvC,EAAA,WAAQ;AACP,QAAA,MAAA,EAAQ,EAAE,IAAA,EAAM,IAAA,CAAK,KAAK,CAAA;AAAA,MAC3B;AAAA,IACD;AAEA,IAAA,MAAM,aAAA,EAAuC,CAAC,CAAA;AAE9C,IAAA,GAAA,CAAI,YAAA,GAAe,OAAA,GAAU,MAAA,CAAO,SAAA,EAAW;AAC9C,MAAA,YAAA,CAAa,mBAAmB,EAAA,EAAI,MAAA;AAAA,IACrC;AAEA,IAAA,GAAA,CAAI,OAAO,MAAA,CAAO,SAAA,IAAa,QAAA,EAAU;AACxC,MAAA,YAAA,CAAa,kBAAkB,EAAA,EAAI,MAAA,CAAO,MAAA,CAAO,QAAQ,CAAA;AAAA,IAC1D;AAEA,IAAA,GAAA,CAAI,OAAO,MAAA,CAAO,eAAA,IAAmB,QAAA,EAAU;AAC9C,MAAA,YAAA,CAAa,kBAAkB,EAAA,EAAI,MAAA,CAAO,cAAA;AAAA,IAC3C;AAEA,IAAA,GAAA,CAAI,OAAO,MAAA,CAAO,SAAA,IAAa,QAAA,EAAU;AACxC,MAAA,YAAA,CAAa,iBAAiB,EAAA,EAAI,IAAA,CAAK,SAAA,CAAU,MAAA,CAAO,QAAQ,CAAA;AAAA,IACjE;AAEA,IAAA,MAAM,QAAA,EAKF;AAAA,MACH,QAAA;AAAA,MACA,QAAA;AAAA,MACA,OAAA,EAAS;AAAA,QACR,mBAAI,IAAA,6BAAM,SAAA;AAAA,QACV,GAAG,OAAA;AAAA,QACH,GAAG,YAAA;AAAA,QACH,cAAA,EAAgB;AAAA,MACjB,CAAA;AAAA,MACA;AAAA,IACD,CAAA;AAEA,IAAA,GAAA,CAAI,SAAA,IAAa,YAAA,EAAc;AAC9B,MAAA,GAAA,CAAI,CAAC,OAAA,CAAQ,QAAA,CAAS,UAAA,CAAW,MAAM,CAAA,EAAG;AACzC,QAAA,OAAA,CAAQ,SAAA,EAAW,CAAA,IAAA,EAAO,KAAA,CAAM,KAAK,CAAA,CAAA;AACtC,MAAA;AACa,MAAA;AACA,MAAA;AACd,IAAA;AAEmB,IAAA;AACiB,MAAA;AACpC,IAAA;AAEyB,IAAA;AAKP,MAAA;AAClB,IAAA;AAEO,IAAA;AACkC,MAAA;AACxC,MAAA;AACI,QAAA;AACM,QAAA;AACQ,UAAA;AACb,UAAA;AACA,UAAA;AAEE,UAAA;AAEN,QAAA;AAC4B,QAAA;AAC7B,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAgBC;AAE6B,EAAA;AACA,IAAA;AAGO,IAAA;AACb,MAAA;AACtB,IAAA;AAEO,IAAA;AACP,EAAA;AACF;AAaC;AAE+B,EAAA;AACb,IAAA;AACiB,MAAA;AAClC,IAAA;AAEI,IAAA;AACA,IAAA;AACkC,MAAA;AAC9B,IAAA;AACa,MAAA;AACrB,IAAA;AAEmB,IAAA;AACC,IAAA;AAGqB,IAAA;AACtB,IAAA;AACA,MAAA;AACZ,QAAA;AACN,MAAA;AACD,IAAA;AACoC,IAAA;AACM,IAAA;AACD,IAAA;AACR,IAAA;AACL,IAAA;AAEiB,IAAA;AACP,IAAA;AACA,IAAA;AAET,IAAA;AAC5B,MAAA;AACA,MAAA;AACqC,MAAA;AACtC,IAAA;AAEgC,IAAA;AAGX,MAAA;AACnB,QAAA;AACA,QAAA;AACD,MAAA;AACiC,MAAA;AACvB,QAAA;AACQ,UAAA;AACC,UAAA;AAClB,QAAA;AACA,MAAA;AACF,IAAA;AAUmB,IAAA;AAIqB,IAAA;AAEC,IAAA;AAClC,MAAA;AACG,MAAA;AACV,IAAA;AACmB,IAAA;AAGqB,IAAA;AACxB,MAAA;AACkB,MAAA;AAMzB,QAAA;AACyB,UAAA;AACzB,UAAA;AACI,UAAA;AACqB,YAAA;AAER,YAAA;AAGvB,UAAA;AACD,QAAA;AACD,MAAA;AACD,IAAA;AAEuB,IAAA;AACe,MAAA;AAC7B,MAAA;AAC6B,MAAA;AACrC,MAAA;AAC+B,MAAA;AAChC,IAAA;AAEmC,IAAA;AACP,MAAA;AAC3B,IAAA;AACF,EAAA;AACD;AAkBC;AAEgC,EAAA;AACA,EAAA;AAGM,EAAA;AACM,EAAA;AAC/B,EAAA;AACM,EAAA;AAIE,EAAA;AAGK,EAAA;AAEZ,EAAA;AAC+B,IAAA;AACd,MAAA;AACM,QAAA;AACH,QAAA;AACP,QAAA;AAEE,QAAA;AACC,UAAA;AACU,UAAA;AACR,UAAA;AAGL,UAAA;AAEnB,UAAA;AAC2B,YAAA;AAKI,YAAA;AAGhB,cAAA;AACe,cAAA;AACD,cAAA;AACf,gBAAA;AACe,gBAAA;AACD,kBAAA;AACL,kBAAA;AAEF,oBAAA;AACF,oBAAA;AACjB,sBAAA;AACmB,sBAAA;AACnB,oBAAA;AACO,oBAAA;AACF,kBAAA;AAGI,oBAAA;AACX,kBAAA;AACD,gBAAA;AACD,cAAA;AAC8B,cAAA;AACd,gBAAA;AAChB,cAAA;AACW,cAAA;AACmB,gBAAA;AAAiB;AAAM;AACrD,cAAA;AACA,cAAA;AACD,YAAA;AAK+B,YAAA;AACV,cAAA;AACf,gBAAA;AACI,gBAAA;AACR,gBAAA;AACA,gBAAA;AACS,gBAAA;AACR,kBAAA;AACQ,oBAAA;AACkB,oBAAA;AACV,oBAAA;AAChB,kBAAA;AACD,gBAAA;AACD,cAAA;AACW,cAAA;AACmB,gBAAA;AAAsB;AAAM;AAC1D,cAAA;AACD,YAAA;AAOyB,YAAA;AACA,cAAA;AACK,gBAAA;AAGA,gBAAA;AACA,gBAAA;AACN,gBAAA;AAGI,gBAAA;AAEX,gBAAA;AAGgC,gBAAA;AACvC,kBAAA;AACR,gBAAA;AAE8B,gBAAA;AAEH,kBAAA;AACC,kBAAA;AACR,kBAAA;AACE,kBAAA;AACI,kBAAA;AACR,oBAAA;AAAA;AAGL,oBAAA;AAKZ,kBAAA;AACM,gBAAA;AAEgB,kBAAA;AACI,oBAAA;AAEhB,sBAAA;AAGT,oBAAA;AACM,kBAAA;AACN,oBAAA;AACD,kBAAA;AACD,gBAAA;AAEkB,gBAAA;AACb,kBAAA;AACI,kBAAA;AACR,kBAAA;AACA,kBAAA;AACS,kBAAA;AACR,oBAAA;AACQ,sBAAA;AACe,sBAAA;AACP,sBAAA;AAChB,oBAAA;AACD,kBAAA;AACD,gBAAA;AACW,gBAAA;AACc,kBAAA;AAAyB;AAAM;AACxD,gBAAA;AACD,cAAA;AACD,YAAA;AACW,UAAA;AAEE,YAAA;AACd,UAAA;AACD,QAAA;AACD,MAAA;AACkB,MAAA;AACI,QAAA;AAED,UAAA;AACd,YAAA;AACI,YAAA;AACR,YAAA;AACA,YAAA;AACS,YAAA;AACR,cAAA;AACQ,gBAAA;AACC,gBAAA;AACsB,gBAAA;AAC/B,cAAA;AACD,YAAA;AACD,UAAA;AACkC,UAAA;AAAmC;AAAO;AAC7E,QAAA;AAGkC,QAAA;AACnC,MAAA;AACA,IAAA;AACF,EAAA;AACD;AD7T8C;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-3L6IRVAV.cjs","sourcesContent":[null,"// ---------------------------------------------------------------------------\n// AI Gateway types (for third-party providers + Workers AI through gateway)\n// ---------------------------------------------------------------------------\n\nexport interface CloudflareAiGateway {\n\trun(request: unknown): Promise<Response>;\n}\n\nexport interface AiGatewayBindingConfig {\n\t/**\n\t * The AI Gateway binding\n\t * @example\n\t * env.AI.gateway('my-gateway-id')\n\t */\n\tbinding: CloudflareAiGateway;\n\t/**\n\t * The Provider API Key if you want to manually pass it, ignore if using Unified Billing or BYOK.\n\t */\n\tapiKey?: string;\n}\n\nexport type AiGatewayCredentialsConfig = {\n\t/**\n\t * The Cloudflare account ID\n\t */\n\taccountId: string;\n\t/**\n\t * The AI Gateway ID\n\t */\n\tgatewayId: string;\n} & (\n\t| {\n\t\t\t/** Cloudflare API Key for AI Gateway */\n\t\t\tcfApiKey: string;\n\t\t\tapiKey?: string;\n\t  }\n\t| {\n\t\t\t/** Provider API Key */\n\t\t\tapiKey: string;\n\t\t\t/** Cloudflare API Key for AI Gateway */\n\t\t\tcfApiKey?: string;\n\t  }\n);\n\nexport interface AiGatewayConfig {\n\tskipCache?: boolean;\n\tcacheTtl?: number;\n\tcustomCacheKey?: string;\n\tmetadata?: Record<string, unknown>;\n}\n\nexport type AiGatewayAdapterConfig = (AiGatewayBindingConfig | AiGatewayCredentialsConfig) &\n\tAiGatewayConfig;\n\n// ---------------------------------------------------------------------------\n// Plain Workers AI types (direct binding or REST, no gateway)\n// ---------------------------------------------------------------------------\n\n/**\n * The Workers AI binding interface (env.AI).\n * Accepts a model name and inputs, returns results directly.\n * Includes `gateway()` which is present on `env.AI` but not on `env.AI.gateway(id)`,\n * enabling structural discrimination from `CloudflareAiGateway`.\n */\nexport interface WorkersAiBinding {\n\trun(\n\t\tmodel: string,\n\t\tinputs: Record<string, unknown>,\n\t\toptions?: Record<string, unknown>,\n\t): Promise<unknown>;\n\tgateway(gatewayId: string): CloudflareAiGateway;\n}\n\nexport interface WorkersAiDirectBindingConfig {\n\t/**\n\t * The Workers AI binding (env.AI).\n\t * @example\n\t * { binding: env.AI }\n\t */\n\tbinding: WorkersAiBinding;\n}\n\nexport interface WorkersAiDirectCredentialsConfig {\n\t/**\n\t * The Cloudflare account ID\n\t */\n\taccountId: string;\n\t/**\n\t * The Cloudflare API key for Workers AI\n\t */\n\tapiKey: string;\n}\n\n/**\n * Config for Workers AI adapters. Supports four modes:\n * - Plain binding: `{ binding: env.AI }`\n * - Plain REST: `{ accountId, apiKey }`\n * - AI Gateway binding: `{ binding: env.AI.gateway(id) }`\n * - AI Gateway REST: `{ accountId, gatewayId, ... }`\n *\n * The third union member intersects `AiGatewayAdapterConfig` with `{ apiKey?: string }`.\n * For the gateway binding variant, `AiGatewayBindingConfig` already includes `apiKey?`,\n * so the intersection is redundant there. For the gateway credentials variant, this\n * `apiKey` represents the Workers AI token (used in the `Authorization` header to the\n * upstream provider), distinct from `cfApiKey` (used in the `cf-aig-authorization`\n * header for authenticated gateways).\n */\nexport type WorkersAiAdapterConfig = (\n\t| WorkersAiDirectBindingConfig\n\t| WorkersAiDirectCredentialsConfig\n\t| (AiGatewayAdapterConfig & { apiKey?: string })\n) & {\n\t/**\n\t * Session affinity key for prefix-cache optimization.\n\t * Routes requests with the same key to the same backend replica.\n\t */\n\tsessionAffinity?: string;\n};\n\n// ---------------------------------------------------------------------------\n// Config detection helpers\n// ---------------------------------------------------------------------------\n\n/** Returns true if this is a plain Workers AI binding config (`{ binding: env.AI }`) */\nexport function isDirectBindingConfig(\n\tconfig: WorkersAiAdapterConfig,\n): config is WorkersAiDirectBindingConfig {\n\t// env.AI has a .gateway() method; env.AI.gateway(id) does not.\n\t// This distinguishes direct bindings from AI Gateway bindings.\n\treturn (\n\t\t\"binding\" in config &&\n\t\ttypeof (config.binding as unknown as Record<string, unknown>).gateway === \"function\"\n\t);\n}\n\n/** Returns true if this is a plain Workers AI REST config (accountId + apiKey, no gatewayId) */\nexport function isDirectCredentialsConfig(\n\tconfig: WorkersAiAdapterConfig,\n): config is WorkersAiDirectCredentialsConfig {\n\treturn \"accountId\" in config && \"apiKey\" in config && !(\"gatewayId\" in config);\n}\n\n/** Returns true if this is an AI Gateway config (has gateway binding or `gatewayId`) */\nexport function isGatewayConfig(config: WorkersAiAdapterConfig): config is AiGatewayAdapterConfig {\n\tif (\"gatewayId\" in config) return true;\n\t// Has `binding` but NOT a direct Workers AI binding (no .gateway method)\n\treturn \"binding\" in config && !isDirectBindingConfig(config);\n}\n\n// ---------------------------------------------------------------------------\n// Config validation\n// ---------------------------------------------------------------------------\n\n/**\n * Validates that a WorkersAiAdapterConfig contains a valid configuration.\n * Throws an error if neither a binding, credentials (accountId + apiKey),\n * nor a gateway configuration is provided.\n */\nexport function validateWorkersAiConfig(config: WorkersAiAdapterConfig): void {\n\tif (\n\t\t!isDirectBindingConfig(config) &&\n\t\t!isDirectCredentialsConfig(config) &&\n\t\t!isGatewayConfig(config)\n\t) {\n\t\tthrow new Error(\n\t\t\t\"Invalid Workers AI configuration: you must provide either a binding (e.g. { binding: env.AI }), \" +\n\t\t\t\t\"credentials ({ accountId, apiKey }), or a gateway configuration ({ binding: env.AI.gateway(id) } \" +\n\t\t\t\t\"or { accountId, gatewayId }).\",\n\t\t);\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// createGatewayFetch -- for routing through AI Gateway\n// ---------------------------------------------------------------------------\n\nexport function createGatewayFetch(\n\tprovider: string,\n\tconfig: AiGatewayAdapterConfig,\n\theaders: Record<string, string> = {},\n): typeof fetch {\n\treturn (input, init) => {\n\t\tlet query: Record<string, unknown> = {};\n\n\t\tconst url =\n\t\t\ttypeof input === \"string\" ? input : input instanceof URL ? input.href : input.url;\n\t\tconst urlObj = new URL(url);\n\n\t\t// Extract endpoint path (remove /v1/ prefix if present)\n\t\tconst endpoint = urlObj.pathname.replace(/^\\/v1\\//, \"\").replace(/^\\//, \"\") + urlObj.search;\n\n\t\tif (init?.body) {\n\t\t\ttry {\n\t\t\t\tquery = JSON.parse(init.body as string);\n\t\t\t} catch {\n\t\t\t\tquery = { _raw: init.body };\n\t\t\t}\n\t\t}\n\n\t\tconst cacheHeaders: Record<string, string> = {};\n\n\t\tif (\"skipCache\" in config && config.skipCache) {\n\t\t\tcacheHeaders[\"cf-aig-skip-cache\"] = \"true\";\n\t\t}\n\n\t\tif (typeof config.cacheTtl === \"number\") {\n\t\t\tcacheHeaders[\"cf-aig-cache-ttl\"] = String(config.cacheTtl);\n\t\t}\n\n\t\tif (typeof config.customCacheKey === \"string\") {\n\t\t\tcacheHeaders[\"cf-aig-cache-key\"] = config.customCacheKey;\n\t\t}\n\n\t\tif (typeof config.metadata === \"object\") {\n\t\t\tcacheHeaders[\"cf-aig-metadata\"] = JSON.stringify(config.metadata);\n\t\t}\n\n\t\tconst request: {\n\t\t\tprovider: string;\n\t\t\tendpoint: string;\n\t\t\theaders: Record<string, string>;\n\t\t\tquery: Record<string, unknown>;\n\t\t} = {\n\t\t\tprovider,\n\t\t\tendpoint,\n\t\t\theaders: {\n\t\t\t\t...(init?.headers as Record<string, string> | undefined),\n\t\t\t\t...headers,\n\t\t\t\t...cacheHeaders,\n\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t},\n\t\t\tquery,\n\t\t};\n\n\t\tif (provider === \"workers-ai\") {\n\t\t\tif (!request.endpoint.startsWith(\"run/\")) {\n\t\t\t\trequest.endpoint = `run/${query.model}`;\n\t\t\t}\n\t\t\tdelete query.model;\n\t\t\tdelete query.instructions;\n\t\t}\n\n\t\tif (config.apiKey) {\n\t\t\trequest.headers[\"authorization\"] = `Bearer ${config.apiKey}`;\n\t\t}\n\n\t\tif (\"binding\" in config) {\n\t\t\treturn (\n\t\t\t\tconfig.binding as {\n\t\t\t\t\trun(req: unknown, opts?: { signal?: AbortSignal }): Promise<Response>;\n\t\t\t\t}\n\t\t\t).run(request, { signal: init?.signal ?? undefined });\n\t\t}\n\n\t\treturn fetch(\n\t\t\t`https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}`,\n\t\t\t{\n\t\t\t\t...init,\n\t\t\t\theaders: {\n\t\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\t\t...headers,\n\t\t\t\t\t...cacheHeaders,\n\t\t\t\t\t...(config.cfApiKey\n\t\t\t\t\t\t? { \"cf-aig-authorization\": `Bearer ${config.cfApiKey}` }\n\t\t\t\t\t\t: {}),\n\t\t\t\t},\n\t\t\t\tbody: JSON.stringify(request),\n\t\t\t},\n\t\t);\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// createWorkersAiBindingFetch -- shim that makes env.AI look like an OpenAI endpoint\n// ---------------------------------------------------------------------------\n\n/**\n * Normalize messages before passing to Workers AI binding.\n *\n * The binding has strict schema validation that may differ from the OpenAI API:\n * - `content` must not be null\n *\n * Content arrays (with image_url parts) are passed through as-is since the\n * Workers AI binding accepts them at runtime for vision-capable models.\n */\nfunction normalizeMessagesForBinding(\n\tmessages: Record<string, unknown>[],\n): Record<string, unknown>[] {\n\treturn messages.map((msg) => {\n\t\tconst normalized = { ...msg };\n\n\t\t// content: null → content: \"\"\n\t\tif (normalized.content === null || normalized.content === undefined) {\n\t\t\tnormalized.content = \"\";\n\t\t}\n\n\t\treturn normalized;\n\t});\n}\n\n/**\n * Creates a fetch function that intercepts OpenAI SDK requests and translates them\n * to Workers AI binding calls (env.AI.run). This allows the WorkersAiTextAdapter\n * to use the OpenAI SDK against a plain Workers AI binding.\n *\n * NOTE: The `input` URL parameter is intentionally ignored. The model name and all\n * request parameters are extracted from the JSON body, matching Workers AI's\n * `binding.run(model, inputs)` calling convention.\n */\nexport function createWorkersAiBindingFetch(\n\tbinding: WorkersAiBinding,\n\toptions?: { extraHeaders?: Record<string, string> },\n): typeof fetch {\n\treturn async (_input, init) => {\n\t\tif (!init?.body) {\n\t\t\treturn new Response(\"No body\", { status: 400 });\n\t\t}\n\n\t\tlet body: Record<string, unknown>;\n\t\ttry {\n\t\t\tbody = JSON.parse(init.body as string);\n\t\t} catch {\n\t\t\treturn new Response(\"Invalid JSON body\", { status: 400 });\n\t\t}\n\n\t\tconst model = body.model as string;\n\t\tconst stream = body.stream as boolean | undefined;\n\n\t\t// Build Workers AI inputs from OpenAI format\n\t\tconst inputs: Record<string, unknown> = {};\n\t\tif (body.messages) {\n\t\t\tinputs.messages = normalizeMessagesForBinding(\n\t\t\t\tbody.messages as Record<string, unknown>[],\n\t\t\t);\n\t\t}\n\t\tif (body.tools) inputs.tools = body.tools;\n\t\tif (typeof body.temperature === \"number\") inputs.temperature = body.temperature;\n\t\tif (typeof body.max_tokens === \"number\") inputs.max_tokens = body.max_tokens;\n\t\tif (body.response_format) inputs.response_format = body.response_format;\n\t\tif (stream) inputs.stream = true;\n\n\t\tconst runOptions: Record<string, unknown> = {};\n\t\tif (options?.extraHeaders) runOptions.extraHeaders = options.extraHeaders;\n\t\tif (init?.signal) runOptions.signal = init.signal;\n\n\t\tconst result = await binding.run(\n\t\t\tmodel,\n\t\t\tinputs,\n\t\t\tObject.keys(runOptions).length > 0 ? runOptions : undefined,\n\t\t);\n\n\t\tif (stream && result instanceof ReadableStream) {\n\t\t\t// Workers AI returns an SSE stream with `data: {\"response\":\"chunk\"}` format.\n\t\t\t// Transform it to OpenAI-compatible SSE format.\n\t\t\tconst transformed = transformWorkersAiStream(\n\t\t\t\tresult as ReadableStream<Uint8Array>,\n\t\t\t\tmodel,\n\t\t\t);\n\t\t\treturn new Response(transformed, {\n\t\t\t\theaders: {\n\t\t\t\t\t\"Content-Type\": \"text/event-stream\",\n\t\t\t\t\t\"Cache-Control\": \"no-cache\",\n\t\t\t\t},\n\t\t\t});\n\t\t}\n\n\t\t// Graceful degradation: some models return a complete (non-streaming)\n\t\t// response even when `stream: true` is requested. Fall through to the\n\t\t// non-streaming wrapper which produces a valid OpenAI Chat Completion\n\t\t// response that the SDK can consume.\n\n\t\t// Non-streaming: Workers AI returns { response: \"text\", tool_calls?: [...] }\n\t\t// Wrap into OpenAI Chat Completion format.\n\t\tconst responseObj =\n\t\t\ttypeof result === \"object\" && result !== null\n\t\t\t\t? (result as Record<string, unknown>)\n\t\t\t\t: { response: String(result) };\n\n\t\tconst responseText = typeof responseObj.response === \"string\" ? responseObj.response : \"\";\n\n\t\tconst message: Record<string, unknown> = {\n\t\t\trole: \"assistant\",\n\t\t\tcontent: responseText,\n\t\t};\n\t\tlet finishReason = \"stop\";\n\n\t\t// Handle tool calls if present in Workers AI response\n\t\tif (Array.isArray(responseObj.tool_calls) && responseObj.tool_calls.length > 0) {\n\t\t\tfinishReason = \"tool_calls\";\n\t\t\tmessage.tool_calls = responseObj.tool_calls.map(\n\t\t\t\t(tc: {\n\t\t\t\t\tid?: string;\n\t\t\t\t\tname?: string;\n\t\t\t\t\targuments: unknown;\n\t\t\t\t\tfunction?: { name: string; arguments?: unknown };\n\t\t\t\t}) => ({\n\t\t\t\t\tid: tc.id || crypto.randomUUID(),\n\t\t\t\t\ttype: \"function\",\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: tc.function?.name || tc.name || \"\",\n\t\t\t\t\t\targuments:\n\t\t\t\t\t\t\ttypeof (tc.function?.arguments ?? tc.arguments) === \"string\"\n\t\t\t\t\t\t\t\t? ((tc.function?.arguments ?? tc.arguments) as string)\n\t\t\t\t\t\t\t\t: JSON.stringify(tc.function?.arguments ?? tc.arguments ?? {}),\n\t\t\t\t\t},\n\t\t\t\t}),\n\t\t\t);\n\t\t}\n\n\t\tconst openAiResponse = {\n\t\t\tid: `workers-ai-${crypto.randomUUID()}`,\n\t\t\tobject: \"chat.completion\",\n\t\t\tcreated: Math.floor(Date.now() / 1000),\n\t\t\tmodel,\n\t\t\tchoices: [{ index: 0, message, finish_reason: finishReason }],\n\t\t};\n\n\t\treturn new Response(JSON.stringify(openAiResponse), {\n\t\t\theaders: { \"Content-Type\": \"application/json\" },\n\t\t});\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// Stream transformer: Workers AI SSE -> OpenAI-compatible SSE\n// Uses TransformStream for proper backpressure.\n// ---------------------------------------------------------------------------\n\n/**\n * Transforms a Workers AI SSE stream (data: {\"response\":\"chunk\"}) into\n * an OpenAI-compatible SSE stream (data: {\"choices\":[{\"delta\":{\"content\":\"chunk\"}}]}).\n *\n * Workers AI binding streams tool calls in an OpenAI-like nested format:\n *   { tool_calls: [{ id, type, index, function: { name, arguments } }] }\n * Arguments are streamed incrementally across multiple SSE chunks, so the\n * transformer must forward them as incremental deltas rather than a single blob.\n */\nfunction transformWorkersAiStream(\n\tsource: ReadableStream<Uint8Array>,\n\tmodel: string,\n): ReadableStream<Uint8Array> {\n\tconst decoder = new TextDecoder();\n\tconst encoder = new TextEncoder();\n\t// Generate a stable ID and timestamp for the entire stream, matching OpenAI's\n\t// convention where all chunks in a single response share the same id/created.\n\tconst streamId = `workers-ai-${crypto.randomUUID()}`;\n\tconst created = Math.floor(Date.now() / 1000);\n\tlet buffer = \"\";\n\tlet hasToolCalls = false;\n\t// When true, the source stream is already in OpenAI format (some models\n\t// like Qwen3, Kimi K2.5 stream OpenAI-compatible SSE through the binding).\n\t// In that case, flush() should only emit [DONE] and skip the finish chunk.\n\tlet isOpenAiFormat = false;\n\t// Track tool call state per index: store the generated/assigned ID so that\n\t// subsequent argument deltas use the same ID (matching the working streaming.ts pattern).\n\tconst toolCallState = new Map<number, { id: string; name: string }>();\n\n\treturn source.pipeThrough(\n\t\tnew TransformStream<Uint8Array, Uint8Array>({\n\t\t\ttransform(chunk, controller) {\n\t\t\t\tbuffer += decoder.decode(chunk, { stream: true });\n\t\t\t\tconst lines = buffer.split(\"\\n\");\n\t\t\t\tbuffer = lines.pop() || \"\";\n\n\t\t\t\tfor (const line of lines) {\n\t\t\t\t\tconst trimmed = line.trim();\n\t\t\t\t\tif (!trimmed || !trimmed.startsWith(\"data: \")) continue;\n\t\t\t\t\tconst data = trimmed.slice(6);\n\n\t\t\t\t\t// Swallow source [DONE]; we emit our own in flush()\n\t\t\t\t\tif (data === \"[DONE]\") continue;\n\n\t\t\t\t\ttry {\n\t\t\t\t\t\tconst parsed = JSON.parse(data);\n\n\t\t\t\t\t\t// Some models (Qwen3, Kimi K2.5) return OpenAI-compatible format\n\t\t\t\t\t\t// directly through the binding, with `choices[].delta.content` and\n\t\t\t\t\t\t// optional `reasoning_content`. Detect this and pass through as-is.\n\t\t\t\t\t\tif (parsed.choices !== undefined) {\n\t\t\t\t\t\t\t// Already OpenAI format — pass through but ensure each tool call\n\t\t\t\t\t\t\t// index gets a unique, stable ID across all chunks.\n\t\t\t\t\t\t\tisOpenAiFormat = true;\n\t\t\t\t\t\t\tconst choice = parsed.choices?.[0];\n\t\t\t\t\t\t\tif (choice?.delta?.tool_calls) {\n\t\t\t\t\t\t\t\thasToolCalls = true;\n\t\t\t\t\t\t\t\tfor (const tc of choice.delta.tool_calls) {\n\t\t\t\t\t\t\t\t\tconst tcIndex = tc.index ?? 0;\n\t\t\t\t\t\t\t\t\tif (!toolCallState.has(tcIndex)) {\n\t\t\t\t\t\t\t\t\t\t// First chunk for this index — generate/store unique ID\n\t\t\t\t\t\t\t\t\t\tconst id = tc.id || `call${streamId}${tcIndex}`;\n\t\t\t\t\t\t\t\t\t\ttoolCallState.set(tcIndex, {\n\t\t\t\t\t\t\t\t\t\t\tid,\n\t\t\t\t\t\t\t\t\t\t\tname: tc.function?.name || \"\",\n\t\t\t\t\t\t\t\t\t\t});\n\t\t\t\t\t\t\t\t\t\ttc.id = id;\n\t\t\t\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t\t\t\t// Subsequent chunk — reuse stored ID, remove id from delta\n\t\t\t\t\t\t\t\t\t\t// (OpenAI format only sends id in first chunk)\n\t\t\t\t\t\t\t\t\t\tdelete tc.id;\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tif (choice?.finish_reason === \"tool_calls\") {\n\t\t\t\t\t\t\t\thasToolCalls = true;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tcontroller.enqueue(\n\t\t\t\t\t\t\t\tencoder.encode(`data: ${JSON.stringify(parsed)}\\n\\n`),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\tcontinue;\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// --- Workers AI native format handling below ---\n\n\t\t\t\t\t\t// Text content\n\t\t\t\t\t\tif (parsed.response != null && parsed.response !== \"\") {\n\t\t\t\t\t\t\tconst openAiChunk = {\n\t\t\t\t\t\t\t\tid: streamId,\n\t\t\t\t\t\t\t\tobject: \"chat.completion.chunk\",\n\t\t\t\t\t\t\t\tcreated,\n\t\t\t\t\t\t\t\tmodel,\n\t\t\t\t\t\t\t\tchoices: [\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t\t\t\t\tdelta: { content: parsed.response },\n\t\t\t\t\t\t\t\t\t\tfinish_reason: null,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\tcontroller.enqueue(\n\t\t\t\t\t\t\t\tencoder.encode(`data: ${JSON.stringify(openAiChunk)}\\n\\n`),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\t// Tool calls — Workers AI binding streams these incrementally:\n\t\t\t\t\t\t//   Chunk A: { id, type, index, function: { name } }          — start\n\t\t\t\t\t\t//   Chunk B: { index, function: { arguments: \"partial...\" } }  — args delta\n\t\t\t\t\t\t//   Chunk C: { index, function: { arguments: \"rest...\" } }     — args delta\n\t\t\t\t\t\t//   Chunk D: { id: null, type: null, index, function: { name: null, arguments: \"\" } } — finalize (skip)\n\t\t\t\t\t\tif (Array.isArray(parsed.tool_calls) && parsed.tool_calls.length > 0) {\n\t\t\t\t\t\t\tfor (const tc of parsed.tool_calls) {\n\t\t\t\t\t\t\t\tconst tcIndex = tc.index ?? 0;\n\n\t\t\t\t\t\t\t\t// Resolve name and arguments from either nested or flat format\n\t\t\t\t\t\t\t\tconst tcName = tc.function?.name ?? tc.name ?? null;\n\t\t\t\t\t\t\t\tconst tcArgs = tc.function?.arguments ?? tc.arguments ?? null;\n\t\t\t\t\t\t\t\tconst tcId = tc.id ?? null;\n\n\t\t\t\t\t\t\t\t// Skip finalization chunks where everything is null/empty\n\t\t\t\t\t\t\t\tif (!tcId && !tcName && (!tcArgs || tcArgs === \"\")) continue;\n\n\t\t\t\t\t\t\t\thasToolCalls = true;\n\n\t\t\t\t\t\t\t\t// Build the OpenAI-compatible tool_calls delta\n\t\t\t\t\t\t\t\tconst toolCallDelta: Record<string, unknown> = {\n\t\t\t\t\t\t\t\t\tindex: tcIndex,\n\t\t\t\t\t\t\t\t};\n\n\t\t\t\t\t\t\t\tif (!toolCallState.has(tcIndex)) {\n\t\t\t\t\t\t\t\t\t// First chunk for this tool call index — emit id, type, name.\n\t\t\t\t\t\t\t\t\tconst id = tcId || `call${streamId}${tcIndex}`;\n\t\t\t\t\t\t\t\t\ttoolCallState.set(tcIndex, { id, name: tcName || \"\" });\n\t\t\t\t\t\t\t\t\ttoolCallDelta.id = id;\n\t\t\t\t\t\t\t\t\ttoolCallDelta.type = \"function\";\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function = {\n\t\t\t\t\t\t\t\t\t\tname: tcName || \"\",\n\t\t\t\t\t\t\t\t\t\t// Include arguments if they arrive in the same chunk\n\t\t\t\t\t\t\t\t\t\targuments:\n\t\t\t\t\t\t\t\t\t\t\ttcArgs != null\n\t\t\t\t\t\t\t\t\t\t\t\t? typeof tcArgs === \"string\"\n\t\t\t\t\t\t\t\t\t\t\t\t\t? tcArgs\n\t\t\t\t\t\t\t\t\t\t\t\t\t: JSON.stringify(tcArgs)\n\t\t\t\t\t\t\t\t\t\t\t\t: \"\",\n\t\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t\t\t// Subsequent chunks — only include arguments delta\n\t\t\t\t\t\t\t\t\tif (tcArgs != null && tcArgs !== \"\") {\n\t\t\t\t\t\t\t\t\t\ttoolCallDelta.function = {\n\t\t\t\t\t\t\t\t\t\t\targuments:\n\t\t\t\t\t\t\t\t\t\t\t\ttypeof tcArgs === \"string\"\n\t\t\t\t\t\t\t\t\t\t\t\t\t? tcArgs\n\t\t\t\t\t\t\t\t\t\t\t\t\t: JSON.stringify(tcArgs),\n\t\t\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t\t\t\tcontinue; // Nothing useful to forward\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\t\tconst toolChunk = {\n\t\t\t\t\t\t\t\t\tid: streamId,\n\t\t\t\t\t\t\t\t\tobject: \"chat.completion.chunk\",\n\t\t\t\t\t\t\t\t\tcreated,\n\t\t\t\t\t\t\t\t\tmodel,\n\t\t\t\t\t\t\t\t\tchoices: [\n\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t\t\t\t\t\tdelta: { tool_calls: [toolCallDelta] },\n\t\t\t\t\t\t\t\t\t\t\tfinish_reason: null,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t\tcontroller.enqueue(\n\t\t\t\t\t\t\t\t\tencoder.encode(`data: ${JSON.stringify(toolChunk)}\\n\\n`),\n\t\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t} catch (e) {\n\t\t\t\t\t\t// Log malformed SSE events for debugging; don't break the stream.\n\t\t\t\t\t\tconsole.warn(\"[tanstack-ai] failed to parse SSE event:\", data, e);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t\tflush(controller) {\n\t\t\t\tif (!isOpenAiFormat) {\n\t\t\t\t\t// Workers AI native format: emit a finish chunk with stop/tool_calls\n\t\t\t\t\tconst finalChunk = {\n\t\t\t\t\t\tid: streamId,\n\t\t\t\t\t\tobject: \"chat.completion.chunk\",\n\t\t\t\t\t\tcreated,\n\t\t\t\t\t\tmodel,\n\t\t\t\t\t\tchoices: [\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tindex: 0,\n\t\t\t\t\t\t\t\tdelta: {},\n\t\t\t\t\t\t\t\tfinish_reason: hasToolCalls ? \"tool_calls\" : \"stop\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t],\n\t\t\t\t\t};\n\t\t\t\t\tcontroller.enqueue(encoder.encode(`data: ${JSON.stringify(finalChunk)}\\n\\n`));\n\t\t\t\t}\n\t\t\t\t// OpenAI format already includes its own finish_reason in the stream.\n\t\t\t\t// Either way, emit a [DONE] sentinel.\n\t\t\t\tcontroller.enqueue(encoder.encode(\"data: [DONE]\\n\\n\"));\n\t\t\t},\n\t\t}),\n\t);\n}\n"]}

package/dist/chunk-3VQDXJLW.cjs

@@ -1,46 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/utils/binary.ts
-function uint8ArrayToBase64(bytes) {
-  let binary = "";
-  for (let i = 0; i < bytes.length; i++) {
-    binary += String.fromCharCode(bytes[i]);
-  }
-  return btoa(binary);
-}
-async function collectStream(stream) {
-  const reader = stream.getReader();
-  const chunks = [];
-  while (true) {
-    const { done, value } = await reader.read();
-    if (done) break;
-    chunks.push(value);
-  }
-  const totalLength = chunks.reduce((acc, c) => acc + c.length, 0);
-  const combined = new Uint8Array(totalLength);
-  let offset = 0;
-  for (const chunk of chunks) {
-    combined.set(chunk, offset);
-    offset += chunk.length;
-  }
-  return combined;
-}
-async function binaryToBase64(result, binaryKey = "image") {
-  if (result instanceof ReadableStream) {
-    return uint8ArrayToBase64(await collectStream(result));
-  }
-  if (result instanceof Uint8Array || result instanceof ArrayBuffer) {
-    const bytes = result instanceof ArrayBuffer ? new Uint8Array(result) : result;
-    return uint8ArrayToBase64(bytes);
-  }
-  if (typeof result === "object" && result !== null && binaryKey in result) {
-    return result[binaryKey];
-  }
-  throw new Error(
-    `Unexpected binary response format from Workers AI (expected Uint8Array, ArrayBuffer, ReadableStream, or { ${binaryKey}: string })`
-  );
-}
-
-
-
-
-exports.uint8ArrayToBase64 = uint8ArrayToBase64; exports.binaryToBase64 = binaryToBase64;
-//# sourceMappingURL=chunk-3VQDXJLW.cjs.map

package/dist/chunk-3VQDXJLW.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-3VQDXJLW.cjs","../src/utils/binary.ts"],"names":[],"mappings":"AAAA;ACYO,SAAS,kBAAA,CAAmB,KAAA,EAA2B;AAC7D,EAAA,IAAI,OAAA,EAAS,EAAA;AACb,EAAA,IAAA,CAAA,IAAS,EAAA,EAAI,CAAA,EAAG,EAAA,EAAI,KAAA,CAAM,MAAA,EAAQ,CAAA,EAAA,EAAK;AACtC,IAAA,OAAA,GAAU,MAAA,CAAO,YAAA,CAAa,KAAA,CAAM,CAAC,CAAE,CAAA;AAAA,EACxC;AACA,EAAA,OAAO,IAAA,CAAK,MAAM,CAAA;AACnB;AAKA,MAAA,SAAsB,aAAA,CAAc,MAAA,EAAyD;AAC5F,EAAA,MAAM,OAAA,EAAS,MAAA,CAAO,SAAA,CAAU,CAAA;AAChC,EAAA,MAAM,OAAA,EAAuB,CAAC,CAAA;AAC9B,EAAA,MAAA,CAAO,IAAA,EAAM;AACZ,IAAA,MAAM,EAAE,IAAA,EAAM,MAAM,EAAA,EAAI,MAAM,MAAA,CAAO,IAAA,CAAK,CAAA;AAC1C,IAAA,GAAA,CAAI,IAAA,EAAM,KAAA;AACV,IAAA,MAAA,CAAO,IAAA,CAAK,KAAK,CAAA;AAAA,EAClB;AACA,EAAA,MAAM,YAAA,EAAc,MAAA,CAAO,MAAA,CAAO,CAAC,GAAA,EAAK,CAAA,EAAA,GAAM,IAAA,EAAM,CAAA,CAAE,MAAA,EAAQ,CAAC,CAAA;AAC/D,EAAA,MAAM,SAAA,EAAW,IAAI,UAAA,CAAW,WAAW,CAAA;AAC3C,EAAA,IAAI,OAAA,EAAS,CAAA;AACb,EAAA,IAAA,CAAA,MAAW,MAAA,GAAS,MAAA,EAAQ;AAC3B,IAAA,QAAA,CAAS,GAAA,CAAI,KAAA,EAAO,MAAM,CAAA;AAC1B,IAAA,OAAA,GAAU,KAAA,CAAM,MAAA;AAAA,EACjB;AACA,EAAA,OAAO,QAAA;AACR;AAcA,MAAA,SAAsB,cAAA,CACrB,MAAA,EACA,UAAA,EAAoB,OAAA,EACF;AAClB,EAAA,GAAA,CAAI,OAAA,WAAkB,cAAA,EAAgB;AACrC,IAAA,OAAO,kBAAA,CAAmB,MAAM,aAAA,CAAc,MAAM,CAAC,CAAA;AAAA,EACtD;AAEA,EAAA,GAAA,CAAI,OAAA,WAAkB,WAAA,GAAc,OAAA,WAAkB,WAAA,EAAa;AAClE,IAAA,MAAM,MAAA,EAAQ,OAAA,WAAkB,YAAA,EAAc,IAAI,UAAA,CAAW,MAAM,EAAA,EAAI,MAAA;AACvE,IAAA,OAAO,kBAAA,CAAmB,KAAK,CAAA;AAAA,EAChC;AAGA,EAAA,GAAA,CAAI,OAAO,OAAA,IAAW,SAAA,GAAY,OAAA,IAAW,KAAA,GAAQ,UAAA,GAAa,MAAA,EAAQ;AACzE,IAAA,OAAQ,MAAA,CAAkC,SAAS,CAAA;AAAA,EACpD;AAEA,EAAA,MAAM,IAAI,KAAA;AAAA,IACT,CAAA,0GAAA,EAA6G,SAAS,CAAA,WAAA;AAAA,EACvH,CAAA;AACD;ADlCA;AACA;AACE;AACA;AACF,yFAAC","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-3VQDXJLW.cjs","sourcesContent":[null,"/**\n * Shared binary-data utilities for Workers AI adapters.\n *\n * Workers AI returns binary data (images, audio) in various formats —\n * Uint8Array, ArrayBuffer, ReadableStream, or JSON objects with a base64 field.\n * These helpers normalise everything into base64 strings for the TanStack AI\n * result types.\n */\n\n/**\n * Convert a Uint8Array to a base64 string.\n */\nexport function uint8ArrayToBase64(bytes: Uint8Array): string {\n\tlet binary = \"\";\n\tfor (let i = 0; i < bytes.length; i++) {\n\t\tbinary += String.fromCharCode(bytes[i]!);\n\t}\n\treturn btoa(binary);\n}\n\n/**\n * Collect all chunks from a ReadableStream<Uint8Array> into a single Uint8Array.\n */\nexport async function collectStream(stream: ReadableStream<Uint8Array>): Promise<Uint8Array> {\n\tconst reader = stream.getReader();\n\tconst chunks: Uint8Array[] = [];\n\twhile (true) {\n\t\tconst { done, value } = await reader.read();\n\t\tif (done) break;\n\t\tchunks.push(value);\n\t}\n\tconst totalLength = chunks.reduce((acc, c) => acc + c.length, 0);\n\tconst combined = new Uint8Array(totalLength);\n\tlet offset = 0;\n\tfor (const chunk of chunks) {\n\t\tcombined.set(chunk, offset);\n\t\toffset += chunk.length;\n\t}\n\treturn combined;\n}\n\n/**\n * Normalise a Workers AI binary response to a base64 string.\n *\n * Handles:\n * - `Uint8Array` / `ArrayBuffer` — raw bytes\n * - `ReadableStream<Uint8Array>` — streamed bytes\n * - `{ [binaryKey]: \"base64...\" }` — JSON wrapper (e.g. `{ image: \"...\" }`)\n *\n * @param result  The raw value returned from Workers AI\n * @param binaryKey  The JSON field to look for when the result is an object\n *                   (defaults to `\"image\"`; pass `\"audio\"` for TTS responses)\n */\nexport async function binaryToBase64(\n\tresult: unknown,\n\tbinaryKey: string = \"image\",\n): Promise<string> {\n\tif (result instanceof ReadableStream) {\n\t\treturn uint8ArrayToBase64(await collectStream(result));\n\t}\n\n\tif (result instanceof Uint8Array || result instanceof ArrayBuffer) {\n\t\tconst bytes = result instanceof ArrayBuffer ? new Uint8Array(result) : result;\n\t\treturn uint8ArrayToBase64(bytes);\n\t}\n\n\t// Some models return { image: \"base64...\" } or { audio: \"base64...\" }\n\tif (typeof result === \"object\" && result !== null && binaryKey in result) {\n\t\treturn (result as Record<string, string>)[binaryKey]!;\n\t}\n\n\tthrow new Error(\n\t\t`Unexpected binary response format from Workers AI (expected Uint8Array, ArrayBuffer, ReadableStream, or { ${binaryKey}: string })`,\n\t);\n}\n"]}

package/dist/chunk-4DE2IREA.cjs

@@ -1,8 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var __defProp = Object.defineProperty;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
-
-
-
-exports.__publicField = __publicField;
-//# sourceMappingURL=chunk-4DE2IREA.cjs.map

package/dist/chunk-4DE2IREA.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-4DE2IREA.cjs"],"names":[],"mappings":"AAAA,6EAAI,UAAU,EAAE,MAAM,CAAC,cAAc;AACrC,IAAI,gBAAgB,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,GAAG,EAAE,EAAE,KAAK;AAC/J,IAAI,cAAc,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,eAAe,CAAC,GAAG,EAAE,OAAO,IAAI,IAAI,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,CAAC;AAC9G;AACA;AACE;AACF,sCAAC","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-4DE2IREA.cjs"}

package/dist/chunk-5WAK4STT.cjs

@@ -1,98 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-
-var _chunk3VQDXJLWcjs = require('./chunk-3VQDXJLW.cjs');
-
-
-var _chunk7HSUHP63cjs = require('./chunk-7HSUHP63.cjs');
-
-
-
-
-
-var _chunk3L6IRVAVcjs = require('./chunk-3L6IRVAV.cjs');
-
-
-var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
-
-// src/adapters/workers-ai-image.ts
-var _adapters = require('@tanstack/ai/adapters');
-var WorkersAiImageAdapter = class extends _adapters.BaseImageAdapter {
-  constructor(config, model) {
-    super({}, model);
-    _chunk4DE2IREAcjs.__publicField.call(void 0, this, "name", "workers-ai-image");
-    _chunk4DE2IREAcjs.__publicField.call(void 0, this, "adapterConfig");
-    _chunk3L6IRVAVcjs.validateWorkersAiConfig.call(void 0, config);
-    this.adapterConfig = config;
-  }
-  async generateImages(options) {
-    const { prompt, size, modelOptions } = options;
-    const extra = { ...modelOptions };
-    if (size) {
-      const [w, h] = size.split("x");
-      if (w) extra.width = Number(w);
-      if (h) extra.height = Number(h);
-    }
-    if (_chunk3L6IRVAVcjs.isDirectBindingConfig.call(void 0, this.adapterConfig)) {
-      return this.generateViaBinding(prompt, extra);
-    }
-    if (_chunk3L6IRVAVcjs.isDirectCredentialsConfig.call(void 0, this.adapterConfig)) {
-      return this.generateViaRest(prompt, extra);
-    }
-    return this.generateViaGateway(prompt, extra);
-  }
-  async generateViaBinding(prompt, options) {
-    const ai = this.adapterConfig.binding;
-    const result = await ai.run(this.model, { prompt, ...options });
-    const b64 = await _chunk3VQDXJLWcjs.binaryToBase64.call(void 0, result, "image");
-    return this.wrapResult(b64);
-  }
-  async generateViaRest(prompt, options) {
-    const config = this.adapterConfig;
-    const response = await _chunk7HSUHP63cjs.workersAiRestFetch.call(void 0, 
-      config,
-      this.model,
-      { prompt, ...options },
-      { label: "Workers AI image", signal: options.signal }
-    );
-    const buffer = await response.arrayBuffer();
-    return this.wrapResult(_chunk3VQDXJLWcjs.uint8ArrayToBase64.call(void 0, new Uint8Array(buffer)));
-  }
-  async generateViaGateway(prompt, options) {
-    const gatewayConfig = this.adapterConfig;
-    const gatewayFetch = _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "workers-ai", gatewayConfig);
-    const response = await gatewayFetch("https://api.cloudflare.com/v1/images/generations", {
-      method: "POST",
-      body: JSON.stringify({
-        model: this.model,
-        prompt,
-        ...options
-      })
-    });
-    if (!response.ok) {
-      const errorText = await response.text();
-      throw new Error(
-        `Workers AI image gateway request failed (${response.status}): ${errorText}`
-      );
-    }
-    const buffer = await response.arrayBuffer();
-    return this.wrapResult(_chunk3VQDXJLWcjs.uint8ArrayToBase64.call(void 0, new Uint8Array(buffer)));
-  }
-  /** Wrap a base64 image string into the standard ImageGenerationResult. */
-  wrapResult(b64) {
-    return {
-      id: this.generateId(),
-      model: this.model,
-      images: [{ b64Json: b64 }]
-    };
-  }
-};
-function createWorkersAiImage(model, config) {
-  return new WorkersAiImageAdapter(config, model);
-}
-
-
-
-
-exports.WorkersAiImageAdapter = WorkersAiImageAdapter; exports.createWorkersAiImage = createWorkersAiImage;
-//# sourceMappingURL=chunk-5WAK4STT.cjs.map

package/dist/chunk-5WAK4STT.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-5WAK4STT.cjs","../src/adapters/workers-ai-image.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACjBA,iDAAiC;AA6B1B,IAAM,sBAAA,EAAN,MAAA,QAAoC,2BAAsC;AAAA,EAIhF,WAAA,CAAY,MAAA,EAAgC,KAAA,EAA4B;AACvE,IAAA,KAAA,CAAM,CAAC,CAAA,EAAG,KAAK,CAAA;AAJhB,IAAA,6CAAA,IAAA,EAAS,MAAA,EAAO,kBAAA,CAAA;AAChB,IAAA,6CAAA,IAAA,EAAQ,eAAA,CAAA;AAIP,IAAA,uDAAA,MAA8B,CAAA;AAC9B,IAAA,IAAA,CAAK,cAAA,EAAgB,MAAA;AAAA,EACtB;AAAA,EAEA,MAAM,cAAA,CAAe,OAAA,EAAiE;AACrF,IAAA,MAAM,EAAE,MAAA,EAAQ,IAAA,EAAM,aAAa,EAAA,EAAI,OAAA;AACvC,IAAA,MAAM,MAAA,EAAiC,EAAE,GAAG,aAAa,CAAA;AAOzD,IAAA,GAAA,CAAI,IAAA,EAAM;AACT,MAAA,MAAM,CAAC,CAAA,EAAG,CAAC,EAAA,EAAI,IAAA,CAAK,KAAA,CAAM,GAAG,CAAA;AAC7B,MAAA,GAAA,CAAI,CAAA,EAAG,KAAA,CAAM,MAAA,EAAQ,MAAA,CAAO,CAAC,CAAA;AAC7B,MAAA,GAAA,CAAI,CAAA,EAAG,KAAA,CAAM,OAAA,EAAS,MAAA,CAAO,CAAC,CAAA;AAAA,IAC/B;AAEA,IAAA,GAAA,CAAI,qDAAA,IAAsB,CAAK,aAAa,CAAA,EAAG;AAC9C,MAAA,OAAO,IAAA,CAAK,kBAAA,CAAmB,MAAA,EAAQ,KAAK,CAAA;AAAA,IAC7C;AAEA,IAAA,GAAA,CAAI,yDAAA,IAA0B,CAAK,aAAa,CAAA,EAAG;AAClD,MAAA,OAAO,IAAA,CAAK,eAAA,CAAgB,MAAA,EAAQ,KAAK,CAAA;AAAA,IAC1C;AAGA,IAAA,OAAO,IAAA,CAAK,kBAAA,CAAmB,MAAA,EAAQ,KAAK,CAAA;AAAA,EAC7C;AAAA,EAEA,MAAc,kBAAA,CACb,MAAA,EACA,OAAA,EACiC;AACjC,IAAA,MAAM,GAAA,EAAM,IAAA,CAAK,aAAA,CAA+C,OAAA;AAChE,IAAA,MAAM,OAAA,EAAS,MAAM,EAAA,CAAG,GAAA,CAAI,IAAA,CAAK,KAAA,EAAO,EAAE,MAAA,EAAQ,GAAG,QAAQ,CAAC,CAAA;AAE9D,IAAA,MAAM,IAAA,EAAM,MAAM,8CAAA,MAAe,EAAQ,OAAO,CAAA;AAChD,IAAA,OAAO,IAAA,CAAK,UAAA,CAAW,GAAG,CAAA;AAAA,EAC3B;AAAA,EAEA,MAAc,eAAA,CACb,MAAA,EACA,OAAA,EACiC;AACjC,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AACpB,IAAA,MAAM,SAAA,EAAW,MAAM,kDAAA;AAAA,MACtB,MAAA;AAAA,MACA,IAAA,CAAK,KAAA;AAAA,MACL,EAAE,MAAA,EAAQ,GAAG,QAAQ,CAAA;AAAA,MACrB,EAAE,KAAA,EAAO,kBAAA,EAAoB,MAAA,EAAS,OAAA,CAAqC,OAAO;AAAA,IACnF,CAAA;AAEA,IAAA,MAAM,OAAA,EAAS,MAAM,QAAA,CAAS,WAAA,CAAY,CAAA;AAC1C,IAAA,OAAO,IAAA,CAAK,UAAA,CAAW,kDAAA,IAAuB,UAAA,CAAW,MAAM,CAAC,CAAC,CAAA;AAAA,EAClE;AAAA,EAEA,MAAc,kBAAA,CACb,MAAA,EACA,OAAA,EACiC;AACjC,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,aAAA;AAC3B,IAAA,MAAM,aAAA,EAAe,kDAAA,YAAmB,EAAc,aAAa,CAAA;AAKnE,IAAA,MAAM,SAAA,EAAW,MAAM,YAAA,CAAa,kDAAA,EAAoD;AAAA,MACvF,MAAA,EAAQ,MAAA;AAAA,MACR,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU;AAAA,QACpB,KAAA,EAAO,IAAA,CAAK,KAAA;AAAA,QACZ,MAAA;AAAA,QACA,GAAG;AAAA,MACJ,CAAC;AAAA,IACF,CAAC,CAAA;AAED,IAAA,GAAA,CAAI,CAAC,QAAA,CAAS,EAAA,EAAI;AACjB,MAAA,MAAM,UAAA,EAAY,MAAM,QAAA,CAAS,IAAA,CAAK,CAAA;AACtC,MAAA,MAAM,IAAI,KAAA;AAAA,QACT,CAAA,yCAAA,EAA4C,QAAA,CAAS,MAAM,CAAA,GAAA,EAAM,SAAS,CAAA;AAAA,MAAA;AAC3E,IAAA;AAGD,IAAA;AACA,IAAA;AAAiE,EAAA;AAClE;AAAA,EAAA;AAIC,IAAA;AAAO,MAAA;AACc,MAAA;AACR,MAAA;AACa,IAAA;AAC1B,EAAA;AAEF;AAyBO;AACN,EAAA;AACD;ADlEA;AACA;AACA;AACA;AACA","file":"/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-5WAK4STT.cjs","sourcesContent":[null,"import { BaseImageAdapter } from \"@tanstack/ai/adapters\";\nimport type { ImageGenerationOptions, ImageGenerationResult } from \"@tanstack/ai\";\nimport type { AiModels, BaseAiTextToImage } from \"@cloudflare/workers-types\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch } from \"../utils/workers-ai-rest\";\nimport { binaryToBase64, uint8ArrayToBase64 } from \"../utils/binary\";\nimport type { WorkersAiDirectCredentialsConfig } from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model type derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiImageModel =\n\t| { [K in keyof AiModels]: AiModels[K] extends BaseAiTextToImage ? K : never }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiImageAdapter: image generation via Workers AI\n// Extends BaseImageAdapter so it works with TanStack AI's generateImage()\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiImageAdapter extends BaseImageAdapter<WorkersAiImageModel> {\n\treadonly name = \"workers-ai-image\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiImageModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync generateImages(options: ImageGenerationOptions): Promise<ImageGenerationResult> {\n\t\tconst { prompt, size, modelOptions } = options;\n\t\tconst extra: Record<string, unknown> = { ...modelOptions };\n\n\t\t// Note: Workers AI Stable Diffusion models only generate a single image\n\t\t// per request — there is no multi-image parameter. `numberOfImages` from\n\t\t// the TanStack AI options is intentionally not forwarded. Use\n\t\t// `modelOptions.num_steps` to control the number of diffusion steps.\n\n\t\tif (size) {\n\t\t\tconst [w, h] = size.split(\"x\");\n\t\t\tif (w) extra.width = Number(w);\n\t\t\tif (h) extra.height = Number(h);\n\t\t}\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaBinding(prompt, extra);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaRest(prompt, extra);\n\t\t}\n\n\t\t// Gateway mode\n\t\treturn this.generateViaGateway(prompt, extra);\n\t}\n\n\tprivate async generateViaBinding(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = await ai.run(this.model, { prompt, ...options });\n\n\t\tconst b64 = await binaryToBase64(result, \"image\");\n\t\treturn this.wrapResult(b64);\n\t}\n\n\tprivate async generateViaRest(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst response = await workersAiRestFetch(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\t{ prompt, ...options },\n\t\t\t{ label: \"Workers AI image\", signal: (options as { signal?: AbortSignal }).signal },\n\t\t);\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\tprivate async generateViaGateway(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/images/generations\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\tprompt,\n\t\t\t\t...options,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI image gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\t/** Wrap a base64 image string into the standard ImageGenerationResult. */\n\tprivate wrapResult(b64: string): ImageGenerationResult {\n\t\treturn {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\timages: [{ b64Json: b64 }],\n\t\t};\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI image generation adapter.\n *\n * Works with TanStack AI's `generateImage()` activity function:\n * ```ts\n * import { generateImage } from \"@tanstack/ai\";\n * import { createWorkersAiImage } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiImage(\n *   \"@cf/stabilityai/stable-diffusion-xl-base-1.0\",\n *   { binding: env.AI },\n * );\n *\n * const result = await generateImage({ adapter, prompt: \"a cat in space\" });\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiImage(model: WorkersAiImageModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiImageAdapter(config, model);\n}\n"]}