@standardagents/builder 0.11.7 → 0.11.9

package/dist/index.js

@@ -2917,8 +2917,22 @@ ${errorLines.join("\n")}`);
           }
           const agentType = agentDef.type;
           if (agentType === "ai_human") {
+            let args2 = {};
+            try {
+              args2 = JSON.parse(call.function.arguments);
+            } catch {
+            }
+            const parentToolConfigs = state.prompt._tools || [];
+            const toolConfig = parentToolConfigs.find(
+              (t) => typeof t === "string" ? t === agentName : t.name === agentName
+            );
             state.pendingHandoff = {
-              agentId: agentName
+              agentId: agentName,
+              args: args2,
+              toolConfig: typeof toolConfig === "object" ? {
+                initUserMessageProperty: toolConfig.initUserMessageProperty,
+                initAttachmentsProperty: toolConfig.initAttachmentsProperty
+              } : void 0
             };
             const text2 = "Handoff complete";
             return {
@@ -3060,10 +3074,11 @@ ${attachmentPaths}`;
           };
           message = await FlowEngine2.runBeforeCreateMessageHook(state, message);
           message.created_at = Date.now() * TIMESTAMP_MULTIPLIER;
+          const currentPrompt = state.promptPath.length > 0 ? state.promptPath[state.promptPath.length - 1] : null;
           await state.storage.sql.exec(
             `
-        INSERT INTO messages (id, role, content, name, tool_call_id, created_at, tool_status, parent_id, depth, attachments)
-        VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10)
+        INSERT INTO messages (id, role, content, name, tool_call_id, created_at, tool_status, parent_id, depth, attachments, prompt)
+        VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11)
       `,
             message.id,
             message.role,
@@ -3074,7 +3089,8 @@ ${attachmentPaths}`;
             message.tool_status,
             message.parent_id,
             message.depth,
-            message.attachments
+            message.attachments,
+            currentPrompt
           );
           await FlowEngine2.runAfterCreateMessageHook(state, message);
           const parentLogId = state.currentLogId || null;
@@ -4722,13 +4738,15 @@ var init_FlowEngine = __esm({
                 status: "completed"
               };
               interruptionMessage = await this.runBeforeCreateMessageHook(state, interruptionMessage);
+              const currentPromptName = state.promptPath.length > 0 ? state.promptPath[state.promptPath.length - 1] : null;
               await state.storage.sql.exec(
-                `INSERT INTO messages (id, role, content, created_at, status) VALUES (?, ?, ?, ?, ?)`,
+                `INSERT INTO messages (id, role, content, created_at, status, prompt) VALUES (?, ?, ?, ?, ?, ?)`,
                 interruptionMessage.id,
                 interruptionMessage.role,
                 interruptionMessage.content,
                 interruptionMessage.created_at,
-                interruptionMessage.status
+                interruptionMessage.status,
+                currentPromptName
               );
               await this.runAfterCreateMessageHook(state, interruptionMessage);
               if (state.emitMessage) {
@@ -5014,8 +5032,8 @@ var init_FlowEngine = __esm({
           );
           return !hasExistingResult;
         });
-        if (toolsToExecute && toolsToExecute.length > 0) {
-          await ToolExecutor.executeSequence(state, toolsToExecute);
+        if (toolsToExecute && toolsToExecute.length > 0 || state.sequence.queue.length > 0) {
+          await ToolExecutor.executeSequence(state, toolsToExecute || []);
         }
         if (await state.thread.instance.shouldStop() || state.abortController?.signal.aborted) {
           throw new Error("aborted");
@@ -5113,10 +5131,72 @@ var init_FlowEngine = __esm({
           state.agentConfig = agentConfig;
           state.prompts.sideA = sideAPrompt;
           state.prompts.sideB = sideBPrompt;
-          state.extraMessages = [];
+          const args = state.pendingHandoff?.args || {};
+          const toolConfig = state.pendingHandoff?.toolConfig;
+          const initUserMessageProperty = toolConfig?.initUserMessageProperty;
+          const initAttachmentsProperty = toolConfig?.initAttachmentsProperty;
+          state.context = { ...state.context, ...args };
+          let userMessageContent;
+          let userMessageAttachments;
+          if (initUserMessageProperty && args[initUserMessageProperty] !== void 0) {
+            const propertyValue = args[initUserMessageProperty];
+            userMessageContent = typeof propertyValue === "string" ? propertyValue : JSON.stringify(propertyValue);
+          }
+          if (initAttachmentsProperty && args[initAttachmentsProperty] !== void 0) {
+            const attachmentValue = args[initAttachmentsProperty];
+            const attachmentPaths = Array.isArray(attachmentValue) ? attachmentValue : [attachmentValue];
+            const resolvedAttachments = [];
+            for (const attachmentPath of attachmentPaths) {
+              if (typeof attachmentPath === "string" && attachmentPath.startsWith("/attachments/")) {
+                try {
+                  const statResult = await state.thread.instance.statFile(attachmentPath);
+                  if (statResult.success && statResult.file) {
+                    resolvedAttachments.push({
+                      id: crypto.randomUUID(),
+                      type: "file",
+                      path: attachmentPath,
+                      name: statResult.file.name,
+                      mimeType: statResult.file.mimeType,
+                      width: statResult.file.width ?? void 0,
+                      height: statResult.file.height ?? void 0
+                    });
+                  }
+                } catch (e) {
+                  console.error(`[FlowEngine] Failed to resolve attachment ${attachmentPath}:`, e);
+                }
+              }
+            }
+            if (resolvedAttachments.length > 0) {
+              userMessageAttachments = JSON.stringify(resolvedAttachments);
+            }
+          }
           state.currentSide = "a";
           state.prompt = state.prompts.sideA;
           state.promptPath = [sideAPrompt.name];
+          if (userMessageContent || userMessageAttachments) {
+            const handoffMessageId = crypto.randomUUID();
+            const handoffTimestamp = Date.now() * 1e3;
+            const handoffMessage = {
+              id: handoffMessageId,
+              role: "user",
+              content: userMessageContent || "",
+              created_at: handoffTimestamp,
+              attachments: userMessageAttachments
+            };
+            await state.storage.sql.exec(
+              `INSERT INTO messages (id, role, content, attachments, created_at, prompt) VALUES (?, ?, ?, ?, ?, ?)`,
+              handoffMessage.id,
+              handoffMessage.role,
+              handoffMessage.content,
+              handoffMessage.attachments || null,
+              handoffMessage.created_at,
+              sideAPrompt.name
+            );
+            state.messageHistory.push(handoffMessage);
+            if (state.emitMessage) {
+              state.emitMessage(handoffMessage);
+            }
+          }
           state.pendingHandoff = void 0;
           state.emitTelemetry?.({
             type: "agent_handoff",
@@ -5459,26 +5539,35 @@ ${msg.content}` : `${imageDescriptions}${nonImageList}`;
               );
               tools.push(...selectedTools);
               const foundToolNames = new Set(selectedTools.map((t) => t.function.name));
