@theia/ai-vercel-ai 1.62.0

package/package.json

@@ -0,0 +1,56 @@
+{
+  "name": "@theia/ai-vercel-ai",
+  "version": "1.62.0",
+  "description": "Theia - Vercel AI SDK Integration",
+  "dependencies": {
+    "@ai-sdk/anthropic": "^1.2.10",
+    "@ai-sdk/openai": "^1.3.21",
+    "@ai-sdk/provider": "^1.1.3",
+    "@theia/ai-core": "1.62.0",
+    "@theia/core": "1.62.0",
+    "@theia/filesystem": "1.62.0",
+    "@theia/workspace": "1.62.0",
+    "ai": "^4.3.13",
+    "tslib": "^2.6.2"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "theiaExtensions": [
+    {
+      "frontend": "lib/browser/vercel-ai-frontend-module",
+      "backend": "lib/node/vercel-ai-backend-module"
+    }
+  ],
+  "keywords": [
+    "theia-extension"
+  ],
+  "license": "EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/eclipse-theia/theia.git"
+  },
+  "bugs": {
+    "url": "https://github.com/eclipse-theia/theia/issues"
+  },
+  "homepage": "https://github.com/eclipse-theia/theia",
+  "files": [
+    "lib",
+    "src"
+  ],
+  "scripts": {
+    "build": "theiaext build",
+    "clean": "theiaext clean",
+    "compile": "theiaext compile",
+    "lint": "theiaext lint",
+    "test": "theiaext test",
+    "watch": "theiaext watch"
+  },
+  "devDependencies": {
+    "@theia/ext-scripts": "1.62.0"
+  },
+  "nyc": {
+    "extends": "../../configs/nyc.json"
+  },
+  "gitHead": "a4e035ccc26fde24c2769a466cb18e3b54c517c5"
+}

package/src/browser/vercel-ai-frontend-application-contribution.ts

