diff --git a/web/src/pages/user-setting/setting-model/components/system-setting.tsx b/web/src/pages/user-setting/setting-model/components/system-setting.tsx
index dc6853e5d0..607f2a99f2 100644
--- a/web/src/pages/user-setting/setting-model/components/system-setting.tsx
+++ b/web/src/pages/user-setting/setting-model/components/system-setting.tsx
@@ -73,6 +73,7 @@ const SystemSetting = ({ onOk, loading }: IProps) => {
value: formData.llm_id,
options: modelOptions as SelectWithSearchFlagOptionType[],
tooltip: t('chatModelTip'),
+ testId: 'default-llm-combobox',
},
{
id: 'embd_id',
@@ -82,6 +83,7 @@ const SystemSetting = ({ onOk, loading }: IProps) => {
LlmModelType.Embedding
] as SelectWithSearchFlagOptionType[],
tooltip: t('embeddingModelTip'),
+ testId: 'default-embedding-combobox',
},
{
id: 'img2txt_id',
@@ -129,6 +131,7 @@ const SystemSetting = ({ onOk, loading }: IProps) => {
tooltip,
id,
isRequired,
+ testId,
}: {
id: string;
label: string;
@@ -136,6 +139,7 @@ const SystemSetting = ({ onOk, loading }: IProps) => {
options: SelectWithSearchFlagOptionType[];
tooltip?: string;
isRequired?: boolean;
+ testId?: string;
}) => {
return (
@@ -162,6 +166,7 @@ const SystemSetting = ({ onOk, loading }: IProps) => {
onChange={(value) => handleFieldChange(id, value)}
placeholder={t('selectModelPlaceholder')}
emptyData={t('modelEmptyTip')}
+ testId={testId}
/>
);
diff --git a/web/src/pages/user-setting/setting-model/components/un-add-model.tsx b/web/src/pages/user-setting/setting-model/components/un-add-model.tsx
index 63ebd9c079..4d6855d2c2 100644
--- a/web/src/pages/user-setting/setting-model/components/un-add-model.tsx
+++ b/web/src/pages/user-setting/setting-model/components/un-add-model.tsx
@@ -79,7 +79,7 @@ export const AvailableModels: FC<{
};
return (
-
+
{t('availableModels')}
@@ -87,6 +87,7 @@ export const AvailableModels: FC<{
{/*
*/}
(
handleAddModel(model.name)}
>
diff --git a/web/src/pages/user-setting/setting-model/components/used-model.tsx b/web/src/pages/user-setting/setting-model/components/used-model.tsx
index 91c7bc0665..6dd3e16dd5 100644
--- a/web/src/pages/user-setting/setting-model/components/used-model.tsx
+++ b/web/src/pages/user-setting/setting-model/components/used-model.tsx
@@ -11,7 +11,7 @@ export const UsedModel = ({
}) => {
const { myLlmList: llmList } = useSelectLlmList();
return (
-
+
{t('setting.addedModels')}
diff --git a/web/src/pages/user-setting/setting-model/modal/api-key-modal/index.tsx b/web/src/pages/user-setting/setting-model/modal/api-key-modal/index.tsx
index 7a84c0e838..4be300c024 100644
--- a/web/src/pages/user-setting/setting-model/modal/api-key-modal/index.tsx
+++ b/web/src/pages/user-setting/setting-model/modal/api-key-modal/index.tsx
@@ -86,6 +86,8 @@ const ApiKeyModal = ({
okText={t('save')}
cancelText={t('cancel')}
className="!w-[600px]"
+ testId="apikey-modal"
+ okButtonTestId="apikey-save"
>