@copilotkitnext/sqlite-runner 0.0.21-alpha.0 → 0.0.22-alpha.0

package/dist/index.js

@@ -36,13 +36,12 @@ module.exports = __toCommonJS(index_exports);
 
 // src/sqlite-runner.ts
 var import_runtime = require("@copilotkitnext/runtime");
-var import_rxjs = require("rxjs");
-var import_client = require("@ag-ui/client");
 var import_better_sqlite3 = __toESM(require("better-sqlite3"));
+var import_rxjs = require("rxjs");
 var SCHEMA_VERSION = 1;
-var ACTIVE_CONNECTIONS = /* @__PURE__ */ new Map();
-var SqliteAgentRunner = class extends import_runtime.AgentRunner {
+var SqliteAgentRunner = class _SqliteAgentRunner extends import_runtime.AgentRunnerBase {
   db;
+  channels = /* @__PURE__ */ new Map();
   constructor(options = {}) {
     super();
     const dbPath = options.dbPath ?? ":memory:";
@@ -51,11 +50,12 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
         "better-sqlite3 is required for SqliteAgentRunner but was not found.\nPlease install it in your project:\n  npm install better-sqlite3\n  or\n  pnpm add better-sqlite3\n  or\n  yarn add better-sqlite3\n\nIf you don't need persistence, use InMemoryAgentRunner instead."
       );
     }
-    this.db = new import_better_sqlite3.default(dbPath);
-    this.initializeSchema();
+    const db = new import_better_sqlite3.default(dbPath);
+    _SqliteAgentRunner.initializeSchema(db);
+    this.db = db;
   }
