feat: add dimcode acp agent (#1336)

zerob13 · web-flow · commit d732e24a391c · 2026-03-08T20:25:29.000+08:00
* fix(think): persist reasoning duration

* feat(acp): add dimcode-acp builtin agent

* fix(session): sync model settings on switch

* fix: test case
diff --git a/src/main/presenter/configPresenter/acpConfHelper.ts b/src/main/presenter/configPresenter/acpConfHelper.ts
@@ -13,7 +13,12 @@ import { McpConfHelper } from './mcpConfHelper'
 const ACP_STORE_VERSION = '2'
 const DEFAULT_PROFILE_NAME = 'Default'
 
-const BUILTIN_ORDER: AcpBuiltinAgentId[] = ['kimi-cli', 'claude-code-acp', 'codex-acp']
+const BUILTIN_ORDER: AcpBuiltinAgentId[] = [
+  'kimi-cli',
+  'claude-code-acp',
+  'codex-acp',
+  'dimcode-acp'
+]
 
 interface BuiltinTemplate {
   name: string
@@ -47,6 +52,15 @@ const BUILTIN_TEMPLATES: Record<AcpBuiltinAgentId, BuiltinTemplate> = {
       args: ['-y', '@zed-industries/codex-acp'],
       env: {}
     })
+  },
+  'dimcode-acp': {
+    name: 'DimCode',
+    defaultProfile: () => ({
+      name: DEFAULT_PROFILE_NAME,
+      command: 'dim',
+      args: ['acp'],
+      env: {}
+    })
   }
 }
 
diff --git a/src/main/presenter/configPresenter/acpInitHelper.ts b/src/main/presenter/configPresenter/acpInitHelper.ts
@@ -67,6 +67,10 @@ const BUILTIN_INIT_COMMANDS: Record<AcpBuiltinAgentId, InitCommandConfig> = {
   'codex-acp': {
     commands: ['npm i -g @zed-industries/codex-acp', 'npm install -g @openai/codex', 'codex'],
     description: 'Initialize Codex CLI ACP'
+  },
+  'dimcode-acp': {
+    commands: ['npm i -g dimcode', 'dim'],
+    description: 'Initialize DimCode ACP'
   }
 }
 
diff --git a/src/main/presenter/deepchatAgentPresenter/accumulator.ts b/src/main/presenter/deepchatAgentPresenter/accumulator.ts
@@ -20,6 +20,16 @@ function getCurrentBlock(
   return block
 }
 
