@opensumi/ide-ai-native 3.9.1-next-1749175927.0 → 3.9.1-next-1749181695.0

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

@@ -0,0 +1,256 @@
+import cls from 'classnames';
+import React, { useEffect, useRef, useState } from 'react';
+
+import { ClickOutside } from '@opensumi/ide-components/lib/click-outside';
+import { Icon, getIcon } from '@opensumi/ide-core-browser/lib/components';
+
+import styles from './mention-select.module.less';
+
+export interface ExtendedModelOption {
+  label: string;
+  value: string;
+  icon?: string;
+  iconClass?: string;
+  tags?: string[];
+  features?: string[];
+  description?: string;
+  disabled?: boolean;
+  badge?: string;
+  badgeColor?: string;
+  selected?: boolean;
+}
+
+export interface MentionSelectProps {
+  options: ExtendedModelOption[];
+  value?: string;
+  onChange?: (value: string) => void;
+  placeholder?: string;
+  disabled?: boolean;
+  className?: string;
+  size?: 'small' | 'medium' | 'large';
+  showThinking?: boolean;
+  thinkingEnabled?: boolean;
+  onThinkingChange?: (enabled: boolean) => void;
+}
+
+const ThinkingToggle: React.FC<{
+  enabled: boolean;
+  onChange: (enabled: boolean) => void;
+}> = ({ enabled, onChange }) => (
+  <div className={styles.thinking_toggle} onClick={() => onChange(!enabled)}>
+    <Icon
+      iconClass={getIcon(enabled ? 'check' : 'circle-outline')}
+      className={cls(styles.thinking_icon, {
+        [styles.enabled]: enabled,
+      })}
+    />
+    <span className={styles.thinking_label}>Thinking</span>
+  </div>
+);
+
+export const MentionSelect: React.FC<MentionSelectProps> = ({
+  options,
+  value,
+  onChange,
+  placeholder,
+  disabled = false,
+  className,
+  size = 'small',
+  showThinking = false,
+  thinkingEnabled = false,
+  onThinkingChange,
+}) => {
+  const [isOpen, setIsOpen] = useState(false);
+  const [activeIndex, setActiveIndex] = useState(-1);
+  const [dropdownDirection, setDropdownDirection] = useState<'up' | 'down'>('up');
+  const selectRef = useRef<HTMLDivElement>(null);
+  const dropdownRef = useRef<HTMLDivElement>(null);
+
+  const selectedOption = options.find((option) => option.selected) || options.find((option) => option.value === value);
+
+  const handleToggle = () => {
+    if (!disabled) {
+      setIsOpen(!isOpen);
+      setActiveIndex(-1);
+    }
+  };
+
+  const handleSelect = (option: ExtendedModelOption) => {
+    if (!option.disabled) {
+      onChange?.(option.value);
+      setIsOpen(false);
+      setActiveIndex(-1);
+    }
+  };
+
+  const handleKeyDown = (e: React.KeyboardEvent) => {
+    if (disabled) {
+      return;
+    }
+
+    switch (e.key) {
+      case 'Enter':
+      case ' ':
+        e.preventDefault();
+        if (!isOpen) {
+          setIsOpen(true);
+        } else if (activeIndex >= 0) {
+          handleSelect(options[activeIndex]);
+        }
+        break;
+      case 'Escape':
+        e.preventDefault();
+        setIsOpen(false);
+        setActiveIndex(-1);
+        break;
+      case 'ArrowDown':
+        e.preventDefault();
+        if (!isOpen) {
+          setIsOpen(true);
+        } else {
+          setActiveIndex((prev) => (prev < options.length - 1 ? prev + 1 : 0));
+        }
+        break;
+      case 'ArrowUp':
+        e.preventDefault();
+        if (isOpen) {
+          setActiveIndex((prev) => (prev > 0 ? prev - 1 : options.length - 1));
+        }
+        break;
+    }
+  };
+
+  const handleClickOutside = () => {
+    setIsOpen(false);
+    setActiveIndex(-1);
+  };
+
+  useEffect(() => {
+    if (isOpen && selectRef.current) {
+      const selectRect = selectRef.current.getBoundingClientRect();
+      const viewportHeight = window.innerHeight;
+      const dropdownHeight = Math.min(400, options.length * 60);
+
+      const spaceAbove = selectRect.top;
+      const spaceBelow = viewportHeight - selectRect.bottom;
+
+      if (spaceAbove < dropdownHeight && spaceBelow > spaceAbove) {
+        setDropdownDirection('down');
+      } else {
+        setDropdownDirection('up');
+      }
+    }
+  }, [isOpen, options.length]);
+
+  useEffect(() => {
+    if (isOpen && activeIndex >= 0 && dropdownRef.current) {
+      const activeElement = dropdownRef.current.children[activeIndex] as HTMLElement;
+      if (activeElement) {
+        activeElement.scrollIntoView({
+          behavior: 'smooth',
+          block: 'nearest',
+        });
+      }
+    }
+  }, [isOpen, activeIndex]);
+
+  return (
+    <ClickOutside onOutsideClick={handleClickOutside}>
+      <div
+        ref={selectRef}
+        className={cls(
+          styles.mention_select,
+          styles[`size_${size}`],
+          {
+            [styles.disabled]: disabled,
+            [styles.open]: isOpen,
+            [styles.dropdown_up]: dropdownDirection === 'up',
+            [styles.dropdown_down]: dropdownDirection === 'down',
+          },
+          className,
+        )}
+        onClick={handleToggle}
+        onKeyDown={handleKeyDown}
+        tabIndex={disabled ? -1 : 0}
+        role='combobox'
+        aria-expanded={isOpen}
+        aria-haspopup='listbox'
+      >
+        <div className={styles.select_trigger}>
+          <div className={styles.select_content}>
+            {selectedOption ? (
+              <div className={styles.selected_option}>
+                <span className={styles.option_label}>{selectedOption.label}</span>
+                {selectedOption.badge && (
+                  <span className={styles.option_badge} style={{ backgroundColor: selectedOption.badgeColor }}>
+                    {selectedOption.badge}
+                  </span>
+                )}
+              </div>
+            ) : (
+              <span className={styles.placeholder}>{placeholder}</span>
+            )}
+          </div>
+          <Icon
+            iconClass={getIcon('down-arrow')}
+            className={cls(styles.dropdown_arrow, {
+              [styles.open]: isOpen,
+            })}
+          />
+        </div>
+
+        {isOpen && (
+          <div ref={dropdownRef} className={styles.dropdown} role='listbox'>
+            {showThinking && onThinkingChange && (
+              <div className={styles.thinking_section}>
+                <ThinkingToggle enabled={thinkingEnabled} onChange={onThinkingChange} />
+                <div className={styles.divider} />
+              </div>
+            )}
+
+            {options.map((option, index) => (
+              <div
+                key={option.value}
+                className={cls(styles.option, {
+                  [styles.active]: index === activeIndex,
+                  [styles.selected]: option.selected || option.value === value,
+                  [styles.disabled]: option.disabled,
+                })}
+                onClick={() => handleSelect(option)}
+                role='option'
+                aria-selected={option.selected || option.value === value}
+              >
+                <div className={styles.option_main}>
+                  <div className={styles.option_header}>
+                    <div className={styles.option_title}>
+                      {option.icon && <Icon icon={option.icon} className={styles.option_icon} />}
+                      {option.iconClass && <Icon iconClass={option.iconClass} className={styles.option_icon} />}
+                      <span className={styles.option_label}>{option.label}</span>
+                      {option.badge && (
+                        <span className={styles.option_badge} style={{ backgroundColor: option.badgeColor }}>
+                          {option.badge}
+                        </span>
+                      )}
+                    </div>
+                  </div>
+
+                  {option.description && <div className={styles.option_description}>{option.description}</div>}
+
+                  {option.tags && option.tags.length > 0 && (
+                    <div className={styles.option_tags}>
+                      {option.tags.map((tag, idx) => (
+                        <span key={idx} className={styles.tag}>
+                          {tag}
+                        </span>
+                      ))}
+                    </div>
+                  )}
+                </div>
+              </div>
+            ))}
+          </div>
+        )}
+      </div>
+    </ClickOutside>
+  );
+};

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

