automatically detect and join blocks of the same role

Author: nunodonato

src/Messages.php

@@ -18,6 +18,7 @@ public function messages(): array
     {
         return $this->messages;
     }
+
     public function addUserTextMessage(string $text): self
     {
         return $this->addMessage(self::ROLE_USER, $text);
@@ -41,7 +42,7 @@ public function addUserImageMessage(string $url, ?string $text = null): self
                 'media_type' => $mediaType
             ],
         ];
-        $message = [ $imageMessage ];
+        $message = [$imageMessage];
 
         if ($text) {
             $textMessage = [
@@ -81,16 +82,31 @@ public function addMessage(string $role, string|array $content): self
                 if ($block['type'] == 'text' && !array_key_exists('text', $block)) {
                     throw new \InvalidArgumentException('Block text property is required for text type. Index: ' . $i);
                 }
-                if ($block['type'] == 'image' && (!array_key_exists('media_type', $block) || !array_key_exists('data', $block))) {
-                    throw new \InvalidArgumentException('Block media type and data property are required for image type. Index: ' . $i);
+                if ($block['type'] == 'image' && (!array_key_exists('media_type', $block) || !array_key_exists(
+                            'data',
+                            $block
+                        ))) {
+                    throw new \InvalidArgumentException(
+                        'Block media type and data property are required for image type. Index: ' . $i
+                    );
                 }
             }
+        } else {
+            // let's standardize the way the blocks are passed
+            $content = [
+                [
+                    'type' => 'text',
+                    'text' => $content
+                ]
+            ];
         }
 
         if (count($this->messages) > 0) {
             $lastMessage = $this->messages[count($this->messages) - 1];
             if ($lastMessage['role'] === $role) {
-                throw new \InvalidArgumentException('Roles must alternate between "user" and "assistant". Message ' . count($this->messages) . ' has the same role as the previous message.');
+                // merge messages of the same role
+                $this->messages[count($this->messages) - 1]['content'] = array_merge($lastMessage['content'], $content);
+                return $this;
             }
         }
 

tests/MessagesTest.php

@@ -27,11 +27,15 @@ public function test_can_add_object_text_message()
         $this->assertCount(1, $messages->messages());
     }
 
-    public function test_prevents_non_alternated_roles()
+    public function test_joins_continuous_messages_with_the_same_role()
     {
         $messages = new Messages();
         $messages->addMessage(Messages::ROLE_USER, 'Hello');
-        $this->expectExceptionMessage('Roles must alternate between "user" and "assistant".');
         $messages->addMessage(Messages::ROLE_USER, 'How are you?');
+        $messages->addMessage(Messages::ROLE_ASSISTANT, 'I am fine');
+        $result = $messages->messages();
+        $this->assertCount(2, $result);
+        $this->assertCount(2, $result[0]['content']);
+        $this->assertCount(1, $result[1]['content']);
     }
 }