ref

Author: Guikingone

src/agent/src/Chat/MessageStore/CacheStore.php

@@ -56,6 +56,6 @@ public function clear(): void
 
     public function withSession(AbstractUid&TimeBasedUidInterface $session): MessageStoreInterface&SessionAwareMessageStoreInterface
     {
-        return new $this($this->cache, $session->toRfc4122(), $this->ttl);
+        return new self($this->cache, $session->toRfc4122(), $this->ttl);
     }
 }

src/agent/src/Chat/MessageStore/InMemoryStore.php

@@ -15,13 +15,14 @@
 use Symfony\AI\Agent\Chat\SessionAwareMessageStoreInterface;
 use Symfony\AI\Platform\Message\MessageBag;
 use Symfony\AI\Platform\Message\MessageBagInterface;
+use Symfony\AI\Platform\Message\MessageInterface;
 use Symfony\Component\Uid\AbstractUid;
 use Symfony\Component\Uid\TimeBasedUidInterface;
 
 final class InMemoryStore implements MessageStoreInterface, SessionAwareMessageStoreInterface
 {
     /**
-     * @var MessageBagInterface[]
+     * @var MessageInterface[]
      */
     private array $messages;
     private (AbstractUid&TimeBasedUidInterface)|null $session = null;
@@ -43,7 +44,7 @@ public function load(): MessageBagInterface
             return new MessageBag(...$this->messages);
         }
 
-        return $this->messages[$this->session->toRfc4122()] ?? new MessageBag();
+        return new MessageBag($this->messages[$this->session->toRfc4122()]);
     }
 
     public function clear(): void