@@ -46,6 +46,17 @@ export interface MentionState {
 interface ModelOption {
   label: string;
   value: string;
+  icon?: string;
+  iconClass?: string;
+  tags?: string[];
+  description?: string;
+  badge?: string;
+  badgeColor?: string;
+}
+
+export interface ExtendedModelOption extends ModelOption {
+  disabled?: boolean;
+  selected?: boolean; // 由外部控制选中状态
 }
 
 export interface ExtendedModelOption {
@@ -89,6 +100,9 @@ export interface FooterConfig {
   buttons?: FooterButton[];
   showModelSelector?: boolean;
   disableModelSelector?: boolean;
+  showThinking?: boolean;
+  thinkingEnabled?: boolean;
+  onThinkingChange?: (enabled: boolean) => void;
 }
 
 export interface MentionInputProps {

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

@@ -1,61 +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; // 最近消息窗口大小（10条）
-  bufferSize: number; // 缓冲区大小（5条）
-}
-
-interface IMemorySummary {
-  content: string;
-  timestamp: number;
-  messageIds: string[];
-}
-
-interface IToolCallInfo {
-  id: string;
-  name: string;
-  args: Record<string, any>;
-  result: any;
-}
-
-export interface MessageBufferManager {
-  messages: IHistoryChatMessage[];
-  add(message: IHistoryChatMessage): void;
-  clear(): void;
-  size(): number;
-}
-
-class SimpleMessageBuffer implements MessageBufferManager {
-  private _messages: IHistoryChatMessage[] = [];
-
-  get messages(): IHistoryChatMessage[] {
-    return this._messages;
-  }
-
-  add(message: IHistoryChatMessage): void {
-    this._messages.push(message);
-  }
-
-  clear(): void {
-    this._messages = [];
-  }
-
-  size(): number {
-    return this._messages.length;
-  }
-}
-
 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 isCompressing = false; // 添加压缩锁，防止重复压缩
 
   private readonly _onMessageChange = new Emitter<IHistoryChatMessage[]>();
   public readonly onMessageChange: Event<IHistoryChatMessage[]> = this._onMessageChange.event;
@@ -63,33 +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 messageBuffer: MessageBufferManager;
-
-  memoryConfig: IMemoryConfig = {
-    shortTermSize: 10,
-    bufferSize: 5,
-  };
-
-  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();
-    this.messageBuffer = new SimpleMessageBuffer();
     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;
-      }
     }
   }
 
