@shareai-lab/kode 1.1.12 → 1.1.13

package/dist/utils/messages.js

@@ -0,0 +1,633 @@
+import { randomUUID } from "crypto";
+import { Box } from "ink";
+import { getCommand, hasCommand } from "../commands.js";
+import { MalformedCommandError } from "./errors.js";
+import { logError } from "./log.js";
+import { resolve } from "path";
+import { last, memoize } from "lodash-es";
+import { logEvent } from "../services/statsig.js";
+import { lastX } from "../utils/generators.js";
+import { NO_CONTENT_MESSAGE } from "../services/claude.js";
+import { setCwd } from "./state.js";
+import { getCwd } from "./state.js";
+import chalk from "chalk";
+import * as React from "react";
+import { UserBashInputMessage } from "../components/messages/UserBashInputMessage.js";
+import { Spinner } from "../components/Spinner.js";
+import { BashTool } from "../tools/BashTool/BashTool.js";
+const INTERRUPT_MESSAGE = "[Request interrupted by user]";
+const INTERRUPT_MESSAGE_FOR_TOOL_USE = "[Request interrupted by user for tool use]";
+const CANCEL_MESSAGE = "The user doesn't want to take this action right now. STOP what you are doing and wait for the user to tell you how to proceed.";
+const REJECT_MESSAGE = "The user doesn't want to proceed with this tool use. The tool use was rejected (eg. if it was a file edit, the new_string was NOT written to the file). STOP what you are doing and wait for the user to tell you how to proceed.";
+const NO_RESPONSE_REQUESTED = "No response requested.";
+const SYNTHETIC_ASSISTANT_MESSAGES = /* @__PURE__ */ new Set([
+  INTERRUPT_MESSAGE,
+  INTERRUPT_MESSAGE_FOR_TOOL_USE,
+  CANCEL_MESSAGE,
+  REJECT_MESSAGE,
+  NO_RESPONSE_REQUESTED
+]);
+function baseCreateAssistantMessage(content, extra) {
+  return {
+    type: "assistant",
+    costUSD: 0,
+    durationMs: 0,
+    uuid: randomUUID(),
+    message: {
+      id: randomUUID(),
+      model: "<synthetic>",
+      role: "assistant",
+      stop_reason: "stop_sequence",
+      stop_sequence: "",
+      type: "message",
+      usage: {
+        input_tokens: 0,
+        output_tokens: 0,
+        cache_creation_input_tokens: 0,
+        cache_read_input_tokens: 0
+      },
+      content
+    },
+    ...extra
+  };
+}
+function createAssistantMessage(content) {
+  return baseCreateAssistantMessage([
+    {
+      type: "text",
+      text: content === "" ? NO_CONTENT_MESSAGE : content,
+      citations: []
+    }
+  ]);
+}
+function createAssistantAPIErrorMessage(content) {
+  return baseCreateAssistantMessage(
+    [
+      {
+        type: "text",
+        text: content === "" ? NO_CONTENT_MESSAGE : content,
+        citations: []
+      }
+    ],
+    { isApiErrorMessage: true }
+  );
+}
+function createUserMessage(content, toolUseResult) {
+  const m = {
+    type: "user",
+    message: {
+      role: "user",
+      content
+    },
+    uuid: randomUUID(),
+    toolUseResult
+  };
+  return m;
+}
+function createProgressMessage(toolUseID, siblingToolUseIDs, content, normalizedMessages, tools) {
+  return {
+    type: "progress",
+    content,
+    normalizedMessages,
+    siblingToolUseIDs,
+    tools,
+    toolUseID,
+    uuid: randomUUID()
+  };
+}
+function createToolResultStopMessage(toolUseID) {
+  return {
+    type: "tool_result",
+    content: CANCEL_MESSAGE,
+    is_error: true,
+    tool_use_id: toolUseID
+  };
+}
+async function processUserInput(input, mode, setToolJSX, context, pastedImage) {
+  if (mode === "bash") {
+    logEvent("tengu_input_bash", {});
+    const userMessage2 = createUserMessage(`<bash-input>${input}</bash-input>`);
+    if (input.startsWith("cd ")) {
+      const oldCwd = getCwd();
+      const newCwd = resolve(oldCwd, input.slice(3));
+      try {
+        await setCwd(newCwd);
+        return [
+          userMessage2,
+          createAssistantMessage(
+            `<bash-stdout>Changed directory to ${chalk.bold(`${newCwd}/`)}</bash-stdout>`
+          )
+        ];
+      } catch (e) {
+        logError(e);
+        return [
+          userMessage2,
+          createAssistantMessage(
+            `<bash-stderr>cwd error: ${e instanceof Error ? e.message : String(e)}</bash-stderr>`
+          )
+        ];
+      }
+    }
+    setToolJSX({
+      jsx: /* @__PURE__ */ React.createElement(Box, { flexDirection: "column", marginTop: 1 }, /* @__PURE__ */ React.createElement(
+        UserBashInputMessage,
+        {
+          addMargin: false,
+          param: { text: `<bash-input>${input}</bash-input>`, type: "text" }
+        }
+      ), /* @__PURE__ */ React.createElement(Spinner, null)),
+      shouldHidePromptInput: false
+    });
+    try {
+      const validationResult = await BashTool.validateInput({
+        command: input
+      });
+      if (!validationResult.result) {
+        return [userMessage2, createAssistantMessage(validationResult.message)];
+      }
+      const { data } = await lastX(BashTool.call({ command: input }, context));
+      return [
+        userMessage2,
+        createAssistantMessage(
+          `<bash-stdout>${data.stdout}</bash-stdout><bash-stderr>${data.stderr}</bash-stderr>`
+        )
+      ];
+    } catch (e) {
+      return [
+        userMessage2,
+        createAssistantMessage(
+          `<bash-stderr>Command failed: ${e instanceof Error ? e.message : String(e)}</bash-stderr>`
+        )
+      ];
+    } finally {
+      setToolJSX(null);
+    }
+  } else if (mode === "koding") {
+    logEvent("tengu_input_koding", {});
+    const userMessage2 = createUserMessage(
+      `<koding-input>${input}</koding-input>`
+    );
+    userMessage2.options = {
+      ...userMessage2.options,
+      isKodingRequest: true
+    };
+    return [userMessage2];
+  }
+  if (input.startsWith("/")) {
+    const words = input.slice(1).split(" ");
+    let commandName = words[0];
+    if (words.length > 1 && words[1] === "(MCP)") {
+      commandName = commandName + " (MCP)";
+    }
+    if (!commandName) {
+      logEvent("tengu_input_slash_missing", { input });
+      return [
+        createAssistantMessage("Commands are in the form `/command [args]`")
+      ];
+    }
+    if (!hasCommand(commandName, context.options.commands)) {
+      logEvent("tengu_input_prompt", {});
+      return [createUserMessage(input)];
+    }
+    const args = input.slice(commandName.length + 2);
+    const newMessages = await getMessagesForSlashCommand(
+      commandName,
+      args,
+      setToolJSX,
+      context
+    );
+    if (newMessages.length === 0) {
+      logEvent("tengu_input_command", { input });
+      return [];
+    }
+    if (newMessages.length === 2 && newMessages[0].type === "user" && newMessages[1].type === "assistant" && typeof newMessages[1].message.content === "string" && newMessages[1].message.content.startsWith("Unknown command:")) {
+      logEvent("tengu_input_slash_invalid", { input });
+      return newMessages;
+    }
+    if (newMessages.length === 2) {
+      logEvent("tengu_input_command", { input });
+      return newMessages;
+    }
+    logEvent("tengu_input_command", { input });
+    return newMessages;
+  }
+  logEvent("tengu_input_prompt", {});
+  const isKodingRequest = context.options?.isKodingRequest === true;
+  const kodingContextInfo = context.options?.kodingContext;
+  let userMessage;
+  if (pastedImage) {
+    userMessage = createUserMessage([
+      {
+        type: "image",
+        source: {
+          type: "base64",
+          media_type: "image/png",
+          data: pastedImage
+        }
+      },
+      {
+        type: "text",
+        text: isKodingRequest && kodingContextInfo ? `${kodingContextInfo}
+
+${input}` : input
+      }
+    ]);
+  } else {
+    let processedInput = isKodingRequest && kodingContextInfo ? `${kodingContextInfo}
+
+${input}` : input;
+    if (input.includes("!`") || input.includes("@")) {
+      try {
+        const { executeBashCommands } = await import("../services/customCommands.js");
+        if (input.includes("!`")) {
+          processedInput = await executeBashCommands(processedInput);
+        }
+        if (input.includes("@")) {
+          const { processMentions } = await import("../services/mentionProcessor.js");
+          await processMentions(input);
+        }
+      } catch (error) {
+        console.warn("Dynamic content processing failed:", error);
+      }
+    }
+    userMessage = createUserMessage(processedInput);
+  }
+  if (isKodingRequest) {
+    userMessage.options = {
+      ...userMessage.options,
+      isKodingRequest: true
+    };
+  }
+  return [userMessage];
+}
+async function getMessagesForSlashCommand(commandName, args, setToolJSX, context) {
+  try {
+    const command = getCommand(commandName, context.options.commands);
+    switch (command.type) {
+      case "local-jsx": {
+        return new Promise((resolve2) => {
+          command.call((r) => {
+            setToolJSX(null);
+            resolve2([
+              createUserMessage(`<command-name>${command.userFacingName()}</command-name>
+          <command-message>${command.userFacingName()}</command-message>
+          <command-args>${args}</command-args>`),
+              r ? createAssistantMessage(r) : createAssistantMessage(NO_RESPONSE_REQUESTED)
+            ]);
+          }, context).then((jsx) => {
+            setToolJSX({
+              jsx,
+              shouldHidePromptInput: true
+            });
+          });
+        });
+      }
+      case "local": {
+        const userMessage = createUserMessage(`<command-name>${command.userFacingName()}</command-name>
+        <command-message>${command.userFacingName()}</command-message>
+        <command-args>${args}</command-args>`);
+        try {
+          const result = await command.call(args, {
+            ...context,
+            options: {
+              commands: context.options.commands || [],
+              tools: context.options.tools || [],
+              slowAndCapableModel: context.options.slowAndCapableModel || "main"
+            }
+          });
+          return [
+            userMessage,
+            createAssistantMessage(
+              `<local-command-stdout>${result}</local-command-stdout>`
+            )
+          ];
+        } catch (e) {
+          logError(e);
+          return [
+            userMessage,
+            createAssistantMessage(
+              `<local-command-stderr>${String(e)}</local-command-stderr>`
+            )
+          ];
+        }
+      }
+      case "prompt": {
+        const prompt = await command.getPromptForCommand(args);
+        return prompt.map((msg) => {
+          const userMessage = createUserMessage(
+            typeof msg.content === "string" ? msg.content : msg.content.map((block) => block.type === "text" ? block.text : "").join("\n")
+          );
+          userMessage.options = {
+            ...userMessage.options,
+            isCustomCommand: true,
+            commandName: command.userFacingName(),
+            commandArgs: args
+          };
+          return userMessage;
+        });
+      }
+    }
+  } catch (e) {
+    if (e instanceof MalformedCommandError) {
+      return [createAssistantMessage(e.message)];
+    }
+    throw e;
+  }
+}
+function extractTagFromMessage(message, tagName) {
+  if (message.type === "progress") {
+    return null;
+  }
+  if (typeof message.message.content !== "string") {
+    return null;
+  }
+  return extractTag(message.message.content, tagName);
+}
+function extractTag(html, tagName) {
+  if (!html.trim() || !tagName.trim()) {
+    return null;
+  }
+  const escapedTag = tagName.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+  const pattern = new RegExp(
+    `<${escapedTag}(?:\\s+[^>]*)?>([\\s\\S]*?)<\\/${escapedTag}>`,
+    // Closing tag
+    "gi"
+  );
+  let match;
+  let depth = 0;
+  let lastIndex = 0;
+  const openingTag = new RegExp(`<${escapedTag}(?:\\s+[^>]*?)?>`, "gi");
+  const closingTag = new RegExp(`<\\/${escapedTag}>`, "gi");
+  while ((match = pattern.exec(html)) !== null) {
+    const content = match[1];
+    const beforeMatch = html.slice(lastIndex, match.index);
+    depth = 0;
+    openingTag.lastIndex = 0;
+    while (openingTag.exec(beforeMatch) !== null) {
+      depth++;
+    }
+    closingTag.lastIndex = 0;
+    while (closingTag.exec(beforeMatch) !== null) {
+      depth--;
+    }
+    if (depth === 0 && content) {
+      return content;
+    }
+    lastIndex = match.index + match[0].length;
+  }
+  return null;
+}
+function isNotEmptyMessage(message) {
+  if (message.type === "progress") {
+    return true;
+  }
+  if (typeof message.message.content === "string") {
+    return message.message.content.trim().length > 0;
+  }
+  if (message.message.content.length === 0) {
+    return false;
+  }
+  if (message.message.content.length > 1) {
+    return true;
+  }
+  if (message.message.content[0].type !== "text") {
+    return true;
+  }
+  return message.message.content[0].text.trim().length > 0 && message.message.content[0].text !== NO_CONTENT_MESSAGE && message.message.content[0].text !== INTERRUPT_MESSAGE_FOR_TOOL_USE;
+}
+function normalizeMessages(messages) {
+  return messages.flatMap((message) => {
+    if (message.type === "progress") {
+      return [message];
+    }
+    if (typeof message.message.content === "string") {
+      return [message];
+    }
+    return message.message.content.map((_) => {
+      switch (message.type) {
+        case "assistant":
+          return {
+            type: "assistant",
+            uuid: randomUUID(),
+            message: {
+              ...message.message,
+              content: [_]
+            },
+            costUSD: message.costUSD / message.message.content.length,
+            durationMs: message.durationMs
+          };
+        case "user":
+          return message;
+      }
+    });
+  });
+}
+function isToolUseRequestMessage(message) {
+  return message.type === "assistant" && "costUSD" in message && // Note: stop_reason === 'tool_use' is unreliable -- it's not always set correctly
+  message.message.content.some((_) => _.type === "tool_use");
+}
+function reorderMessages(messages) {
+  const ms = [];
+  const toolUseMessages = [];
+  for (const message of messages) {
+    if (isToolUseRequestMessage(message)) {
+      toolUseMessages.push(message);
+    }
+    if (message.type === "progress") {
+      const existingProgressMessage = ms.find(
+        (_) => _.type === "progress" && _.toolUseID === message.toolUseID
+      );
+      if (existingProgressMessage) {
+        ms[ms.indexOf(existingProgressMessage)] = message;
+        continue;
+      }
+      const toolUseMessage = toolUseMessages.find(
+        (_) => _.message.content[0]?.id === message.toolUseID
+      );
+      if (toolUseMessage) {
+        ms.splice(ms.indexOf(toolUseMessage) + 1, 0, message);
+        continue;
+      }
+    }
+    if (message.type === "user" && Array.isArray(message.message.content) && message.message.content[0]?.type === "tool_result") {
+      const toolUseID = message.message.content[0]?.tool_use_id;
+      const lastProgressMessage = ms.find(
+        (_) => _.type === "progress" && _.toolUseID === toolUseID
+      );
+      if (lastProgressMessage) {
+        ms.splice(ms.indexOf(lastProgressMessage) + 1, 0, message);
+        continue;
+      }
+      const toolUseMessage = toolUseMessages.find(
+        (_) => _.message.content[0]?.id === toolUseID
+      );
+      if (toolUseMessage) {
+        ms.splice(ms.indexOf(toolUseMessage) + 1, 0, message);
+        continue;
+      }
+    } else {
+      ms.push(message);
+    }
+  }
+  return ms;
+}
+const getToolResultIDs = memoize(
+  (normalizedMessages) => Object.fromEntries(
+    normalizedMessages.flatMap(
+      (_) => _.type === "user" && _.message.content[0]?.type === "tool_result" ? [
+        [
+          _.message.content[0].tool_use_id,
+          _.message.content[0].is_error ?? false
+        ]
+      ] : []
+    )
+  )
+);
+function getUnresolvedToolUseIDs(normalizedMessages) {
+  const toolResults = getToolResultIDs(normalizedMessages);
+  return new Set(
+    normalizedMessages.filter(
+      (_) => _.type === "assistant" && Array.isArray(_.message.content) && _.message.content[0]?.type === "tool_use" && !(_.message.content[0]?.id in toolResults)
+    ).map((_) => _.message.content[0].id)
+  );
+}
+function getInProgressToolUseIDs(normalizedMessages) {
+  const unresolvedToolUseIDs = getUnresolvedToolUseIDs(normalizedMessages);
+  const toolUseIDsThatHaveProgressMessages = new Set(
+    normalizedMessages.filter((_) => _.type === "progress").map((_) => _.toolUseID)
+  );
+  return new Set(
+    normalizedMessages.filter((_) => {
+      if (_.type !== "assistant") {
+        return false;
+      }
+      if (_.message.content[0]?.type !== "tool_use") {
+        return false;
+      }
+      const toolUseID = _.message.content[0].id;
+      if (toolUseID === unresolvedToolUseIDs.values().next().value) {
+        return true;
+      }
+      if (toolUseIDsThatHaveProgressMessages.has(toolUseID) && unresolvedToolUseIDs.has(toolUseID)) {
+        return true;
+      }
+      return false;
+    }).map((_) => _.message.content[0].id)
+  );
+}
+function getErroredToolUseMessages(normalizedMessages) {
+  const toolResults = getToolResultIDs(normalizedMessages);
+  return normalizedMessages.filter(
+    (_) => _.type === "assistant" && Array.isArray(_.message.content) && _.message.content[0]?.type === "tool_use" && _.message.content[0]?.id in toolResults && toolResults[_.message.content[0]?.id]
+  );
+}
+function normalizeMessagesForAPI(messages) {
+  const result = [];
+  messages.filter((_) => _.type !== "progress").forEach((message) => {
+    switch (message.type) {
+      case "user": {
+        if (!Array.isArray(message.message.content) || message.message.content[0]?.type !== "tool_result") {
+          result.push(message);
+          return;
+        }
+        const lastMessage = last(result);
+        if (!lastMessage || lastMessage?.type === "assistant" || !Array.isArray(lastMessage.message.content) || lastMessage.message.content[0]?.type !== "tool_result") {
+          result.push(message);
+          return;
+        }
+        result[result.indexOf(lastMessage)] = {
+          ...lastMessage,
+          message: {
+            ...lastMessage.message,
+            content: [
+              ...lastMessage.message.content,
+              ...message.message.content
+            ]
+          }
+        };
+        return;
+      }
+      case "assistant":
+        result.push(message);
+        return;
+    }
+  });
+  return result;
+}
+function normalizeContentFromAPI(content) {
+  const filteredContent = content.filter(
+    (_) => _.type !== "text" || _.text.trim().length > 0
+  );
+  if (filteredContent.length === 0) {
+    return [{ type: "text", text: NO_CONTENT_MESSAGE, citations: [] }];
+  }
+  return filteredContent;
+}
+function isEmptyMessageText(text) {
+  return stripSystemMessages(text).trim() === "" || text.trim() === NO_CONTENT_MESSAGE;
+}
+const STRIPPED_TAGS = [
+  "commit_analysis",
+  "context",
+  "function_analysis",
+  "pr_analysis"
+];
+function stripSystemMessages(content) {
+  const regex = new RegExp(`<(${STRIPPED_TAGS.join("|")})>.*?</\\1>
+?`, "gs");
+  return content.replace(regex, "").trim();
+}
+function getToolUseID(message) {
+  switch (message.type) {
+    case "assistant":
+      if (message.message.content[0]?.type !== "tool_use") {
+        return null;
+      }
+      return message.message.content[0].id;
+    case "user":
+      if (message.message.content[0]?.type !== "tool_result") {
+        return null;
+      }
+      return message.message.content[0].tool_use_id;
+    case "progress":
+      return message.toolUseID;
+  }
+}
+function getLastAssistantMessageId(messages) {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const message = messages[i];
+    if (message && message.type === "assistant") {
+      return message.message.id;
+    }
+  }
+  return void 0;
+}
+export {
+  CANCEL_MESSAGE,
+  INTERRUPT_MESSAGE,
+  INTERRUPT_MESSAGE_FOR_TOOL_USE,
+  NO_RESPONSE_REQUESTED,
+  REJECT_MESSAGE,
+  SYNTHETIC_ASSISTANT_MESSAGES,
+  createAssistantAPIErrorMessage,
+  createAssistantMessage,
+  createProgressMessage,
+  createToolResultStopMessage,
+  createUserMessage,
+  extractTag,
+  extractTagFromMessage,
+  getErroredToolUseMessages,
+  getInProgressToolUseIDs,
+  getLastAssistantMessageId,
+  getToolUseID,
+  getUnresolvedToolUseIDs,
+  isEmptyMessageText,
+  isNotEmptyMessage,
+  normalizeContentFromAPI,
+  normalizeMessages,
+  normalizeMessagesForAPI,
+  processUserInput,
+  reorderMessages,
+  stripSystemMessages
+};
+//# sourceMappingURL=messages.js.map