@opensumi/ide-ai-native 3.8.1-next-1741091353.0 → 3.8.1-next-1741092802.0

package/src/browser/widget/inline-stream-diff/inline-stream-diff.handler.tsx

@@ -44,13 +44,10 @@ export enum EComputerMode {
 
 const inlineStreamDiffComputer = new InlineStreamDiffComputer();
 
-/**
- * Abstract base class for inline streaming diff handlers
- */
 @Injectable({ multiple: true })
-export abstract class BaseInlineStreamDiffHandler extends Disposable implements IInlineDiffPreviewerNode {
+export class InlineStreamDiffHandler extends Disposable implements IInlineDiffPreviewerNode {
   @Autowired(INJECTOR_TOKEN)
-  protected readonly injector: Injector;
+  private readonly injector: Injector;
 
   protected readonly _onDidEditChange = this.registerDispose(new Emitter<void>());
   public readonly onDidEditChange: Event<void> = this._onDidEditChange.event;
@@ -60,14 +57,15 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
 
   public previewerOptions: IDiffPreviewerOptions;
 
-  protected savedModel: ITextModel;
-  protected virtualModel: ITextModel;
+  private originalModel: ITextModel;
+  private virtualModel: ITextModel;
 
-  protected rawSavedTextLines: string[];
-  protected rawOriginTextLinesTokens: LineTokens[] | undefined;
-  protected undoRedoGroup: UndoRedoGroup = new UndoRedoGroup();
+  private rawOriginalTextLines: string[];
+  private rawOriginalTextLinesTokens: LineTokens[] = [];
+  private undoRedoGroup: UndoRedoGroup = new UndoRedoGroup();
 
-  protected readonly finallyDiffModel: ISettableObservable<IComputeDiffData | undefined> = observableValue(
+  private readonly diffModel: ISettableObservable<IComputeDiffData | undefined> = observableValue(this, undefined);
+  private readonly finallyDiffModel: ISettableObservable<IComputeDiffData | undefined> = observableValue(
     this,
     undefined,
   );
@@ -75,15 +73,14 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
   public livePreviewDiffDecorationModel: LivePreviewDiffDecorationModel;
 
   public get uri() {
-    return this.savedModel.uri;
+    return this.originalModel.uri;
   }
 
-  constructor(protected readonly monacoEditor: ICodeEditor) {
+  constructor(private readonly monacoEditor: ICodeEditor) {
     super();
 
     const modelService = StandaloneServices.get(IModelService);
-    const savedModel = this.monacoEditor.getModel()!;
-    const setModel = modelService.createModel(
+    this.virtualModel = modelService.createModel(
       '',
       null,
       Uri.from({
@@ -92,8 +89,7 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
       }),
       true,
     );
-    this.savedModel = savedModel;
-    this.virtualModel = setModel;
+    this.originalModel = this.monacoEditor.getModel()!;
 
     this.livePreviewDiffDecorationModel = this.injector.get(LivePreviewDiffDecorationModel, [this.monacoEditor]);
     this.addDispose(this.livePreviewDiffDecorationModel);
@@ -117,21 +113,45 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
     });
   }
 
-  abstract initialize(selection: Selection): void;
+  initialize(selection: Selection): void {
+    const eol = this.originalModel.getEOL();
+    const startPosition = selection.getStartPosition();
+    const endPosition = selection.getEndPosition();
+
+    this.rawOriginalTextLines = this.originalModel
+      .getValueInRange(Range.fromPositions(startPosition, endPosition))
+      .split(eol);
+
+    this.rawOriginalTextLinesTokens = this.rawOriginalTextLines.map((_, index) => {
+      const lineNumber = startPosition.lineNumber + index;
+      this.originalModel.tokenization.forceTokenization(lineNumber);
+      const lineTokens = this.originalModel.tokenization.getLineTokens(lineNumber);
+      return lineTokens;
+    });
+
+    const zone = LineRange.fromRangeInclusive(
+      Range.fromPositions(
+        { lineNumber: selection.startLineNumber, column: 1 },
+        { lineNumber: selection.endLineNumber, column: Number.MAX_SAFE_INTEGER },
+      ),
+    );
+
+    this.livePreviewDiffDecorationModel.initialize(zone);
+  }
 
   getVirtualModelValue() {
     return this.virtualModel.getValue();
   }
 
   getOriginModelValue() {
-    return this.rawSavedTextLines.join('\n');
+    return this.rawOriginalTextLines.join('\n');
   }
 
   get onPartialEditWidgetListChange() {
     return this.livePreviewDiffDecorationModel.onPartialEditWidgetListChange;
   }
 
-  protected computeDiff(
+  private computeDiff(
     originalTextLines: string[],
     newTextLines: string[],
     computerMode: EComputerMode = EComputerMode.default,
@@ -265,12 +285,7 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
           },
           textLines: removedTextLines.map((text, index) => ({
             text,
-            lineTokens:
-              this.rawOriginTextLinesTokens?.[removedLinesOriginalRange.startLineNumber - 1 + index] ||
-              LineTokens.createEmpty(text, {
-                encodeLanguageId: () => 0,
-                decodeLanguageId: () => 'plaintext',
-              }),
+            lineTokens: this.rawOriginalTextLinesTokens[removedLinesOriginalRange.startLineNumber - 1 + index],
           })),
         });
       }
@@ -283,11 +298,6 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
     });
   }
 
