@opensumi/ide-ai-native 3.9.1-next-1749538805.0 → 3.9.1-next-1749540423.0

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

@@ -28,7 +28,7 @@ import {
 import { MsgHistoryManager } from '../model/msg-history-manager';
 import { IChatSlashCommandItem } from '../types';
 
-import { ChatFeatureRegistry } from './chat.feature.registry';
+import type { ImagePart, TextPart, ToolCallPart } from 'ai';
 
 export type IChatProgressResponseContent =
   | IChatMarkdownContent
@@ -296,13 +296,10 @@ export class ChatRequestModel implements IChatRequestModel {
 export class ChatModel extends Disposable implements IChatModel {
   private requestIdPool = 0;
 
-  constructor(
-    private chatFeatureRegistry: ChatFeatureRegistry,
-    initParams?: { sessionId?: string; history?: MsgHistoryManager; modelId?: string },
-  ) {
+  constructor(initParams?: { sessionId?: string; history?: MsgHistoryManager; modelId?: string }) {
     super();
     this.#sessionId = initParams?.sessionId ?? uuid();
-    this.history = initParams?.history ?? new MsgHistoryManager(this.chatFeatureRegistry);
+    this.history = initParams?.history ?? new MsgHistoryManager();
     this.#modelId = initParams?.modelId;
   }
 
@@ -339,142 +336,88 @@ export class ChatModel extends Disposable implements IChatModel {
     this.#modelId = modelId;
   }
 
-  private processMemorySummaries(): CoreMessage[] {
-    const memorySummaries = this.history.getMemorySummaries();
-    if (memorySummaries.length === 0) {
-      return [];
-    }
-
-    const processedSummaries = memorySummaries
-      .map((summary) => {
-        try {
-          const parsed = JSON.parse(summary.content);
-          return parsed.memory || parsed.content || summary.content;
-        } catch {
-          return summary.content;
-        }
-      })
-      .filter((content) => content && content !== 'no_memory_needed')
-      .filter((content, index, self) => self.indexOf(content) === index);
-
-    return processedSummaries.length > 0
-      ? [
-        {
-          role: 'system',
-          content: '以下是之前对话的总结：\n' + processedSummaries.join('\n\n'),
-        },
-      ]
-      : [];
-  }
-
-  private processToolCall(part: IChatToolContent, history: CoreMessage[]): void {
-    if (history[history.length - 1].role !== 'assistant') {
-      history.push({
-        role: 'assistant',
-        content: [],
-      });
-    }
-
-    const toolCallId = part.content.id;
-    const toolCallInfo = {
-      id: toolCallId,
-      name: part.content.function.name,
-      args: this.parseJsonSafely(part.content.function.arguments || '{}', 'tool call arguments'),
-      result: this.parseJsonSafely(part.content.result || '{}', 'tool result'),
-    };
-
-    this.history.addToolCall(toolCallInfo);
-    this.history.setMessageAdditional(part.content.id, { toolCallId });
-
-    const lastMessage = history[history.length - 1];
-    lastMessage.content = [
-      {
-        type: 'tool-call',
-        toolCallId: part.content.id,
-        toolName: part.content.function.name,
-        args: toolCallInfo.args,
-      },
-    ];
-
-    history.push({
-      role: 'tool',
-      content: [
-        {
-          type: 'tool-result',
-          toolCallId: part.content.id,
-          toolName: part.content.function.name,
-          result: toolCallInfo.result,
-        },
-      ],
-    });
-  }
-
-  private parseJsonSafely(jsonString: string, context: string): any {
-    try {
-      return JSON.parse(jsonString);
-    } catch (e) {
-      console.error(`[ChatModel] Failed to parse ${context}:`, e);
-      return {};
-    }
-  }
-
-  private processRecentMessages(): CoreMessage[] {
+  getMessageHistory(contextWindow?: number) {
     const history: CoreMessage[] = [];
-    // 从 MsgHistoryManager 获取消息，并过滤掉已总结的消息
-    const recentMessages = this.history
-      .getMessages()
-      .filter((msg) => !msg.isSummarized)
-      .slice(-this.history.memoryConfig.shortTermSize);
-
-    for (const message of recentMessages) {
+    for (const request of this.requests) {
+      if (!request.response.isComplete) {
+        continue;
+      }
       history.push({
         role: 'user',
-        content: message.content,
+        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,
       });
-
-      // 如果有响应内容，添加响应
-      const additional = this.history.getMessageAdditional(message.id);
-      if (additional?.response) {
-        const response = additional.response;
-        if (response.kind === 'toolCall') {
-          this.processToolCall(response, history);
-        } else {
+      for (const part of request.response.responseParts) {
+        if (part.kind === 'treeData' || part.kind === 'component') {
+          continue;
+        }
+        if (part.kind !== 'toolCall') {
           history.push({
             role: 'assistant',
-            content: response.kind === 'markdownContent' ? response.content.value : response.content,
+            content: [
+              {
+                type: 'text',
+                text: part.kind === 'markdownContent' ? part.content.value : part.content,
+              },
+            ],
+          });
+        } else {
+          // 直接开始toolCall场景
+          if (history[history.length - 1].role !== 'assistant') {
+            history.push({
+              role: 'assistant',
+              content: [],
+            });
+          }
+          (history[history.length - 1].content as Array<TextPart | ToolCallPart>).push({
+            type: 'tool-call',
+            toolCallId: part.content.id,
+            toolName: part.content.function.name,
+            args: (() => {
+              try {
+                return JSON.parse(part.content.function.arguments || '{}');
+              } catch (e) {
+                console.error('Failed to parse tool call arguments:', e);
+                return {};
+              }
+            })(),
+          });
+          history.push({
+            role: 'tool',
+            content: [
+              {
+                type: 'tool-result',
+                toolCallId: part.content.id,
+                toolName: part.content.function.name,
+                result: (() => {
+                  try {
+                    return JSON.parse(part.content.result || '{}');
+                  } catch (e) {
+                    console.error('Failed to parse tool result:', e);
+                    return {};
+                  }
+                })(),
+              },
+            ],
           });
         }
       }
     }
-
-    return history;
-  }
-
-  private limitTokens(history: CoreMessage[], contextWindow?: number): CoreMessage[] {
-    if (!contextWindow) {
-      return history;
-    }
-
-    let currentHistory = history;
-    let tokenCount = JSON.stringify(currentHistory).length / 3;
-
-    while (tokenCount > contextWindow && currentHistory.length > 1) {
-      if (currentHistory[0].role === 'system' && currentHistory.length > 2) {
-        currentHistory = [currentHistory[0], ...currentHistory.slice(2)];
-      } else {
-        currentHistory = currentHistory.slice(1);
+    if (contextWindow) {
+      while (this.#slicedMessageCount < history.length) {
+        // 简单的使用 JSON.stringify 计算 token 数量
+        const tokenCount = JSON.stringify(history.slice(this.#slicedMessageCount)).length / 3;
+        if (tokenCount <= contextWindow) {
+          break;
+        }
+        this.#slicedMessageCount++;
       }
-      tokenCount = JSON.stringify(currentHistory).length / 3;
     }
-
-    return currentHistory;
-  }
-
-  getMessageHistory(contextWindow?: number): CoreMessage[] {
-    const memorySummaries = this.processMemorySummaries();
-    const recentMessages = this.processRecentMessages();
-    const history = [...memorySummaries, ...recentMessages];
-    return this.limitTokens(history, contextWindow);
+    return history.slice(this.#slicedMessageCount);
   }
 
   addRequest(message: IChatRequestMessage): ChatRequestModel {

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

@@ -1,16 +1,18 @@
 import { Autowired, Injectable } from '@opensumi/di';
-import { PreferenceService } from '@opensumi/ide-core-browser';
+import { AppConfig, PreferenceService } from '@opensumi/ide-core-browser';
 import {
   AIBackSerivcePath,
   CancellationToken,
   ChatAgentViewServiceToken,
   ChatFeatureRegistryToken,
+  ChatServiceToken,
   Deferred,
   Disposable,
   IAIBackService,
   IAIReporter,
   IApplicationService,
   IChatProgress,
+  getOperatingSystemName,
 } from '@opensumi/ide-core-common';
 import { AINativeSettingSectionsId } from '@opensumi/ide-core-common/lib/settings/ai-native';
 import { MonacoCommandRegistry } from '@opensumi/ide-editor/lib/browser/monaco-contrib/command/command.service';
@@ -25,10 +27,10 @@ import {
   IChatAgentService,
   IChatAgentWelcomeMessage,
 } from '../../common';
-import { DEFAULT_SYSTEM_PROMPT } from '../../common/prompts/system-prompt';
 import { ChatToolRender } from '../components/ChatToolRender';
 import { IChatAgentViewService } from '../types';
 
+import { ChatService } from './chat.api.service';
 import { ChatFeatureRegistry } from './chat.feature.registry';
 
 /**
@@ -51,6 +53,9 @@ export class ChatProxyService extends Disposable {
   @Autowired(MonacoCommandRegistry)
   private readonly monacoCommandRegistry: MonacoCommandRegistry;
 
+  @Autowired(ChatServiceToken)
+  private aiChatService: ChatService;
+
   @Autowired(IAIReporter)
   private readonly aiReporter: IAIReporter;
 
@@ -66,6 +71,9 @@ export class ChatProxyService extends Disposable {
   @Autowired(IMessageService)
   private readonly messageService: IMessageService;
 
+  @Autowired(AppConfig)
+  private readonly appConfig: AppConfig;
+
   private chatDeferred: Deferred<void> = new Deferred<void>();
 
   public getRequestOptions() {
@@ -104,74 +112,79 @@ export class ChatProxyService extends Disposable {
       initialProps: {},
     });
 
-    this.applicationService.getBackendOS().then(() => {
-      this.addDispose(
-        this.chatAgentService.registerAgent({
-          id: ChatProxyService.AGENT_ID,
-          metadata: {
-            systemPrompt: this.preferenceService.get<string>(
+    this.addDispose(
+      this.chatAgentService.registerAgent({
+        id: ChatProxyService.AGENT_ID,
+        metadata: {
+          systemPrompt:
+            this.preferenceService.get<string>(
               AINativeSettingSectionsId.SystemPrompt,
-              DEFAULT_SYSTEM_PROMPT,
-            ),
-          },
-          invoke: async (
-            request: IChatAgentRequest,
-            progress: (part: IChatProgress) => void,
-            history: CoreMessage[],
-            token: CancellationToken,
-          ): Promise<IChatAgentResult> => {
-            this.chatDeferred = new Deferred<void>();
-            const { message, command } = request;
-            let prompt: string = message;
-            if (command) {
-              const commandHandler = this.chatFeatureRegistry.getSlashCommandHandler(command);
-              if (commandHandler && commandHandler.providerPrompt) {
-                const editor = this.monacoCommandRegistry.getActiveCodeEditor();
-                const slashCommandPrompt = await commandHandler.providerPrompt(message, editor);
-                prompt = slashCommandPrompt;
-              }
+              'You are a powerful AI coding assistant working in OpenSumi, a top IDE framework. You collaborate with a USER to solve coding tasks, which may involve creating, modifying, or debugging code, or answering questions. When the USER sends a message, relevant context (e.g., open files, cursor position, edit history, linter errors) may be attached. Use this information as needed.\n\n<tool_calling>\nYou have access to tools to assist with tasks. Follow these rules:\n1. Always adhere to the tool call schema and provide all required parameters.\n2. Only use tools explicitly provided; ignore unavailable ones.\n3. Avoid mentioning tool names to the USER (e.g., say "I will edit your file" instead of "I need to use the edit_file tool").\n4. Only call tools when necessary; respond directly if the task is general or you already know the answer.\n5. Explain to the USER why you’re using a tool before calling it.\n</tool_calling>\n\n<making_code_changes>\nWhen modifying code:\n1. Use code edit tools instead of outputting code unless explicitly requested.\n2. Limit tool calls to one per turn.\n3. Ensure generated code is immediately executable by including necessary imports, dependencies, and endpoints.\n4. For new projects, create a dependency management file (e.g., requirements.txt) and a README.\n5. For web apps, design a modern, user-friendly UI.\n6. Avoid generating non-textual or excessively long code.\n7. Read file contents before editing, unless appending a small change or creating a new file.\n8. Fix introduced linter errors if possible, but stop after 3 attempts and ask the USER for guidance.\n9. Reapply reasonable code edits if they weren’t followed initially.\n</making_code_changes>\n\nUse the appropriate tools to fulfill the USER’s request, ensuring all required parameters are provided or inferred from context.',
+            ) +
+            `\n\n<user_info>\nThe user's OS is ${getOperatingSystemName()}. The absolute path of the user's workspace is ${
+              this.appConfig.workspaceDir
+            }.\n</user_info>`,
+        },
+        invoke: async (
+          request: IChatAgentRequest,
+          progress: (part: IChatProgress) => void,
+          history: CoreMessage[],
+          token: CancellationToken,
+        ): Promise<IChatAgentResult> => {
+          this.chatDeferred = new Deferred<void>();
+
+          const { message, command } = request;
+          let prompt: string = message;
+
+          if (command) {
+            const commandHandler = this.chatFeatureRegistry.getSlashCommandHandler(command);
+            if (commandHandler && commandHandler.providerPrompt) {
+              const editor = this.monacoCommandRegistry.getActiveCodeEditor();
+              const slashCommandPrompt = await commandHandler.providerPrompt(message, editor);
+              prompt = slashCommandPrompt;
             }
-
-            const stream = await this.aiBackService.requestStream(
-              prompt,
-              {
-                requestId: request.requestId,
-                sessionId: request.sessionId,
-                history,
-                images: request.images,
-                ...this.getRequestOptions(),
-              },
-              token,
-            );
-
-            listenReadable<IChatProgress>(stream, {
-              onData: (data) => {
-                progress(data);
-              },
-              onEnd: () => {
-                this.chatDeferred.resolve();
-              },
-              onError: (error) => {
-                this.messageService.error(error.message);
-                this.aiReporter.end(request.sessionId + '_' + request.requestId, {
-                  message: error.message,
-                  success: false,
-                  command,
-                });
-              },
-            });
-
-            await this.chatDeferred.promise;
-            return {};
-          },
-          provideSlashCommands: async (): Promise<IChatAgentCommand[]> =>
-            this.chatFeatureRegistry
-              .getAllSlashCommand()
-              .map((s) => ({ ...s, name: s.name, description: s.description || '' })),
-          provideChatWelcomeMessage: async (): Promise<IChatAgentWelcomeMessage | undefined> => undefined,
-        }),
-      );
-    });
+          }
+
+          const stream = await this.aiBackService.requestStream(
+            prompt,
+            {
+              requestId: request.requestId,
+              sessionId: request.sessionId,
+              history,
+              images: request.images,
+              ...this.getRequestOptions(),
+            },
+            token,
+          );
+
+          listenReadable<IChatProgress>(stream, {
+            onData: (data) => {
+              progress(data);
+            },
+            onEnd: () => {
+              this.chatDeferred.resolve();
+            },
+            onError: (error) => {
+              this.messageService.error(error.message);
+              this.aiReporter.end(request.sessionId + '_' + request.requestId, {
+                message: error.message,
+                success: false,
+                command,
+              });
+            },
+          });
+
+          await this.chatDeferred.promise;
+          return {};
+        },
+        provideSlashCommands: async (token: CancellationToken): Promise<IChatAgentCommand[]> =>
+          this.chatFeatureRegistry
+            .getAllSlashCommand()
+            .map((s) => ({ ...s, name: s.name, description: s.description || '' })),
+        provideChatWelcomeMessage: async (token: CancellationToken): Promise<IChatAgentWelcomeMessage | undefined> =>
+          undefined,
+      }),
+    );
 
     queueMicrotask(() => {
       this.chatAgentService.updateAgent(ChatProxyService.AGENT_ID, {});

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

@@ -2,13 +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,
-  IImageUploadProvider,
-  IMessageSummaryProvider,
-} from '../types';
+import { IChatFeatureRegistry, IChatSlashCommandHandler, IChatSlashCommandItem, IImageUploadProvider } from '../types';
 
 import { ChatSlashCommandItemModel, ChatWelcomeMessageModel } from './chat-model';
 import { ChatProxyService } from './chat-proxy.service';
@@ -20,8 +14,6 @@ export class ChatFeatureRegistry extends Disposable implements IChatFeatureRegis
   private slashCommandsHandlerMap: Map<string, IChatSlashCommandHandler> = new Map();
   private imageUploadProvider: IImageUploadProvider | undefined;
 
-  private messageSummaryProvider?: IMessageSummaryProvider;
-
   public registerImageUploadProvider(provider: IImageUploadProvider): void {
     this.imageUploadProvider = provider;
   }
@@ -30,14 +22,6 @@ export class ChatFeatureRegistry extends Disposable implements IChatFeatureRegis
     return this.imageUploadProvider;
   }
 
-  public registerMessageSummaryProvider(provider: IMessageSummaryProvider): void {
-    this.messageSummaryProvider = provider;
-  }
-
-  public getMessageSummaryProvider(): IMessageSummaryProvider | undefined {
-    return this.messageSummaryProvider;
-  }
-
   public chatWelcomeMessageModel?: ChatWelcomeMessageModel;
 
   private _onDidWelcomeMessageChange: Emitter<void> = new Emitter<void>();

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

@@ -1,4 +1,3 @@
-import debounce from 'lodash/debounce';
 import * as React from 'react';
 import { MessageList } from 'react-chat-elements';
 
@@ -618,12 +617,7 @@ export const AIChatView = () => {
       const { message, images, agentId, command, reportExtra } = value;
       const { actionType, actionSource } = reportExtra || {};
 
-      const request = aiChatService.createRequest(
-        message.replaceAll(LLM_CONTEXT_KEY_REGEX, ''),
-        agentId!,
-        images,
-        command,
-      );
+      const request = aiChatService.createRequest(message, agentId!, images, command);
       if (!request) {
         return;
       }
@@ -647,7 +641,7 @@ export const AIChatView = () => {
         600 * 1000,
       );
       msgHistoryManager.addUserMessage({
-        content: message,
+        content: message.replaceAll(LLM_CONTEXT_KEY_REGEX, ''),
         images: images || [],
         agentId: agentId!,
         agentCommand: command!,
@@ -701,9 +695,14 @@ export const AIChatView = () => {
       let processedContent = message;
       const filePattern = /\{\{@file:(.*?)\}\}/g;
       const fileMatches = message.match(filePattern);
+      let isCleanContext = false;
       if (fileMatches) {
         for (const match of fileMatches) {
           const filePath = match.replace(/\{\{@file:(.*?)\}\}/, '$1');
+          if (filePath && !isCleanContext) {
+            isCleanContext = true;
+            llmContextService.cleanFileContext();
+          }
           const fileUri = new URI(filePath);
           const relativePath = (await workspaceService.asRelativePath(fileUri))?.path || fileUri.displayName;
           processedContent = processedContent.replace(match, `\`${LLM_CONTEXT_KEY.AttachedFile}${relativePath}\``);
@@ -739,18 +738,6 @@ export const AIChatView = () => {
           );
         }
       }
-      const rulePattern = /\{\{@rule:(.*?)\}\}/g;
-      const ruleMatches = processedContent.match(rulePattern);
-      if (ruleMatches) {
-        for (const match of ruleMatches) {
-          const ruleName = match.replace(/\{\{@rule:(.*?)\}\}/, '$1');
-          const ruleUri = new URI(ruleName);
-          processedContent = processedContent.replace(
-            match,
-            `\`${LLM_CONTEXT_KEY.AttachedFile}${ruleUri.displayName}\``,
-          );
-        }
-      }
       return handleAgentReply({ message: processedContent, images, agentId, command, reportExtra });
     },
     [handleAgentReply],
@@ -931,7 +918,6 @@ export function DefaultChatViewHeader({
 }) {
   const aiChatService = useInjectable<ChatInternalService>(IChatInternalService);
   const messageService = useInjectable<IMessageService>(IMessageService);
-  const chatFeatureRegistry = useInjectable<ChatFeatureRegistry>(ChatFeatureRegistryToken);
 
   const [historyList, setHistoryList] = React.useState<IChatHistoryItem[]>([]);
   const [currentTitle, setCurrentTitle] = React.useState<string>('');
@@ -957,60 +943,13 @@ export function DefaultChatViewHeader({
     [aiChatService],
   );
 
-  // 防抖函数，避免频繁触发摘要生成
-  const debouncedGetSummary = React.useCallback(
-    debounce(
-      async (messages: { role: ChatMessageRole; content: string }[], currentTitle: string): Promise<string> => {
-        const summaryProvider = chatFeatureRegistry.getMessageSummaryProvider();
-        if (!summaryProvider || !aiChatService.sessionModel.sessionId) {
-          return currentTitle;
-        }
-
-        try {
-          const summary = await summaryProvider.getMessageSummary(messages);
-          return summary ? summary.slice(0, MAX_TITLE_LENGTH) : currentTitle;
-        } catch (error) {
-          return currentTitle;
-        }
-      },
-      1000,
-      { leading: false, trailing: true },
-    ),
-    [chatFeatureRegistry, aiChatService.sessionModel.sessionId],
-  );
-
-  // 使用 ref 来跟踪最新的请求
-  const latestSummaryRequestRef = React.useRef<number>(0);
-
   React.useEffect(() => {
-    const getHistoryList = async () => {
+    const getHistoryList = () => {
       const currentMessages = aiChatService.sessionModel.history.getMessages();
-      const latestUserMessage = [...currentMessages].find((m) => m.role === ChatMessageRole.User);
-      const currentTitle = latestUserMessage
-        ? cleanAttachedTextWrapper(latestUserMessage.content).slice(0, MAX_TITLE_LENGTH)
-        : '';
-
-      // 设置初始标题
-      setCurrentTitle(currentTitle);
-
-      const messages = currentMessages.map((msg) => ({
-        role: msg.role,
-        content: msg.content,
-      }));
-
-      // 只有当消息数量超过阈值时才生成摘要
-      if (messages.length > 2) {
-        const requestId = Date.now();
-        latestSummaryRequestRef.current = requestId;
-
-        const summary = await debouncedGetSummary(messages, currentTitle);
-
-        // 检查是否是最新请求
-        if (requestId === latestSummaryRequestRef.current && summary) {
-          setCurrentTitle(summary);
-        }
-      }
-
+      const latestUserMessage = currentMessages.findLast((m) => m.role === ChatMessageRole.User);
+      setCurrentTitle(
+        latestUserMessage ? cleanAttachedTextWrapper(latestUserMessage.content).slice(0, MAX_TITLE_LENGTH) : '',
+      );
       setHistoryList(
         aiChatService.getSessions().map((session) => {
           const history = session.history;

package/src/browser/components/ChatEditor.tsx

@@ -25,7 +25,7 @@ import {
   localize,
   runWhenIdle,
 } from '@opensumi/ide-core-common';
-import { insertSnippetWithMonacoEditor } from '@opensumi/ide-editor/lib/browser/editor-collection.service';
+import { insertSnippetWithMonacoEditor } from '@opensumi/ide-editor/lib/browser/base-editor-wrapper';
 import { MonacoCommandRegistry } from '@opensumi/ide-editor/lib/browser/monaco-contrib/command/command.service';
 import { ITheme, IThemeService } from '@opensumi/ide-theme';
 import { WorkbenchThemeService } from '@opensumi/ide-theme/lib/browser/workbench.theme.service';

package/src/browser/components/ChatHistory.tsx

@@ -216,8 +216,7 @@ const ChatHistory: FC<IChatHistoryProps> = memo(
     // 渲染历史记录列表
     const renderHistory = useCallback(() => {
       const filteredList = historyList
-        .slice(-MAX_HISTORY_LIST)
-        .reverse()
+        .slice(0, MAX_HISTORY_LIST)
         .filter((item) => item.title && item.title.includes(searchValue));
 
       const groupedHistoryList = formatHistory(filteredList);