+function updateReasoningMetadata(state: StreamState, start: number, end: number): void {
+  const relativeStart = Math.max(0, start - state.startTime)
+  const relativeEnd = Math.max(0, end - state.startTime)
+
+  if (state.metadata.reasoningStartTime === undefined) {
+    state.metadata.reasoningStartTime = relativeStart
+  }
+  state.metadata.reasoningEndTime = relativeEnd
+}
+
 /**
  * Apply a single stream event to the accumulator state.
  * Pure block mutations only — no I/O, no finalization, no emit.
@@ -34,9 +44,25 @@ export function accumulate(state: StreamState, event: LLMCoreStreamEvent): void
       break
     }
     case 'reasoning': {
-      if (state.firstTokenTime === null) state.firstTokenTime = Date.now()
+      const currentTime = Date.now()
+      if (state.firstTokenTime === null) state.firstTokenTime = currentTime
       const block = getCurrentBlock(state.blocks, 'reasoning_content')
       block.content += event.reasoning_content
+      if (
+        typeof block.reasoning_time !== 'object' ||
+        block.reasoning_time === null ||
+        typeof block.reasoning_time.start !== 'number' ||
+        typeof block.reasoning_time.end !== 'number'
+      ) {
+        block.reasoning_time = {
+          start: currentTime,
+          end: currentTime
+        }
+      } else {
+        block.reasoning_time.end = currentTime
+      }
+      const reasoningTime = block.reasoning_time as { start: number; end: number }
+      updateReasoningMetadata(state, reasoningTime.start, reasoningTime.end)
       state.dirty = true
       break
     }
diff --git a/src/main/presenter/deepchatAgentPresenter/index.ts b/src/main/presenter/deepchatAgentPresenter/index.ts
@@ -618,12 +618,9 @@ export class DeepChatAgentPresenter implements IAgentImplementation {
     }
 
     const currentGeneration = await this.getEffectiveSessionGenerationSettings(sessionId)
-    const sanitized = await this.sanitizeGenerationSettings(
-      nextProviderId,
-      nextModelId,
-      {},
-      currentGeneration
-    )
+    const sanitized = await this.sanitizeGenerationSettings(nextProviderId, nextModelId, {
+      systemPrompt: currentGeneration.systemPrompt
+    })
 
     if (state) {
       state.providerId = nextProviderId
diff --git a/src/renderer/src/assets/llm-icons/dimcode.svg b/src/renderer/src/assets/llm-icons/dimcode.svg
@@ -0,0 +1,3 @@
+<svg width="16" height="16" viewBox="0 0 16 16" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M3.12109 11.0078H1.99902V5.49316H3.12109V11.0078ZM3.7041 5.49316C4.91979 5.49316 5.31142 5.57449 5.80762 5.7373C6.50208 5.97365 6.85299 6.40958 6.86133 7.04492V9.45605C6.86125 10.207 6.37767 10.6797 5.41016 10.874C4.95546 10.9633 4.69632 11.0078 3.7041 11.0078V10.6064C4.72131 10.6064 4.95994 10.5507 5.34863 10.4404C5.91072 10.2671 6.19576 9.93907 6.2041 9.45605V7.04492C6.20402 6.4883 5.83211 6.13886 5.08789 5.99707C4.74057 5.93405 4.58897 5.89978 3.7041 5.89453V5.49316ZM9.16797 5.49316V11.0078H8.0459V5.49316H9.16797ZM14 6.79297V11.0078H12.8779V8.0791L13.8877 6.79297H14ZM14 5.49316V5.7373L11.3594 8.97852H10.7852L9.74219 6.94531V6.07812H9.86719L11.0723 8.33203L13.4258 5.49316H14Z" fill="currentColor"/>
+</svg>
diff --git a/src/renderer/src/components/chat/ChatStatusBar.vue b/src/renderer/src/components/chat/ChatStatusBar.vue
@@ -10,7 +10,7 @@
               class="h-6 px-2 gap-1 text-xs text-muted-foreground hover:text-foreground backdrop-blur-lg"
             >
               <ModelIcon
-                :model-id="displayProviderId"
+                :model-id="displayIconId"
                 custom-class="w-3.5 h-3.5"
                 :is-dark="themeStore.isDark"
               />
@@ -25,7 +25,7 @@
                 @click="selectModel(group.providerId, group.model.id)"
               >
                 <ModelIcon
-                  :model-id="group.providerId"
+                  :model-id="resolveModelIconId(group.providerId, group.model.id)"
                   custom-class="w-3.5 h-3.5"
                   :is-dark="themeStore.isDark"
                 />
@@ -42,7 +42,7 @@
           :disabled="true"
         >
           <ModelIcon
-            :model-id="displayProviderId"
+            :model-id="displayIconId"
             custom-class="w-3.5 h-3.5"
             :is-dark="themeStore.isDark"
           />
@@ -635,18 +635,27 @@ watch(
   { immediate: true }
 )
 
-const displayProviderId = computed(() => {
+const resolveModelIconId = (providerId?: string | null, modelId?: string | null): string => {
+  if (providerId === 'acp' && modelId) {
+    return modelId
+  }
+  return providerId || 'anthropic'
+}
+
+const displayIconId = computed(() => {
   if (hasActiveSession.value) {
-    return (
-      activeSessionSelection.value?.providerId ||
-      draftModelSelection.value?.providerId ||
-      'anthropic'
+    return resolveModelIconId(
+      activeSessionSelection.value?.providerId || draftModelSelection.value?.providerId,
+      activeSessionSelection.value?.modelId || draftModelSelection.value?.modelId
     )
   }
   if (isAcpAgent.value) {
-    return agentStore.selectedAgentId ?? 'acp'
+    return resolveModelIconId('acp', agentStore.selectedAgentId)
   }
-  return draftModelSelection.value?.providerId || 'anthropic'
+  return resolveModelIconId(
+    draftModelSelection.value?.providerId,
+    draftModelSelection.value?.modelId
+  )
 })
 
 const displayModelName = computed(() => {
diff --git a/src/renderer/src/components/icons/ModelIcon.vue b/src/renderer/src/components/icons/ModelIcon.vue
@@ -32,6 +32,7 @@ import deepseekColorIcon from '@/assets/llm-icons/deepseek-color.svg?url'
 import openaiColorIcon from '@/assets/llm-icons/openai.svg?url'
 import ollamaColorIcon from '@/assets/llm-icons/ollama.svg?url'
 import doubaoColorIcon from '@/assets/llm-icons/doubao-color.svg?url'
+import dimcodeColorIcon from '@/assets/llm-icons/dimcode.svg?url'
 import minimaxColorIcon from '@/assets/llm-icons/minimax-color.svg?url'
 import fireworksColorIcon from '@/assets/llm-icons/fireworks-color.svg?url'
 import zerooneColorIcon from '@/assets/llm-icons/zeroone.svg?url'
@@ -75,6 +76,7 @@ const icons = {
   'kimi-cli': moonshotColorIcon,
   'claude-code-acp': claudeColorIcon,
   'codex-acp': openaiColorIcon,
+  'dimcode-acp': dimcodeColorIcon,
   o3fan: o3fanColorIcon,
   cherryin: cherryinColorIcon,
   modelscope: modelscopeColorIcon,
@@ -197,6 +199,7 @@ const iconKey = computed(() => {
 // Monochrome icon URLs that need inversion in dark mode
 const monoIconUrls = new Set([
   openaiColorIcon,
+  dimcodeColorIcon,
   ollamaColorIcon,
   zerooneColorIcon,
   xaiColorIcon,
diff --git a/src/renderer/src/i18n/zh-CN/routes.json b/src/renderer/src/i18n/zh-CN/routes.json
@@ -14,6 +14,6 @@
   "settings-prompt": "Prompt管理",
   "settings-mcp-market": "MCP市场",
   "settings-acp": "ACP Agent",
-  "settings-skills": "skills设置",
+  "settings-skills": "Skills设置",
   "settings-notifications-hooks": "通知与Hooks"
 }
diff --git a/src/renderer/src/i18n/zh-CN/settings.json b/src/renderer/src/i18n/zh-CN/settings.json
@@ -1108,7 +1108,7 @@
     "resetToDefaultFailed": "重置失败，请重试"
   },
   "skills": {
-    "title": "skills管理",
+    "title": "Skills设置",
     "description": "管理和配置 AI 助手的 skill 模块",
     "openFolder": "打开文件夹",
     "addSkill": "添加 skill",
diff --git a/src/renderer/src/i18n/zh-HK/routes.json b/src/renderer/src/i18n/zh-HK/routes.json
@@ -14,6 +14,6 @@
   "settings-mcp-market": "MCP市場",
   "playground": "Playground 實驗室",
   "settings-acp": "ACP Agent",
-  "settings-skills": "skills設置",
+  "settings-skills": "Skills設置",
   "settings-notifications-hooks": "通知與 Hooks"
 }
diff --git a/src/renderer/src/i18n/zh-HK/settings.json b/src/renderer/src/i18n/zh-HK/settings.json
@@ -1256,6 +1256,6 @@
       "syncing": "正在匯入...",
       "title": "外部工具"
     },
-    "title": "skills管理"
+    "title": "Skills設置"
   }
 }
diff --git a/src/renderer/src/i18n/zh-TW/routes.json b/src/renderer/src/i18n/zh-TW/routes.json
@@ -14,6 +14,6 @@
   "settings-mcp-market": "MCP市場",
   "playground": "Playground 實驗室",
   "settings-acp": "ACP Agent",
-  "settings-skills": "skills管理",
+  "settings-skills": "Skills設定",
   "settings-notifications-hooks": "通知與 Hooks"
 }
diff --git a/src/renderer/src/i18n/zh-TW/settings.json b/src/renderer/src/i18n/zh-TW/settings.json
@@ -1256,6 +1256,6 @@
       "syncing": "正在匯入...",
       "title": "外部工具"
     },
-    "title": "skills管理"
+    "title": "Skills設定"
   }
 }
diff --git a/src/renderer/src/pages/ChatPage.vue b/src/renderer/src/pages/ChatPage.vue
@@ -187,8 +187,8 @@ function buildUsage(metadata: MessageMetadata): Message['usage'] {
     total_tokens: metadata.totalTokens ?? 0,
     generation_time: metadata.generationTime ?? 0,
     first_token_time: metadata.firstTokenTime ?? 0,
-    reasoning_start_time: 0,
-    reasoning_end_time: 0,
+    reasoning_start_time: metadata.reasoningStartTime ?? 0,
+    reasoning_end_time: metadata.reasoningEndTime ?? 0,
     input_tokens: metadata.inputTokens ?? 0,
     output_tokens: metadata.outputTokens ?? 0
   }
diff --git a/src/shared/types/agent-interface.d.ts b/src/shared/types/agent-interface.d.ts
@@ -234,6 +234,8 @@ export interface MessageMetadata {
   outputTokens?: number
   generationTime?: number
   firstTokenTime?: number
+  reasoningStartTime?: number
+  reasoningEndTime?: number
   tokensPerSecond?: number
   model?: string
   provider?: string
diff --git a/src/shared/types/presenters/legacy.presenters.d.ts b/src/shared/types/presenters/legacy.presenters.d.ts
@@ -890,7 +890,7 @@ export interface AcpDebugRunResult {
   events: AcpDebugEventEntry[]
 }
 
-export type AcpBuiltinAgentId = 'kimi-cli' | 'claude-code-acp' | 'codex-acp'
+export type AcpBuiltinAgentId = 'kimi-cli' | 'claude-code-acp' | 'codex-acp' | 'dimcode-acp'
 
 export interface AcpAgentProfile {
   id: string
diff --git a/test/main/presenter/configPresenter/acpConfHelper.test.ts b/test/main/presenter/configPresenter/acpConfHelper.test.ts
@@ -0,0 +1,86 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+const mockStores = vi.hoisted(() => new Map<string, Record<string, any>>())
+
+const clone = <T>(value: T): T => {
+  const cloneFn = (globalThis as typeof globalThis & { structuredClone?: (input: T) => T })
+    .structuredClone
+
+  if (typeof cloneFn === 'function') {
+    return cloneFn(value)
+  }
+
+  return JSON.parse(JSON.stringify(value)) as T
+}
+
+vi.mock('electron-store', () => ({
+  default: class MockElectronStore {
+    private readonly data: Record<string, any>
+
+    constructor(options: { name: string; defaults?: Record<string, any> }) {
+      if (!mockStores.has(options.name)) {
+        mockStores.set(options.name, clone(options.defaults ?? {}))
+      }
+      this.data = mockStores.get(options.name)!
+    }
+
+    get(key: string) {
+      return this.data[key]
+    }
+
+    set(key: string, value: any) {
+      this.data[key] = value
+    }
+
+    delete(key: string) {
+      delete this.data[key]
+    }
+  }
+}))
+
+vi.mock('../../../../src/main/presenter/configPresenter/mcpConfHelper', () => ({
+  McpConfHelper: class MockMcpConfHelper {
+    async getMcpServers() {
+      return {}
+    }
+  }
+}))
+
+describe('AcpConfHelper DimCode builtin', () => {
+  beforeEach(() => {
+    mockStores.clear()
+    vi.resetModules()
+  })
+
+  it('includes DimCode in default builtins with dim acp profile', async () => {
+    const { AcpConfHelper } =
+      await import('../../../../src/main/presenter/configPresenter/acpConfHelper')
+    const helper = new AcpConfHelper()
+
+    const dimcode = helper.getBuiltins().find((agent) => agent.id === 'dimcode-acp')
+
+    expect(dimcode).toBeDefined()
+    expect(dimcode?.name).toBe('DimCode')
+    expect(dimcode?.profiles).toHaveLength(1)
+    expect(dimcode?.profiles[0]?.command).toBe('dim')
+    expect(dimcode?.profiles[0]?.args).toEqual(['acp'])
+  })
+
+  it('returns enabled DimCode from getEnabledAgents', async () => {
+    const { AcpConfHelper } =
+      await import('../../../../src/main/presenter/configPresenter/acpConfHelper')
+    const helper = new AcpConfHelper()
+
+    helper.setGlobalEnabled(true)
+    helper.setBuiltinEnabled('dimcode-acp', true)
+
+    expect(helper.getEnabledAgents()).toContainEqual(
+      expect.objectContaining({
+        id: 'dimcode-acp',
+        name: 'DimCode - Default',
+        command: 'dim',
+        args: ['acp']
+      })
+    )
+  })
+})
diff --git a/test/main/presenter/deepchatAgentPresenter/accumulator.test.ts b/test/main/presenter/deepchatAgentPresenter/accumulator.test.ts
diff --git a/test/main/presenter/deepchatAgentPresenter/deepchatAgentPresenter.test.ts b/test/main/presenter/deepchatAgentPresenter/deepchatAgentPresenter.test.ts
diff --git a/test/renderer/components/ChatPage.test.ts b/test/renderer/components/ChatPage.test.ts
diff --git a/test/renderer/components/ChatStatusBar.test.ts b/test/renderer/components/ChatStatusBar.test.ts
diff --git a/test/renderer/components/ModelIcon.test.ts b/test/renderer/components/ModelIcon.test.ts
diff --git a/test/renderer/components/message/MessageBlockThink.test.ts b/test/renderer/components/message/MessageBlockThink.test.ts

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,10 @@ const BUILTIN_INIT_COMMANDS: Record<AcpBuiltinAgentId, InitCommandConfig> = {`
`67`	`67`	`'codex-acp': {`
`68`	`68`	`commands: ['npm i -g @zed-industries/codex-acp', 'npm install -g @openai/codex', 'codex'],`
`69`	`69`	`description: 'Initialize Codex CLI ACP'`
	`70`	`+ },`
	`71`	`+ 'dimcode-acp': {`
	`72`	`+ commands: ['npm i -g dimcode', 'dim'],`
	`73`	`+ description: 'Initialize DimCode ACP'`
`70`	`74`	`}`
`71`	`75`	`}`
`72`	`76`
Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,6 @@`
`14`	`14`	`"settings-prompt": "Prompt管理",`
`15`	`15`	`"settings-mcp-market": "MCP市场",`
`16`	`16`	`"settings-acp": "ACP Agent",`
`17`		`- "settings-skills": "skills设置",`
	`17`	`+ "settings-skills": "Skills设置",`
`18`	`18`	`"settings-notifications-hooks": "通知与Hooks"`
`19`	`19`	`}`