From 411cf01e7f81da2c416059f6eeb17c453d2cb001 Mon Sep 17 00:00:00 2001 From: Prasang Prajapati Date: Tue, 26 Aug 2025 12:38:04 -0400 Subject: [PATCH] fix providers issue --- packages/types/src/provider-settings.ts | 2 ++ src/shared/api.ts | 2 ++ .../settings/__tests__/ApiOptions.provider-filtering.spec.tsx | 1 + webview-ui/src/utils/__tests__/validate.test.ts | 1 + 4 files changed, 6 insertions(+) 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 = {