feat: gpt 프롬프트 반환

src/main/java/com/project/oof/api/controller/ChatGPTController.java → src/main/java/com/project/oof/gpt/controller/ChatGPTController.java

@@ -1,7 +1,6 @@
-package com.project.oof.api.controller;
+package com.project.oof.gpt.controller;
 
-import com.project.oof.dto.ChatCompletionDto;
-import com.project.oof.service.ChatGPTService;
+import com.project.oof.gpt.service.ChatGPTService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
@@ -10,8 +9,6 @@
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
-import java.util.Map;
-
 
 @Slf4j
 @RestController
@@ -25,9 +22,10 @@ public ChatGPTController(ChatGPTService chatGPTService) {
     }
 
     @PostMapping("/prompt")
-    public ResponseEntity<Map<String, Object>> selectPrompt(@RequestBody ChatCompletionDto chatCompletionDto) {
-        log.debug("param :: " + chatCompletionDto.toString());
-        Map<String, Object> result = chatGPTService.prompt(chatCompletionDto);
+    public ResponseEntity<String> selectPrompt(@RequestBody String message) {
+        String result = chatGPTService.prompt(message);
         return new ResponseEntity<>(result, HttpStatus.OK);
     }
+
+
 }

src/main/java/com/project/oof/gpt/service/ChatGPTService.java

@@ -0,0 +1,10 @@
+package com.project.oof.gpt.service;
+
+import org.springframework.stereotype.Service;
+
+
+@Service
+public interface ChatGPTService {
+
+    String prompt(String userMessage);
+}

src/main/java/com/project/oof/gpt/service/ChatGPTServiceImpl.java

@@ -0,0 +1,66 @@
+package com.project.oof.gpt.service;
+
+import com.project.oof.gpt.config.ChatGPTConfig;
+import com.project.oof.storage.service.ChatHistoryService;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+import java.util.Map;
+
+@Slf4j
+@Service
+@RequiredArgsConstructor
+public class ChatGPTServiceImpl implements ChatGPTService {
+
+
+    private final ChatGPTConfig chatGPTConfig;
+    private final ChatHistoryService chatHistoryService;
+
+    @Value("${openai.url.prompt}")
+    private String promptUrl;
+
+
+    @Override
+    public String prompt(String userMessage) {
+        log.debug("[+] 신규 프롬프트를 수행합니다.");
+        chatHistoryService.addMessage("user", userMessage);
+
+        List<Map<String, String>> messages = chatHistoryService.getChatHistory();
+        Map<String, Object> requestBody = Map.of(
+                "model", "gpt-3.5-turbo",
+                "messages", messages
+        );
+
+        // [STEP1] 토큰 정보가 포함된 Header를 가져옵니다.
+        HttpHeaders headers = chatGPTConfig.httpHeaders();
+
+
+        // [STEP5] 통신을 위한 RestTemplate을 구성합니다.
+        HttpEntity<Map<String, Object>> requestEntity = new HttpEntity<>(requestBody, headers);
+        ResponseEntity<Map> response = chatGPTConfig
+                .restTemplate()
+                .exchange(promptUrl, HttpMethod.POST, requestEntity, Map.class);
+
+        // OpenAI API의 응답을 처리
+        Map<String, Object> responseBody = response.getBody();
+
+        List<Map<String, Object>> choices = (List<Map<String, Object>>) responseBody.get("choices");
+
+        Map<String, Object> firstChoice = choices.get(0);
+        Map<String, Object> message = (Map<String, Object>) firstChoice.get("message");
+
+        String assistantMessage = message.get("content").toString();
+
+        System.out.println("Assistant response: " + assistantMessage);
+        chatHistoryService.addMessage("assistant", assistantMessage);
+        return assistantMessage;
+    }
+
+}