Skip to content

Commit c60073c

Browse files
committed
add o1 unit test
1 parent 5ec8a82 commit c60073c

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

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

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -957,4 +957,18 @@ void toolCallingStrictTest(){
957957
assertEquals("asc",arguments.get("order_by").asText());
958958
}
959959

960+
@Test
961+
void o1ModelTest(){
962+
ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder()
963+
.model("o1-mini")
964+
.messages(Arrays.asList(new UserMessage("9.11和9.8比较,那个大")))
965+
.build();
966+
ChatCompletionResult chatCompletion = service.createChatCompletion(chatCompletionRequest);
967+
assertNotNull(chatCompletion);
968+
assertNotNull(chatCompletion.getChoices());
969+
ChatCompletionChoice choice = chatCompletion.getChoices().get(0);
970+
assertNotNull(choice);
971+
assertNotNull(choice.getMessage());
972+
assertNotNull(choice.getMessage().getContent());
973+
}
960974
}

0 commit comments

Comments
 (0)