|
| 1 | +package com.mbtips.domain.addition.service; |
| 2 | + |
| 3 | +import com.mbtips.clova.dto.Message; |
| 4 | +import com.mbtips.common.mbtiinfo.MbtiTraits; |
| 5 | +import com.mbtips.common.mbtiinfo.MbtiType; |
| 6 | +import com.mbtips.domain.conversation.service.ConversationService; |
| 7 | +import com.mbtips.domain.converstation.Conversation; |
| 8 | +import com.mbtips.domain.message.dto.response.GetMessageResponseDto; |
| 9 | +import com.mbtips.domain.virtualfriend.VirtualFriend; |
| 10 | +import com.mbtips.domain.virtualfriend.VirtualFriendService; |
| 11 | +import com.mbtips.domain.virtualfriend.response.VirtualFriendInfoResponse; |
| 12 | +import com.mbtips.message.application.manager.MessageManager; |
| 13 | +import com.mbtips.message.application.service.MessageService; |
| 14 | +import lombok.RequiredArgsConstructor; |
| 15 | +import lombok.extern.slf4j.Slf4j; |
| 16 | +import org.springframework.stereotype.Service; |
| 17 | + |
| 18 | +import java.util.ArrayList; |
| 19 | +import java.util.List; |
| 20 | + |
| 21 | +@Service |
| 22 | +@RequiredArgsConstructor |
| 23 | +@Slf4j |
| 24 | +public class AdditionService { |
| 25 | + |
| 26 | + public final MessageManager messageManager; |
| 27 | + public final ConversationService conversationService; |
| 28 | + public final MessageService messageService; |
| 29 | + public final VirtualFriendService virtualFriendService; |
| 30 | + public String requestConversationTips(Long virtualFriendId) { |
| 31 | + VirtualFriendInfoResponse virtualFriend = virtualFriendService.findFriendInfoById(virtualFriendId); |
| 32 | + |
| 33 | + String mbtiTrait = MbtiTraits.getTrait(MbtiType.valueOf(virtualFriend.mbti())); |
| 34 | + mbtiTrait = mbtiTrait.substring(mbtiTrait.indexOf("특징을 설명해주면 다음과 같아")); |
| 35 | + String requestPrompt = mbtiTrait + ", 이 mbti와 대화할 때, 꿀팁 알려줘"; |
| 36 | + log.debug("requstPrompt : {}", requestPrompt); |
| 37 | + String result = messageManager.messageRequest(requestPrompt); |
| 38 | + return result; |
| 39 | + } |
| 40 | + |
| 41 | + public String requestTemperature(Long conversationId) { |
| 42 | + Conversation conversation = conversationService.findById(conversationId); |
| 43 | + List<GetMessageResponseDto> messages = messageService.getMessagesOfConversationId(conversationId); |
| 44 | + |
| 45 | + String[] messageList = { |
| 46 | + "사용자: 안녕하세요, 오늘 기분이 어떠세요?", |
| 47 | + "AI: 안녕하세요, 오늘 기분이 활기차고 즐거워요.", |
| 48 | + "사용자: 오늘 특별한 계획이 있으세요?", |
| 49 | + "AI: 네, 친구와 함께 신나는 시간을 보낼 예정입니다." |
| 50 | + }; |
| 51 | + String requestConversation = String.join(", ", messageList); |
| 52 | + String prompt = requestConversation + "현재까지의 대화 온도 측정해서 0~100 사이의 숫자로만 대답해줘 냉정하게 판단해주고, 숫자로만 대답해."; |
| 53 | + log.debug("prompt : {}", prompt); |
| 54 | + String temperatureResponse = messageManager.messageRequest(prompt); |
| 55 | + String result = temperatureResponse.replaceAll("[^0-9]", ""); |
| 56 | + return result; |
| 57 | + } |
| 58 | + |
| 59 | + public String requestRecommendTopic(Long virtualFriendId) { |
| 60 | + VirtualFriendInfoResponse virtualFriend = virtualFriendService.findFriendInfoById(virtualFriendId); |
| 61 | + String mbtiTrait = MbtiTraits.getTrait(MbtiType.valueOf(virtualFriend.mbti())); |
| 62 | + mbtiTrait = mbtiTrait.substring(mbtiTrait.indexOf("특징을 설명해주면 다음과 같아")); |
| 63 | + String requestPrompt = mbtiTrait + ", 이 mbti와 대화할 때, 맞는 대화 주제 키워드 5개만 추천해줘, 짧게 단어로만 이야기해줘"; |
| 64 | + |
| 65 | + String result = messageManager.messageRequest(requestPrompt); |
| 66 | + log.debug("result : {}", result); |
| 67 | + return result; |
| 68 | + } |
| 69 | +} |
0 commit comments