@voltagent/libsql 1.0.9 → 1.0.11

package/dist/index.mjs

@@ -79,6 +79,7 @@ var LibSQLMemoryAdapter = class {
     const messagesTable = `${this.tablePrefix}_messages`;
     const usersTable = `${this.tablePrefix}_users`;
     const workflowStatesTable = `${this.tablePrefix}_workflow_states`;
+    const stepsTable = `${this.tablePrefix}_steps`;
     const isMemoryDb = this.url === ":memory:" || this.url.includes("mode=memory");
     if (!isMemoryDb && (this.url.startsWith("file:") || this.url.startsWith("libsql:"))) {
       try {
@@ -139,11 +140,34 @@ var LibSQLMemoryAdapter = class {
         workflow_name TEXT NOT NULL,
         status TEXT NOT NULL,
         suspension TEXT,
+        events TEXT,
+        output TEXT,
+        cancellation TEXT,
         user_id TEXT,
         conversation_id TEXT,
         metadata TEXT,
         created_at TEXT NOT NULL,
         updated_at TEXT NOT NULL
+      )`,
+        // Create conversation steps table
+        `CREATE TABLE IF NOT EXISTS ${stepsTable} (
+        id TEXT PRIMARY KEY,
+        conversation_id TEXT NOT NULL,
+        user_id TEXT NOT NULL,
+        agent_id TEXT NOT NULL,
+        agent_name TEXT,
+        operation_id TEXT,
+        step_index INTEGER NOT NULL,
+        type TEXT NOT NULL,
+        role TEXT NOT NULL,
+        content TEXT,
+        arguments TEXT,
+        result TEXT,
+        usage TEXT,
+        sub_agent_id TEXT,
+        sub_agent_name TEXT,
+        created_at TEXT NOT NULL,
+        FOREIGN KEY (conversation_id) REFERENCES ${conversationsTable}(id) ON DELETE CASCADE
       )`,
         // Create indexes for better performance
         `CREATE INDEX IF NOT EXISTS idx_${conversationsTable}_user_id ON ${conversationsTable}(user_id)`,
@@ -151,11 +175,14 @@ var LibSQLMemoryAdapter = class {
         `CREATE INDEX IF NOT EXISTS idx_${messagesTable}_conversation_id ON ${messagesTable}(conversation_id)`,
         `CREATE INDEX IF NOT EXISTS idx_${messagesTable}_created_at ON ${messagesTable}(created_at)`,
         `CREATE INDEX IF NOT EXISTS idx_${workflowStatesTable}_workflow_id ON ${workflowStatesTable}(workflow_id)`,
-        `CREATE INDEX IF NOT EXISTS idx_${workflowStatesTable}_status ON ${workflowStatesTable}(status)`
+        `CREATE INDEX IF NOT EXISTS idx_${workflowStatesTable}_status ON ${workflowStatesTable}(status)`,
+        `CREATE INDEX IF NOT EXISTS idx_${stepsTable}_conversation ON ${stepsTable}(conversation_id, step_index)`,
+        `CREATE INDEX IF NOT EXISTS idx_${stepsTable}_operation ON ${stepsTable}(conversation_id, operation_id)`
       ]);
     }, "initialize database schema");
     await this.addV2ColumnsToMessagesTable();
     await this.migrateDefaultUserIds();
+    await this.addWorkflowStateColumns();
     this.initialized = true;
     this.logger.debug("Database schema initialized");
   }
@@ -287,6 +314,42 @@ var LibSQLMemoryAdapter = class {
       this.logger.error("Failed to migrate default user_ids", error);
     }
   }
+  /**
+   * Add new columns to workflow_states table for event persistence
+   * This migration adds support for events, output, and cancellation tracking
+   */
+  async addWorkflowStateColumns() {
+    const workflowStatesTable = `${this.tablePrefix}_workflow_states`;
+    try {
+      const tableInfo = await this.client.execute(`PRAGMA table_info(${workflowStatesTable})`);
+      const columns = tableInfo.rows.map((row) => row.name);
+      if (!columns.includes("events")) {
+        try {
+          await this.client.execute(`ALTER TABLE ${workflowStatesTable} ADD COLUMN events TEXT`);
+          this.logger.debug("Added 'events' column to workflow_states table");
+        } catch (_e) {
+        }
+      }
+      if (!columns.includes("output")) {
+        try {
+          await this.client.execute(`ALTER TABLE ${workflowStatesTable} ADD COLUMN output TEXT`);
+          this.logger.debug("Added 'output' column to workflow_states table");
+        } catch (_e) {
+        }
+      }
+      if (!columns.includes("cancellation")) {
+        try {
+          await this.client.execute(
+            `ALTER TABLE ${workflowStatesTable} ADD COLUMN cancellation TEXT`
+          );
+          this.logger.debug("Added 'cancellation' column to workflow_states table");
+        } catch (_e) {
+        }
+      }
+    } catch (error) {
+      this.logger.warn("Failed to add workflow state columns (non-critical)", error);
+    }
+  }
   // ============================================================================
   // Message Operations
   // ============================================================================
@@ -349,6 +412,72 @@ var LibSQLMemoryAdapter = class {
       );
     }, "add batch messages");
   }
+  async saveConversationSteps(steps) {
+    if (steps.length === 0) return;
+    await this.initialize();
+    const stepsTable = `${this.tablePrefix}_steps`;
+    await this.executeWithRetry(async () => {
+      await this.client.batch(
+        steps.map((step) => {
+          const createdAt = step.createdAt ?? (/* @__PURE__ */ new Date()).toISOString();
+          return {
+            sql: `INSERT INTO ${stepsTable} (
+              id,
+              conversation_id,
+              user_id,
+              agent_id,
+              agent_name,
+              operation_id,
+              step_index,
+              type,
+              role,
+              content,
+              arguments,
+              result,
+              usage,
+              sub_agent_id,
+              sub_agent_name,
+              created_at
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            ON CONFLICT(id) DO UPDATE SET
+              conversation_id = excluded.conversation_id,
+              user_id = excluded.user_id,
+              agent_id = excluded.agent_id,
+              agent_name = excluded.agent_name,
+              operation_id = excluded.operation_id,
+              step_index = excluded.step_index,
+              type = excluded.type,
+              role = excluded.role,
+              content = excluded.content,
+              arguments = excluded.arguments,
+              result = excluded.result,
+              usage = excluded.usage,
+              sub_agent_id = excluded.sub_agent_id,
+              sub_agent_name = excluded.sub_agent_name,
+              created_at = excluded.created_at`,
+            args: [
+              step.id,
+              step.conversationId,
+              step.userId,
+              step.agentId,
+              step.agentName ?? null,
+              step.operationId ?? null,
+              step.stepIndex,
+              step.type,
+              step.role,
+              step.content ?? null,
+              step.arguments ? safeStringify(step.arguments) : null,
+              step.result ? safeStringify(step.result) : null,
+              step.usage ? safeStringify(step.usage) : null,
+              step.subAgentId ?? null,
+              step.subAgentName ?? null,
+              createdAt
+            ]
+          };
+        })
+      );
+    }, "save conversation steps");
+  }
   /**
    * Get messages with optional filtering
    */
@@ -410,6 +539,51 @@ var LibSQLMemoryAdapter = class {
       };
     });
   }
+  async getConversationSteps(userId, conversationId, options) {
+    await this.initialize();
+    const stepsTable = `${this.tablePrefix}_steps`;
+    const limit = options?.limit && options.limit > 0 ? options.limit : void 0;
+    let sql = `SELECT * FROM ${stepsTable} WHERE conversation_id = ? AND user_id = ?`;
+    const args = [conversationId, userId];
+    if (options?.operationId) {
+      sql += " AND operation_id = ?";
+      args.push(options.operationId);
+    }
+    sql += " ORDER BY step_index ASC";
+    if (limit !== void 0) {
+      sql += " LIMIT ?";
+      args.push(limit);
+    }
+    const result = await this.client.execute({ sql, args });
+    const parseJsonField = /* @__PURE__ */ __name((value) => {
+      if (typeof value !== "string" || value.length === 0) {
+        return void 0;
+      }
+      try {
+        return JSON.parse(value);
+      } catch {
+        return void 0;
+      }
+    }, "parseJsonField");
+    return result.rows.map((row) => ({
+      id: row.id,
+      conversationId: row.conversation_id,
+      userId: row.user_id,
+      agentId: row.agent_id,
+      agentName: row.agent_name ?? void 0,
+      operationId: row.operation_id ?? void 0,
+      stepIndex: typeof row.step_index === "number" ? row.step_index : Number(row.step_index ?? 0),
+      type: row.type,
+      role: row.role,
+      content: row.content ?? void 0,
+      arguments: parseJsonField(row.arguments),
+      result: parseJsonField(row.result),
+      usage: parseJsonField(row.usage),
+      subAgentId: row.sub_agent_id ?? void 0,
+      subAgentName: row.sub_agent_name ?? void 0,
+      createdAt: row.created_at ?? (/* @__PURE__ */ new Date()).toISOString()
+    }));
+  }
   /**
    * Clear messages for a user
    */
@@ -417,11 +591,16 @@ var LibSQLMemoryAdapter = class {
     await this.initialize();
     const messagesTable = `${this.tablePrefix}_messages`;
     const conversationsTable = `${this.tablePrefix}_conversations`;
+    const stepsTable = `${this.tablePrefix}_steps`;
     if (conversationId) {
       await this.client.execute({
         sql: `DELETE FROM ${messagesTable} WHERE conversation_id = ? AND user_id = ?`,
         args: [conversationId, userId]
       });
+      await this.client.execute({
+        sql: `DELETE FROM ${stepsTable} WHERE conversation_id = ? AND user_id = ?`,
+        args: [conversationId, userId]
+      });
     } else {
       await this.client.execute({
         sql: `DELETE FROM ${messagesTable}
@@ -430,6 +609,13 @@ var LibSQLMemoryAdapter = class {
               )`,
         args: [userId]
       });
+      await this.client.execute({
+        sql: `DELETE FROM ${stepsTable}
+              WHERE conversation_id IN (
+                SELECT id FROM ${conversationsTable} WHERE user_id = ?
+              )`,
+        args: [userId]
+      });
     }
   }
   // ============================================================================
@@ -721,6 +907,9 @@ var LibSQLMemoryAdapter = class {
       workflowName: row.workflow_name,
       status: row.status,
       suspension: row.suspension ? JSON.parse(row.suspension) : void 0,
+      events: row.events ? JSON.parse(row.events) : void 0,
+      output: row.output ? JSON.parse(row.output) : void 0,
+      cancellation: row.cancellation ? JSON.parse(row.cancellation) : void 0,
       userId: row.user_id,
       conversationId: row.conversation_id,
       metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
@@ -735,15 +924,18 @@ var LibSQLMemoryAdapter = class {
     await this.initialize();
     const workflowStatesTable = `${this.tablePrefix}_workflow_states`;
     await this.client.execute({
-      sql: `INSERT OR REPLACE INTO ${workflowStatesTable} 
-            (id, workflow_id, workflow_name, status, suspension, user_id, conversation_id, metadata, created_at, updated_at) 
-            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+      sql: `INSERT OR REPLACE INTO ${workflowStatesTable}
+            (id, workflow_id, workflow_name, status, suspension, events, output, cancellation, user_id, conversation_id, metadata, created_at, updated_at)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
       args: [
         executionId,
         state.workflowId,
         state.workflowName,
         state.status,
         state.suspension ? safeStringify(state.suspension) : null,
+        state.events ? safeStringify(state.events) : null,
+        state.output ? safeStringify(state.output) : null,
+        state.cancellation ? safeStringify(state.cancellation) : null,
         state.userId || null,
         state.conversationId || null,
         state.metadata ? safeStringify(state.metadata) : null,
@@ -784,6 +976,9 @@ var LibSQLMemoryAdapter = class {
       workflowName: row.workflow_name,
       status: "suspended",
       suspension: row.suspension ? JSON.parse(row.suspension) : void 0,
+      events: row.events ? JSON.parse(row.events) : void 0,
+      output: row.output ? JSON.parse(row.output) : void 0,
+      cancellation: row.cancellation ? JSON.parse(row.cancellation) : void 0,
       userId: row.user_id,
       conversationId: row.conversation_id,
       metadata: row.metadata ? JSON.parse(row.metadata) : void 0,