Skip to content

Commit fb1f268

Browse files
committed
add test case
1 parent 3a78e04 commit fb1f268

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

service/src/test/java/com/launchableinc/openai/service/ChatCompletionTest.java

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,27 @@ void createChatCompletion_with_json_mode() {
101101
assertTrue(isValidJson(choices.getMessage().getContent()));
102102
}
103103

104+
@Test
105+
void createChatCompletion_with_gpt4o() {
106+
final List<ChatMessage> messages = new ArrayList<>();
107+
final ChatMessage systemMessage = new ChatMessage(ChatMessageRole.SYSTEM.value(),
108+
"You are a cat and will speak as such.");
109+
messages.add(systemMessage);
110+
111+
ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest
112+
.builder()
113+
.model("gpt-4o")
114+
.messages(messages)
115+
.n(5)
116+
.maxTokens(50)
117+
.logitBias(new HashMap<>())
118+
.build();
119+
120+
List<ChatCompletionChoice> choices = service.createChatCompletion(chatCompletionRequest)
121+
.getChoices();
122+
assertEquals(5, choices.size());
123+
}
124+
104125
private boolean isValidJson(String jsonString) {
105126
ObjectMapper objectMapper = new ObjectMapper();
106127
try {

0 commit comments

Comments
 (0)