diff --git a/chat2db-client/src/blocks/Setting/AiSetting/index.tsx b/chat2db-client/src/blocks/Setting/AiSetting/index.tsx index 85faaa27..4129a0d9 100644 --- a/chat2db-client/src/blocks/Setting/AiSetting/index.tsx +++ b/chat2db-client/src/blocks/Setting/AiSetting/index.tsx @@ -81,6 +81,7 @@ export default function SettingAI(props: IProps) {
{i18n('setting.title.aiSource')}:
Chat2DB AI + 智普AI Open AI Azure AI {i18n('setting.tab.custom')} @@ -102,6 +103,45 @@ export default function SettingAI(props: IProps) { )} + {aiConfig?.aiSqlSource === AiSqlSourceType.ZHIPUAI && ( +
+
Api Key
+
+ { + setAiConfig({ ...aiConfig, apiKey: e.target.value }); + }} + /> +
+
Host
+
+ { + setAiConfig({ ...aiConfig, apiHost: e.target.value }); + }} + /> +
+
Model
+
+ { + setAiConfig({ ...aiConfig, model: e.target.value }); + }} + /> +
+
+ )} {aiConfig?.aiSqlSource === AiSqlSourceType.OPENAI && (
Api Key
diff --git a/chat2db-client/src/typings/ai.ts b/chat2db-client/src/typings/ai.ts index 9946c678..14ddc987 100644 --- a/chat2db-client/src/typings/ai.ts +++ b/chat2db-client/src/typings/ai.ts @@ -1,5 +1,6 @@ export enum AiSqlSourceType { CHAT2DBAI = 'CHAT2DBAI', + ZHIPUAI = 'ZHIPUAI', OPENAI = 'OPENAI', AZUREAI = 'AZUREAI', RESTAI = 'RESTAI', diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/ConfigController.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/ConfigController.java index dc328157..66b683d4 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/ConfigController.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/ConfigController.java @@ -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(); } /**