Merge pull request #45 from Princess-in-silvertown/feat/43

Feat: 키워드 자동 생성 기능

build.gradle

@@ -25,19 +25,17 @@ repositories {
 }
 
 dependencies {
-    // Spring Boot 설정
     implementation 'org.springframework.boot:spring-boot-starter-web'
     implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
     developmentOnly 'org.springframework.boot:spring-boot-devtools'
 
-    // 데이터베이스 의존성 설정
+    implementation 'org.springframework.boot:spring-boot-starter-webflux'
+
     runtimeOnly 'com.h2database:h2'
 
-    // Lombok 설정
     compileOnly 'org.projectlombok:lombok'
     annotationProcessor 'org.projectlombok:lombok'
 
-    // 테스트 환경 설정
     testImplementation 'org.springframework.boot:spring-boot-starter-test'
     testRuntimeOnly 'org.junit.platform:junit-platform-launcher'
 }

src/main/java/slvtwn/khu/toyouserver/agent/gpt/ChatGptAgent.java

@@ -0,0 +1,47 @@
+package slvtwn.khu.toyouserver.agent.gpt;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import lombok.AllArgsConstructor;
+import org.springframework.http.HttpStatusCode;
+import org.springframework.stereotype.Component;
+import org.springframework.web.reactive.function.client.WebClient;
+
+@AllArgsConstructor
+@Component
+public class ChatGptAgent {
+
+    private ChatGptConfiguration configuration;
+
+    public ChatGptResponse requestWithPrompt(String prompt) {
+        HashMap<String, Object> body = setupBody(prompt);
+
+        return WebClient.create()
+                .post()
+                .uri(configuration.getTextModel())
+                .header("Content-Type", "application/json")
+                .header("Authorization", "Bearer " + configuration.getOpenaiKey())
+                .bodyValue(body)
+                .retrieve()
+                .onStatus(HttpStatusCode::is4xxClientError,
+                        clientResponse -> clientResponse.bodyToMono(String.class).map(Exception::new))
+                .bodyToMono(ChatGptResponse.class)
+                .block();
+    }
+
+
+    private HashMap<String, Object> setupBody(String prompt) {
+        HashMap<String, Object> body = new HashMap<>();
+        body.put("model", configuration.getTextModel());
+
+        HashMap<String, String> userMessage = new HashMap<>();
+        userMessage.put("role", "user");
+        userMessage.put("content", prompt);
+
+        List<HashMap<String, String>> messages = new ArrayList<>();
+        messages.add(userMessage);
+        body.put("messages", messages);
+        return body;
+    }
+}

src/main/java/slvtwn/khu/toyouserver/agent/gpt/ChatGptConfiguration.java

@@ -0,0 +1,17 @@
+package slvtwn.khu.toyouserver.agent.gpt;
+
+import lombok.Getter;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+
+@Getter
+@Component
+public class ChatGptConfiguration {
+
+    @Value("${openai.key}")
+    private String openaiKey;
+    @Value("${openai.text-generation.endpoint}")
+    private String textModelEndpoint;
+    @Value("${openai.text-generation.model}")
+    private String textModel;
+}

src/main/java/slvtwn/khu/toyouserver/agent/gpt/ChatGptResponse.java

@@ -0,0 +1,21 @@
+package slvtwn.khu.toyouserver.agent.gpt;
+
+import java.util.List;
+
+public record ChatGptResponse(String id, String object, Long created, List<Choice> choices, Usage usage) {
+
+    public record Choice(
+            String text,
+            int index,
+            String logprobs,
+            String finish_reason
+    ) {
+    }
+
+    public record Usage(
+            int prompt_tokens,
+            int completion_tokens,
+            int total_tokens
+    ) {
+    }
+}

src/main/java/slvtwn/khu/toyouserver/application/AgentService.java

