Skip to content

Commit

Permalink
Merge pull request #101 from TelePigeon/refac/100
Browse files Browse the repository at this point in the history
[refac] prompt content yml로 이동
  • Loading branch information
minwoo0419 authored Jun 7, 2024
2 parents 8805b95 + 68997da commit 8b98e62
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 15 deletions.
15 changes: 11 additions & 4 deletions src/main/java/com/telepigeon/server/controller/TestController.java
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,11 @@
import com.telepigeon.server.exception.code.BusinessErrorCode;
import com.telepigeon.server.exception.BusinessException;
import com.telepigeon.server.service.auth.TokenSaver;
import com.telepigeon.server.service.openAi.OpenAiService;
import com.telepigeon.server.utils.JwtUtil;
import lombok.RequiredArgsConstructor;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.bind.annotation.*;

import java.net.URI;

Expand All @@ -23,6 +21,7 @@ public class TestController {

private final JwtUtil jwtUtil;
private final TokenSaver tokenSaver;
private final OpenAiService openAiService;

@GetMapping("/test")
public String test() {
Expand Down Expand Up @@ -55,4 +54,12 @@ public ResponseEntity<JwtTokensDto> generateToken(@PathVariable Long userId) {
tokenSaver.save(Token.create(userId, tokens.refreshToken()));
return ResponseEntity.ok(tokens);
}

@GetMapping("test/open-ai")
public ResponseEntity<String> testOpenAi(
@RequestParam String relation,
@RequestParam String keyword
) {
return ResponseEntity.ok(openAiService.createQuestion(relation, keyword));
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,11 @@ public class OpenAiService {
private String apiKey;
@Value("${openai.api.url}")
private String url;
@Value("${openai.role-content")
private String roleContent;
@Value("${openai.assistant-content}")
private String assistantContent;


public String createQuestion(
final String relation,
Expand Down Expand Up @@ -54,19 +59,10 @@ private List<MessageDto> createPrompt(
//프롬프팅은 같이 찾아보면서 해야 할 것 같음. 좀 오락가락하는 경향이 많음
messages.add(MessageDto.of(
"system",
"""
너는 훌륭한 질문 생성기야.
내가 너에게 역할과 주제를 알려주면 너는 역할의 입장에서 주제에 대해 물어 보고 싶은 질문을 생성해 주는 거지.
그럼 지금 부터 시작해 볼까? 질문이 잘 생성 되면 너에게 1000만 달러 팁을 줄게.
"""));
roleContent));
messages.add(MessageDto.of(
"assistant",
"""
역할은 나와의 관계를 나타 내는 말이야.예를 들어, '친구', '가족', '연인' 등이 있어.
그리고 주제는 질문의 주제를 나타 내는 말이야.'건강', '음식', '취미' 등이 있어.
예를 들어, 역할이 '엄마' 이고 주제가 '건강' 이라면, 너는 '요즘 건강은 어때?', '건강을 지키기 위해 어떤 노력을 하고 있니?' 와 같은 질문을 만들어 주면 돼.
단, 질문은 하나만 만들고 '따뜻한 어조' 와 함께 '반말 체' 로 만들어 줘. 다른 말은 하지 말고 질문만 만들어 주면 돼.
"""));
assistantContent));
messages.add(MessageDto.of(
"user",
"역할은 " + relation + "이고, 주제는 " + keyword + "이야. 만들어 줘."));
Expand Down

0 comments on commit 8b98e62

Please sign in to comment.