대화 부가서비스 관련 API구현

Author: yhj0214

api/src/main/java/com/mbtips/domain/addition/controller/AdditionController.java

@@ -24,7 +24,6 @@ public class AdditionController {
     @Operation(summary = "대화 꿀팁", description = "해당 AI와의 대화 꿀팁을 반환합니다.")
     public ApiResponse<String> requestConversationTips(@PathVariable Long virtualFriendId, @LoginUser User user){
         String result = additionService.requestConversationTips(virtualFriendId);
-
         return ApiResponse.success(result);
     }
 

api/src/main/java/com/mbtips/domain/addition/service/AdditionService.java

@@ -1,26 +1,39 @@
 package com.mbtips.domain.addition.service;
 
 import com.mbtips.clova.dto.Message;
+import com.mbtips.common.mbtiinfo.MbtiTraits;
+import com.mbtips.common.mbtiinfo.MbtiType;
 import com.mbtips.domain.conversation.service.ConversationService;
 import com.mbtips.domain.converstation.Conversation;
 import com.mbtips.domain.message.dto.response.GetMessageResponseDto;
+import com.mbtips.domain.virtualfriend.VirtualFriend;
+import com.mbtips.domain.virtualfriend.VirtualFriendService;
+import com.mbtips.domain.virtualfriend.response.VirtualFriendInfoResponse;
 import com.mbtips.message.application.manager.MessageManager;
 import com.mbtips.message.application.service.MessageService;
 import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 
 import java.util.ArrayList;
 import java.util.List;
 
 @Service
 @RequiredArgsConstructor
+@Slf4j
 public class AdditionService {
 
     public final MessageManager messageManager;
     public final ConversationService conversationService;
     public final MessageService messageService;
+    public final VirtualFriendService virtualFriendService;
     public String requestConversationTips(Long virtualFriendId) {
-        String requestPrompt = "대화 주제 추천해줘";
+        VirtualFriendInfoResponse virtualFriend = virtualFriendService.findFriendInfoById(virtualFriendId);
+
+        String mbtiTrait = MbtiTraits.getTrait(MbtiType.valueOf(virtualFriend.mbti()));
+        mbtiTrait = mbtiTrait.substring(mbtiTrait.indexOf("특징을 설명해주면 다음과 같아"));
+        String requestPrompt = mbtiTrait + ", 이 mbti와 대화할 때, 꿀팁 알려줘";
+        log.debug("requstPrompt : {}", requestPrompt);
         String result = messageManager.messageRequest(requestPrompt);
         return result;
     }
@@ -29,16 +42,28 @@ public String requestTemperature(Long conversationId) {
         Conversation conversation = conversationService.findById(conversationId);
         List<GetMessageResponseDto> messages =  messageService.getMessagesOfConversationId(conversationId);
 
-        String prompt = "현재까지의 대화 온도 측정해서 0~100 사이의 숫자로만 대답해줘";
-        String result = messageManager.messageRequest(prompt);
+        String[] messageList = {
+                "사용자: 안녕하세요, 오늘 기분이 어떠세요?",
+                "AI: 안녕하세요, 오늘 기분이 활기차고 즐거워요.",
+                "사용자: 오늘 특별한 계획이 있으세요?",
+                "AI: 네, 친구와 함께 신나는 시간을 보낼 예정입니다."
+        };
+        String requestConversation = String.join(", ", messageList);
+        String prompt = requestConversation + "현재까지의 대화 온도 측정해서 0~100 사이의 숫자로만 대답해줘 냉정하게 판단해주고, 숫자로만 대답해.";
+        log.debug("prompt : {}", prompt);
+        String temperatureResponse = messageManager.messageRequest(prompt);
+        String result = temperatureResponse.replaceAll("[^0-9]", "");
         return result;
     }
 
     public String requestRecommendTopic(Long virtualFriendId) {
+        VirtualFriendInfoResponse virtualFriend = virtualFriendService.findFriendInfoById(virtualFriendId);
+        String mbtiTrait = MbtiTraits.getTrait(MbtiType.valueOf(virtualFriend.mbti()));
+        mbtiTrait = mbtiTrait.substring(mbtiTrait.indexOf("특징을 설명해주면 다음과 같아"));
+        String requestPrompt = mbtiTrait + ", 이 mbti와 대화할 때, 맞는 대화 주제 키워드 5개만 추천해줘, 짧게 단어로만 이야기해줘";
 
-        String prompt = "지금까지 말한 MBTI와 맞는 대화 주제 5개만 추천해줘";
-        String result = messageManager.messageRequest(prompt);
-
+        String result = messageManager.messageRequest(requestPrompt);
+        log.debug("result : {}", result);
         return result;
     }
 }

api/src/main/java/com/mbtips/domain/virtualfriend/VirtualFriendController.java

@@ -33,7 +33,7 @@ public ApiResponse<List<VirtualFriendResponse>> getVirtualFriends(@LoginUser Use
     @GetMapping("/{virtualFriendId}")
     @Operation(summary = "가상친구 상세 조회", description = "virtualFriendId를 기준으로 가상친구를 상세조회합니다.")
     public ApiResponse<VirtualFriendInfoResponse> getVirtualFriend(@LoginUser User user, @PathVariable Long virtualFriendId) {
-        VirtualFriendInfoResponse result = virtualFriendService.findById(virtualFriendId);
+        VirtualFriendInfoResponse result = virtualFriendService.findFriendInfoById(virtualFriendId);
         return ApiResponse.success(result);
     }
 

api/src/main/java/com/mbtips/domain/virtualfriend/VirtualFriendService.java

@@ -91,7 +91,7 @@ public String makePrompt(Long conversationId) {
         return result;
     }
 
-    public VirtualFriendInfoResponse findById(Long virtualFriendId) {
+    public VirtualFriendInfoResponse findFriendInfoById(Long virtualFriendId) {
         VirtualFriend virtualFriend = virtualFriendRepository.findById(virtualFriendId);
         List<String> interest = interestRepository.findTopicsByVirtualFriendId(virtualFriendId);
         log.debug("interest : {}", interest);

repository/src/main/resources/local/application-repository.yaml

@@ -3,7 +3,7 @@ spring:
     driver-class-name: com.mysql.cj.jdbc.Driver
     url: "jdbc:mysql://localhost:3306/problem"
     username: root
-    password: "!qotjrdyd1"
+    password: root
 
   jpa:
     show-sql: true