@@ -0,0 +1,196 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { FrontendApplicationContribution, PreferenceService, PreferenceChange } from '@theia/core/lib/browser';
+import { inject, injectable } from '@theia/core/shared/inversify';
+import { VercelAiLanguageModelsManager, VercelAiModelDescription, VercelAiProvider } from '../common';
+import { ANTHROPIC_API_KEY_PREF, CUSTOM_ENDPOINTS_PREF, MODELS_PREF, OPENAI_API_KEY_PREF, VERCEL_AI_PROVIDER_ID } from './vercel-ai-preferences';
+
+interface ModelConfig {
+    id: string;
+    model: string;
+    provider: VercelAiProvider;
+}
+
+@injectable()
+export class VercelAiFrontendApplicationContribution implements FrontendApplicationContribution {
+
+    @inject(PreferenceService)
+    protected preferenceService: PreferenceService;
+
+    @inject(VercelAiLanguageModelsManager)
+    protected manager: VercelAiLanguageModelsManager;
+
+    onStart(): void {
+        this.preferenceService.ready.then(() => {
+            // Set up provider-specific API keys
+            const openaiApiKey = this.preferenceService.get<string>(OPENAI_API_KEY_PREF, undefined);
+            const anthropicApiKey = this.preferenceService.get<string>(ANTHROPIC_API_KEY_PREF, undefined);
+
+            // Set provider configs
+            if (openaiApiKey) {
+                this.manager.setProviderConfig('openai', { provider: 'openai', apiKey: openaiApiKey });
+            }
+
+            if (anthropicApiKey) {
+                this.manager.setProviderConfig('anthropic', { provider: 'anthropic', apiKey: anthropicApiKey });
+            }
+
+            // Initial setup of models
+            const models = this.preferenceService.get<ModelConfig[]>(MODELS_PREF, []);
+            this.manager.createOrUpdateLanguageModels(...models.map(model => this.createVercelAiModelDescription(model)));
+
+            const customModels = this.preferenceService.get<Partial<VercelAiModelDescription>[]>(CUSTOM_ENDPOINTS_PREF, []);
+            this.manager.createOrUpdateLanguageModels(...this.createCustomModelDescriptionsFromPreferences(customModels));
+
+            // Set up listeners for preference changes
+            this.preferenceService.onPreferenceChanged(this.handlePreferenceChange.bind(this));
+        });
+    }
+
+    protected handlePreferenceChange(event: PreferenceChange): void {
+        switch (event.preferenceName) {
+            case OPENAI_API_KEY_PREF:
+                this.manager.setProviderConfig('openai', { provider: 'openai', apiKey: event.newValue });
+                break;
+            case ANTHROPIC_API_KEY_PREF:
+                this.manager.setProviderConfig('anthropic', { provider: 'anthropic', apiKey: event.newValue });
+                break;
+            case MODELS_PREF:
+                this.handleModelChanges(event);
+                break;
+            case CUSTOM_ENDPOINTS_PREF:
+                this.handleCustomModelChanges(event);
+                break;
+        }
+    }
+
+    protected handleModelChanges(event: PreferenceChange): void {
+        const newModels = this.ensureModelConfigArray(event.newValue);
+        const oldModels = this.ensureModelConfigArray(event.oldValue);
+
+        const oldModelIds = new Set(oldModels.map(m => m.id));
+        const newModelIds = new Set(newModels.map(m => m.id));
+
+        const modelsToRemove = [...oldModelIds].filter(modelId => !newModelIds.has(modelId));
+        const modelsToAdd = newModels.filter(model => !oldModelIds.has(model.id));
+
+        this.manager.removeLanguageModels(...modelsToRemove);
+        this.manager.createOrUpdateLanguageModels(...modelsToAdd.map(model => this.createVercelAiModelDescription(model)));
+    }
+
+    protected handleCustomModelChanges(event: PreferenceChange): void {
+        const newCustomModels = this.ensureCustomModelArray(event.newValue);
+        const oldCustomModels = this.ensureCustomModelArray(event.oldValue);
+
+        const oldModels = this.createCustomModelDescriptionsFromPreferences(oldCustomModels);
+        const newModels = this.createCustomModelDescriptionsFromPreferences(newCustomModels);
+
+        const modelsToRemove = oldModels.filter(model => !newModels.some(newModel => newModel.id === model.id));
+        const modelsToAddOrUpdate = newModels.filter(newModel =>
+            !oldModels.some(model =>
+                model.id === newModel.id &&
+                model.model === newModel.model &&
+                model.url === newModel.url &&
+                model.apiKey === newModel.apiKey &&
+                model.supportsStructuredOutput === newModel.supportsStructuredOutput &&
+                model.enableStreaming === newModel.enableStreaming &&
+                model.provider === newModel.provider));
+
+        this.manager.removeLanguageModels(...modelsToRemove.map(model => model.id));
+        this.manager.createOrUpdateLanguageModels(...modelsToAddOrUpdate);
+    }
+
+    protected ensureModelConfigArray(value: unknown): ModelConfig[] {
+        if (!value || !Array.isArray(value)) {
+            return [];
+        }
+
+        return value.filter(item =>
+            item &&
+            typeof item === 'object' &&
+            'id' in item &&
+            'model' in item &&
+            'provider' in item &&
+            typeof item.id === 'string' &&
+            typeof item.model === 'string' &&
+            (typeof item.provider === 'string' || item.provider === undefined)
+        ) as ModelConfig[];
+    }
+
+    protected ensureCustomModelArray(value: unknown): Partial<VercelAiModelDescription>[] {
+        if (!value || !Array.isArray(value)) {
+            return [];
+        }
+
+        return value.filter(item =>
+            item &&
+            typeof item === 'object'
+        ) as Partial<VercelAiModelDescription>[];
+    }
+
+    protected createVercelAiModelDescription(modelInfo: ModelConfig): VercelAiModelDescription {
+        // The model ID already includes the 'vercel' prefix from preferences
+        return {
+            id: modelInfo.id,
+            model: modelInfo.model,
+            provider: modelInfo.provider,
+            apiKey: true,
+            enableStreaming: true,
+            supportsStructuredOutput: modelsSupportingStructuredOutput.includes(modelInfo.model)
+        };
+    }
+
+    protected createCustomModelDescriptionsFromPreferences(
+        preferences: Partial<VercelAiModelDescription>[]
+    ): VercelAiModelDescription[] {
+        return preferences.reduce((acc, pref) => {
+            if (!pref.model || !pref.url || typeof pref.model !== 'string' || typeof pref.url !== 'string') {
+                return acc;
+            }
+
+            // Ensure custom model IDs have the 'vercel' prefix
+            const modelId = pref.id && typeof pref.id === 'string' ? pref.id : pref.model;
+            const prefixedId = modelId.startsWith('vercel/') ? modelId : `${VERCEL_AI_PROVIDER_ID}/${modelId}`;
+
+            return [
+                ...acc,
+                {
+                    id: prefixedId,
+                    model: pref.model,
+                    url: pref.url,
+                    provider: pref.provider || 'openai',
+                    apiKey: typeof pref.apiKey === 'string' || pref.apiKey === true ? pref.apiKey : undefined,
+                    supportsStructuredOutput: pref.supportsStructuredOutput ?? true,
+                    enableStreaming: pref.enableStreaming ?? true
+                }
+            ];
+        }, []);
+    }
+}
+
+// List of models that support structured output via JSON schema
+const modelsSupportingStructuredOutput = [
+    'gpt-4.1',
+    'gpt-4.1-mini',
+    'gpt-4.1-nano',
+    'gpt-4o',
+    'gpt-4o-mini',
+    'gpt-4-turbo',
+    'claude-3-7-sonnet-20250219',
+    'claude-3-5-haiku-20241022',
+    'claude-3-opus-20240229'
+];

