opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-02 21:50:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

perf: Add Zhipu AI

Browse Source
This commit is contained in:
Jerry Fan
2023-11-01 14:24:24 +08:00
parent 47b8a24153
commit 9e9e99b39b
3 changed files with 42 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
chat2db-client/src/blocks/Setting/AiSetting/index.tsx
View File

@ -81,6 +81,7 @@ export default function SettingAI(props: IProps) {
<div className={styles.aiSqlSourceTitle}>{i18n('setting.title.aiSource')}:</div>
<Radio.Group onChange={handleAiTypeChange} value={aiConfig?.aiSqlSource}>
<Radio value={AiSqlSourceType.CHAT2DBAI}>Chat2DB AI</Radio>
<Radio value={AiSqlSourceType.ZHIPUAI}>智普AI</Radio>
<Radio value={AiSqlSourceType.OPENAI}>Open AI</Radio>
<Radio value={AiSqlSourceType.AZUREAI}>Azure AI</Radio>
<Radio value={AiSqlSourceType.RESTAI}>{i18n('setting.tab.custom')}</Radio>
@ -102,6 +103,45 @@ export default function SettingAI(props: IProps) {
</div>
</div>
)}
{aiConfig?.aiSqlSource === AiSqlSourceType.ZHIPUAI && (
<div>
<div className={styles.title}>Api Key</div>
<div className={classnames(styles.content, styles.chatGPTKey)}>
<Input
autoComplete="off"
// placeholder={i18n('setting.placeholder.chat2dbApiHost')}
value={aiConfig.apiKey}
onChange={(e) => {
setAiConfig({ ...aiConfig, apiKey: e.target.value });
}}
/>
</div>
<div className={styles.title}>Host</div>
<div className={classnames(styles.content, styles.chatGPTKey)}>
<Input
autoComplete="off"
// placeholder={i18n('setting.placeholder.apiHost')}
defaultValue={'https://open.bigmodel.cn/api/paas/v3/model-api/'}
value={aiConfig.apiHost}
onChange={(e) => {
setAiConfig({ ...aiConfig, apiHost: e.target.value });
}}
/>
</div>
<div className={styles.title}>Model</div>
<div className={classnames(styles.content, styles.chatGPTKey)}>
<Input
autoComplete="off"
// placeholder={i18n('setting.placeholder.azureEndpoint')}
defaultValue={'chatglm_turbo'}
value={aiConfig.model}
onChange={(e) => {
setAiConfig({ ...aiConfig, model: e.target.value });
}}
/>
</div>
</div>
)}
{aiConfig?.aiSqlSource === AiSqlSourceType.OPENAI && (
<div>
<div className={styles.title}>Api Key</div>

1
chat2db-client/src/typings/ai.ts
View File

@ -1,5 +1,6 @@
export enum AiSqlSourceType {
CHAT2DBAI = 'CHAT2DBAI',
ZHIPUAI = 'ZHIPUAI',
OPENAI = 'OPENAI',
AZUREAI = 'AZUREAI',
RESTAI = 'RESTAI',

2
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/ConfigController.java
View File

@ -212,7 +212,7 @@ public class ConfigController {
SystemConfigParam modelParam = SystemConfigParam.builder().code(ZhipuChatAIClient.ZHIPU_MODEL)
.content(request.getModel()).build();
configService.createOrUpdate(modelParam);
FastChatAIClient.refresh();
ZhipuChatAIClient.refresh();
}
/**