@wrongstack/providers 0.1.9 → 0.1.10

package/dist/index.js

@@ -37,6 +37,8 @@ async function aggregateStream(stream, onEvent) {
   const textBuffers = [];
   let currentTextIndex = -1;
   const toolBuffers = /* @__PURE__ */ new Map();
+  const thinkingBuffers = [];
+  let currentThinkingIndex = -1;
   const blockOrder = [];
   for await (const ev of stream) {
     if (onEvent) onEvent(ev);
@@ -74,10 +76,45 @@ async function aggregateStream(stream, onEvent) {
           } else {
             b.input = { __raw: ev.input };
           }
+          if (ev.providerMeta) b.providerMeta = ev.providerMeta;
         }
         currentTextIndex = -1;
         break;
       }
+      case "thinking_start": {
+        currentTextIndex = -1;
+        currentThinkingIndex = thinkingBuffers.length;
+        thinkingBuffers.push({
+          textBuf: "",
+          ...ev.providerMeta ? { providerMeta: ev.providerMeta } : {}
+        });
+        blockOrder.push({ kind: "thinking", idx: currentThinkingIndex });
+        break;
+      }
+      case "thinking_delta": {
+        if (currentThinkingIndex === -1) {
+          currentThinkingIndex = thinkingBuffers.length;
+          thinkingBuffers.push({ textBuf: "" });
+          blockOrder.push({ kind: "thinking", idx: currentThinkingIndex });
+        }
+        const t = thinkingBuffers[currentThinkingIndex];
+        if (t) t.textBuf += ev.text;
+        break;
+      }
+      case "thinking_signature": {
+        if (currentThinkingIndex === -1) {
+          currentThinkingIndex = thinkingBuffers.length;
+          thinkingBuffers.push({ textBuf: "" });
+          blockOrder.push({ kind: "thinking", idx: currentThinkingIndex });
+        }
+        const t = thinkingBuffers[currentThinkingIndex];
+        if (t) t.signature = ev.signature;
+        break;
+      }
+      case "thinking_stop": {
+        currentThinkingIndex = -1;
+        break;
+      }
       case "message_stop":
         stopReason = ev.stopReason;
         usage = ev.usage;
@@ -89,15 +126,28 @@ async function aggregateStream(stream, onEvent) {
     if (b.kind === "text") {
       const text = textBuffers[b.idx] ?? "";
       if (text) content.push({ type: "text", text });
+    } else if (b.kind === "thinking") {
+      const t = thinkingBuffers[b.idx];
+      if (!t || !t.textBuf && !t.signature) continue;
+      const block = { type: "thinking", thinking: t.textBuf };
+      if (t.signature) block.signature = t.signature;
+      if (t.providerMeta && Object.keys(t.providerMeta).length > 0) {
+        block.providerMeta = t.providerMeta;
+      }
+      content.push(block);
     } else {
       const tb = toolBuffers.get(b.id);
       if (tb) {
-        content.push({
+        const block = {
           type: "tool_use",
           id: b.id,
           name: tb.name,
           input: tb.input && typeof tb.input === "object" && !Array.isArray(tb.input) ? tb.input : {}
-        });
+        };
+        if (tb.providerMeta && Object.keys(tb.providerMeta).length > 0) {
+          block.providerMeta = tb.providerMeta;
+        }
+        content.push(block);
       }
     }
   }
@@ -109,6 +159,9 @@ var init_aggregate = __esm({
     init_tool_input();
   }
 });
