Merge pull request #32 from TeamAiary/feat/#31-prompt-engineering

feat : 프롬프트와 모델 변경

Author: BaeJunH0

src/main/java/com/aiary/be/report/application/ReportClient.java

@@ -4,6 +4,7 @@
 import com.aiary.be.global.annotation.Client;
 
 import com.aiary.be.report.application.dto.AiResponse;
+import com.aiary.be.report.domain.ReportType;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.web.reactive.function.client.WebClient;
@@ -25,11 +26,11 @@ public ReportClient(WebClient.Builder webClientBuilder) {
         this.webClient = webClientBuilder.baseUrl("https://api.openai.com/v1").build();
     }
 
-    public Mono<AiResponse> analyze(Long userId, List<DiaryInfo> diaryInfos) {
-        String prompt = buildPrompt(diaryInfos);
+    public Mono<AiResponse> analyze(Long userId, List<DiaryInfo> diaryInfos, ReportType reportType) {
+        String prompt = buildPrompt(diaryInfos, reportType);
 
         Map<String, Object> requestBody = new HashMap<>();
-        requestBody.put("model", "gpt-3.5-turbo");
+        requestBody.put("model", "gpt-4o-mini");
         requestBody.put("messages", List.of(Map.of("role", "user", "content", prompt)));
 
         return webClient.post()
@@ -49,8 +50,20 @@ public Mono<AiResponse> analyze(Long userId, List<DiaryInfo> diaryInfos) {
                    });
     }
 
-    private String buildPrompt(List<DiaryInfo> diaryInfos) {
-        StringBuilder prompt = new StringBuilder("아래의 사용자의 일기를 읽고 우울감과 같은 심리 지표를 분석하고, 간단한 조언을 포함한 리포트를 작성해줘.\n");
+    private String buildPrompt(List<DiaryInfo> diaryInfos, ReportType reportType) {
+        String startStr = reportType.equals(ReportType.WEEKLY) ?
+                              "아래는 사용자가 이번 주에 작성한 일기야.\n" :  "아래는 사용자가 이번 달에 작성한 일기야.\n";
+        
+        StringBuilder prompt = new StringBuilder(startStr);
+        prompt.append("아래의 일기들을 읽고, 사용자에게 공감과 위로가 담긴 따뜻한 리포트를 작성해 줘.\n");
+        prompt.append("리포트에는 다음 내용을 포함해:\n");
+        prompt.append("1. 일기를 읽고 느낀 전반적인 감정과 분위기 요약\n");
+        prompt.append("2. 사용자에게 공감과 위로의 말\n");
+        prompt.append("3. 오늘 하루를 되돌아볼 수 있는 질문 또는 생각할 거리\n");
+        prompt.append("4. 앞으로의 마음가짐이나 행동에 도움이 될 수 있는 따뜻한 조언\n");
+        prompt.append("너무 차갑거나 기계적으로 분석하지 말고, 진심 어린 친구처럼 부드럽고 따뜻하게 작성해 줘.\n");
+        prompt.append("일기는 오직 Plain text로만 작성하고, 마크다운 문법을 사용하지 않아야 해\n");
+        prompt.append("그리고, 1, 2, 3, 4 이런식의 첨자를 사용하지 말고 평서문으로 작성해줘\n");
 
         for (int i = 0; i < diaryInfos.size(); i++) {
             prompt.append("\n일기 ")
@@ -62,8 +75,6 @@ private String buildPrompt(List<DiaryInfo> diaryInfos) {
                 .append("\n");
         }
 
-        prompt.append("\n\n분석의 형식은 두 줄로, 첫 줄에는 리포트의 제목을, 두 번째 줄에는 리포트의 내용을 적어줘");
-        
         log.info(prompt.toString());
         return prompt.toString();
     }

src/main/java/com/aiary/be/report/application/ReportFacade.java

@@ -48,10 +48,10 @@ public void createReport(ReportType reportType) {
 
         // 유저-주간 다이어리 쌍
         Map<Long, List<DiaryInfo>> userIdDiary = users.stream()
-                                                     .collect(Collectors.toMap(
-                                                         User::getId,
-                                                         user -> diaryService.readDiaryInfos(user.getId(), searchRange)
-                                                     ));
+                                                     .map(user -> Map.entry(user.getId(), diaryService.readDiaryInfos(user.getId(), searchRange)))
+                                                     .filter(entry -> !entry.getValue().isEmpty())
+                                                     .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
+        
         // 유저-주간 감정 쌍
         Map<Long, Emotion> userIdEmotion = userIdDiary.keySet().stream()
                                           .collect(Collectors.toMap(
@@ -70,15 +70,15 @@ public void createReport(ReportType reportType) {
 
         // 핵심 : 외부 api 호출을 non-blocking으로 하고, 이후에 응답을 모아서 한 번에 bulk로 저장
         Flux.fromIterable(userIdDiary.entrySet())
-            .flatMap(entry -> reportClient.analyze(entry.getKey(), entry.getValue()))
+            .flatMap(entry -> reportClient.analyze(entry.getKey(), entry.getValue(), reportType))
             .collectList()
             .publishOn(Schedulers.boundedElastic())
             .flatMap((apiResponses) -> {
                 List<Report> reports = new ArrayList<>();
                 for (AiResponse apiResponse : apiResponses) {
                     reports.add(new Report(
                         userIdMap.get(apiResponse.userId()),
-                        start + " - " + end,
+                        start + "부터 " + end + "까지의 리포트입니다.",
                         apiResponse.content(),
                         reportType,
                         start,