ability to add messages from base64 encoded data

nunodonato · Jul 19, 2024 · 870afb6 · 870afb6
1 parent 9ec5514
commit 870afb6
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/src/Messages.php b/src/Messages.php
@@ -60,6 +60,34 @@ public function addUserImageMessage(string $url, ?string $text = null): self
         return $this;
     }
 
+    public function addUserImageMessageFromBase64(string $base64Encoded, string $mimeType, ?string $text = null): self
+    {
+        $imageMessage = [
+            'type' => 'image',
+            'source' => [
+                'type' => 'base64',
+                'data' => $base64Encoded,
+                'media_type' => $mimeType
+            ],
+        ];
+        $message = [$imageMessage];
+
+        if ($text) {
+            $textMessage = [
+                'role' => self::ROLE_USER,
+                'content' => $text
+            ];
+            $message[] = $textMessage;
+        }
+
+        $this->messages[] = [
+            'role' => self::ROLE_USER,
+            'content' => $message,
+        ];
+
+        return $this;
+    }
+
     public function addAssistantTextMessage(string $text): self
     {
         return $this->addMessage(self::ROLE_ASSISTANT, $text);