@opensumi/ide-ai-native 3.7.2-next-1739848467.0 → 3.7.2-next-1739945875.0

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

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