@opensumi/ide-ai-native 3.8.3-next-1742194253.0 → 3.8.3-next-1742203639.0

package/src/browser/chat/chat-manager.service.ts

@@ -150,7 +150,7 @@ export class ChatManagerService extends Disposable {
     this.saveSessions();
   }
 
-  createRequest(sessionId: string, message: string, agentId: string, command?: string) {
+  createRequest(sessionId: string, message: string, agentId: string, command?: string, images?: string[]) {
     const model = this.getSession(sessionId);
     if (!model) {
       throw new Error(`Unknown session: ${sessionId}`);
@@ -160,7 +160,7 @@ export class ChatManagerService extends Disposable {
       return;
     }
 
-    return model.addRequest({ prompt: message, agentId, command });
+    return model.addRequest({ prompt: message, agentId, command, images });
   }
 
   async sendRequest(sessionId: string, request: ChatRequestModel, regenerate: boolean) {
@@ -191,6 +191,7 @@ export class ChatManagerService extends Disposable {
         requestId: request.requestId,
         message: request.message.prompt,
         command: request.message.command,
+        images: request.message.images,
         regenerate,
       };
       const result = await this.chatAgentService.invokeAgent(

package/src/browser/chat/chat-model.ts

@@ -27,7 +27,7 @@ import {
 import { MsgHistoryManager } from '../model/msg-history-manager';
 import { IChatSlashCommandItem } from '../types';
 
-import type { TextPart, ToolCallPart } from 'ai';
+import type { ImagePart, TextPart, ToolCallPart } from 'ai';
 
 export type IChatProgressResponseContent =
   | IChatMarkdownContent
@@ -330,7 +330,15 @@ export class ChatModel extends Disposable implements IChatModel {
       if (!request.response.isComplete) {
         continue;
       }
-      history.push({ role: 'user', content: request.message.prompt });
+      history.push({
+        role: 'user',
+        content: request.message.images?.length
+          ? [
+              { type: 'text', text: request.message.prompt },
+              ...request.message.images.map((image) => ({ type: 'image', image: new URL(image) } as ImagePart)),
+            ]
+          : request.message.prompt,
+      });
       for (const part of request.response.responseParts) {
         if (part.kind === 'treeData' || part.kind === 'component') {
           continue;

package/src/browser/chat/chat-proxy.service.ts

@@ -151,6 +151,7 @@ export class ChatProxyService extends Disposable {
               requestId: request.requestId,
               sessionId: request.sessionId,
               history,
+              images: request.images,
               ...this.getRequestOptions(),
             },
             token,

package/src/browser/chat/chat.feature.registry.ts

@@ -2,7 +2,7 @@ import { Injectable } from '@opensumi/di';
 import { Disposable, Emitter, Event, getDebugLogger } from '@opensumi/ide-core-common';
 
 import { IChatWelcomeMessageContent, ISampleQuestions, SLASH_SYMBOL } from '../../common';
-import { IChatFeatureRegistry, IChatSlashCommandHandler, IChatSlashCommandItem } from '../types';
+import { IChatFeatureRegistry, IChatSlashCommandHandler, IChatSlashCommandItem, IImageUploadProvider } from '../types';
 
 import { ChatSlashCommandItemModel, ChatWelcomeMessageModel } from './chat-model';
 import { ChatProxyService } from './chat-proxy.service';
@@ -12,6 +12,15 @@ export class ChatFeatureRegistry extends Disposable implements IChatFeatureRegis
   private readonly logger = getDebugLogger();
   private slashCommandsMap: Map<string, ChatSlashCommandItemModel> = new Map();
   private slashCommandsHandlerMap: Map<string, IChatSlashCommandHandler> = new Map();
+  private imageUploadProvider: IImageUploadProvider | undefined;
+
+  public registerImageUploadProvider(provider: IImageUploadProvider): void {
+    this.imageUploadProvider = provider;
+  }
+
+  public getImageUploadProvider(): IImageUploadProvider | undefined {
+    return this.imageUploadProvider;
+  }
 
   public chatWelcomeMessageModel?: ChatWelcomeMessageModel;
 

package/src/browser/chat/chat.internal.service.ts

@@ -62,8 +62,8 @@ export class ChatInternalService extends Disposable {
     this._onChangeRequestId.fire(id);
   }
 
-  createRequest(input: string, agentId: string, command?: string) {
-    return this.chatManagerService.createRequest(this.#sessionModel.sessionId, input, agentId, command);
+  createRequest(input: string, agentId: string, images?: string[], command?: string) {
+    return this.chatManagerService.createRequest(this.#sessionModel.sessionId, input, agentId, command, images);
   }
 
   sendRequest(request: ChatRequestModel, regenerate = false) {

package/src/browser/chat/chat.view.tsx

@@ -281,7 +281,7 @@ export const AIChatView = () => {
           if (loading) {
             return;
           }
-          await handleSend(message.message, message.agentId, message.command);
+          await handleSend(message.message, message.images, message.agentId, message.command);
         } else {
           if (message.agentId) {
             setAgentId(message.agentId);
@@ -462,10 +462,16 @@ export const AIChatView = () => {
   );
 
   const renderUserMessage = React.useCallback(
-    async (renderModel: { message: string; agentId?: string; relationId: string; command?: string }) => {
+    async (renderModel: {
+      message: string;
+      images?: string[];
+      agentId?: string;
+      relationId: string;
+      command?: string;
+    }) => {
       const ChatUserRoleRender = chatRenderRegistry.chatUserRoleRender;
 
-      const { message, agentId, relationId, command } = renderModel;
+      const { message, images, agentId, relationId, command } = renderModel;
 
       const visibleAgentId = agentId === ChatProxyService.AGENT_ID ? '' : agentId;
 
@@ -474,12 +480,13 @@ export const AIChatView = () => {
           id: uuid(6),
           relationId,
           text: ChatUserRoleRender ? (
-            <ChatUserRoleRender content={message} agentId={visibleAgentId} command={command} />
+            <ChatUserRoleRender content={message} images={images} agentId={visibleAgentId} command={command} />
           ) : (
             <CodeBlockWrapperInput
               labelService={labelService}
               relationId={relationId}
               text={message}
+              images={images}
               agentId={visibleAgentId}
               command={command}
               workspaceService={workspaceService}
@@ -598,10 +605,10 @@ export const AIChatView = () => {
 
   const handleAgentReply = React.useCallback(
     async (value: IChatMessageStructure) => {
-      const { message, agentId, command, reportExtra } = value;
+      const { message, images, agentId, command, reportExtra } = value;
       const { actionType, actionSource } = reportExtra || {};
 
-      const request = aiChatService.createRequest(message, agentId!, command);
+      const request = aiChatService.createRequest(message, agentId!, images, command);
       if (!request) {
         return;
       }
@@ -627,6 +634,7 @@ export const AIChatView = () => {
 
       msgHistoryManager.addUserMessage({
         content: message,
+        images: images || [],
         agentId: agentId!,
         agentCommand: command!,
         relationId,
@@ -635,6 +643,7 @@ export const AIChatView = () => {
       await renderUserMessage({
         relationId,
         message,
+        images,
         command,
         agentId,
       });
@@ -668,7 +677,7 @@ export const AIChatView = () => {
   );
 
   const handleSend = React.useCallback(
-    async (message: string, agentId?: string, command?: string) => {
+    async (message: string, images?: string[], agentId?: string, command?: string) => {
       const reportExtra = {
         actionSource: ActionSourceEnum.Chat,
         actionType: ActionTypeEnum.Send,
@@ -707,7 +716,7 @@ export const AIChatView = () => {
           processedContent = processedContent.replace(match, `\`<attached_folder>${relativePath}\``);
         }
       }
-      return handleAgentReply({ message: processedContent, agentId, command, reportExtra });
+      return handleAgentReply({ message: processedContent, images, agentId, command, reportExtra });
     },
     [handleAgentReply],
   );
@@ -750,6 +759,7 @@ export const AIChatView = () => {
             message: msg.content,
             agentId: msg.agentId,
             command: msg.agentCommand,
+            images: msg.images,
           });
         } else if (msg.role === ChatMessageRole.Assistant && msg.requestId) {
           const request = aiChatService.sessionModel.getRequest(msg.requestId)!;

package/src/browser/components/ChatEditor.tsx

@@ -2,6 +2,7 @@ import capitalize from 'lodash/capitalize';
 import React, { useCallback, useEffect, useMemo, useState } from 'react';
 import Highlight from 'react-highlight';
 
+import { Image } from '@opensumi/ide-components/lib/image';
 import {
   EDITOR_COMMANDS,
   FILE_COMMANDS,
@@ -336,6 +337,7 @@ export const CodeBlockWrapper = ({
 
 export const CodeBlockWrapperInput = ({
   text,
+  images,
   relationId,
   agentId,
   command,
@@ -344,6 +346,7 @@ export const CodeBlockWrapperInput = ({
   commandService,
 }: {
   text: string;
+  images?: string[];
   relationId: string;
   agentId?: string;
   command?: string;
@@ -370,6 +373,11 @@ export const CodeBlockWrapperInput = ({
 
   return (
     <div className={styles.ai_chat_code_wrapper}>
+      {images?.map((image) => (
+        <div className={styles.image_wrapper}>
+          <Image src={image} />
+        </div>
+      ))}
       <div className={styles.render_text}>
         {tag && (
           <div className={styles.tag_wrapper}>

package/src/browser/components/ChatInput.tsx

@@ -162,7 +162,7 @@ const AgentWidget = ({ agentId, command }) => (
 );
 
 export interface IChatInputProps {
-  onSend: (value: string, agentId?: string, command?: string) => void;
+  onSend: (value: string, images?: string[], agentId?: string, command?: string) => void;
   onValueChange?: (value: string) => void;
   onExpand?: (value: boolean) => void;
   placeholder?: string;
@@ -341,7 +341,7 @@ export const ChatInput = React.forwardRef((props: IChatInputProps, ref) => {
     }
 
     const handleSendLogic = (newValue: string = value) => {
-      onSend(newValue, agentId, command);
+      onSend(newValue, [], agentId, command);
       setValue('');
       setTheme('');
       setAgentId('');

package/src/browser/components/ChatMentionInput.tsx

@@ -1,14 +1,18 @@
+import { DataContent } from 'ai';
 import React, { useCallback, useEffect, useMemo, useState } from 'react';
 
+import { Image } from '@opensumi/ide-components/lib/image';
 import { LabelService, RecentFilesManager, useInjectable } from '@opensumi/ide-core-browser';
-import { getIcon } from '@opensumi/ide-core-browser/lib/components';
-import { URI, localize } from '@opensumi/ide-core-common';
+import { Icon, getIcon } from '@opensumi/ide-core-browser/lib/components';
+import { ChatFeatureRegistryToken, URI, localize } from '@opensumi/ide-core-common';
 import { CommandService } from '@opensumi/ide-core-common/lib/command';
 import { WorkbenchEditorService } from '@opensumi/ide-editor';
 import { FileSearchServicePath, IFileSearchService } from '@opensumi/ide-file-search';
+import { IMessageService } from '@opensumi/ide-overlay';
 import { IWorkspaceService } from '@opensumi/ide-workspace';
 
 import { IChatInternalService } from '../../common';
+import { ChatFeatureRegistry } from '../chat/chat.feature.registry';
 import { ChatInternalService } from '../chat/chat.internal.service';
 import { OPEN_MCP_CONFIG_COMMAND } from '../mcp/config/mcp-config.commands';
 
@@ -17,7 +21,13 @@ import { MentionInput } from './mention-input/mention-input';
 import { FooterButtonPosition, FooterConfig, MentionItem, MentionType } from './mention-input/types';
 
 export interface IChatMentionInputProps {
-  onSend: (value: string, agentId?: string, command?: string, option?: { model: string; [key: string]: any }) => void;
+  onSend: (
+    value: string,
+    images?: string[],
+    agentId?: string,
+    command?: string,
+    option?: { model: string; [key: string]: any },
+  ) => void;
   onValueChange?: (value: string) => void;
   onExpand?: (value: boolean) => void;
   placeholder?: string;
@@ -26,6 +36,7 @@ export interface IChatMentionInputProps {
   sendBtnClassName?: string;
   defaultHeight?: number;
   value?: string;
+  images?: Array<DataContent | URL>;
   autoFocus?: boolean;
   theme?: string | null;
   setTheme: (theme: string | null) => void;
@@ -41,6 +52,7 @@ export const ChatMentionInput = (props: IChatMentionInputProps) => {
   const { onSend, disabled = false } = props;
 
   const [value, setValue] = useState(props.value || '');
+  const [images, setImages] = useState(props.images || []);
   const aiChatService = useInjectable<ChatInternalService>(IChatInternalService);
   const commandService = useInjectable<CommandService>(CommandService);
   const searchService = useInjectable<IFileSearchService>(FileSearchServicePath);
@@ -48,7 +60,8 @@ export const ChatMentionInput = (props: IChatMentionInputProps) => {
   const workspaceService = useInjectable<IWorkspaceService>(IWorkspaceService);
   const editorService = useInjectable<WorkbenchEditorService>(WorkbenchEditorService);
   const labelService = useInjectable<LabelService>(LabelService);
-
+  const messageService = useInjectable<IMessageService>(IMessageService);
+  const chatFeatureRegistry = useInjectable<ChatFeatureRegistry>(ChatFeatureRegistryToken);
   const handleShowMCPConfig = React.useCallback(() => {
     commandService.executeCommand(OPEN_MCP_CONFIG_COMMAND.id);
   }, [commandService]);
@@ -239,6 +252,24 @@ export const ChatMentionInput = (props: IChatMentionInputProps) => {
           onClick: handleShowMCPConfig,
           position: FooterButtonPosition.LEFT,
         },
+        {
+          id: 'upload-image',
+          icon: 'image',
+          title: localize('aiNative.chat.imageUpload'),
+          onClick: () => {
+            const input = document.createElement('input');
+            input.type = 'file';
+            input.accept = 'image/*';
+            input.onchange = (e) => {
+              const files = (e.target as HTMLInputElement).files;
+              if (files?.length) {
+                handleImageUpload(Array.from(files));
+              }
+            };
+            input.click();
+          },
+          position: FooterButtonPosition.LEFT,
+        },
       ],
       showModelSelector: true,
     }),
@@ -254,13 +285,54 @@ export const ChatMentionInput = (props: IChatMentionInputProps) => {
       if (disabled) {
         return;
       }
-      onSend(content, undefined, undefined, option);
+      onSend(
+        content,
+        images.map((image) => image.toString()),
+        undefined,
+        undefined,
+        option,
+      );
+      setImages(props.images || []);
     },
-    [onSend, editorService, disabled],
+    [onSend, images, disabled],
+  );
+
+  const handleImageUpload = useCallback(
+    async (files: File[]) => {
+      const allowedTypes = ['image/jpeg', 'image/jpg', 'image/png', 'image/webp', 'image/gif'];
+
+      // Validate file types
+      const invalidFiles = files.filter((file) => !allowedTypes.includes(file.type));
+      if (invalidFiles.length > 0) {
+        messageService.error('Only JPG, PNG, WebP and GIF images are supported');
+        return;
+      }
+
+      const imageUploadProvider = chatFeatureRegistry.getImageUploadProvider();
+      if (!imageUploadProvider) {
+        messageService.error('No image upload provider found');
+        return;
+      }
+
+      // Upload all files
+      const uploadedData = await Promise.all(files.map((file) => imageUploadProvider.imageUpload(file)));
+
+      const newImages = [...images, ...uploadedData];
+      setImages(newImages);
+    },
+    [images],
+  );
+
+  const handleDeleteImage = useCallback(
+    (index: number) => {
+      setImages(images.filter((_, i) => i !== index));
+    },
+    [images],
   );
 
   return (
     <div className={styles.chat_input_container}>
+      {images.length > 0 && <ImagePreviewer images={images} onDelete={handleDeleteImage} />}
       <MentionInput
         mentionItems={defaultMenuItems}
         onSend={handleSend}
@@ -270,7 +342,29 @@ export const ChatMentionInput = (props: IChatMentionInputProps) => {
         workspaceService={workspaceService}
         placeholder={localize('aiNative.chat.input.placeholder.default')}
         footerConfig={defaultMentionInputFooterOptions}
+        onImageUpload={handleImageUpload}
       />
     </div>
   );
 };
+
+const ImagePreviewer = ({
+  images,
+  onDelete,
+}: {
+  images: Array<DataContent | URL>;
+  onDelete: (index: number) => void;
+}) => (
+  <div>
+    <div className={styles.thumbnail_container}>
+      {images.map((image, index) => (
+        <div key={index} className={styles.thumbnail}>
+          <Image src={image.toString()} />
+          <button onClick={() => onDelete(index)} className={styles.delete_button}>
+            <Icon iconClass='codicon codicon-close' />
+          </button>
+        </div>
+      ))}
+    </div>
+  </div>
+);

package/src/browser/components/components.module.less

@@ -601,3 +601,52 @@
   vertical-align: middle;
   font-size: 12px;
 }
+
+.thumbnail_container {
+  display: flex;
+  gap: 4px;
+  padding: 4px 12px;
+  .thumbnail {
+    width: 36px;
+    height: 36px;
+    border-radius: 4px;
+    position: relative;
+    box-shadow: 0 0 10px 0 rgba(0, 0, 0, 0.3);
+    padding: 2px;
+    display: inline-flex;
+    align-items: center;
+    img {
+      width: 100%;
+      height: 100%;
+      object-fit: cover;
+    }
+  }
+  .delete_button {
+    position: absolute;
+    top: -5px;
+    right: -5px;
+    font-size: 12px;
+    padding: 2px;
+    border: 0;
+    border-radius: 50%;
+    transition: transform 0.2s ease-in-out;
+    width: 16px;
+    height: 16px;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    background-color: var(--badge-background);
+    &:hover {
+      cursor: pointer;
+      transform: scale(1.2);
+    }
+    :global(.codicon) {
+      font-size: 12px;
+      color: var(--badge-foreground);
+    }
+  }
+}
+
+.image_wrapper {
+  display: inline-block;
+}

package/src/browser/components/mention-input/mention-input.tsx

@@ -18,6 +18,7 @@ export const MentionInput: React.FC<MentionInputProps> = ({
   loading = false,
   mentionKeyword = MENTION_KEYWORD,
   onSelectionChange,
+  onImageUpload,
   labelService,
   workspaceService,
   placeholder = 'Ask anything, @ to mention',
@@ -447,20 +448,35 @@ export const MentionInput: React.FC<MentionInputProps> = ({
   };
 
   // 添加粘贴事件处理
-  const handlePaste = (e: React.ClipboardEvent<HTMLDivElement>) => {
-    // 阻止默认粘贴行为
+  const handlePaste = async (e: React.ClipboardEvent<HTMLDivElement>) => {
+    const items = e.clipboardData.items;
+
+    // 先收集所有图片文件
+    const imageFiles: File[] = [];
+    // eslint-disable-next-line @typescript-eslint/prefer-for-of
+    for (let i = 0; i < items.length; i++) {
+      if (items[i].kind === 'file' && items[i].type.startsWith('image/')) {
+        const file = items[i].getAsFile();
+        if (file) {
+          imageFiles.push(file);
+        }
+      }
+    }
+
     e.preventDefault();
 
-    // 获取剪贴板中的纯文本内容
+    // 处理所有收集到的图片
+    if (imageFiles.length > 0 && onImageUpload) {
+      await onImageUpload(imageFiles);
+      return;
+    }
+
     const text = e.clipboardData.getData('text/plain');
 
     // 处理文本，保留换行和缩进
     const processedText = text
-      // 将制表符转换为4个空格
       .replace(/\t/g, '    ')
-      // 将连续的换行符转换为单个换行
       .replace(/\n\s*\n/g, '\n\n')
-      // 移除行尾空格
       .replace(/[ \t]+$/gm, '');
 
     const selection = window.getSelection();
@@ -931,7 +947,7 @@ export const MentionInput: React.FC<MentionInputProps> = ({
             title={button.title}
           >
             <EnhanceIcon
-              className={cls(getIcon(button.icon), styles[`${button.id}_logo`])}
+              className={cls(button.icon ? getIcon(button.icon) : button.iconClass, styles[`${button.id}_logo`])}
               tabIndex={0}
               role='button'
               ariaLabel={button.title}
@@ -965,8 +981,8 @@ export const MentionInput: React.FC<MentionInputProps> = ({
           contentEditable={true}
           onInput={handleInput}
           onKeyDown={handleKeyDown}
-          onCompositionEnd={handleCompositionEnd}
           onPaste={handlePaste}
+          onCompositionEnd={handleCompositionEnd}
         />
       </div>
       <div className={styles.footer}>

package/src/browser/components/mention-input/types.ts

@@ -55,7 +55,8 @@ export enum MentionType {
 
 interface FooterButton {
   id: string;
-  icon: string;
+  icon?: string;
+  iconClass?: string;
   title: string;
   onClick?: () => void;
   position: FooterButtonPosition;
@@ -75,6 +76,7 @@ export interface MentionInputProps {
   placeholder?: string;
   loading?: boolean;
   onSelectionChange?: (value: string) => void;
+  onImageUpload?: (files: File[]) => Promise<void>;
   footerConfig?: FooterConfig; // 新增配置项
   mentionKeyword?: string;
   labelService?: LabelService;

package/src/browser/context/llm-context.service.ts

@@ -1,3 +1,5 @@
+import { DataContent } from 'ai';
+
 import { Autowired, Injectable } from '@opensumi/di';
 import { AppConfig } from '@opensumi/ide-core-browser/lib/react-providers/config-provider';
 import { WithEventBus } from '@opensumi/ide-core-common/lib/event-bus/event-decorator';

package/src/browser/mcp/config/components/mcp-config.view.tsx

@@ -137,7 +137,7 @@ export const MCPConfigView: React.FC = () => {
     async (serverName: string) => {
       const servers = preferenceService.get<MCPServerFormData[]>(AINativeSettingSectionsId.MCPServers, []);
       const updatedServers = servers.filter((s) => s.name !== serverName);
-      sumiMCPServerBackendProxy.removeServer(serverName);
+      sumiMCPServerBackendProxy.$removeServer(serverName);
       await preferenceService.set(AINativeSettingSectionsId.MCPServers, updatedServers, PreferenceScope.User);
       await loadServers();
     },
@@ -156,7 +156,7 @@ export const MCPConfigView: React.FC = () => {
       }
       setServers(servers as MCPServer[]);
       setFormVisible(false);
-      await sumiMCPServerBackendProxy.addOrUpdateServer(data as MCPServerDescription);
+      await sumiMCPServerBackendProxy.$addOrUpdateServer(data as MCPServerDescription);
       await preferenceService.set(AINativeSettingSectionsId.MCPServers, servers, PreferenceScope.User);
       await loadServers();
     },

package/src/browser/mcp/config/components/mcp-server-form.tsx

@@ -167,7 +167,14 @@ export const MCPServerForm: FC<Props> = ({ visible, initialData, onSave, onCance
 
   const handleTypeChange = useCallback(
     (value: MCP_SERVER_TYPE) => {
-      setFormData({ ...formData, type: value });
+      setFormData({
+        ...formData,
+        type: value,
+        command: '',
+        args: [],
+        env: {},
+        serverHost: '',
+      });
     },
     [formData],
   );

package/src/browser/mcp/config/mcp-config.contribution.ts

@@ -1,6 +1,7 @@
 import { Autowired } from '@opensumi/di';
+import { getIcon } from '@opensumi/ide-components';
 import { LabelService } from '@opensumi/ide-core-browser/lib/services';
-import { Domain, URI } from '@opensumi/ide-core-common';
+import { Domain, URI, localize } from '@opensumi/ide-core-common';
 import {
   BrowserEditorContribution,
   EditorComponentRegistry,
@@ -53,8 +54,8 @@ export class MCPConfigContribution implements BrowserEditorContribution {
 
         return {
           uri,
-          name: 'MCP Configuration',
-          icon: 'settings',
+          name: localize('ai.native.mcp.config.title'),
+          icon: getIcon('mcp'),
           metadata: {
             configType,
           },

package/src/browser/mcp/mcp-server-proxy.service.ts

@@ -50,18 +50,18 @@ export class MCPServerProxyService implements IMCPServerProxyService {
   }
 
   async getAllMCPTools() {
-    return this.sumiMCPServerProxyService.getAllMCPTools();
+    return this.sumiMCPServerProxyService.$getAllMCPTools();
   }
 
   async $getServers() {
-    return this.sumiMCPServerProxyService.getServers();
+    return this.sumiMCPServerProxyService.$getServers();
   }
 
   async $startServer(serverName: string) {
-    await this.sumiMCPServerProxyService.startServer(serverName);
+    await this.sumiMCPServerProxyService.$startServer(serverName);
   }
 
   async $stopServer(serverName: string) {
-    await this.sumiMCPServerProxyService.stopServer(serverName);
+    await this.sumiMCPServerProxyService.$stopServer(serverName);
   }
 }

package/src/browser/model/msg-history-manager.ts

@@ -62,7 +62,7 @@ export class MsgHistoryManager extends Disposable {
   }
 
   public addUserMessage(
-    message: Required<Pick<IExcludeMessage, 'agentId' | 'agentCommand' | 'content' | 'relationId'>>,
+    message: Required<Pick<IExcludeMessage, 'agentId' | 'agentCommand' | 'content' | 'relationId' | 'images'>>,
   ): string {
     return this.doAddMessage({
       ...message,