@copilotkitnext/sqlite-runner 1.51.4-next.7 → 1.51.4-next.8

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @copilotkitnext/sqlite-runner
 
+## 1.51.4-next.8
+
+### Patch Changes
+
+- @copilotkitnext/runtime@1.51.4-next.8
+
 ## 1.51.4-next.7
 
 ### Patch Changes

package/dist/index.js

@@ -85,9 +85,13 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
         applied_at INTEGER NOT NULL
       )
     `);
-    const currentVersion = this.db.prepare("SELECT version FROM schema_version ORDER BY version DESC LIMIT 1").get();
+    const currentVersion = this.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());
+      this.db.prepare(
+        "INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)"
+      ).run(SCHEMA_VERSION, Date.now());
     }
   }
   storeRun(threadId, runId, events, input, parentRunId) {

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 {\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(\n        \"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\",\n      )\n      .get() as { version: number } | undefined;\n\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      this.db\n        .prepare(\n          \"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\",\n        )\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(\n    threadId: string,\n    isRunning: boolean,\n    runId?: string,\n  ): 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): {\n    isRunning: boolean;\n    currentRunId: string | null;\n  } {\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\n      | { is_running: number; current_run_id: string | null }\n      | 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 (\n      activeConnection &&\n      (runState.isRunning || activeConnection.stopRequested)\n    ) {\n      activeConnection.subject.subscribe({\n        next: (event) => {\n          // Skip message events that we've already emitted from historic\n          if (\n            \"messageId\" in event &&\n            typeof event.messageId === \"string\" &&\n            emittedMessageIds.has(event.messageId)\n          ) {\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;AAAA,MACC;AAAA,IACF,EACC,IAAI;AAEP,QAAI,CAAC,kBAAkB,eAAe,UAAU,gBAAgB;AAC9D,WAAK,GACF;AAAA,QACC;AAAA,MACF,EACC,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,CAAC,SAAS;AAAA,MACxB,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,YACN,UACA,WACA,OACM;AACN,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,UAGlB;AACA,UAAM,OAAO,KAAK,GAAG,QAAQ;AAAA;AAAA,KAE5B;AACD,UAAM,SAAS,KAAK,IAAI,QAAQ;AAIhC,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,QACE,qBACC,SAAS,aAAa,iBAAiB,gBACxC;AACA,uBAAiB,QAAQ,UAAU;AAAA,QACjC,MAAM,CAAC,UAAU;AAEf,cACE,eAAe,SACf,OAAO,MAAM,cAAc,YAC3B,kBAAkB,IAAI,MAAM,SAAS,GACrC;AACA;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"]}

package/dist/index.mjs

@@ -55,9 +55,13 @@ var SqliteAgentRunner = class extends AgentRunner {
         applied_at INTEGER NOT NULL
       )
     `);
