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

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

@@ -1,18 +1,16 @@
 import { Autowired, Injectable } from '@opensumi/di';
-import { AppConfig, PreferenceService } from '@opensumi/ide-core-browser';
+import { 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';
@@ -27,10 +25,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';
 
 /**
@@ -53,9 +51,6 @@ export class ChatProxyService extends Disposable {
   @Autowired(MonacoCommandRegistry)
   private readonly monacoCommandRegistry: MonacoCommandRegistry;
 
-  @Autowired(ChatServiceToken)
-  private aiChatService: ChatService;
-
   @Autowired(IAIReporter)
   private readonly aiReporter: IAIReporter;
 
@@ -71,9 +66,6 @@ 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() {
@@ -112,79 +104,74 @@ export class ChatProxyService extends Disposable {
       initialProps: {},
     });
 
-    this.addDispose(
-      this.chatAgentService.registerAgent({
-        id: ChatProxyService.AGENT_ID,
-        metadata: {
-          systemPrompt:
-            this.preferenceService.get<string>(
+    this.applicationService.getBackendOS().then(() => {
+      this.addDispose(
+        this.chatAgentService.registerAgent({
+          id: ChatProxyService.AGENT_ID,
+          metadata: {
+            systemPrompt: this.preferenceService.get<string>(
               AINativeSettingSectionsId.SystemPrompt,
-              '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;
+              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;
+              }
             }
-          }
-
-          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,
-      }),
-    );
+
+            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,
+        }),
+      );
+    });
 
     queueMicrotask(() => {
       this.chatAgentService.updateAgent(ChatProxyService.AGENT_ID, {});

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

@@ -2,7 +2,13 @@ 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 } from '../types';
+import {
+  IChatFeatureRegistry,
+  IChatSlashCommandHandler,
+  IChatSlashCommandItem,
+  IImageUploadProvider,
+  IMessageSummaryProvider,
+} from '../types';
 
 import { ChatSlashCommandItemModel, ChatWelcomeMessageModel } from './chat-model';
 import { ChatProxyService } from './chat-proxy.service';
@@ -14,6 +20,8 @@ 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;
   }
@@ -22,6 +30,14 @@ 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,3 +1,4 @@
+import debounce from 'lodash/debounce';
 import * as React from 'react';
 import { MessageList } from 'react-chat-elements';
 
@@ -617,7 +618,12 @@ export const AIChatView = () => {
       const { message, images, agentId, command, reportExtra } = value;
       const { actionType, actionSource } = reportExtra || {};
 
-      const request = aiChatService.createRequest(message, agentId!, images, command);
+      const request = aiChatService.createRequest(
+        message.replaceAll(LLM_CONTEXT_KEY_REGEX, ''),
+        agentId!,
+        images,
+        command,
+      );
       if (!request) {
         return;
       }
@@ -641,7 +647,7 @@ export const AIChatView = () => {
         600 * 1000,
       );
       msgHistoryManager.addUserMessage({
-        content: message.replaceAll(LLM_CONTEXT_KEY_REGEX, ''),
+        content: message,
         images: images || [],
         agentId: agentId!,
         agentCommand: command!,
@@ -695,14 +701,9 @@ 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}\``);
@@ -738,6 +739,18 @@ 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],
@@ -918,6 +931,7 @@ 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>('');
@@ -943,13 +957,60 @@ 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 = () => {
+    const getHistoryList = async () => {
       const currentMessages = aiChatService.sessionModel.history.getMessages();
-      const latestUserMessage = currentMessages.findLast((m) => m.role === ChatMessageRole.User);
-      setCurrentTitle(
-        latestUserMessage ? cleanAttachedTextWrapper(latestUserMessage.content).slice(0, MAX_TITLE_LENGTH) : '',
-      );
+      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);
+        }
+      }
+
       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/base-editor-wrapper';
+import { insertSnippetWithMonacoEditor } from '@opensumi/ide-editor/lib/browser/editor-collection.service';
 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,7 +216,8 @@ const ChatHistory: FC<IChatHistoryProps> = memo(
     // 渲染历史记录列表
     const renderHistory = useCallback(() => {
       const filteredList = historyList
-        .slice(0, MAX_HISTORY_LIST)
+        .slice(-MAX_HISTORY_LIST)
+        .reverse()
         .filter((item) => item.title && item.title.includes(searchValue));
 
       const groupedHistoryList = formatHistory(filteredList);