@@ -105,91 +33,6 @@ export class MsgHistoryManager extends Disposable {
   public clearMessages() {
     this.messageMap.clear();
     this.messageAdditionalMap.clear();
-    this.memorySummaries = [];
-    this.toolCallMap.clear();
-  }
-
-  /**
-   * 压缩历史消息的方法：
-   * 1. 保持最新的10条消息不变
-   * 2. 当累积了5条超出限制的消息时，对这5条（最早的消息）进行总结
-   * 3. 总结完成后删除这5条消息
-   * 4. 重复这个过程，确保消息数量始终在可控范围内
-   */
-  private async compressMemory() {
-    // 如果正在压缩中，直接返回
-    if (this.isCompressing) {
-      return;
-    }
-
-    const messages = this.messageList;
-    // 只有当消息总数超过短期记忆限制时才进行压缩
-    if (messages.length <= this.memoryConfig.shortTermSize) {
-      return;
-    }
-
-    try {
-      this.isCompressing = true;
-
-      // 1. 获取超出短期记忆限制的最新一条消息
-      const latestExcessMessage = messages[messages.length - this.memoryConfig.shortTermSize - 1];
-
-      // 如果这条消息已经被总结过，就不需要再处理
-      if (latestExcessMessage.isSummarized) {
-        return;
-      }
-
-      // 2. 将这条消息添加到缓冲区
-      this.messageBuffer.add(latestExcessMessage);
-
-      // 3. 当缓冲区达到指定大小时，进行总结
-      if (this.messageBuffer.size() >= this.memoryConfig.bufferSize) {
-        const messagesToSummarize = this.messageBuffer.messages;
-
-        const summaryProvider = this.chatFeatureRegistry.getMessageSummaryProvider?.();
-        if (summaryProvider) {
-          const messageContents = messagesToSummarize.map((msg) => ({
-            role: msg.role,
-            content: msg.content,
-          }));
-
-          const memorize = await summaryProvider.generateMemorizedMessage(messageContents);
-          if (memorize) {
-            // 添加新的记忆总结
-            this.memorySummaries.push({
-              content: memorize,
-              timestamp: Date.now(),
-              messageIds: messagesToSummarize.map((msg) => msg.id),
-            });
-
-            // 标记消息为已总结
-            for (const msg of messagesToSummarize) {
-              const existingMsg = this.messageMap.get(msg.id);
-              if (existingMsg) {
-                this.messageMap.set(msg.id, {
-                  ...existingMsg,
-                  isSummarized: true,
-                });
-              }
-            }
-
-            // 清空缓冲区
-            this.messageBuffer.clear();
-          }
-        }
-      }
-    } finally {
-      this.isCompressing = false;
-    }
-  }
-
-  public addToolCall(toolCall: IToolCallInfo): void {
-    this.toolCallMap.set(toolCall.id, toolCall);
-  }
-
-  public getToolCall(id: string): IToolCallInfo | undefined {
-    const toolCall = this.toolCallMap.get(id);
-    return toolCall;
   }
 
   private doAddMessage(message: IExcludeMessage): string {
@@ -205,19 +48,11 @@ export class MsgHistoryManager extends Disposable {
 
     this.messageMap.set(id, msg);
 
-    // 在添加新消息后尝试压缩记忆
-    this.compressMemory().catch((error) => {
-      // eslint-disable-next-line no-console
-      console.error('[MsgHistoryManager] Error compressing memory', error);
-    });
-
-    // 无论压缩是否完成，都触发消息变更事件
     this._onMessageChange.fire(this.getMessages());
     return id;
   }
 
   private get messageList(): IHistoryChatMessage[] {
-    // 按 order 升序排序，保持消息的原始顺序
     return Array.from(this.messageMap.values()).sort((a, b) => a.order - b.order);
   }
 
@@ -230,17 +65,6 @@ export class MsgHistoryManager extends Disposable {
     return this.messageList;
   }
 
-  public getMemorySummaries(): IMemorySummary[] {
-    return this.memorySummaries;
-  }
-
-  public setMemoryConfig(config: Partial<IMemoryConfig>) {
-    this.memoryConfig = {
-      ...this.memoryConfig,
-      ...config,
-    };
-  }
-
   public addUserMessage(
     message: Required<Pick<IExcludeMessage, 'agentId' | 'agentCommand' | 'content' | 'relationId' | 'images'>>,
   ): string {
@@ -294,12 +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()),
     };
-    return data;
   }
 }

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

