diff --git a/packages/types/src/provider-settings.ts b/packages/types/src/provider-settings.ts index 85e22f22f6..956ea3e6b2 100644 --- a/packages/types/src/provider-settings.ts +++ b/packages/types/src/provider-settings.ts @@ -436,6 +436,7 @@ export const MODEL_ID_KEYS: Partial[] = [ "litellmModelId", "huggingFaceModelId", "ioIntelligenceModelId", + "watsonxModelId", ] export const getModelId = (settings: ProviderSettings): string | undefined => { @@ -566,6 +567,7 @@ export const dynamicProviders = [ "openrouter", "requesty", "unbound", + "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 3dde992c6f..061a604faa 100644 --- a/src/shared/api.ts +++ b/src/shared/api.ts @@ -27,6 +27,7 @@ const routerNames = [ "ollama", "lmstudio", "io-intelligence", + "watsonx", ] as const export type RouterName = (typeof routerNames)[number] @@ -151,3 +152,4 @@ export type GetModelsOptions = | { provider: "ollama"; baseUrl?: string } | { provider: "lmstudio"; baseUrl?: string } | { provider: "io-intelligence"; apiKey: string } + | { 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 52048de95d..4686337d21 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 @@ -159,6 +159,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 f14fb89200..6ee84c8f77 100644 --- a/webview-ui/src/utils/__tests__/validate.test.ts +++ b/webview-ui/src/utils/__tests__/validate.test.ts @@ -41,6 +41,7 @@ describe("Model Validation Functions", () => { ollama: {}, lmstudio: {}, "io-intelligence": {}, + watsonx: {}, } const allowAllOrganization: OrganizationAllowList = {