@copilotkitnext/sqlite-runner 0.0.17-alpha.0 → 0.0.18

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
 
-> @copilotkitnext/sqlite-runner@0.0.17-alpha.0 build /Users/brandonmcconnell/Projects/@CopilotKit/vnext_experimental/packages/sqlite-runner
+> @copilotkitnext/sqlite-runner@0.0.18 build /Users/mme/Projects/CopilotKit2/main/packages/sqlite-runner
 > tsup
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.5.0
-CLI Using tsup config: /Users/brandonmcconnell/Projects/@CopilotKit/vnext_experimental/packages/sqlite-runner/tsup.config.ts
+CLI Using tsup config: /Users/mme/Projects/CopilotKit2/main/packages/sqlite-runner/tsup.config.ts
 CLI Target: es2022
 CLI Cleaning output folder
 CJS Build start
 ESM Build start
-ESM dist/index.mjs     9.50 KB
-ESM dist/index.mjs.map 18.71 KB
-ESM ⚡️ Build success in 12ms
-CJS dist/index.js     11.26 KB
-CJS dist/index.js.map 18.80 KB
+ESM dist/index.mjs     11.55 KB
+ESM dist/index.mjs.map 22.22 KB
+ESM ⚡️ Build success in 11ms
+CJS dist/index.js     13.32 KB
+CJS dist/index.js.map 22.31 KB
 CJS ⚡️ Build success in 12ms
 DTS Build start
-DTS ⚡️ Build success in 793ms
-DTS dist/index.d.ts  1010.00 B
-DTS dist/index.d.mts 1010.00 B
+DTS ⚡️ Build success in 1174ms
+DTS dist/index.d.ts  1009.00 B
+DTS dist/index.d.mts 1009.00 B

package/dist/index.d.mts

@@ -17,7 +17,7 @@ declare class SqliteAgentRunner extends AgentRunner {
     run(request: AgentRunnerRunRequest): Observable<BaseEvent>;
     connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;
     isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;
-    stop(_request: AgentRunnerStopRequest): Promise<boolean | undefined>;
+    stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;
     /**
      * Close the database connection (for cleanup)
      */

package/dist/index.d.ts

@@ -17,7 +17,7 @@ declare class SqliteAgentRunner extends AgentRunner {
     run(request: AgentRunnerRunRequest): Observable<BaseEvent>;
     connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;
     isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;
-    stop(_request: AgentRunnerStopRequest): Promise<boolean | undefined>;
+    stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;
     /**
      * Close the database connection (for cleanup)
      */

package/dist/index.js

@@ -188,9 +188,16 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
       }
     }
     const nextSubject = new import_rxjs.ReplaySubject(Infinity);
-    const prevSubject = ACTIVE_CONNECTIONS.get(request.threadId);
-    ACTIVE_CONNECTIONS.set(request.threadId, nextSubject);
+    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);
+    const prevSubject = prevConnection?.subject;
     const runSubject = new import_rxjs.ReplaySubject(Infinity);
+    ACTIVE_CONNECTIONS.set(request.threadId, {
+      subject: nextSubject,
+      agent: request.agent,
+      runSubject,
+      currentEvents: currentRunEvents,
+      stopRequested: false
+    });
     const runAgent = async () => {
       const parentRunId = this.getLatestRunId(request.threadId);
       try {
@@ -232,6 +239,14 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
             }
           }
         });
+        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        const appendedEvents = (0, import_runtime.finalizeRunEvents)(currentRunEvents, {
+          stopRequested: connection?.stopRequested ?? false
+        });
+        for (const event of appendedEvents) {
+          runSubject.next(event);
+          nextSubject.next(event);
+        }
         this.storeRun(
           request.threadId,
           request.input.runId,
@@ -240,9 +255,24 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
           parentRunId
         );
         this.setRunState(request.threadId, false);
+        if (connection) {
+          connection.agent = void 0;
+          connection.runSubject = void 0;
+          connection.currentEvents = void 0;
+          connection.stopRequested = false;
+        }
         runSubject.complete();
         nextSubject.complete();