@@ -0,0 +1,62 @@
+package slvtwn.khu.toyouserver.application;
+
+import java.util.Arrays;
+import java.util.List;
+import lombok.AllArgsConstructor;
+import org.springframework.stereotype.Service;
+import slvtwn.khu.toyouserver.agent.gpt.ChatGptAgent;
+import slvtwn.khu.toyouserver.agent.gpt.ChatGptResponse;
+import slvtwn.khu.toyouserver.dto.KeywordRequest;
+import slvtwn.khu.toyouserver.dto.KeywordResponse;
+
+@AllArgsConstructor
+@Service
+public class AgentService {
+
+    private final ChatGptAgent chatGptAgent;
+
+    public KeywordResponse generateKeywords(KeywordRequest request) {
+        String content = request.content();
+        String prompt = String.format("""
+                Suggest 3 keywords that could represent emotions or characteristics in the content.
+                
+                <example>
+                    <request>
+                        content:
+                            안녕, 저번 학기에 만나서 정말 반가웠어! 앞으로 또 만날 수 있을거라 기대하고 있어.
+                            정말 아쉽지만 건강하게 잘 지내!
+                        keywords:
+                    </request>
+                    <response>
+                        반가움, 기대, 아쉬움
+                    </response>
+                </example>
+                
+                <example>
+                    <request>
+                        content:
+                            생일 축하해! 생일 축하 파티에 직접 참여하진 못해서 아쉽지만, 즐겁게 하루를 보냈으면 좋겠어.
+                        keywords:
+                    </request>
+                    <response>
+                        축하, 아쉬움, 즐거움
+                    </response>
+                </example>
+                
+                content: %s
+                keywords:                                 
+                """, content);
+
+        ChatGptResponse chatGptResponse = chatGptAgent.requestWithPrompt(prompt);
+        ChatGptResponse.Choice firstChoice = chatGptResponse.choices().get(0);
+        List<String> keywords = parseKeywordFromChoice(firstChoice);
+
+        return new KeywordResponse(keywords);
+    }
+
+    private List<String> parseKeywordFromChoice(ChatGptResponse.Choice choice) {
+        String text = choice.text();
+        return Arrays.stream(text.split(","))
+                .toList();
+    }
+}

src/main/java/slvtwn/khu/toyouserver/dto/KeywordRequest.java

@@ -0,0 +1,5 @@
+package slvtwn.khu.toyouserver.dto;
+
+public record KeywordRequest(String content) {
+
+}

src/main/java/slvtwn/khu/toyouserver/dto/KeywordResponse.java

@@ -0,0 +1,7 @@
+package slvtwn.khu.toyouserver.dto;
+
+import java.util.List;
+
+public record KeywordResponse(List<String> keywords) {
+
+}

src/main/java/slvtwn/khu/toyouserver/presentation/AgentController.java

@@ -0,0 +1,23 @@
+package slvtwn.khu.toyouserver.presentation;
+
+import lombok.AllArgsConstructor;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RestController;
+import slvtwn.khu.toyouserver.application.AgentService;
+import slvtwn.khu.toyouserver.common.response.ToyouResponse;
+import slvtwn.khu.toyouserver.dto.KeywordRequest;
+import slvtwn.khu.toyouserver.dto.KeywordResponse;
+
+@AllArgsConstructor
+@RestController
+public class AgentController {
+
+    private final AgentService agentService;
+
+    @PostMapping("/generate-keywords")
+    public ToyouResponse<KeywordResponse> generateKeywords(@RequestBody KeywordRequest keywordRequest) {
+        KeywordResponse response = agentService.generateKeywords(keywordRequest);
+        return ToyouResponse.from(response);
+    }
+}

src/main/resources/application.yml

@@ -1,6 +1,5 @@
 spring:
   jpa:
-#    database: mysql
     open-in-view: false
     generate-ddl: true
     show-sql: true
@@ -14,3 +13,9 @@ spring:
     username: sa
     password:
 
+openai:
+  key: ${OPEN_AI_KEY}
+
+  text-generation:
+    endpoint: https://api.openai.com/v1/chat/completions
+    model: gpt-4o-mini

src/test/resources/application.yml

@@ -0,0 +1,21 @@
+spring:
+  jpa:
+    open-in-view: false
+    generate-ddl: true
+    show-sql: true
+    properties:
+      hibernate:
+        format_sql: true
+
+  datasource:
+    url: jdbc:h2:mem:testdb
+    driver-class-name: org.h2.Driver
+    username: sa
+    password:
+
+openai:
+  key: test-key
+
+  text-generation:
+    endpoint: https://api.openai.com/v1/chat/completions
+    model: gpt-4o-mini