Merge pull request #1626 from tmlx1990/fixAI

fix: 修复自定义AI保存后再次进入没有显示的问题。
CodePhiliaX · Jan 14, 2025 · 698323a · 698323a
2 parents aff2564 + 54452f5
commit 698323a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...r-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/ConfigController.java b/...r-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/ConfigController.java
@@ -333,6 +333,13 @@ public DataResult<AIConfig> getChatAiSystemConfig(String aiSqlSource) {
                 config.setModel(Objects.nonNull(azureDeployId.getData()) ? azureDeployId.getData().getContent() : "");
                 break;
             case RESTAI:
+                DataResult<Config> restAIApiKey = configService.find(RestAIClient.REST_AI_API_KEY);
+                DataResult<Config> restAIApiHost = configService.find(RestAIClient.REST_AI_URL);
+                DataResult<Config> restAIModel = configService.find(RestAIClient.REST_AI_MODEL);
+                config.setApiKey(Objects.nonNull(restAIApiKey.getData()) ? restAIApiKey.getData().getContent() : "");
+                config.setApiHost(Objects.nonNull(restAIApiHost.getData()) ? restAIApiHost.getData().getContent() : "");
+                config.setModel(Objects.nonNull(restAIModel.getData()) ? restAIModel.getData().getContent() : "");
+                break;
             case FASTCHATAI:
                 DataResult<Config> fastChatApiKey = configService.find(FastChatAIClient.FASTCHAT_API_KEY);
                 DataResult<Config> fastChatApiHost = configService.find(FastChatAIClient.FASTCHAT_HOST);