Trigger AI SDK chat

.changeset/curly-radios-visit.md

@@ -0,0 +1,10 @@
+---
+"@trigger.dev/ai": minor
+---
+
+Add a new `@trigger.dev/ai` package with:
+
+- `ai.tool(...)` and `ai.currentToolOptions()` helpers for AI SDK tool calling ergonomics
+- a typed `TriggerChatTransport` that plugs into AI SDK UI `useChat()` and runs chat backends as Trigger.dev tasks
+- rich default task payloads (`chatId`, trigger metadata, messages, request context) with optional payload mapping
+- reconnect-aware stream handling on top of Trigger.dev Realtime Streams v2

docs/tasks/schemaTask.mdx

@@ -81,7 +81,7 @@ await myTask.trigger({ name: "Alice", age: 30, dob: "2020-01-01" }); // this is
 The `ai.tool` function allows you to create an AI tool from an existing `schemaTask` to use with the Vercel [AI SDK](https://vercel.com/docs/ai-sdk):
 
 ```ts
-import { ai } from "@trigger.dev/sdk/ai";
+import { ai } from "@trigger.dev/ai";
 import { schemaTask } from "@trigger.dev/sdk";
 import { z } from "zod";
 import { generateText } from "ai";
@@ -118,7 +118,7 @@ You can also pass the `experimental_toToolResultContent` option to the `ai.tool`
 ```ts
 import { openai } from "@ai-sdk/openai";
 import { Sandbox } from "@e2b/code-interpreter";
-import { ai } from "@trigger.dev/sdk/ai";
+import { ai } from "@trigger.dev/ai";
 import { schemaTask } from "@trigger.dev/sdk";
 import { generateObject } from "ai";
 import { z } from "zod";
@@ -183,7 +183,7 @@ export const chartTool = ai.tool(chartTask, {
 You can access the current tool execution options inside the task run function using the `ai.currentToolOptions()` function:
 
 ```ts
-import { ai } from "@trigger.dev/sdk/ai";
+import { ai } from "@trigger.dev/ai";
 import { schemaTask } from "@trigger.dev/sdk";
 import { z } from "zod";
 

docs/tasks/streams.mdx

@@ -517,6 +517,95 @@ const { parts, error } = useRealtimeStream(streamDef, runId, {
 });
 ```
 
+## AI SDK `useChat` transport with Trigger.dev tasks
+
+If you want to use AI SDK UI's `useChat()` on the frontend and run the backend as a Trigger.dev task,
+use the `@trigger.dev/ai` transport.
+
+### Install
+
+```bash
+npm add @trigger.dev/ai @ai-sdk/react ai
+```
+
+### Define a typed stream
+
+```ts
+// app/streams.ts
+import { streams } from "@trigger.dev/sdk";
+import { UIMessageChunk } from "ai";
+
+export const aiStream = streams.define<UIMessageChunk>({
+  id: "ai",
+});
+```
+
+### Create a task that accepts rich chat transport payload
+
+```ts
+// trigger/chat-task.ts
+import { openai } from "@ai-sdk/openai";
+import type { TriggerChatTransportPayload } from "@trigger.dev/ai";
+import { task } from "@trigger.dev/sdk";
+import { convertToModelMessages, streamText, UIMessage } from "ai";
+import { aiStream } from "@/app/streams";
+
+type ChatPayload = TriggerChatTransportPayload<UIMessage>;
+
+export const aiChatTask = task({
+  id: "ai-chat",
+  run: async (payload: ChatPayload) => {
+    const result = streamText({
+      model: openai("gpt-4o"),
+      messages: convertToModelMessages(payload.messages),
+    });
+
+    const { waitUntilComplete } = aiStream.pipe(result.toUIMessageStream());
+    await waitUntilComplete();
+  },
+});
+```
+
+### Use `useChat()` with Trigger chat transport
+
+```tsx
+"use client";
+
+import { useChat } from "@ai-sdk/react";
+import { TriggerChatTransport } from "@trigger.dev/ai";
+import { aiStream } from "@/app/streams";
+
+export function Chat({ triggerToken }: { triggerToken: string }) {
+  const chat = useChat({
+    transport: new TriggerChatTransport({
+      task: "ai-chat",
+      stream: aiStream,
+      accessToken: triggerToken,
+      timeoutInSeconds: 120,
+    }),
+  });
+
+  return (
+    <form
+      onSubmit={(event) => {
+        event.preventDefault();
+        chat.sendMessage({ text: "Hello!" });
+      }}
+    >
+      <button type="submit">Send</button>
+    </form>
+  );
+}
+```
+
+The default payload sent to your task is a rich, typed object that includes:
+
+- `chatId`
+- `trigger` (`"submit-message"` or `"regenerate-message"`)
+- `messageId`
+- `messages`
+- `request` (`headers`, `body`, and `metadata`)
+
 ## Complete Example: AI Streaming
 
 ### Define the stream

packages/ai/package.json

@@ -0,0 +1,75 @@
+{
+  "name": "@trigger.dev/ai",
+  "version": "4.3.3",
+  "description": "Trigger.dev AI SDK integrations and chat transport",
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/triggerdotdev/trigger.dev",
+    "directory": "packages/ai"
+  },
+  "type": "module",
+  "files": [
+    "dist"
+  ],
+  "tshy": {
+    "selfLink": false,
+    "main": true,
+    "module": true,
+    "project": "./tsconfig.json",
+    "exports": {
+      "./package.json": "./package.json",
+      ".": "./src/index.ts"
+    },
+    "sourceDialects": [
+      "@triggerdotdev/source"
+    ]
+  },
+  "scripts": {
+    "clean": "rimraf dist .tshy .tshy-build .turbo",
+    "build": "tshy && pnpm run update-version",
+    "dev": "tshy --watch",
+    "typecheck": "tsc --noEmit",
+    "test": "vitest",
+    "update-version": "tsx ../../scripts/updateVersion.ts",
+    "check-exports": "attw --pack ."
+  },
+  "dependencies": {
+    "@trigger.dev/core": "workspace:^4.3.3"
+  },
+  "devDependencies": {
+    "@arethetypeswrong/cli": "^0.15.4",
+    "ai": "^6.0.0",
+    "rimraf": "^3.0.2",
+    "tshy": "^3.0.2",
+    "tsx": "4.17.0",
+    "zod": "3.25.76"
+  },
+  "peerDependencies": {
+    "ai": "^4.2.0 || ^5.0.0 || ^6.0.0",
+    "zod": "^3.0.0 || ^4.0.0"
+  },
+  "engines": {
+    "node": ">=18.20.0"
+  },
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "import": {
+        "@triggerdotdev/source": "./src/index.ts",
+        "types": "./dist/esm/index.d.ts",
+        "default": "./dist/esm/index.js"
+      },
+      "require": {
+        "types": "./dist/commonjs/index.d.ts",
+        "default": "./dist/commonjs/index.js"
+      }
+    }
+  },
+  "main": "./dist/commonjs/index.js",
+  "types": "./dist/commonjs/index.d.ts",
+  "module": "./dist/esm/index.js"
+}

