@@ -216,7 +216,8 @@ public function testGetUserMessageText()
216216 Message::ofAssistant ('How can I help you? ' ),
217217 );
218218
219- $ userText = $ messageBag ->getUserMessageText ();
219+ $ userMessage = $ messageBag ->getUserMessage ();
220+ $ userText = $ userMessage ?->asText();
220221
221222 $ this ->assertSame ('Hello, world! ' , $ userText );
222223 }
@@ -228,7 +229,9 @@ public function testGetUserMessageTextReturnsNullWithoutUserMessage()
228229 Message::ofAssistant ('It is time to sleep. ' ),
229230 );
230231
231- $ this ->assertNull ($ messageBag ->getUserMessageText ());
232+ $ userMessage = $ messageBag ->getUserMessage ();
233+
234+ $ this ->assertNull ($ userMessage ?->asText());
232235 }
233236
234237 public function testGetUserMessageTextWithMultipleTextParts ()
@@ -239,7 +242,8 @@ public function testGetUserMessageTextWithMultipleTextParts()
239242 Message::ofAssistant ('Response ' ),
240243 );
241244
242- $ userText = $ messageBag ->getUserMessageText ();
245+ $ userMessage = $ messageBag ->getUserMessage ();
246+ $ userText = $ userMessage ?->asText();
243247
244248 $ this ->assertSame ('Part one Part two Part three ' , $ userText );
245249 }
@@ -252,7 +256,8 @@ public function testGetUserMessageTextIgnoresNonTextContent()
252256 Message::ofAssistant ('Response ' ),
253257 );
254258
255- $ userText = $ messageBag ->getUserMessageText ();
259+ $ userMessage = $ messageBag ->getUserMessage ();
260+ $ userText = $ userMessage ?->asText();
256261
257262 // Should only return the text content, ignoring the image
258263 $ this ->assertSame ('Text content ' , $ userText );
0 commit comments