add support for Baidu yiyan (#2049)

### What problem does this PR solve?

add support for Baidu yiyan

### Type of change

- [x] New Feature (non-breaking change which adds functionality)

---------

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
This commit is contained in:
黄腾
2024-08-22 16:45:15 +08:00
committed by GitHub
parent 21f2c5838b
commit 733219cc3f
17 changed files with 307 additions and 13 deletions

View File

@@ -39,6 +39,7 @@ import {
useSubmitSpark,
useSubmitSystemModelSetting,
useSubmitVolcEngine,
useSubmityiyan,
} from './hooks';
import HunyuanModal from './hunyuan-modal';
import styles from './index.less';
@@ -46,6 +47,7 @@ import OllamaModal from './ollama-modal';
import SparkModal from './spark-modal';
import SystemModelSettingModal from './system-model-setting-modal';
import VolcEngineModal from './volcengine-modal';
import YiyanModal from './yiyan-modal';
const LlmIcon = ({ name }: { name: string }) => {
const icon = IconMap[name as keyof typeof IconMap];
@@ -95,7 +97,8 @@ const ModelCard = ({ item, clickApiKey }: IModelCardProps) => {
{isLocalLlmFactory(item.name) ||
item.name === 'VolcEngine' ||
item.name === 'Tencent Hunyuan' ||
item.name === 'XunFei Spark'
item.name === 'XunFei Spark' ||
item.name === 'BaiduYiyan'
? t('addTheModel')
: 'API-Key'}
<SettingOutlined />
@@ -185,6 +188,14 @@ const UserSettingModel = () => {
SparkAddingLoading,
} = useSubmitSpark();
const {
yiyanAddingVisible,
hideyiyanAddingModal,
showyiyanAddingModal,
onyiyanAddingOk,
yiyanAddingLoading,
} = useSubmityiyan();
const {
bedrockAddingLoading,
onBedrockAddingOk,
@@ -199,12 +210,14 @@ const UserSettingModel = () => {
VolcEngine: showVolcAddingModal,
'Tencent Hunyuan': showHunyuanAddingModal,
'XunFei Spark': showSparkAddingModal,
BaiduYiyan: showyiyanAddingModal,
}),
[
showBedrockAddingModal,
showVolcAddingModal,
showHunyuanAddingModal,
showSparkAddingModal,
showyiyanAddingModal,
],
);
@@ -330,6 +343,13 @@ const UserSettingModel = () => {
loading={SparkAddingLoading}
llmFactory={'XunFei Spark'}
></SparkModal>
<YiyanModal
visible={yiyanAddingVisible}
hideModal={hideyiyanAddingModal}
onOk={onyiyanAddingOk}
loading={yiyanAddingLoading}
llmFactory={'BaiduYiyan'}
></YiyanModal>
<BedrockModal
visible={bedrockAddingVisible}
hideModal={hideBedrockAddingModal}