@shareai-lab/kode 1.1.12 → 1.1.13

package/src/tools/MultiEditTool/MultiEditTool.tsx

@@ -19,7 +19,28 @@ import { logError } from '../../utils/log'
 import { getCwd } from '../../utils/state'
 import { getTheme } from '../../utils/theme'
 import { NotebookEditTool } from '../NotebookEditTool/NotebookEditTool'
-import { applyEdit } from '../FileEditTool/utils'
+// Local content-based edit function for MultiEditTool
+function applyContentEdit(
+  content: string,
+  oldString: string,
+  newString: string,
+  replaceAll: boolean = false
+): { newContent: string; occurrences: number } {
+  if (replaceAll) {
+    const regex = new RegExp(oldString.replace(/[.*+?^${}()|[\]\\]/g, '\\$&'), 'g')
+    const matches = content.match(regex)
+    const occurrences = matches ? matches.length : 0
+    const newContent = content.replace(regex, newString)
+    return { newContent, occurrences }
+  } else {
+    if (content.includes(oldString)) {
+      const newContent = content.replace(oldString, newString)
+      return { newContent, occurrences: 1 }
+    } else {
+      throw new Error(`String not found: ${oldString.substring(0, 50)}...`)
+    }
+  }
+}
 import { hasWritePermission } from '../../utils/permissions/filesystem'
 import { PROJECT_FILE } from '../../constants/product'
 import { DESCRIPTION, PROMPT } from './prompt'
@@ -274,7 +295,7 @@ export const MultiEditTool = {
         const { old_string, new_string, replace_all } = edit
 
         try {
-          const result = applyEdit(
+          const result = applyContentEdit(
             modifiedContent,
             old_string,
             new_string,
@@ -302,8 +323,9 @@ export const MultiEditTool = {
       }
 
       // Write the modified content
-      const lineEndings = fileExists ? detectLineEndings(currentContent) : '\n'
-      writeTextContent(filePath, modifiedContent, lineEndings)
+      const lineEndings = fileExists ? detectLineEndings(currentContent) : 'LF'
+      const encoding = fileExists ? detectFileEncoding(filePath) : 'utf8'
+      writeTextContent(filePath, modifiedContent, encoding, lineEndings)
 
       // Record Agent edit operation for file freshness tracking
       recordFileEdit(filePath, modifiedContent)

package/src/tools/NotebookReadTool/NotebookReadTool.tsx

@@ -176,7 +176,7 @@ function processOutput(output: NotebookCellOutput) {
     case 'display_data':
       return {
         output_type: output.output_type,
-        text: processOutputText(output.data?.['text/plain']),
+        text: processOutputText(output.data?.['text/plain'] as string | string[] | undefined),
         image: output.data && extractImage(output.data),
       }
     case 'error':

package/src/tools/StickerRequestTool/StickerRequestTool.tsx

@@ -1,7 +1,7 @@
 import { z } from 'zod'
 import React from 'react'
 import { Text } from 'ink'
-import { Tool, ToolUseContext } from '../../Tool'
+import { Tool, ToolUseContext, ExtendedToolUseContext } from '../../Tool'
 import { DESCRIPTION, PROMPT } from './prompt'
 import {
   StickerRequestForm,
@@ -38,8 +38,11 @@ export const StickerRequestTool: Tool = {
       resolveForm = success => resolve(success)
     })
 
-    context.setToolJSX?.({
-      jsx: (
+    // Check if setToolJSX is available (cast context if needed)
+    const extendedContext = context as ExtendedToolUseContext
+    if (extendedContext.setToolJSX) {
+      extendedContext.setToolJSX({
+        jsx: (
         <StickerRequestForm
           onSubmit={(formData: FormData) => {
             // Log successful completion with form data
@@ -48,18 +51,27 @@ export const StickerRequestTool: Tool = {
               has_optional_address: Boolean(formData.address2).toString(),
             })
             resolveForm(true)
-            context.setToolJSX?.(null) // Clear the JSX
+            if (extendedContext.setToolJSX) {
+              extendedContext.setToolJSX(null) // Clear the JSX
+            }
           }}
           onClose={() => {
             // Log form cancellation
             logEvent('sticker_request_form_cancelled', {})
             resolveForm(false)
-            context.setToolJSX?.(null) // Clear the JSX
+            if (extendedContext.setToolJSX) {
+              extendedContext.setToolJSX(null) // Clear the JSX
+            }
           }}
         />
-      ),
-      shouldHidePromptInput: true,
-    })
+        ),
+        shouldHidePromptInput: true,
+      })
+    } else {
+      // Fallback if setToolJSX is not available
+      console.log('Sticker form would be displayed here, but setToolJSX is not available')
+      resolveForm(false)
+    }
 
     // Wait for form completion and get status
     const success = await formComplete
@@ -82,12 +94,14 @@ export const StickerRequestTool: Tool = {
     return ''
   },
 
-  renderToolUseRejectedMessage: _input => (
-    <Text>
-      &nbsp;&nbsp;⎿ &nbsp;
-      <Text color={getTheme().error}>No (Sticker request cancelled)</Text>
-    </Text>
-  ),
+  renderToolUseRejectedMessage() {
+    return (
+      <Text>
+        &nbsp;&nbsp;⎿ &nbsp;
+        <Text color={getTheme().error}>No (Sticker request cancelled)</Text>
+      </Text>
+    )
+  },
 
   renderResultForAssistant: (content: string) => content,
 }

package/src/tools/TaskTool/TaskTool.tsx

@@ -72,7 +72,7 @@ export const TaskTool = {
       options: { safeMode = false, forkNumber, messageLogName, verbose },
       readFileTimestamps,
     },
-  ) {
+  ): AsyncGenerator<{ type: 'result'; data: TextBlock[]; resultForAssistant?: string }, void, unknown> {
     const startTime = Date.now()
     
     // Default to general-purpose if no subagent_type specified
@@ -95,7 +95,7 @@ export const TaskTool = {
         
         yield {
           type: 'result',
-          data: { error: helpMessage },
+          data: [{ type: 'text', text: helpMessage }] as TextBlock[],
           resultForAssistant: helpMessage,
         }
         return
@@ -135,14 +135,7 @@ export const TaskTool = {
       }
     }
 
-    // We yield an initial message immediately so the UI
-    // doesn't move around when messages start streaming back.
-    yield {
-      type: 'progress',
-      content: createAssistantMessage(chalk.dim(`[${agentType}] ${description}`)),
-      normalizedMessages: normalizeMessages(messages),
-      tools,
-    }
+    // Skip initial progress yield - only yield results for Tool interface
 
     const [taskPrompt, context, maxThinkingTokens] = await Promise.all([
       getAgentPrompt(),
@@ -194,6 +187,7 @@ export const TaskTool = {
         messageId: getLastAssistantMessageId(messages),
         agentId: taskId,
         readFileTimestamps,
+        setToolJSX: () => {}, // No-op implementation for TaskTool
       },
     )) {
       messages.push(message)
@@ -214,12 +208,7 @@ export const TaskTool = {
         if (content.type === 'text' && content.text && content.text !== INTERRUPT_MESSAGE) {
           // Show agent's reasoning/responses
           const preview = content.text.length > 200 ? content.text.substring(0, 200) + '...' : content.text
-          yield {
-            type: 'progress',
-            content: createAssistantMessage(`[${agentType}] ${preview}`),
-            normalizedMessages,
-            tools,
-          }
+          // Skip progress yield - only yield results for Tool interface
         } else if (content.type === 'tool_use') {
           toolUseCount++
           
@@ -250,12 +239,7 @@ export const TaskTool = {
               }
             }
             
-            yield {
-              type: 'progress',
-              content: modifiedMessage,
-              normalizedMessages,
-              tools,
-            }
+            // Skip progress yield - only yield results for Tool interface
           }
         }
       }
@@ -273,12 +257,7 @@ export const TaskTool = {
         _ => _.type === 'text' && _.text === INTERRUPT_MESSAGE,
       )
     ) {
-      yield {
-        type: 'progress',
-        content: lastMessage,
-        normalizedMessages,
-        tools,
-      }
+      // Skip progress yield - only yield final result
     } else {
       const result = [
         toolUseCount === 1 ? '1 tool use' : `${toolUseCount} tool uses`,
@@ -290,12 +269,7 @@ export const TaskTool = {
         ) + ' tokens',
         formatDuration(Date.now() - startTime),
       ]
-      yield {
-        type: 'progress',
-        content: createAssistantMessage(`[${agentType}] Completed (${result.join(' · ')})`),
-        normalizedMessages,
-        tools,
-      }
+      // Skip progress yield - only yield final result
     }
 
     // Output is an AssistantMessage, but since TaskTool is a tool, it needs
@@ -304,9 +278,7 @@ export const TaskTool = {
     yield {
       type: 'result',
       data,
-      normalizedMessages,
       resultForAssistant: this.renderResultForAssistant(data),
-      tools,
     }
   },
 

package/src/types/common.d.ts

@@ -0,0 +1,2 @@
+// UUID 类型定义
+export type UUID = `${string}-${string}-${string}-${string}-${string}`;

package/src/utils/generators.ts

@@ -51,7 +51,7 @@ export async function* all<A>(
       promises.add(next(generator))
       // TODO: Clean this up
       if (value !== undefined) {
-        yield value
+        yield value as A
       }
     } else if (waiting.length > 0) {
       // Start a new generator when one finishes

package/src/utils/messageContextManager.ts

@@ -1,5 +1,7 @@
 import { Message } from '../query'
+import type { UUID } from '../types/common'
 import { countTokens } from './tokens'
+import crypto from 'crypto'
 
 export interface MessageRetentionStrategy {
   type:
@@ -144,6 +146,9 @@ export class MessageContextManager {
           },
         ],
       },
+      costUSD: 0,
+      durationMs: 0,
+      uuid: crypto.randomUUID() as UUID
     }
 
     const truncatedMessages = [summaryMessage, ...recentMessages]

