@dbx-tools/shared-model 0.3.26 → 0.3.28

package/package.json

@@ -12,14 +12,14 @@
   },
   "dependencies": {
     "zod": "^4.3.6",
-    "@dbx-tools/shared-core": "0.3.26"
+    "@dbx-tools/shared-core": "0.3.28"
   },
   "main": "index.ts",
   "license": "UNLICENSED",
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.3.26",
+  "version": "0.3.28",
   "types": "index.ts",
   "type": "module",
   "exports": {

package/src/openai-responses.ts

@@ -29,6 +29,9 @@
  *   - {@link sanitizeResponsesTools} keeps only `function` tools on a Responses
  *     request body (for `/open-responses` / Anthropic, which reject Codex
  *     built-ins like `web_search`).
+ *   - {@link sanitizeOpenResponsesInput} rewrites `output_*` content part types
+ *     in `input` to `input_*`, and drops Claude thinking / reasoning blocks
+ *     that Open Responses rejects on replay (`redacted_thinking`, …).
  *
  * Only the surface real clients exercise is translated; unknown fields are
  * ignored rather than rejected, so a newer client degrades instead of breaking.
@@ -643,3 +646,95 @@ export function sanitizeResponsesTools(
   }
   return next;
 }
+
+/** Content part / input item types Claude extended-thinking uses on the wire. */
+const REASONING_TYPES = new Set(["reasoning", "thinking", "redacted_thinking"]);
+
+/**
+ * Rewrite Responses `input` so Open Responses will accept it.
+ *
+ * Two OpenAI / Codex shapes that Databricks Open Responses (Claude, Gemini, …)
+ * reject:
+ *
+ * 1. Prior assistant turns with `output_text` / `output_*` content parts —
+ *    Open Responses only allows `input_text`, `input_image`, `input_file`,
+ *    `input_audio`. Map every `output_<kind>` part to `input_<kind>`.
+ * 2. Replayed extended-thinking blocks (`thinking`, `redacted_thinking`,
+ *    `reasoning` content parts, and top-level `reasoning` input items). Claude
+ *    signs those; any client round-trip that mutates them (or a cross-provider
+ *    replay) fails with e.g.
+ *      "messages.N.content.M: Invalid `data` in `redacted_thinking` block"
+ *    Strip them — the UI/client already showed the thinking; replay does not
+ *    need the signed blob. Same policy as appkit-mastra's serving sanitize.
+ *
+ * Returns a shallow copy when anything changes; otherwise the input body.
+ */
+export function sanitizeOpenResponsesInput(
+  body: Record<string, unknown>,
+): Record<string, unknown> {
+  if (!Array.isArray(body.input) || body.input.length === 0) return body;
+
+  let changed = false;
+  const input: unknown[] = [];
+
+  for (const raw of body.input) {
+    if (!raw || typeof raw !== "object") {
+      input.push(raw);
+      continue;
+    }
+    const item = raw as Record<string, unknown>;
+
+    // Top-level Responses reasoning items (not message content parts).
+    if (typeof item.type === "string" && REASONING_TYPES.has(item.type)) {
+      changed = true;
+      continue;
+    }
+
+    if (!Array.isArray(item.content)) {
+      input.push(raw);
+      continue;
+    }
+
+    let partChanged = false;
+    const content: unknown[] = [];
+    for (const part of item.content) {
+      if (!part || typeof part !== "object") {
+        content.push(part);
+        continue;
+      }
+      const p = part as Record<string, unknown>;
+      if (typeof p.type === "string" && REASONING_TYPES.has(p.type)) {
+        partChanged = true;
+        continue;
+      }
+      if (typeof p.type === "string" && p.type.startsWith("output_")) {
+        partChanged = true;
+        content.push({ ...p, type: `input_${p.type.slice("output_".length)}` });
+        continue;
+      }
+      content.push(part);
+    }
+
+    if (!partChanged) {
+      input.push(raw);
+      continue;
+    }
+    changed = true;
+    // Drop assistant turns that only carried thinking (nothing left to replay).
+    if (content.length === 0 && item.role === "assistant") continue;
+    input.push({ ...item, content });
+  }
+
+  return changed ? { ...body, input } : body;
+}
+
+/**
+ * Full Open Responses request sanitizer: strip non-`function` tools, rewrite
+ * `output_*` content parts, and drop thinking / reasoning blocks. Safe no-op
+ * when nothing needs changing.
+ */
+export function sanitizeOpenResponsesRequest(
+  body: Record<string, unknown>,
+): Record<string, unknown> {
+  return sanitizeOpenResponsesInput(sanitizeResponsesTools(body));
+}

package/test/openai-responses.test.ts

@@ -6,6 +6,7 @@ import {
   createResponsesStreamTranslator,
   readResponsesOutput,
   responsesToChat,
+  sanitizeOpenResponsesRequest,
 } from "../src/openai-responses";
 
 /** The `messages` a lowered Responses body produced, typed for assertions. */
@@ -240,3 +241,72 @@ describe("readResponsesOutput", () => {
     assert.deepEqual(readResponsesOutput({}), { text: "", citations: [] });
   });
 });
+
+describe("sanitizeOpenResponsesRequest", () => {
+  it("rewrites output_text content parts to input_text", () => {
+    const out = sanitizeOpenResponsesRequest({
+      input: [
+        {
+          type: "message",
+          role: "assistant",
+          content: [{ type: "output_text", text: "hi" }],
+        },
+      ],
+    });
+    assert.deepEqual(out.input, [
+      {
+        type: "message",
+        role: "assistant",
+        content: [{ type: "input_text", text: "hi" }],
+      },
+    ]);
+  });
+
+  it("strips redacted_thinking / thinking content parts and reasoning items", () => {
+    const out = sanitizeOpenResponsesRequest({
+      input: [
+        { type: "reasoning", summary: [{ type: "summary_text", text: "plan" }] },
+        {
+          type: "message",
+          role: "assistant",
+          content: [
+            { type: "redacted_thinking", data: "not-a-valid-blob" },
+            { type: "thinking", thinking: "secret" },
+            { type: "output_text", text: "visible" },
+          ],
+        },
+      ],
+    });
+    assert.deepEqual(out.input, [
+      {
+        type: "message",
+        role: "assistant",
+        content: [{ type: "input_text", text: "visible" }],
+      },
+    ]);
+  });
+
+  it("drops assistant turns that only carried thinking", () => {
+    const out = sanitizeOpenResponsesRequest({
+      input: [
+        {
+          type: "message",
+          role: "assistant",
+          content: [{ type: "redacted_thinking", data: "x" }],
+        },
+        { type: "message", role: "user", content: [{ type: "input_text", text: "next" }] },
+      ],
+    });
+    assert.deepEqual(out.input, [
+      { type: "message", role: "user", content: [{ type: "input_text", text: "next" }] },
+    ]);
+  });
+
+  it("strips non-function tools", () => {
+    const out = sanitizeOpenResponsesRequest({
+      input: [],
+      tools: [{ type: "web_search" }, { type: "function", name: "f", parameters: {} }],
+    });
+    assert.deepEqual(out.tools, [{ type: "function", name: "f", parameters: {} }]);
+  });
+});