Skip to content

Commit

Permalink
Add watsonx icon for inference endpoints management page (elastic#197116
Browse files Browse the repository at this point in the history
)

This PR adds IBM Watsonx ai support for the Inference Endpoints
Management UI.

<img width="1249" alt="Screenshot 2024-10-21 at 12 22 09 PM"
src="https://github.com/user-attachments/assets/58516bef-4800-4198-91bf-c428d878d787">

(cherry picked from commit 8e1fcc1)
  • Loading branch information
saikatsarkar056 committed Oct 22, 2024
1 parent 6c8699f commit 9e13609
Show file tree
Hide file tree
Showing 4 changed files with 19 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -287,6 +287,16 @@ export type InferenceServiceSettings =
};
};
}
| {
service: 'watsonxai';
service_settings: {
api_key: string;
url: string;
model_id: string;
project_id: string;
api_version: string;
};
}
| {
service: 'amazonbedrock';
service_settings: {
Expand Down
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ import googleAIStudioIcon from '../../../../assets/images/providers/google_ai_st
import mistralIcon from '../../../../assets/images/providers/mistral.svg';
import amazonBedrockIcon from '../../../../assets/images/providers/amazon_bedrock.svg';
import alibabaCloudAISearchIcon from '../../../../assets/images/providers/alibaba_cloud_ai_search.svg';
import watsonxAIIcon from '../../../../assets/images/providers/watsonx_ai.svg';
import { ServiceProviderKeys } from '../../types';
import * as i18n from './translations';

Expand Down Expand Up @@ -78,6 +79,10 @@ export const SERVICE_PROVIDERS: Record<ServiceProviderKeys, ServiceProviderRecor
icon: openAIIcon,
name: 'OpenAI',
},
[ServiceProviderKeys.watsonxai]: {
icon: watsonxAIIcon,
name: 'WatsonxAI',
},
};

export const ServiceProvider: React.FC<ServiceProviderProps> = ({ providerEndpoint }) => {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ export enum ServiceProviderKeys {
hugging_face = 'hugging_face',
mistral = 'mistral',
openai = 'openai',
watsonxai = 'watsonxai',
}

export enum SortFieldInferenceEndpoint {
Expand Down

0 comments on commit 9e13609

Please sign in to comment.