pass through thread id

Author: tymondesigns

src/Agents/Agent.php

@@ -101,7 +101,7 @@ public function __construct(
         protected array $middleware = [],
     ) {
         $this->prompt = self::buildPromptTemplate($prompt, $strict, $initialPromptVariables);
-        $this->memory = self::buildMemory($this->prompt, $this->memoryStore);
+        $this->memory = self::buildMemory($this->prompt, $runtimeConfig?->threadId, $this->memoryStore);
         $this->middleware = [...$this->defaultMiddleware(), ...$this->middleware];
 
         // Reset the prompt to only the message placeholders, since the initial
@@ -432,6 +432,7 @@ protected function invokePipeline(
         $messages = Utils::toMessageCollection($messages);
 
         $this->memory
+            ->setThreadId($config->threadId)
             ->setMessages($this->memory->getMessages()->merge($messages))
             ->setVariables([
                 ...$this->initialPromptVariables,
@@ -502,9 +503,12 @@ protected static function buildPromptTemplate(
     /**
      * Build the memory instance for the agent.
      */
-    protected static function buildMemory(ChatPromptTemplate $prompt, ?Store $memoryStore = null): ChatMemoryContract
-    {
-        $memoryStore ??= new InMemoryStore(Str::uuid7()->toString());
+    protected static function buildMemory(
+        ChatPromptTemplate $prompt,
+        ?string $threadId = null,
+        ?Store $memoryStore = null,
+    ): ChatMemoryContract {
+        $memoryStore ??= new InMemoryStore($threadId ?? Str::uuid7()->toString());
         $memoryStore->setMessages($prompt->messages->withoutPlaceholders());
 
         return new ChatMemory($memoryStore);

src/Contracts/ChatMemory.php

@@ -50,4 +50,9 @@ public function getVariables(): array;
      * Delegates to the underlying store - the store is the source of truth for threadId.
      */
     public function getThreadId(): string;
+
+    /**
+     * Set the thread ID for this memory instance.
+     */
+    public function setThreadId(string $threadId): static;
 }

src/Memory/ChatMemory.php

@@ -99,4 +99,11 @@ public function getThreadId(): string
     {
         return $this->store->getThreadId();
     }
+
+    public function setThreadId(string $threadId): static
+    {
+        $this->store->setThreadId($threadId);
+
+        return $this;
+    }
 }

src/Memory/Contracts/Store.php

@@ -40,4 +40,9 @@ public function reset(): void;
      * Get the thread ID for this store.
      */
     public function getThreadId(): string;
+
+    /**
+     * Set the thread ID for this store.
+     */
+    public function setThreadId(string $threadId): void;
 }

src/Memory/Stores/CacheStore.php

@@ -70,4 +70,9 @@ public function getThreadId(): string
     {
         return $this->threadId;
     }
+
+    public function setThreadId(string $threadId): void
+    {
+        $this->threadId = $threadId;
+    }
 }

src/Memory/Stores/InMemoryStore.php

@@ -44,4 +44,9 @@ public function getThreadId(): string
     {
         return $this->threadId;
     }
+
+    public function setThreadId(string $threadId): void
+    {
+        $this->threadId = $threadId;
+    }
 }