@@ -264,8 +264,25 @@ def test_prompt_loading(self, mock_load_prompt):
264264 # Verify the loaded prompt was used in the API call
265265 call_args = self .mock_client .chat .completions .create .call_args
266266 messages = call_args [1 ]["messages" ]
267- self .assertEqual (len (messages ), 1 )
268- self .assertIn ("Test prompt content" , messages [0 ]["content" ])
267+ self .assertEqual (len (messages ), 2 )
268+ system_message = messages [0 ]
269+ user_message = messages [1 ]
270+
271+ self .assertEqual (system_message .get ("role" ), "system" )
272+ system_content = system_message .get ("content" , [])
273+ self .assertTrue (system_content )
274+ self .assertIn (
275+ "Test prompt content" ,
276+ [part .get ("text" , "" ) for part in system_content if isinstance (part , dict )],
277+ )
278+
279+ self .assertEqual (user_message .get ("role" ), "user" )
280+ user_content = user_message .get ("content" , [])
281+ self .assertTrue (user_content )
282+ self .assertIn (
283+ "Sentence: Test sentence with more words." ,
284+ [part .get ("text" , "" ) for part in user_content if isinstance (part , dict )],
285+ )
269286
270287 def test_sentence_splitting (self ):
271288 """Test that text is properly split into sentences."""
0 commit comments