fixes

Author: eldadfux

src/Agents/Conversation.php

@@ -48,7 +48,7 @@ public function listen(callable $listener): self
     /**
      * Add a message to the conversation
      *
-     * @param  array<Message>  $attachments
+     * @param  array<int, mixed>  $attachments
      */
     public function message(Role $from, Message $message, array $attachments = []): self
     {
@@ -201,7 +201,7 @@ public function getTotalTokens(): int
     }
 
     /**
-     * @param  array<Message>  $attachments
+     * @param  array<int, mixed>  $attachments
      */
     protected function validateAttachments(Message $message, array $attachments): void
     {

src/Agents/Message.php

@@ -15,6 +15,8 @@ abstract class Message
 
     /**
      * Create a new message
+     *
+     * @param  array<int, mixed>  $attachments
      */
     public function __construct(string $content, ?string $role = null, array $attachments = [])
     {
@@ -62,13 +64,13 @@ public function addAttachment(Message $attachment): self
     }
 
     /**
-     * @param  array<Message>  $attachments
+     * @param  array<int, mixed>  $attachments
      */
     public function setAttachments(array $attachments): self
     {
         $this->attachments = [];
         foreach ($attachments as $attachment) {
-            if (! $attachment instanceof self) {
+            if (! $attachment instanceof Message) {
                 throw new \InvalidArgumentException('Attachments must be Message instances');
             }
             $this->attachments[] = $attachment;

src/Agents/Messages/Image.php

@@ -10,6 +10,7 @@ class Image extends Message
      * Create a new image message
      *
      * @param  string  $content  Binary content of the image
+     * @param  array<int, mixed>  $attachments
      */
     public function __construct(string $content, ?string $role = null, array $attachments = [])
     {

src/Agents/Messages/Text.php

@@ -8,6 +8,8 @@ class Text extends Message
 {
     /**
      * Create a new text message
+     *
+     * @param  array<int, mixed>  $attachments
      */
     public function __construct(string $content, ?string $role = null, array $attachments = [])
     {

tests/Agents/ConversationTest.php

@@ -48,7 +48,8 @@ public function testSendPassesAttachmentsToAdapter(): void
             ->send();
 
         $this->assertNotNull($adapter->lastSentMessage);
-        $this->assertCount(1, $adapter->lastSentMessage?->getAttachments() ?? []);
+        $this->assertInstanceOf(Message::class, $adapter->lastSentMessage);
+        $this->assertCount(1, $adapter->lastSentMessage->getAttachments());
     }
 
     public function testRejectsTooManyAttachments(): void
@@ -153,8 +154,7 @@ class ConversationFakeAdapter extends Adapter
     public function __construct(
         private readonly string $response,
         private readonly bool $supportsImageAttachments = true
-    ) {
-    }
+    ) {}
 
     public function supportsAttachments(): bool
     {