-  initializeSchema() {
-    this.db.exec(`
+  static initializeSchema(db) {
+    db.exec(`
       CREATE TABLE IF NOT EXISTS agent_runs (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
         thread_id TEXT NOT NULL,
@@ -67,7 +67,7 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
         version INTEGER NOT NULL
       )
     `);
-    this.db.exec(`
+    db.exec(`
       CREATE TABLE IF NOT EXISTS run_state (
         thread_id TEXT PRIMARY KEY,
         is_running INTEGER DEFAULT 0,
@@ -75,305 +75,143 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
         updated_at INTEGER NOT NULL
       )
     `);
-    this.db.exec(`
+    db.exec(`
       CREATE INDEX IF NOT EXISTS idx_thread_id ON agent_runs(thread_id);
       CREATE INDEX IF NOT EXISTS idx_parent_run_id ON agent_runs(parent_run_id);
     `);
-    this.db.exec(`
+    db.exec(`
       CREATE TABLE IF NOT EXISTS schema_version (
         version INTEGER PRIMARY KEY,
         applied_at INTEGER NOT NULL
       )
     `);
-    const currentVersion = this.db.prepare("SELECT version FROM schema_version ORDER BY version DESC LIMIT 1").get();
+    const currentVersion = db.prepare("SELECT version FROM schema_version ORDER BY version DESC LIMIT 1").get();
     if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {
-      this.db.prepare("INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)").run(SCHEMA_VERSION, Date.now());
+      db.prepare("INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)").run(SCHEMA_VERSION, Date.now());
     }
   }
-  storeRun(threadId, runId, events, input, parentRunId) {
-    const compactedEvents = (0, import_client.compactEvents)(events);
-    const stmt = this.db.prepare(`
-      INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version)
-      VALUES (?, ?, ?, ?, ?, ?, ?)
-    `);
+  // Hooks implementation using SQLite
+  async acquireRun(threadId, runId) {
+    const row = this.db.prepare("SELECT is_running FROM run_state WHERE thread_id = ?").get(threadId);
+    if (row?.is_running === 1) return false;
+    this.db.prepare(
+      "INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at) VALUES (?, ?, ?, ?)"
+    ).run(threadId, 1, runId, Date.now());
+    return true;
+  }
+  async releaseRun(threadId) {
+    this.db.prepare("INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at) VALUES (?, 0, NULL, ?)").run(threadId, Date.now());
+  }
+  async isRunningState(threadId) {
+    const row = this.db.prepare("SELECT is_running FROM run_state WHERE thread_id = ?").get(threadId);
+    return row?.is_running === 1;
+  }
+  async listRuns(threadId) {
+    const rows = this.db.prepare(
+      "SELECT run_id, events, created_at FROM agent_runs WHERE thread_id = ? ORDER BY created_at ASC"
+    ).all(threadId);
+    return rows.map((r) => ({ runId: r.run_id, events: JSON.parse(r.events), createdAt: r.created_at }));
+  }
+  async saveRun(threadId, runId, events, input, parentRunId) {
+    const stmt = this.db.prepare(
+      "INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version) VALUES (?, ?, ?, ?, ?, ?, ?)"
+    );
     stmt.run(
       threadId,
       runId,
       parentRunId ?? null,
-      JSON.stringify(compactedEvents),
-      // Store only this run's compacted events
+      JSON.stringify(events),
       JSON.stringify(input),
       Date.now(),
       SCHEMA_VERSION
     );
   }
-  getHistoricRuns(threadId) {
-    const stmt = this.db.prepare(`
-      WITH RECURSIVE run_chain AS (
-        -- Base case: find the root runs (those without parent)
-        SELECT * FROM agent_runs 
-        WHERE thread_id = ? AND parent_run_id IS NULL
-        
-        UNION ALL
-        
-        -- Recursive case: find children of current level
-        SELECT ar.* FROM agent_runs ar
-        INNER JOIN run_chain rc ON ar.parent_run_id = rc.run_id
-        WHERE ar.thread_id = ?
-      )
-      SELECT * FROM run_chain
-      ORDER BY created_at ASC
-    `);
-    const rows = stmt.all(threadId, threadId);
-    return rows.map((row) => ({
-      id: row.id,
-      thread_id: row.thread_id,
-      run_id: row.run_id,
-      parent_run_id: row.parent_run_id,
-      events: JSON.parse(row.events),
-      input: JSON.parse(row.input),
-      created_at: row.created_at,
-      version: row.version
-    }));
-  }
-  getLatestRunId(threadId) {
-    const stmt = this.db.prepare(`
-      SELECT run_id FROM agent_runs 
-      WHERE thread_id = ? 
-      ORDER BY created_at DESC 
-      LIMIT 1
-    `);
-    const result = stmt.get(threadId);
-    return result?.run_id ?? null;
+  async pageThreads(params) {
+    const limit = params.limit ?? 20;
+    const offset = params.offset ?? 0;
+    const totalRow = this.db.prepare("SELECT COUNT(DISTINCT thread_id) AS total FROM agent_runs").get();
+    const rows = this.db.prepare(
+      "SELECT thread_id, MAX(created_at) AS last FROM agent_runs GROUP BY thread_id ORDER BY last DESC LIMIT ? OFFSET ?"
+    ).all(limit, offset);
+    return { threadIds: rows.map((r) => r.thread_id), total: totalRow.total };
   }
-  setRunState(threadId, isRunning, runId) {
-    const stmt = this.db.prepare(`
-      INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at)
-      VALUES (?, ?, ?, ?)
-    `);
-    stmt.run(threadId, isRunning ? 1 : 0, runId ?? null, Date.now());
+  async deleteThreadStorage(threadId) {
+    this.db.prepare("DELETE FROM agent_runs WHERE thread_id = ?").run(threadId);
+    this.db.prepare("DELETE FROM run_state WHERE thread_id = ?").run(threadId);
   }
-  getRunState(threadId) {
-    const stmt = this.db.prepare(`
-      SELECT is_running, current_run_id FROM run_state WHERE thread_id = ?
-    `);
-    const result = stmt.get(threadId);
-    return {
-      isRunning: result?.is_running === 1,
-      currentRunId: result?.current_run_id ?? null
-    };
+  ensureChannel(threadId) {
+    let s = this.channels.get(threadId);
+    if (!s || s.closed) {
+      s = new import_rxjs.ReplaySubject(Infinity);
+      this.channels.set(threadId, s);
+    }
+    return s;
   }
-  run(request) {
-    const runState = this.getRunState(request.threadId);
-    if (runState.isRunning) {
-      throw new Error("Thread already running");
+  publishLive(threadId, event) {
+    if (event.type === "RUN_STARTED") {
+      const s = new import_rxjs.ReplaySubject(Infinity);
+      this.channels.set(threadId, s);
     }
-    this.setRunState(request.threadId, true, request.input.runId);
-    const seenMessageIds = /* @__PURE__ */ new Set();
-    const currentRunEvents = [];
-    const historicRuns = this.getHistoricRuns(request.threadId);
-    const historicMessageIds = /* @__PURE__ */ new Set();
-    for (const run of historicRuns) {
-      for (const event of run.events) {
-        if ("messageId" in event && typeof event.messageId === "string") {
-          historicMessageIds.add(event.messageId);
-        }
-        if (event.type === import_client.EventType.RUN_STARTED) {
-          const runStarted = event;
-          const messages = runStarted.input?.messages ?? [];
-          for (const message of messages) {
-            historicMessageIds.add(message.id);
-          }
+    this.ensureChannel(threadId).next(event);
+  }
+  completeLive(threadId) {
+    const s = this.ensureChannel(threadId);
+    if (!s.closed) s.complete();
+  }
+  subscribeLive(threadId) {
+    return this.ensureChannel(threadId).asObservable();
+  }
+  async closeLive(threadId) {
+    const s = this.channels.get(threadId);
+    if (s && !s.closed) s.complete();
+    this.channels.delete(threadId);
+  }
+  // Override connect to preserve original sqlite-runner ordering without re-compacting across runs
+  connect(request) {
+    const { threadId } = request;
+    const subject = new import_rxjs.ReplaySubject(Infinity);
+    void (async () => {
+      const runs = await this.listRuns(threadId);
+      const emittedMessageIds = /* @__PURE__ */ new Set();
+      for (const r of runs) {
+        for (const e of r.events) {
+          subject.next(e);
+          const id = e.messageId;
+          if (typeof id === "string") emittedMessageIds.add(id);
         }
       }
-    }
-    const nextSubject = new import_rxjs.ReplaySubject(Infinity);
-    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);
-    const prevSubject = prevConnection?.subject;
-    const runSubject = new import_rxjs.ReplaySubject(Infinity);
-    ACTIVE_CONNECTIONS.set(request.threadId, {
-      subject: nextSubject,
-      agent: request.agent,
-      runSubject,
-      currentEvents: currentRunEvents,
-      stopRequested: false
-    });
-    const runAgent = async () => {
-      const parentRunId = this.getLatestRunId(request.threadId);
-      try {
-        await request.agent.runAgent(request.input, {
-          onEvent: ({ event }) => {
-            let processedEvent = event;
-            if (event.type === import_client.EventType.RUN_STARTED) {
-              const runStartedEvent = event;
-              if (!runStartedEvent.input) {
-                const sanitizedMessages = request.input.messages ? request.input.messages.filter(
-                  (message) => !historicMessageIds.has(message.id)
-                ) : void 0;
-                const updatedInput = {
-                  ...request.input,
-                  ...sanitizedMessages !== void 0 ? { messages: sanitizedMessages } : {}
-                };
-                processedEvent = {
-                  ...runStartedEvent,
-                  input: updatedInput
-                };
-              }
-            }
-            runSubject.next(processedEvent);
-            nextSubject.next(processedEvent);
-            currentRunEvents.push(processedEvent);
-          },
-          onNewMessage: ({ message }) => {
-            if (!seenMessageIds.has(message.id)) {
-              seenMessageIds.add(message.id);
-            }
-          },
-          onRunStartedEvent: () => {
-            if (request.input.messages) {
-              for (const message of request.input.messages) {
-                if (!seenMessageIds.has(message.id)) {
-                  seenMessageIds.add(message.id);
-                }
-              }
+      const running = await this.isRunningState(threadId);
+      if (!running) {
+        subject.complete();
+        return;
+      }
+      const live$ = this.subscribeLive(threadId);
+      let completed = false;
+      live$.subscribe({
+        next: (e) => {
+          const id = e.messageId;
+          if (typeof id === "string" && emittedMessageIds.has(id)) return;
+          subject.next(e);
+          if (e.type === "RUN_FINISHED" || e.type === "RUN_ERROR") {
+            if (!completed) {
+              completed = true;
+              subject.complete();
             }
           }
-        });
-        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
-        const appendedEvents = (0, import_runtime.finalizeRunEvents)(currentRunEvents, {
-          stopRequested: connection?.stopRequested ?? false
-        });
-        for (const event of appendedEvents) {
-          runSubject.next(event);
-          nextSubject.next(event);
-        }
-        this.storeRun(
-          request.threadId,
-          request.input.runId,
-          currentRunEvents,
-          request.input,
-          parentRunId
-        );
-        this.setRunState(request.threadId, false);
-        if (connection) {
-          connection.agent = void 0;
-          connection.runSubject = void 0;
-          connection.currentEvents = void 0;
-          connection.stopRequested = false;
-        }
-        runSubject.complete();
-        nextSubject.complete();
-        ACTIVE_CONNECTIONS.delete(request.threadId);
-      } catch {
-        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
-        const appendedEvents = (0, import_runtime.finalizeRunEvents)(currentRunEvents, {
-          stopRequested: connection?.stopRequested ?? false
-        });
-        for (const event of appendedEvents) {
-          runSubject.next(event);
-          nextSubject.next(event);
-        }
-        if (currentRunEvents.length > 0) {
-          this.storeRun(
-            request.threadId,
-            request.input.runId,
-            currentRunEvents,
-            request.input,
-            parentRunId
-          );
-        }
-        this.setRunState(request.threadId, false);
-        if (connection) {
-          connection.agent = void 0;
-          connection.runSubject = void 0;
-          connection.currentEvents = void 0;
-          connection.stopRequested = false;
-        }
-        runSubject.complete();
-        nextSubject.complete();
-        ACTIVE_CONNECTIONS.delete(request.threadId);
-      }
-    };
-    if (prevSubject) {
-      prevSubject.subscribe({
-        next: (e) => nextSubject.next(e),
-        error: (err) => nextSubject.error(err),
+        },
+        error: (err) => subject.error(err),
         complete: () => {
-        }
-      });
-    }
-    runAgent();
-    return runSubject.asObservable();
-  }
-  connect(request) {
-    const connectionSubject = new import_rxjs.ReplaySubject(Infinity);
-    const historicRuns = this.getHistoricRuns(request.threadId);
-    const allHistoricEvents = [];
-    for (const run of historicRuns) {
-      allHistoricEvents.push(...run.events);
-    }
-    const compactedEvents = (0, import_client.compactEvents)(allHistoricEvents);
-    const emittedMessageIds = /* @__PURE__ */ new Set();
-    for (const event of compactedEvents) {
-      connectionSubject.next(event);
-      if ("messageId" in event && typeof event.messageId === "string") {
-        emittedMessageIds.add(event.messageId);
-      }
-    }
-    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);
-    const runState = this.getRunState(request.threadId);
-    if (activeConnection && (runState.isRunning || activeConnection.stopRequested)) {
-      activeConnection.subject.subscribe({
-        next: (event) => {
-          if ("messageId" in event && typeof event.messageId === "string" && emittedMessageIds.has(event.messageId)) {
-            return;
+          if (!completed) {
+            completed = true;
+            subject.complete();
           }
-          connectionSubject.next(event);
-        },
-        complete: () => connectionSubject.complete(),
-        error: (err) => connectionSubject.error(err)
+        }
       });
-    } else {
-      connectionSubject.complete();
-    }
-    return connectionSubject.asObservable();
-  }
-  isRunning(request) {
-    const runState = this.getRunState(request.threadId);
-    return Promise.resolve(runState.isRunning);
+    })();
+    return subject.asObservable();
   }
-  stop(request) {
-    const runState = this.getRunState(request.threadId);
-    if (!runState.isRunning) {
-      return Promise.resolve(false);
-    }
-    const connection = ACTIVE_CONNECTIONS.get(request.threadId);
-    const agent = connection?.agent;
-    if (!connection || !agent) {
-      return Promise.resolve(false);
-    }
-    if (connection.stopRequested) {
-      return Promise.resolve(false);
-    }
-    connection.stopRequested = true;
-    this.setRunState(request.threadId, false);
-    try {
-      agent.abortRun();
-      return Promise.resolve(true);
-    } catch (error) {
-      console.error("Failed to abort sqlite agent run", error);
-      connection.stopRequested = false;
-      this.setRunState(request.threadId, true);
-      return Promise.resolve(false);
-    }
-  }
-  /**
-   * Close the database connection (for cleanup)
-   */
   close() {
-    if (this.db) {
-      this.db.close();
-    }
+    if (this.db) this.db.close();
   }
 };
 // Annotate the CommonJS export names for ESM import in node:

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/sqlite-runner.ts"],"sourcesContent":["export * from \"./sqlite-runner\";\n","import {\n  AgentRunner,\n  finalizeRunEvents,\n  type AgentRunnerConnectRequest,\n  type AgentRunnerIsRunningRequest,\n  type AgentRunnerRunRequest,\n  type AgentRunnerStopRequest,\n} from \"@copilotkitnext/runtime\";\nimport { Observable, ReplaySubject } from \"rxjs\";\nimport {\n  AbstractAgent,\n  BaseEvent,\n  RunAgentInput,\n  EventType,\n  RunStartedEvent,\n  compactEvents,\n} from \"@ag-ui/client\";\nimport Database from \"better-sqlite3\";\n\nconst SCHEMA_VERSION = 1;\n\ninterface AgentRunRecord {\n  id: number;\n  thread_id: string;\n  run_id: string;\n  parent_run_id: string | null;\n  events: BaseEvent[];\n  input: RunAgentInput;\n  created_at: number;\n  version: number;\n}\n\nexport interface SqliteAgentRunnerOptions {\n  dbPath?: string;\n}\n\ninterface ActiveConnectionContext {\n  subject: ReplaySubject<BaseEvent>;\n  agent?: AbstractAgent;\n  runSubject?: ReplaySubject<BaseEvent>;\n  currentEvents?: BaseEvent[];\n  stopRequested?: boolean;\n}\n\n// Active connections for streaming events and stop support\nconst ACTIVE_CONNECTIONS = new Map<string, ActiveConnectionContext>();\n\nexport class SqliteAgentRunner extends AgentRunner {\n  private db: any;\n\n  constructor(options: SqliteAgentRunnerOptions = {}) {\n    super();\n    const dbPath = options.dbPath ?? \":memory:\";\n    \n    if (!Database) {\n      throw new Error(\n        'better-sqlite3 is required for SqliteAgentRunner but was not found.\\n' +\n        'Please install it in your project:\\n' +\n        '  npm install better-sqlite3\\n' +\n        '  or\\n' +\n        '  pnpm add better-sqlite3\\n' +\n        '  or\\n' +\n        '  yarn add better-sqlite3\\n\\n' +\n        'If you don\\'t need persistence, use InMemoryAgentRunner instead.'\n      );\n    }\n    \n    this.db = new Database(dbPath);\n    this.initializeSchema();\n  }\n\n  private initializeSchema(): void {\n    // Create the agent_runs table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS agent_runs (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        thread_id TEXT NOT NULL,\n        run_id TEXT NOT NULL UNIQUE,\n        parent_run_id TEXT,\n        events TEXT NOT NULL,\n        input TEXT NOT NULL,\n        created_at INTEGER NOT NULL,\n        version INTEGER NOT NULL\n      )\n    `);\n\n    // Create run_state table to track active runs\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS run_state (\n        thread_id TEXT PRIMARY KEY,\n        is_running INTEGER DEFAULT 0,\n        current_run_id TEXT,\n        updated_at INTEGER NOT NULL\n      )\n    `);\n\n    // Create indexes for efficient queries\n    this.db.exec(`\n      CREATE INDEX IF NOT EXISTS idx_thread_id ON agent_runs(thread_id);\n      CREATE INDEX IF NOT EXISTS idx_parent_run_id ON agent_runs(parent_run_id);\n    `);\n\n    // Create schema version table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS schema_version (\n        version INTEGER PRIMARY KEY,\n        applied_at INTEGER NOT NULL\n      )\n    `);\n\n    // Check and set schema version\n    const currentVersion = this.db\n      .prepare(\"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\")\n      .get() as { version: number } | undefined;\n\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      this.db\n        .prepare(\"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\")\n        .run(SCHEMA_VERSION, Date.now());\n    }\n  }\n\n  private storeRun(\n    threadId: string,\n    runId: string,\n    events: BaseEvent[],\n    input: RunAgentInput,\n    parentRunId?: string | null\n  ): void {\n    // Compact ONLY the events from this run\n    const compactedEvents = compactEvents(events);\n    \n    const stmt = this.db.prepare(`\n      INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version)\n      VALUES (?, ?, ?, ?, ?, ?, ?)\n    `);\n\n    stmt.run(\n      threadId,\n      runId,\n      parentRunId ?? null,\n      JSON.stringify(compactedEvents), // Store only this run's compacted events\n      JSON.stringify(input),\n      Date.now(),\n      SCHEMA_VERSION\n    );\n  }\n\n  private getHistoricRuns(threadId: string): AgentRunRecord[] {\n    const stmt = this.db.prepare(`\n      WITH RECURSIVE run_chain AS (\n        -- Base case: find the root runs (those without parent)\n        SELECT * FROM agent_runs \n        WHERE thread_id = ? AND parent_run_id IS NULL\n        \n        UNION ALL\n        \n        -- Recursive case: find children of current level\n        SELECT ar.* FROM agent_runs ar\n        INNER JOIN run_chain rc ON ar.parent_run_id = rc.run_id\n        WHERE ar.thread_id = ?\n      )\n      SELECT * FROM run_chain\n      ORDER BY created_at ASC\n    `);\n\n    const rows = stmt.all(threadId, threadId) as any[];\n    \n    return rows.map(row => ({\n      id: row.id,\n      thread_id: row.thread_id,\n      run_id: row.run_id,\n      parent_run_id: row.parent_run_id,\n      events: JSON.parse(row.events),\n      input: JSON.parse(row.input),\n      created_at: row.created_at,\n      version: row.version\n    }));\n  }\n\n  private getLatestRunId(threadId: string): string | null {\n    const stmt = this.db.prepare(`\n      SELECT run_id FROM agent_runs \n      WHERE thread_id = ? \n      ORDER BY created_at DESC \n      LIMIT 1\n    `);\n\n    const result = stmt.get(threadId) as { run_id: string } | undefined;\n    return result?.run_id ?? null;\n  }\n\n  private setRunState(threadId: string, isRunning: boolean, runId?: string): void {\n    const stmt = this.db.prepare(`\n      INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at)\n      VALUES (?, ?, ?, ?)\n    `);\n    stmt.run(threadId, isRunning ? 1 : 0, runId ?? null, Date.now());\n  }\n\n  private getRunState(threadId: string): { isRunning: boolean; currentRunId: string | null } {\n    const stmt = this.db.prepare(`\n      SELECT is_running, current_run_id FROM run_state WHERE thread_id = ?\n    `);\n    const result = stmt.get(threadId) as { is_running: number; current_run_id: string | null } | undefined;\n    \n    return {\n      isRunning: result?.is_running === 1,\n      currentRunId: result?.current_run_id ?? null\n    };\n  }\n\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    // Check if thread is already running in database\n    const runState = this.getRunState(request.threadId);\n    if (runState.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n\n    // Mark thread as running in database\n    this.setRunState(request.threadId, true, request.input.runId);\n\n    // Track seen message IDs and current run events in memory for this run\n    const seenMessageIds = new Set<string>();\n    const currentRunEvents: BaseEvent[] = [];\n    \n    // Get all previously seen message IDs from historic runs\n    const historicRuns = this.getHistoricRuns(request.threadId);\n    const historicMessageIds = new Set<string>();\n    for (const run of historicRuns) {\n      for (const event of run.events) {\n        if ('messageId' in event && typeof event.messageId === 'string') {\n          historicMessageIds.add(event.messageId);\n        }\n        if (event.type === EventType.RUN_STARTED) {\n          const runStarted = event as RunStartedEvent;\n          const messages = runStarted.input?.messages ?? [];\n          for (const message of messages) {\n            historicMessageIds.add(message.id);\n          }\n        }\n      }\n    }\n\n    // Get or create subject for this thread's connections\n    const nextSubject = new ReplaySubject<BaseEvent>(Infinity);\n    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const prevSubject = prevConnection?.subject;\n    \n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Update the active connection for this thread\n    ACTIVE_CONNECTIONS.set(request.threadId, {\n      subject: nextSubject,\n      agent: request.agent,\n      runSubject,\n      currentEvents: currentRunEvents,\n      stopRequested: false,\n    });\n\n    // Helper function to run the agent and handle errors\n    const runAgent = async () => {\n      // Get parent run ID for chaining\n      const parentRunId = this.getLatestRunId(request.threadId);\n      \n      try {\n        await request.agent.runAgent(request.input, {\n          onEvent: ({ event }) => {\n            let processedEvent: BaseEvent = event;\n            if (event.type === EventType.RUN_STARTED) {\n              const runStartedEvent = event as RunStartedEvent;\n              if (!runStartedEvent.input) {\n                const sanitizedMessages = request.input.messages\n                  ? request.input.messages.filter(\n                      (message) => !historicMessageIds.has(message.id),\n                    )\n                  : undefined;\n                const updatedInput = {\n                  ...request.input,\n                  ...(sanitizedMessages !== undefined\n                    ? { messages: sanitizedMessages }\n                    : {}),\n                };\n                processedEvent = {\n                  ...runStartedEvent,\n                  input: updatedInput,\n                } as RunStartedEvent;\n              }\n            }\n\n            runSubject.next(processedEvent); // For run() return - only agent events\n            nextSubject.next(processedEvent); // For connect() / store - all events\n            currentRunEvents.push(processedEvent); // Accumulate for database storage\n          },\n          onNewMessage: ({ message }) => {\n            // Called for each new message\n            if (!seenMessageIds.has(message.id)) {\n              seenMessageIds.add(message.id);\n            }\n          },\n          onRunStartedEvent: () => {\n            // Mark input messages as seen without emitting duplicates\n            if (request.input.messages) {\n              for (const message of request.input.messages) {\n                if (!seenMessageIds.has(message.id)) {\n                  seenMessageIds.add(message.id);\n                }\n              }\n            }\n          },\n        });\n        \n        const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run in database\n        this.storeRun(\n          request.threadId,\n          request.input.runId,\n          currentRunEvents,\n          request.input,\n          parentRunId\n        );\n        \n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Complete the subjects\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      } catch {\n        const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run even if it failed (partial events)\n        if (currentRunEvents.length > 0) {\n          this.storeRun(\n            request.threadId,\n            request.input.runId,\n            currentRunEvents,\n            request.input,\n            parentRunId\n          );\n        }\n        \n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Don't emit error to the subject, just complete it\n        // This allows subscribers to get events emitted before the error\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      }\n    };\n\n    // Bridge previous events if they exist\n    if (prevSubject) {\n      prevSubject.subscribe({\n        next: (e) => nextSubject.next(e),\n        error: (err) => nextSubject.error(err),\n        complete: () => {\n          // Don't complete nextSubject here - it needs to stay open for new events\n        },\n      });\n    }\n\n    // Start the agent execution immediately (not lazily)\n    runAgent();\n\n    // Return the run subject (only agent events, no injected messages)\n    return runSubject.asObservable();\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const connectionSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Load historic runs from database\n    const historicRuns = this.getHistoricRuns(request.threadId);\n    \n    // Collect all historic events from database\n    const allHistoricEvents: BaseEvent[] = [];\n    for (const run of historicRuns) {\n      allHistoricEvents.push(...run.events);\n    }\n    \n    // Compact all events together before emitting\n    const compactedEvents = compactEvents(allHistoricEvents);\n    \n    // Emit compacted events and track message IDs\n    const emittedMessageIds = new Set<string>();\n    for (const event of compactedEvents) {\n      connectionSubject.next(event);\n      if ('messageId' in event && typeof event.messageId === 'string') {\n        emittedMessageIds.add(event.messageId);\n      }\n    }\n    \n    // Bridge active run to connection if exists\n    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const runState = this.getRunState(request.threadId);\n\n    if (activeConnection && (runState.isRunning || activeConnection.stopRequested)) {\n      activeConnection.subject.subscribe({\n        next: (event) => {\n          // Skip message events that we've already emitted from historic\n          if ('messageId' in event && typeof event.messageId === 'string' && emittedMessageIds.has(event.messageId)) {\n            return;\n          }\n          connectionSubject.next(event);\n        },\n        complete: () => connectionSubject.complete(),\n        error: (err) => connectionSubject.error(err)\n      });\n    } else {\n      // No active run, complete after historic events\n      connectionSubject.complete();\n    }\n    \n    return connectionSubject.asObservable();\n  }\n\n  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean> {\n    const runState = this.getRunState(request.threadId);\n    return Promise.resolve(runState.isRunning);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const runState = this.getRunState(request.threadId);\n    if (!runState.isRunning) {\n      return Promise.resolve(false);\n    }\n\n    const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const agent = connection?.agent;\n\n    if (!connection || !agent) {\n      return Promise.resolve(false);\n    }\n\n    if (connection.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    connection.stopRequested = true;\n    this.setRunState(request.threadId, false);\n\n    try {\n      agent.abortRun();\n      return Promise.resolve(true);\n    } catch (error) {\n      console.error(\"Failed to abort sqlite agent run\", error);\n      connection.stopRequested = false;\n      this.setRunState(request.threadId, true);\n      return Promise.resolve(false);\n    }\n  }\n\n  /**\n   * Close the database connection (for cleanup)\n   */\n  close(): void {\n    if (this.db) {\n      this.db.close();\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,qBAOO;AACP,kBAA0C;AAC1C,oBAOO;AACP,4BAAqB;AAErB,IAAM,iBAAiB;AA0BvB,IAAM,qBAAqB,oBAAI,IAAqC;AAE7D,IAAM,oBAAN,cAAgC,2BAAY;AAAA,EACzC;AAAA,EAER,YAAY,UAAoC,CAAC,GAAG;AAClD,UAAM;AACN,UAAM,SAAS,QAAQ,UAAU;AAEjC,QAAI,CAAC,sBAAAA,SAAU;AACb,YAAM,IAAI;AAAA,QACR;AAAA,MAQF;AAAA,IACF;AAEA,SAAK,KAAK,IAAI,sBAAAA,QAAS,MAAM;AAC7B,SAAK,iBAAiB;AAAA,EACxB;AAAA,EAEQ,mBAAyB;AAE/B,SAAK,GAAG,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAWZ;AAGD,SAAK,GAAG,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAOZ;AAGD,SAAK,GAAG,KAAK;AAAA;AAAA;AAAA,KAGZ;AAGD,SAAK,GAAG,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA,KAKZ;AAGD,UAAM,iBAAiB,KAAK,GACzB,QAAQ,kEAAkE,EAC1E,IAAI;AAEP,QAAI,CAAC,kBAAkB,eAAe,UAAU,gBAAgB;AAC9D,WAAK,GACF,QAAQ,2EAA2E,EACnF,IAAI,gBAAgB,KAAK,IAAI,CAAC;AAAA,IACnC;AAAA,EACF;AAAA,EAEQ,SACN,UACA,OACA,QACA,OACA,aACM;AAEN,UAAM,sBAAkB,6BAAc,MAAM;AAE5C,UAAM,OAAO,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA,KAG5B;AAED,SAAK;AAAA,MACH;AAAA,MACA;AAAA,MACA,eAAe;AAAA,MACf,KAAK,UAAU,eAAe;AAAA;AAAA,MAC9B,KAAK,UAAU,KAAK;AAAA,MACpB,KAAK,IAAI;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,gBAAgB,UAAoC;AAC1D,UAAM,OAAO,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAe5B;AAED,UAAM,OAAO,KAAK,IAAI,UAAU,QAAQ;AAExC,WAAO,KAAK,IAAI,UAAQ;AAAA,MACtB,IAAI,IAAI;AAAA,MACR,WAAW,IAAI;AAAA,MACf,QAAQ,IAAI;AAAA,MACZ,eAAe,IAAI;AAAA,MACnB,QAAQ,KAAK,MAAM,IAAI,MAAM;AAAA,MAC7B,OAAO,KAAK,MAAM,IAAI,KAAK;AAAA,MAC3B,YAAY,IAAI;AAAA,MAChB,SAAS,IAAI;AAAA,IACf,EAAE;AAAA,EACJ;AAAA,EAEQ,eAAe,UAAiC;AACtD,UAAM,OAAO,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA,KAK5B;AAED,UAAM,SAAS,KAAK,IAAI,QAAQ;AAChC,WAAO,QAAQ,UAAU;AAAA,EAC3B;AAAA,EAEQ,YAAY,UAAkB,WAAoB,OAAsB;AAC9E,UAAM,OAAO,KAAK,GAAG,QAAQ;AAAA;AAAA;AAAA,KAG5B;AACD,SAAK,IAAI,UAAU,YAAY,IAAI,GAAG,SAAS,MAAM,KAAK,IAAI,CAAC;AAAA,EACjE;AAAA,EAEQ,YAAY,UAAuE;AACzF,UAAM,OAAO,KAAK,GAAG,QAAQ;AAAA;AAAA,KAE5B;AACD,UAAM,SAAS,KAAK,IAAI,QAAQ;AAEhC,WAAO;AAAA,MACL,WAAW,QAAQ,eAAe;AAAA,MAClC,cAAc,QAAQ,kBAAkB;AAAA,IAC1C;AAAA,EACF;AAAA,EAEA,IAAI,SAAuD;AAEzD,UAAM,WAAW,KAAK,YAAY,QAAQ,QAAQ;AAClD,QAAI,SAAS,WAAW;AACtB,YAAM,IAAI,MAAM,wBAAwB;AAAA,IAC1C;AAGA,SAAK,YAAY,QAAQ,UAAU,MAAM,QAAQ,MAAM,KAAK;AAG5D,UAAM,iBAAiB,oBAAI,IAAY;AACvC,UAAM,mBAAgC,CAAC;AAGvC,UAAM,eAAe,KAAK,gBAAgB,QAAQ,QAAQ;AAC1D,UAAM,qBAAqB,oBAAI,IAAY;AAC3C,eAAW,OAAO,cAAc;AAC9B,iBAAW,SAAS,IAAI,QAAQ;AAC9B,YAAI,eAAe,SAAS,OAAO,MAAM,cAAc,UAAU;AAC/D,6BAAmB,IAAI,MAAM,SAAS;AAAA,QACxC;AACA,YAAI,MAAM,SAAS,wBAAU,aAAa;AACxC,gBAAM,aAAa;AACnB,gBAAM,WAAW,WAAW,OAAO,YAAY,CAAC;AAChD,qBAAW,WAAW,UAAU;AAC9B,+BAAmB,IAAI,QAAQ,EAAE;AAAA,UACnC;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAGA,UAAM,cAAc,IAAI,0BAAyB,QAAQ;AACzD,UAAM,iBAAiB,mBAAmB,IAAI,QAAQ,QAAQ;AAC9D,UAAM,cAAc,gBAAgB;AAGpC,UAAM,aAAa,IAAI,0BAAyB,QAAQ;AAGxD,uBAAmB,IAAI,QAAQ,UAAU;AAAA,MACvC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf;AAAA,MACA,eAAe;AAAA,MACf,eAAe;AAAA,IACjB,CAAC;AAGD,UAAM,WAAW,YAAY;AAE3B,YAAM,cAAc,KAAK,eAAe,QAAQ,QAAQ;AAExD,UAAI;AACF,cAAM,QAAQ,MAAM,SAAS,QAAQ,OAAO;AAAA,UAC1C,SAAS,CAAC,EAAE,MAAM,MAAM;AACtB,gBAAI,iBAA4B;AAChC,gBAAI,MAAM,SAAS,wBAAU,aAAa;AACxC,oBAAM,kBAAkB;AACxB,kBAAI,CAAC,gBAAgB,OAAO;AAC1B,sBAAM,oBAAoB,QAAQ,MAAM,WACpC,QAAQ,MAAM,SAAS;AAAA,kBACrB,CAAC,YAAY,CAAC,mBAAmB,IAAI,QAAQ,EAAE;AAAA,gBACjD,IACA;AACJ,sBAAM,eAAe;AAAA,kBACnB,GAAG,QAAQ;AAAA,kBACX,GAAI,sBAAsB,SACtB,EAAE,UAAU,kBAAkB,IAC9B,CAAC;AAAA,gBACP;AACA,iCAAiB;AAAA,kBACf,GAAG;AAAA,kBACH,OAAO;AAAA,gBACT;AAAA,cACF;AAAA,YACF;AAEA,uBAAW,KAAK,cAAc;AAC9B,wBAAY,KAAK,cAAc;AAC/B,6BAAiB,KAAK,cAAc;AAAA,UACtC;AAAA,UACA,cAAc,CAAC,EAAE,QAAQ,MAAM;AAE7B,gBAAI,CAAC,eAAe,IAAI,QAAQ,EAAE,GAAG;AACnC,6BAAe,IAAI,QAAQ,EAAE;AAAA,YAC/B;AAAA,UACF;AAAA,UACA,mBAAmB,MAAM;AAEvB,gBAAI,QAAQ,MAAM,UAAU;AAC1B,yBAAW,WAAW,QAAQ,MAAM,UAAU;AAC5C,oBAAI,CAAC,eAAe,IAAI,QAAQ,EAAE,GAAG;AACnC,iCAAe,IAAI,QAAQ,EAAE;AAAA,gBAC/B;AAAA,cACF;AAAA,YACF;AAAA,UACF;AAAA,QACF,CAAC;AAED,cAAM,aAAa,mBAAmB,IAAI,QAAQ,QAAQ;AAC1D,cAAM,qBAAiB,kCAAkB,kBAAkB;AAAA,UACzD,eAAe,YAAY,iBAAiB;AAAA,QAC9C,CAAC;AACD,mBAAW,SAAS,gBAAgB;AAClC,qBAAW,KAAK,KAAK;AACrB,sBAAY,KAAK,KAAK;AAAA,QACxB;AAGA,aAAK;AAAA,UACH,QAAQ;AAAA,UACR,QAAQ,MAAM;AAAA,UACd;AAAA,UACA,QAAQ;AAAA,UACR;AAAA,QACF;AAGA,aAAK,YAAY,QAAQ,UAAU,KAAK;AAExC,YAAI,YAAY;AACd,qBAAW,QAAQ;AACnB,qBAAW,aAAa;AACxB,qBAAW,gBAAgB;AAC3B,qBAAW,gBAAgB;AAAA,QAC7B;AAGA,mBAAW,SAAS;AACpB,oBAAY,SAAS;AAErB,2BAAmB,OAAO,QAAQ,QAAQ;AAAA,MAC5C,QAAQ;AACN,cAAM,aAAa,mBAAmB,IAAI,QAAQ,QAAQ;AAC1D,cAAM,qBAAiB,kCAAkB,kBAAkB;AAAA,UACzD,eAAe,YAAY,iBAAiB;AAAA,QAC9C,CAAC;AACD,mBAAW,SAAS,gBAAgB;AAClC,qBAAW,KAAK,KAAK;AACrB,sBAAY,KAAK,KAAK;AAAA,QACxB;AAGA,YAAI,iBAAiB,SAAS,GAAG;AAC/B,eAAK;AAAA,YACH,QAAQ;AAAA,YACR,QAAQ,MAAM;AAAA,YACd;AAAA,YACA,QAAQ;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAGA,aAAK,YAAY,QAAQ,UAAU,KAAK;AAExC,YAAI,YAAY;AACd,qBAAW,QAAQ;AACnB,qBAAW,aAAa;AACxB,qBAAW,gBAAgB;AAC3B,qBAAW,gBAAgB;AAAA,QAC7B;AAIA,mBAAW,SAAS;AACpB,oBAAY,SAAS;AAErB,2BAAmB,OAAO,QAAQ,QAAQ;AAAA,MAC5C;AAAA,IACF;AAGA,QAAI,aAAa;AACf,kBAAY,UAAU;AAAA,QACpB,MAAM,CAAC,MAAM,YAAY,KAAK,CAAC;AAAA,QAC/B,OAAO,CAAC,QAAQ,YAAY,MAAM,GAAG;AAAA,QACrC,UAAU,MAAM;AAAA,QAEhB;AAAA,MACF,CAAC;AAAA,IACH;AAGA,aAAS;AAGT,WAAO,WAAW,aAAa;AAAA,EACjC;AAAA,EAEA,QAAQ,SAA2D;AACjE,UAAM,oBAAoB,IAAI,0BAAyB,QAAQ;AAG/D,UAAM,eAAe,KAAK,gBAAgB,QAAQ,QAAQ;AAG1D,UAAM,oBAAiC,CAAC;AACxC,eAAW,OAAO,cAAc;AAC9B,wBAAkB,KAAK,GAAG,IAAI,MAAM;AAAA,IACtC;AAGA,UAAM,sBAAkB,6BAAc,iBAAiB;AAGvD,UAAM,oBAAoB,oBAAI,IAAY;AAC1C,eAAW,SAAS,iBAAiB;AACnC,wBAAkB,KAAK,KAAK;AAC5B,UAAI,eAAe,SAAS,OAAO,MAAM,cAAc,UAAU;AAC/D,0BAAkB,IAAI,MAAM,SAAS;AAAA,MACvC;AAAA,IACF;AAGA,UAAM,mBAAmB,mBAAmB,IAAI,QAAQ,QAAQ;AAChE,UAAM,WAAW,KAAK,YAAY,QAAQ,QAAQ;AAElD,QAAI,qBAAqB,SAAS,aAAa,iBAAiB,gBAAgB;AAC9E,uBAAiB,QAAQ,UAAU;AAAA,QACjC,MAAM,CAAC,UAAU;AAEf,cAAI,eAAe,SAAS,OAAO,MAAM,cAAc,YAAY,kBAAkB,IAAI,MAAM,SAAS,GAAG;AACzG;AAAA,UACF;AACA,4BAAkB,KAAK,KAAK;AAAA,QAC9B;AAAA,QACA,UAAU,MAAM,kBAAkB,SAAS;AAAA,QAC3C,OAAO,CAAC,QAAQ,kBAAkB,MAAM,GAAG;AAAA,MAC7C,CAAC;AAAA,IACH,OAAO;AAEL,wBAAkB,SAAS;AAAA,IAC7B;AAEA,WAAO,kBAAkB,aAAa;AAAA,EACxC;AAAA,EAEA,UAAU,SAAwD;AAChE,UAAM,WAAW,KAAK,YAAY,QAAQ,QAAQ;AAClD,WAAO,QAAQ,QAAQ,SAAS,SAAS;AAAA,EAC3C;AAAA,EAEA,KAAK,SAA+D;AAClE,UAAM,WAAW,KAAK,YAAY,QAAQ,QAAQ;AAClD,QAAI,CAAC,SAAS,WAAW;AACvB,aAAO,QAAQ,QAAQ,KAAK;AAAA,IAC9B;AAEA,UAAM,aAAa,mBAAmB,IAAI,QAAQ,QAAQ;AAC1D,UAAM,QAAQ,YAAY;AAE1B,QAAI,CAAC,cAAc,CAAC,OAAO;AACzB,aAAO,QAAQ,QAAQ,KAAK;AAAA,IAC9B;AAEA,QAAI,WAAW,eAAe;AAC5B,aAAO,QAAQ,QAAQ,KAAK;AAAA,IAC9B;AAEA,eAAW,gBAAgB;AAC3B,SAAK,YAAY,QAAQ,UAAU,KAAK;AAExC,QAAI;AACF,YAAM,SAAS;AACf,aAAO,QAAQ,QAAQ,IAAI;AAAA,IAC7B,SAAS,OAAO;AACd,cAAQ,MAAM,oCAAoC,KAAK;AACvD,iBAAW,gBAAgB;AAC3B,WAAK,YAAY,QAAQ,UAAU,IAAI;AACvC,aAAO,QAAQ,QAAQ,KAAK;AAAA,IAC9B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,QAAc;AACZ,QAAI,KAAK,IAAI;AACX,WAAK,GAAG,MAAM;AAAA,IAChB;AAAA,EACF;AACF;","names":["Database"]}
+{"version":3,"sources":["../src/index.ts","../src/sqlite-runner.ts"],"sourcesContent":["export * from \"./sqlite-runner\";\n","import { AgentRunnerBase } from \"@copilotkitnext/runtime\";\nimport { BaseEvent, RunAgentInput } from \"@ag-ui/client\";\nimport Database from \"better-sqlite3\";\nimport { Observable, ReplaySubject } from \"rxjs\";\n\nconst SCHEMA_VERSION = 1;\n\nexport interface SqliteAgentRunnerOptions {\n  dbPath?: string;\n}\nexport class SqliteAgentRunner extends AgentRunnerBase {\n  private db: any;\n  private channels = new Map<string, ReplaySubject<BaseEvent>>();\n\n  constructor(options: SqliteAgentRunnerOptions = {}) {\n    super();\n    const dbPath = options.dbPath ?? \":memory:\";\n    if (!Database) {\n      throw new Error(\n        'better-sqlite3 is required for SqliteAgentRunner but was not found.\\n' +\n          'Please install it in your project:\\n' +\n          '  npm install better-sqlite3\\n' +\n          '  or\\n' +\n          '  pnpm add better-sqlite3\\n' +\n          '  or\\n' +\n          '  yarn add better-sqlite3\\n\\n' +\n          \"If you don't need persistence, use InMemoryAgentRunner instead.\",\n      );\n    }\n    const db = new Database(dbPath);\n    SqliteAgentRunner.initializeSchema(db);\n    this.db = db;\n  }\n\n  private static initializeSchema(db: any): void {\n    db.exec(`\n      CREATE TABLE IF NOT EXISTS agent_runs (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        thread_id TEXT NOT NULL,\n        run_id TEXT NOT NULL UNIQUE,\n        parent_run_id TEXT,\n        events TEXT NOT NULL,\n        input TEXT NOT NULL,\n        created_at INTEGER NOT NULL,\n        version INTEGER NOT NULL\n      )\n    `);\n\n    db.exec(`\n      CREATE TABLE IF NOT EXISTS run_state (\n        thread_id TEXT PRIMARY KEY,\n        is_running INTEGER DEFAULT 0,\n        current_run_id TEXT,\n        updated_at INTEGER NOT NULL\n      )\n    `);\n\n    db.exec(`\n      CREATE INDEX IF NOT EXISTS idx_thread_id ON agent_runs(thread_id);\n      CREATE INDEX IF NOT EXISTS idx_parent_run_id ON agent_runs(parent_run_id);\n    `);\n\n    db.exec(`\n      CREATE TABLE IF NOT EXISTS schema_version (\n        version INTEGER PRIMARY KEY,\n        applied_at INTEGER NOT NULL\n      )\n    `);\n\n    const currentVersion = db\n      .prepare(\"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\")\n      .get() as { version: number } | undefined;\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      db\n        .prepare(\"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\")\n        .run(SCHEMA_VERSION, Date.now());\n    }\n  }\n\n  // Hooks implementation using SQLite\n  protected async acquireRun(threadId: string, runId: string): Promise<boolean> {\n    const row = this.db.prepare(\"SELECT is_running FROM run_state WHERE thread_id = ?\").get(threadId) as\n      | { is_running: number }\n      | undefined;\n    if (row?.is_running === 1) return false;\n    this.db\n      .prepare(\n        \"INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at) VALUES (?, ?, ?, ?)\",\n      )\n      .run(threadId, 1, runId, Date.now());\n    return true;\n  }\n\n  protected async releaseRun(threadId: string): Promise<void> {\n    this.db\n      .prepare(\"INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at) VALUES (?, 0, NULL, ?)\")\n      .run(threadId, Date.now());\n  }\n\n  protected async isRunningState(threadId: string): Promise<boolean> {\n    const row = this.db.prepare(\"SELECT is_running FROM run_state WHERE thread_id = ?\").get(threadId) as\n      | { is_running: number }\n      | undefined;\n    return row?.is_running === 1;\n  }\n\n  protected async listRuns(threadId: string): Promise<Array<{ runId: string; events: BaseEvent[]; createdAt: number }>> {\n    const rows = this.db\n      .prepare(\n        \"SELECT run_id, events, created_at FROM agent_runs WHERE thread_id = ? ORDER BY created_at ASC\",\n      )\n      .all(threadId) as Array<{ run_id: string; events: string; created_at: number }>;\n    return rows.map((r) => ({ runId: r.run_id, events: JSON.parse(r.events), createdAt: r.created_at }));\n  }\n\n  protected async saveRun(\n    threadId: string,\n    runId: string,\n    events: BaseEvent[],\n    input: RunAgentInput,\n    parentRunId: string | null,\n  ): Promise<void> {\n    const stmt = this.db.prepare(\n      \"INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version) VALUES (?, ?, ?, ?, ?, ?, ?)\",\n    );\n    stmt.run(\n      threadId,\n      runId,\n      parentRunId ?? null,\n      JSON.stringify(events),\n      JSON.stringify(input),\n      Date.now(),\n      SCHEMA_VERSION,\n    );\n  }\n\n  protected async pageThreads(params: { scope?: any; limit?: number; offset?: number }): Promise<{ threadIds: string[]; total: number }> {\n    const limit = params.limit ?? 20;\n    const offset = params.offset ?? 0;\n    const totalRow = this.db\n      .prepare(\"SELECT COUNT(DISTINCT thread_id) AS total FROM agent_runs\")\n      .get() as { total: number };\n    const rows = this.db\n      .prepare(\n        \"SELECT thread_id, MAX(created_at) AS last FROM agent_runs GROUP BY thread_id ORDER BY last DESC LIMIT ? OFFSET ?\",\n      )\n      .all(limit, offset) as Array<{ thread_id: string; last: number }>;\n    return { threadIds: rows.map((r) => r.thread_id), total: totalRow.total };\n  }\n\n  protected async deleteThreadStorage(threadId: string): Promise<void> {\n    this.db.prepare(\"DELETE FROM agent_runs WHERE thread_id = ?\").run(threadId);\n    this.db.prepare(\"DELETE FROM run_state WHERE thread_id = ?\").run(threadId);\n  }\n\n  private ensureChannel(threadId: string): ReplaySubject<BaseEvent> {\n    let s = this.channels.get(threadId);\n    if (!s || s.closed) {\n      s = new ReplaySubject<BaseEvent>(Infinity);\n      this.channels.set(threadId, s);\n    }\n    return s;\n  }\n\n  protected publishLive(threadId: string, event: BaseEvent): void {\n    // Reset the live channel at the start of each run to avoid replaying\n    // previous runs' events to new concurrent connections.\n    if (event.type === 'RUN_STARTED') {\n      const s = new ReplaySubject<BaseEvent>(Infinity);\n      this.channels.set(threadId, s);\n    }\n    this.ensureChannel(threadId).next(event);\n  }\n\n  protected completeLive(threadId: string): void {\n    const s = this.ensureChannel(threadId);\n    if (!s.closed) s.complete();\n  }\n\n  protected subscribeLive(threadId: string): Observable<BaseEvent> {\n    return this.ensureChannel(threadId).asObservable();\n  }\n\n  protected async closeLive(threadId: string): Promise<void> {\n    const s = this.channels.get(threadId);\n    if (s && !s.closed) s.complete();\n    this.channels.delete(threadId);\n  }\n\n  // Override connect to preserve original sqlite-runner ordering without re-compacting across runs\n  connect(request: { threadId: string }) {\n    const { threadId } = request;\n    const subject = new ReplaySubject<BaseEvent>(Infinity);\n    void (async () => {\n      // Load runs and emit stored events in order\n      const runs = await this.listRuns(threadId);\n      const emittedMessageIds = new Set<string>();\n      for (const r of runs) {\n        for (const e of r.events) {\n          subject.next(e);\n          const id = (e as any).messageId;\n          if (typeof id === \"string\") emittedMessageIds.add(id);\n        }\n      }\n\n      const running = await this.isRunningState(threadId);\n      if (!running) {\n        subject.complete();\n        return;\n      }\n\n      const live$ = this.subscribeLive(threadId);\n      let completed = false;\n      live$.subscribe({\n        next: (e) => {\n          const id = (e as any).messageId;\n          if (typeof id === \"string\" && emittedMessageIds.has(id)) return;\n          subject.next(e);\n          if (e.type === 'RUN_FINISHED' || e.type === 'RUN_ERROR') {\n            if (!completed) { completed = true; subject.complete(); }\n          }\n        },\n        error: (err) => subject.error(err),\n        complete: () => { if (!completed) { completed = true; subject.complete(); } },\n      });\n    })();\n    return subject.asObservable();\n  }\n\n  close(): void {\n    if (this.db) this.db.close();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,qBAAgC;AAEhC,4BAAqB;AACrB,kBAA0C;AAE1C,IAAM,iBAAiB;AAKhB,IAAM,oBAAN,MAAM,2BAA0B,+BAAgB;AAAA,EAC7C;AAAA,EACA,WAAW,oBAAI,IAAsC;AAAA,EAE7D,YAAY,UAAoC,CAAC,GAAG;AAClD,UAAM;AACN,UAAM,SAAS,QAAQ,UAAU;AACjC,QAAI,CAAC,sBAAAA,SAAU;AACb,YAAM,IAAI;AAAA,QACR;AAAA,MAQF;AAAA,IACF;AACA,UAAM,KAAK,IAAI,sBAAAA,QAAS,MAAM;AAC9B,uBAAkB,iBAAiB,EAAE;AACrC,SAAK,KAAK;AAAA,EACZ;AAAA,EAEA,OAAe,iBAAiB,IAAe;AAC7C,OAAG,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAWP;AAED,OAAG,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAOP;AAED,OAAG,KAAK;AAAA;AAAA;AAAA,KAGP;AAED,OAAG,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA,KAKP;AAED,UAAM,iBAAiB,GACpB,QAAQ,kEAAkE,EAC1E,IAAI;AACP,QAAI,CAAC,kBAAkB,eAAe,UAAU,gBAAgB;AAC9D,SACG,QAAQ,2EAA2E,EACnF,IAAI,gBAAgB,KAAK,IAAI,CAAC;AAAA,IACnC;AAAA,EACF;AAAA;AAAA,EAGA,MAAgB,WAAW,UAAkB,OAAiC;AAC5E,UAAM,MAAM,KAAK,GAAG,QAAQ,sDAAsD,EAAE,IAAI,QAAQ;AAGhG,QAAI,KAAK,eAAe,EAAG,QAAO;AAClC,SAAK,GACF;AAAA,MACC;AAAA,IACF,EACC,IAAI,UAAU,GAAG,OAAO,KAAK,IAAI,CAAC;AACrC,WAAO;AAAA,EACT;AAAA,EAEA,MAAgB,WAAW,UAAiC;AAC1D,SAAK,GACF,QAAQ,6GAA6G,EACrH,IAAI,UAAU,KAAK,IAAI,CAAC;AAAA,EAC7B;AAAA,EAEA,MAAgB,eAAe,UAAoC;AACjE,UAAM,MAAM,KAAK,GAAG,QAAQ,sDAAsD,EAAE,IAAI,QAAQ;AAGhG,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,MAAgB,SAAS,UAA6F;AACpH,UAAM,OAAO,KAAK,GACf;AAAA,MACC;AAAA,IACF,EACC,IAAI,QAAQ;AACf,WAAO,KAAK,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,QAAQ,KAAK,MAAM,EAAE,MAAM,GAAG,WAAW,EAAE,WAAW,EAAE;AAAA,EACrG;AAAA,EAEA,MAAgB,QACd,UACA,OACA,QACA,OACA,aACe;AACf,UAAM,OAAO,KAAK,GAAG;AAAA,MACnB;AAAA,IACF;AACA,SAAK;AAAA,MACH;AAAA,MACA;AAAA,MACA,eAAe;AAAA,MACf,KAAK,UAAU,MAAM;AAAA,MACrB,KAAK,UAAU,KAAK;AAAA,MACpB,KAAK,IAAI;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAgB,YAAY,QAA2G;AACrI,UAAM,QAAQ,OAAO,SAAS;AAC9B,UAAM,SAAS,OAAO,UAAU;AAChC,UAAM,WAAW,KAAK,GACnB,QAAQ,2DAA2D,EACnE,IAAI;AACP,UAAM,OAAO,KAAK,GACf;AAAA,MACC;AAAA,IACF,EACC,IAAI,OAAO,MAAM;AACpB,WAAO,EAAE,WAAW,KAAK,IAAI,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,SAAS,MAAM;AAAA,EAC1E;AAAA,EAEA,MAAgB,oBAAoB,UAAiC;AACnE,SAAK,GAAG,QAAQ,4CAA4C,EAAE,IAAI,QAAQ;AAC1E,SAAK,GAAG,QAAQ,2CAA2C,EAAE,IAAI,QAAQ;AAAA,EAC3E;AAAA,EAEQ,cAAc,UAA4C;AAChE,QAAI,IAAI,KAAK,SAAS,IAAI,QAAQ;AAClC,QAAI,CAAC,KAAK,EAAE,QAAQ;AAClB,UAAI,IAAI,0BAAyB,QAAQ;AACzC,WAAK,SAAS,IAAI,UAAU,CAAC;AAAA,IAC/B;AACA,WAAO;AAAA,EACT;AAAA,EAEU,YAAY,UAAkB,OAAwB;AAG9D,QAAI,MAAM,SAAS,eAAe;AAChC,YAAM,IAAI,IAAI,0BAAyB,QAAQ;AAC/C,WAAK,SAAS,IAAI,UAAU,CAAC;AAAA,IAC/B;AACA,SAAK,cAAc,QAAQ,EAAE,KAAK,KAAK;AAAA,EACzC;AAAA,EAEU,aAAa,UAAwB;AAC7C,UAAM,IAAI,KAAK,cAAc,QAAQ;AACrC,QAAI,CAAC,EAAE,OAAQ,GAAE,SAAS;AAAA,EAC5B;AAAA,EAEU,cAAc,UAAyC;AAC/D,WAAO,KAAK,cAAc,QAAQ,EAAE,aAAa;AAAA,EACnD;AAAA,EAEA,MAAgB,UAAU,UAAiC;AACzD,UAAM,IAAI,KAAK,SAAS,IAAI,QAAQ;AACpC,QAAI,KAAK,CAAC,EAAE,OAAQ,GAAE,SAAS;AAC/B,SAAK,SAAS,OAAO,QAAQ;AAAA,EAC/B;AAAA;AAAA,EAGA,QAAQ,SAA+B;AACrC,UAAM,EAAE,SAAS,IAAI;AACrB,UAAM,UAAU,IAAI,0BAAyB,QAAQ;AACrD,UAAM,YAAY;AAEhB,YAAM,OAAO,MAAM,KAAK,SAAS,QAAQ;AACzC,YAAM,oBAAoB,oBAAI,IAAY;AAC1C,iBAAW,KAAK,MAAM;AACpB,mBAAW,KAAK,EAAE,QAAQ;AACxB,kBAAQ,KAAK,CAAC;AACd,gBAAM,KAAM,EAAU;AACtB,cAAI,OAAO,OAAO,SAAU,mBAAkB,IAAI,EAAE;AAAA,QACtD;AAAA,MACF;AAEA,YAAM,UAAU,MAAM,KAAK,eAAe,QAAQ;AAClD,UAAI,CAAC,SAAS;AACZ,gBAAQ,SAAS;AACjB;AAAA,MACF;AAEA,YAAM,QAAQ,KAAK,cAAc,QAAQ;AACzC,UAAI,YAAY;AAChB,YAAM,UAAU;AAAA,QACd,MAAM,CAAC,MAAM;AACX,gBAAM,KAAM,EAAU;AACtB,cAAI,OAAO,OAAO,YAAY,kBAAkB,IAAI,EAAE,EAAG;AACzD,kBAAQ,KAAK,CAAC;AACd,cAAI,EAAE,SAAS,kBAAkB,EAAE,SAAS,aAAa;AACvD,gBAAI,CAAC,WAAW;AAAE,0BAAY;AAAM,sBAAQ,SAAS;AAAA,YAAG;AAAA,UAC1D;AAAA,QACF;AAAA,QACA,OAAO,CAAC,QAAQ,QAAQ,MAAM,GAAG;AAAA,QACjC,UAAU,MAAM;AAAE,cAAI,CAAC,WAAW;AAAE,wBAAY;AAAM,oBAAQ,SAAS;AAAA,UAAG;AAAA,QAAE;AAAA,MAC9E,CAAC;AAAA,IACH,GAAG;AACH,WAAO,QAAQ,aAAa;AAAA,EAC9B;AAAA,EAEA,QAAc;AACZ,QAAI,KAAK,GAAI,MAAK,GAAG,MAAM;AAAA,EAC7B;AACF;","names":["Database"]}