@cloudflare/tanstack-ai 0.1.5 → 0.1.7

package/dist/chunk-IQAGFHRT.cjs

@@ -1,521 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } 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; }
-
-
-
-
-
-var _chunk3L6IRVAVcjs = require('./chunk-3L6IRVAV.cjs');
-
-
-var _chunk4DE2IREAcjs = require('./chunk-4DE2IREA.cjs');
-
-// src/adapters/workers-ai.ts
-
-
-var _adapters = require('@tanstack/ai/adapters');
-var _openai = require('openai'); var _openai2 = _interopRequireDefault(_openai);
-function buildWorkersAiClient(config) {
-  _chunk3L6IRVAVcjs.validateWorkersAiConfig.call(void 0, config);
-  const sessionHeaders = config.sessionAffinity ? { "x-session-affinity": config.sessionAffinity } : void 0;
-  if (_chunk3L6IRVAVcjs.isDirectBindingConfig.call(void 0, config)) {
-    return new (0, _openai2.default)({
-      apiKey: "unused",
-      fetch: _chunk3L6IRVAVcjs.createWorkersAiBindingFetch.call(void 0, 
-        config.binding,
-        sessionHeaders ? { extraHeaders: sessionHeaders } : void 0
-      )
-    });
-  }
-  if (_chunk3L6IRVAVcjs.isDirectCredentialsConfig.call(void 0, config)) {
-    return new (0, _openai2.default)({
-      baseURL: `https://api.cloudflare.com/client/v4/accounts/${config.accountId}/ai/v1`,
-      apiKey: config.apiKey,
-      defaultHeaders: sessionHeaders
-    });
-  }
-  const gatewayConfig = config;
-  return new (0, _openai2.default)({
-    fetch: _chunk3L6IRVAVcjs.createGatewayFetch.call(void 0, "workers-ai", gatewayConfig, sessionHeaders),
-    apiKey: _nullishCoalesce(gatewayConfig.apiKey, () => ( "unused"))
-  });
-}
-function extractTextContent(content) {
-  if (content === null) return "";
-  if (typeof content === "string") return content;
-  return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
-}
-function convertContentPart(part) {
-  switch (part.type) {
-    case "text":
-      if (part.content) {
-        return { type: "text", text: part.content };
-      }
-      return void 0;
-    case "image": {
-      let url;
-      if (part.source.type === "data") {
-        url = part.source.value.startsWith("data:") ? part.source.value : `data:${part.source.mimeType};base64,${part.source.value}`;
-      } else {
-        url = part.source.value;
-      }
-      return { type: "image_url", image_url: { url } };
-    }
-    default:
-      console.warn(
-        `[@cloudflare/tanstack-ai] Unsupported content part type "${part.type}" \u2014 skipping`
-      );
-      return void 0;
-  }
-}
-function buildUserContent(content) {
-  if (content === null) return "";
-  if (typeof content === "string") return content;
-  const hasImages = content.some((p) => p.type === "image");
-  if (!hasImages) {
-    return content.flatMap((p) => p.type === "text" ? [p.content] : []).join("");
-  }
-  const parts = [];
-  for (const part of content) {
-    const converted = convertContentPart(part);
-    if (converted) {
-      parts.push(converted);
-    }
-  }
-  return parts;
-}
-function buildOpenAIMessages(systemPrompts, messages, options) {
-  const includeTools = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _ => _.includeToolMessages]), () => ( true));
-  const openAIMessages = [];
-  if (systemPrompts && systemPrompts.length > 0) {
-    openAIMessages.push({
-      role: "system",
-      content: systemPrompts.join("\n")
-    });
-  }
-  for (const message of messages) {
-    if (message.role === "user") {
-      openAIMessages.push({
-        role: "user",
-        content: buildUserContent(message.content)
-      });
-    } else if (message.role === "assistant") {
-      const assistantMessage = {
-        role: "assistant",
-        content: extractTextContent(message.content)
-      };
-      if (includeTools && message.toolCalls && message.toolCalls.length > 0) {
-        assistantMessage.tool_calls = message.toolCalls.map((tc) => ({
-          id: tc.id,
-          type: "function",
-          function: {
-            name: tc.function.name,
-            arguments: tc.function.arguments
-          }
-        }));
-      }
-      openAIMessages.push(assistantMessage);
-    } else if (includeTools && message.role === "tool") {
-      let toolContent;
-      if (typeof message.content === "string") {
-        try {
-          JSON.parse(message.content);
-          toolContent = message.content;
-        } catch (e) {
-          toolContent = JSON.stringify(message.content);
-        }
-      } else {
-        toolContent = JSON.stringify(message.content);
-      }
-      openAIMessages.push({
-        role: "tool",
-        tool_call_id: message.toolCallId || `tool_${crypto.randomUUID().slice(0, 8)}`,
-        content: toolContent
-      });
-    }
-  }
-  return openAIMessages;
-}
-function buildOpenAITools(tools) {
-  if (!tools) return void 0;
-  return tools.map((tool) => ({
-    type: "function",
-    function: {
-      name: tool.name,
-      description: tool.description,
-      parameters: tool.inputSchema
-    }
-  }));
-}
-function generateId(prefix = "chatcmpl") {
-  return `${prefix}-${crypto.randomUUID().replace(/-/g, "").slice(0, 16)}`;
-}
-var WorkersAiTextAdapter = class extends _adapters.BaseTextAdapter {
-  constructor(model, config) {
-    super({ apiKey: "unused" }, model);
-    _chunk4DE2IREAcjs.__publicField.call(void 0, this, "name", "workers-ai");
-    _chunk4DE2IREAcjs.__publicField.call(void 0, this, "client");
-    this.client = buildWorkersAiClient(config);
-  }
-  async *chatStream(options) {
-    const { systemPrompts, messages, tools, temperature, model } = options;
-    const openAIMessages = buildOpenAIMessages(systemPrompts, messages);
-    const openAITools = buildOpenAITools(tools);
-    const timestamp = Date.now();
-    const runId = generateId();
-    const messageId = generateId();
-    let hasEmittedRunStarted = false;
-    let hasEmittedTextMessageStart = false;
-    let accumulatedContent = "";
-    let hasEmittedStepStarted = false;
-    let accumulatedReasoning = "";
-    const stepId = generateId();
-    let hasReceivedFinishReason = false;
-    const toolCallsInProgress = /* @__PURE__ */ new Map();
-    try {
-      let stream;
-      try {
-        stream = await this.client.chat.completions.create({
-          model: _nullishCoalesce(model, () => ( this.model)),
-          messages: openAIMessages,
-          tools: openAITools,
-          temperature,
-          stream: true,
-          stream_options: { include_usage: true }
-        });
-      } catch (streamError) {
-        console.warn(
-          "[tanstack-ai] Streaming failed, falling back to non-streaming:",
-          streamError instanceof Error ? streamError.message : streamError
-        );
-        const nonStreamResult = await this.client.chat.completions.create({
-          model: _nullishCoalesce(model, () => ( this.model)),
-          messages: openAIMessages,
-          tools: openAITools,
-          temperature
-        });
-        yield {
-          type: "RUN_STARTED",
-          runId,
-          model: nonStreamResult.model || model || this.model,
-          timestamp
-        };
-        const msg = _optionalChain([nonStreamResult, 'access', _2 => _2.choices, 'access', _3 => _3[0], 'optionalAccess', _4 => _4.message]);
-        if (_optionalChain([msg, 'optionalAccess', _5 => _5.content])) {
-          yield {
-            type: "TEXT_MESSAGE_START",
-            messageId,
-            model: nonStreamResult.model || model || this.model,
-            timestamp,
-            role: "assistant"
-          };
-          yield {
-            type: "TEXT_MESSAGE_CONTENT",
-            messageId,
-            model: nonStreamResult.model || model || this.model,
-            timestamp,
-            delta: msg.content,
-            content: msg.content
-          };
-          yield {
-            type: "TEXT_MESSAGE_END",
-            messageId,
-            model: nonStreamResult.model || model || this.model,
-            timestamp
-          };
-        }
-        if (_optionalChain([msg, 'optionalAccess', _6 => _6.tool_calls])) {
-          for (const tc of msg.tool_calls) {
-            if (tc.type !== "function") continue;
-            const fn = tc.function;
-            let parsedInput = {};
-            try {
-              parsedInput = fn.arguments ? JSON.parse(fn.arguments) : {};
-            } catch (e2) {
-              parsedInput = {};
-            }
-            yield {
-              type: "TOOL_CALL_START",
-              toolCallId: tc.id,
-              toolName: fn.name,
-              model: nonStreamResult.model || model || this.model,
-              timestamp,
-              index: 0
-            };
-            yield {
-              type: "TOOL_CALL_END",
-              toolCallId: tc.id,
-              toolName: fn.name,
-              model: nonStreamResult.model || model || this.model,
-              timestamp,
-              input: parsedInput
-            };
-          }
-        }
-        const finishReason = _optionalChain([nonStreamResult, 'access', _7 => _7.choices, 'access', _8 => _8[0], 'optionalAccess', _9 => _9.finish_reason]);
-        yield {
-          type: "RUN_FINISHED",
-          runId,
-          model: nonStreamResult.model || model || this.model,
-          timestamp,
-          usage: nonStreamResult.usage ? {
-            promptTokens: nonStreamResult.usage.prompt_tokens,
-            completionTokens: nonStreamResult.usage.completion_tokens,
-            totalTokens: nonStreamResult.usage.total_tokens
-          } : void 0,
-          finishReason: finishReason === "tool_calls" || finishReason === "function_call" ? "tool_calls" : _nullishCoalesce(finishReason, () => ( "stop"))
-        };
-        return;
-      }
-      for await (const chunk of stream) {
-        if (!chunk.choices || chunk.choices.length === 0) continue;
-        const choice = chunk.choices[0];
-        if (!choice) continue;
-        if (!hasEmittedRunStarted) {
-          hasEmittedRunStarted = true;
-          yield {
-            type: "RUN_STARTED",
-            runId,
-            model: chunk.model || model || this.model,
-            timestamp
-          };
-        }
-        const delta = choice.delta;
-        const reasoningContent = _nullishCoalesce(delta.reasoning_content, () => ( delta.reasoning));
-        if (reasoningContent) {
-          if (!hasEmittedStepStarted) {
-            hasEmittedStepStarted = true;
-            yield {
-              type: "STEP_STARTED",
-              stepId,
-              stepType: "thinking",
-              model: chunk.model || model || this.model,
-              timestamp
-            };
-          }
-          accumulatedReasoning += reasoningContent;
-          yield {
-            type: "STEP_FINISHED",
-            stepId,
-            delta: reasoningContent,
-            content: accumulatedReasoning,
-            model: chunk.model || model || this.model,
-            timestamp
-          };
-        }
-        if (delta.content) {
-          if (!hasEmittedTextMessageStart) {
-            hasEmittedTextMessageStart = true;
-            yield {
-              type: "TEXT_MESSAGE_START",
-              messageId,
-              model: chunk.model || model || this.model,
-              timestamp,
-              role: "assistant"
-            };
-          }
-          accumulatedContent += delta.content;
-          yield {
-            type: "TEXT_MESSAGE_CONTENT",
-            messageId,
-            model: chunk.model || model || this.model,
-            timestamp,
-            delta: delta.content,
-            content: accumulatedContent
-          };
-        }
-        if (delta.tool_calls) {
-          for (const toolCallDelta of delta.tool_calls) {
-            const index = toolCallDelta.index;
-            if (!toolCallsInProgress.has(index)) {
-              toolCallsInProgress.set(index, {
-                id: generateId("chatcmpl-tool"),
-                name: _optionalChain([toolCallDelta, 'access', _10 => _10.function, 'optionalAccess', _11 => _11.name]) || "",
-                arguments: "",
-                started: false
-              });
-            }
-            const toolCall = toolCallsInProgress.get(index);
-            if (_optionalChain([toolCallDelta, 'access', _12 => _12.function, 'optionalAccess', _13 => _13.name])) {
-              toolCall.name = toolCallDelta.function.name;
-            }
-            if (_optionalChain([toolCallDelta, 'access', _14 => _14.function, 'optionalAccess', _15 => _15.arguments])) {
-              toolCall.arguments += toolCallDelta.function.arguments;
-            }
-            if (toolCall.id && toolCall.name && !toolCall.started) {
-              toolCall.started = true;
-              yield {
-                type: "TOOL_CALL_START",
-                toolCallId: toolCall.id,
-                toolName: toolCall.name,
-                model: chunk.model || model || this.model,
-                timestamp,
-                index
-              };
-            }
-            if (_optionalChain([toolCallDelta, 'access', _16 => _16.function, 'optionalAccess', _17 => _17.arguments]) && toolCall.started) {
-              yield {
-                type: "TOOL_CALL_ARGS",
-                toolCallId: toolCall.id,
-                model: chunk.model || model || this.model,
-                timestamp,
-                delta: toolCallDelta.function.arguments
-              };
-            }
-          }
-        }
-        if (choice.finish_reason) {
-          hasReceivedFinishReason = true;
-          if (choice.finish_reason === "tool_calls" || toolCallsInProgress.size > 0) {
-            for (const [, toolCall] of toolCallsInProgress) {
-              let parsedInput = {};
-              try {
-                parsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};
-              } catch (e3) {
-                parsedInput = {};
-              }
-              yield {
-                type: "TOOL_CALL_END",
-                toolCallId: toolCall.id,
-                toolName: toolCall.name,
-                model: chunk.model || model || this.model,
-                timestamp,
-                input: parsedInput
-              };
-            }
-          }
-          const computedFinishReason = choice.finish_reason === "tool_calls" || choice.finish_reason === "function_call" || toolCallsInProgress.size > 0 ? "tool_calls" : choice.finish_reason;
-          if (hasEmittedTextMessageStart) {
-            yield {
-              type: "TEXT_MESSAGE_END",
-              messageId,
-              model: chunk.model || model || this.model,
-              timestamp
-            };
-          }
-          yield {
-            type: "RUN_FINISHED",
-            runId,
-            model: chunk.model || model || this.model,
-            timestamp,
-            usage: chunk.usage ? {
-              promptTokens: chunk.usage.prompt_tokens,
-              completionTokens: chunk.usage.completion_tokens,
-              totalTokens: chunk.usage.total_tokens
-            } : void 0,
-            finishReason: computedFinishReason
-          };
-        }
-      }
-      if (hasEmittedRunStarted && !hasReceivedFinishReason) {
-        console.warn(
-          "[tanstack-ai] Stream ended without finish_reason \u2014 possible truncation or connection drop"
-        );
-        for (const [, toolCall] of toolCallsInProgress) {
-          if (toolCall.started) {
-            let parsedInput = {};
-            try {
-              parsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};
-            } catch (e4) {
-              parsedInput = {};
-            }
-            yield {
-              type: "TOOL_CALL_END",
-              toolCallId: toolCall.id,
-              toolName: toolCall.name,
-              model: _nullishCoalesce(model, () => ( this.model)),
-              timestamp,
-              input: parsedInput
-            };
-          }
-        }
-        if (hasEmittedTextMessageStart) {
-          yield {
-            type: "TEXT_MESSAGE_END",
-            messageId,
-            model: _nullishCoalesce(model, () => ( this.model)),
-            timestamp
-          };
-        }
-        yield {
-          type: "RUN_FINISHED",
-          runId,
-          model: _nullishCoalesce(model, () => ( this.model)),
-          timestamp,
-          finishReason: "stop"
-        };
-      }
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      const code = error instanceof Error ? error.code : void 0;
-      if (!hasEmittedRunStarted) {
-        yield {
-          type: "RUN_STARTED",
-          runId,
-          model: _nullishCoalesce(model, () => ( this.model)),
-          timestamp
-        };
-      }
-      yield {
-        type: "RUN_ERROR",
-        runId,
-        model: _nullishCoalesce(model, () => ( this.model)),
-        timestamp,
-        error: {
-          message: message || "Unknown error",
-          code
-        }
-      };
-    }
-  }
-  async structuredOutput(options) {
-    const { outputSchema, chatOptions } = options;
-    const { systemPrompts, messages, temperature, model } = chatOptions;
-    const openAIMessages = buildOpenAIMessages(systemPrompts, messages, {
-      includeToolMessages: false
-    });
-    const response = await this.client.chat.completions.create({
-      model: _nullishCoalesce(model, () => ( this.model)),
-      messages: openAIMessages,
-      temperature,
-      stream: false,
-      response_format: {
-        type: "json_schema",
-        json_schema: {
-          name: "structured_output",
-          strict: true,
-          schema: outputSchema
-        }
-      }
-    });
-    const choice = _optionalChain([response, 'access', _18 => _18.choices, 'optionalAccess', _19 => _19[0]]);
-    if (!choice) {
-      throw new Error(
-        `Workers AI structured output returned no choices: ${JSON.stringify(response)}`
-      );
-    }
-    const rawContent = _nullishCoalesce(_optionalChain([choice, 'access', _20 => _20.message, 'optionalAccess', _21 => _21.content]), () => ( ""));
-    let data;
-    let rawText;
-    if (typeof rawContent === "string") {
-      rawText = rawContent;
-      try {
-        data = JSON.parse(rawText);
-      } catch (e5) {
-        data = rawText;
-      }
-    } else {
-      data = rawContent;
-      rawText = JSON.stringify(rawContent);
-    }
-    return { data, rawText };
-  }
-};
-function createWorkersAiChat(model, config) {
-  return new WorkersAiTextAdapter(model, config);
-}
-
-
-
-
-exports.WorkersAiTextAdapter = WorkersAiTextAdapter; exports.createWorkersAiChat = createWorkersAiChat;
-//# sourceMappingURL=chunk-IQAGFHRT.cjs.map

