@voltagent/libsql 1.0.10 → 1.0.12

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 {
@@ -147,6 +148,26 @@ var LibSQLMemoryAdapter = class {
         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)`,
@@ -154,7 +175,9 @@ 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();
@@ -389,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
    */
@@ -450,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
    */
@@ -457,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}
@@ -470,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]
+      });
     }
   }
   // ============================================================================