-    const currentVersion = this.db.prepare("SELECT version FROM schema_version ORDER BY version DESC LIMIT 1").get();
+    const currentVersion = this.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());
+      this.db.prepare(
+        "INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)"
+      ).run(SCHEMA_VERSION, Date.now());
     }
   }
   storeRun(threadId, runId, events, input, parentRunId) {

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/sqlite-runner.ts"],"sourcesContent":["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,EACE;AAAA,EACA;AAAA,OAKK;AACP,SAAqB,qBAAqB;AAC1C;AAAA,EAIE;AAAA,EAEA;AAAA,OACK;AACP,OAAO,cAAc;AAErB,IAAM,iBAAiB;AA0BvB,IAAM,qBAAqB,oBAAI,IAAqC;AAE7D,IAAM,oBAAN,cAAgC,YAAY;AAAA,EACzC;AAAA,EAER,YAAY,UAAoC,CAAC,GAAG;AAClD,UAAM;AACN,UAAM,SAAS,QAAQ,UAAU;AAEjC,QAAI,CAAC,UAAU;AACb,YAAM,IAAI;AAAA,QACR;AAAA,MAQF;AAAA,IACF;AAEA,SAAK,KAAK,IAAI,SAAS,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,kBAAkB,cAAc,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,UAAU,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,cAAyB,QAAQ;AACzD,UAAM,iBAAiB,mBAAmB,IAAI,QAAQ,QAAQ;AAC9D,UAAM,cAAc,gBAAgB;AAGpC,UAAM,aAAa,IAAI,cAAyB,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,UAAU,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,iBAAiB,kBAAkB,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,iBAAiB,kBAAkB,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,cAAyB,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,kBAAkB,cAAc,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":[]}
+{"version":3,"sources":["../src/sqlite-runner.ts"],"sourcesContent":["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(\n        \"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\",\n      )\n      .get() as { version: number } | undefined;\n\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      this.db\n        .prepare(\n          \"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\",\n        )\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(\n    threadId: string,\n    isRunning: boolean,\n    runId?: string,\n  ): 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): {\n    isRunning: boolean;\n    currentRunId: string | null;\n  } {\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\n      | { is_running: number; current_run_id: string | null }\n      | 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 (\n      activeConnection &&\n      (runState.isRunning || activeConnection.stopRequested)\n    ) {\n      activeConnection.subject.subscribe({\n        next: (event) => {\n          // Skip message events that we've already emitted from historic\n          if (\n            \"messageId\" in event &&\n            typeof event.messageId === \"string\" &&\n            emittedMessageIds.has(event.messageId)\n          ) {\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,EACE;AAAA,EACA;AAAA,OAKK;AACP,SAAqB,qBAAqB;AAC1C;AAAA,EAIE;AAAA,EAEA;AAAA,OACK;AACP,OAAO,cAAc;AAErB,IAAM,iBAAiB;AA0BvB,IAAM,qBAAqB,oBAAI,IAAqC;AAE7D,IAAM,oBAAN,cAAgC,YAAY;AAAA,EACzC;AAAA,EAER,YAAY,UAAoC,CAAC,GAAG;AAClD,UAAM;AACN,UAAM,SAAS,QAAQ,UAAU;AAEjC,QAAI,CAAC,UAAU;AACb,YAAM,IAAI;AAAA,QACR;AAAA,MAQF;AAAA,IACF;AAEA,SAAK,KAAK,IAAI,SAAS,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;AAAA,MACC;AAAA,IACF,EACC,IAAI;AAEP,QAAI,CAAC,kBAAkB,eAAe,UAAU,gBAAgB;AAC9D,WAAK,GACF;AAAA,QACC;AAAA,MACF,EACC,IAAI,gBAAgB,KAAK,IAAI,CAAC;AAAA,IACnC;AAAA,EACF;AAAA,EAEQ,SACN,UACA,OACA,QACA,OACA,aACM;AAEN,UAAM,kBAAkB,cAAc,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,CAAC,SAAS;AAAA,MACxB,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,YACN,UACA,WACA,OACM;AACN,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,UAGlB;AACA,UAAM,OAAO,KAAK,GAAG,QAAQ;AAAA;AAAA,KAE5B;AACD,UAAM,SAAS,KAAK,IAAI,QAAQ;AAIhC,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,UAAU,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,cAAyB,QAAQ;AACzD,UAAM,iBAAiB,mBAAmB,IAAI,QAAQ,QAAQ;AAC9D,UAAM,cAAc,gBAAgB;AAGpC,UAAM,aAAa,IAAI,cAAyB,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,UAAU,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,iBAAiB,kBAAkB,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,iBAAiB,kBAAkB,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,cAAyB,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,kBAAkB,cAAc,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,QACE,qBACC,SAAS,aAAa,iBAAiB,gBACxC;AACA,uBAAiB,QAAQ,UAAU;AAAA,QACjC,MAAM,CAAC,UAAU;AAEf,cACE,eAAe,SACf,OAAO,MAAM,cAAc,YAC3B,kBAAkB,IAAI,MAAM,SAAS,GACrC;AACA;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":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkitnext/sqlite-runner",
-  "version": "1.51.4-next.7",
+  "version": "1.51.4-next.8",
   "description": "SQLite-backed agent runner for CopilotKit2",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -22,13 +22,13 @@
     "tsup": "^8.5.0",
     "typescript": "5.8.2",
     "vitest": "^3.0.5",
-    "@copilotkitnext/eslint-config": "1.51.4-next.7",
-    "@copilotkitnext/typescript-config": "1.51.4-next.7"
+    "@copilotkitnext/eslint-config": "1.51.4-next.8",
+    "@copilotkitnext/typescript-config": "1.51.4-next.8"
   },
   "dependencies": {
     "@ag-ui/client": "0.0.42",
     "rxjs": "7.8.1",
-    "@copilotkitnext/runtime": "1.51.4-next.7"
+    "@copilotkitnext/runtime": "1.51.4-next.8"
   },
   "peerDependencies": {
     "better-sqlite3": "^12.2.0"
@@ -46,7 +46,6 @@
     "dev": "tsup --watch",
     "lint": "eslint .",
     "check-types": "tsc --noEmit",
-    "clean": "rm -rf dist",
     "test": "vitest run",
     "test:watch": "vitest",
     "test:coverage": "vitest run --coverage"

package/src/__tests__/sqlite-runner.e2e.test.ts

@@ -39,7 +39,10 @@ interface EmitAgentOptions {
   emitDefaultRunStarted?: boolean;
   includeRunFinished?: boolean;
   runFinishedEvent?: RunFinishedEvent;
-  afterEvent?: (args: { event: BaseEvent; index: number }) => void | Promise<void>;
+  afterEvent?: (args: {
+    event: BaseEvent;
+    index: number;
+  }) => void | Promise<void>;
 }
 
 class EmitAgent extends AbstractAgent {
@@ -85,12 +88,11 @@ class EmitAgent extends AbstractAgent {
       runFinishedEvent?.type === EventType.RUN_FINISHED;
 
     if (includeRunFinished && !hasRunFinishedEvent) {
-      const finishEvent: RunFinishedEvent =
-        runFinishedEvent ?? {
-          type: EventType.RUN_FINISHED,
-          threadId: input.threadId,
-          runId: input.runId,
-        };
+      const finishEvent: RunFinishedEvent = runFinishedEvent ?? {
+        type: EventType.RUN_FINISHED,
+        threadId: input.threadId,
+        runId: input.runId,
+      };
       await emit(finishEvent);
     }
   }
@@ -112,7 +114,10 @@ class EmitAgent extends AbstractAgent {
 }
 
 class ReplayAgent extends AbstractAgent {
-  constructor(private readonly replayEvents: BaseEvent[], threadId: string) {
+  constructor(
+    private readonly replayEvents: BaseEvent[],
+    threadId: string,
+  ) {
     super({ threadId });
   }
 
@@ -130,7 +135,10 @@ class ReplayAgent extends AbstractAgent {
 }
 
 class RunnerConnectAgent extends AbstractAgent {
-  constructor(private readonly runner: SqliteAgentRunner, threadId: string) {
+  constructor(
+    private readonly runner: SqliteAgentRunner,
+    threadId: string,
+  ) {
     super({ threadId });
   }
 
@@ -142,7 +150,9 @@ class RunnerConnectAgent extends AbstractAgent {
     return EMPTY;
   }
 
-  protected connect(input: RunAgentInput): ReturnType<AbstractAgent["connect"]> {
+  protected connect(
+    input: RunAgentInput,
+  ): ReturnType<AbstractAgent["connect"]> {
     return this.runner.connect({ threadId: input.threadId });
   }
 }
@@ -163,7 +173,11 @@ function createDeferred<T>(): Deferred<T> {
   return { promise, resolve, reject };
 }
 
-async function collectEvents(observable: ReturnType<SqliteAgentRunner["run"]> | ReturnType<SqliteAgentRunner["connect"]>) {
+async function collectEvents(
+  observable:
+    | ReturnType<SqliteAgentRunner["run"]>
+    | ReturnType<SqliteAgentRunner["connect"]>,
+) {
   return firstValueFrom(observable.pipe(toArray()));
 }
 
@@ -348,9 +362,9 @@ describe("SqliteAgentRunner e2e", () => {
       await replayAgent.connectAgent({ runId: "replay-run" });
 
       expect(replayAgent.messages).toEqual([existingMessage, newMessage]);
-      expect(new Set(replayAgent.messages.map((message) => message.id)).size).toBe(
-        replayAgent.messages.length,
-      );
+      expect(
+        new Set(replayAgent.messages.map((message) => message.id)).size,
+      ).toBe(replayAgent.messages.length);
     });
   });
 
@@ -438,7 +452,9 @@ describe("SqliteAgentRunner e2e", () => {
       );
 
       expectRunStartedEvent(runEvents[0], baseMessages);
-      expect(runEvents.filter((event) => event.type === EventType.TOOL_CALL_RESULT)).toHaveLength(1);
+      expect(
+        runEvents.filter((event) => event.type === EventType.TOOL_CALL_RESULT),
+      ).toHaveLength(1);
 
       const replayEvents = await collectEvents(runner.connect({ threadId }));
       const replayAgent = new ReplayAgent(replayEvents, threadId);
@@ -470,7 +486,9 @@ describe("SqliteAgentRunner e2e", () => {
           toolCallId,
         },
       ]);
-      expect(replayAgent.messages.filter((message) => message.role === "tool")).toHaveLength(1);
+      expect(
+        replayAgent.messages.filter((message) => message.role === "tool"),
+      ).toHaveLength(1);
     });
   });
 
@@ -538,11 +556,16 @@ describe("SqliteAgentRunner e2e", () => {
       await replayAgent.connectAgent({ runId: "replay-final" });
 
       const finalMessages = replayAgent.messages;
-      expect(new Set(finalMessages.map((message) => message.id)).size).toBe(finalMessages.length);
-      const roleCounts = finalMessages.reduce<Record<string, number>>((counts, message) => {
-        counts[message.role] = (counts[message.role] ?? 0) + 1;
-        return counts;
-      }, {});
+      expect(new Set(finalMessages.map((message) => message.id)).size).toBe(
+        finalMessages.length,
+      );
+      const roleCounts = finalMessages.reduce<Record<string, number>>(
+        (counts, message) => {
+          counts[message.role] = (counts[message.role] ?? 0) + 1;
+          return counts;
+        },
+        {},
+      );
       expect(roleCounts.system).toBe(1);
       expect(roleCounts.user).toBe(3);
       expect(roleCounts.assistant).toBe(3);
@@ -605,14 +628,16 @@ describe("SqliteAgentRunner e2e", () => {
       );
 
       expect(runEvents[0]).toEqual(customRunStarted);
-      expect(runEvents.filter((event) => event.type === EventType.RUN_FINISHED)).toHaveLength(1);
+      expect(
+        runEvents.filter((event) => event.type === EventType.RUN_FINISHED),
+      ).toHaveLength(1);
 
       const replayEvents = await collectEvents(runner.connect({ threadId }));
       const replayAgent = new ReplayAgent(replayEvents, threadId);
       await replayAgent.connectAgent({ runId: "replay-run" });
-      expect(replayAgent.messages.find((message) => message.id === "custom-user")).toEqual(
-        customMessages[0],
-      );
+      expect(
+        replayAgent.messages.find((message) => message.id === "custom-user"),
+      ).toEqual(customMessages[0]);
     });
   });
 

package/src/__tests__/sqlite-runner.test.ts

@@ -48,8 +48,10 @@ class MockAgent extends AbstractAgent {
       await callbacks.onEvent({ event });
     }
 
-    const hasTerminalEvent = this.events.some((event) =>
-      event.type === EventType.RUN_FINISHED || event.type === EventType.RUN_ERROR,
+    const hasTerminalEvent = this.events.some(
+      (event) =>
+        event.type === EventType.RUN_FINISHED ||
+        event.type === EventType.RUN_ERROR,
     );
 
     if (!hasTerminalEvent) {
@@ -84,10 +86,7 @@ class StoppableAgent extends AbstractAgent {
     this.eventDelay = eventDelay;
   }
 
-  async runAgent(
-    input: RunAgentInput,
-    callbacks: RunCallbacks,
-  ): Promise<void> {
+  async runAgent(input: RunAgentInput, callbacks: RunCallbacks): Promise<void> {
     this.shouldStop = false;
     let counter = 0;
 
@@ -123,10 +122,7 @@ class StoppableAgent extends AbstractAgent {
 class OpenEventsAgent extends AbstractAgent {
   private shouldStop = false;
 
-  async runAgent(
-    input: RunAgentInput,
-    callbacks: RunCallbacks,
-  ): Promise<void> {
+  async runAgent(input: RunAgentInput, callbacks: RunCallbacks): Promise<void> {
     this.shouldStop = false;
     const messageId = "open-message";
     const toolCallId = "open-tool";
@@ -182,6 +178,8 @@ describe("SqliteAgentRunner", () => {
   });
 
   afterEach(() => {
+    runner.close();
+
     if (fs.existsSync(dbPath)) fs.unlinkSync(dbPath);
     if (fs.existsSync(tempDir)) fs.rmdirSync(tempDir);
   });
@@ -189,10 +187,25 @@ describe("SqliteAgentRunner", () => {
   it("emits RUN_STARTED and agent events", async () => {
     const threadId = "sqlite-basic";
     const agent = new MockAgent([
-      { type: EventType.TEXT_MESSAGE_START, messageId: "msg-1", role: "assistant" } as TextMessageStartEvent,
-      { type: EventType.TEXT_MESSAGE_CONTENT, messageId: "msg-1", delta: "Hello" } as TextMessageContentEvent,
-      { type: EventType.TEXT_MESSAGE_END, messageId: "msg-1" } as TextMessageEndEvent,
-      { type: EventType.RUN_FINISHED, threadId, runId: "run-1" } as RunFinishedEvent,
+      {
+        type: EventType.TEXT_MESSAGE_START,
+        messageId: "msg-1",
+        role: "assistant",
+      } as TextMessageStartEvent,
+      {
+        type: EventType.TEXT_MESSAGE_CONTENT,
+        messageId: "msg-1",
+        delta: "Hello",
+      } as TextMessageContentEvent,
+      {
+        type: EventType.TEXT_MESSAGE_END,
+        messageId: "msg-1",
+      } as TextMessageEndEvent,
+      {
+        type: EventType.RUN_FINISHED,
+        threadId,
+        runId: "run-1",
+      } as RunFinishedEvent,
     ]);
 
     const events = await firstValueFrom(
@@ -228,7 +241,11 @@ describe("SqliteAgentRunner", () => {
         .pipe(toArray()),
     );
 
-    const newMessage: Message = { id: "new", role: "user", content: "follow up" };
+    const newMessage: Message = {
+      id: "new",
+      role: "user",
+      content: "follow up",
+    };
 
     const secondRun = await firstValueFrom(
       runner
@@ -250,18 +267,26 @@ describe("SqliteAgentRunner", () => {
 
     const db = new Database(dbPath);
     const rows = db
-      .prepare("SELECT events FROM agent_runs WHERE thread_id = ? ORDER BY created_at")
-      .all(threadId) as { events: string }[];
+      .prepare(
+        "SELECT events FROM agent_runs WHERE thread_id = ? ORDER BY created_at",
+      )
+      .all(threadId) as {
+      events: string;
+    }[];
     db.close();
 
     expect(rows).toHaveLength(2);
     const run1Stored = JSON.parse(rows[0].events) as BaseEvent[];
     const run2Stored = JSON.parse(rows[1].events) as BaseEvent[];
 
-    const run1Started = run1Stored.find((event) => event.type === EventType.RUN_STARTED) as RunStartedEvent;
+    const run1Started = run1Stored.find(
+      (event) => event.type === EventType.RUN_STARTED,
+    ) as RunStartedEvent;
     expect(run1Started.input?.messages?.map((m) => m.id)).toEqual(["existing"]);
 
-    const run2Started = run2Stored.find((event) => event.type === EventType.RUN_STARTED) as RunStartedEvent;
+    const run2Started = run2Stored.find(
+      (event) => event.type === EventType.RUN_STARTED,
+    ) as RunStartedEvent;
     expect(run2Started.input?.messages?.map((m) => m.id)).toEqual(["new"]);
   });
 
@@ -317,10 +342,25 @@ describe("SqliteAgentRunner", () => {
   it("persists events across runner instances", async () => {
     const threadId = "sqlite-persist";
     const agent = new MockAgent([
-      { type: EventType.TEXT_MESSAGE_START, messageId: "msg", role: "assistant" } as TextMessageStartEvent,
-      { type: EventType.TEXT_MESSAGE_CONTENT, messageId: "msg", delta: "hi" } as TextMessageContentEvent,
-      { type: EventType.TEXT_MESSAGE_END, messageId: "msg" } as TextMessageEndEvent,
-      { type: EventType.RUN_FINISHED, threadId, runId: "run-1" } as RunFinishedEvent,
+      {
+        type: EventType.TEXT_MESSAGE_START,
+        messageId: "msg",
+        role: "assistant",
+      } as TextMessageStartEvent,
+      {
+        type: EventType.TEXT_MESSAGE_CONTENT,
+        messageId: "msg",
+        delta: "hi",
+      } as TextMessageContentEvent,
+      {
+        type: EventType.TEXT_MESSAGE_END,
+        messageId: "msg",
+      } as TextMessageEndEvent,
+      {
+        type: EventType.RUN_FINISHED,
+        threadId,
+        runId: "run-1",
+      } as RunFinishedEvent,
     ]);
 
     await firstValueFrom(
@@ -334,19 +374,27 @@ describe("SqliteAgentRunner", () => {
     );
 
     const newRunner = new SqliteAgentRunner({ dbPath });
-    const replayed = await firstValueFrom(newRunner.connect({ threadId }).pipe(toArray()));
-
-    expect(replayed[0].type).toBe(EventType.RUN_STARTED);
-    expect(replayed.slice(1).map((event) => event.type)).toEqual([
-      EventType.TEXT_MESSAGE_START,
-      EventType.TEXT_MESSAGE_CONTENT,
-      EventType.TEXT_MESSAGE_END,
-      EventType.RUN_FINISHED,
-    ]);
+    try {
+      const replayed = await firstValueFrom(
+        newRunner.connect({ threadId }).pipe(toArray()),
+      );
+
+      expect(replayed[0].type).toBe(EventType.RUN_STARTED);
+      expect(replayed.slice(1).map((event) => event.type)).toEqual([
+        EventType.TEXT_MESSAGE_START,
+        EventType.TEXT_MESSAGE_CONTENT,
+        EventType.TEXT_MESSAGE_END,
+        EventType.RUN_FINISHED,
+      ]);
+    } finally {
+      newRunner.close();
+    }
   });
 
   it("returns false when stopping a thread that is not running", async () => {
-    await expect(runner.stop({ threadId: "sqlite-missing" })).resolves.toBe(false);
+    await expect(runner.stop({ threadId: "sqlite-missing" })).resolves.toBe(
+      false,
+    );
   });
 
   it("stops an active run and completes observables", async () => {

package/src/sqlite-runner.ts

@@ -51,20 +51,20 @@ export class SqliteAgentRunner extends AgentRunner {
   constructor(options: SqliteAgentRunnerOptions = {}) {
     super();
     const dbPath = options.dbPath ?? ":memory:";
-    
+
     if (!Database) {
       throw new Error(
-        'better-sqlite3 is required for SqliteAgentRunner but was not found.\n' +
-        'Please 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\n' +
-        'If you don\'t need persistence, use InMemoryAgentRunner instead.'
+        "better-sqlite3 is required for SqliteAgentRunner but was not found.\n" +
+          "Please 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\n" +
+          "If you don't need persistence, use InMemoryAgentRunner instead.",
       );
     }
-    
+
     this.db = new Database(dbPath);
     this.initializeSchema();
   }
@@ -110,12 +110,16 @@ export class SqliteAgentRunner extends AgentRunner {
 
     // Check and set schema version
     const currentVersion = this.db
-      .prepare("SELECT version FROM schema_version ORDER BY version DESC LIMIT 1")
+      .prepare(
+        "SELECT version FROM schema_version ORDER BY version DESC LIMIT 1",
+      )
       .get() as { version: number } | undefined;
 
     if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {
       this.db
-        .prepare("INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)")
+        .prepare(
+          "INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)",
+        )
         .run(SCHEMA_VERSION, Date.now());
     }
   }
@@ -125,11 +129,11 @@ export class SqliteAgentRunner extends AgentRunner {
     runId: string,
     events: BaseEvent[],
     input: RunAgentInput,
-    parentRunId?: string | null
+    parentRunId?: string | null,
   ): void {
     // Compact ONLY the events from this run
     const compactedEvents = compactEvents(events);
-    
+
     const stmt = this.db.prepare(`
       INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version)
       VALUES (?, ?, ?, ?, ?, ?, ?)
@@ -142,7 +146,7 @@ export class SqliteAgentRunner extends AgentRunner {
       JSON.stringify(compactedEvents), // Store only this run's compacted events
       JSON.stringify(input),
       Date.now(),
-      SCHEMA_VERSION
+      SCHEMA_VERSION,
     );
   }
 
@@ -165,8 +169,8 @@ export class SqliteAgentRunner extends AgentRunner {
     `);
 
     const rows = stmt.all(threadId, threadId) as any[];
-    
-    return rows.map(row => ({
+
+    return rows.map((row) => ({
       id: row.id,
       thread_id: row.thread_id,
       run_id: row.run_id,
@@ -174,7 +178,7 @@ export class SqliteAgentRunner extends AgentRunner {
       events: JSON.parse(row.events),
       input: JSON.parse(row.input),
       created_at: row.created_at,
-      version: row.version
+      version: row.version,
     }));
   }
 
@@ -190,7 +194,11 @@ export class SqliteAgentRunner extends AgentRunner {
     return result?.run_id ?? null;
   }
 
-  private setRunState(threadId: string, isRunning: boolean, runId?: string): void {
+  private setRunState(
+    threadId: string,
+    isRunning: boolean,
+    runId?: string,
+  ): void {
     const stmt = this.db.prepare(`
       INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at)
       VALUES (?, ?, ?, ?)
@@ -198,15 +206,20 @@ export class SqliteAgentRunner extends AgentRunner {
     stmt.run(threadId, isRunning ? 1 : 0, runId ?? null, Date.now());
   }
 
-  private getRunState(threadId: string): { isRunning: boolean; currentRunId: string | null } {
+  private getRunState(threadId: string): {
+    isRunning: boolean;
+    currentRunId: string | null;
+  } {
     const stmt = this.db.prepare(`
       SELECT is_running, current_run_id FROM run_state WHERE thread_id = ?
     `);
-    const result = stmt.get(threadId) as { is_running: number; current_run_id: string | null } | undefined;
-    
+    const result = stmt.get(threadId) as
+      | { is_running: number; current_run_id: string | null }
+      | undefined;
+
     return {
       isRunning: result?.is_running === 1,
-      currentRunId: result?.current_run_id ?? null
+      currentRunId: result?.current_run_id ?? null,
     };
   }
 
@@ -223,13 +236,13 @@ export class SqliteAgentRunner extends AgentRunner {
     // Track seen message IDs and current run events in memory for this run
     const seenMessageIds = new Set<string>();
     const currentRunEvents: BaseEvent[] = [];
-    
+
     // Get all previously seen message IDs from historic runs
     const historicRuns = this.getHistoricRuns(request.threadId);
     const historicMessageIds = new Set<string>();
     for (const run of historicRuns) {
       for (const event of run.events) {
-        if ('messageId' in event && typeof event.messageId === 'string') {
+        if ("messageId" in event && typeof event.messageId === "string") {
           historicMessageIds.add(event.messageId);
         }
         if (event.type === EventType.RUN_STARTED) {
@@ -246,7 +259,7 @@ export class SqliteAgentRunner extends AgentRunner {
     const nextSubject = new ReplaySubject<BaseEvent>(Infinity);
     const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);
     const prevSubject = prevConnection?.subject;
-    
+
     // Create a subject for run() return value
     const runSubject = new ReplaySubject<BaseEvent>(Infinity);
 
@@ -263,7 +276,7 @@ export class SqliteAgentRunner extends AgentRunner {
     const runAgent = async () => {
       // Get parent run ID for chaining
       const parentRunId = this.getLatestRunId(request.threadId);
-      
+
       try {
         await request.agent.runAgent(request.input, {
           onEvent: ({ event }) => {
@@ -310,7 +323,7 @@ export class SqliteAgentRunner extends AgentRunner {
             }
           },
         });
-        
+
         const connection = ACTIVE_CONNECTIONS.get(request.threadId);
         const appendedEvents = finalizeRunEvents(currentRunEvents, {
           stopRequested: connection?.stopRequested ?? false,
@@ -326,9 +339,9 @@ export class SqliteAgentRunner extends AgentRunner {
           request.input.runId,
           currentRunEvents,
           request.input,
-          parentRunId
+          parentRunId,
         );
-        
+
         // Mark run as complete in database
         this.setRunState(request.threadId, false);
 
@@ -361,10 +374,10 @@ export class SqliteAgentRunner extends AgentRunner {
             request.input.runId,
             currentRunEvents,
             request.input,
-            parentRunId
+            parentRunId,
           );
         }
-        
+
         // Mark run as complete in database
         this.setRunState(request.threadId, false);
 
@@ -407,46 +420,53 @@ export class SqliteAgentRunner extends AgentRunner {
 
     // Load historic runs from database
     const historicRuns = this.getHistoricRuns(request.threadId);
-    
+
     // Collect all historic events from database
     const allHistoricEvents: BaseEvent[] = [];
     for (const run of historicRuns) {
       allHistoricEvents.push(...run.events);
     }
-    
+
     // Compact all events together before emitting
     const compactedEvents = compactEvents(allHistoricEvents);
-    
+
     // Emit compacted events and track message IDs
     const emittedMessageIds = new Set<string>();
     for (const event of compactedEvents) {
       connectionSubject.next(event);
-      if ('messageId' in event && typeof event.messageId === 'string') {
+      if ("messageId" in event && typeof event.messageId === "string") {
         emittedMessageIds.add(event.messageId);
       }
     }
-    
+
     // Bridge active run to connection if exists
     const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);
     const runState = this.getRunState(request.threadId);
 
-    if (activeConnection && (runState.isRunning || activeConnection.stopRequested)) {
+    if (
+      activeConnection &&
+      (runState.isRunning || activeConnection.stopRequested)
+    ) {
       activeConnection.subject.subscribe({
         next: (event) => {
           // Skip message events that we've already emitted from historic
-          if ('messageId' in event && typeof event.messageId === 'string' && emittedMessageIds.has(event.messageId)) {
+          if (
+            "messageId" in event &&
+            typeof event.messageId === "string" &&
+            emittedMessageIds.has(event.messageId)
+          ) {
             return;
           }
           connectionSubject.next(event);
         },
         complete: () => connectionSubject.complete(),
-        error: (err) => connectionSubject.error(err)
+        error: (err) => connectionSubject.error(err),
       });
     } else {
       // No active run, complete after historic events
       connectionSubject.complete();
     }
-    
+
     return connectionSubject.asObservable();
   }
 

package/tsup.config.ts

@@ -1,11 +1,11 @@
-import { defineConfig } from 'tsup';
+import { defineConfig } from "tsup";
 
 export default defineConfig({
-  entry: ['src/index.ts'],
-  format: ['cjs', 'esm'],
+  entry: ["src/index.ts"],
+  format: ["cjs", "esm"],
   dts: true,
   sourcemap: true,
   clean: true,
-  target: 'es2022',
-  outDir: 'dist',
+  target: "es2022",
+  outDir: "dist",
 });