+
+// src/anthropic.ts
+init_tool_input();
 function parseProviderHttpError(providerId, status, rawText) {
   const body = parseBody(rawText);
   const retryable = isRetryable(status, body.type);
@@ -162,71 +215,69 @@ function stringOf(v) {
   return typeof v === "string" && v.length > 0 ? v : void 0;
 }
 
-// src/anthropic.ts
-init_tool_input();
-
-// src/tool-format/to-anthropic.ts
-function toolsToAnthropic(tools) {
-  return tools.map((t) => ({
-    name: t.name,
-    description: t.description,
-    input_schema: t.inputSchema ?? {
-      type: "object",
-      properties: {}
-    }
-  }));
-}
-
-// src/stop-reason.ts
-function normalizeAnthropic(stop) {
-  switch (stop) {
-    case "end_turn":
-      return "end_turn";
-    case "tool_use":
-      return "tool_use";
-    case "max_tokens":
-      return "max_tokens";
-    case "stop_sequence":
-      return "stop_sequence";
-    case "refusal":
-      return "refusal";
-    default:
-      return "end_turn";
-  }
-}
-function normalizeOpenAI(stop) {
-  switch (stop) {
-    case "stop":
-      return "end_turn";
-    case "tool_calls":
-    case "function_call":
-      return "tool_use";
-    case "length":
-      return "max_tokens";
-    case "content_filter":
-      return "refusal";
-    default:
-      return "end_turn";
-  }
-}
-function normalizeGemini(stop) {
-  switch (stop) {
-    case "SAFETY":
-    case "RECITATION":
-    case "hallucination":
-      return "refusal";
-    case "stop":
-    case "STOP":
-      return "end_turn";
-    case "max_tokens":
-    case "MAX_TOKENS":
-      return "max_tokens";
-    case "tool_use":
-    case "TOOL_USE":
-      return "tool_use";
-    default:
-      return "end_turn";
+// src/family-capabilities.ts
+var CAPABILITIES_BY_FAMILY = {
+  anthropic: {
+    tools: true,
+    parallelTools: true,
+    vision: true,
+    streaming: true,
+    promptCache: true,
+    systemPrompt: true,
+    jsonMode: false,
+    maxContext: 2e5,
+    cacheControl: "native"
+  },
+  openai: {
+    tools: true,
+    parallelTools: true,
+    vision: true,
+    streaming: true,
+    promptCache: false,
+    systemPrompt: true,
+    jsonMode: true,
+    maxContext: 128e3,
+    cacheControl: "auto"
+  },
+  "openai-compatible": {
+    tools: true,
+    parallelTools: true,
+    vision: false,
+    streaming: true,
+    promptCache: false,
+    systemPrompt: true,
+    jsonMode: false,
+    maxContext: 32e3,
+    cacheControl: "none"
+  },
+  google: {
+    tools: true,
+    parallelTools: true,
+    vision: true,
+    streaming: true,
+    promptCache: false,
+    systemPrompt: true,
+    jsonMode: true,
+    maxContext: 1e6,
+    cacheControl: "none"
+  },
+  unsupported: {
+    tools: false,
+    parallelTools: false,
+    vision: false,
+    streaming: false,
+    promptCache: false,
+    systemPrompt: false,
+    jsonMode: false,
+    maxContext: 0,
+    cacheControl: "none"
   }
+};
+function capabilitiesForFamily(family, overrides = {}) {
+  return {
+    ...CAPABILITIES_BY_FAMILY[family] ?? CAPABILITIES_BY_FAMILY.unsupported,
+    ...overrides
+  };
 }
 
 // src/sse.ts
@@ -318,6 +369,70 @@ function splitBuffer(buf) {
 function isNodeReadable(b) {
   return !!b && typeof b === "object" && typeof b.pipe === "function" && typeof b.on === "function";
 }
+
+// src/stop-reason.ts
+function normalizeAnthropic(stop) {
+  switch (stop) {
+    case "end_turn":
+      return "end_turn";
+    case "tool_use":
+      return "tool_use";
+    case "max_tokens":
+      return "max_tokens";
+    case "stop_sequence":
+      return "stop_sequence";
+    case "refusal":
+      return "refusal";
+    default:
+      return "end_turn";
+  }
+}
+function normalizeOpenAI(stop) {
+  switch (stop) {
+    case "stop":
+      return "end_turn";
+    case "tool_calls":
+    case "function_call":
+      return "tool_use";
+    case "length":
+      return "max_tokens";
+    case "content_filter":
+      return "refusal";
+    default:
+      return "end_turn";
+  }
+}
+function normalizeGemini(stop) {
+  switch (stop) {
+    case "SAFETY":
+    case "RECITATION":
+    case "hallucination":
+      return "refusal";
+    case "stop":
+    case "STOP":
+      return "end_turn";
+    case "max_tokens":
+    case "MAX_TOKENS":
+      return "max_tokens";
+    case "tool_use":
+    case "TOOL_USE":
+      return "tool_use";
+    default:
+      return "end_turn";
+  }
+}
+
+// src/tool-format/to-anthropic.ts
+function toolsToAnthropic(tools) {
+  return tools.map((t) => ({
+    name: t.name,
+    description: t.description,
+    input_schema: t.inputSchema ?? {
+      type: "object",
+      properties: {}
+    }
+  }));
+}
 async function safeText(res) {
   try {
     return await res.text();
@@ -353,13 +468,10 @@ var WireAdapter = class {
       });
     } catch (err) {
       if (opts.signal.aborted) throw err;
-      throw new ProviderError(
-        err instanceof Error ? err.message : String(err),
-        0,
-        true,
-        this.id,
-        { cause: err, body: { message: err instanceof Error ? err.message : String(err) } }
-      );
+      throw new ProviderError(err instanceof Error ? err.message : String(err), 0, true, this.id, {
+        cause: err,
+        body: { message: err instanceof Error ? err.message : String(err) }
+      });
     }
     if (!httpRes.ok) {
       const text = await safeText(httpRes);
@@ -380,71 +492,6 @@ var WireAdapter = class {
   }
 };
 
-// src/family-capabilities.ts
-var CAPABILITIES_BY_FAMILY = {
-  anthropic: {
-    tools: true,
-    parallelTools: true,
-    vision: true,
-    streaming: true,
-    promptCache: true,
-    systemPrompt: true,
-    jsonMode: false,
-    maxContext: 2e5,
-    cacheControl: "native"
-  },
-  openai: {
-    tools: true,
-    parallelTools: true,
-    vision: true,
-    streaming: true,
-    promptCache: false,
-    systemPrompt: true,
-    jsonMode: true,
-    maxContext: 128e3,
-    cacheControl: "auto"
-  },
-  "openai-compatible": {
-    tools: true,
-    parallelTools: true,
-    vision: false,
-    streaming: true,
-    promptCache: false,
-    systemPrompt: true,
-    jsonMode: false,
-    maxContext: 32e3,
-    cacheControl: "none"
-  },
-  google: {
-    tools: true,
-    parallelTools: true,
-    vision: true,
-    streaming: true,
-    promptCache: false,
-    systemPrompt: true,
-    jsonMode: true,
-    maxContext: 1e6,
-    cacheControl: "none"
-  },
-  unsupported: {
-    tools: false,
-    parallelTools: false,
-    vision: false,
-    streaming: false,
-    promptCache: false,
-    systemPrompt: false,
-    jsonMode: false,
-    maxContext: 0,
-    cacheControl: "none"
-  }
-};
-function capabilitiesForFamily(family, overrides = {}) {
-  return {
-    ...CAPABILITIES_BY_FAMILY[family] ?? CAPABILITIES_BY_FAMILY.unsupported,
-    ...overrides
-  };
-}
-
 // src/anthropic.ts
 var DEFAULT_BASE = "https://api.anthropic.com";
 var DEFAULT_VERSION = "2023-06-01";
@@ -539,6 +586,9 @@ async function* parseAnthropicStream(body, fallbackModel) {
           }
         } else if (cb?.type === "text") {
           blocks.set(index, { kind: "text", partial: "" });
+        } else if (cb?.type === "thinking" || cb?.type === "redacted_thinking") {
+          blocks.set(index, { kind: "thinking", partial: "" });
+          yield { type: "thinking_start" };
         } else {
           blocks.set(index, { kind: "unknown", partial: "" });
         }
@@ -556,6 +606,10 @@ async function* parseAnthropicStream(body, fallbackModel) {
             block.partial += delta.partial_json;
             yield { type: "tool_use_input_delta", id: block.id, partial: delta.partial_json };
           }
+        } else if (delta.type === "thinking_delta" && typeof delta.thinking === "string") {
+          yield { type: "thinking_delta", text: delta.thinking };
+        } else if (delta.type === "signature_delta" && typeof delta.signature === "string") {
+          yield { type: "thinking_signature", signature: delta.signature };
         }
         break;
       }
@@ -565,6 +619,8 @@ async function* parseAnthropicStream(body, fallbackModel) {
         if (block?.kind === "tool_use" && block.id) {
           const input = parseToolInput(block.partial);
           yield { type: "tool_use_stop", id: block.id, input };
+        } else if (block?.kind === "thinking") {
+          yield { type: "thinking_stop" };
         }
         break;
       }
@@ -582,20 +638,237 @@ async function* parseAnthropicStream(body, fallbackModel) {
         break;
       case "error": {
         const err = ev["error"];
-        throw new ProviderError(
-          err?.message ?? "Anthropic stream error",
-          0,
-          false,
-          "anthropic",
-          { body: { type: err?.type, message: err?.message } }
-        );
+        throw new ProviderError(err?.message ?? "Anthropic stream error", 0, false, "anthropic", {
+          body: { type: err?.type, message: err?.message }
+        });
+      }
+    }
+  }
+  if (started) {
+    yield { type: "message_stop", stopReason, usage };
+  }
+}
+var DEFAULT_BASE2 = "https://generativelanguage.googleapis.com/v1beta";
+var GoogleProvider = class extends WireAdapter {
+  id;
+  capabilities;
+  opts;
+  constructor(opts) {
+    super(opts.apiKey, opts.baseUrl ?? DEFAULT_BASE2, opts.fetchImpl);
+    this.opts = opts;
+    this.id = opts.id ?? "google";
+    this.capabilities = capabilitiesForFamily("google", {
+      ...opts.capabilities
+    });
+  }
+  buildUrl(req) {
+    return `${this.baseUrl}/models/${encodeURIComponent(req.model)}:streamGenerateContent?alt=sse`;
+  }
+  buildHeaders(req) {
+    return {
+      ...super.buildHeaders(req),
+      "x-goog-api-key": this.apiKey
+    };
+  }
+  buildBody(req) {
+    const body = {
+      contents: messagesToGemini(req.messages),
+      generationConfig: this.buildGenConfig(req)
+    };
+    if (req.system && req.system.length > 0) {
+      body["systemInstruction"] = {
+        parts: req.system.map((b) => ({ text: b.text }))
+      };
+    }
+    if (req.tools && req.tools.length > 0) {
+      body["tools"] = [{ functionDeclarations: toolsToGemini(req.tools) }];
+    }
+    return body;
+  }
+  parseStream(body, fallbackModel) {
+    return parseGoogleStream(body, fallbackModel);
+  }
+  translateError(status, text) {
+    return parseProviderHttpError(this.id, status, text);
+  }
+  buildGenConfig(req) {
+    const cfg = { maxOutputTokens: req.maxTokens };
+    if (req.temperature !== void 0) cfg["temperature"] = req.temperature;
+    if (req.topP !== void 0) cfg["topP"] = req.topP;
+    if (req.stopSequences) cfg["stopSequences"] = req.stopSequences;
+    return cfg;
+  }
+};
+function toolsToGemini(tools) {
+  return tools.map((t) => ({
+    name: t.name,
+    description: t.description,
+    parameters: sanitizeSchemaForGemini(t.inputSchema) ?? {
+      type: "object",
+      properties: {}
+    }
+  }));
+}
+var GEMINI_ALLOWED_KEYS = /* @__PURE__ */ new Set([
+  "type",
+  "format",
+  "description",
+  "nullable",
+  "enum",
+  "items",
+  "properties",
+  "required",
+  "anyOf",
+  "minLength",
+  "maxLength",
+  "pattern",
+  "minimum",
+  "maximum",
+  "minItems",
+  "maxItems",
+  "minProperties",
+  "maxProperties",
+  "propertyOrdering",
+  "title"
+]);
+function sanitizeSchemaForGemini(node) {
+  if (node === null || node === void 0) return void 0;
+  if (Array.isArray(node)) {
+    return void 0;
+  }
+  if (typeof node !== "object") return void 0;
+  const src = node;
+  const out = {};
+  for (const [k, v] of Object.entries(src)) {
+    if (!GEMINI_ALLOWED_KEYS.has(k)) continue;
+    if (k === "properties" && v && typeof v === "object") {
+      const props = {};
+      for (const [pname, pschema] of Object.entries(v)) {
+        const cleaned = sanitizeSchemaForGemini(pschema);
+        if (cleaned) props[pname] = cleaned;
+      }
+      out["properties"] = props;
+    } else if (k === "items") {
+      const cleaned = sanitizeSchemaForGemini(v);
+      if (cleaned) out["items"] = cleaned;
+    } else if (k === "anyOf" && Array.isArray(v)) {
+      const cleaned = v.map((s) => sanitizeSchemaForGemini(s)).filter((s) => s !== void 0);
+      if (cleaned.length > 0) out["anyOf"] = cleaned;
+    } else if (k === "required" && Array.isArray(v)) {
+      out["required"] = v.filter((s) => typeof s === "string");
+    } else if (k === "enum" && Array.isArray(v)) {
+      out["enum"] = v;
+    } else {
+      out[k] = v;
+    }
+  }
+  return out;
+}
+function messagesToGemini(messages) {
+  const out = [];
+  for (const m of messages) {
+    if (m.role === "system") continue;
+    const blocks = typeof m.content === "string" ? [{ type: "text", text: m.content }] : m.content;
+    if (m.role === "assistant") {
+      const parts = [];
+      for (const b of blocks) {
+        if (b.type === "text" && b.text) parts.push({ text: b.text });
+        else if (b.type === "tool_use") {
+          const part = {
+            functionCall: { name: b.name, args: b.input }
+          };
+          const sig = b.providerMeta?.["google.thoughtSignature"];
+          if (typeof sig === "string" && sig.length > 0) {
+            part.thoughtSignature = sig;
+          }
+          parts.push(part);
+        }
+      }
+      if (parts.length > 0) out.push({ role: "model", parts });
+      continue;
+    }
+    const textParts = [];
+    const functionParts = [];
+    for (const b of blocks) {
+      if (b.type === "text" && b.text) textParts.push({ text: b.text });
+      else if (b.type === "tool_result") {
+        const responseText = typeof b.content === "string" ? b.content : JSON.stringify(b.content);
+        const fnName = b.name ?? b.tool_use_id;
+        functionParts.push({
+          functionResponse: {
+            name: fnName,
+            response: { content: responseText }
+          }
+        });
+      } else if (b.type === "image" && b.source.type === "base64") {
+        textParts.push({
+          inlineData: {
+            mimeType: b.source.media_type ?? "image/png",
+            data: b.source.data ?? ""
+          }
+        });
+      }
+    }
+    const userParts = [...textParts];
+    if (functionParts.length > 0) userParts.push(...functionParts);
+    if (userParts.length > 0) out.push({ role: "user", parts: userParts });
+  }
+  return out;
+}
+async function* parseGoogleStream(body, fallbackModel) {
+  let model = fallbackModel;
+  let usage = { input: 0, output: 0 };
+  let stopReason = "end_turn";
+  let started = false;
+  let sawFunctionCall = false;
+  for await (const msg of parseSSE(body)) {
+    if (!msg.data || msg.data === "[DONE]") continue;
+    const parsed = safeParse(msg.data);
+    if (!parsed.ok || !parsed.value) continue;
+    const obj = parsed.value;
+    if (obj.modelVersion) model = obj.modelVersion;
+    if (!started) {
+      started = true;
+      yield { type: "message_start", model };
+    }
+    const candidate = obj.candidates?.[0];
+    for (const part of candidate?.content?.parts ?? []) {
+      if (typeof part.text === "string" && part.text.length > 0) {
+        yield { type: "text_delta", text: part.text };
+      } else if (part.functionCall) {
+        sawFunctionCall = true;
+        const id = randomUUID();
+        yield { type: "tool_use_start", id, name: part.functionCall.name };
+        const providerMeta = typeof part.thoughtSignature === "string" ? { "google.thoughtSignature": part.thoughtSignature } : void 0;
+        yield {
+          type: "tool_use_stop",
+          id,
+          input: part.functionCall.args ?? {},
+          ...providerMeta ? { providerMeta } : {}
+        };
       }
     }
+    if (candidate?.finishReason) {
+      stopReason = normalizeGemini(candidate.finishReason);
+    }
+    const u = obj.usageMetadata;
+    if (u) {
+      const cached = u.cachedContentTokenCount ?? 0;
+      const promptTotal = u.promptTokenCount ?? usage.input + cached;
+      usage = {
+        input: Math.max(0, promptTotal - cached),
+        output: u.candidatesTokenCount ?? usage.output,
+        cacheRead: cached || usage.cacheRead
+      };
+    }
   }
   if (started) {
-    yield { type: "message_stop", stopReason, usage };
+    const finalStop = sawFunctionCall ? "tool_use" : stopReason;
+    yield { type: "message_stop", stopReason: finalStop, usage };
   }
 }