package/src/browser/vercel-ai-frontend-module.ts

@@ -0,0 +1,31 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { ContainerModule } from '@theia/core/shared/inversify';
+import { VercelAiPreferencesSchema } from './vercel-ai-preferences';
+import { FrontendApplicationContribution, PreferenceContribution, RemoteConnectionProvider, ServiceConnectionProvider } from '@theia/core/lib/browser';
+import { VercelAiFrontendApplicationContribution } from './vercel-ai-frontend-application-contribution';
+import { VERCEL_AI_LANGUAGE_MODELS_MANAGER_PATH, VercelAiLanguageModelsManager } from '../common';
+
+export default new ContainerModule(bind => {
+    bind(PreferenceContribution).toConstantValue({ schema: VercelAiPreferencesSchema });
+    bind(VercelAiFrontendApplicationContribution).toSelf().inSingletonScope();
+    bind(FrontendApplicationContribution).toService(VercelAiFrontendApplicationContribution);
+    bind(VercelAiLanguageModelsManager).toDynamicValue(ctx => {
+        const provider = ctx.container.get<ServiceConnectionProvider>(RemoteConnectionProvider);
+        return provider.createProxy<VercelAiLanguageModelsManager>(VERCEL_AI_LANGUAGE_MODELS_MANAGER_PATH);
+    }).inSingletonScope();
+});

package/src/browser/vercel-ai-preferences.ts

