topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-11 03:58:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[feature][chat]Add API to get ChatModelType list.#1739

Browse Source
This commit is contained in:
jerryjzhang
2024-10-09 14:56:34 +08:00
parent 485a4a19f3
commit 9ffe67c962
3 changed files with 36 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
chat/server/src/main/java/com/tencent/supersonic/chat/server/rest/ChatModelController.java
View File

@@ -5,14 +5,18 @@ import javax.servlet.http.HttpServletResponse;
import com.tencent.supersonic.auth.api.authentication.pojo.User;
import com.tencent.supersonic.auth.api.authentication.utils.UserHolder;
import com.tencent.supersonic.chat.api.pojo.response.ChatModelTypeResp;
import com.tencent.supersonic.chat.server.pojo.ChatModel;
import com.tencent.supersonic.chat.server.service.ChatModelService;
import com.tencent.supersonic.chat.server.util.ModelConfigHelper;
import com.tencent.supersonic.common.pojo.ChatModelConfig;
import com.tencent.supersonic.common.pojo.enums.ChatModelType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
@RestController
@RequestMapping({"/api/chat/model", "/openapi/chat/model"})
@@ -45,6 +49,13 @@ public class ChatModelController {
return chatModelService.getChatModels();
}
@RequestMapping("/getModelTypeList")
public List<ChatModelTypeResp> getModelTypeList() {
return Arrays.stream(ChatModelType.values()).map(t -> ChatModelTypeResp.builder()
.type(t.toString()).name(t.getName()).description(t.getDescription()).build())
.collect(Collectors.toList());
}
@PostMapping("/testConnection")
public boolean testConnection(@RequestBody ChatModelConfig modelConfig) {
return ModelConfigHelper.testConnection(modelConfig);