@codeproxy/core 0.1.9 → 0.1.12

package/dist/index.d.cts

@@ -1,51 +1,3 @@
-type UpstreamFormat = 'anthropic' | 'openai-chat';
-interface CacheStats {
-    cachedTokens: number;
-    cacheCreationTokens: number;
-    inputTokens: number;
-    outputTokens: number;
-    totalTokens: number;
-}
-interface CreateResponsesFetchOptions {
-    /** Upstream API format. If omitted, inferred from `baseUrl`. */
-    upstreamFormat?: UpstreamFormat;
-    /** Upstream endpoint URL. Required. */
-    baseUrl: string;
-    /** Override upstream API version header (Anthropic only). */
-    apiVersion?: string;
-    /** Replace the caller-provided `model` field before translation. */
-    model?: string;
-    /** Extra headers merged into every upstream call. */
-    defaultHeaders?: Record<string, string>;
-    /** Underlying fetch. Defaults to `globalThis.fetch`. */
-    fetch?: typeof fetch;
-    /** Non-/responses traffic forward target. Defaults to `options.fetch`. */
-    passthroughFetch?: typeof fetch;
-    /** Drop image/file parts from user messages (e.g. DeepSeek text-only models). */
-    dropImages?: boolean;
-    /** Fallback thought signature for Gemini OpenAI-compatible tool histories. */
-    fallbackThoughtSignature?: string;
-    /** Optional callback to receive cache statistics. */
-    onCacheStats?: (stats: CacheStats) => void;
-    /** Override reasoning_effort sent to the upstream model (OpenAI Chat / Anthropic). */
-    reasoning_effort?: string;
-    /** Override thinking configuration sent to the upstream model. */
-    thinking?: unknown;
-    /** Timeout in milliseconds for upstream requests. Defaults to no timeout. */
-    timeoutMs?: number;
-    /** Fallback upstream for requests where the last user message contains images. */
-    fallbackUpstream?: {
-        baseUrl: string;
-        upstreamFormat?: UpstreamFormat;
-        model?: string;
-        defaultHeaders?: Record<string, string>;
-        apiVersion?: string;
-        reasoning_effort?: string;
-        thinking?: unknown;
-    };
-}
-declare function createResponsesFetch(options: CreateResponsesFetchOptions): typeof fetch;
-
 /**
  * OpenAI Responses API type definitions (subset used by this library).
  * Only the fields relevant to request translation and event emission are typed.
@@ -240,6 +192,7 @@ interface ResponsesOutputFunctionCall {
     type: 'function_call' | 'local_shell_call' | string;
     status: 'completed' | 'in_progress';
     name?: string;
+    namespace?: string;
     arguments?: string;
     call_id?: string;
     thought_signature?: string;
@@ -295,6 +248,58 @@ interface ResponsesStreamEvent {
     [key: string]: unknown;
 }
 
+type UpstreamFormat = 'anthropic' | 'openai-chat';
+interface CacheStats {
+    cachedTokens: number;
+    cacheCreationTokens: number;
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+}
+interface CreateResponsesFetchOptions {
+    /** Upstream API format. If omitted, inferred from `baseUrl`. */
+    upstreamFormat?: UpstreamFormat;
+    /** Upstream endpoint URL. Required. */
+    baseUrl: string;
+    /** Override upstream API version header (Anthropic only). */
+    apiVersion?: string;
+    /** Replace the caller-provided `model` field before translation. */
+    model?: string;
+    /** Extra headers merged into every upstream call. */
+    defaultHeaders?: Record<string, string>;
+    /** Underlying fetch. Defaults to `globalThis.fetch`. */
+    fetch?: typeof fetch;
+    /** Non-/responses traffic forward target. Defaults to `options.fetch`. */
+    passthroughFetch?: typeof fetch;
+    /** Drop image/file parts from user messages (e.g. DeepSeek text-only models). */
+    dropImages?: boolean;
+    /** Drop tools from the request before translation. Return true to drop a tool.
+     *  Use to reduce payload size when the upstream has a limit (e.g. DeepSeek ECONNRESET
+     *  on 252 KB requests with 143 tools). Applied before namespace flattening. */
+    dropTools?: (tool: ResponsesTool) => boolean;
+    /** Fallback thought signature for Gemini OpenAI-compatible tool histories. */
+    fallbackThoughtSignature?: string;
+    /** Optional callback to receive cache statistics. */
+    onCacheStats?: (stats: CacheStats) => void;
+    /** Override reasoning_effort sent to the upstream model (OpenAI Chat / Anthropic). */
+    reasoning_effort?: string;
+    /** Override thinking configuration sent to the upstream model. */
+    thinking?: unknown;
+    /** Timeout in milliseconds for upstream requests. Defaults to no timeout. */
+    timeoutMs?: number;
+    /** Fallback upstream for requests where the last user message contains images. */
+    fallbackUpstream?: {
+        baseUrl: string;
+        upstreamFormat?: UpstreamFormat;
+        model?: string;
+        defaultHeaders?: Record<string, string>;
+        apiVersion?: string;
+        reasoning_effort?: string;
+        thinking?: unknown;
+    };
+}
+declare function createResponsesFetch(options: CreateResponsesFetchOptions): typeof fetch;
+
 /** Anthropic Messages API type definitions (subset). */
 interface AnthropicTextBlock {
     type: 'text';
@@ -640,6 +645,9 @@ interface TranslateResponseOptions {
     responseId?: string;
     createdAt?: number;
     model?: string;
+    /** Chat Completions tool list from the translated request — used to recover
+     *  namespace when the upstream omits the "namespace." prefix in a tool call. */
+    requestTools?: unknown[];
 }
 /** Convert an OpenAI Chat response into a Responses-API response. */
 declare function translateResponse(body: OpenAiChatResponse, options?: TranslateResponseOptions): ResponsesResponse;

package/dist/index.d.ts

@@ -1,51 +1,3 @@
-type UpstreamFormat = 'anthropic' | 'openai-chat';
-interface CacheStats {
-    cachedTokens: number;
-    cacheCreationTokens: number;
-    inputTokens: number;
-    outputTokens: number;
-    totalTokens: number;
-}
-interface CreateResponsesFetchOptions {
-    /** Upstream API format. If omitted, inferred from `baseUrl`. */
-    upstreamFormat?: UpstreamFormat;
-    /** Upstream endpoint URL. Required. */
-    baseUrl: string;
-    /** Override upstream API version header (Anthropic only). */
-    apiVersion?: string;
-    /** Replace the caller-provided `model` field before translation. */
-    model?: string;
-    /** Extra headers merged into every upstream call. */
-    defaultHeaders?: Record<string, string>;
-    /** Underlying fetch. Defaults to `globalThis.fetch`. */
-    fetch?: typeof fetch;
-    /** Non-/responses traffic forward target. Defaults to `options.fetch`. */
-    passthroughFetch?: typeof fetch;
-    /** Drop image/file parts from user messages (e.g. DeepSeek text-only models). */
-    dropImages?: boolean;
-    /** Fallback thought signature for Gemini OpenAI-compatible tool histories. */
-    fallbackThoughtSignature?: string;
-    /** Optional callback to receive cache statistics. */
-    onCacheStats?: (stats: CacheStats) => void;
-    /** Override reasoning_effort sent to the upstream model (OpenAI Chat / Anthropic). */
-    reasoning_effort?: string;
-    /** Override thinking configuration sent to the upstream model. */
-    thinking?: unknown;
-    /** Timeout in milliseconds for upstream requests. Defaults to no timeout. */
-    timeoutMs?: number;
-    /** Fallback upstream for requests where the last user message contains images. */
-    fallbackUpstream?: {
-        baseUrl: string;
-        upstreamFormat?: UpstreamFormat;
-        model?: string;
-        defaultHeaders?: Record<string, string>;
-        apiVersion?: string;
-        reasoning_effort?: string;
-        thinking?: unknown;
-    };
-}
-declare function createResponsesFetch(options: CreateResponsesFetchOptions): typeof fetch;
-
 /**
  * OpenAI Responses API type definitions (subset used by this library).
  * Only the fields relevant to request translation and event emission are typed.
@@ -240,6 +192,7 @@ interface ResponsesOutputFunctionCall {
     type: 'function_call' | 'local_shell_call' | string;
     status: 'completed' | 'in_progress';
     name?: string;
+    namespace?: string;
     arguments?: string;
     call_id?: string;
     thought_signature?: string;
@@ -295,6 +248,58 @@ interface ResponsesStreamEvent {
     [key: string]: unknown;
 }
 
+type UpstreamFormat = 'anthropic' | 'openai-chat';
+interface CacheStats {
+    cachedTokens: number;
+    cacheCreationTokens: number;
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+}
+interface CreateResponsesFetchOptions {
+    /** Upstream API format. If omitted, inferred from `baseUrl`. */
+    upstreamFormat?: UpstreamFormat;
+    /** Upstream endpoint URL. Required. */
+    baseUrl: string;
+    /** Override upstream API version header (Anthropic only). */
+    apiVersion?: string;
+    /** Replace the caller-provided `model` field before translation. */
+    model?: string;
+    /** Extra headers merged into every upstream call. */
+    defaultHeaders?: Record<string, string>;
+    /** Underlying fetch. Defaults to `globalThis.fetch`. */
+    fetch?: typeof fetch;
+    /** Non-/responses traffic forward target. Defaults to `options.fetch`. */
+    passthroughFetch?: typeof fetch;
+    /** Drop image/file parts from user messages (e.g. DeepSeek text-only models). */
+    dropImages?: boolean;
+    /** Drop tools from the request before translation. Return true to drop a tool.
+     *  Use to reduce payload size when the upstream has a limit (e.g. DeepSeek ECONNRESET
+     *  on 252 KB requests with 143 tools). Applied before namespace flattening. */
+    dropTools?: (tool: ResponsesTool) => boolean;
+    /** Fallback thought signature for Gemini OpenAI-compatible tool histories. */
+    fallbackThoughtSignature?: string;
+    /** Optional callback to receive cache statistics. */
+    onCacheStats?: (stats: CacheStats) => void;
+    /** Override reasoning_effort sent to the upstream model (OpenAI Chat / Anthropic). */
+    reasoning_effort?: string;
+    /** Override thinking configuration sent to the upstream model. */
+    thinking?: unknown;
+    /** Timeout in milliseconds for upstream requests. Defaults to no timeout. */
+    timeoutMs?: number;
+    /** Fallback upstream for requests where the last user message contains images. */
+    fallbackUpstream?: {
+        baseUrl: string;
+        upstreamFormat?: UpstreamFormat;
+        model?: string;
+        defaultHeaders?: Record<string, string>;
+        apiVersion?: string;
+        reasoning_effort?: string;
+        thinking?: unknown;
+    };
+}
+declare function createResponsesFetch(options: CreateResponsesFetchOptions): typeof fetch;
+
 /** Anthropic Messages API type definitions (subset). */
 interface AnthropicTextBlock {
     type: 'text';
@@ -640,6 +645,9 @@ interface TranslateResponseOptions {
     responseId?: string;
     createdAt?: number;
     model?: string;
+    /** Chat Completions tool list from the translated request — used to recover
+     *  namespace when the upstream omits the "namespace." prefix in a tool call. */
+    requestTools?: unknown[];
 }
 /** Convert an OpenAI Chat response into a Responses-API response. */
 declare function translateResponse(body: OpenAiChatResponse, options?: TranslateResponseOptions): ResponsesResponse;

package/dist/index.js

@@ -1412,6 +1412,32 @@ function mapTools2(tools) {
       });
       continue;
     }
