Add the chat client and strip agent crumbs

Author: ericallam

apps/webapp/app/routes/_app.orgs.$organizationSlug.projects.$projectParam.env.$envParam.playground.$agentParam/route.tsx

@@ -157,12 +157,12 @@ export const loader = async ({ request, params }: LoaderFunctionArgs) => {
 };
 
 export default function PlaygroundAgentPage() {
-  const { activeConversation } = useTypedLoaderData<typeof loader>();
-  // Key on conversation chatId so React remounts all stateful children when
-  // navigating between conversations (Link changes search params, loader re-runs,
-  // but without a key change the component instance is reused and useState
-  // initializers / useRef initializations don't re-run).
-  const conversationKey = activeConversation?.chatId ?? "new";
+  const { agent, activeConversation } = useTypedLoaderData<typeof loader>();
+  // Key on agent slug + conversation chatId so React remounts all stateful
+  // children when switching agents or navigating between conversations.
+  // Without the agent slug, switching agents keeps key="new" and React
+  // reuses the component — useState initializers don't re-run.
+  const conversationKey = `${agent.slug}:${activeConversation?.chatId ?? "new"}`;
   return <PlaygroundChat key={conversationKey} />;
 }
 
@@ -974,6 +974,7 @@ function PlaygroundSidebar({
               getCurrentPayload={getCurrentClientData}
               generateButtonLabel="Generate client data"
               placeholder="e.g. generate client data for a free-tier user"
+              isAgent={true}
               examplePromptsOverride={[
                 "Generate valid client data",
                 "Generate client data with all fields",

apps/webapp/app/routes/_app.orgs.$organizationSlug.projects.$projectParam.env.$envParam.test.tasks.$taskParam/AIPayloadTabContent.tsx

@@ -25,6 +25,7 @@ export function AIPayloadTabContent({
   generateButtonLabel = "Generate payload",
   placeholder,
   examplePromptsOverride,
+  isAgent = false,
 }: {
   onPayloadGenerated: (payload: string) => void;
   payloadSchema?: unknown;
@@ -33,6 +34,7 @@ export function AIPayloadTabContent({
   generateButtonLabel?: string;
   placeholder?: string;
   examplePromptsOverride?: string[];
+  isAgent?: boolean;
 }) {
   const [prompt, setPrompt] = useState("");
   const [isLoading, setIsLoading] = useState(false);
@@ -70,6 +72,7 @@ export function AIPayloadTabContent({
         const formData = new FormData();
         formData.append("prompt", queryPrompt);
         formData.append("taskIdentifier", taskIdentifier);
+        formData.append("isAgent", isAgent ? "true" : "false");
         if (payloadSchema) {
           formData.append("payloadSchema", JSON.stringify(payloadSchema));
         }
@@ -141,7 +144,7 @@ export function AIPayloadTabContent({
         setIsLoading(false);
       }
     },
-    [resourcePath, taskIdentifier, payloadSchema, getCurrentPayload]
+    [resourcePath, taskIdentifier, payloadSchema, getCurrentPayload, isAgent]
   );
 
   const processStreamEvent = useCallback(

apps/webapp/app/routes/resources.orgs.$organizationSlug.projects.$projectParam.env.$envParam.test.ai-generate-payload.tsx

@@ -20,6 +20,7 @@ const RequestSchema = z.object({
   taskIdentifier: z.string().max(256),
   payloadSchema: z.string().max(50_000).optional(),
   currentPayload: z.string().max(50_000).optional(),
+  isAgent: z.enum(["true", "false"]).optional(),
 });
 
 export async function action({ request, params }: ActionFunctionArgs) {
@@ -63,16 +64,20 @@ export async function action({ request, params }: ActionFunctionArgs) {
     );
   }
 
-  const { prompt, taskIdentifier, payloadSchema, currentPayload } = submission.data;
+  const { prompt, taskIdentifier, payloadSchema, currentPayload, isAgent } = submission.data;
+  const agentMode = isAgent === "true";
 
   logger.info("[AI payload] Generating payload", {
     taskIdentifier,
     hasPayloadSchema: !!payloadSchema,
     hasCurrentPayload: !!currentPayload,
     promptLength: prompt.length,
+    agentMode,
   });
 
-  const systemPrompt = buildSystemPrompt(taskIdentifier, payloadSchema, currentPayload);
+  const systemPrompt = agentMode
+    ? buildAgentClientDataPrompt(taskIdentifier, payloadSchema, currentPayload)
+    : buildSystemPrompt(taskIdentifier, payloadSchema, currentPayload);
 
   const stream = new ReadableStream({
     async start(controller) {
@@ -233,6 +238,60 @@ async function getTaskFromDeployment(environmentId: string, taskIdentifier: stri
   return { fileId: task.fileId };
 }
 
+function buildAgentClientDataPrompt(
+  taskIdentifier: string,
+  payloadSchema?: string,
+  currentPayload?: string
+): string {
+  let prompt = `You are a JSON generator for client data (metadata) of a Trigger.dev chat agent with id "${taskIdentifier}".
+
+IMPORTANT: You are generating ONLY the client data object — this is the metadata sent alongside each chat message. It is NOT the full task payload. Do NOT generate fields like "chatId", "messages", "trigger", or "idleTimeoutInSeconds" — those are internal transport fields managed by the framework.
+
+The client data typically contains user context like user IDs, preferences, configuration, or session info. Return ONLY valid JSON wrapped in a \`\`\`json code block.
+
+Requirements:
+- Generate realistic, meaningful example data
+- All string values should be plausible (real-looking IDs, names, etc.)
+- The JSON must be valid and parseable
+- Keep it simple — client data is usually a flat or shallow object`;
+
+  if (payloadSchema) {
+    prompt += `
+
+The agent has the following JSON Schema for its client data:
+\`\`\`json
+${payloadSchema}
+\`\`\`
+
+Generate client data that strictly conforms to this schema.`;
+  } else {
+    prompt += `
+
+No JSON Schema is available for this agent's client data. Use the getTaskSourceCode tool to look up the agent's source code file.
+
+IMPORTANT instructions for reading the source code:
+- The file may contain multiple task/agent definitions. Find the one with id "${taskIdentifier}".
+- Look for \`withClientData({ schema: ... })\` or \`clientDataSchema\` to find the expected client data shape.
+- If using \`chat.agent()\` or \`chat.customAgent()\`, the client data is accessed via \`clientData\` in hooks and \`payload.metadata\` in raw tasks.
+- Look for how \`clientData\` or \`payload.metadata\` is accessed/destructured to infer the shape.
+- Do NOT generate the full ChatTaskWirePayload (messages, chatId, trigger, etc.) — ONLY the metadata/clientData portion.
+- If no client data schema or usage is found, generate a simple \`{ "userId": "user_..." }\` object.`;
+  }
+
+  if (currentPayload) {
+    prompt += `
+
+The current client data in the editor is:
+\`\`\`json
+${currentPayload}
+\`\`\`
+
+Use this as context but generate new client data based on the user's prompt.`;
+  }
+
+  return prompt;
+}
+
 function buildSystemPrompt(
   taskIdentifier: string,
   payloadSchema?: string,

apps/webapp/app/v3/services/createBackgroundWorker.server.ts

@@ -270,19 +270,13 @@ async function createWorkerTask(
       );
     }
 
-    // @crumbs
-    console.log(`[crumbs:webapp] createWorkerTask task=${task.id} triggerSource=${task.triggerSource} agentConfig=${JSON.stringify(task.agentConfig)} taskKeys=${Object.keys(task).join(",")}`); // @crumbs
-
     const resolvedTriggerSource =
       task.triggerSource === "schedule"
         ? ("SCHEDULED" as const)
         : task.triggerSource === "agent"
           ? ("AGENT" as const)
           : ("STANDARD" as const);
 
-    // @crumbs
-    console.log(`[crumbs:webapp] createWorkerTask resolved triggerSource=${resolvedTriggerSource} for task=${task.id}`); // @crumbs
-
     await prisma.backgroundWorkerTask.create({
       data: {
         friendlyId: generateFriendlyId("task"),

packages/cli-v3/src/dev/devSupervisor.ts

@@ -1,5 +1,3 @@
-import { trail } from "agentcrumbs"; // @crumbs
-const _cliCrumb = trail("cli"); // @crumbs
 import { spawn, type ChildProcess } from "node:child_process";
 import { readFileSync, writeFileSync, renameSync, unlinkSync, existsSync, mkdirSync } from "node:fs";
 import { join } from "node:path";
@@ -347,23 +345,6 @@ class DevSupervisor implements WorkerRuntime {
 
     const sourceFiles = resolveSourceFiles(manifest.sources, backgroundWorker.manifest.tasks);
 
-    // #region @crumbs
-    const _agentTasksSupervisor = (backgroundWorker.manifest.tasks as any[]).filter(
-      (t: any) => t.triggerSource || t.agentConfig
-    );
-    _cliCrumb("devSupervisor sending worker metadata to API", {
-      totalTasks: backgroundWorker.manifest.tasks.length,
-      agentTasks: _agentTasksSupervisor.map((t: any) => ({
-        id: t.id,
-        triggerSource: t.triggerSource,
-        agentConfig: t.agentConfig,
-      })),
-      manifestTaskKeys: backgroundWorker.manifest.tasks[0]
-        ? Object.keys(backgroundWorker.manifest.tasks[0])
-        : [],
-    });
-    // #endregion @crumbs
-
     const backgroundWorkerBody: CreateBackgroundWorkerRequestBody = {
       localOnly: true,
       metadata: {

packages/cli-v3/src/entryPoints/dev-index-worker.ts

@@ -1,5 +1,3 @@
-import { trail } from "agentcrumbs"; // @crumbs
-const _cliCrumb = trail("cli"); // @crumbs
 import {
   BuildManifest,
   type HandleErrorFunction,
@@ -106,18 +104,6 @@ const { buildManifest, importErrors, config, timings } = await bootstrap();
 
 let tasks = await convertSchemasToJsonSchemas(resourceCatalog.listTaskManifests());
 
-// #region @crumbs
-const _agentTasks = tasks.filter((t: any) => t.triggerSource || t.agentConfig);
-_cliCrumb("dev-index-worker tasks after listTaskManifests", {
-  totalTasks: tasks.length,
-  agentTasks: _agentTasks.map((t: any) => ({
-    id: t.id,
-    triggerSource: t.triggerSource,
-    agentConfig: t.agentConfig,
-  })),
-});
-// #endregion @crumbs
-
 // If the config has retry defaults, we need to apply them to all tasks that don't have any retry settings
 if (config.retries?.default) {
   tasks = tasks.map((task) => {

packages/core/package.json

@@ -41,6 +41,7 @@
       "./v3/utils/omit": "./src/v3/utils/omit.ts",
       "./v3/utils/retries": "./src/v3/utils/retries.ts",
       "./v3/utils/structuredLogger": "./src/v3/utils/structuredLogger.ts",
+      "./v3/chat-client": "./src/v3/chat-client.ts",
       "./v3/zodfetch": "./src/v3/zodfetch.ts",
       "./v3/zodMessageHandler": "./src/v3/zodMessageHandler.ts",
       "./v3/zodNamespace": "./src/v3/zodNamespace.ts",
@@ -446,6 +447,17 @@
         "default": "./dist/commonjs/v3/utils/structuredLogger.js"
       }
     },
+    "./v3/chat-client": {
+      "import": {
+        "@triggerdotdev/source": "./src/v3/chat-client.ts",
+        "types": "./dist/esm/v3/chat-client.d.ts",
+        "default": "./dist/esm/v3/chat-client.js"
+      },
+      "require": {
+        "types": "./dist/commonjs/v3/chat-client.d.ts",
+        "default": "./dist/commonjs/v3/chat-client.js"
+      }
+    },
     "./v3/zodfetch": {
       "import": {
         "@triggerdotdev/source": "./src/v3/zodfetch.ts",

packages/core/src/v3/chat-client.ts

@@ -0,0 +1,13 @@
+/**
+ * Chat constants shared between backend (ai.ts) and frontend (chat.ts).
+ * The ChatClient class lives in @trigger.dev/sdk/chat.
+ */
+
+/** The output stream key where UIMessageChunks are written. */
+export const CHAT_STREAM_KEY = "chat";
+
+/** Input stream ID for sending chat messages to the running task. */
+export const CHAT_MESSAGES_STREAM_ID = "chat-messages";
+
+/** Input stream ID for sending stop signals to abort the current generation. */
+export const CHAT_STOP_STREAM_ID = "chat-stop";

packages/core/src/v3/resource-catalog/standardResourceCatalog.ts

@@ -1,5 +1,3 @@
-import { trail } from "agentcrumbs"; // @crumbs
-const _coreCrumb = trail("core"); // @crumbs
 import {
   PromptManifest,
   PromptMetadata,
@@ -75,15 +73,6 @@ export class StandardResourceCatalog implements ResourceCatalog {
       return;
     }
 
-    // #region @crumbs
-    _coreCrumb("registerTaskMetadata", {
-      taskId: task.id,
-      triggerSource: metadata.triggerSource,
-      agentConfig: metadata.agentConfig,
-      metadataKeys: Object.keys(metadata),
-    });
-    // #endregion @crumbs
-
     this._taskFileMetadata.set(task.id, {
       ...this._currentFileContext,
     });
@@ -140,18 +129,6 @@ export class StandardResourceCatalog implements ResourceCatalog {
         ...fileMetadata,
       };
 
-      // #region @crumbs
-      if (metadata.triggerSource || metadata.agentConfig) {
-        _coreCrumb("listTaskManifests building manifest", {
-          taskId: id,
-          triggerSource: metadata.triggerSource,
-          agentConfig: metadata.agentConfig,
-          manifestTriggerSource: taskManifest.triggerSource,
-          manifestAgentConfig: (taskManifest as any).agentConfig,
-        });
-      }
-      // #endregion @crumbs
-
       result.push(taskManifest);
     }