opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-09-23 13:37:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

add azure openai support

Browse Source
This commit is contained in:
robin
2023-06-27 21:59:14 +08:00
parent cd8b798547
commit 4b36d7dd6c
3 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,7 +14,7 @@ import ai.chat2db.server.domain.api.service.ConfigService;
import ai.chat2db.server.tools.base.wrapper.result.ActionResult;
import ai.chat2db.server.tools.base.wrapper.result.DataResult;
import ai.chat2db.server.web.api.aspect.ConnectionInfoAspect;
import ai.chat2db.server.web.api.controller.config.request.ChatGptSystemConfigRequest;
import ai.chat2db.server.web.api.controller.config.request.AISystemConfigRequest;
import ai.chat2db.server.web.api.controller.config.request.SystemConfigRequest;
import ai.chat2db.server.web.api.util.OpenAIClient;
import ai.chat2db.server.web.api.controller.ai.rest.client.RestAIClient;
@ -58,7 +58,7 @@ public class ConfigController {
* @return
*/
@PostMapping("/system_config/chatgpt")
public ActionResult addChatGptSystemConfig(@RequestBody ChatGptSystemConfigRequest request) {
public ActionResult addChatGptSystemConfig(@RequestBody AISystemConfigRequest request) {
String sqlSource = StringUtils.isNotBlank(request.getAiSqlSource()) ? request.getAiSqlSource()
: AiSqlSourceEnum.OPENAI.getCode();
SystemConfigParam param = SystemConfigParam.builder().code(RestAIClient.AI_SQL_SOURCE).content(sqlSource)
@ -77,7 +77,7 @@ public class ConfigController {
*
* @param request
*/
private void saveOpenAIConfig(ChatGptSystemConfigRequest request) {
private void saveOpenAIConfig(AISystemConfigRequest request) {
SystemConfigParam param = SystemConfigParam.builder().code(OpenAIClient.OPENAI_KEY).content(
request.getApiKey())
.build();
@ -100,7 +100,7 @@ public class ConfigController {
*
* @param request
*/
private void saveRestAIConfig(ChatGptSystemConfigRequest request) {
private void saveRestAIConfig(AISystemConfigRequest request) {
SystemConfigParam restParam = SystemConfigParam.builder().code(RestAIClient.REST_AI_URL).content(
request.getRestAiUrl())
.build();

17
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/request/ChatGptSystemConfigRequest.java → chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/config/request/AISystemConfigRequest.java
View File

@ -13,7 +13,7 @@ import lombok.Data;
* @version : SystemConfigRequest.java
*/
@Data
public class ChatGptSystemConfigRequest {
public class AISystemConfigRequest {
/**
* OpenAi APIKEY
@ -56,4 +56,19 @@ public class ChatGptSystemConfigRequest {
* 非必填，默认值为TRUE
*/
private Boolean restAiStream = Boolean.TRUE;
/**
* Get Azure OpenAI key credential from the Azure Portal
*/
private String azureApiKey;
/**
* Get Azure OpenAI endpoint from the Azure Portal
*/
private String azureEndpoint;
/**
* deployment id of the deployed model
*/
private String azureDeploymentId;
}