+    if (tt === "namespace") {
+      const ns = tool.name;
+      const nested = tool.tools;
+      if (ns && Array.isArray(nested)) {
+        for (const sub of nested) {
+          if (!sub || typeof sub !== "object" || sub.type !== "function") {
+            continue;
+          }
+          const subName = sub.name;
+          if (!subName) {
+            continue;
+          }
+          const params = sub.parameters ?? { type: "object" };
+          out.push({
+            type: "function",
+            function: {
+              name: `${ns}.${subName}`,
+              description: sub.description ?? "",
+              // eslint-disable-next-line no-restricted-syntax -- schema is Record<string,unknown> by OpenAPI convention
+              parameters: params
+            }
+          });
+        }
+      }
+      continue;
+    }
   }
   return out;
 }
@@ -1490,6 +1516,36 @@ function repairToolMessageOrder(messages) {
 
 // src/translate/openai/translateResponse.ts
 var SHELL_TOOL_NAMES3 = /* @__PURE__ */ new Set(["shell", "container.exec", "shell_command"]);
+function buildShortNameToNamespace(tools) {
+  const map = /* @__PURE__ */ new Map();
+  for (const tool of tools) {
+    if (!tool || typeof tool !== "object") {
+      continue;
+    }
+    const entry = tool;
+    const fn = entry.function;
+    const flatName = typeof fn?.name === "string" ? fn.name : typeof entry.name === "string" ? entry.name : "";
+    const dotIdx = flatName.indexOf(".");
+    if (dotIdx !== -1) {
+      map.set(flatName.slice(dotIdx + 1), flatName.slice(0, dotIdx));
+      continue;
+    }
+    if (entry.type === "namespace" && typeof entry.name === "string" && Array.isArray(entry.tools)) {
+      const ns = entry.name;
+      for (const sub of entry.tools) {
+        if (!sub || typeof sub !== "object") {
+          continue;
+        }
+        const subEntry = sub;
+        const subName = typeof subEntry.name === "string" ? subEntry.name : "";
+        if (subName) {
+          map.set(subName, ns);
+        }
+      }
+    }
+  }
+  return map;
+}
 function translateResponse2(body, options = {}) {
   const createdAt = options.createdAt ?? body.created ?? Math.floor(Date.now() / 1e3);
   const id = options.responseId ?? body.id ?? makeId("resp");
@@ -1497,9 +1553,10 @@ function translateResponse2(body, options = {}) {
   const choice = body.choices?.[0];
   const message = choice?.message;
   const output = [];
+  const shortNameToNs = buildShortNameToNamespace(options.requestTools ?? []);
   if (message?.tool_calls?.length) {
     for (const tc of message.tool_calls) {
-      const item = mapToolCallToOutput(tc);
+      const item = mapToolCallToOutput(tc, shortNameToNs);
       if (item) {
         output.push(item);
       }
@@ -1536,7 +1593,7 @@ function translateResponse2(body, options = {}) {
     }
   };
 }
-function mapToolCallToOutput(tc) {
+function mapToolCallToOutput(tc, shortNameToNs) {
   const name = tc.function?.name;
   if (!name) {
     return void 0;
@@ -1546,11 +1603,23 @@ function mapToolCallToOutput(tc) {
   if (typeof args !== "string") {
     args = jsonStringifySafe(args ?? {});
   }
+  let resolvedName = name;
+  let resolvedNamespace;
+  if (!SHELL_TOOL_NAMES3.has(name)) {
+    const dotIdx = name.indexOf(".");
+    if (dotIdx !== -1) {
+      resolvedNamespace = name.slice(0, dotIdx);
+      resolvedName = name.slice(dotIdx + 1);
+    } else {
+      resolvedNamespace = shortNameToNs?.get(name);
+    }
+  }
   const item = {
     id: callId,
     type: "function_call",
     status: "completed",
-    name,
+    name: resolvedName,
+    ...resolvedNamespace ? { namespace: resolvedNamespace } : {},
     arguments: args,
     call_id: callId
   };
@@ -1558,7 +1627,7 @@ function mapToolCallToOutput(tc) {
   if (thoughtSignature) {
     item.thought_signature = thoughtSignature;
   }
-  if (SHELL_TOOL_NAMES3.has(name)) {
+  if (SHELL_TOOL_NAMES3.has(resolvedName)) {
     item.type = "local_shell_call";
     const parsed = safeJsonParse(args);
     item.action = { type: "exec", command: parsed?.command ?? [] };
@@ -1572,6 +1641,36 @@ function getThoughtSignature(tc) {
 
 // src/translate/openai/translateStream.ts
 var SHELL_TOOL_NAMES4 = /* @__PURE__ */ new Set(["shell", "container.exec", "shell_command"]);
+function buildShortNameToNamespace2(tools) {
+  const map = /* @__PURE__ */ new Map();
+  for (const tool of tools) {
+    if (!tool || typeof tool !== "object") {
+      continue;
+    }
+    const entry = tool;
+    const fn = entry.function;
+    const flatName = typeof fn?.name === "string" ? fn.name : typeof entry.name === "string" ? entry.name : "";
+    const dotIdx = flatName.indexOf(".");
+    if (dotIdx !== -1) {
+      map.set(flatName.slice(dotIdx + 1), flatName.slice(0, dotIdx));
+      continue;
+    }
+    if (entry.type === "namespace" && typeof entry.name === "string" && Array.isArray(entry.tools)) {
+      const ns = entry.name;
+      for (const sub of entry.tools) {
+        if (!sub || typeof sub !== "object") {
+          continue;
+        }
+        const subEntry = sub;
+        const subName = typeof subEntry.name === "string" ? subEntry.name : "";
+        if (subName) {
+          map.set(subName, ns);
+        }
+      }
+    }
+  }
+  return map;
+}
 async function* translateStream2(stream, options = {}) {
   const translator = new StreamTranslator2(options);
   yield translator.createInitialEvent();
@@ -1611,6 +1710,10 @@ var StreamTranslator2 = class {
     __publicField(this, "textItemIndex", -1);
     __publicField(this, "textBuffer", "");
     __publicField(this, "toolCalls", /* @__PURE__ */ new Map());
+    // shortName → namespace reverse map built from flattened namespace tools in
+    // the translated request.  Used to restore the namespace when an upstream
+    // (e.g. DeepSeek) omits the "namespace." prefix in its tool-call response.
+    __publicField(this, "shortNameToNamespace");
     __publicField(this, "inputTokens", 0);
     __publicField(this, "outputTokens", 0);
     __publicField(this, "cachedTokens", 0);
@@ -1618,6 +1721,7 @@ var StreamTranslator2 = class {
     this.responseId = options.responseId ?? makeId("resp");
     this.createdAt = options.createdAt ?? Math.floor(Date.now() / 1e3);
     this.metadata = options.requestMetadata ?? {};
+    this.shortNameToNamespace = buildShortNameToNamespace2(this.metadata.tools ?? []);
   }
   createInitialEvent() {
     const toolsArr = this.metadata.tools ?? [];
@@ -1739,6 +1843,18 @@ var StreamTranslator2 = class {
     for (const state of this.toolCalls.values()) {
       const item = state.item;
       item.status = "completed";
+      if (item.name && !SHELL_TOOL_NAMES4.has(item.name)) {
+        const dotIdx = item.name.indexOf(".");
+        if (dotIdx !== -1) {
+          item.namespace = item.name.slice(0, dotIdx);
+          item.name = item.name.slice(dotIdx + 1);
+        } else {
+          const ns = this.shortNameToNamespace.get(item.name);
+          if (ns) {
+            item.namespace = ns;
+          }
+        }
+      }
       if (item.name && SHELL_TOOL_NAMES4.has(item.name)) {
         item.type = "local_shell_call";
         const parsed = safeJsonParse(item.arguments ?? "");
@@ -1800,6 +1916,109 @@ function getThoughtSignature2(tc) {
   return typeof sig === "string" && sig ? sig : void 0;
 }
 
+// src/tool-name-sanitizer.ts
+function sanitizeUpstreamToolNames(body) {
+  const map = /* @__PURE__ */ new Map();
+  const tools = body.tools;
+  if (!Array.isArray(tools)) {
+    return map;
+  }
+  for (const tool of tools) {
+    if (!tool || typeof tool !== "object") {
+      continue;
+    }
+    const toolRecord = tool;
+    const fn = toolRecord.function;
+    if (!fn || typeof fn !== "object") {
+      continue;
+    }
+    const fnRecord = fn;
+    const original = fnRecord.name;
+    if (typeof original !== "string") {
+      continue;
+    }
+    const sanitized = original.replace(/[^a-zA-Z0-9_-]/g, "_");
+    if (sanitized !== original) {
+      map.set(sanitized, original);
+      fnRecord.name = sanitized;
+    }
+  }
+  return map;
+}
+function restoreToolNamesInChatResponse(body, map) {
+  if (map.size === 0) {
+    return body;
+  }
+  const choices = body.choices;
+  if (!Array.isArray(choices)) {
+    return body;
+  }
+  for (const choice of choices) {
+    if (!choice || typeof choice !== "object") {
+      continue;
+    }
+    const choiceRecord = choice;
+    const message = choiceRecord.message;
+    if (!message || typeof message !== "object") {
+      continue;
+    }
+    const msg = message;
+    const toolCalls = msg.tool_calls;
+    if (!Array.isArray(toolCalls)) {
+      continue;
+    }
+    for (const tc of toolCalls) {
+      if (!tc || typeof tc !== "object") {
+        continue;
+      }
+      const callRecord = tc;
+      const fn = callRecord.function;
+      if (!fn || typeof fn !== "object") {
+        continue;
+      }
+      const fnRecord = fn;
+      const name = fnRecord.name;
+      if (typeof name === "string" && map.has(name)) {
+        fnRecord.name = map.get(name);
+      }
+    }
+  }
+  return body;
+}
+function createToolNameRestoreStream(body, map) {
+  if (map.size === 0) {
+    return body;
+  }
+  const entries = Array.from(map.entries());
+  const decoder = new TextDecoder();
+  const encoder = new TextEncoder();
+  const reader = body.getReader();
+  return new ReadableStream({
+    async pull(controller) {
+      try {
+        const { value, done } = await reader.read();
+        if (done) {
+          controller.close();
+          return;
+        }
+        let text = decoder.decode(value, { stream: true });
+        for (const [sanitized, original] of entries) {
+          const escapedSanitized = sanitized.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+          const re = new RegExp(`("name"\\s*:\\s*")${escapedSanitized}(")`, "g");
+          text = text.replace(re, `$1${original}$2`);
+        }
+        controller.enqueue(encoder.encode(text));
+      } catch (err) {
+        controller.error(err);
+      }
+    },
+    cancel() {
+      reader.cancel().catch(() => {
+      });
+    }
+  });
+}
+
 // src/fetch.ts
 function createResponsesFetch(options) {
   if (!options.baseUrl) {
@@ -1968,6 +2187,10 @@ async function handleResponses(request, format, options, baseFetch, incomingHead
     const fbModel = fb.model ? `, model: ${fb.model}` : "";
     console.warn(`[fallback] last user message has image, routing to ${fb.baseUrl}${fbModel}`);
   }
+  if (options.dropTools && request.tools) {
+    const { dropTools } = options;
+    request = { ...request, tools: request.tools.filter((tool) => !dropTools(tool)) };
+  }
   const streaming = request.stream ?? false;
   const resolvedUrl = normalizeBaseUrl(options.baseUrl, format);
   const { upstreamBody, requestMetadata } = buildUpstreamBody(
@@ -1981,6 +2204,10 @@ async function handleResponses(request, format, options, baseFetch, incomingHead
     options.fallbackThoughtSignature
   );
   const upstreamHeaders = buildUpstreamHeaders(format, options, incomingHeaders);
+  const toolNameMap = format === "openai-chat" ? (
+    // eslint-disable-next-line no-restricted-syntax -- upstreamBody is unknown; cast to Record for tool mutation
+    sanitizeUpstreamToolNames(upstreamBody)
+  ) : /* @__PURE__ */ new Map();
   const upstream = await baseFetch(resolvedUrl, {
     method: "POST",
     headers: upstreamHeaders,
@@ -1994,13 +2221,23 @@ async function handleResponses(request, format, options, baseFetch, incomingHead
     });
   }
   if (!streaming) {
-    const body = await upstream.json();
+    const rawBody = await upstream.json();
+    const restoredBody = restoreToolNamesInChatResponse(
+      // eslint-disable-next-line no-restricted-syntax -- upstream json() returns unknown; cast to Record for tool name restoration
+      rawBody,
+      toolNameMap
+    );
     const translated = format === "anthropic" ? (
       // eslint-disable-next-line no-restricted-syntax -- union type narrowing requires type assertion
-      translateResponse(body, { model: request.model })
+      translateResponse(restoredBody, {
+        model: request.model
+      })
     ) : (
       // eslint-disable-next-line no-restricted-syntax -- union type narrowing requires type assertion
-      translateResponse2(body, { model: request.model })
+      translateResponse2(restoredBody, {
+        model: request.model,
+        requestTools: request.tools ?? []
+      })
     );
     options.onCacheStats?.(extractCacheStatsFromResponse(translated));
     return new Response(JSON.stringify(translated), {
@@ -2011,7 +2248,8 @@ async function handleResponses(request, format, options, baseFetch, incomingHead
   if (!upstream.body) {
     return jsonErrorResponse(502, "Upstream streaming response has no body");
   }
-  const events = format === "anthropic" ? translateStream(upstream.body, { model: request.model, requestMetadata }) : translateStream2(upstream.body, { model: request.model, requestMetadata });
+  const upstreamBodyStream = createToolNameRestoreStream(upstream.body, toolNameMap);
+  const events = format === "anthropic" ? translateStream(upstreamBodyStream, { model: request.model, requestMetadata }) : translateStream2(upstreamBodyStream, { model: request.model, requestMetadata });
   return new Response(
     responsesEventsToSseStream(collectCacheStatsFromStream(events, options.onCacheStats)),
     {