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

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

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