@@ -0,0 +1,139 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { PreferenceSchema } from '@theia/core/lib/browser/preferences/preference-contribution';
+import { AI_CORE_PREFERENCES_TITLE } from '@theia/ai-core/lib/browser/ai-core-preferences';
+import { nls } from '@theia/core';
+
+export const OPENAI_API_KEY_PREF = 'ai-features.vercelAi.openaiApiKey';
+export const ANTHROPIC_API_KEY_PREF = 'ai-features.vercelAi.anthropicApiKey';
+export const MODELS_PREF = 'ai-features.vercelAi.officialModels';
+export const CUSTOM_ENDPOINTS_PREF = 'ai-features.vercelAi.customModels';
+
+export const VERCEL_AI_PROVIDER_ID = 'vercel-ai';
+
+export const VercelAiPreferencesSchema: PreferenceSchema = {
+    type: 'object',
+    properties: {
+        [OPENAI_API_KEY_PREF]: {
+            type: 'string',
+            markdownDescription: nls.localize('theia/ai/vercelai/openaiApiKey/mdDescription',
+                'Enter an API Key for OpenAI models used by the Vercel AI SDK. \
+                **Please note:** By using this preference the API key will be stored in clear text \
+on the machine running Theia. Use the environment variable `OPENAI_API_KEY` to set the key securely.'),
+            title: AI_CORE_PREFERENCES_TITLE,
+        },
+        [ANTHROPIC_API_KEY_PREF]: {
+            type: 'string',
+            markdownDescription: nls.localize('theia/ai/vercelai/anthropicApiKey/mdDescription',
+                'Enter an API Key for Anthropic models used by the Vercel AI SDK. \
+                **Please note:** By using this preference the API key will be stored in clear text \
+on the machine running Theia. Use the environment variable `ANTHROPIC_API_KEY` to set the key securely.'),
+            title: AI_CORE_PREFERENCES_TITLE,
+        },
+        [MODELS_PREF]: {
+            type: 'array',
+            description: nls.localize('theia/ai/vercelai/models/description', 'Official models to use with Vercel AI SDK'),
+            title: AI_CORE_PREFERENCES_TITLE,
+            default: [
+                { id: 'vercel/openai/gpt-4.1', model: 'gpt-4.1', provider: 'openai' },
+                { id: 'vercel/openai/gpt-4.1-nano', model: 'gpt-4.1-nano', provider: 'openai' },
+                { id: 'vercel/openai/gpt-4.1-mini', model: 'gpt-4.1-mini', provider: 'openai' },
+                { id: 'vercel/openai/gpt-4-turbo', model: 'gpt-4-turbo', provider: 'openai' },
+                { id: 'vercel/openai/gpt-4o', model: 'gpt-4o', provider: 'openai' },
+                { id: 'vercel/openai/gpt-4o-mini', model: 'gpt-4o-mini', provider: 'openai' },
+                { id: 'vercel/anthropic/claude-3-7-sonnet-20250219', model: 'claude-3-7-sonnet-20250219', provider: 'anthropic' },
+                { id: 'vercel/anthropic/claude-3-5-haiku-20241022', model: 'claude-3-5-haiku-20241022', provider: 'anthropic' },
+                { id: 'vercel/anthropic/claude-3-opus-20240229', model: 'claude-3-opus-20240229', provider: 'anthropic' }
+            ],
+            items: {
+                type: 'object',
+                properties: {
+                    id: {
+                        type: 'string',
+                        title: nls.localize('theia/ai/vercelai/models/id/title', 'Model ID')
+                    },
+                    model: {
+                        type: 'string',
+                        title: nls.localize('theia/ai/vercelai/models/model/title', 'Model Name')
+                    },
+                    provider: {
+                        type: 'string',
+                        enum: ['openai', 'anthropic'],
+                        title: nls.localize('theia/ai/vercelai/models/provider/title', 'Provider')
+                    }
+                },
+                required: ['id', 'model', 'provider']
+            }
+        },
+        [CUSTOM_ENDPOINTS_PREF]: {
+            type: 'array',
+            title: AI_CORE_PREFERENCES_TITLE,
+            markdownDescription: nls.localize('theia/ai/vercelai/customEndpoints/mdDescription',
+                'Integrate custom models compatible with the Vercel AI SDK. The required attributes are `model` and `url`.\
+            \n\
+            Optionally, you can\
+            \n\
+            - specify a unique `id` to identify the custom model in the UI. If none is given `model` will be used as `id`.\
+            \n\
+            - provide an `apiKey` to access the API served at the given url. Use `true` to indicate the use of the global API key.\
+            \n\
+            - specify `supportsStructuredOutput: false` to indicate that structured output shall not be used.\
+            \n\
+            - specify `enableStreaming: false` to indicate that streaming shall not be used.\
+            \n\
+            - specify `provider` to indicate which provider the model is from (openai, anthropic).'),
+            default: [],
+            items: {
+                type: 'object',
+                properties: {
+                    model: {
+                        type: 'string',
+                        title: nls.localize('theia/ai/vercelai/customEndpoints/modelId/title', 'Model ID')
+                    },
+                    url: {
+                        type: 'string',
+                        title: nls.localize('theia/ai/vercelai/customEndpoints/url/title', 'The API endpoint where the model is hosted')
+                    },
+                    id: {
+                        type: 'string',
+                        title: nls.localize('theia/ai/vercelai/customEndpoints/id/title', 'A unique identifier which is used in the UI to identify the custom model'),
+                    },
+                    provider: {
+                        type: 'string',
+                        enum: ['openai', 'anthropic'],
+                        title: nls.localize('theia/ai/vercelai/customEndpoints/provider/title', 'Provider')
+                    },
+                    apiKey: {
+                        type: ['string', 'boolean'],
+                        title: nls.localize('theia/ai/vercelai/customEndpoints/apiKey/title',
+                            'Either the key to access the API served at the given url or `true` to use the global API key'),
+                    },
+                    supportsStructuredOutput: {
+                        type: 'boolean',
+                        title: nls.localize('theia/ai/vercelai/customEndpoints/supportsStructuredOutput/title',
+                            'Indicates whether the model supports structured output. `true` by default.'),
+                    },
+                    enableStreaming: {
+                        type: 'boolean',
+                        title: nls.localize('theia/ai/vercelai/customEndpoints/enableStreaming/title',
+                            'Indicates whether the streaming API shall be used. `true` by default.'),
+                    }
+                }
+            }
+        }
+    }
+};