+              const { z: z4 } = await import('zod');
               for (const toolName of regularToolNames) {
                 if (foundToolNames.has(toolName)) continue;
                 try {
                   const agentDef = await state.thread.instance.loadAgent(toolName);
                   if (agentDef && agentDef.type === "ai_human") {
+                    let parameters;
+                    const sideAPromptName = agentDef.sideA?.prompt;
+                    if (sideAPromptName) {
+                      try {
+                        const promptDef = await state.thread.instance.loadPrompt(sideAPromptName);
+                        if (promptDef.requiredSchema) {
+                          const jsonSchema = z4.toJSONSchema(promptDef.requiredSchema);
+                          parameters = {
+                            type: "object",
+                            properties: jsonSchema.properties || {},
+                            required: jsonSchema.required || []
+                          };
+                        }
+                      } catch (promptError) {
+                        console.error(`Error loading prompt schema for agent ${toolName}:`, promptError);
+                      }
+                    }
                     tools.push({
                       type: "function",
                       function: {
-                        name: agentDef.title || toolName,
+                        name: toolName,
                         description: agentDef.toolDescription || `Hand off conversation to ${agentDef.title || toolName} agent`,
-                        parameters: {
-                          type: "object",
-                          properties: {
-                            input: {
-                              type: "string",
-                              description: "Input message for the agent"
-                            }
-                          },
-                          required: ["input"]
-                        }
+                        parameters
                       }
                     });
                   }
@@ -5713,25 +5802,34 @@ ${msg.content}` : `${imageDescriptions}${nonImageList}`;
         const tools = [];
         try {
           const agentNames = state.thread.instance.getAgentNames();
+          const { z: z4 } = await import('zod');
           for (const name of agentNames) {
             try {
               const agentDef = await state.thread.instance.loadAgent(name);
               if (agentDef.exposeAsTool) {
+                let parameters;
+                const sideAPromptName = agentDef.sideA?.prompt;
+                if (sideAPromptName) {
+                  try {
+                    const promptDef = await state.thread.instance.loadPrompt(sideAPromptName);
+                    if (promptDef.requiredSchema) {
+                      const jsonSchema = z4.toJSONSchema(promptDef.requiredSchema);
+                      parameters = {
+                        type: "object",
+                        properties: jsonSchema.properties || {},
+                        required: jsonSchema.required || []
+                      };
+                    }
+                  } catch (promptError) {
+                    console.error(`Error loading prompt schema for agent ${name}:`, promptError);
+                  }
+                }
                 tools.push({
                   type: "function",
                   function: {
-                    name: agentDef.title || name,
+                    name,
                     description: agentDef.toolDescription || `Execute the ${agentDef.title || name} agent`,
-                    parameters: {
-                      type: "object",
-                      properties: {
-                        input: {
-                          type: "string",
-                          description: "Input message for the agent"
-                        }
-                      },
-                      required: ["input"]
-                    }
+                    parameters
                   }
                 });
               }
@@ -5985,13 +6083,14 @@ ${msg.content}` : `${imageDescriptions}${nonImageList}`;
         const now = Date.now() * TIMESTAMP_MULTIPLIER;
         message.created_at = now;
         message.request_sent_at = now;
+        const currentPrompt = state.promptPath.length > 0 ? state.promptPath[state.promptPath.length - 1] : null;
         await state.storage.sql.exec(
           `
       INSERT INTO messages (
         id, role, content, tool_calls, log_id, created_at,
-        request_sent_at, response_completed_at, status, parent_id, depth, reasoning_content
+        request_sent_at, response_completed_at, status, parent_id, depth, reasoning_content, prompt
       )
-      VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12)
+      VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12, ?13)
     `,
           message.id,
           message.role,
@@ -6004,8 +6103,9 @@ ${msg.content}` : `${imageDescriptions}${nonImageList}`;
           message.status,
           message.parent_id,
           message.depth,
-          null
+          null,
           // reasoning_content - will be set when message is updated with LLM response
+          currentPrompt
         );
         await this.runAfterCreateMessageHook(state, message);
         state.messageHistory.push(message);
@@ -6263,10 +6363,11 @@ ${msg.content}` : `${imageDescriptions}${nonImageList}`;
               depth: state.depth,
               attachments: JSON.stringify([attachmentRef])
             };
+            const currentPromptForAttachment = state.promptPath.length > 0 ? state.promptPath[state.promptPath.length - 1] : null;
             await state.storage.sql.exec(
               `
-          INSERT INTO messages (id, role, content, name, tool_call_id, created_at, tool_status, parent_id, depth, attachments)
-          VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10)
+          INSERT INTO messages (id, role, content, name, tool_call_id, created_at, tool_status, parent_id, depth, attachments, prompt)
+          VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11)
           `,
               toolMessage.id,
               toolMessage.role,
@@ -6277,7 +6378,8 @@ ${msg.content}` : `${imageDescriptions}${nonImageList}`;
               toolMessage.tool_status,
               toolMessage.parent_id,
               toolMessage.depth,
-              toolMessage.attachments
+              toolMessage.attachments,
+              currentPromptForAttachment
             );
             state.messageHistory.push(toolMessage);
             if (state.emitMessage) {
@@ -6807,8 +6909,9 @@ async function injectMessage(flow, options) {
     }
     message.created_at = targetMessage.created_at - 1;
   }
+  const currentPrompt = flow.promptPath.length > 0 ? flow.promptPath[flow.promptPath.length - 1] : null;
   await flow.storage.sql.exec(
-    `INSERT INTO messages (id, role, content, name, tool_call_id, created_at, status, silent, parent_id, depth) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+    `INSERT INTO messages (id, role, content, name, tool_call_id, created_at, status, silent, parent_id, depth, prompt) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
     messageId,
     message.role,
     message.content,
@@ -6818,7 +6921,8 @@ async function injectMessage(flow, options) {
     message.status,
     silent ? 1 : 0,
     message.parent_id,
-    message.depth
+    message.depth,
+    currentPrompt
   );
   if (!silent) {
     flow.messageHistory.push(message);
@@ -7671,6 +7775,74 @@ var TSCONFIG_CONTENT = `{
   "include": ["types.d.ts", "virtual-module.d.ts"]
 }
 `;
+function extractSchemaFields(content) {
+  const fields = [];
+  const schemaMatch = content.match(/requiredSchema:\s*z\.object\(\s*\{([^}]+)\}/s);
+  if (!schemaMatch) {
+    return fields;
+  }
+  const schemaContent = schemaMatch[1];
+  const fieldMatches = schemaContent.matchAll(/(?:['"]([^'"]+)['"]|(\w+))\s*:/g);
+  for (const match of fieldMatches) {
+    const fieldName = match[1] || match[2];
+    if (fieldName && !fieldName.startsWith("z.")) {
+      fields.push(fieldName);
+    }
+  }
+  return fields;
+}
+function scanPromptsWithSchemas(dir) {
+  const results = [];
+  if (!fs2.existsSync(dir)) {
+    return results;
+  }
+  try {
+    const entries = fs2.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.isFile() && entry.name.endsWith(".ts")) {
+        const filePath = path3.join(dir, entry.name);
+        const content = fs2.readFileSync(filePath, "utf-8");
+        const nameMatch = content.match(/name:\s*['"]([^'"]+)['"]/);
+        if (nameMatch) {
+          const name = nameMatch[1];
+          const schemaFields = extractSchemaFields(content);
+          results.push({ name, schemaFields });
+        }
+      }
+    }
+  } catch (error) {
+    console.error(`Error scanning prompts directory ${dir}:`, error);
+  }
+  return results;
+}
+function scanAgentsWithSideA(dir) {
+  const results = [];
+  if (!fs2.existsSync(dir)) {
+    return results;
+  }
+  try {
+    const entries = fs2.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.isFile() && entry.name.endsWith(".ts")) {
+        const filePath = path3.join(dir, entry.name);
+        const content = fs2.readFileSync(filePath, "utf-8");
+        const nameMatch = content.match(/name:\s*['"]([^'"]+)['"]/);
+        if (nameMatch) {
+          const name = nameMatch[1];
+          let sideAPrompt = null;
+          const sideAMatch = content.match(/sideA:\s*\{[^}]*prompt:\s*['"]([^'"]+)['"]/);
+          if (sideAMatch) {
+            sideAPrompt = sideAMatch[1];
+          }
+          results.push({ name, sideAPrompt });
+        }
+      }
+    }
+  } catch (error) {
+    console.error(`Error scanning agents directory ${dir}:`, error);
+  }
+  return results;
+}
 function scanForNames(dir, useFilename = false) {
   const names = [];
   if (!fs2.existsSync(dir)) {
@@ -7706,11 +7878,38 @@ function generateRegistryProperties(names) {
   }
   return names.map((n) => `'${n}': true;`).join("\n      ");
 }
+function generateSchemaRegistryProperties(promptResults, agentResults) {
+  const entries = [];
+  const promptSchemaMap = /* @__PURE__ */ new Map();
+  for (const prompt of promptResults) {
+    if (prompt.schemaFields.length > 0) {
+      promptSchemaMap.set(prompt.name, prompt.schemaFields);
+    }
+  }
+  for (const prompt of promptResults) {
+    if (prompt.schemaFields.length > 0) {
+      const fieldsUnion = prompt.schemaFields.map((f) => `'${f}'`).join(" | ");
+      entries.push(`'${prompt.name}': ${fieldsUnion};`);
+    }
+  }
+  for (const agent of agentResults) {
+    if (agent.sideAPrompt) {
+      const schemaFields = promptSchemaMap.get(agent.sideAPrompt);
+      if (schemaFields && schemaFields.length > 0) {
+        const fieldsUnion = schemaFields.map((f) => `'${f}'`).join(" | ");
+        entries.push(`'${agent.name}': ${fieldsUnion};`);
+      }
+    }
+  }
+  return entries.join("\n      ");
+}
 function generateTypesContent(config) {
   const models = scanForNames(config.modelsDir);
-  const prompts = scanForNames(config.promptsDir);
-  const agents = scanForNames(config.agentsDir);
+  const promptResults = scanPromptsWithSchemas(config.promptsDir);
+  const agentResults = scanAgentsWithSideA(config.agentsDir);
   const tools = scanForNames(config.toolsDir, true);
+  const prompts = promptResults.map((p) => p.name);
+  const agents = agentResults.map((a) => a.name);
   const callables = [...prompts, ...agents, ...tools];
   return `// Auto-generated by @standardagents/builder - DO NOT EDIT
 // Generated at: ${(/* @__PURE__ */ new Date()).toISOString()}
@@ -7751,6 +7950,11 @@ declare global {
     interface CallableRegistry {
       ${generateRegistryProperties(callables)}
     }
+
+    /** Schema field names for prompts and agents (used for initUserMessageProperty autocomplete) */
+    interface PromptSchemaRegistry {
+      ${generateSchemaRegistryProperties(promptResults, agentResults)}
+    }
   }
 }
 
@@ -12074,8 +12278,24 @@ var migration21 = {
   }
 };
 
+// src/durable-objects/migrations/022_add_prompt_to_messages.ts
+var migration22 = {
+  version: 22,
+  async up(sql) {
+    await sql.exec(`
+      ALTER TABLE messages ADD COLUMN prompt TEXT
+    `);
+    await sql.exec(`
+      CREATE INDEX IF NOT EXISTS idx_messages_prompt ON messages(prompt)
+    `);
+    await sql.exec(`
+      UPDATE _metadata SET value = '22' WHERE key = 'schema_version'
+    `);
+  }
+};
+
 // src/durable-objects/migrations/index.ts
-var migrations = [migration, migration2, migration3, migration4, migration5, migration6, migration7, migration8, migration9, migration10, migration11, migration12, migration13, migration14, migration15, migration16, migration17, migration18, migration19, migration20, migration21];
+var migrations = [migration, migration2, migration3, migration4, migration5, migration6, migration7, migration8, migration9, migration10, migration11, migration12, migration13, migration14, migration15, migration16, migration17, migration18, migration19, migration20, migration21, migration22];
 var LATEST_SCHEMA_VERSION = migrations.length;
 
 // src/durable-objects/DurableThread.ts
@@ -12659,9 +12879,9 @@ var DurableThread = class extends DurableObject {
    * Each migration is run in order, starting from the current version + 1.
    */
   async runMigrations(fromVersion) {
-    for (const migration24 of migrations) {
-      if (migration24.version > fromVersion) {
-        await migration24.up(this.ctx.storage.sql);
+    for (const migration25 of migrations) {
+      if (migration25.version > fromVersion) {
+        await migration25.up(this.ctx.storage.sql);
       }
     }
   }
@@ -12818,14 +13038,15 @@ var DurableThread = class extends DurableObject {
               if (responses.length === 0) {
                 const toolResponseId = crypto.randomUUID();
                 await this.ctx.storage.sql.exec(
-                  `INSERT INTO messages (id, role, content, tool_call_id, tool_status, created_at, status) VALUES (?, ?, ?, ?, ?, ?, ?)`,
+                  `INSERT INTO messages (id, role, content, tool_call_id, tool_status, created_at, status, prompt) VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
                   toolResponseId,
                   "tool",
                   "Execution stopped by user",
                   tc.id,
                   "error",
                   now,
-                  "completed"
+                  "completed",
+                  null
                 );
                 this.broadcastMessage({
                   id: toolResponseId,
@@ -12846,12 +13067,13 @@ var DurableThread = class extends DurableObject {
       const interruptionMessageId = crypto.randomUUID();
       const timestamp = Date.now() * TIMESTAMP_MULTIPLIER;
       await this.ctx.storage.sql.exec(
-        `INSERT INTO messages (id, role, content, created_at, status) VALUES (?, ?, ?, ?, ?)`,
+        `INSERT INTO messages (id, role, content, created_at, status, prompt) VALUES (?, ?, ?, ?, ?, ?)`,
         interruptionMessageId,
         "system",
         "Execution stopped by user",
         timestamp,
-        "completed"
+        "completed",
+        null
       );
       this.broadcastMessage({
         id: interruptionMessageId,
@@ -13090,14 +13312,15 @@ var DurableThread = class extends DurableObject {
       for (const msg of args.messages) {
         await this.ctx.storage.sql.exec(
           `
-          INSERT INTO messages (id, role, content, created_at, log_id)
-          VALUES (?, ?, ?, ?, ?)
+          INSERT INTO messages (id, role, content, created_at, log_id, prompt)
+          VALUES (?, ?, ?, ?, ?, ?)
         `,
           msg.id,
           msg.role,
           msg.content,
           msg.created_at,
-          msg.log_id ?? null
+          msg.log_id ?? null,
+          null
         );
       }
       return {
@@ -13675,14 +13898,15 @@ var DurableThread = class extends DurableObject {
         }));
         for (const msg of messagesToStore) {
           await this.ctx.storage.sql.exec(
-            `INSERT INTO messages (id, role, content, name, tool_calls, tool_call_id, created_at) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7)`,
+            `INSERT INTO messages (id, role, content, name, tool_calls, tool_call_id, created_at, prompt) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8)`,
             msg.id,
             msg.role,
             msg.content || null,
             msg.name || null,
             msg.tool_calls ? JSON.stringify(msg.tool_calls) : null,
             msg.tool_call_id || null,
-            msg.created_at
+            msg.created_at,
+            null
           );
           if (msg.role === "user") {
             rootMessageId = msg.id;
@@ -13692,11 +13916,12 @@ var DurableThread = class extends DurableObject {
       if (!rootMessageId) {
         rootMessageId = crypto.randomUUID();
         await this.ctx.storage.sql.exec(
-          `INSERT INTO messages (id, role, content, created_at) VALUES (?1, ?2, ?3, ?4)`,
+          `INSERT INTO messages (id, role, content, created_at, prompt) VALUES (?1, ?2, ?3, ?4, ?5)`,
           rootMessageId,
           "system",
           null,
-          Date.now() * 1e3
+          Date.now() * 1e3,
+          null
         );
       }
       await FlowEngine2.execute({
@@ -13819,13 +14044,15 @@ var DurableThread = class extends DurableObject {
       };
       const { FlowEngine: FlowEngine2 } = await Promise.resolve().then(() => (init_FlowEngine(), FlowEngine_exports));
       message = await FlowEngine2.runBeforeCreateMessageHook(state, message);
+      const currentPrompt = agentDef.sideA?.prompt || null;
       await this.ctx.storage.sql.exec(
-        `INSERT INTO messages (id, role, content, attachments, created_at) VALUES (?, ?, ?, ?, ?)`,
+        `INSERT INTO messages (id, role, content, attachments, created_at, prompt) VALUES (?, ?, ?, ?, ?, ?)`,
         message.id,
         message.role,
         message.content,
         message.attachments,
-        message.created_at
+        message.created_at,
+        currentPrompt
       );
       await FlowEngine2.runAfterCreateMessageHook(state, message);
       this.broadcastMessage({
@@ -14036,12 +14263,13 @@ var DurableThread = class extends DurableObject {
         }
       ]);
       await this.ctx.storage.sql.exec(
-        `INSERT INTO messages (id, role, content, tool_calls, created_at)
-         VALUES (?, 'assistant', ?, ?, ?)`,
+        `INSERT INTO messages (id, role, content, tool_calls, created_at, prompt)
+         VALUES (?, 'assistant', ?, ?, ?, ?)`,
         messageId,
         params.content || null,
         toolCalls,
-        Date.now() * 1e3
+        Date.now() * 1e3,
+        null
       );
       return Response.json({
         status: "inserted",
@@ -14553,7 +14781,7 @@ var DurableThread = class extends DurableObject {
 };
 
 // src/durable-objects/agentbuilder-migrations/0001_initial.ts
-var migration22 = {
+var migration23 = {
   version: 1,
   async up(sql) {
     sql.exec(`
@@ -14652,7 +14880,7 @@ var migration22 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0002_add_tenvs_properties.ts
-var migration23 = {
+var migration24 = {
   version: 2,
   async up(sql) {
     sql.exec(`ALTER TABLE threads ADD COLUMN tenvs TEXT`);
@@ -14664,7 +14892,7 @@ var migration23 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/index.ts
-var migrations2 = [migration22, migration23];
+var migrations2 = [migration23, migration24];
 var LATEST_SCHEMA_VERSION2 = 2;
 
 // src/utils/crypto.ts
@@ -14841,9 +15069,9 @@ var DurableAgentBuilder = class extends DurableObject {
     }
   }
   async runMigrations(fromVersion) {
-    for (const migration24 of migrations2) {
-      if (migration24.version > fromVersion) {
-        await migration24.up(this.ctx.storage.sql);
+    for (const migration25 of migrations2) {
+      if (migration25.version > fromVersion) {
+        await migration25.up(this.ctx.storage.sql);
       }
     }
   }
@@ -15611,7 +15839,7 @@ var FlowStateSdk = class _FlowStateSdk {
    * The tool will be executed as part of the normal flow processing.
    *
    * @param flow - The current FlowState
-   * @param toolName - The name of the tool to call
+   * @param toolName - The name of the tool to call (autocompletes from available tools)
    * @param args - The arguments to pass to the tool
    *
    * @example
@@ -15697,8 +15925,9 @@ var FlowStateSdk = class _FlowStateSdk {
       }
       message.created_at = targetMessage.created_at - 1;
     }
+    const currentPrompt = flow.promptPath.length > 0 ? flow.promptPath[flow.promptPath.length - 1] : null;
     await flow.storage.sql.exec(
-      `INSERT INTO messages (id, role, content, name, tool_call_id, created_at, status, silent, parent_id, depth) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+      `INSERT INTO messages (id, role, content, name, tool_call_id, created_at, status, silent, parent_id, depth, prompt) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
       messageId,
       message.role,
       message.content,
@@ -15708,7 +15937,8 @@ var FlowStateSdk = class _FlowStateSdk {
       message.status,
       silent ? 1 : 0,
       message.parent_id,
-      message.depth
+      message.depth,
+      currentPrompt
     );
     if (!silent) {
       flow.messageHistory.push(message);