package/dist/chunk-IQAGFHRT.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/home/runner/work/ai/ai/packages/tanstack-ai/dist/chunk-IQAGFHRT.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-IQAGFHRT.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"]}

package/dist/chunk-J5DSSZTO.js

@@ -1,71 +0,0 @@
-// src/adapters/gemini.ts
-import {
-  GeminiTextAdapter,
-  GeminiImageAdapter,
-  GeminiSummarizeAdapter,
-  GeminiTTSAdapter,
-  GeminiTextModels,
-  GeminiImageModels,
-  GeminiSummarizeModels,
-  GeminiTTSModels
-} from "@tanstack/ai-gemini";
-function buildGeminiGatewayConfig(config) {
-  if ("binding" in config) {
-    throw new Error(
-      "Gemini adapters do not support binding config. The Google GenAI SDK does not support a custom fetch function \u2014 only credential-based config ({ accountId, gatewayId }) is supported. See https://github.com/googleapis/js-genai/issues/999"
-    );
-  }
-  const headers = {};
-  if (config.apiKey && config.cfApiKey) {
-    headers["cf-aig-authorization"] = `Bearer ${config.cfApiKey}`;
-  }
-  if (config.skipCache) {
-    headers["cf-aig-skip-cache"] = "true";
-  }
-  if (typeof config.cacheTtl === "number") {
-    headers["cf-aig-cache-ttl"] = String(config.cacheTtl);
-  }
-  if (typeof config.customCacheKey === "string") {
-    headers["cf-aig-cache-key"] = config.customCacheKey;
-  }
-  if (typeof config.metadata === "object") {
-    headers["cf-aig-metadata"] = JSON.stringify(config.metadata);
-  }
-  const apiKey = config.apiKey ?? config.cfApiKey;
-  if (!apiKey) {
-    throw new Error(
-      "If you want to use BYOK or unified billing, you need to pass the Cloudflare AI Gateway API key."
-    );
-  }
-  return {
-    apiKey,
-    httpOptions: {
-      baseUrl: `https://gateway.ai.cloudflare.com/v1/${config.accountId}/${config.gatewayId}/google-ai-studio`,
-      headers: Object.keys(headers).length > 0 ? headers : void 0
-    }
-  };
-}
-function createGeminiChat(model, config) {
-  return new GeminiTextAdapter(buildGeminiGatewayConfig(config), model);
-}
-function createGeminiImage(model, config) {
-  return new GeminiImageAdapter(buildGeminiGatewayConfig(config), model);
-}
-function createGeminiSummarize(model, config) {
-  return new GeminiSummarizeAdapter(buildGeminiGatewayConfig(config), model);
-}
-function createGeminiTts(model, config) {
-  return new GeminiTTSAdapter(buildGeminiGatewayConfig(config), model);
-}
-
-export {
-  GeminiTextModels,
-  GeminiImageModels,
-  GeminiSummarizeModels,
-  GeminiTTSModels,
-  createGeminiChat,
-  createGeminiImage,
-  createGeminiSummarize,
-  createGeminiTts
-};
-//# sourceMappingURL=chunk-J5DSSZTO.js.map