diff --git a/packages/types/src/provider-settings.ts b/packages/types/src/provider-settings.ts index 5f3ad5f162..8dbdd6f010 100644 --- a/packages/types/src/provider-settings.ts +++ b/packages/types/src/provider-settings.ts @@ -466,6 +466,7 @@ export const MODEL_ID_KEYS: Partial[] = [ "ioIntelligenceModelId", "vercelAiGatewayModelId", "deepInfraModelId", + "watsonxModelId", ] export const getModelId = (settings: ProviderSettings): string | undefined => { @@ -605,6 +606,7 @@ export const dynamicProviders = [ "unbound", "deepinfra", "vercel-ai-gateway", + "watsonx", ] as const satisfies readonly ProviderName[] export type DynamicProvider = (typeof dynamicProviders)[number] diff --git a/src/shared/api.ts b/src/shared/api.ts index 2a32759574..efa3770d0b 100644 --- a/src/shared/api.ts +++ b/src/shared/api.ts @@ -29,6 +29,7 @@ const routerNames = [ "io-intelligence", "deepinfra", "vercel-ai-gateway", + "watsonx", ] as const export type RouterName = (typeof routerNames)[number] @@ -155,3 +156,4 @@ export type GetModelsOptions = | { provider: "deepinfra"; apiKey?: string; baseUrl?: string } | { provider: "io-intelligence"; apiKey: string } | { provider: "vercel-ai-gateway" } + | { provider: "watsonx"; apiKey: string } diff --git a/webview-ui/src/components/settings/__tests__/ApiOptions.provider-filtering.spec.tsx b/webview-ui/src/components/settings/__tests__/ApiOptions.provider-filtering.spec.tsx index 17f898a5b7..ebb7151677 100644 --- a/webview-ui/src/components/settings/__tests__/ApiOptions.provider-filtering.spec.tsx +++ b/webview-ui/src/components/settings/__tests__/ApiOptions.provider-filtering.spec.tsx @@ -160,6 +160,7 @@ describe("ApiOptions Provider Filtering", () => { expect(providerValues).toContain("unbound") expect(providerValues).toContain("requesty") expect(providerValues).toContain("io-intelligence") + expect(providerValues).toContain("watsonx") }) it("should filter static providers based on organization allow list", () => { diff --git a/webview-ui/src/utils/__tests__/validate.test.ts b/webview-ui/src/utils/__tests__/validate.test.ts index c9fb7bfd42..4b7c52ca14 100644 --- a/webview-ui/src/utils/__tests__/validate.test.ts +++ b/webview-ui/src/utils/__tests__/validate.test.ts @@ -42,6 +42,7 @@ describe("Model Validation Functions", () => { deepinfra: {}, "io-intelligence": {}, "vercel-ai-gateway": {}, + watsonx: {}, } const allowAllOrganization: OrganizationAllowList = {