@standardagents/builder 0.9.9 → 0.9.10

package/dist/index.d.ts

@@ -904,6 +904,16 @@ declare class DurableThread<Env extends ThreadEnv = ThreadEnv> extends DurableOb
         success: boolean;
         error: any;
     }>;
+    /**
+     * Update a message's content (RPC method)
+     */
+    updateMessageContent(messageId: string, content: string): Promise<{
+        success: boolean;
+        error?: undefined;
+    } | {
+        success: boolean;
+        error: any;
+    }>;
     /**
      * Seed messages directly into the database (RPC method - for testing)
      * This bypasses the normal message processing flow
@@ -914,6 +924,7 @@ declare class DurableThread<Env extends ThreadEnv = ThreadEnv> extends DurableOb
             role: string;
             content: string;
             created_at: number;
+            log_id?: string | null;
         }>;
     }): Promise<{
         success: boolean;
@@ -924,6 +935,27 @@ declare class DurableThread<Env extends ThreadEnv = ThreadEnv> extends DurableOb
         error: any;
         count?: undefined;
     }>;
+    /**
+     * Seed a log directly into the database (RPC method - for testing)
+     * This bypasses the normal log creation flow
+     */
+    seedLog(args: {
+        id: string;
+        message_id: string;
+        provider: string;
+        model: string;
+        created_at: number;
+        is_complete?: boolean;
+        prompt_name?: string;
+    }): Promise<{
+        success: boolean;
+        id: string;
+        error?: undefined;
+    } | {
+        success: boolean;
+        error: any;
+        id?: undefined;
+    }>;
     /**
      * Get logs (RPC method)
      */
@@ -1254,6 +1286,11 @@ declare class DurableAgentBuilder<Env extends AgentBuilderEnv = AgentBuilderEnv>
      * Delete a thread from the registry.
      */
     deleteThread(id: string): Promise<boolean>;
+    /**
+     * Delete all threads from the registry.
+     * Useful for clearing test data between test runs.
+     */
+    deleteAllThreads(): Promise<number>;
     /**
      * Update a thread's agent_name (used during agent handoff).
      */

package/dist/index.js

@@ -8374,6 +8374,34 @@ var DurableThread = class extends DurableObject {
       };
     }
   }
+  /**
+   * Update a message's content (RPC method)
+   */
+  async updateMessageContent(messageId, content) {
+    await this.ensureMigrated();
+    try {
+      const checkResult = await this.ctx.storage.sql.exec(
+        `SELECT id FROM messages WHERE id = ?`,
+        messageId
+      );
+      const messageExists = checkResult.toArray().length > 0;
+      if (!messageExists) {
+        return { success: false, error: "Message not found" };
+      }
+      await this.ctx.storage.sql.exec(
+        `UPDATE messages SET content = ? WHERE id = ?`,
+        content,
+        messageId
+      );
+      return { success: true };
+    } catch (error) {
+      console.error("Error in updateMessageContent:", error);
+      return {
+        success: false,
+        error: error.message || "Failed to update message"
+      };
+    }
+  }
   /**
    * Seed messages directly into the database (RPC method - for testing)
    * This bypasses the normal message processing flow
@@ -8384,13 +8412,14 @@ 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)
-          VALUES (?, ?, ?, ?)
+          INSERT INTO messages (id, role, content, created_at, log_id)
+          VALUES (?, ?, ?, ?, ?)
         `,
           msg.id,
           msg.role,
           msg.content,
-          msg.created_at
+          msg.created_at,
+          msg.log_id ?? null
         );
       }
       return {
@@ -8405,6 +8434,38 @@ var DurableThread = class extends DurableObject {
       };
     }
   }
+  /**
+   * Seed a log directly into the database (RPC method - for testing)
+   * This bypasses the normal log creation flow
+   */
+  async seedLog(args) {
+    await this.ensureMigrated();
+    try {
+      await this.ctx.storage.sql.exec(
+        `
+        INSERT INTO logs (id, message_id, provider, model, is_complete, prompt_name, created_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?)
+      `,
+        args.id,
+        args.message_id,
+        args.provider,
+        args.model,
+        args.is_complete ? 1 : 0,
+        args.prompt_name ?? null,
+        args.created_at
+      );
+      return {
+        success: true,
+        id: args.id
+      };
+    } catch (error) {
+      console.error("Error in seedLog:", error);
+      return {
+        success: false,
+        error: error.message || "Failed to seed log"
+      };
+    }
+  }
   /**
    * Get logs (RPC method)
    */
@@ -9518,6 +9579,19 @@ var DurableAgentBuilder = class extends DurableObject {
     this.broadcastEvent({ type: "thread_deleted", threadId: id });
     return true;
   }
+  /**
+   * Delete all threads from the registry.
+   * Useful for clearing test data between test runs.
+   */
+  async deleteAllThreads() {
+    await this.ensureMigrated();
+    const countCursor = await this.ctx.storage.sql.exec(
+      `SELECT COUNT(*) as count FROM threads`
+    );
+    const count = countCursor.one().count;
+    await this.ctx.storage.sql.exec(`DELETE FROM threads`);
+    return count;
+  }
   /**
    * Update a thread's agent_name (used during agent handoff).
    */