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

package/src/browser/mcp/tools/getOpenEditorFileDiagnostics.ts

@@ -2,7 +2,7 @@ import * as path from 'path';
 
 import { z } from 'zod';
 
-import { Autowired, Injectable } from '@opensumi/di';
+import { Autowired } from '@opensumi/di';
 import { Domain, URI } from '@opensumi/ide-core-common';
 import { WorkbenchEditorService } from '@opensumi/ide-editor';
 import { IWorkspaceService } from '@opensumi/ide-workspace';

package/src/browser/mcp/tools/handlers/ListDir.ts

@@ -1,5 +1,5 @@
 import { Autowired, Injectable } from '@opensumi/di';
-import { AppConfig, Throttler, URI } from '@opensumi/ide-core-browser';
+import { AppConfig, URI } from '@opensumi/ide-core-browser';
 import { IFileServiceClient } from '@opensumi/ide-file-service';
 
 /**

package/src/browser/mcp/tools/runTerminalCmd.ts

@@ -2,7 +2,7 @@ import { z } from 'zod';
 
 import { Autowired } from '@opensumi/di';
 import { AppConfig } from '@opensumi/ide-core-browser';
-import { Deferred, Domain } from '@opensumi/ide-core-common';
+import { Domain } from '@opensumi/ide-core-common';
 import { ITerminalController, ITerminalGroupViewService } from '@opensumi/ide-terminal-next/lib/common/controller';
 
 import { IMCPServerRegistry, MCPLogger, MCPServerContribution, MCPToolDefinition } from '../../types';

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

@@ -1,40 +1,11 @@
 import { Disposable, Emitter, Event, uuid } from '@opensumi/ide-core-common';
 import { ChatMessageRole, IHistoryChatMessage } from '@opensumi/ide-core-common/lib/types/ai-native';
 
-import { ChatFeatureRegistry } from '../chat/chat.feature.registry';
-
 type IExcludeMessage = Omit<IHistoryChatMessage, 'id' | 'order'>;
 
-interface IMemoryConfig {
-  shortTermSize: number;  // 短期记忆窗口大小
-  midTermSize: number;    // 中期记忆窗口大小
-  longTermSize: number;   // 长期记忆窗口大小
-}
-
-interface IMemorySummary {
-  content: string;
-  timestamp: number;
-  messageIds: string[];
-  toolCalls?: Array<{
-    id: string;
-    name: string;
-    args?: Record<string, any>;
-    result?: string;
-  }>;
-}
-
-interface IToolCallInfo {
-  id: string;
-  name: string;
-  args: Record<string, any>;
-  result: any;
-}
-
 export class MsgHistoryManager extends Disposable {
   private messageMap: Map<string, IHistoryChatMessage> = new Map();
   private messageAdditionalMap: Map<string, Record<string, any>> = new Map();
-  private memorySummaries: IMemorySummary[] = [];
-  private toolCallMap: Map<string, IToolCallInfo> = new Map();
 
   private readonly _onMessageChange = new Emitter<IHistoryChatMessage[]>();
   public readonly onMessageChange: Event<IHistoryChatMessage[]> = this._onMessageChange.event;
@@ -42,36 +13,11 @@ export class MsgHistoryManager extends Disposable {
   private readonly _onMessageAdditionalChange = new Emitter<Record<string, any>>();
   public readonly onMessageAdditionalChange: Event<Record<string, any>> = this._onMessageAdditionalChange.event;
 
-  private memoryConfig: IMemoryConfig = {
-    shortTermSize: 10,    // 最近的10条消息保持完整
-    midTermSize: 20,      // 接下来的20条消息进行总结
-    longTermSize: 50,     // 最多保留50条消息
-  };
-
-  constructor(
-    private chatFeatureRegistry: ChatFeatureRegistry,
-    data?: {
-      additional: Record<string, any>;
-      messages: IHistoryChatMessage[];
-      toolCalls?: Record<string, IToolCallInfo>;
-      memorySummaries?: IMemorySummary[];
-    },
-  ) {
+  constructor(data?: { additional: Record<string, any>; messages: IHistoryChatMessage[] }) {
     super();
     if (data) {
       this.messageMap = new Map(data.messages.map((item) => [item.id, item]));
       this.messageAdditionalMap = new Map(Object.entries(data.additional));
-      if (data.toolCalls) {
-        this.toolCallMap = new Map(Object.entries(data.toolCalls));
-      }
-      if (data.memorySummaries) {
-        this.memorySummaries = data.memorySummaries;
-      }
-      console.log('[MsgHistoryManager] Initialized with data:', {
-        messagesCount: this.messageMap.size,
-        toolCallsCount: this.toolCallMap.size,
-        memorySummariesCount: this.memorySummaries.length,
-      });
     }
   }
 
@@ -85,133 +31,8 @@ export class MsgHistoryManager extends Disposable {
   }
 
   public clearMessages() {
-    console.log('[MsgHistoryManager] Clearing all messages and related data');
     this.messageMap.clear();
     this.messageAdditionalMap.clear();
-    this.memorySummaries = [];
-    this.toolCallMap.clear();
-  }
-
-  private async compressMemory() {
-    const messages = this.messageList;
-    const summaryProvider = this.chatFeatureRegistry.getMessageSummaryProvider();
-
-    // 1. 短期记忆处理（保持完整信息）
-    if (messages.length <= this.memoryConfig.shortTermSize) {
-      return;
-    }
-
-    // 2. 中期记忆处理
-    const midTermStart = this.memoryConfig.shortTermSize;
-    const midTermEnd = Math.min(
-      messages.length,
-      this.memoryConfig.shortTermSize + this.memoryConfig.midTermSize,
-    );
-
-    if (midTermStart < midTermEnd) {
-      const midTermMessages = messages.slice(midTermStart, midTermEnd);
-
-      // 收集中期记忆的 toolCall 信息（保留参数和简化结果）
-      const midTermToolCalls: IToolCallInfo[] = [];
-      midTermMessages.forEach((msg) => {
-        const additional = this.messageAdditionalMap.get(msg.id);
-        if (additional?.toolCallId) {
-          const toolCall = this.toolCallMap.get(additional.toolCallId);
-          if (toolCall) {
-            midTermToolCalls.push({
-              id: toolCall.id,
-              name: toolCall.name,
-              args: toolCall.args,
-              // 中期记忆只保留简化的结果
-              result: typeof toolCall.result === 'object' ?
-                JSON.stringify(toolCall.result).slice(0, 100) + '...' :
-                String(toolCall.result).slice(0, 100) + '...',
-            });
-          }
-        }
-      });
-
-      // 对中期记忆进行总结
-      const midTermContents = midTermMessages.map((msg) => ({
-        role: msg.role,
-        content: msg.content,
-      }));
-
-      if (summaryProvider) {
-        const midTermSummary = await summaryProvider.getMessageSummary(midTermContents);
-        if (midTermSummary) {
-          this.memorySummaries.push({
-            content: midTermSummary,
-            timestamp: Date.now(),
-            messageIds: midTermMessages.map((msg) => msg.id),
-            toolCalls: midTermToolCalls.length > 0 ? midTermToolCalls : undefined,
-          });
-        }
-      }
-    }
-
-    // 3. 长期记忆处理
-    const longTermStart = midTermEnd;
-    const longTermEnd = Math.min(
-      messages.length,
-      this.memoryConfig.shortTermSize + this.memoryConfig.midTermSize + this.memoryConfig.longTermSize,
-    );
-
-    if (longTermStart < longTermEnd) {
-      const longTermMessages = messages.slice(longTermStart, longTermEnd);
-
-      // 对长期记忆进行更简洁的总结，不包含 toolCall 信息
-      const longTermContents = longTermMessages.map((msg) => ({
-        role: msg.role,
-        content: msg.content,
-      }));
-
-      if (summaryProvider) {
-        const longTermSummary = await summaryProvider.getMessageSummary(longTermContents);
-        if (longTermSummary) {
-          this.memorySummaries.push({
-            content: longTermSummary,
-            timestamp: Date.now(),
-            messageIds: longTermMessages.map((msg) => msg.id),
-            // 长期记忆不保留 toolCall 信息
-          });
-        }
-      }
-    }
-
-    // 4. 删除超出范围的消息
-    const maxSize = this.memoryConfig.shortTermSize +
-      this.memoryConfig.midTermSize +
-      this.memoryConfig.longTermSize;
-
-    if (messages.length > maxSize) {
-      const toRemove = messages.slice(maxSize);
-      toRemove.forEach((msg) => {
-        const additional = this.messageAdditionalMap.get(msg.id);
-        if (additional?.toolCallId) {
-          this.toolCallMap.delete(additional.toolCallId);
-        }
-        this.messageMap.delete(msg.id);
-        this.messageAdditionalMap.delete(msg.id);
-      });
-    }
-  }
-
-  public addToolCall(toolCall: IToolCallInfo): void {
-    console.log('[MsgHistoryManager] Adding toolCall', {
-      toolCallId: toolCall.id,
-      toolName: toolCall.name,
-    });
-    this.toolCallMap.set(toolCall.id, toolCall);
-  }
-
-  public getToolCall(id: string): IToolCallInfo | undefined {
-    const toolCall = this.toolCallMap.get(id);
-    console.log('[MsgHistoryManager] Getting toolCall', {
-      toolCallId: id,
-      found: !!toolCall,
-    });
-    return toolCall;
   }
 
   private doAddMessage(message: IExcludeMessage): string {
@@ -225,18 +46,9 @@ export class MsgHistoryManager extends Disposable {
       type: message.type || 'string',
     };
 
-    console.log('[MsgHistoryManager] Adding new message', {
-      messageId: id,
-      role: message.role,
-      order,
-    });
-
     this.messageMap.set(id, msg);
 
-    // 在添加新消息后尝试压缩记忆
-    this.compressMemory().then(() => {
-      this._onMessageChange.fire(this.getMessages());
-    });
+    this._onMessageChange.fire(this.getMessages());
     return id;
   }
 
@@ -253,31 +65,9 @@ export class MsgHistoryManager extends Disposable {
     return this.messageList;
   }
 
-  public getMemorySummaries(): IMemorySummary[] {
-    console.log('[MsgHistoryManager] Getting memory summaries', {
-      summariesCount: this.memorySummaries.length,
-    });
-    return this.memorySummaries;
-  }
-
-  public setMemoryConfig(config: Partial<IMemoryConfig>) {
-    console.log('[MsgHistoryManager] Updating memory config', {
-      oldConfig: this.memoryConfig,
-      newConfig: { ...this.memoryConfig, ...config },
-    });
-    this.memoryConfig = {
-      ...this.memoryConfig,
-      ...config,
-    };
-  }
-
   public addUserMessage(
     message: Required<Pick<IExcludeMessage, 'agentId' | 'agentCommand' | 'content' | 'relationId' | 'images'>>,
   ): string {
-    console.log('[MsgHistoryManager] Adding user message', {
-      agentId: message.agentId,
-      command: message.agentCommand,
-    });
     return this.doAddMessage({
       ...message,
       role: ChatMessageRole.User,
@@ -285,7 +75,6 @@ export class MsgHistoryManager extends Disposable {
   }
 
   public addAssistantMessage(message: Omit<IExcludeMessage, 'role'>): string {
-    console.log('[MsgHistoryManager] Adding assistant message');
     return this.doAddMessage({
       ...message,
       role: ChatMessageRole.Assistant,
@@ -294,12 +83,10 @@ export class MsgHistoryManager extends Disposable {
 
   public updateAssistantMessage(id: string, message: Omit<IExcludeMessage, 'role'>) {
     if (!this.messageMap.has(id)) {
-      console.log('[MsgHistoryManager] Failed to update assistant message: message not found', { messageId: id });
       return;
     }
 
     const oldMessage = this.messageMap.get(id);
-    console.log('[MsgHistoryManager] Updating assistant message', { messageId: id });
 
     this.messageMap.set(id, {
       ...oldMessage!,
@@ -309,7 +96,6 @@ export class MsgHistoryManager extends Disposable {
 
   public setMessageAdditional(id: string, additional: Record<string, any>) {
     if (!this.messageMap.has(id)) {
-      console.log('[MsgHistoryManager] Failed to set message additional: message not found', { messageId: id });
       return;
     }
 
@@ -319,11 +105,6 @@ export class MsgHistoryManager extends Disposable {
       ...additional,
     };
 
-    console.log('[MsgHistoryManager] Setting message additional', {
-      messageId: id,
-      hasToolCallId: !!additional.toolCallId,
-    });
-
     this.messageAdditionalMap.set(id, newAdditional);
     this._onMessageAdditionalChange.fire(newAdditional);
   }
@@ -337,17 +118,9 @@ export class MsgHistoryManager extends Disposable {
   }
 
   toJSON() {
-    const data = {
+    return {
       messages: this.getMessages(),
       additional: Object.fromEntries(this.messageAdditionalMap.entries()),
-      memorySummaries: this.memorySummaries,
-      toolCalls: Object.fromEntries(this.toolCallMap.entries()),
     };
-    console.log('[MsgHistoryManager] Serializing to JSON', {
-      messagesCount: data.messages.length,
-      toolCallsCount: Object.keys(data.toolCalls).length,
-      memorySummariesCount: data.memorySummaries.length,
-    });
-    return data;
   }
 }

package/src/browser/preferences/schema.ts

@@ -177,5 +177,10 @@ export const aiNativePreferenceSchema: PreferenceSchema = {
       type: 'string',
       description: '%preference.ai.native.chat.system.prompt.description%',
     },
+    [AINativeSettingSectionsId.GlobalRules]: {
+      type: 'string',
+      default: '',
+      description: '%preference.ai.native.globalRules.description%',
+    },
   },
 };

package/src/browser/rules/rules.contribution.ts

@@ -0,0 +1,105 @@
+import { Autowired } from '@opensumi/di';
+import { getIcon } from '@opensumi/ide-components';
+import { ClientAppContribution } from '@opensumi/ide-core-browser';
+import { LabelService } from '@opensumi/ide-core-browser/lib/services';
+import {
+  CommandContribution,
+  CommandRegistry,
+  Domain,
+  RulesServiceToken,
+  URI,
+  localize,
+} from '@opensumi/ide-core-common';
+import {
+  BrowserEditorContribution,
+  EditorComponentRegistry,
+  EditorComponentRenderMode,
+  IResource,
+  ResourceService,
+  WorkbenchEditorService,
+} from '@opensumi/ide-editor/lib/browser/types';
+import { IconService } from '@opensumi/ide-theme/lib/browser';
+import { IWorkspaceService } from '@opensumi/ide-workspace/lib/common';
+
+import { RulesService } from './rules.service';
+import { RulesView } from './rules.view';
+
+export namespace RulesCommands {
+  export const OPEN_RULES_FILE = {
+    id: 'rules.openRulesConfig',
+    label: 'Open Rules Configuration',
+  };
+}
+
+const COMPONENTS_ID = 'opensumi-rules-viewer';
+export const RULES_COMPONENTS_SCHEME_ID = 'rules';
+
+export type IRulesResource = IResource<{ configType: string }>;
+
+@Domain(BrowserEditorContribution, CommandContribution, ClientAppContribution)
+export class RulesContribution implements BrowserEditorContribution, CommandContribution, ClientAppContribution {
+  @Autowired(IWorkspaceService)
+  protected readonly workspaceService: IWorkspaceService;
+
+  @Autowired(IconService)
+  protected readonly iconService: IconService;
+
+  @Autowired(WorkbenchEditorService)
+  protected readonly editorService: WorkbenchEditorService;
+
+  @Autowired()
+  labelService: LabelService;
+
+  @Autowired(RulesServiceToken)
+  protected readonly rulesService: RulesService;
+
+  onStart() {
+    this.rulesService.initProjectRules();
+  }
+
+  registerEditorComponent(registry: EditorComponentRegistry) {
+    registry.registerEditorComponent({
+      uid: COMPONENTS_ID,
+      scheme: RULES_COMPONENTS_SCHEME_ID,
+      component: RulesView,
+      renderMode: EditorComponentRenderMode.ONE_PER_WORKBENCH,
+    });
+
+    registry.registerEditorComponentResolver(RULES_COMPONENTS_SCHEME_ID, (resource, results) => {
+      results.push({
+        type: 'component',
+        componentId: COMPONENTS_ID,
+      });
+    });
+  }
+
+  registerResource(service: ResourceService) {
+    service.registerResourceProvider({
+      scheme: RULES_COMPONENTS_SCHEME_ID,
+      provideResource: async (uri: URI): Promise<IRulesResource> => {
+        const { configType } = uri.getParsedQuery();
+
+        return {
+          uri,
+          name: localize('ai.native.rules.title'),
+          icon: getIcon('rules'),
+          metadata: {
+            configType,
+          },
+        };
+      },
+    });
+  }
+
+  registerCommands(registry: CommandRegistry) {
+    registry.registerCommand(RulesCommands.OPEN_RULES_FILE, {
+      execute: () => {
+        const uri = new URI().withScheme(RULES_COMPONENTS_SCHEME_ID);
+        this.editorService.open(uri, {
+          preview: false,
+          focus: true,
+        });
+      },
+    });
+  }
+}

package/src/browser/rules/rules.module.less

@@ -0,0 +1,175 @@
+.container {
+  padding: 20px;
+  color: var(--foreground);
+  textarea {
+    resize: vertical;
+    min-height: 60px;
+  }
+  input,
+  textarea {
+    width: 100%;
+    padding: 8px;
+    border: 1px solid var(--input-border);
+    border-radius: 4px;
+    background-color: var(--input-background);
+    color: var(--input-foreground);
+    font-size: 13px;
+    font-family: var(--monaco-monospace-font);
+
+    &:focus {
+      border-color: var(--focusBorder);
+      outline: none;
+    }
+
+    &::placeholder {
+      color: var(--descriptionForeground);
+      opacity: 0.6;
+    }
+  }
+}
+
+.header {
+  display: flex;
+  justify-content: space-between;
+  align-items: flex-start;
+  margin-bottom: 24px;
+}
+
+.title {
+  margin: 0 0 8px 0;
+  font-size: 24px;
+  font-weight: 600;
+}
+
+.description {
+  margin: 0;
+  color: var(--descriptionForeground);
+  font-size: 14px;
+  line-height: 1.5;
+}
+
+.user_rules {
+  display: flex;
+  flex-direction: column;
+  gap: 16px;
+  .user_rules_header {
+    display: flex;
+    flex-direction: column;
+  }
+}
+
+.project_rules {
+  margin-top: 20px;
+  display: flex;
+  flex-direction: column;
+  gap: 16px;
+  h3 {
+    position: relative;
+  }
+  .actionButton {
+    position: absolute;
+    right: 0;
+    top: 0;
+  }
+}
+
+.project_rules_header {
+  display: flex;
+  justify-content: space-between;
+  align-items: flex-start;
+}
+
+.project_rules_header_left {
+  width: 100%;
+  display: flex;
+  flex-direction: column;
+}
+
+.actionButton {
+  padding: 8px 16px;
+  border-radius: 4px;
+  background-color: var(--button-primary-background);
+  color: var(--button-primary-foreground);
+  border: none;
+  cursor: pointer;
+  font-size: 13px;
+  .actionButtonIcon {
+    margin-right: 5px;
+  }
+
+  &:hover {
+    background-color: var(--button-primary-hover-background);
+  }
+}
+
+.project_rules_list {
+  display: flex;
+  flex-direction: column;
+  gap: 8px;
+}
+
+.rule_item {
+  display: flex;
+  justify-content: space-between;
+  align-items: center;
+  padding: 12px;
+  border: 1px solid var(--input-border);
+  border-radius: 4px;
+  background-color: var(--input-background);
+  opacity: 0.6;
+
+  &:hover {
+    opacity: 1;
+  }
+}
+
+.rule_item_left {
+  display: flex;
+  flex-direction: column;
+  gap: 4px;
+  flex: 1;
+}
+
+.rule_item_right {
+  display: flex;
+  align-items: center;
+  flex-shrink: 0;
+  margin-left: 16px;
+}
+
+.rule_filename {
+  font-size: 13px;
+  font-weight: 500;
+  color: var(--foreground);
+}
+
+.rule_description {
+  font-size: 12px;
+  color: var(--descriptionForeground);
+}
+
+.rule_warning {
+  font-size: 12px;
+  color: var(--editorWarning-foreground);
+  display: flex;
+  align-items: center;
+  gap: 4px;
+
+  &::before {
+    content: '⚠';
+    color: var(--editorWarning-foreground);
+  }
+}
+
+.rule_applies_to {
+  font-size: 12px;
+  color: var(--descriptionForeground);
+  font-family: var(--monaco-monospace-font);
+}
+
+.empty_state {
+  text-align: center;
+  padding: 32px;
+  color: var(--descriptionForeground);
+  font-style: italic;
+}