@opensumi/ide-ai-native 3.8.1-next-1741250387.0 → 3.8.1-next-1741251026.0

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

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

@@ -12,13 +12,18 @@ import {
   IMarkdownString,
   Uri,
 } from '@opensumi/ide-core-common';
-import { IChatMessage } from '@opensumi/ide-core-common/lib/types/ai-native';
 import { DESIGN_MENUBAR_CONTAINER_VIEW_ID } from '@opensumi/ide-design/lib/common/constants';
 import { IPosition, ITextModel, InlineCompletionContext } from '@opensumi/ide-monaco/lib/common';
 
 import { MCPServerDescription } from './mcp-server-manager';
 import { MCPTool } from './types';
 
+import type { CoreMessage } from 'ai';
+
+export type { CoreMessage };
+
+export * from './model';
+
 export const IAINativeService = Symbol('IAINativeService');
 
 /**
@@ -31,6 +36,9 @@ export const AI_CHAT_CONTAINER_ID = 'AI-Chat-Container';
 export const AI_CHAT_LOGO_AVATAR_ID = 'AI-Chat-Logo-Avatar';
 export const AI_MENU_BAR_DEBUG_TOOLBAR = 'AI_MENU_BAR_DEBUG_TOOLBAR';
 
+// 内置 MCP 服务器名称
+export const BUILTIN_MCP_SERVER_NAME = 'sumi-builtin';
+
 /**
  * @deprecated Use {@link DESIGN_MENUBAR_CONTAINER_VIEW_ID} instead
  */
@@ -123,9 +131,12 @@ export const ChatProxyServiceToken = Symbol('ChatProxyServiceToken');
 export const TokenMCPServerProxyService = Symbol('TokenMCPServerProxyService');
 
 export interface ISumiMCPServerBackend {
-  initBuiltinMCPServer(): void;
+  initBuiltinMCPServer(enabled: boolean): void;
   initExternalMCPServers(servers: MCPServerDescription[]): void;
   getAllMCPTools(): Promise<MCPTool[]>;
+  getServers(): Promise<Array<{ name: string; isStarted: boolean }>>;
+  startServer(serverName: string): Promise<void>;
+  stopServer(serverName: string): Promise<void>;
 }
 
 export const SumiMCPServerProxyServicePath = 'SumiMCPServerProxyServicePath';
@@ -138,7 +149,7 @@ export interface IChatAgentService {
     id: string,
     request: IChatAgentRequest,
     progress: (part: IChatProgress) => void,
-    history: IChatMessage[],
+    history: CoreMessage[],
     token: CancellationToken,
   ): Promise<IChatAgentResult>;
   getAgents(): Array<IChatAgent>;
@@ -159,7 +170,7 @@ export interface IChatAgent extends IChatAgentData {
   invoke(
     request: IChatAgentRequest,
     progress: (part: IChatProgress) => void,
-    history: IChatMessage[],
+    history: CoreMessage[],
     token: CancellationToken,
   ): Promise<IChatAgentResult>;
   provideFollowups?(sessionId: string, token: CancellationToken): Promise<IChatFollowup[]>;

package/src/common/llm-context.ts

@@ -15,13 +15,13 @@ export interface LLMContextService {
    */
   cleanFileContext(): void;
 
-  onDidContextFilesChangeEvent: Event<FileContext[]>;
+  onDidContextFilesChangeEvent: Event<{ viewed: FileContext[]; attached: FileContext[]; version: number }>;
 
   /**
    * 从 context 中移除文件
    * @param uri URI
    */
-  removeFileFromContext(uri: URI): void;
+  removeFileFromContext(uri: URI, isManual?: boolean): void;
 
   /** 导出为可序列化格式 */
   serialize(): SerializedContext;
@@ -30,12 +30,18 @@ export interface LLMContextService {
 export interface FileContext {
   uri: URI;
   selection?: [number, number];
-  isManual: boolean;
 }
 
 export const LLMContextServiceToken = Symbol('LLMContextService');
 
+export interface AttachFileContext {
+  content: string;
+  lineErrors: string[];
+  path: string;
+  language: string;
+}
+
 export interface SerializedContext {
   recentlyViewFiles: string[];
-  attachedFiles: Array<{ content: string; lineErrors: string[]; path: string; language: string }>;
+  attachedFiles: Array<AttachFileContext>;
 }

package/src/common/mcp-server-manager.ts

@@ -1,5 +1,15 @@
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 
+export interface IMCPServer {
+  isStarted(): boolean;
+  start(): Promise<void>;
+  getServerName(): string;
+  callTool(toolName: string, toolCallId: string, arg_string: string): ReturnType<Client['callTool']>;
+  getTools(): ReturnType<Client['listTools']>;
+  update(command: string, args?: string[], env?: { [key: string]: string }): void;
+  stop(): void;
+}
+
 export interface MCPServerManager {
   callTool(
     serverName: string,
@@ -11,7 +21,7 @@ export interface MCPServerManager {
   addOrUpdateServer(description: MCPServerDescription): void;
   // invoke in node.js only
   addOrUpdateServerDirectly(server: any): void;
-  initBuiltinServer(builtinMCPServer: any): void;
+  initBuiltinServer(builtinMCPServer: any, enabled: boolean): void;
   getTools(serverName: string): ReturnType<Client['listTools']>;
   getServerNames(): Promise<string[]>;
   startServer(serverName: string): Promise<void>;
@@ -19,6 +29,7 @@ export interface MCPServerManager {
   getStartedServers(): Promise<string[]>;
   registerTools(serverName: string): Promise<void>;
   addExternalMCPServers(servers: MCPServerDescription[]): void;
+  getServers(): Map<string, IMCPServer>;
 }
 
 export type MCPTool = Awaited<ReturnType<MCPServerManager['getTools']>>['tools'][number];
@@ -45,6 +56,11 @@ export interface MCPServerDescription {
    * Optional environment variables to set when starting the server.
    */
   env?: { [key: string]: string };
+
+  /**
+   * Whether to enable the MCP server.
+   */
+  enabled?: boolean;
 }
 
 export const MCPServerManager = Symbol('MCPServerManager');

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