package/src/common/index.ts

@@ -0,0 +1,16 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+export * from './vercel-ai-language-models-manager';

package/src/common/vercel-ai-language-models-manager.ts

@@ -0,0 +1,63 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+export const VERCEL_AI_LANGUAGE_MODELS_MANAGER_PATH = '/services/vercel-ai/language-model-manager';
+
+export type VercelAiProvider = 'openai' | 'anthropic';
+
+export interface VercelAiProviderConfig {
+    provider: VercelAiProvider;
+    apiKey?: string;
+    baseURL?: string;
+}
+
+export interface VercelAiModelDescription {
+    /**
+     * The identifier of the model which will be shown in the UI.
+     */
+    id: string;
+    /**
+     * The model ID as used by the Vercel AI SDK.
+     */
+    model: string;
+    /**
+     * The provider of the model (openai, anthropic, etc.)
+     */
+    provider?: VercelAiProvider;
+    /**
+     * The API base URL where the model is hosted. If not provided the default provider endpoint will be used.
+     */
+    url?: string;
+    /**
+     * The key for the model. If 'true' is provided the global provider API key will be used.
+     */
+    apiKey: string | true | undefined;
+    /**
+     * Controls whether streaming is enabled for this model.
+     */
+    enableStreaming: boolean;
+    /**
+     * Flag to configure whether the model supports structured output. Default is `true`.
+     */
+    supportsStructuredOutput: boolean;
+}
+
+export const VercelAiLanguageModelsManager = Symbol('VercelAiLanguageModelsManager');
+export interface VercelAiLanguageModelsManager {
+    apiKey: string | undefined;
+    setProviderConfig(provider: VercelAiProvider, config: Partial<VercelAiProviderConfig>): void;
+    createOrUpdateLanguageModels(...models: VercelAiModelDescription[]): Promise<void>;
+    removeLanguageModels(...modelIds: string[]): void;
+}

package/src/node/vercel-ai-backend-module.ts