+
+// src/openai.ts
 init_tool_input();
 
 // src/tool-format/to-openai.ts
@@ -645,7 +918,9 @@ function messagesToOpenAI(system, messages, opts = {}) {
       const blocks = normalizeContent(msg.content);
       const textBlocks = blocks.filter((b) => b.type === "text");
       const toolUses = blocks.filter((b) => b.type === "tool_use");
+      const thinkingBlocks = blocks.filter((b) => b.type === "thinking");
       const text = textBlocks.map((b) => b.text).join("");
+      const reasoning = thinkingBlocks.map((b) => b.thinking).filter((t) => t && t.length > 0).join("");
       const toolCalls = toolUses.map((u) => ({
         id: u.id,
         type: "function",
@@ -662,6 +937,9 @@ function messagesToOpenAI(system, messages, opts = {}) {
       } else {
         message.content = text;
       }
+      if (reasoning.length > 0) {
+        message.reasoning_content = reasoning;
+      }
       out.push(message);
     }
   }
@@ -693,13 +971,13 @@ function blocksToContentArray(blocks) {
 }
 
 // src/openai.ts
-var DEFAULT_BASE2 = "https://api.openai.com/v1";
+var DEFAULT_BASE3 = "https://api.openai.com/v1";
 var OpenAIProvider = class extends WireAdapter {
   id;
   capabilities;
   opts;
   constructor(opts) {
-    super(opts.apiKey, opts.baseUrl ?? DEFAULT_BASE2, opts.fetchImpl);
+    super(opts.apiKey, opts.baseUrl ?? DEFAULT_BASE3, opts.fetchImpl);
     this.opts = opts;
     this.id = opts.id ?? "openai";
     this.capabilities = capabilitiesForFamily("openai", {
@@ -771,6 +1049,7 @@ async function* parseOpenAIStream(body, fallbackModel) {
   let usage = { input: 0, output: 0 };
   let stopReason = "end_turn";
   let started = false;
+  let thinkingOpen = false;
   const toolByIndex = /* @__PURE__ */ new Map();
   for await (const msg of parseSSE(body)) {
     if (!msg.data || msg.data === "[DONE]") continue;
@@ -784,10 +1063,26 @@ async function* parseOpenAIStream(body, fallbackModel) {
     }
     const choices = obj["choices"];
     const choice = choices?.[0];
+    const reasoningDelta = typeof choice?.delta?.reasoning_content === "string" ? choice.delta.reasoning_content : typeof choice?.delta?.reasoning === "string" ? choice.delta.reasoning : void 0;
+    if (reasoningDelta && reasoningDelta.length > 0) {
+      if (!thinkingOpen) {
+        thinkingOpen = true;
+        yield { type: "thinking_start" };
+      }
+      yield { type: "thinking_delta", text: reasoningDelta };
+    }
     if (choice?.delta?.content) {
+      if (thinkingOpen) {
+        thinkingOpen = false;
+        yield { type: "thinking_stop" };
+      }
       yield { type: "text_delta", text: choice.delta.content };
     }
     if (choice?.delta?.tool_calls) {
+      if (thinkingOpen) {
+        thinkingOpen = false;
+        yield { type: "thinking_stop" };
+      }
       for (const tc of choice.delta.tool_calls) {
         const idx = tc.index ?? 0;
         let entry = toolByIndex.get(idx);
@@ -811,13 +1106,18 @@ async function* parseOpenAIStream(body, fallbackModel) {
     }
     const u = obj["usage"];
     if (u) {
+      const cached = u.prompt_tokens_details?.cached_tokens ?? 0;
+      const promptTotal = u.prompt_tokens ?? usage.input + cached;
       usage = {
-        input: u.prompt_tokens ?? usage.input,
+        input: Math.max(0, promptTotal - cached),
         output: u.completion_tokens ?? usage.output,
-        cacheRead: u.prompt_tokens_details?.cached_tokens ?? usage.cacheRead
+        cacheRead: cached || usage.cacheRead
       };
     }
   }
+  if (thinkingOpen) {
+    yield { type: "thinking_stop" };
+  }
   for (const entry of toolByIndex.values()) {
     const input = parseToolInput(entry.argBuf);
     yield { type: "tool_use_stop", id: entry.id, input };
@@ -860,220 +1160,6 @@ var OpenAICompatibleProvider = class extends OpenAIProvider {
     };
   }
 };
-var DEFAULT_BASE3 = "https://generativelanguage.googleapis.com/v1beta";
-var GoogleProvider = class extends WireAdapter {
-  id;
-  capabilities;
-  opts;
-  constructor(opts) {
-    super(opts.apiKey, opts.baseUrl ?? DEFAULT_BASE3, opts.fetchImpl);
-    this.opts = opts;
-    this.id = opts.id ?? "google";
-    this.capabilities = capabilitiesForFamily("google", {
-      ...opts.capabilities
-    });
-  }
-  buildUrl(req) {
-    return `${this.baseUrl}/models/${encodeURIComponent(req.model)}:streamGenerateContent?alt=sse`;
-  }
-  buildHeaders(req) {
-    return {
-      ...super.buildHeaders(req),
-      "x-goog-api-key": this.apiKey
-    };
-  }
-  buildBody(req) {
-    const body = {
-      contents: messagesToGemini(req.messages),
-      generationConfig: this.buildGenConfig(req)
-    };
-    if (req.system && req.system.length > 0) {
-      body["systemInstruction"] = {
-        parts: req.system.map((b) => ({ text: b.text }))
-      };
-    }
-    if (req.tools && req.tools.length > 0) {
-      body["tools"] = [{ functionDeclarations: toolsToGemini(req.tools) }];
-    }
-    return body;
-  }
-  parseStream(body, fallbackModel) {
-    return parseGoogleStream(body, fallbackModel);
-  }
-  translateError(status, text) {
-    return parseProviderHttpError(this.id, status, text);
-  }
-  buildGenConfig(req) {
-    const cfg = { maxOutputTokens: req.maxTokens };
-    if (req.temperature !== void 0) cfg["temperature"] = req.temperature;
-    if (req.topP !== void 0) cfg["topP"] = req.topP;
-    if (req.stopSequences) cfg["stopSequences"] = req.stopSequences;
-    return cfg;
-  }
-};
-function toolsToGemini(tools) {
-  return tools.map((t) => ({
-    name: t.name,
-    description: t.description,
-    parameters: sanitizeSchemaForGemini(t.inputSchema) ?? { type: "object", properties: {} }
-  }));
-}
-var GEMINI_ALLOWED_KEYS = /* @__PURE__ */ new Set([
-  "type",
-  "format",
-  "description",
-  "nullable",
-  "enum",
-  "items",
-  "properties",
-  "required",
-  "anyOf",
-  "minLength",
-  "maxLength",
-  "pattern",
-  "minimum",
-  "maximum",
-  "minItems",
-  "maxItems",
-  "minProperties",
-  "maxProperties",
-  "propertyOrdering",
-  "title"
-]);
-function sanitizeSchemaForGemini(node) {
-  if (node === null || node === void 0) return void 0;
-  if (Array.isArray(node)) {
-    return void 0;
-  }
-  if (typeof node !== "object") return void 0;
-  const src = node;
-  const out = {};
-  for (const [k, v] of Object.entries(src)) {
-    if (!GEMINI_ALLOWED_KEYS.has(k)) continue;
-    if (k === "properties" && v && typeof v === "object") {
-      const props = {};
-      for (const [pname, pschema] of Object.entries(v)) {
-        const cleaned = sanitizeSchemaForGemini(pschema);
-        if (cleaned) props[pname] = cleaned;
-      }
-      out["properties"] = props;
-    } else if (k === "items") {
-      const cleaned = sanitizeSchemaForGemini(v);
-      if (cleaned) out["items"] = cleaned;
-    } else if (k === "anyOf" && Array.isArray(v)) {
-      const cleaned = v.map((s) => sanitizeSchemaForGemini(s)).filter((s) => s !== void 0);
-      if (cleaned.length > 0) out["anyOf"] = cleaned;
-    } else if (k === "required" && Array.isArray(v)) {
-      out["required"] = v.filter((s) => typeof s === "string");
-    } else if (k === "enum" && Array.isArray(v)) {
-      out["enum"] = v;
-    } else {
-      out[k] = v;
-    }
-  }
-  return out;
-}
-function messagesToGemini(messages) {
-  const out = [];
-  for (const m of messages) {
-    if (m.role === "system") continue;
-    const blocks = typeof m.content === "string" ? [{ type: "text", text: m.content }] : m.content;
-    if (m.role === "assistant") {
-      const parts = [];
-      for (const b of blocks) {
-        if (b.type === "text" && b.text) parts.push({ text: b.text });
-        else if (b.type === "tool_use") {
-          const part = {
-            functionCall: { name: b.name, args: b.input }
-          };
-          const sig = b.providerMeta?.["google.thoughtSignature"];
-          if (typeof sig === "string" && sig.length > 0) {
-            part.thoughtSignature = sig;
-          }
-          parts.push(part);
-        }
-      }
-      if (parts.length > 0) out.push({ role: "model", parts });
-      continue;
-    }
-    const textParts = [];
-    const functionParts = [];
-    for (const b of blocks) {
-      if (b.type === "text" && b.text) textParts.push({ text: b.text });
-      else if (b.type === "tool_result") {
-        const responseText = typeof b.content === "string" ? b.content : JSON.stringify(b.content);
-        const fnName = b.name ?? b.tool_use_id;
-        functionParts.push({
-          functionResponse: {
-            name: fnName,
-            response: { content: responseText }
-          }
-        });
-      } else if (b.type === "image" && b.source.type === "base64") {
-        textParts.push({
-          inlineData: {
-            mimeType: b.source.media_type ?? "image/png",
-            data: b.source.data ?? ""
-          }
-        });
-      }
-    }
-    const userParts = [...textParts];
-    if (functionParts.length > 0) userParts.push(...functionParts);
-    if (userParts.length > 0) out.push({ role: "user", parts: userParts });
-  }
-  return out;
-}
-async function* parseGoogleStream(body, fallbackModel) {
-  let model = fallbackModel;
-  let usage = { input: 0, output: 0 };
-  let stopReason = "end_turn";
-  let started = false;
-  let sawFunctionCall = false;
-  for await (const msg of parseSSE(body)) {
-    if (!msg.data || msg.data === "[DONE]") continue;
-    const parsed = safeParse(msg.data);
-    if (!parsed.ok || !parsed.value) continue;
-    const obj = parsed.value;
-    if (obj.modelVersion) model = obj.modelVersion;
-    if (!started) {
-      started = true;
-      yield { type: "message_start", model };
-    }
-    const candidate = obj.candidates?.[0];
-    for (const part of candidate?.content?.parts ?? []) {
-      if (typeof part.text === "string" && part.text.length > 0) {
-        yield { type: "text_delta", text: part.text };
-      } else if (part.functionCall) {
-        sawFunctionCall = true;
-        const id = randomUUID();
-        yield { type: "tool_use_start", id, name: part.functionCall.name };
-        const providerMeta = typeof part.thoughtSignature === "string" ? { "google.thoughtSignature": part.thoughtSignature } : void 0;
-        yield {
-          type: "tool_use_stop",
-          id,
-          input: part.functionCall.args ?? {},
-          ...providerMeta ? { providerMeta } : {}
-        };
-      }
-    }
-    if (candidate?.finishReason) {
-      stopReason = normalizeGemini(candidate.finishReason);
-    }
-    const u = obj.usageMetadata;
-    if (u) {
-      usage = {
-        input: u.promptTokenCount ?? usage.input,
-        output: u.candidatesTokenCount ?? usage.output,
-        cacheRead: u.cachedContentTokenCount ?? usage.cacheRead
-      };
-    }
-  }
-  if (started) {
-    const finalStop = sawFunctionCall ? "tool_use" : stopReason;
-    yield { type: "message_stop", stopReason: finalStop, usage };
-  }
-}
 
 // src/wire-format.ts
 var WireFormatProvider = class extends WireAdapter {
@@ -1356,13 +1442,9 @@ var anthropicWireFormat = defineWireFormat({
         break;
       case "error": {
         const err = ev["error"];
-        throw new ProviderError(
-          err?.message ?? "Anthropic stream error",
-          0,
-          false,
-          "anthropic",
-          { body: { type: err?.type, message: err?.message } }
-        );
+        throw new ProviderError(err?.message ?? "Anthropic stream error", 0, false, "anthropic", {
+          body: { type: err?.type, message: err?.message }
+        });
       }
     }
     return out;
@@ -1421,6 +1503,7 @@ var openaiWireFormat = defineWireFormat({
     stopReason: "end_turn",
     started: false,
     textOpen: false,
+    thinkingOpen: false,
     toolByIndex: /* @__PURE__ */ new Map(),
     finalEmitted: false
   }),
@@ -1437,11 +1520,27 @@ var openaiWireFormat = defineWireFormat({
     }
     const choices = obj["choices"];
     const choice = choices?.[0];
+    const reasoningDelta = typeof choice?.delta?.reasoning_content === "string" ? choice.delta.reasoning_content : typeof choice?.delta?.reasoning === "string" ? choice.delta.reasoning : void 0;
+    if (reasoningDelta && reasoningDelta.length > 0) {
+      if (!state.thinkingOpen) {
+        state.thinkingOpen = true;
+        out.push({ type: "thinking_start" });
+      }
+      out.push({ type: "thinking_delta", text: reasoningDelta });
+    }
     if (choice?.delta?.content) {
+      if (state.thinkingOpen) {
+        state.thinkingOpen = false;
+        out.push({ type: "thinking_stop" });
+      }
       if (!state.textOpen) state.textOpen = true;
       out.push({ type: "text_delta", text: choice.delta.content });
     }
     if (choice?.delta?.tool_calls) {
+      if (state.thinkingOpen) {
+        state.thinkingOpen = false;
+        out.push({ type: "thinking_stop" });
+      }
       for (const tc of choice.delta.tool_calls) {
         const idx = tc.index ?? 0;
         let entry = state.toolByIndex.get(idx);
@@ -1466,10 +1565,12 @@ var openaiWireFormat = defineWireFormat({
     }
     const u = obj["usage"];
     if (u) {
+      const cached = u.prompt_tokens_details?.cached_tokens ?? 0;
+      const promptTotal = u.prompt_tokens ?? state.usage.input + cached;
       state.usage = {
-        input: u.prompt_tokens ?? state.usage.input,
+        input: Math.max(0, promptTotal - cached),
         output: u.completion_tokens ?? state.usage.output,
-        cacheRead: u.prompt_tokens_details?.cached_tokens ?? state.usage.cacheRead
+        cacheRead: cached || state.usage.cacheRead
       };
     }
     return out;
@@ -1478,6 +1579,10 @@ var openaiWireFormat = defineWireFormat({
     if (state.finalEmitted) return [];
     state.finalEmitted = true;
     const out = [];
+    if (state.thinkingOpen) {
+      state.thinkingOpen = false;
+      out.push({ type: "thinking_stop" });
+    }
     for (const entry of state.toolByIndex.values()) {
       const input = parseToolInput(entry.argBuf);
       out.push({ type: "tool_use_stop", id: entry.id, input });
@@ -1542,7 +1647,7 @@ var googleWireFormat = defineWireFormat({
         out.push({ type: "text_delta", text: part.text });
       } else if (part.functionCall) {
         state.sawFunctionCall = true;
-        const id = `${part.functionCall.name}_${Math.random().toString(36).slice(2, 10)}`;
+        const id = `${part.functionCall.name}_${randomUUID().slice(0, 8)}`;
         out.push({ type: "tool_use_start", id, name: part.functionCall.name });
         const providerMeta = typeof part.thoughtSignature === "string" ? { "google.thoughtSignature": part.thoughtSignature } : void 0;
         out.push({
@@ -1558,10 +1663,12 @@ var googleWireFormat = defineWireFormat({
     }
     const u = obj.usageMetadata;
     if (u) {
+      const cached = u.cachedContentTokenCount ?? 0;
+      const promptTotal = u.promptTokenCount ?? state.usage.input + cached;
       state.usage = {
-        input: u.promptTokenCount ?? state.usage.input,
+        input: Math.max(0, promptTotal - cached),
         output: u.candidatesTokenCount ?? state.usage.output,
-        cacheRead: u.cachedContentTokenCount ?? state.usage.cacheRead
+        cacheRead: cached || state.usage.cacheRead
       };
     }
     return out;
@@ -1725,6 +1832,12 @@ function contentFromAnthropic(blocks, opts = {}) {
         content: normalizeToolResultContent(b.content, opts),
         is_error: b.is_error
       });
+    } else if (b.type === "thinking" && typeof b.thinking === "string") {
+      out.push({
+        type: "thinking",
+        thinking: b.thinking,
+        ...b.signature ? { signature: b.signature } : {}
+      });
     } else if (b.type === "image" && b.source) {
       const src = b.source;
       const kind = src.type === "url" ? "url" : "base64";