mirror of
https://gitcode.com/ageerle/ruoyi-ai.git
synced 2026-03-25 18:43:46 +08:00
114 lines
3.4 KiB
Java
114 lines
3.4 KiB
Java
package com.xmzs.controller;
|
|
|
|
|
|
import com.xmzs.common.chat.domain.request.ChatRequest;
|
|
import com.xmzs.common.chat.domain.request.Dall3Request;
|
|
import com.xmzs.common.chat.entity.Tts.TextToSpeech;
|
|
import com.xmzs.common.chat.entity.files.UploadFileResponse;
|
|
import com.xmzs.common.chat.entity.images.Item;
|
|
import com.xmzs.common.chat.entity.whisper.WhisperResponse;
|
|
import com.xmzs.common.core.domain.R;
|
|
import com.xmzs.common.core.domain.model.LoginUser;
|
|
import com.xmzs.common.core.exception.base.BaseException;
|
|
import com.xmzs.common.mybatis.core.page.PageQuery;
|
|
import com.xmzs.common.mybatis.core.page.TableDataInfo;
|
|
import com.xmzs.common.satoken.utils.LoginHelper;
|
|
import com.xmzs.system.domain.bo.ChatMessageBo;
|
|
import com.xmzs.system.domain.vo.ChatMessageVo;
|
|
import com.xmzs.system.service.IChatMessageService;
|
|
import com.xmzs.system.service.ISseService;
|
|
import jakarta.servlet.http.HttpServletResponse;
|
|
import jakarta.validation.Valid;
|
|
import lombok.RequiredArgsConstructor;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.core.io.Resource;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
|
|
|
|
import java.util.List;
|
|
|
|
/**
|
|
* 描述：
|
|
*
|
|
* @author https:www.unfbx.com
|
|
* @date 2023-03-01
|
|
*/
|
|
@Controller
|
|
@Slf4j
|
|
@RequiredArgsConstructor
|
|
public class ChatController {
|
|
|
|
private final ISseService ISseService;
|
|
|
|
private final IChatMessageService chatMessageService;
|
|
|
|
/**
|
|
* 聊天接口
|
|
*/
|
|
@PostMapping("/chat")
|
|
@ResponseBody
|
|
public SseEmitter sseChat(@RequestBody @Valid ChatRequest chatRequest, HttpServletResponse response) {
|
|
return ISseService.sseChat(chatRequest);
|
|
}
|
|
|
|
/**
|
|
* 上传文件
|
|
*/
|
|
@PostMapping("/v1/upload")
|
|
@ResponseBody
|
|
public UploadFileResponse upload(@RequestPart("file") MultipartFile file) {
|
|
return ISseService.upload(file);
|
|
}
|
|
|
|
|
|
/**
|
|
* 语音转文本
|
|
*
|
|
* @param file
|
|
*/
|
|
@PostMapping("/audio")
|
|
@ResponseBody
|
|
public WhisperResponse audio(@RequestParam("file") MultipartFile file) {
|
|
WhisperResponse whisperResponse = ISseService.speechToTextTranscriptionsV2(file);
|
|
return whisperResponse;
|
|
}
|
|
|
|
/**
|
|
* 文本转语音
|
|
*
|
|
* @param textToSpeech
|
|
*/
|
|
@PostMapping("/speech")
|
|
@ResponseBody
|
|
public ResponseEntity<Resource> speech(@RequestBody TextToSpeech textToSpeech) {
|
|
return ISseService.textToSpeed(textToSpeech);
|
|
}
|
|
|
|
|
|
@PostMapping("/dall3")
|
|
@ResponseBody
|
|
public R<List<Item>> dall3(@RequestBody @Valid Dall3Request request) {
|
|
return R.ok(ISseService.dall3(request));
|
|
}
|
|
|
|
/**
|
|
* 聊天记录
|
|
*/
|
|
@PostMapping("/chatList")
|
|
@ResponseBody
|
|
public R<TableDataInfo<ChatMessageVo>> list(@RequestBody @Valid ChatMessageBo chatRequest, @RequestBody PageQuery pageQuery) {
|
|
// 默认查询当前登录用户消息记录
|
|
LoginUser loginUser = LoginHelper.getLoginUser();
|
|
if (loginUser == null) {
|
|
throw new BaseException("用户未登录！");
|
|
}
|
|
chatRequest.setUserId(loginUser.getUserId());
|
|
TableDataInfo<ChatMessageVo> chatMessageVoTableDataInfo = chatMessageService.queryPageList(chatRequest, pageQuery);
|
|
return R.ok(chatMessageVoTableDataInfo);
|
|
}
|
|
|
|
}
|