스프링부트 OpenAI 를 적용한 AI 채팅 서비스 개발하기

implementation 'io.github.flashvayne:chatgpt-spring-boot-starter:1.0.0'
annotationProcessor "org.springframework.boot:spring-boot-configuration-processor"

spring:
  profiles:
    include: API-KEY

openai:
  api-key: qowjfkopqw-UWnein12321313  // api key

chatgpt:
  api-key: qowjfkopqw-UWnein12321313  // api key

@RequestMapping("/chatGPT")
@RestController
public class MyChatGPTController {
    String qreQuestion = "안녕, ChatGPT! 나 질문이 있어.";
    private final MyChatGPTService chatGPTService;
    private final JwtService jwtService;

    @Autowired
    public MyChatGPTController(MyChatGPTService chatGPTService, JwtService jwtService){
        this.chatGPTService = chatGPTService;
        this.jwtService = jwtService;
    }

    @ResponseBody
    @PostMapping("/askChatGPT")
    public BaseResponse<ChatGptRes> askToChatGPT(@RequestBody ChatGptReq chatGptReq){
        try {
            int userIdx = jwtService.getUserIdx();
            String resultQuestion = qreQuestion + chatGptReq.getQuestion();
            ChatGptRes chatGptRes = chatGPTService.getChatResponse(resultQuestion);

            return new BaseResponse(chatGptRes);
        } catch (BaseException baseException){
            return new BaseResponse(baseException.getStatus());
        }
    }
}

@Data
@AllArgsConstructor
@NoArgsConstructor
public class ChatGptReq {
    String question;
}

@Service
public class MyChatGPTService {
    private ChatgptService chatgptService;

    @Autowired
    public MyChatGPTService(ChatgptService chatgptService){
        this.chatgptService = chatgptService;
    }

    public ChatGptRes getChatResponse(String prompt) throws BaseException {
        try {
            // ChatGPT 에게 질문을 던집니다.
            String responseMessage = chatgptService.sendMessage(prompt);
            return new ChatGptRes(responseMessage, category);
        } catch (Exception exception){
            throw new BaseException(BaseResponseStatus.SERVER_ERROR);
        }
    }
}

package io.github.flashvayne.chatgpt.service;

import io.github.flashvayne.chatgpt.dto.ChatRequest;
import io.github.flashvayne.chatgpt.dto.ChatResponse;

public interface ChatgptService {

    String sendMessage(String message);

    ChatResponse sendChatRequest(ChatRequest request);

}

{
    "message": "요청에 성공했습니다.",
    "result": {
        "answer": "chatgpt는 최근 인공지능 기술의 발전에 따라 만들어진 챗봇입니다. 이 챗봇은 사람들이 챗봇과 대화하는 것처럼 사람과 대화할 수 있는 기술을 사용합니다. 이는 사람들이 챗봇과",
    }
}

@Service
public class MyChatGPTService {
    private ChatgptService chatgptService;

    @Autowired
    public MyChatGPTService(ChatgptService chatgptService){
        this.chatgptService = chatgptService;
    }

    public ChatGptRes getChatResponse(String prompt) throws BaseException {
        try {
            String resultMessage = chatgptService.sendMessage(prompt);
            String splitResultMessage = splitMessage(resultMessage);
            return new ChatGptRes(splitResultMessage, category);
        } catch (Exception exception){
            throw new BaseException(BaseResponseStatus.SERVER_ERROR);
        }
    }

    public String splitMessage(String message){
        String[] tempArr = message.split("[.]");
        String[] resultArr = new String[tempArr.length];
        for(int i=0; i< 2; i++){
            resultArr[i] = tempArr[i];
        }
        String res = String.join(".", resultArr);
        String splitResult = res.replace(String.valueOf("null"), "");
        String realSplitResult = splitResult.substring(0, splitResult.length() - 1);
        return realSplitResult;
    }
}