@opensumi/ide-ai-native 3.9.1-next-1748944723.0 → 3.9.1-next-1749007675.0

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

@@ -29,6 +29,7 @@ import { MsgHistoryManager } from '../model/msg-history-manager';
 import { IChatSlashCommandItem } from '../types';
 
 import type { ImagePart, TextPart, ToolCallPart } from 'ai';
+import { ChatFeatureRegistry } from './chat.feature.registry';
 
 export type IChatProgressResponseContent =
   | IChatMarkdownContent
@@ -296,10 +297,13 @@ export class ChatRequestModel implements IChatRequestModel {
 export class ChatModel extends Disposable implements IChatModel {
   private requestIdPool = 0;
 
-  constructor(initParams?: { sessionId?: string; history?: MsgHistoryManager; modelId?: string }) {
+  constructor(
+    private chatFeatureRegistry: ChatFeatureRegistry,
+    initParams?: { sessionId?: string; history?: MsgHistoryManager; modelId?: string },
+  ) {
     super();
     this.#sessionId = initParams?.sessionId ?? uuid();
-    this.history = initParams?.history ?? new MsgHistoryManager();
+    this.history = initParams?.history ?? new MsgHistoryManager(this.chatFeatureRegistry);
     this.#modelId = initParams?.modelId;
   }
 
@@ -346,9 +350,9 @@ export class ChatModel extends Disposable implements IChatModel {
         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)),
-            ]
+            { 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) {
@@ -375,18 +379,58 @@ export class ChatModel extends Disposable implements IChatModel {
               content: [],
             });
           }
-          (history[history.length - 1].content as Array<TextPart | ToolCallPart>).push({
-            type: 'tool-call',
-            toolCallId: part.content.id,
-            toolName: part.content.function.name,
+          const toolCallId = part.content.id;
+          console.log('[ChatModel] Processing toolCall', {
+            toolCallId,
+            functionName: part.content.function.name,
+            hasArguments: !!part.content.function.arguments,
+            hasResult: !!part.content.result,
+          });
+
+          const toolCallInfo = {
+            id: toolCallId,
+            name: part.content.function.name,
             args: (() => {
               try {
-                return JSON.parse(part.content.function.arguments || '{}');
+                const args = JSON.parse(part.content.function.arguments || '{}');
+                console.log('[ChatModel] Parsed toolCall arguments', {
+                  toolCallId,
+                  args,
+                });
+                return args;
+              } catch (e) {
+                console.error('[ChatModel] Failed to parse tool call arguments:', e);
+                return {};
+              }
+            })(),
+            result: (() => {
+              try {
+                const result = JSON.parse(part.content.result || '{}');
+                console.log('[ChatModel] Parsed toolCall result', {
+                  toolCallId,
+                  resultSize: JSON.stringify(result).length,
+                });
+                return result;
               } catch (e) {
-                console.error('Failed to parse tool call arguments:', e);
+                console.error('[ChatModel] Failed to parse tool result:', e);
                 return {};
               }
             })(),
+          };
+
+          // 保存 toolCall 信息到 history manager
+          console.log('[ChatModel] Saving toolCall to history manager', {
+            toolCallId,
+            requestId: request.requestId,
+          });
+          this.history.addToolCall(toolCallInfo);
+          this.history.setMessageAdditional(request.requestId, { toolCallId });
+
+          (history[history.length - 1].content as Array<TextPart | ToolCallPart>).push({
+            type: 'tool-call',
+            toolCallId: part.content.id,
+            toolName: part.content.function.name,
+            args: toolCallInfo.args,
           });
           history.push({
             role: 'tool',
@@ -395,14 +439,7 @@ export class ChatModel extends Disposable implements IChatModel {
                 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 {};
-                  }
-                })(),
+                result: toolCallInfo.result,
               },
             ],
           });

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,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

@@ -617,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;
       }
@@ -646,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!,
@@ -700,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}\``);
@@ -738,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],
@@ -930,6 +918,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>('');
@@ -959,9 +948,26 @@ export function DefaultChatViewHeader({
     const getHistoryList = () => {
       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 summaryProvider = chatFeatureRegistry.getMessageSummaryProvider();
+      const currentTitle = latestUserMessage
+        ? cleanAttachedTextWrapper(latestUserMessage.content).slice(0, MAX_TITLE_LENGTH)
+        : '';
+      const messages = aiChatService.sessionModel.history.getMessages().map((msg) => ({
+        role: msg.role,
+        content: msg.content,
+      }));
+      if (messages.length > 2 && summaryProvider && aiChatService.sessionModel.sessionId) {
+        summaryProvider.getMessageSummary(messages).then((summary) => {
+          if (summary) {
+            setCurrentTitle(summary.slice(0, MAX_TITLE_LENGTH));
+          } else {
+            setCurrentTitle(currentTitle);
+          }
+        });
+      } else {
+        setCurrentTitle(currentTitle);
+      }
+
       setHistoryList(
         aiChatService.getSessions().map((session) => {
           const history = session.history;