-  /**
-   * Get the original model for diff operations
-   */
-  protected abstract getOriginalModel(): ITextModel;
-
   /**
    * 令当前的 inline diff 在流式渲染过程当中使用 pushEditOperations 进行编辑的操作都放在同一组 undo/redo 堆栈里
    * 一旦撤销到最顶层则关闭当前的 inline diff
@@ -300,24 +310,22 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
     });
   }
 
-  protected renderDiffEdits(diffModel: IComputeDiffData): void {
+  private renderDiffEdits(diffModel: IComputeDiffData): void {
     const { activeLine, newFullRangeTextLines, pendingRange } = diffModel;
+    const eol = this.originalModel.getEOL();
     const zone = this.getZone();
 
-    const originalModel = this.getOriginalModel();
-    const eol = originalModel.getEOL();
-
     const validZone =
       zone.startLineNumber < zone.endLineNumberExclusive
         ? new Range(
             zone.startLineNumber,
             1,
             zone.endLineNumberExclusive - 1,
-            originalModel.getLineMaxColumn(zone.endLineNumberExclusive - 1),
+            this.originalModel.getLineMaxColumn(zone.endLineNumberExclusive - 1),
           )
         : new Range(zone.startLineNumber, 1, zone.startLineNumber, 1);
 
-    const newOriginalTextLines = originalModel.getValueInRange(validZone).split(eol);
+    const newOriginalTextLines = this.originalModel.getValueInRange(validZone).split(eol);
     const diffComputation = linesDiffComputers.getDefault().computeDiff(newOriginalTextLines, newFullRangeTextLines, {
       computeMoves: false,
       maxComputationTimeMs: 200,
@@ -362,7 +370,7 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
             validZone.startLineNumber + change.original.startLineNumber - 1,
             1,
             validZone.startLineNumber + change.original.endLineNumberExclusive - 2,
-            originalModel.getLineMaxColumn(validZone.startLineNumber + change.original.endLineNumberExclusive - 2),
+            this.originalModel.getLineMaxColumn(validZone.startLineNumber + change.original.endLineNumberExclusive - 2),
           );
         }
         const edit = {
@@ -373,7 +381,7 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
         realTimeChanges.push(edit);
       }
     }
-    originalModel.pushEditOperations(null, realTimeChanges, () => null, this.undoRedoGroup);
+    this.originalModel.pushEditOperations(null, realTimeChanges, () => null, this.undoRedoGroup);
 
     /**
      * 根据 newFullRangeTextLines 内容长度重新计算 zone，避免超过最大长度，进而影响未选中的代码区域
@@ -413,20 +421,17 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
     this._onDidEditChange.fire();
   }
 
-  public recompute(computerMode: EComputerMode, content?: string): IComputeDiffData {
-    if (content) {
-      this.virtualModel.setValue(content);
+  public recompute(computerMode: EComputerMode, newContent?: string): IComputeDiffData {
+    if (newContent) {
+      this.virtualModel.setValue(newContent);
     }
 
-    const textLines = this.virtualModel.getLinesContent();
-    return this.processDiffComputation(textLines, computerMode);
+    const newTextLines = this.virtualModel.getLinesContent();
+    return this.computeDiff(this.rawOriginalTextLines, newTextLines, computerMode);
   }
 
-  protected currentEditLine = 0;
-  protected isEditing = false;
-
-  protected abstract processDiffComputation(currentText: string[], computerMode?: EComputerMode): IComputeDiffData;
-
+  private currentEditLine = 0;
+  private isEditing = false;
   public async rateRenderEditController(): Promise<void> {
     if (this.isEditing === false) {
       this.isEditing = true;
@@ -438,14 +443,19 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
 
         const virtualTextLines = this.virtualModel.getLinesContent();
         const currentText = virtualTextLines.slice(0, this.currentEditLine);
-        const currentDiffModel = this.processDiffComputation(currentText);
+        const currentDiffModel = this.computeDiff(this.rawOriginalTextLines, currentText);
+        transaction((tx) => {
+          this.diffModel.set(currentDiffModel, tx);
+        });
+
+        this.onDiffFinishedEmitter.fire(currentDiffModel);
 
-        if (this.savedModel.id === this.monacoEditor.getModel()?.id) {
+        if (this.originalModel.id === this.monacoEditor.getModel()?.id) {
           this.renderDiffEdits(currentDiffModel);
         }
 
         this.currentEditLine += 1;
-        // 这个 sleep 会带来潜在的时序问题，如 finallyRender 时模型已经被 dispose
+
         await sleep(FRAME_THREE);
       }
 
@@ -459,7 +469,10 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
   }
 
   public addLinesToDiff(newText: string, computerMode: EComputerMode = EComputerMode.default): void {
-    this.recompute(computerMode, newText);
+    const diffModel = this.recompute(computerMode, newText);
+    transaction((tx) => {
+      this.diffModel.set(diffModel, tx);
+    });
   }
 
   public pushRateFinallyDiffStack(diffModel: IComputeDiffData): void {
@@ -475,17 +488,14 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
   public finallyRender(diffModel: IComputeDiffData): void {
     transaction((tx) => {
       this.finallyDiffModel.set(diffModel, tx);
+      this.diffModel.set(diffModel, tx);
     });
 
-    if (this.savedModel.id !== this.monacoEditor.getModel()?.id) {
+    if (this.originalModel.id !== this.monacoEditor.getModel()?.id) {
       return;
     }
 
     this.onDiffFinishedEmitter.fire(diffModel);
-
-    if (this.livePreviewDiffDecorationModel.disposed) {
-      return;
-    }
     this.renderPartialEditWidgets(diffModel);
     this.renderDiffEdits(diffModel);
     this.pushStackElement();
@@ -527,84 +537,3 @@ export abstract class BaseInlineStreamDiffHandler extends Disposable implements
     return this.livePreviewDiffDecorationModel.getTotalCodeInfo();
   }
 }
-
-/**
- * Regular inline stream diff handler (non-reverse mode)
- */
-@Injectable({ multiple: true })
-export class InlineStreamDiffHandler extends BaseInlineStreamDiffHandler {
-  initialize(selection: Selection): void {
-    const eol = this.savedModel.getEOL();
-    const startPosition = selection.getStartPosition();
-    const endPosition = selection.getEndPosition();
-
-    this.rawSavedTextLines = this.savedModel
-      .getValueInRange(Range.fromPositions(startPosition, endPosition))
-      .split(eol);
-
-    const zone = LineRange.fromRangeInclusive(
-      Range.fromPositions(
-        { lineNumber: selection.startLineNumber, column: 1 },
-        { lineNumber: selection.endLineNumber, column: Number.MAX_SAFE_INTEGER },
-      ),
-    );
-
-    this.livePreviewDiffDecorationModel.initialize(zone);
-    this.rawOriginTextLinesTokens = this.rawSavedTextLines.map((_, index) => {
-      const lineNumber = startPosition.lineNumber + index;
-      this.savedModel.tokenization.forceTokenization(lineNumber);
-      const lineTokens = this.savedModel.tokenization.getLineTokens(lineNumber);
-      return lineTokens;
-    });
-  }
-
-  protected processDiffComputation(currentText: string[], computerMode?: EComputerMode): IComputeDiffData {
-    return this.computeDiff(this.rawSavedTextLines, currentText, computerMode);
-  }
-
-  protected getOriginalModel(): ITextModel {
-    return this.savedModel;
-  }
-}
-
-/**
- * Reverse inline stream diff handler
- * In reverse mode, the roles of savedModel and virtualModel are swapped
- */
-@Injectable({ multiple: true })
-export class ReverseInlineStreamDiffHandler extends BaseInlineStreamDiffHandler {
-  recompute(computerMode: EComputerMode, content?: string): IComputeDiffData {
-    const result = super.recompute(computerMode, content);
-    this.rawOriginTextLinesTokens = this.virtualModel.getLinesContent().map((_, index) => {
-      const lineNumber = index + 1;
-      this.virtualModel.tokenization.forceTokenization(lineNumber);
-      const lineTokens = this.virtualModel.tokenization.getLineTokens(lineNumber);
-      return lineTokens;
-    });
-    return result;
-  }
-
-  initialize(): void {
-    const eol = this.savedModel.getEOL();
-
-    // reverse 模式不支持 range
-    this.rawSavedTextLines = this.savedModel.getValue().split(eol);
-
-    // TODO: reverse 模式暂不支持 range
-    const zone = LineRange.fromRangeInclusive(
-      Range.fromPositions(
-        { lineNumber: 1, column: 1 },
-        { lineNumber: this.virtualModel.getLineCount(), column: Number.MAX_SAFE_INTEGER },
-      ),
-    );
-    this.livePreviewDiffDecorationModel.initialize(zone);
-  }
-
-  protected getOriginalModel(): ITextModel {
-    return this.virtualModel;
-  }
-
-  protected processDiffComputation(currentText: string[], computerMode?: EComputerMode): IComputeDiffData {
-    return this.computeDiff(currentText, this.rawSavedTextLines, computerMode);
-  }
-}