package/src/utils/messages.tsx

@@ -298,7 +298,6 @@ export async function processUserInput(
       newMessages[0]!.type === 'user' &&
       newMessages[1]!.type === 'assistant' &&
       typeof newMessages[1]!.message.content === 'string' &&
-      // @ts-expect-error: TODO: this is probably a bug
       newMessages[1]!.message.content.startsWith('Unknown command:')
     ) {
       logEvent('tengu_input_slash_invalid', { input })
@@ -436,7 +435,14 @@ async function getMessagesForSlashCommand(
 
         try {
           // Use the context's abortController for local commands
-          const result = await command.call(args, context)
+          const result = await command.call(args, {
+            ...context,
+            options: {
+              commands: context.options.commands || [],
+              tools: context.options.tools || [],
+              slowAndCapableModel: context.options.slowAndCapableModel || 'main'
+            }
+          })
 
           return [
             userMessage,

package/src/utils/thinking.ts

@@ -112,7 +112,7 @@ export async function getReasoningEffort(
     // 🔧 Fix: Use ModelManager fallback instead of legacy config
     const modelManager = getModelManager()
     const fallbackProfile = modelManager.getModel('main')
-    reasoningEffort = fallbackProfile?.reasoningEffort || 'medium'
+    reasoningEffort = (fallbackProfile?.reasoningEffort === 'minimal' ? 'low' : fallbackProfile?.reasoningEffort) || 'medium'
   }
 
   const maxEffort =