@@ -80,6 +80,7 @@
 }
 
 .project_rules_header_left {
+  width: 100%;
   display: flex;
   flex-direction: column;
 }

package/src/browser/rules/rules.view.tsx

@@ -37,7 +37,7 @@ export const RulesView: React.FC = () => {
     [rulesService],
   );
 
-  const getFileNameFromPath = (path: string) => path.split('/').pop() || path;
+  const getFileNameFromPath = (path: string) => decodeURIComponent(path.split('/').pop() || path);
 
   const hasWarning = (rule: ProjectRule) => !rule.description && !rule.alwaysApply && !rule.globs;
 

package/src/browser/types.ts

@@ -308,12 +308,6 @@ export interface IMessageSummaryProvider {
       content: string;
     }>,
   ): Promise<string | undefined>;
-  generateMemorizedMessage(
-    messages: Array<{
-      role: ChatMessageRole;
-      content: string;
-    }>,
-  ): Promise<string | undefined>;
 }
 
 export const AINativeCoreContribution = Symbol('AINativeCoreContribution');

package/src/common/model.ts

@@ -5,6 +5,7 @@ export interface ModelInfo {
   temperature?: number;
   topP?: number;
   topK?: number;
+  maxSteps?: number;
 }
 
 export const deepSeekModels = {

package/src/node/base-language-model.ts

@@ -116,7 +116,7 @@ export abstract class BaseLanguageModel {
         messages,
         abortSignal: abortController.signal,
         experimental_toolCallStreaming: true,
-        maxSteps: 12,
+        maxSteps: modelInfo?.maxSteps ?? 25,
         maxTokens,
         system: systemPrompt,
         providerOptions,