package/src/browser/widget/inline-stream-diff/live-preview.decoration.tsx

@@ -7,13 +7,11 @@ import {
   Emitter,
   Event,
   IAIReporter,
-  localize,
   runWhenIdle,
 } from '@opensumi/ide-core-common';
 import { ISingleEditOperation } from '@opensumi/ide-editor';
 import { ICodeEditor, IEditorDecorationsCollection, ITextModel, Position, Range } from '@opensumi/ide-monaco';
 import { StandaloneServices } from '@opensumi/ide-monaco/lib/browser/monaco-api/services';
-import { IMessageService } from '@opensumi/ide-overlay';
 import { EditOperation } from '@opensumi/monaco-editor-core/esm/vs/editor/common/core/editOperation';
 import { LineRange } from '@opensumi/monaco-editor-core/esm/vs/editor/common/core/lineRange';
 import { ModelDecorationOptions } from '@opensumi/monaco-editor-core/esm/vs/editor/common/model/textModel';
@@ -69,9 +67,6 @@ export class LivePreviewDiffDecorationModel extends Disposable {
   @Autowired(InlineDiffService)
   private readonly inlineDiffService: InlineDiffService;
 
-  @Autowired(IMessageService)
-  private readonly messageService: IMessageService;
-
   private activeLineDec: IEditorDecorationsCollection;
   private pendingRangeDec: IEditorDecorationsCollection;
   private aiNativeContextKey: AINativeContextKey;
@@ -92,10 +87,6 @@ export class LivePreviewDiffDecorationModel extends Disposable {
   private removedZoneWidgets: RemovedZoneWidget[] = [];
   private zone: LineRange;
 
-  public get partialEditWidgetCount() {
-    return this.partialEditWidgetList.length;
-  }
-
   constructor(private readonly monacoEditor: ICodeEditor) {
     super();
     this.model = this.monacoEditor.getModel()!;
@@ -625,7 +616,7 @@ export class LivePreviewDiffDecorationModel extends Disposable {
     if (this.options.renderRemovedWidgetImmediately) {
       run();
     } else {
-      this.addDispose(runWhenIdle(run));
+      runWhenIdle(run);
     }
   }
 
@@ -690,26 +681,6 @@ export class LivePreviewDiffDecorationModel extends Disposable {
     }
   }
 
-  currentChangeIndex: number = 0;
-
-  revealSiblingChange(direction: 'up' | 'down') {
-    this.currentChangeIndex = this.currentChangeIndex + (direction === 'up' ? -1 : 1);
-    if (this.currentChangeIndex >= 0 && this.currentChangeIndex < this.partialEditWidgetList.length) {
-      const siblingChange = this.partialEditWidgetList[this.currentChangeIndex];
-      const pos = siblingChange.getPosition();
-      if (pos?.position) {
-        this.monacoEditor.revealLineInCenter(pos.position!.lineNumber);
-        return this.currentChangeIndex;
-      }
-    } else {
-      this.messageService.info(
-        direction === 'up'
-          ? localize('aiNative.inlineDiff.noMoreChangesUp')
-          : localize('aiNative.inlineDiff.noMoreChangesDown'),
-      );
-    }
-  }
-
   setPreviewerOptions(options: IModelOptions) {
     this.options = options;
   }

package/src/common/index.ts

@@ -19,6 +19,8 @@ import { IPosition, ITextModel, InlineCompletionContext } from '@opensumi/ide-mo
 import { MCPServerDescription } from './mcp-server-manager';
 import { MCPTool } from './types';
 
+export * from './model';
+
 export const IAINativeService = Symbol('IAINativeService');
 
 /**

package/src/common/model.ts

@@ -0,0 +1,90 @@
+export interface ModelInfo {
+  maxTokens?: number;
+  contextWindow?: number;
+  supportsImages?: boolean;
+  temperature?: number;
+  topP?: number;
+  topK?: number;
+}
+
+export const deepSeekModels = {
+  'deepseek-chat': {
+    maxTokens: 8_000,
+    contextWindow: 64_000,
+    supportsImages: false,
+  },
+  'deepseek-reasoner': {
+    maxTokens: 8_000,
+    contextWindow: 64_000,
+    supportsImages: false,
+    temperature: 0.7,
+    topP: 0.95,
+  },
+} as Record<string, ModelInfo>;
+
+export const anthropicModels = {
+  'claude-3-7-sonnet-20250219': {
+    maxTokens: 8192,
+    contextWindow: 200_000,
+    supportsImages: true,
+  },
+  'claude-3-5-sonnet-20241022': {
+    maxTokens: 8192,
+    contextWindow: 200_000,
+    supportsImages: true,
+  },
+  'claude-3-5-haiku-20241022': {
+    maxTokens: 8192,
+    contextWindow: 200_000,
+    supportsImages: false,
+  },
+  'claude-3-opus-20240229': {
+    maxTokens: 4096,
+    contextWindow: 200_000,
+    supportsImages: true,
+  },
+  'claude-3-haiku-20240307': {
+    maxTokens: 4096,
+    contextWindow: 200_000,
+    supportsImages: true,
+  },
+} as Record<string, ModelInfo>; // as const assertion makes the object deeply readonly
+
+export const openAiNativeModels = {
+  'o3-mini': {
+    maxTokens: 100_000,
+    contextWindow: 200_000,
+    supportsImages: false,
+  },
+  // don't support tool use yet
+  o1: {
+    maxTokens: 100_000,
+    contextWindow: 200_000,
+    supportsImages: true,
+  },
+  'o1-preview': {
+    maxTokens: 32_768,
+    contextWindow: 128_000,
+    supportsImages: true,
+  },
+  'o1-mini': {
+    maxTokens: 65_536,
+    contextWindow: 128_000,
+    supportsImages: true,
+  },
+  'gpt-4o': {
+    maxTokens: 4_096,
+    contextWindow: 128_000,
+    supportsImages: true,
+  },
+  'gpt-4o-mini': {
+    maxTokens: 16_384,
+    contextWindow: 128_000,
+    supportsImages: true,
+  },
+  'gpt-4.5-preview': {
+    maxTokens: 16_384,
+    contextWindow: 128_000,
+    supportsImages: true,
+  },
+} as Record<string, ModelInfo>;

package/src/common/types.ts

@@ -52,7 +52,6 @@ export interface MCPTool {
 export interface CodeBlockData {
   toolCallId: string;
   codeEdit: string;
-  originalCode: string;
   updatedCode?: string;
   relativePath: string;
   status: CodeBlockStatus;

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

@@ -4,6 +4,7 @@ import { Injectable } from '@opensumi/di';
 import { IAIBackServiceOption } from '@opensumi/ide-core-common';
 import { AINativeSettingSectionsId } from '@opensumi/ide-core-common/lib/settings/ai-native';
 
+import { ModelInfo, anthropicModels } from '../../common/model';
 import { BaseLanguageModel } from '../base-language-model';
 
 export const AnthropicModelIdentifier = Symbol('AnthropicModelIdentifier');
@@ -19,7 +20,11 @@ export class AnthropicModel extends BaseLanguageModel {
     return createAnthropic({ apiKey });
   }
 
-  protected getModelIdentifier(provider: AnthropicProvider) {
-    return provider('claude-3-5-sonnet-20241022');
+  protected getModelIdentifier(provider: AnthropicProvider, modelId = 'claude-3-5-sonnet-20241022') {
+    return provider(modelId);
+  }
+
+  protected getModelInfo(modelId: string): ModelInfo | undefined {
+    return anthropicModels[modelId];
   }
 }

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

@@ -5,6 +5,7 @@ import { ChatMessageRole, IAIBackServiceOption, IChatMessage } from '@opensumi/i
 import { ChatReadableStream } from '@opensumi/ide-core-node';
 import { CancellationToken } from '@opensumi/ide-utils';
 
+import { ModelInfo } from '../common';
 import {
   IToolInvocationRegistryManager,
   ToolInvocationRegistryManager,
@@ -13,6 +14,8 @@ import {
 
 @Injectable()
 export abstract class BaseLanguageModel {
+  static ModelOptions: Record<string, ModelInfo>;
+
   @Autowired(ToolInvocationRegistryManager)
   protected readonly toolInvocationRegistryManager: IToolInvocationRegistryManager;
 
@@ -56,9 +59,6 @@ export abstract class BaseLanguageModel {
       chatReadableStream,
       options.history || [],
       options.modelId,
-      options.temperature,
-      options.topP,
-      options.topK,
       options.providerOptions,
       options.trimTexts,
       options.system,
@@ -78,6 +78,8 @@ export abstract class BaseLanguageModel {
 
   protected abstract getModelIdentifier(provider: any, modelId?: string): any;
 
+  protected abstract getModelInfo(modelId: string): ModelInfo | undefined;
+
   protected async handleStreamingRequest(
     provider: any,
     request: string,
@@ -85,9 +87,6 @@ export abstract class BaseLanguageModel {
     chatReadableStream: ChatReadableStream,
     history: IChatMessage[] = [],
     modelId?: string,
-    temperature?: number,
-    topP?: number,
-    topK?: number,
     providerOptions?: Record<string, any>,
     trimTexts?: [string, string],
     systemPrompt?: string,
@@ -110,6 +109,7 @@ export abstract class BaseLanguageModel {
         })),
         { role: 'user', content: request },
       ];
+      const modelInfo = modelId ? this.getModelInfo(modelId) : undefined;
       const stream = streamText({
         model: this.getModelIdentifier(provider, modelId),
         tools: aiTools,
@@ -117,9 +117,9 @@ export abstract class BaseLanguageModel {
         abortSignal: abortController.signal,
         experimental_toolCallStreaming: true,
         maxSteps: 12,
-        temperature,
-        topP: topP || 0.8,
-        topK: topK || 1,
+        temperature: modelInfo?.temperature || 0,
+        topP: modelInfo?.topP || 0.8,
+        topK: modelInfo?.topK || 1,
         system: systemPrompt,
         providerOptions,
       });

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

@@ -4,6 +4,7 @@ import { Injectable } from '@opensumi/di';
 import { IAIBackServiceOption } from '@opensumi/ide-core-common';
 import { AINativeSettingSectionsId } from '@opensumi/ide-core-common/lib/settings/ai-native';
 
+import { ModelInfo, deepSeekModels } from '../../common';
 import { BaseLanguageModel } from '../base-language-model';
 
 export const DeepSeekModelIdentifier = Symbol('DeepSeekModelIdentifier');
@@ -19,7 +20,11 @@ export class DeepSeekModel extends BaseLanguageModel {
     return createDeepSeek({ apiKey });
   }
 
-  protected getModelIdentifier(provider: DeepSeekProvider) {
-    return provider('deepseek-chat');
+  protected getModelIdentifier(provider: DeepSeekProvider, modelId = 'deepseek-chat') {
+    return provider(modelId);
+  }
+
+  protected getModelInfo(modelId: string): ModelInfo | undefined {
+    return deepSeekModels[modelId];
   }
 }

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

@@ -1,28 +1,29 @@
-import { OpenAICompatibleProvider, createOpenAICompatible } from '@ai-sdk/openai-compatible';
+import { OpenAIProvider, createOpenAI } from '@ai-sdk/openai';
 import { LanguageModelV1 } from 'ai';
 
 import { Injectable } from '@opensumi/di';
 import { AINativeSettingSectionsId, IAIBackServiceOption } from '@opensumi/ide-core-common';
 
+import { ModelInfo, openAiNativeModels } from '../../common/model';
 import { BaseLanguageModel } from '../base-language-model';
 
-export const DeepSeekModelIdentifier = Symbol('DeepSeekModelIdentifier');
-
 @Injectable()
 export class OpenAIModel extends BaseLanguageModel {
-  protected initializeProvider(options: IAIBackServiceOption): OpenAICompatibleProvider {
+  protected initializeProvider(options: IAIBackServiceOption): OpenAIProvider {
     const apiKey = options.apiKey;
     if (!apiKey) {
       throw new Error(`Please provide OpenAI API Key in preferences (${AINativeSettingSectionsId.OpenaiApiKey})`);
     }
-    return createOpenAICompatible({
+    return createOpenAI({
       apiKey,
-      baseURL: options.baseURL || 'https://dashscope.aliyuncs.com/compatible-mode/v1',
-      name: 'openai',
     });
   }
 
-  protected getModelIdentifier(provider: OpenAICompatibleProvider, modelId?: string): LanguageModelV1 {
-    return provider(modelId || 'qwen-max');
+  protected getModelIdentifier(provider: OpenAIProvider, modelId: string) {
+    return provider(modelId) as LanguageModelV1;
+  }
+
+  protected getModelInfo(modelId: string): ModelInfo | undefined {
+    return openAiNativeModels[modelId];
   }
 }