Allow tuple header inputs in transport send options

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

Author: cursoragent

packages/ai/src/chatTransport.test.ts

@@ -374,7 +374,7 @@ describe("TriggerChatTransport", function () {
       messageId: undefined,
       messages: [],
       abortSignal: undefined,
-      headers: [["x-tuple-header", "tuple-value"]] as unknown as Record<string, string>,
+      headers: [["x-tuple-header", "tuple-value"]],
     });
 
     const chunks = await readChunks(stream);

packages/ai/src/chatTransport.ts

@@ -62,6 +62,12 @@ type TriggerTaskRequestBody = {
   options?: TriggerTaskRequestOptions;
 };
 
+type TriggerChatRequestOptionsWithTupleHeaders = Omit<ChatRequestOptions, "headers"> & {
+  headers?:
+    | ChatRequestOptions["headers"]
+    | Array<[string, string]>;
+};
+
 type TriggerChatTransportCommonOptions<
   UI_MESSAGE extends UIMessage = UIMessage,
 > = {
@@ -158,7 +164,7 @@ export class TriggerChatTransport<
       messageId: string | undefined;
       messages: UI_MESSAGE[];
       abortSignal: AbortSignal | undefined;
-    } & ChatRequestOptions
+    } & TriggerChatRequestOptionsWithTupleHeaders
   ): Promise<ReadableStream<UIMessageChunk>> {
     const transportRequest = createTransportRequest<UI_MESSAGE>(options);
     const payload = await this.payloadMapper(transportRequest);
@@ -342,7 +348,7 @@ function createTransportRequest<UI_MESSAGE extends UIMessage>(
     messageId: string | undefined;
     messages: UI_MESSAGE[];
     abortSignal: AbortSignal | undefined;
-  } & ChatRequestOptions
+  } & TriggerChatRequestOptionsWithTupleHeaders
 ): TriggerChatTransportRequest<UI_MESSAGE> {
   return {
     chatId: options.chatId,

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

@@ -122,3 +122,15 @@ it("accepts typed stream definition objects", function () {
 
   expectTypeOf(transport).toBeObject();
 });
+
+it("accepts tuple-style headers in sendMessages options", function () {
+  const transport = new TriggerChatTransport({
+    task: "ai-chat",
+    accessToken: "pk_test",
+  });
+
+  type SendMessagesParams = Parameters<typeof transport.sendMessages>[0];
+  const tupleHeaders: SendMessagesParams["headers"] = [["x-header", "x-value"]];
+  expectTypeOf(transport.sendMessages).toBeFunction();
+  void tupleHeaders;
+});