@shareai-lab/kode 2.1.0 → 2.1.2

package/dist/sdk/core.js

@@ -1963,18 +1963,18 @@ var init_package = __esm({
         "zod-to-json-schema": "^3.24.6"
       },
       optionalDependencies: {
-        "@shareai-lab/kode-bin-darwin-arm64": "2.1.1",
-        "@shareai-lab/kode-bin-darwin-x64": "2.1.1",
-        "@shareai-lab/kode-bin-linux-arm64": "2.1.1",
-        "@shareai-lab/kode-bin-linux-x64": "2.1.1",
-        "@shareai-lab/kode-bin-win32-arm64": "2.1.1",
-        "@shareai-lab/kode-bin-win32-x64": "2.1.1",
-        "@shareai-lab/kode-ripgrep-darwin-arm64": "2.1.1",
-        "@shareai-lab/kode-ripgrep-darwin-x64": "2.1.1",
-        "@shareai-lab/kode-ripgrep-linux-arm64": "2.1.1",
-        "@shareai-lab/kode-ripgrep-linux-x64": "2.1.1",
-        "@shareai-lab/kode-ripgrep-win32-arm64": "2.1.1",
-        "@shareai-lab/kode-ripgrep-win32-x64": "2.1.1"
+        "@shareai-lab/kode-bin-darwin-arm64": "2.1.0",
+        "@shareai-lab/kode-bin-darwin-x64": "2.1.0",
+        "@shareai-lab/kode-bin-linux-arm64": "2.1.0",
+        "@shareai-lab/kode-bin-linux-x64": "2.1.0",
+        "@shareai-lab/kode-bin-win32-arm64": "2.1.0",
+        "@shareai-lab/kode-bin-win32-x64": "2.1.0",
+        "@shareai-lab/kode-ripgrep-darwin-arm64": "2.1.0",
+        "@shareai-lab/kode-ripgrep-darwin-x64": "2.1.0",
+        "@shareai-lab/kode-ripgrep-linux-arm64": "2.1.0",
+        "@shareai-lab/kode-ripgrep-linux-x64": "2.1.0",
+        "@shareai-lab/kode-ripgrep-win32-arm64": "2.1.0",
+        "@shareai-lab/kode-ripgrep-win32-x64": "2.1.0"
       },
       devDependencies: {
         "@codemirror/lang-css": "^6.3.1",
@@ -2946,7 +2946,11 @@ var init_constants3 = __esm({
 });
 
 // packages/core/src/engine/messages/create.ts
-import { randomUUID as randomUUID3 } from "crypto";
+import { createHash, randomUUID as randomUUID3 } from "crypto";
+function stableUuidFromSeed(seed) {
+  const hex = createHash("sha256").update(seed).digest("hex").slice(0, 32);
+  return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20, 32)}`;
+}
 function baseCreateAssistantMessage(content, extra) {
   return {
     type: "assistant",
@@ -3012,7 +3016,7 @@ function createProgressMessage(toolUseID, siblingToolUseIDs, content, normalized
     siblingToolUseIDs,
     tools,
     toolUseID,
-    uuid: randomUUID3()
+    uuid: stableUuidFromSeed(`progress:${toolUseID}`)
   };
 }
 var init_create = __esm({
@@ -6774,9 +6778,21 @@ var init_openaiAdapter = __esm({
 
 // packages/core/src/utils/requestStatus.ts
 function setRequestStatus(status) {
-  current = { ...status, updatedAt: Date.now() };
+  current = { ...current, ...status, updatedAt: Date.now() };
   for (const listener of listeners) listener(current);
 }
+function setRequestInputTokens(inputTokens) {
+  if (current.kind !== "idle") {
+    current = { ...current, inputTokens, outputTokens: void 0, updatedAt: Date.now() };
+    for (const listener of listeners) listener(current);
+  }
+}
+function updateRequestTokens(outputTokens) {
+  if (current.kind !== "idle") {
+    current = { ...current, outputTokens, updatedAt: Date.now() };
+    for (const listener of listeners) listener(current);
+  }
+}
 var current, listeners;
 var init_requestStatus = __esm({
   "packages/core/src/utils/requestStatus.ts"() {
@@ -9494,6 +9510,8 @@ async function handleMessageStream(stream, signal) {
   let ttftMs;
   let chunkCount = 0;
   let errorCount = 0;
+  let hasMarkedStreaming = false;
+  let outputTokenCount = 0;
   debug2.api("OPENAI_STREAM_START", {
     streamStartTime: String(streamStartTime)
   });
@@ -9532,9 +9550,18 @@ async function handleMessageStream(stream, signal) {
         }
         if (!usage) {
           usage = chunk.usage;
+          if (chunk.usage?.prompt_tokens) {
+            setRequestInputTokens(chunk.usage.prompt_tokens);
+          }
         }
         message = messageReducer(message, chunk);
         if (chunk?.choices?.[0]?.delta?.content) {
+          if (!hasMarkedStreaming) {
+            setRequestStatus({ kind: "streaming" });
+            hasMarkedStreaming = true;
+          }
+          outputTokenCount++;
+          updateRequestTokens(outputTokenCount);
           if (!ttftMs) {
             ttftMs = Date.now() - streamStartTime;
             debug2.api("OPENAI_STREAM_FIRST_TOKEN", {
@@ -9543,6 +9570,9 @@ async function handleMessageStream(stream, signal) {
             });
           }
         }
+        if (chunk?.usage?.completion_tokens) {
+          updateRequestTokens(chunk.usage.completion_tokens);
+        }
       } catch (chunkError) {
         errorCount++;
         debug2.error("OPENAI_STREAM_CHUNK_ERROR", {
@@ -9587,6 +9617,7 @@ async function handleMessageStream(stream, signal) {
 var init_stream2 = __esm({
   "packages/core/src/ai/llm/openai/stream.ts"() {
     init_debugLogger2();
+    init_requestStatus();
   }
 });
 
@@ -10356,7 +10387,7 @@ async function createAnthropicStreamingMessage(anthropic2, params, signal, optio
     },
     {
       signal
-      // ← CRITICAL: Connect the AbortSignal to API call
+      // CRITICAL: Connect the AbortSignal to API call
     }
   );
   let finalResponse = null;
@@ -10367,6 +10398,7 @@ async function createAnthropicStreamingMessage(anthropic2, params, signal, optio
   let stopReason = null;
   let stopSequence = null;
   let hasMarkedStreaming = false;
+  let outputTokenCount = 0;
   for await (const event of stream) {
     try {
       options?.onStreamEvent?.(event);
@@ -10387,6 +10419,9 @@ async function createAnthropicStreamingMessage(anthropic2, params, signal, optio
           content: []
           // Will be populated from content blocks
         };
+        if (event.message?.usage?.input_tokens) {
+          setRequestInputTokens(event.message.usage.input_tokens);
+        }
         break;
       case "content_block_start":
         contentBlocks[event.index] = { ...event.content_block };
@@ -10419,6 +10454,8 @@ async function createAnthropicStreamingMessage(anthropic2, params, signal, optio
             hasMarkedStreaming = true;
           }
           contentBlocks[blockIndex].text += event.delta.text;
+          outputTokenCount++;
+          updateRequestTokens(outputTokenCount);
         } else if (event.delta.type === "input_json_delta") {
           const currentBuffer = inputJSONBuffers.get(blockIndex) || "";
           const nextBuffer = currentBuffer + event.delta.partial_json;
@@ -10434,7 +10471,12 @@ async function createAnthropicStreamingMessage(anthropic2, params, signal, optio
       case "message_delta":
         if (event.delta.stop_reason) stopReason = event.delta.stop_reason;
         if (event.delta.stop_sequence) stopSequence = event.delta.stop_sequence;
-        if (event.usage) usage = { ...usage, ...event.usage };
+        if (event.usage) {
+          usage = { ...usage, ...event.usage };
+          if (event.usage.output_tokens) {
+            updateRequestTokens(event.usage.output_tokens);
+          }
+        }
         break;
       case "content_block_stop":
         const stopIndex = event.index;
@@ -11732,7 +11774,7 @@ var init_anthropic2 = __esm({
 });
 
 // packages/core/src/services/vcr.ts
-import { createHash } from "crypto";
+import { createHash as createHash2 } from "crypto";
 import { mkdirSync as mkdirSync8, readFileSync as readFileSync9, writeFileSync as writeFileSync5 } from "fs";
 import { dirname as dirname8 } from "path";
 import { existsSync as existsSync17 } from "fs";
@@ -11746,7 +11788,7 @@ async function withVCR(messages, f) {
     messages.map((_) => _.message.content),
     dehydrateValue
   );
-  const filename = `./fixtures/${dehydratedInput.map((_) => createHash("sha1").update(JSON.stringify(_)).digest("hex").slice(0, 6)).join("-")}.json`;
+  const filename = `./fixtures/${dehydratedInput.map((_) => createHash2("sha1").update(JSON.stringify(_)).digest("hex").slice(0, 6)).join("-")}.json`;
   if (existsSync17(filename)) {
     const cached = JSON.parse(readFileSync9(filename, "utf-8"));
     return mapAssistantMessage(cached.output, hydrateValue);
@@ -14979,7 +15021,10 @@ function countTokens(messages) {
     const message = messages[i];
     if (message?.type === "assistant" && "usage" in message.message && !(message.message.content[0]?.type === "text" && SYNTHETIC_ASSISTANT_MESSAGES.has(message.message.content[0].text))) {
       const { usage } = message.message;
-      return usage.input_tokens + (usage.cache_creation_input_tokens ?? 0) + (usage.cache_read_input_tokens ?? 0) + usage.output_tokens;
+      const total = usage.input_tokens + (usage.cache_creation_input_tokens ?? 0) + (usage.cache_read_input_tokens ?? 0) + usage.output_tokens;
+      if (total > 0) {
+        return total;
+      }
     }
     i--;
   }
@@ -17225,8 +17270,14 @@ var AbortError = class extends Error {
 init_log2();
 init_product();
 
+// packages/core/src/types/PermissionMode.ts
+function normalizePermissionMode(mode) {
+  return mode === "default" ? "cautious" : mode;
+}
+
 // packages/core/src/utils/permissionModeState.ts
 var DEFAULT_CONVERSATION_KEY2 = "default";
+var ACTUAL_DEFAULT_MODE = "yolo";
 var permissionModeByConversationKey = /* @__PURE__ */ new Map();
 function getConversationKey2(context) {
   const messageLogName = context?.options?.messageLogName ?? DEFAULT_CONVERSATION_KEY2;
@@ -17236,31 +17287,34 @@ function getConversationKey2(context) {
 function getPermissionModeForConversationKey(options) {
   const existing = permissionModeByConversationKey.get(options.conversationKey);
   if (existing) {
-    if (existing === "bypassPermissions" && !options.isBypassPermissionsModeAvailable) {
-      permissionModeByConversationKey.set(options.conversationKey, "default");
-      return "default";
+    const normalized = normalizePermissionMode(existing);
+    if (normalized === "bypassPermissions" && !options.isBypassPermissionsModeAvailable) {
+      permissionModeByConversationKey.set(options.conversationKey, ACTUAL_DEFAULT_MODE);
+      return ACTUAL_DEFAULT_MODE;
     }
-    return existing;
+    return normalized;
   }
-  permissionModeByConversationKey.set(options.conversationKey, "default");
-  return "default";
+  permissionModeByConversationKey.set(options.conversationKey, ACTUAL_DEFAULT_MODE);
+  return ACTUAL_DEFAULT_MODE;
 }
 function getPermissionMode(context) {
   const conversationKey = getConversationKey2(context);
   const safeMode = context?.options?.safeMode ?? false;
   const fromToolPermissionContext = context?.options?.toolPermissionContext?.mode;
-  if (fromToolPermissionContext === "default" || fromToolPermissionContext === "acceptEdits" || fromToolPermissionContext === "plan" || fromToolPermissionContext === "dontAsk" || fromToolPermissionContext === "bypassPermissions") {
-    if (fromToolPermissionContext === "bypassPermissions" && safeMode) {
-      return "default";
+  if (fromToolPermissionContext) {
+    const normalized = normalizePermissionMode(fromToolPermissionContext);
+    if (normalized === "bypassPermissions" && safeMode) {
+      return ACTUAL_DEFAULT_MODE;
     }
-    return fromToolPermissionContext;
+    return normalized;
   }
   const override = context?.options?.permissionMode;
-  if (override === "default" || override === "acceptEdits" || override === "plan" || override === "dontAsk" || override === "bypassPermissions") {
-    if (override === "bypassPermissions" && safeMode) {
-      return "default";
+  if (override) {
+    const normalized = normalizePermissionMode(override);
+    if (normalized === "bypassPermissions" && safeMode) {
+      return ACTUAL_DEFAULT_MODE;
     }
-    return override;
+    return normalized;
   }
   return getPermissionModeForConversationKey({
     conversationKey,
@@ -17271,7 +17325,7 @@ function getPermissionMode(context) {
 // packages/core/src/types/toolPermissionContext.ts
 function createDefaultToolPermissionContext(options) {
   return {
-    mode: "default",
+    mode: "yolo",
     additionalWorkingDirectories: /* @__PURE__ */ new Map(),
     alwaysAllowRules: {},
     alwaysDenyRules: {},
@@ -21649,8 +21703,10 @@ function flattenPermissionRuleGroups(groups) {
   return out;
 }
 var hasPermissionsToUseTool = async (tool, input, context, assistantMessage) => {
-  const permissionMode = getPermissionMode(context);
+  const rawPermissionMode = getPermissionMode(context);
+  const permissionMode = normalizePermissionMode(rawPermissionMode);
   const isDontAskMode = permissionMode === "dontAsk";
+  const isYoloMode = permissionMode === "yolo";
   const shouldAvoidPermissionPrompts = context.options?.shouldAvoidPermissionPrompts === true;
   const safeMode = Boolean(context.options?.safeMode ?? context.safeMode);
   const requiresUserInteraction = tool.requiresUserInteraction?.(input) ?? false;
@@ -21853,6 +21909,9 @@ var hasPermissionsToUseTool = async (tool, input, context, assistantMessage) =>
   if (shouldAvoidPermissionPrompts && permissionResult.result === false && permissionResult.shouldPromptUser !== false) {
     return promptsUnavailableDenied;
   }
+  if (isYoloMode && !requiresUserInteraction && permissionResult.result === false && permissionResult.shouldPromptUser !== false) {
+    return { result: true };
+  }
   return permissionResult;
 };
 
@@ -22099,7 +22158,7 @@ function getToolPermissionContextForConversationKey(options) {
       };
     }
     if (!options.isBypassPermissionsModeAvailable && next.mode === "bypassPermissions") {
-      next = { ...next, mode: "default" };
+      next = { ...next, mode: "yolo" };
     }
     if (next !== existing) {
       toolPermissionContextByConversationKey.set(options.conversationKey, next);