File tree Expand file tree Collapse file tree 1 file changed +21
-0
lines changed
service/src/test/java/com/launchableinc/openai/service Expand file tree Collapse file tree 1 file changed +21
-0
lines changed Original file line number Diff line number Diff 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 {
You can’t perform that action at this time.
0 commit comments