+        ACTIVE_CONNECTIONS.delete(request.threadId);
       } catch {
+        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        const appendedEvents = (0, import_runtime.finalizeRunEvents)(currentRunEvents, {
+          stopRequested: connection?.stopRequested ?? false
+        });
+        for (const event of appendedEvents) {
+          runSubject.next(event);
+          nextSubject.next(event);
+        }
         if (currentRunEvents.length > 0) {
           this.storeRun(
             request.threadId,
@@ -253,8 +283,15 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
           );
         }
         this.setRunState(request.threadId, false);
+        if (connection) {
+          connection.agent = void 0;
+          connection.runSubject = void 0;
+          connection.currentEvents = void 0;
+          connection.stopRequested = false;
+        }
         runSubject.complete();
         nextSubject.complete();
+        ACTIVE_CONNECTIONS.delete(request.threadId);
       }
     };
     if (prevSubject) {
@@ -283,10 +320,10 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
         emittedMessageIds.add(event.messageId);
       }
     }
-    const activeSubject = ACTIVE_CONNECTIONS.get(request.threadId);
+    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);
     const runState = this.getRunState(request.threadId);
-    if (activeSubject && runState.isRunning) {
-      activeSubject.subscribe({
+    if (activeConnection && (runState.isRunning || activeConnection.stopRequested)) {
+      activeConnection.subject.subscribe({
         next: (event) => {
           if ("messageId" in event && typeof event.messageId === "string" && emittedMessageIds.has(event.messageId)) {
             return;
@@ -305,9 +342,30 @@ var SqliteAgentRunner = class extends import_runtime.AgentRunner {
     const runState = this.getRunState(request.threadId);
     return Promise.resolve(runState.isRunning);
   }
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  stop(_request) {
-    throw new Error("Method not implemented.");
+  stop(request) {
+    const runState = this.getRunState(request.threadId);
+    if (!runState.isRunning) {
+      return Promise.resolve(false);
+    }
+    const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+    const agent = connection?.agent;
+    if (!connection || !agent) {
+      return Promise.resolve(false);
+    }
+    if (connection.stopRequested) {
+      return Promise.resolve(false);
+    }
+    connection.stopRequested = true;
+    this.setRunState(request.threadId, false);
+    try {
+      agent.abortRun();
+      return Promise.resolve(true);
+    } catch (error) {
+      console.error("Failed to abort sqlite agent run", error);
+      connection.stopRequested = false;
+      this.setRunState(request.threadId, true);
+      return Promise.resolve(false);
+    }
   }
   /**
    * Close the database connection (for cleanup)

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  type AgentRunnerConnectRequest,\n  type AgentRunnerIsRunningRequest,\n  type AgentRunnerRunRequest,\n  type AgentRunnerStopRequest,\n} from \"@copilotkitnext/runtime\";\nimport { Observable, ReplaySubject } from \"rxjs\";\nimport {\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\n// Active connections for streaming events\n// This is the only in-memory state we need - just for active streaming\nconst ACTIVE_CONNECTIONS = new Map<string, ReplaySubject<BaseEvent>>();\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 prevSubject = ACTIVE_CONNECTIONS.get(request.threadId);\n    \n    // Update the active connection for this thread\n    ACTIVE_CONNECTIONS.set(request.threadId, nextSubject);\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\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        // 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        // Complete the subjects\n        runSubject.complete();\n        nextSubject.complete();\n      } catch {\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        // 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    };\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 activeSubject = ACTIVE_CONNECTIONS.get(request.threadId);\n    const runState = this.getRunState(request.threadId);\n    \n    if (activeSubject && runState.isRunning) {\n      activeSubject.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  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  stop(_request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    throw new Error(\"Method not implemented.\");\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,qBAMO;AACP,kBAA0C;AAC1C,oBAMO;AACP,4BAAqB;AAErB,IAAM,iBAAiB;AAmBvB,IAAM,qBAAqB,oBAAI,IAAsC;AAE9D,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,cAAc,mBAAmB,IAAI,QAAQ,QAAQ;AAG3D,uBAAmB,IAAI,QAAQ,UAAU,WAAW;AAGpD,UAAM,aAAa,IAAI,0BAAyB,QAAQ;AAGxD,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;AAGD,aAAK;AAAA,UACH,QAAQ;AAAA,UACR,QAAQ,MAAM;AAAA,UACd;AAAA,UACA,QAAQ;AAAA,UACR;AAAA,QACF;AAGA,aAAK,YAAY,QAAQ,UAAU,KAAK;AAGxC,mBAAW,SAAS;AACpB,oBAAY,SAAS;AAAA,MACvB,QAAQ;AAEN,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;AAIxC,mBAAW,SAAS;AACpB,oBAAY,SAAS;AAAA,MACvB;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,gBAAgB,mBAAmB,IAAI,QAAQ,QAAQ;AAC7D,UAAM,WAAW,KAAK,YAAY,QAAQ,QAAQ;AAElD,QAAI,iBAAiB,SAAS,WAAW;AACvC,oBAAc,UAAU;AAAA,QACtB,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;AAAA,EAGA,KAAK,UAAgE;AACnE,UAAM,IAAI,MAAM,yBAAyB;AAAA,EAC3C;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(\"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"]}

package/dist/index.mjs

@@ -1,6 +1,7 @@
 // src/sqlite-runner.ts
 import {
-  AgentRunner
+  AgentRunner,
+  finalizeRunEvents
 } from "@copilotkitnext/runtime";
 import { ReplaySubject } from "rxjs";
 import {
@@ -157,9 +158,16 @@ var SqliteAgentRunner = class extends AgentRunner {
       }
     }
     const nextSubject = new ReplaySubject(Infinity);
-    const prevSubject = ACTIVE_CONNECTIONS.get(request.threadId);
-    ACTIVE_CONNECTIONS.set(request.threadId, nextSubject);
+    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);
+    const prevSubject = prevConnection?.subject;
     const runSubject = new ReplaySubject(Infinity);
+    ACTIVE_CONNECTIONS.set(request.threadId, {
+      subject: nextSubject,
+      agent: request.agent,
+      runSubject,
+      currentEvents: currentRunEvents,
+      stopRequested: false
+    });
     const runAgent = async () => {
       const parentRunId = this.getLatestRunId(request.threadId);
       try {
@@ -201,6 +209,14 @@ var SqliteAgentRunner = class extends AgentRunner {
             }
           }
         });
+        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        const appendedEvents = finalizeRunEvents(currentRunEvents, {
+          stopRequested: connection?.stopRequested ?? false
+        });
+        for (const event of appendedEvents) {
+          runSubject.next(event);
+          nextSubject.next(event);
+        }
         this.storeRun(
           request.threadId,
           request.input.runId,
@@ -209,9 +225,24 @@ var SqliteAgentRunner = class extends AgentRunner {
           parentRunId
         );
         this.setRunState(request.threadId, false);
+        if (connection) {
+          connection.agent = void 0;
+          connection.runSubject = void 0;
+          connection.currentEvents = void 0;
+          connection.stopRequested = false;
+        }
         runSubject.complete();
         nextSubject.complete();
+        ACTIVE_CONNECTIONS.delete(request.threadId);
       } catch {
+        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        const appendedEvents = finalizeRunEvents(currentRunEvents, {
+          stopRequested: connection?.stopRequested ?? false
+        });
+        for (const event of appendedEvents) {
+          runSubject.next(event);
+          nextSubject.next(event);
+        }
         if (currentRunEvents.length > 0) {
           this.storeRun(
             request.threadId,
@@ -222,8 +253,15 @@ var SqliteAgentRunner = class extends AgentRunner {
           );
         }
         this.setRunState(request.threadId, false);
+        if (connection) {
+          connection.agent = void 0;
+          connection.runSubject = void 0;
+          connection.currentEvents = void 0;
+          connection.stopRequested = false;
+        }
         runSubject.complete();
         nextSubject.complete();
+        ACTIVE_CONNECTIONS.delete(request.threadId);
       }
     };
     if (prevSubject) {
@@ -252,10 +290,10 @@ var SqliteAgentRunner = class extends AgentRunner {
         emittedMessageIds.add(event.messageId);
       }
     }
-    const activeSubject = ACTIVE_CONNECTIONS.get(request.threadId);
+    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);
     const runState = this.getRunState(request.threadId);
-    if (activeSubject && runState.isRunning) {
-      activeSubject.subscribe({
+    if (activeConnection && (runState.isRunning || activeConnection.stopRequested)) {
+      activeConnection.subject.subscribe({
         next: (event) => {
           if ("messageId" in event && typeof event.messageId === "string" && emittedMessageIds.has(event.messageId)) {
             return;
@@ -274,9 +312,30 @@ var SqliteAgentRunner = class extends AgentRunner {
     const runState = this.getRunState(request.threadId);
     return Promise.resolve(runState.isRunning);
   }
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  stop(_request) {
-    throw new Error("Method not implemented.");
+  stop(request) {
+    const runState = this.getRunState(request.threadId);
+    if (!runState.isRunning) {
+      return Promise.resolve(false);
+    }
+    const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+    const agent = connection?.agent;
+    if (!connection || !agent) {
+      return Promise.resolve(false);
+    }
+    if (connection.stopRequested) {
+      return Promise.resolve(false);
+    }
+    connection.stopRequested = true;
+    this.setRunState(request.threadId, false);
+    try {
+      agent.abortRun();
+      return Promise.resolve(true);
+    } catch (error) {
+      console.error("Failed to abort sqlite agent run", error);
+      connection.stopRequested = false;
+      this.setRunState(request.threadId, true);
+      return Promise.resolve(false);
+    }
   }
   /**
    * Close the database connection (for cleanup)

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/sqlite-runner.ts"],"sourcesContent":["import {\n  AgentRunner,\n  type AgentRunnerConnectRequest,\n  type AgentRunnerIsRunningRequest,\n  type AgentRunnerRunRequest,\n  type AgentRunnerStopRequest,\n} from \"@copilotkitnext/runtime\";\nimport { Observable, ReplaySubject } from \"rxjs\";\nimport {\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\n// Active connections for streaming events\n// This is the only in-memory state we need - just for active streaming\nconst ACTIVE_CONNECTIONS = new Map<string, ReplaySubject<BaseEvent>>();\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 prevSubject = ACTIVE_CONNECTIONS.get(request.threadId);\n    \n    // Update the active connection for this thread\n    ACTIVE_CONNECTIONS.set(request.threadId, nextSubject);\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\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        // 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        // Complete the subjects\n        runSubject.complete();\n        nextSubject.complete();\n      } catch {\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        // 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    };\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 activeSubject = ACTIVE_CONNECTIONS.get(request.threadId);\n    const runState = this.getRunState(request.threadId);\n    \n    if (activeSubject && runState.isRunning) {\n      activeSubject.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  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  stop(_request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    throw new Error(\"Method not implemented.\");\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,OAKK;AACP,SAAqB,qBAAqB;AAC1C;AAAA,EAGE;AAAA,EAEA;AAAA,OACK;AACP,OAAO,cAAc;AAErB,IAAM,iBAAiB;AAmBvB,IAAM,qBAAqB,oBAAI,IAAsC;AAE9D,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,cAAc,mBAAmB,IAAI,QAAQ,QAAQ;AAG3D,uBAAmB,IAAI,QAAQ,UAAU,WAAW;AAGpD,UAAM,aAAa,IAAI,cAAyB,QAAQ;AAGxD,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;AAGD,aAAK;AAAA,UACH,QAAQ;AAAA,UACR,QAAQ,MAAM;AAAA,UACd;AAAA,UACA,QAAQ;AAAA,UACR;AAAA,QACF;AAGA,aAAK,YAAY,QAAQ,UAAU,KAAK;AAGxC,mBAAW,SAAS;AACpB,oBAAY,SAAS;AAAA,MACvB,QAAQ;AAEN,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;AAIxC,mBAAW,SAAS;AACpB,oBAAY,SAAS;AAAA,MACvB;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,gBAAgB,mBAAmB,IAAI,QAAQ,QAAQ;AAC7D,UAAM,WAAW,KAAK,YAAY,QAAQ,QAAQ;AAElD,QAAI,iBAAiB,SAAS,WAAW;AACvC,oBAAc,UAAU;AAAA,QACtB,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;AAAA,EAGA,KAAK,UAAgE;AACnE,UAAM,IAAI,MAAM,yBAAyB;AAAA,EAC3C;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(\"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":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkitnext/sqlite-runner",
-  "version": "0.0.17-alpha.0",
+  "version": "0.0.18",
   "description": "SQLite-backed agent runner for CopilotKit2",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -28,7 +28,7 @@
   "dependencies": {
     "@ag-ui/client": "0.0.40-alpha.6",
     "rxjs": "7.8.1",
-    "@copilotkitnext/runtime": "0.0.17-alpha.0"
+    "@copilotkitnext/runtime": "0.0.18"
   },
   "peerDependencies": {
     "better-sqlite3": "^12.2.0"

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

@@ -6,6 +6,7 @@ import {
   EventType,
   Message,
   RunAgentInput,
+  RunFinishedEvent,
   RunStartedEvent,
   TextMessageContentEvent,
   TextMessageEndEvent,
@@ -46,6 +47,19 @@ class MockAgent extends AbstractAgent {
     for (const event of this.events) {
       await callbacks.onEvent({ event });
     }
+
+    const hasTerminalEvent = this.events.some((event) =>
+      event.type === EventType.RUN_FINISHED || event.type === EventType.RUN_ERROR,
+    );
+
+    if (!hasTerminalEvent) {
+      const runFinished: RunFinishedEvent = {
+        type: EventType.RUN_FINISHED,
+        threadId: input.threadId,
+        runId: input.runId,
+      };
+      await callbacks.onEvent({ event: runFinished });
+    }
   }
 
   protected run(): ReturnType<AbstractAgent["run"]> {
@@ -61,6 +75,101 @@ class MockAgent extends AbstractAgent {
   }
 }
 
+class StoppableAgent extends AbstractAgent {
+  private shouldStop = false;
+  private eventDelay: number;
+
+  constructor(eventDelay = 5) {
+    super();
+    this.eventDelay = eventDelay;
+  }
+
+  async runAgent(
+    input: RunAgentInput,
+    callbacks: RunCallbacks,
+  ): Promise<void> {
+    this.shouldStop = false;
+    let counter = 0;
+
+    const runStarted: RunStartedEvent = {
+      type: EventType.RUN_STARTED,
+      threadId: input.threadId,
+      runId: input.runId,
+    };
+    await callbacks.onEvent({ event: runStarted });
+    await callbacks.onRunStartedEvent?.();
+
+    while (!this.shouldStop && counter < 10_000) {
+      await new Promise((resolve) => setTimeout(resolve, this.eventDelay));
+      const event: BaseEvent = {
+        type: EventType.TEXT_MESSAGE_CONTENT,
+        messageId: `sqlite-stop-${counter}`,
+        delta: `chunk-${counter}`,
+      } as TextMessageContentEvent;
+      await callbacks.onEvent({ event });
+      counter += 1;
+    }
+  }
+
+  abortRun(): void {
+    this.shouldStop = true;
+  }
+
+  clone(): AbstractAgent {
+    return new StoppableAgent(this.eventDelay);
+  }
+}
+
+class OpenEventsAgent extends AbstractAgent {
+  private shouldStop = false;
+
+  async runAgent(
+    input: RunAgentInput,
+    callbacks: RunCallbacks,
+  ): Promise<void> {
+    this.shouldStop = false;
+    const messageId = "open-message";
+    const toolCallId = "open-tool";
+
+    await callbacks.onEvent({
+      event: {
+        type: EventType.TEXT_MESSAGE_START,
+        messageId,
+        role: "assistant",
+      } as BaseEvent,
+    });
+
+    await callbacks.onEvent({
+      event: {
+        type: EventType.TEXT_MESSAGE_CONTENT,
+        messageId,
+        delta: "Partial content",
+      } as BaseEvent,
+    });
+
+    await callbacks.onEvent({
+      event: {
+        type: EventType.TOOL_CALL_START,
+        toolCallId,
+        toolCallName: "testTool",
+        parentMessageId: messageId,
+      } as BaseEvent,
+    });
+
+    while (!this.shouldStop) {
+      await new Promise((resolve) => setTimeout(resolve, 5));
+    }
+  }
+
+  abortRun(): void {
+    this.shouldStop = true;
+  }
+
+  clone(): AbstractAgent {
+    return new OpenEventsAgent();
+  }
+}
+
 describe("SqliteAgentRunner", () => {
   let tempDir: string;
   let dbPath: string;
@@ -83,6 +192,7 @@ describe("SqliteAgentRunner", () => {
       { 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(
@@ -100,6 +210,7 @@ describe("SqliteAgentRunner", () => {
       EventType.TEXT_MESSAGE_START,
       EventType.TEXT_MESSAGE_CONTENT,
       EventType.TEXT_MESSAGE_END,
+      EventType.RUN_FINISHED,
     ]);
   });
 
@@ -171,6 +282,11 @@ describe("SqliteAgentRunner", () => {
           runId: "run-keep",
           input: providedInput,
         } as RunStartedEvent,
+        {
+          type: EventType.RUN_FINISHED,
+          threadId,
+          runId: "run-keep",
+        } as RunFinishedEvent,
       ],
       false,
     );
@@ -190,7 +306,10 @@ describe("SqliteAgentRunner", () => {
         .pipe(toArray()),
     );
 
-    expect(events).toHaveLength(1);
+    expect(events.map((event) => event.type)).toEqual([
+      EventType.RUN_STARTED,
+      EventType.RUN_FINISHED,
+    ]);
     const runStarted = events[0] as RunStartedEvent;
     expect(runStarted.input).toBe(providedInput);
   });
@@ -201,6 +320,7 @@ describe("SqliteAgentRunner", () => {
       { 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(
@@ -221,6 +341,62 @@ describe("SqliteAgentRunner", () => {
       EventType.TEXT_MESSAGE_START,
       EventType.TEXT_MESSAGE_CONTENT,
       EventType.TEXT_MESSAGE_END,
+      EventType.RUN_FINISHED,
+    ]);
+  });
+
+  it("returns false when stopping a thread that is not running", async () => {
+    await expect(runner.stop({ threadId: "sqlite-missing" })).resolves.toBe(false);
+  });
+
+  it("stops an active run and completes observables", async () => {
+    const threadId = "sqlite-stop";
+    const agent = new StoppableAgent(2);
+    const input: RunAgentInput = {
+      threadId,
+      runId: "sqlite-stop-run",
+      messages: [],
+      state: {},
+    };
+
+    const run$ = runner.run({ threadId, agent, input });
+    const collected = firstValueFrom(run$.pipe(toArray()));
+
+    await new Promise((resolve) => setTimeout(resolve, 20));
+    expect(await runner.isRunning({ threadId })).toBe(true);
+
+    const stopped = await runner.stop({ threadId });
+    expect(stopped).toBe(true);
+
+    const events = await collected;
+    expect(events.length).toBeGreaterThan(0);
+    expect(events[events.length - 1].type).toBe(EventType.RUN_FINISHED);
+    expect(await runner.isRunning({ threadId })).toBe(false);
+  });
+
+  it("closes open text and tool events when stopping", async () => {
+    const threadId = "sqlite-open-events";
+    const agent = new OpenEventsAgent();
+    const input: RunAgentInput = {
+      threadId,
+      runId: "sqlite-open-run",
+      messages: [],
+      state: {},
+    };
+
+    const run$ = runner.run({ threadId, agent, input });
+    const collected = firstValueFrom(run$.pipe(toArray()));
+
+    await new Promise((resolve) => setTimeout(resolve, 20));
+    await runner.stop({ threadId });
+
+    const events = await collected;
+    const endingTypes = events.slice(-4).map((event) => event.type);
+    expect(endingTypes).toEqual([
+      EventType.TEXT_MESSAGE_END,
+      EventType.TOOL_CALL_END,
+      EventType.TOOL_CALL_RESULT,
+      EventType.RUN_FINISHED,
     ]);
   });
 });

package/src/sqlite-runner.ts

@@ -1,5 +1,6 @@
 import {
   AgentRunner,
+  finalizeRunEvents,
   type AgentRunnerConnectRequest,
   type AgentRunnerIsRunningRequest,
   type AgentRunnerRunRequest,
@@ -7,6 +8,7 @@ import {
 } from "@copilotkitnext/runtime";
 import { Observable, ReplaySubject } from "rxjs";
 import {
+  AbstractAgent,
   BaseEvent,
   RunAgentInput,
   EventType,
@@ -32,9 +34,16 @@ export interface SqliteAgentRunnerOptions {
   dbPath?: string;
 }
 
-// Active connections for streaming events
-// This is the only in-memory state we need - just for active streaming
-const ACTIVE_CONNECTIONS = new Map<string, ReplaySubject<BaseEvent>>();
+interface ActiveConnectionContext {
+  subject: ReplaySubject<BaseEvent>;
+  agent?: AbstractAgent;
+  runSubject?: ReplaySubject<BaseEvent>;
+  currentEvents?: BaseEvent[];
+  stopRequested?: boolean;
+}
+
+// Active connections for streaming events and stop support
+const ACTIVE_CONNECTIONS = new Map<string, ActiveConnectionContext>();
 
 export class SqliteAgentRunner extends AgentRunner {
   private db: any;
@@ -235,14 +244,21 @@ export class SqliteAgentRunner extends AgentRunner {
 
     // Get or create subject for this thread's connections
     const nextSubject = new ReplaySubject<BaseEvent>(Infinity);
-    const prevSubject = ACTIVE_CONNECTIONS.get(request.threadId);
+    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);
+    const prevSubject = prevConnection?.subject;
     
-    // Update the active connection for this thread
-    ACTIVE_CONNECTIONS.set(request.threadId, nextSubject);
-
     // Create a subject for run() return value
     const runSubject = new ReplaySubject<BaseEvent>(Infinity);
 
+    // Update the active connection for this thread
+    ACTIVE_CONNECTIONS.set(request.threadId, {
+      subject: nextSubject,
+      agent: request.agent,
+      runSubject,
+      currentEvents: currentRunEvents,
+      stopRequested: false,
+    });
+
     // Helper function to run the agent and handle errors
     const runAgent = async () => {
       // Get parent run ID for chaining
@@ -295,6 +311,15 @@ export class SqliteAgentRunner extends AgentRunner {
           },
         });
         
+        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        const appendedEvents = finalizeRunEvents(currentRunEvents, {
+          stopRequested: connection?.stopRequested ?? false,
+        });
+        for (const event of appendedEvents) {
+          runSubject.next(event);
+          nextSubject.next(event);
+        }
+
         // Store the run in database
         this.storeRun(
           request.threadId,
@@ -306,11 +331,29 @@ export class SqliteAgentRunner extends AgentRunner {
         
         // Mark run as complete in database
         this.setRunState(request.threadId, false);
-        
+
+        if (connection) {
+          connection.agent = undefined;
+          connection.runSubject = undefined;
+          connection.currentEvents = undefined;
+          connection.stopRequested = false;
+        }
+
         // Complete the subjects
         runSubject.complete();
         nextSubject.complete();
+
+        ACTIVE_CONNECTIONS.delete(request.threadId);
       } catch {
+        const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        const appendedEvents = finalizeRunEvents(currentRunEvents, {
+          stopRequested: connection?.stopRequested ?? false,
+        });
+        for (const event of appendedEvents) {
+          runSubject.next(event);
+          nextSubject.next(event);
+        }
+
         // Store the run even if it failed (partial events)
         if (currentRunEvents.length > 0) {
           this.storeRun(
@@ -324,11 +367,20 @@ export class SqliteAgentRunner extends AgentRunner {
         
         // Mark run as complete in database
         this.setRunState(request.threadId, false);
-        
+
+        if (connection) {
+          connection.agent = undefined;
+          connection.runSubject = undefined;
+          connection.currentEvents = undefined;
+          connection.stopRequested = false;
+        }
+
         // Don't emit error to the subject, just complete it
         // This allows subscribers to get events emitted before the error
         runSubject.complete();
         nextSubject.complete();
+
+        ACTIVE_CONNECTIONS.delete(request.threadId);
       }
     };
 
@@ -375,11 +427,11 @@ export class SqliteAgentRunner extends AgentRunner {
     }
     
     // Bridge active run to connection if exists
-    const activeSubject = ACTIVE_CONNECTIONS.get(request.threadId);
+    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);
     const runState = this.getRunState(request.threadId);
-    
-    if (activeSubject && runState.isRunning) {
-      activeSubject.subscribe({
+
+    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)) {
@@ -403,9 +455,35 @@ export class SqliteAgentRunner extends AgentRunner {
     return Promise.resolve(runState.isRunning);
   }
 
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  stop(_request: AgentRunnerStopRequest): Promise<boolean | undefined> {
-    throw new Error("Method not implemented.");
+  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {
+    const runState = this.getRunState(request.threadId);
+    if (!runState.isRunning) {
+      return Promise.resolve(false);
+    }
+
+    const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+    const agent = connection?.agent;
+
+    if (!connection || !agent) {
+      return Promise.resolve(false);
+    }
+
+    if (connection.stopRequested) {
+      return Promise.resolve(false);
+    }
+
+    connection.stopRequested = true;
+    this.setRunState(request.threadId, false);
+
+    try {
+      agent.abortRun();
+      return Promise.resolve(true);
+    } catch (error) {
+      console.error("Failed to abort sqlite agent run", error);
+      connection.stopRequested = false;
+      this.setRunState(request.threadId, true);
+      return Promise.resolve(false);
+    }
   }
 
   /**

package/.turbo/turbo-test.log

@@ -1,61 +0,0 @@
-
-
-> @copilotkitnext/sqlite-runner@0.0.16 test /Users/brandonmcconnell/Projects/@CopilotKit/vnext_experimental/packages/sqlite-runner
-> vitest run
-
-[?25l
- RUN  v3.2.4 /Users/brandonmcconnell/Projects/@CopilotKit/vnext_experimental/packages/sqlite-runner
-
-[?2026h
- ❯ src/__tests__/sqlite-runner.test.ts [queued]
-
- Test Files 0 passed (2)
-      Tests 0 passed (0)
-   Start at 10:10:47
-   Duration 778ms
-[?2026l[?2026h
- ❯ src/__tests__/sqlite-runner.e2e.test.ts [queued]
- ❯ src/__tests__/sqlite-runner.test.ts [queued]
-
- Test Files 0 passed (2)
-      Tests 0 passed (0)
-   Start at 10:10:47
-   Duration 1.02s
-[?2026l[?2026h
- ❯ src/__tests__/sqlite-runner.e2e.test.ts 0/8
- ❯ src/__tests__/sqlite-runner.test.ts [queued]
-
- Test Files 0 passed (2)
-      Tests 0 passed (8)
-   Start at 10:10:47
-   Duration 1.75s
-[?2026l[?2026h ✓ src/__tests__/sqlite-runner.e2e.test.ts (8 tests) 77ms
-
- ❯ src/__tests__/sqlite-runner.test.ts [queued]
-
- Test Files 1 passed (2)
-      Tests 8 passed (8)
-   Start at 10:10:47
-   Duration 1.96s
-[?2026l[?2026h
- ❯ src/__tests__/sqlite-runner.test.ts 0/4
-
- Test Files 1 passed (2)
-      Tests 8 passed (12)
-   Start at 10:10:47
-   Duration 2.17s
-[?2026l[?2026h
- ❯ src/__tests__/sqlite-runner.test.ts 1/4
-
- Test Files 1 passed (2)
-      Tests 9 passed (12)
-   Start at 10:10:47
-   Duration 2.37s
-[?2026l ✓ src/__tests__/sqlite-runner.test.ts (4 tests) 182ms
-
- Test Files  2 passed (2)
-      Tests  12 passed (12)
-   Start at  10:10:47
-   Duration  2.47s (transform 444ms, setup 0ms, collect 2.36s, tests 259ms, environment 1ms, prepare 592ms)
-
-[?25h