@codeproxy/core 0.1.6 → 0.1.8

package/dist/index.d.cts

@@ -23,6 +23,8 @@ interface CreateResponsesFetchOptions {
     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). */
@@ -94,6 +96,7 @@ interface ResponsesFunctionCallItem {
     call_id?: string;
     name?: string;
     arguments?: string | Record<string, unknown>;
+    thought_signature?: string;
     [key: string]: unknown;
 }
 interface ResponsesFunctionCallOutputItem {
@@ -239,6 +242,7 @@ interface ResponsesOutputFunctionCall {
     name?: string;
     arguments?: string;
     call_id?: string;
+    thought_signature?: string;
     action?: {
         type?: string;
         command?: string[];
@@ -488,10 +492,18 @@ declare namespace index$2 {
 interface OpenAiChatToolCall {
     id?: string;
     type?: 'function' | string;
+    extra_content?: {
+        google?: {
+            thought_signature?: string;
+            [key: string]: unknown;
+        };
+        [key: string]: unknown;
+    };
     function?: {
         name?: string;
         arguments?: string | Record<string, unknown>;
     };
+    thought_signature?: string;
 }
 interface OpenAiChatMessage {
     role: 'system' | 'user' | 'assistant' | 'tool' | string;
@@ -575,10 +587,12 @@ interface OpenAiChatStreamDeltaToolCall {
     index: number;
     id?: string;
     type?: string;
+    extra_content?: OpenAiChatToolCall['extra_content'];
     function?: {
         name?: string;
         arguments?: string | Record<string, unknown>;
     };
+    thought_signature?: string;
 }
 interface OpenAiChatStreamDelta {
     role?: string;
@@ -613,6 +627,8 @@ interface TranslateRequestOptions {
     /** If true, strip `strict` from function tools (some upstreams reject it). */
     /** If true, drop image/file parts from user messages (e.g. DeepSeek text-only models). */
     dropImages?: boolean;
+    /** Fallback signature for Gemini OpenAI histories that lack returned signatures. */
+    fallbackThoughtSignature?: string;
 }
 interface TranslateRequestResult {
     request: OpenAiChatRequest;

package/dist/index.d.ts

@@ -23,6 +23,8 @@ interface CreateResponsesFetchOptions {
     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). */
@@ -94,6 +96,7 @@ interface ResponsesFunctionCallItem {
     call_id?: string;
     name?: string;
     arguments?: string | Record<string, unknown>;
+    thought_signature?: string;
     [key: string]: unknown;
 }
 interface ResponsesFunctionCallOutputItem {
@@ -239,6 +242,7 @@ interface ResponsesOutputFunctionCall {
     name?: string;
     arguments?: string;
     call_id?: string;
+    thought_signature?: string;
     action?: {
         type?: string;
         command?: string[];
@@ -488,10 +492,18 @@ declare namespace index$2 {
 interface OpenAiChatToolCall {
     id?: string;
     type?: 'function' | string;
+    extra_content?: {
+        google?: {
+            thought_signature?: string;
+            [key: string]: unknown;
+        };
+        [key: string]: unknown;
+    };
     function?: {
         name?: string;
         arguments?: string | Record<string, unknown>;
     };
+    thought_signature?: string;
 }
 interface OpenAiChatMessage {
     role: 'system' | 'user' | 'assistant' | 'tool' | string;
@@ -575,10 +587,12 @@ interface OpenAiChatStreamDeltaToolCall {
     index: number;
     id?: string;
     type?: string;
+    extra_content?: OpenAiChatToolCall['extra_content'];
     function?: {
         name?: string;
         arguments?: string | Record<string, unknown>;
     };
+    thought_signature?: string;
 }
 interface OpenAiChatStreamDelta {
     role?: string;
@@ -613,6 +627,8 @@ interface TranslateRequestOptions {
     /** If true, strip `strict` from function tools (some upstreams reject it). */
     /** If true, drop image/file parts from user messages (e.g. DeepSeek text-only models). */
     dropImages?: boolean;
+    /** Fallback signature for Gemini OpenAI histories that lack returned signatures. */
+    fallbackThoughtSignature?: string;
 }
 interface TranslateRequestResult {
     request: OpenAiChatRequest;

package/dist/index.js

@@ -525,9 +525,14 @@ function ensureEndsWithUser(messages) {
   return [...messages, { role: "user", content: [{ type: "text", text: "Continue." }] }];
 }
 function markBlocksForCache(blocks) {
+  let count = 0;
   for (const block of blocks) {
     if (!block.cache_control) {
       block.cache_control = { type: "ephemeral" };
+      count++;
+      if (count >= 3) {
+        break;
+      }
     }
   }
   return blocks;
@@ -836,6 +841,8 @@ var StreamTranslator = class {
   *finalize() {
     const items = [];
     if (this.textItem) {
+      this.textItem.status = "completed";
+      this.textItem.content[0].text = this.textBuffer;
       items.push({ index: this.textItemIndex, item: this.textItem });
     }
     for (const block of this.blocks.values()) {
@@ -947,15 +954,22 @@ var StreamTranslator = class {
     if (btype === "tool_use") {
       const outputIndex = this.outputCounter++;
       const callId = block.id ?? makeId("call");
+      const initialInput = typeof block.input === "object" && block.input !== null ? jsonStringifySafe(block.input) : "";
+      const hasInitialInput = initialInput !== "" && initialInput !== "{}";
       const item = {
         id: callId,
         type: "function_call",
         status: "in_progress",
         name: block.name ?? "",
-        arguments: "",
+        arguments: hasInitialInput ? initialInput : "",
         call_id: callId
       };
-      this.blocks.set(index, { type: "tool_use", outputIndex, item, buffer: "" });
+      this.blocks.set(index, {
+        type: "tool_use",
+        outputIndex,
+        item,
+        buffer: hasInitialInput ? initialInput : ""
+      });
       yield this.makeEvent("response.output_item.added", {
         response_id: this.responseId,
         output_index: outputIndex,
@@ -1057,9 +1071,66 @@ __export(openai_exports, {
   translateStream: () => translateStream2
 });
 
+// src/translate/openai/toolSchema.ts
+function sanitizeJsonSchema(schema, depth = 0) {
+  if (!schema || typeof schema !== "object" || Array.isArray(schema) || depth > 20) {
+    return schema;
+  }
+  const src = schema;
+  if ("$ref" in src) {
+    return src.description ? { description: src.description } : {};
+  }
+  const out = {};
+  for (const [key, val] of Object.entries(src)) {
+    if (isDroppedSchemaKeyword(key)) {
+      continue;
+    }
+    if (key === "properties" && val && typeof val === "object" && !Array.isArray(val)) {
+      const props = {};
+      for (const [propName, propSchema] of Object.entries(val)) {
+        props[propName] = sanitizeJsonSchema(propSchema, depth + 1);
+      }
+      out[key] = props;
+    } else if (key === "additionalProperties") {
+      if (typeof val !== "boolean") {
+        out[key] = sanitizeJsonSchema(val, depth + 1);
+      }
+    } else if (key === "items") {
+      out[key] = sanitizeJsonSchema(val, depth + 1);
+    } else if (isSchemaCompositionKeyword(key) && Array.isArray(val)) {
+      out[key] = val.map((schemaItem) => sanitizeJsonSchema(schemaItem, depth + 1));
+    } else {
+      out[key] = val;
+    }
+  }
+  return out;
+}
+function getValidFunctionNames(tools) {
+  const names = /* @__PURE__ */ new Set();
+  for (const tool of tools) {
+    const maybeTool = tool;
+    if (maybeTool.type === "function" && typeof maybeTool.function?.name === "string") {
+      names.add(maybeTool.function.name);
+    }
+  }
+  return names.size ? names : void 0;
+}
+function isDroppedSchemaKeyword(key) {
+  return key === "$schema" || key === "$defs" || key === "definitions" || key === "$id" || key === "$anchor" || key === "$comment";
+}
+function isSchemaCompositionKeyword(key) {
+  return key === "anyOf" || key === "oneOf" || key === "allOf" || key === "not";
+}
+
 // src/translate/openai/translateRequest.ts
 function translateRequest2(data, options = {}) {
   const messages = [];
+  const tools = mapTools2(data.tools ?? []);
+  const validFunctionNames = getValidFunctionNames(tools);
+  const context = {
+    validFunctionNames,
+    ignoredToolCallIds: /* @__PURE__ */ new Set()
+  };
   const systemContent = buildSystemContent(data.instructions);
   if (systemContent) {
     messages.push({ role: "system", content: systemContent });
@@ -1074,7 +1145,7 @@ function translateRequest2(data, options = {}) {
       continue;
     }
     const rawItem = raw;
-    processInputItem(rawItem, messages, options.dropImages);
+    processInputItem(rawItem, messages, options, context);
   }
   const request = {
     model: data.model,
@@ -1095,7 +1166,6 @@ function translateRequest2(data, options = {}) {
   if (typeof maxTokens === "number") {
     request.max_tokens = maxTokens;
   }
-  const tools = mapTools2(data.tools ?? []);
   if (tools.length) {
     request.tools = tools;
     const toolChoice = mapToolChoice2(data.tool_choice);
@@ -1134,7 +1204,7 @@ function buildSystemContent(instructions) {
   }
   return out;
 }
-function processInputItem(item, messages, dropImages) {
+function processInputItem(item, messages, options, context) {
   const itemType = String(item.type) || "message";
   const getLastAssistant = () => {
     const last = messages[messages.length - 1];
@@ -1196,7 +1266,7 @@ function processInputItem(item, messages, dropImages) {
             } else if (contentPart.type === "reasoning_text") {
               reasoningContent += String(contentPart.text ?? "");
             } else if (contentPart.type === "input_image" || contentPart.type === "image" || contentPart.type === "image_url") {
-              if (dropImages) {
+              if (options.dropImages) {
                 continue;
               }
               let url = "";
@@ -1271,15 +1341,15 @@ function processInputItem(item, messages, dropImages) {
     return;
   }
   if (itemType === "function_call" || itemType === "commandExecution" || itemType === "local_shell_call" || itemType === "fileChange" || itemType === "custom_tool_call" || itemType === "web_search_call") {
-    processToolCall(item, messages, getLastAssistant);
+    processToolCall(item, messages, getLastAssistant, options.fallbackThoughtSignature, context);
     return;
   }
   if (itemType === "function_call_output" || itemType === "commandExecutionOutput" || itemType === "fileChangeOutput" || itemType === "custom_tool_call_output") {
-    processToolOutput(item, messages);
+    processToolOutput(item, messages, context);
     return;
   }
 }
-function processToolCall(item, messages, getLastAssistant) {
+function processToolCall(item, messages, getLastAssistant, fallbackThoughtSignature, context) {
   const callId = String(item.call_id ?? "") || String(item.id ?? "") || makeId("call");
   let name = item.name === void 0 ? void 0 : String(item.name);
   const itemType = item.type === void 0 ? void 0 : String(item.type);
@@ -1321,25 +1391,35 @@ function processToolCall(item, messages, getLastAssistant) {
   if (!name) {
     return;
   }
+  if (context?.validFunctionNames?.size && !context.validFunctionNames.has(name)) {
+    context.ignoredToolCallIds.add(callId);
+    const amsg2 = getLastAssistant();
+    const note = `Unsupported tool call omitted: ${name}`;
+    amsg2.content = typeof amsg2.content === "string" && amsg2.content ? `${amsg2.content}
+${note}` : note;
+    return;
+  }
   const amsg = getLastAssistant();
   if (!amsg.tool_calls) {
     amsg.tool_calls = [];
   }
-  amsg.tool_calls.push({
+  const toolCall = {
     id: callId,
     type: "function",
     function: { name, arguments: argsStr }
-  });
-  const sig = item.thought_signature;
+  };
+  const sig = item.thought_signature ?? fallbackThoughtSignature;
   const thought = item.thought;
   if (typeof sig === "string" && sig) {
+    toolCall.extra_content = { google: { thought_signature: sig } };
     amsg.thought_signature = sig;
   }
+  amsg.tool_calls.push(toolCall);
   if (typeof thought === "string" && thought) {
     amsg.reasoning_content = (amsg.reasoning_content ?? "") + thought;
   }
 }
-function processToolOutput(item, messages) {
+function processToolOutput(item, messages, context) {
   const callId = item.call_id === void 0 ? void 0 : String(item.call_id);
   const outputRaw = item.output ?? item.content ?? item.stdout ?? "";
   let content = "";
@@ -1366,6 +1446,15 @@ function processToolOutput(item, messages) {
   if (!content && typeof item.stderr === "string" && item.stderr) {
     content = `Error: ${item.stderr}`;
   }
+  if (callId && context?.ignoredToolCallIds.has(callId)) {
+    if (content) {
+      messages.push({
+        role: "user",
+        content: `Output for omitted unsupported tool call: ${content}`
+      });
+    }
+    return;
+  }
   messages.push({
     role: "tool",
     tool_call_id: callId,
@@ -1385,7 +1474,8 @@ function mapTools2(tools) {
       if (!name) {
         continue;
       }
-      const params = fn?.parameters ?? tool.parameters ?? { type: "object" };
+      const rawParams = fn?.parameters ?? tool.parameters ?? { type: "object" };
+      const params = sanitizeJsonSchema(rawParams);
       out.push({
         type: "function",
         function: {
@@ -1538,6 +1628,10 @@ function mapToolCallToOutput(tc) {
     arguments: args,
     call_id: callId
   };
+  const thoughtSignature = getThoughtSignature(tc);
+  if (thoughtSignature) {
+    item.thought_signature = thoughtSignature;
+  }
   if (SHELL_TOOL_NAMES3.has(name)) {
     item.type = "local_shell_call";
     const parsed = safeJsonParse(args);
@@ -1545,6 +1639,10 @@ function mapToolCallToOutput(tc) {
   }
   return item;
 }
+function getThoughtSignature(tc) {
+  const sig = tc.extra_content?.google?.thought_signature ?? tc.thought_signature;
+  return typeof sig === "string" && sig ? sig : void 0;
+}
 
 // src/translate/openai/translateStream.ts
 var SHELL_TOOL_NAMES4 = /* @__PURE__ */ new Set(["shell", "container.exec", "shell_command"]);
@@ -1647,6 +1745,10 @@ var StreamTranslator2 = class {
           });
         }
         const fn = tc.function;
+        const thoughtSignature = getThoughtSignature2(tc);
+        if (thoughtSignature) {
+          state.item.thought_signature = thoughtSignature;
+        }
         if (fn?.name) {
           state.item.name = (state.item.name ?? "") + fn.name;
         }
@@ -1758,6 +1860,10 @@ var StreamTranslator2 = class {
     };
   }
 };
+function getThoughtSignature2(tc) {
+  const sig = tc.extra_content?.google?.thought_signature ?? tc.thought_signature;
+  return typeof sig === "string" && sig ? sig : void 0;
+}
 
 // src/fetch.ts
 function createResponsesFetch(options) {
@@ -1765,7 +1871,7 @@ function createResponsesFetch(options) {
     throw new Error("baseUrl is required");
   }
   const rawFormat = options.upstreamFormat;
-  const format = rawFormat ? normalizeFormat(rawFormat) : inferFormatFromUrl(options.baseUrl) ?? "openai-chat";
+  const format = rawFormat ? normalizeFormat(rawFormat) : inferFormatFromUrl(options.baseUrl) ?? inferFormatFromModel(options.model) ?? "openai-chat";
   if (!format) {
     throw new Error(
       `Unsupported upstream format: ${options.upstreamFormat}. Use 'anthropic' or 'openai-chat'`
@@ -1936,7 +2042,8 @@ async function handleResponses(request, format, options, baseFetch, incomingHead
     options.baseUrl,
     dropImages,
     options.reasoning_effort,
-    options.thinking
+    options.thinking,
+    options.fallbackThoughtSignature
   );
   const upstreamHeaders = buildUpstreamHeaders(format, options, incomingHeaders);
   const upstream = await baseFetch(resolvedUrl, {
@@ -1992,7 +2099,7 @@ function buildRequestMetadata(request, temperature, top_p) {
     metadata: request.metadata ?? {}
   };
 }
-function buildUpstreamBody(request, format, streaming, baseUrl, dropImages, reasoning_effort, thinking) {
+function buildUpstreamBody(request, format, streaming, baseUrl, dropImages, reasoning_effort, thinking, fallbackThoughtSignature) {
   if (format === "anthropic") {
     const { request: ar } = translateRequest(request);
     ar.stream = streaming;
@@ -2023,7 +2130,10 @@ function buildUpstreamBody(request, format, streaming, baseUrl, dropImages, reas
       requestMetadata: buildRequestMetadata(request, ar.temperature, ar.top_p)
     };
   }
-  const { request: cr } = translateRequest2(request, { dropImages });
+  const { request: cr } = translateRequest2(request, {
+    dropImages,
+    fallbackThoughtSignature
+  });
   cr.stream = streaming;
   if (streaming) {
     cr.stream_options = { include_usage: true };