BaiduAiController.java 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package com.goafanti.baiduAI.controller;
  2. import com.goafanti.baiduAI.bo.InputSendChat;
  3. import com.goafanti.baiduAI.bo.Messages;
  4. import com.goafanti.baiduAI.bo.SseMap;
  5. import com.goafanti.baiduAI.bo.SseResult;
  6. import com.goafanti.baiduAI.service.BaiduAiService;
  7. import com.goafanti.common.bo.Result;
  8. import com.goafanti.common.bo.SseEmitterUTF8;
  9. import com.goafanti.common.constant.ErrorConstants;
  10. import com.goafanti.common.controller.BaseController;
  11. import com.goafanti.common.controller.CertifyApiController;
  12. import com.goafanti.common.error.BusinessException;
  13. import com.goafanti.common.model.AftUser;
  14. import com.goafanti.common.utils.LoggerUtils;
  15. import com.goafanti.common.utils.StringUtils;
  16. import com.goafanti.core.shiro.token.TokenManager;
  17. import org.springframework.beans.factory.annotation.Autowired;
  18. import org.springframework.web.bind.annotation.*;
  19. import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
  20. import java.io.IOException;
  21. import java.util.ArrayList;
  22. import java.util.Calendar;
  23. import java.util.List;
  24. import java.util.UUID;
  25. @RestController
  26. @RequestMapping(value = "/chat")
  27. public class BaiduAiController extends CertifyApiController{
  28. @Autowired
  29. private BaiduAiService baiduAiService;
  30. @RequestMapping(value = "/send")
  31. public Object sendChat(@RequestBody InputSendChat in){
  32. if (in.getUserId()==null ){
  33. in.setUserId(UUID.randomUUID().toString());
  34. in.setStream(true);
  35. }
  36. if (in.getMessages()==null||in.getMessages().isEmpty()){
  37. Result res= new Result();
  38. res.getError().add(buildError(ErrorConstants.PARAM_ERROR,"访问参数"));
  39. return res;
  40. }
  41. return baiduAiService.sendChat(in);
  42. }
  43. }