@@ -0,0 +1,35 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { ContainerModule } from '@theia/core/shared/inversify';
+import { VERCEL_AI_LANGUAGE_MODELS_MANAGER_PATH, VercelAiLanguageModelsManager } from '../common/vercel-ai-language-models-manager';
+import { ConnectionHandler, RpcConnectionHandler } from '@theia/core';
+import { VercelAiLanguageModelsManagerImpl } from './vercel-ai-language-models-manager-impl';
+import { ConnectionContainerModule } from '@theia/core/lib/node/messaging/connection-container-module';
+import { VercelAiLanguageModelFactory } from './vercel-ai-language-model-factory';
+
+const vercelAiConnectionModule = ConnectionContainerModule.create(({ bind, bindBackendService, bindFrontendService }) => {
+    bind(VercelAiLanguageModelsManagerImpl).toSelf().inSingletonScope();
+    bind(VercelAiLanguageModelsManager).toService(VercelAiLanguageModelsManagerImpl);
+    bind(VercelAiLanguageModelFactory).toSelf().inSingletonScope();
+    bind(ConnectionHandler).toDynamicValue(ctx =>
+        new RpcConnectionHandler(VERCEL_AI_LANGUAGE_MODELS_MANAGER_PATH, () => ctx.container.get(VercelAiLanguageModelsManager))
+    ).inSingletonScope();
+});
+
+export default new ContainerModule(bind => {
+    bind(ConnectionContainerModule).toConstantValue(vercelAiConnectionModule);
+});

package/src/node/vercel-ai-language-model-factory.ts

@@ -0,0 +1,82 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { createAnthropic } from '@ai-sdk/anthropic';
+import { createOpenAI } from '@ai-sdk/openai';
+import { LanguageModelV1 } from '@ai-sdk/provider';
+import { injectable } from '@theia/core/shared/inversify';
+import { VercelAiModelDescription } from '../common';
+
+export type VercelAiProvider = 'openai' | 'anthropic';
+
+export interface VercelAiProviderConfig {
+    provider: VercelAiProvider;
+    apiKey?: string;
+    baseURL?: string;
+}
+
+@injectable()
+export class VercelAiLanguageModelFactory {
+
+    createLanguageModel(modelDescription: VercelAiModelDescription, providerConfig: VercelAiProviderConfig): LanguageModelV1 {
+        const apiKey = this.resolveApiKey(modelDescription, providerConfig);
+        if (!apiKey) {
+            throw new Error(`Please provide an API key for ${providerConfig.provider} in preferences or via environment variable`);
+        }
+
+        const baseURL = modelDescription.url || providerConfig.baseURL;
+
+        switch (providerConfig.provider) {
+            case 'openai':
+                return createOpenAI({
+                    apiKey,
+                    baseURL,
+                    compatibility: 'strict'
+                }).languageModel(modelDescription.model);
+            case 'anthropic':
+                return createAnthropic({
+                    apiKey,
+                    baseURL
+                }).languageModel(modelDescription.model);
+            default:
+                throw new Error(`Unsupported provider: ${providerConfig.provider}`);
+        }
+    }
+
+    private resolveApiKey(modelDescription: VercelAiModelDescription, providerConfig: VercelAiProviderConfig): string | undefined {
+        if (modelDescription.apiKey === true) {
+            return this.getApiKeyBasedOnProvider(providerConfig);
+        }
+        if (modelDescription.apiKey) {
+            return modelDescription.apiKey;
+        }
+        return this.getApiKeyBasedOnProvider(providerConfig);
+    }
+
+    private getApiKeyBasedOnProvider(providerConfig: VercelAiProviderConfig): string | undefined {
+        if (providerConfig.apiKey) {
+            return providerConfig.apiKey;
+        }
+        switch (providerConfig.provider) {
+            case 'openai':
+                return process.env.OPENAI_API_KEY;
+            case 'anthropic':
+                return process.env.ANTHROPIC_API_KEY;
+            default:
+                return undefined;
+        }
+    }
+}