diff --git a/src/Capabilities/Chat/Parameters/Messages.php b/src/Capabilities/Chat/Parameters/Messages.php index 4f21ff4..85e776a 100644 --- a/src/Capabilities/Chat/Parameters/Messages.php +++ b/src/Capabilities/Chat/Parameters/Messages.php @@ -6,7 +6,7 @@ class Messages { public array $messages = []; - public function system(string $content, ?string $name = null): self + public function system(string|array $content, ?string $name = null): self { $message = [ 'role' => 'system', @@ -20,7 +20,7 @@ public function system(string $content, ?string $name = null): self return $this; } - public function user(string $content, ?string $name = null): self + public function user(string|array $content, ?string $name = null): self { $message = [ 'role' => 'user', @@ -34,7 +34,7 @@ public function user(string $content, ?string $name = null): self return $this; } - public function assistant(?string $content = null, ?string $name = null, ?array $toolCalls): self + public function assistant(string|array|null $content = null, ?string $name = null, ?array $toolCalls): self { $message = [ 'role' => 'assistant', @@ -49,7 +49,7 @@ public function assistant(?string $content = null, ?string $name = null, ?array return $this; } - public function tool(string $content, string $toolCallId): self + public function tool(string|array $content, string $toolCallId): self { $message = [ 'role' => 'system', diff --git a/tests/Unit/ChatTest.php b/tests/Unit/ChatTest.php index 8cda1b8..f2893df 100644 --- a/tests/Unit/ChatTest.php +++ b/tests/Unit/ChatTest.php @@ -44,12 +44,32 @@ public function test_chat_json_response(): void public function test_chat_stream(): void { - $response = OpenAI::chat()->stream(function (string $newChunk, string $message) { - })->create( + $response = OpenAI::chat()->stream(function (string $newChunk, string $message) {})->create( model: 'gpt-4o-mini', messages: (new Messages)->system('You are a helpful assistant.')->user('Hello!'), ); $this->assertTrue($response instanceof StreamedChatResponse); $this->assertIsArray($response->choices); } + + public function test_chat_image_url(): void + { + $response = OpenAI::chat()->create( + model: 'gpt-4o-mini', + messages: (new Messages)->system('You are a helpful assistant.')->user([ + [ + 'type' => 'text', + 'text' => 'Describe what\'s on the attached photo', + ], + [ + 'type' => 'image_url', + 'image_url' => [ + 'url' => 'https://img-cdn.pixlr.com/image-generator/history/65bb506dcb310754719cf81f/ede935de-1138-4f66-8ed7-44bd16efc709/medium.webp', + ], + ], + ]), + ); + $this->assertTrue($response instanceof ChatResponse); + $this->assertIsArray($response->choices); + } }