packages/ai/src/ai.test.ts

@@ -0,0 +1,79 @@
+import { describe, expect, it } from "vitest";
+import { z } from "zod";
+import { ai } from "./ai.js";
+import type { TaskWithSchema } from "@trigger.dev/core/v3";
+
+describe("ai helper", function () {
+  it("creates a tool from a schema task and executes through triggerAndWait", async function () {
+    let receivedInput: unknown = undefined;
+
+    const fakeTask = {
+      id: "fake-task",
+      description: "A fake task",
+      schema: z.object({
+        name: z.string(),
+      }),
+      triggerAndWait: function (payload: { name: string }) {
+        receivedInput = payload;
+        const resultPromise = Promise.resolve({
+          ok: true,
+          id: "run_123",
+          taskIdentifier: "fake-task",
+          output: {
+            greeting: `Hello ${payload.name}`,
+          },
+        });
+
+        return Object.assign(resultPromise, {
+          unwrap: async function () {
+            return {
+              greeting: `Hello ${payload.name}`,
+            };
+          },
+        });
+      },
+    } as unknown as TaskWithSchema<
+      "fake-task",
+      z.ZodObject<{ name: z.ZodString }>,
+      { greeting: string }
+    >;
+
+    const tool = ai.tool(fakeTask);
+    const result = await tool.execute?.(
+      {
+        name: "Ada",
+      },
+      undefined as never
+    );
+
+    expect(receivedInput).toEqual({
+      name: "Ada",
+    });
+    expect(result).toEqual({
+      greeting: "Hello Ada",
+    });
+  });
+
+  it("throws when creating a tool from a task without schema", function () {
+    const fakeTask = {
+      id: "no-schema",
+      description: "No schema task",
+      schema: undefined,
+      triggerAndWait: async function () {
+        return {
+          unwrap: async function () {
+            return {};
+          },
+        };
+      },
+    } as unknown as TaskWithSchema<"no-schema", undefined, unknown>;
+
+    expect(function () {
+      ai.tool(fakeTask);
+    }).toThrowError("task has no schema");
+  });
+
+  it("returns undefined for current tool options outside task execution context", function () {
+    expect(ai.currentToolOptions()).toBeUndefined();
+  });
+});