ChatGPT API를 스프링 프레임워크에서 써보기

https://platform.openai.com/docs/api-reference

OpenAI API

 

API Key 발급 받기

 

 

 

발급 받은 키는 다시 볼 수 없다. 

다른곳에 보관하자 

API키를 포함해서 프로젝트를 깃헙에 업로드하면 다시 생성해야된다. 

 

 

 

모델 선택

text-davinci-003 을 선택했다. 

 

 

 

RestTemplate으로 API 요청하기

@Service
public class ChatGptService {
    
    private String API_KEY = "";
    private static final String ENDPOINT = "https://api.openai.com/v1/completions";
    
    public ResponseEntity<String> generateText(String prompt, float temperature, int maxTokens) {

        //HttpHeader 오브젝트 생성
        HttpHeaders headers = new HttpHeaders();
        headers.add("Content-type", "application/json");
        headers.add("Authorization", "Bearer " + API_KEY);

        //HttpBody 오브젝트 생성
        Map<String, Object> requestBody = new HashMap<>();
        requestBody.put("model", "text-davinci-003");
        requestBody.put("prompt", prompt);
        requestBody.put("max_tokens", maxTokens);
        requestBody.put("temperature", temperature);

        // HttpHeader와 HttpBody를 하나의 오브젝트에 담기
        HttpEntity<Map<String, Object>> requestEntity = new HttpEntity<>(requestBody, headers);
        
        RestTemplate rt = new RestTemplate();

        //Http POST 방식 요청 - response 변수 응답
        ResponseEntity<String> response = rt.exchange(
                ENDPOINT,   // 사용자 정보 요청 주소
                HttpMethod.POST,
                requestEntity,
                String.class
        );
        
        return response;
    }
}

발급 받은 API 를 작성하면 된다.

 

파라미터 

• model : 어떤 GPT 모델을 사용할지 지정
• prompt : 질문할 문장
• max_tokens : 응답의 컨텍스트 길이
• temerature : 얼마나 창의적인 답을 작성하도록 할지 지정하는 값. 클수록 창의적이고 작을수록 잘 정의된 답을 제출한다.

 

 

@RequestMapping("/chatGpt")
@ResponseBody
public ResponseEntity<String> chatGpt() throws JsonProcessingException {
    log.info("ok");
    ResponseEntity<String> response = chatGptService.generateText("ChatGPT는 무엇인가요?", 1.0f, 1000);
    return response;
}

 

 

 

 

Response 데이터

 

JSON 형식으로 응답이 온다. 

 

 

 

JSON 파일을 Java 클래스로 만들자  

 

 

https://www.jsonschema2pojo.org/

jsonschema2pojo

 

jsonschema2pojo는 JSON 스키마를 사용하여 자바 클래스를 생성하는 오픈소스 라이브러리입니다.

 

 

 

ObjectMapper objectMapper = new ObjectMapper();
ResponseData responseData = null;
try {
    responseData = objectMapper.readValue(response.getBody(), ResponseData.class);
} catch (JsonProcessingException e) {
    throw new RuntimeException(e);
}

JSON 파일을 jsonschema2pojo 통해서 생성한 Java 클래스 객체로 deserialization 하기 위해서는

 

ObjectMapper의 readValue() 메서드를 이용한다.

 

 

 

 

는 에러

 

다른거 다 바꿔가면서 해봤는데 

자바 클래스를 잘못만들었는지 아니면 deserialization 작업이 잘못된 것 같다. 

다음에 해결하자...

 

 

 

 

4/14(금)

교수님께 물어봐서 해결했다.

 jsonschema2pojo 통해서 생성한 Java 클래스에 문제가 있었다.

 

원인은 바로 Inner Class가 static(정적)으로 선언되지 않는 한 단독(Outer 클래스를 참조하지 않고)으로 Inner Class의 디폴트 생성자를 호출해 인스턴스를 생성할 수 없는 것이다. 즉, 위와 같은 예외를 피하려면 Inner Class를 별도의 클래스로 생성하던가, 아니면 static Inner Class로 선언해주어야 한다.

 

 

@Data
public class ResponseData {
    public String id;
    public String object;
    public Integer created;
    public String model;
    public List<Choice> choices;
    public Usage usage;
}

 

Choice 클래스를  ResponseData 에 Inner Class 로 생성했었다.

 

@Data
public class Choice {
    public String text;
    public Integer index;
    public Object logprobs;
    public String finish_reason;
}

 

별도의 클래스를 생성하여 문제를 해결하였다. 

 

 

 

@RequestMapping("/chatGpt")
@ResponseBody
public  String chatGpt() throws JsonProcessingException {
    log.info("ok");
    ResponseData responseData = chatGptService.generateText("chatGPT는 무엇인가요?", 1.0f, 1000);
    return responseData.getChoices().get(0).getText();
}

 

응답 결과를 확인해보자 

 

 

 

text 만 출력되는 것을 확인 !