@fudrouter/fsrouter 0.6.59 → 0.6.60

package/open-sse/translator/request/claude-to-kiro.js

@@ -1,478 +0,0 @@
-/**
- * Claude → Kiro Request Translator (DIRECT route, no OpenAI pivot)
- *
- * Converts Anthropic Messages API requests straight to Kiro / AWS
- * CodeWhisperer `GenerateAssistantResponse` payloads. This is the function the
- * direct `claude:kiro` route in ../index.js uses; it is NOT reached through the
- * claude→openai→kiro pivot.
- *
- * It reproduces the two 400-guards that live in openai-to-kiro.js so that a
- * Claude client which omits the `tools` array on a follow-up turn (typical
- * after client-side compaction) does not trip Kiro's schema validator and get
- * "Improperly formed request" (HTTP 400):
- *
- *   1. flattenClaudeToolInteractions — when the client sent NO tools, collapse
- *      every tool_use / tool_result block to plain text so no structured tool
- *      reference survives to trigger the "tools required" rule.
- *   2. reconcileOrphanedToolResults — when tools ARE present, fold any
- *      tool_result whose tool_use_id has no matching tool_use back into the
- *      user text instead of leaving a dangling structured reference.
- *
- * It also handles the 9router-synthetic `-agentic` / `-thinking` suffixes and
- * the `<thinking_mode>enabled</thinking_mode>` reasoning trigger, matching
- * buildKiroPayload.
- */
-import { register } from "../index.js";
-import { FORMATS } from "../formats.js";
-import { v4 as uuidv4 } from "uuid";
-import {
-  resolveKiroModel,
-  buildThinkingSystemPrefix,
-  KIRO_AGENTIC_SYSTEM_PROMPT,
-  resolveDefaultProfileArn,
-} from "../../config/kiroConstants.js";
-import { resolveKiroThinkingBudget } from "../../services/thinkingBudget.js";
-import { DEFAULT_IMAGE_MIME } from "../schema/index.js";
-import { ROLE, CLAUDE_BLOCK } from "../schema/index.js";
-
-/** Stringify a tool_use input as a readable line. */
-function toolUseToText(name, input) {
-  let argStr;
-  try {
-    argStr = typeof input === "string" ? input : JSON.stringify(input ?? {});
-  } catch {
-    argStr = "{}";
-  }
-  return `[Tool call: ${name || "unknown"}(${argStr})]`;
-}
-
-/** Render a Claude tool_result block's content as a readable line. */
-function toolResultBlockToText(content) {
-  let text = "";
-  if (typeof content === "string") {
-    text = content;
-  } else if (Array.isArray(content)) {
-    text = content
-      .map((c) => (typeof c === "string" ? c : c?.text || ""))
-      .filter(Boolean)
-      .join("\n");
-  } else if (content) {
-    try {
-      text = JSON.stringify(content);
-    } catch {
-      text = "";
-    }
-  }
-  return `[Tool result: ${text}]`;
-}
-
-/**
- * When the client sent no tools, rewrite every tool_use (assistant) and
- * tool_result (user) content block into plain text. Keeps text + images.
- * Returns a new messages array; never mutates the input.
- */
-function flattenClaudeToolInteractions(messages) {
-  const out = [];
-  for (const msg of messages) {
-    if (!msg) continue;
-
-    if (msg.role === ROLE.ASSISTANT && Array.isArray(msg.content)) {
-      const parts = [];
-      for (const block of msg.content) {
-        if (block.type === CLAUDE_BLOCK.TEXT && block.text) {
-          parts.push(block.text);
-        } else if (block.type === CLAUDE_BLOCK.TOOL_USE) {
-          parts.push(toolUseToText(block.name, block.input));
-        }
-      }
-      out.push({ ...msg, content: parts.join("\n") });
-      continue;
-    }
-
-    if (msg.role === ROLE.USER && Array.isArray(msg.content)) {
-      const newContent = msg.content.map((block) =>
-        block.type === CLAUDE_BLOCK.TOOL_RESULT
-          ? { type: CLAUDE_BLOCK.TEXT, text: toolResultBlockToText(block.content) }
-          : block
-      );
-      out.push({ ...msg, content: newContent });
-      continue;
-    }
-
-    out.push(msg);
-  }
-  return out;
-}
-
-/**
- * Convert Claude messages to Kiro history + currentMessage.
- * Kiro requires alternating user/assistant turns; consecutive same-role
- * messages are merged.
- */
-function convertClaudeMessagesToKiro(messages, tools, model) {
-  const history = [];
-  let currentMessage = null;
-
-  let pendingUserContent = [];
-  let pendingAssistantContent = [];
-  let pendingToolResults = [];
-  let pendingImages = [];
-  let currentRole = null;
-  let toolsInjected = false;
-
-  const clientProvidedTools = Array.isArray(tools) && tools.length > 0;
-
-  const buildToolSpecs = () =>
-    tools.map((t) => {
-      const name = t.name;
-      const description = t.description || `Tool: ${name}`;
-      const schema = t.input_schema || {};
-      const normalizedSchema =
-        Object.keys(schema).length === 0
-          ? { type: "object", properties: {}, required: [] }
-          : { ...schema, required: schema.required ?? [] };
-      return {
-        toolSpecification: {
-          name,
-          description,
-          inputSchema: { json: normalizedSchema },
-        },
-      };
-    });
-
-  const flushPending = () => {
-    if (currentRole === ROLE.USER) {
-      const content = pendingUserContent.join("\n\n").trim() || "continue";
-      const userMsg = { userInputMessage: { content, modelId: model } };
-
-      if (pendingImages.length > 0) {
-        userMsg.userInputMessage.images = pendingImages;
-      }
-      if (pendingToolResults.length > 0) {
-        userMsg.userInputMessage.userInputMessageContext = {
-          toolResults: pendingToolResults,
-        };
-      }
-      // Attach tools to the first user turn only.
-      if (clientProvidedTools && !toolsInjected) {
-        if (!userMsg.userInputMessage.userInputMessageContext) {
-          userMsg.userInputMessage.userInputMessageContext = {};
-        }
-        userMsg.userInputMessage.userInputMessageContext.tools = buildToolSpecs();
-        toolsInjected = true;
-      }
-
-      history.push(userMsg);
-      currentMessage = userMsg;
-      pendingUserContent = [];
-      pendingToolResults = [];
-      pendingImages = [];
-    } else if (currentRole === ROLE.ASSISTANT) {
-      const content = pendingAssistantContent.join("\n\n").trim() || "...";
-      history.push({ assistantResponseMessage: { content } });
-      pendingAssistantContent = [];
-    }
-  };
-
-  for (const msg of messages) {
-    const role = msg.role;
-    if (role !== currentRole && currentRole !== null) flushPending();
-    currentRole = role;
-
-    if (role === ROLE.USER) {
-      if (typeof msg.content === "string") {
-        pendingUserContent.push(msg.content);
-      } else if (Array.isArray(msg.content)) {
-        for (const block of msg.content) {
-          if (block.type === CLAUDE_BLOCK.TEXT) {
-            pendingUserContent.push(block.text);
-          } else if (block.type === CLAUDE_BLOCK.IMAGE && block.source?.type === "base64") {
-            const mediaType = block.source.media_type || DEFAULT_IMAGE_MIME;
-            const format = mediaType.split("/")[1] || mediaType;
-            pendingImages.push({ format, source: { bytes: block.source.data } });
-          } else if (block.type === CLAUDE_BLOCK.TOOL_RESULT) {
-            let resultContent = "";
-            if (typeof block.content === "string") {
-              resultContent = block.content;
-            } else if (Array.isArray(block.content)) {
-              resultContent =
-                block.content
-                  .filter((c) => c.type === CLAUDE_BLOCK.TEXT)
-                  .map((c) => c.text)
-                  .join("\n") || JSON.stringify(block.content);
-            } else if (block.content) {
-              resultContent = JSON.stringify(block.content);
-            }
-            pendingToolResults.push({
-              toolUseId: block.tool_use_id,
-              status: "success",
-              content: [{ text: resultContent }],
-            });
-          }
-        }
-      }
-    } else if (role === ROLE.ASSISTANT) {
-      let textContent = "";
-      const toolUses = [];
-      if (typeof msg.content === "string") {
-        textContent = msg.content;
-      } else if (Array.isArray(msg.content)) {
-        for (const block of msg.content) {
-          if (block.type === CLAUDE_BLOCK.TEXT) {
-            textContent += block.text;
-          } else if (block.type === CLAUDE_BLOCK.TOOL_USE) {
-            toolUses.push({
-              toolUseId: block.id,
-              name: block.name,
-              input: block.input || {},
-            });
-          }
-        }
-      }
-      if (textContent) pendingAssistantContent.push(textContent);
-
-      if (toolUses.length > 0) {
-        flushPending();
-        const lastMsg = history[history.length - 1];
-        if (lastMsg?.assistantResponseMessage) {
-          lastMsg.assistantResponseMessage.toolUses = toolUses;
-        }
-        currentRole = null;
-      }
-    }
-  }
-
-  if (currentRole !== null) flushPending();
-
-  // Pop the last user turn as currentMessage (skip trailing assistant turns).
-  for (let i = history.length - 1; i >= 0; i--) {
-    if (history[i].userInputMessage) {
-      currentMessage = history.splice(i, 1)[0];
-      break;
-    }
-  }
-
-  // Grab tools from the first history user turn before cleanup strips them.
-  const firstHistoryTools =
-    history[0]?.userInputMessage?.userInputMessageContext?.tools;
-
-  history.forEach((item) => {
-    if (item.userInputMessage?.userInputMessageContext?.tools) {
-      delete item.userInputMessage.userInputMessageContext.tools;
-    }
-    if (
-      item.userInputMessage?.userInputMessageContext &&
-      Object.keys(item.userInputMessage.userInputMessageContext).length === 0
-    ) {
-      delete item.userInputMessage.userInputMessageContext;
-    }
-    if (item.userInputMessage && !item.userInputMessage.modelId) {
-      item.userInputMessage.modelId = model;
-    }
-  });
-
-  // Merge consecutive user turns (Kiro requires alternating roles).
-  const mergedHistory = [];
-  for (const current of history) {
-    const prev = mergedHistory[mergedHistory.length - 1];
-    if (current.userInputMessage && prev?.userInputMessage) {
-      prev.userInputMessage.content += "\n\n" + current.userInputMessage.content;
-      const prevCtx = prev.userInputMessage.userInputMessageContext;
-      const curCtx = current.userInputMessage.userInputMessageContext;
-      if (curCtx) {
-        if (!prevCtx) {
-          prev.userInputMessage.userInputMessageContext = curCtx;
-        } else {
-          if (curCtx.toolResults?.length > 0) {
-            prevCtx.toolResults = [
-              ...(prevCtx.toolResults || []),
-              ...curCtx.toolResults,
-            ];
-          }
-          if (curCtx.tools?.length > 0) {
-            prevCtx.tools = [...(prevCtx.tools || []), ...curCtx.tools];
-          }
-        }
-      }
-    } else {
-      mergedHistory.push(current);
-    }
-  }
-
-  if (!currentMessage) {
-    currentMessage = { userInputMessage: { content: "", modelId: model } };
-  }
-
-  // Inject tools into currentMessage after cleanup if not already present.
-  if (
-    firstHistoryTools?.length > 0 &&
-    !currentMessage.userInputMessage.userInputMessageContext?.tools
-  ) {
-    if (!currentMessage.userInputMessage.userInputMessageContext) {
-      currentMessage.userInputMessage.userInputMessageContext = {};
-    }
-    currentMessage.userInputMessage.userInputMessageContext.tools =
-      firstHistoryTools;
-  }
-
-  return { history: mergedHistory, currentMessage };
-}
-
-/**
- * Fold orphaned toolResults (those whose toolUseId has no matching toolUse in
- * any assistant turn) back into the user text, removing the dangling
- * structured reference that makes Kiro 400.
- */
-function reconcileOrphanedToolResults(history, currentMessage) {
-  const validIds = new Set();
-  for (const h of history) {
-    const arm = h.assistantResponseMessage;
-    if (!arm) continue;
-    for (const tu of arm.toolUses || []) {
-      if (tu.toolUseId) validIds.add(tu.toolUseId);
-    }
-  }
-
-  const carriers = currentMessage ? [...history, currentMessage] : history;
-  for (const item of carriers) {
-    const uim = item.userInputMessage;
-    const ctx = uim?.userInputMessageContext;
-    if (!ctx?.toolResults?.length) continue;
-
-    const kept = [];
-    const salvaged = [];
-    for (const tr of ctx.toolResults) {
-      if (validIds.has(tr.toolUseId)) {
-        kept.push(tr);
-      } else {
-        const text = Array.isArray(tr.content)
-          ? tr.content.map((c) => c?.text || "").join("\n")
-          : "";
-        salvaged.push(`[Tool result: ${text}]`);
-      }
-    }
-
-    if (salvaged.length === 0) continue;
-
-    const extra = salvaged.join("\n");
-    uim.content = uim.content ? `${uim.content}\n\n${extra}` : extra;
-    ctx.toolResults = kept;
-    if (kept.length === 0 && !ctx.tools?.length) {
-      delete uim.userInputMessageContext;
-    }
-  }
-}
-
-/**
- * Build a Kiro payload directly from a Claude Messages API request body.
- */
-export function claudeToKiroRequest(model, body, stream, credentials) {
-  let messages = Array.isArray(body.messages) ? body.messages : [];
-  const tools = Array.isArray(body.tools) ? body.tools : [];
-  const clientProvidedTools = tools.length > 0;
-  const maxTokens = body.max_tokens || 32000;
-  const temperature = body.temperature;
-  const topP = body.top_p;
-
-  const { upstream: upstreamModel, agentic } = resolveKiroModel(model);
-  const thinkingBudget = resolveKiroThinkingBudget(body, credentials?.rawHeaders, model);
-
-  // Guard 1: no client tools → flatten all tool interactions to text.
-  if (!clientProvidedTools) {
-    messages = flattenClaudeToolInteractions(messages);
-  }
-
-  const { history, currentMessage } = convertClaudeMessagesToKiro(
-    messages,
-    tools,
-    upstreamModel
-  );
-
-  // Guard 2: tools present → reconcile dangling tool_results.
-  if (clientProvidedTools) {
-    reconcileOrphanedToolResults(history, currentMessage);
-  }
-
-  // api_key / idc / external_idp must never use the shared default ARN (belongs
-  // to another account → 403 "bearer token invalid"); OAuth/social fall back to it.
-  const authMethod = credentials?.providerSpecificData?.authMethod;
-  const accountBoundAuth =
-    authMethod === "api_key" || authMethod === "idc" || authMethod === "external_idp";
-  const profileArn = accountBoundAuth
-    ? (credentials?.providerSpecificData?.profileArn || "")
-    : (credentials?.providerSpecificData?.profileArn || resolveDefaultProfileArn(authMethod));
-
-  let finalContent = currentMessage?.userInputMessage?.content || "";
-
-  // System prompt: pass via native systemInstruction field (Kiro/Q API supports it)
-  // and also prepend as <instructions> in user content as fallback for upstreams
-  // that don't support the native field.
-  let systemInstruction = undefined;
-  if (body.system) {
-    let systemText = "";
-    if (typeof body.system === "string") {
-      systemText = body.system;
-    } else if (Array.isArray(body.system)) {
-      systemText = body.system.map((s) => s.text || "").join("\n");
-    }
-    if (systemText) {
-      systemInstruction = systemText;
-      finalContent = `<instructions>\n${systemText}\n</instructions>\n\n${finalContent}`;
-    }
-  }
-
-  // Prefix order: thinking_mode tag, timestamp marker, then agentic prompt.
-  const timestamp = new Date().toISOString();
-  const prefixParts = [];
-  if (thinkingBudget !== null) prefixParts.push(buildThinkingSystemPrefix(thinkingBudget));
-  prefixParts.push(`[Context: Current time is ${timestamp}]`);
-  if (agentic) prefixParts.push(KIRO_AGENTIC_SYSTEM_PROMPT);
-  finalContent = `${prefixParts.join("\n\n")}\n\n${finalContent}`;
-
-  const userInputMessage = {
-    content: finalContent,
-    modelId: upstreamModel,
-    origin: "AI_EDITOR",
-    ...(currentMessage?.userInputMessage?.userInputMessageContext && {
-      userInputMessageContext:
-        currentMessage.userInputMessage.userInputMessageContext,
-    }),
-    ...(currentMessage?.userInputMessage?.images && {
-      images: currentMessage.userInputMessage.images,
-    }),
-  };
-
-  if (systemInstruction) {
-    userInputMessage.systemInstruction = systemInstruction;
-  }
-
-  const payload = {
-    conversationState: {
-      chatTriggerType: "MANUAL",
-      conversationId: uuidv4(),
-      currentMessage: {
-        userInputMessage,
-      },
-      history,
-    },
-  };
-
-  if (profileArn) payload.profileArn = profileArn;
-
-  if (maxTokens || temperature !== undefined || topP !== undefined) {
-    payload.inferenceConfig = {};
-    if (maxTokens) payload.inferenceConfig.maxTokens = maxTokens;
-    if (temperature !== undefined) payload.inferenceConfig.temperature = temperature;
-    if (topP !== undefined) payload.inferenceConfig.topP = topP;
-  }
-
-  // Non-enumerable hint so the executor can route the upstream model id.
-  Object.defineProperty(payload, "_kiroUpstreamModel", {
-    value: upstreamModel,
-    enumerable: false,
-  });
-
-  return payload;
-}
-
-register(FORMATS.CLAUDE, FORMATS.KIRO, claudeToKiroRequest, null);