Add type coverage for custom run store options

Co-authored-by: Eric Allam <eric@trigger.dev>

Author: cursoragent

packages/ai/src/chatTransport.types.test.ts

@@ -3,6 +3,7 @@ import type { InferUIMessageChunk, UIMessage } from "ai";
 import {
   createTriggerChatTransport,
   TriggerChatTransport,
+  InMemoryTriggerChatRunStore,
   TriggerChatTransportOptions,
   type TriggerChatOnError,
   type TriggerChatTransportError,
@@ -176,3 +177,14 @@ it("accepts tuple-style headers in sendMessages options", function () {
   expectTypeOf(transport.sendMessages).toBeFunction();
   void tupleHeaders;
 });
+
+it("accepts custom run store implementations via options typing", function () {
+  const runStore = new InMemoryTriggerChatRunStore();
+  const transport = new TriggerChatTransport({
+    task: "ai-chat",
+    accessToken: "pk_test",
+    runStore,
+  });
+
+  expectTypeOf(transport).toBeObject();
+});