@copilotkit/sqlite-runner 1.75.1 → 1.76.0

package/dist/sqlite-runner.cjs

@@ -134,6 +134,21 @@ var SqliteAgentRunner = class extends _copilotkit_runtime_v2.AgentRunner {
 		});
 		const runAgent = async () => {
 			const parentRunId = this.getLatestRunId(request.threadId);
+			const ensureStoppedRunStarted = (stopRequested) => {
+				if (!stopRequested || currentRunEvents.some((event) => event.type === _ag_ui_client.EventType.RUN_STARTED)) return;
+				const started = {
+					type: _ag_ui_client.EventType.RUN_STARTED,
+					threadId: request.threadId,
+					runId: request.input.runId,
+					input: {
+						...request.input,
+						messages: request.input.messages.filter((message) => !historicMessageIds.has(message.id))
+					}
+				};
+				currentRunEvents.unshift(started);
+				runSubject.next(started);
+				nextSubject.next(started);
+			};
 			try {
 				await request.agent.runAgent(request.input, {
 					onEvent: ({ event }) => {
@@ -166,7 +181,11 @@ var SqliteAgentRunner = class extends _copilotkit_runtime_v2.AgentRunner {
 					}
 				});
 				const connection = ACTIVE_CONNECTIONS.get(request.threadId);
-				const appendedEvents = (0, _copilotkit_runtime_v2.finalizeRunEvents)(currentRunEvents, { stopRequested: connection?.stopRequested ?? false });
+				ensureStoppedRunStarted(connection?.stopRequested ?? false);
+				const appendedEvents = (0, _copilotkit_runtime_v2.finalizeRunEvents)(currentRunEvents, {
+					stopRequested: connection?.stopRequested ?? false,
+					protocolVersion: request.input.protocolVersion
+				});
 				for (const event of appendedEvents) {
 					runSubject.next(event);
 					nextSubject.next(event);
@@ -184,7 +203,11 @@ var SqliteAgentRunner = class extends _copilotkit_runtime_v2.AgentRunner {
 				ACTIVE_CONNECTIONS.delete(request.threadId);
 			} catch {
 				const connection = ACTIVE_CONNECTIONS.get(request.threadId);
-				const appendedEvents = (0, _copilotkit_runtime_v2.finalizeRunEvents)(currentRunEvents, { stopRequested: connection?.stopRequested ?? false });
+				ensureStoppedRunStarted(connection?.stopRequested ?? false);
+				const appendedEvents = (0, _copilotkit_runtime_v2.finalizeRunEvents)(currentRunEvents, {
+					stopRequested: connection?.stopRequested ?? false,
+					protocolVersion: request.input.protocolVersion
+				});
 				for (const event of appendedEvents) {
 					runSubject.next(event);
 					nextSubject.next(event);

package/dist/sqlite-runner.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"sqlite-runner.cjs","names":["AgentRunner","Database","EventType","ReplaySubject"],"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 \"@copilotkit/runtime/v2\";\nimport { Observable, ReplaySubject } from \"rxjs\";\nimport {\n  AbstractAgent,\n  BaseEvent,\n  RunAgentInput,\n  EventType,\n  RunStartedEvent,\n  compactEvents,\n} from \"@ag-ui/client\";\nimport Database from \"better-sqlite3\";\n\nconst SCHEMA_VERSION = 1;\n\ninterface AgentRunRecord {\n  id: number;\n  thread_id: string;\n  run_id: string;\n  parent_run_id: string | null;\n  events: BaseEvent[];\n  input: RunAgentInput;\n  created_at: number;\n  version: number;\n}\n\nexport interface SqliteAgentRunnerOptions {\n  dbPath?: string;\n}\n\ninterface ActiveConnectionContext {\n  subject: ReplaySubject<BaseEvent>;\n  agent?: AbstractAgent;\n  runSubject?: ReplaySubject<BaseEvent>;\n  currentEvents?: BaseEvent[];\n  stopRequested?: boolean;\n}\n\n// Active connections for streaming events and stop support\nconst ACTIVE_CONNECTIONS = new Map<string, ActiveConnectionContext>();\n\nexport class SqliteAgentRunner extends AgentRunner {\n  private db: any;\n\n  constructor(options: SqliteAgentRunnerOptions = {}) {\n    super();\n    const dbPath = options.dbPath ?? \":memory:\";\n\n    if (!Database) {\n      throw new Error(\n        \"better-sqlite3 is required for SqliteAgentRunner but was not found.\\n\" +\n          \"Please install it in your project:\\n\" +\n          \"  npm install better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  pnpm add better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  yarn add better-sqlite3\\n\\n\" +\n          \"If you don't need persistence, use InMemoryAgentRunner instead.\",\n      );\n    }\n\n    this.db = new Database(dbPath);\n    this.initializeSchema();\n  }\n\n  private initializeSchema(): void {\n    // Create the agent_runs table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS agent_runs (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        thread_id TEXT NOT NULL,\n        run_id TEXT NOT NULL UNIQUE,\n        parent_run_id TEXT,\n        events TEXT NOT NULL,\n        input TEXT NOT NULL,\n        created_at INTEGER NOT NULL,\n        version INTEGER NOT NULL\n      )\n    `);\n\n    // Create run_state table to track active runs\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS run_state (\n        thread_id TEXT PRIMARY KEY,\n        is_running INTEGER DEFAULT 0,\n        current_run_id TEXT,\n        updated_at INTEGER NOT NULL\n      )\n    `);\n\n    // Create indexes for efficient queries\n    this.db.exec(`\n      CREATE INDEX IF NOT EXISTS idx_thread_id ON agent_runs(thread_id);\n      CREATE INDEX IF NOT EXISTS idx_parent_run_id ON agent_runs(parent_run_id);\n    `);\n\n    // Create schema version table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS schema_version (\n        version INTEGER PRIMARY KEY,\n        applied_at INTEGER NOT NULL\n      )\n    `);\n\n    // Check and set schema version\n    const currentVersion = this.db\n      .prepare(\n        \"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\",\n      )\n      .get() as { version: number } | undefined;\n\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      this.db\n        .prepare(\n          \"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\",\n        )\n        .run(SCHEMA_VERSION, Date.now());\n    }\n  }\n\n  private storeRun(\n    threadId: string,\n    runId: string,\n    events: BaseEvent[],\n    input: RunAgentInput,\n    parentRunId?: string | null,\n  ): void {\n    // Compact ONLY the events from this run\n    const compactedEvents = compactEvents(events);\n\n    const stmt = this.db.prepare(`\n      INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version)\n      VALUES (?, ?, ?, ?, ?, ?, ?)\n    `);\n\n    stmt.run(\n      threadId,\n      runId,\n      parentRunId ?? null,\n      JSON.stringify(compactedEvents), // Store only this run's compacted events\n      JSON.stringify(input),\n      Date.now(),\n      SCHEMA_VERSION,\n    );\n  }\n\n  private getHistoricRuns(threadId: string): AgentRunRecord[] {\n    const stmt = this.db.prepare(`\n      WITH RECURSIVE run_chain AS (\n        -- Base case: find the root runs (those without parent)\n        SELECT * FROM agent_runs \n        WHERE thread_id = ? AND parent_run_id IS NULL\n        \n        UNION ALL\n        \n        -- Recursive case: find children of current level\n        SELECT ar.* FROM agent_runs ar\n        INNER JOIN run_chain rc ON ar.parent_run_id = rc.run_id\n        WHERE ar.thread_id = ?\n      )\n      SELECT * FROM run_chain\n      ORDER BY created_at ASC\n    `);\n\n    const rows = stmt.all(threadId, threadId) as any[];\n\n    return rows.map((row) => ({\n      id: row.id,\n      thread_id: row.thread_id,\n      run_id: row.run_id,\n      parent_run_id: row.parent_run_id,\n      events: JSON.parse(row.events),\n      input: JSON.parse(row.input),\n      created_at: row.created_at,\n      version: row.version,\n    }));\n  }\n\n  private getLatestRunId(threadId: string): string | null {\n    const stmt = this.db.prepare(`\n      SELECT run_id FROM agent_runs \n      WHERE thread_id = ? \n      ORDER BY created_at DESC \n      LIMIT 1\n    `);\n\n    const result = stmt.get(threadId) as { run_id: string } | undefined;\n    return result?.run_id ?? null;\n  }\n\n  private setRunState(\n    threadId: string,\n    isRunning: boolean,\n    runId?: string,\n  ): void {\n    const stmt = this.db.prepare(`\n      INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at)\n      VALUES (?, ?, ?, ?)\n    `);\n    stmt.run(threadId, isRunning ? 1 : 0, runId ?? null, Date.now());\n  }\n\n  private getRunState(threadId: string): {\n    isRunning: boolean;\n    currentRunId: string | null;\n  } {\n    const stmt = this.db.prepare(`\n      SELECT is_running, current_run_id FROM run_state WHERE thread_id = ?\n    `);\n    const result = stmt.get(threadId) as\n      | { is_running: number; current_run_id: string | null }\n      | undefined;\n\n    return {\n      isRunning: result?.is_running === 1,\n      currentRunId: result?.current_run_id ?? null,\n    };\n  }\n\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    // Check if thread is already running in database\n    const runState = this.getRunState(request.threadId);\n    if (runState.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n\n    // Mark thread as running in database\n    this.setRunState(request.threadId, true, request.input.runId);\n\n    // Track seen message IDs and current run events in memory for this run\n    const seenMessageIds = new Set<string>();\n    const currentRunEvents: BaseEvent[] = [];\n\n    // Get all previously seen message IDs from historic runs\n    const historicRuns = this.getHistoricRuns(request.threadId);\n    const historicMessageIds = new Set<string>();\n    for (const run of historicRuns) {\n      for (const event of run.events) {\n        if (\"messageId\" in event && typeof event.messageId === \"string\") {\n          historicMessageIds.add(event.messageId);\n        }\n        if (event.type === EventType.RUN_STARTED) {\n          const runStarted = event as RunStartedEvent;\n          const messages = runStarted.input?.messages ?? [];\n          for (const message of messages) {\n            historicMessageIds.add(message.id);\n          }\n        }\n      }\n    }\n\n    // Get or create subject for this thread's connections\n    const nextSubject = new ReplaySubject<BaseEvent>(Infinity);\n    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const prevSubject = prevConnection?.subject;\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Update the active connection for this thread\n    ACTIVE_CONNECTIONS.set(request.threadId, {\n      subject: nextSubject,\n      agent: request.agent,\n      runSubject,\n      currentEvents: currentRunEvents,\n      stopRequested: false,\n    });\n\n    // Helper function to run the agent and handle errors\n    const runAgent = async () => {\n      // Get parent run ID for chaining\n      const parentRunId = this.getLatestRunId(request.threadId);\n\n      try {\n        await request.agent.runAgent(request.input, {\n          onEvent: ({ event }) => {\n            let processedEvent: BaseEvent = event;\n            if (event.type === EventType.RUN_STARTED) {\n              const runStartedEvent = event as RunStartedEvent;\n              if (!runStartedEvent.input) {\n                const sanitizedMessages = request.input.messages\n                  ? request.input.messages.filter(\n                      (message) => !historicMessageIds.has(message.id),\n                    )\n                  : undefined;\n                const updatedInput = {\n                  ...request.input,\n                  ...(sanitizedMessages !== undefined\n                    ? { messages: sanitizedMessages }\n                    : {}),\n                };\n                processedEvent = {\n                  ...runStartedEvent,\n                  input: updatedInput,\n                } as RunStartedEvent;\n              }\n            }\n\n            runSubject.next(processedEvent); // For run() return - only agent events\n            nextSubject.next(processedEvent); // For connect() / store - all events\n            currentRunEvents.push(processedEvent); // Accumulate for database storage\n          },\n          onNewMessage: ({ message }) => {\n            // Called for each new message\n            if (!seenMessageIds.has(message.id)) {\n              seenMessageIds.add(message.id);\n            }\n          },\n          onRunStartedEvent: () => {\n            // Mark input messages as seen without emitting duplicates\n            if (request.input.messages) {\n              for (const message of request.input.messages) {\n                if (!seenMessageIds.has(message.id)) {\n                  seenMessageIds.add(message.id);\n                }\n              }\n            }\n          },\n        });\n\n        const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run in database\n        this.storeRun(\n          request.threadId,\n          request.input.runId,\n          currentRunEvents,\n          request.input,\n          parentRunId,\n        );\n\n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Complete the subjects\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      } catch {\n        const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run even if it failed (partial events)\n        if (currentRunEvents.length > 0) {\n          this.storeRun(\n            request.threadId,\n            request.input.runId,\n            currentRunEvents,\n            request.input,\n            parentRunId,\n          );\n        }\n\n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Don't emit error to the subject, just complete it\n        // This allows subscribers to get events emitted before the error\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      }\n    };\n\n    // Bridge previous events if they exist\n    if (prevSubject) {\n      prevSubject.subscribe({\n        next: (e) => nextSubject.next(e),\n        error: (err) => nextSubject.error(err),\n        complete: () => {\n          // Don't complete nextSubject here - it needs to stay open for new events\n        },\n      });\n    }\n\n    // Start the agent execution immediately (not lazily)\n    runAgent();\n\n    // Return the run subject (only agent events, no injected messages)\n    return runSubject.asObservable();\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const connectionSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Load historic runs from database\n    const historicRuns = this.getHistoricRuns(request.threadId);\n\n    // Collect all historic events from database\n    const allHistoricEvents: BaseEvent[] = [];\n    for (const run of historicRuns) {\n      allHistoricEvents.push(...run.events);\n    }\n\n    // Compact all events together before emitting\n    const compactedEvents = compactEvents(allHistoricEvents);\n\n    // Emit compacted events and track message IDs\n    const emittedMessageIds = new Set<string>();\n    for (const event of compactedEvents) {\n      connectionSubject.next(event);\n      if (\"messageId\" in event && typeof event.messageId === \"string\") {\n        emittedMessageIds.add(event.messageId);\n      }\n    }\n\n    // Bridge active run to connection if exists\n    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const runState = this.getRunState(request.threadId);\n\n    if (\n      activeConnection &&\n      (runState.isRunning || activeConnection.stopRequested)\n    ) {\n      activeConnection.subject.subscribe({\n        next: (event) => {\n          // Skip message events that we've already emitted from historic\n          if (\n            \"messageId\" in event &&\n            typeof event.messageId === \"string\" &&\n            emittedMessageIds.has(event.messageId)\n          ) {\n            return;\n          }\n          connectionSubject.next(event);\n        },\n        complete: () => connectionSubject.complete(),\n        error: (err) => connectionSubject.error(err),\n      });\n    } else {\n      // No active run, complete after historic events\n      connectionSubject.complete();\n    }\n\n    return connectionSubject.asObservable();\n  }\n\n  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean> {\n    const runState = this.getRunState(request.threadId);\n    return Promise.resolve(runState.isRunning);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const runState = this.getRunState(request.threadId);\n    if (!runState.isRunning) {\n      return Promise.resolve(false);\n    }\n\n    const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const agent = connection?.agent;\n\n    if (!connection || !agent) {\n      return Promise.resolve(false);\n    }\n\n    if (connection.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    connection.stopRequested = true;\n    this.setRunState(request.threadId, false);\n\n    try {\n      agent.abortRun();\n      return Promise.resolve(true);\n    } catch (error) {\n      console.error(\"Failed to abort sqlite agent run\", error);\n      connection.stopRequested = false;\n      this.setRunState(request.threadId, true);\n      return Promise.resolve(false);\n    }\n  }\n\n  /**\n   * Close the database connection (for cleanup)\n   */\n  close(): void {\n    if (this.db) {\n      this.db.close();\n    }\n  }\n}\n"],"mappings":";;;;;;;;AAmBA,MAAM,iBAAiB;AA0BvB,MAAM,qCAAqB,IAAI,KAAsC;AAErE,IAAa,oBAAb,cAAuCA,mCAAY;CACjD,AAAQ;CAER,YAAY,UAAoC,EAAE,EAAE;AAClD,SAAO;EACP,MAAM,SAAS,QAAQ,UAAU;AAEjC,MAAI,CAACC,uBACH,OAAM,IAAI,MACR,6QAQD;AAGH,OAAK,KAAK,IAAIA,uBAAS,OAAO;AAC9B,OAAK,kBAAkB;;CAGzB,AAAQ,mBAAyB;AAE/B,OAAK,GAAG,KAAK;;;;;;;;;;;MAWX;AAGF,OAAK,GAAG,KAAK;;;;;;;MAOX;AAGF,OAAK,GAAG,KAAK;;;MAGX;AAGF,OAAK,GAAG,KAAK;;;;;MAKX;EAGF,MAAM,iBAAiB,KAAK,GACzB,QACC,mEACD,CACA,KAAK;AAER,MAAI,CAAC,kBAAkB,eAAe,UAAU,eAC9C,MAAK,GACF,QACC,4EACD,CACA,IAAI,gBAAgB,KAAK,KAAK,CAAC;;CAItC,AAAQ,SACN,UACA,OACA,QACA,OACA,aACM;EAEN,MAAM,mDAAgC,OAAO;AAO7C,EALa,KAAK,GAAG,QAAQ;;;MAG3B,CAEG,IACH,UACA,OACA,eAAe,MACf,KAAK,UAAU,gBAAgB,EAC/B,KAAK,UAAU,MAAM,EACrB,KAAK,KAAK,EACV,eACD;;CAGH,AAAQ,gBAAgB,UAAoC;AAoB1D,SAnBa,KAAK,GAAG,QAAQ;;;;;;;;;;;;;;;MAe3B,CAEgB,IAAI,UAAU,SAAS,CAE7B,KAAK,SAAS;GACxB,IAAI,IAAI;GACR,WAAW,IAAI;GACf,QAAQ,IAAI;GACZ,eAAe,IAAI;GACnB,QAAQ,KAAK,MAAM,IAAI,OAAO;GAC9B,OAAO,KAAK,MAAM,IAAI,MAAM;GAC5B,YAAY,IAAI;GAChB,SAAS,IAAI;GACd,EAAE;;CAGL,AAAQ,eAAe,UAAiC;AAStD,SARa,KAAK,GAAG,QAAQ;;;;;MAK3B,CAEkB,IAAI,SAAS,EAClB,UAAU;;CAG3B,AAAQ,YACN,UACA,WACA,OACM;AAKN,EAJa,KAAK,GAAG,QAAQ;;;MAG3B,CACG,IAAI,UAAU,YAAY,IAAI,GAAG,SAAS,MAAM,KAAK,KAAK,CAAC;;CAGlE,AAAQ,YAAY,UAGlB;EAIA,MAAM,SAHO,KAAK,GAAG,QAAQ;;MAE3B,CACkB,IAAI,SAAS;AAIjC,SAAO;GACL,WAAW,QAAQ,eAAe;GAClC,cAAc,QAAQ,kBAAkB;GACzC;;CAGH,IAAI,SAAuD;AAGzD,MADiB,KAAK,YAAY,QAAQ,SAAS,CACtC,UACX,OAAM,IAAI,MAAM,yBAAyB;AAI3C,OAAK,YAAY,QAAQ,UAAU,MAAM,QAAQ,MAAM,MAAM;EAG7D,MAAM,iCAAiB,IAAI,KAAa;EACxC,MAAM,mBAAgC,EAAE;EAGxC,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAC3D,MAAM,qCAAqB,IAAI,KAAa;AAC5C,OAAK,MAAM,OAAO,aAChB,MAAK,MAAM,SAAS,IAAI,QAAQ;AAC9B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,oBAAmB,IAAI,MAAM,UAAU;AAEzC,OAAI,MAAM,SAASC,wBAAU,aAAa;IAExC,MAAM,WADa,MACS,OAAO,YAAY,EAAE;AACjD,SAAK,MAAM,WAAW,SACpB,oBAAmB,IAAI,QAAQ,GAAG;;;EAO1C,MAAM,cAAc,IAAIC,mBAAyB,SAAS;EAE1D,MAAM,cADiB,mBAAmB,IAAI,QAAQ,SAAS,EAC3B;EAGpC,MAAM,aAAa,IAAIA,mBAAyB,SAAS;AAGzD,qBAAmB,IAAI,QAAQ,UAAU;GACvC,SAAS;GACT,OAAO,QAAQ;GACf;GACA,eAAe;GACf,eAAe;GAChB,CAAC;EAGF,MAAM,WAAW,YAAY;GAE3B,MAAM,cAAc,KAAK,eAAe,QAAQ,SAAS;AAEzD,OAAI;AACF,UAAM,QAAQ,MAAM,SAAS,QAAQ,OAAO;KAC1C,UAAU,EAAE,YAAY;MACtB,IAAI,iBAA4B;AAChC,UAAI,MAAM,SAASD,wBAAU,aAAa;OACxC,MAAM,kBAAkB;AACxB,WAAI,CAAC,gBAAgB,OAAO;QAC1B,MAAM,oBAAoB,QAAQ,MAAM,WACpC,QAAQ,MAAM,SAAS,QACpB,YAAY,CAAC,mBAAmB,IAAI,QAAQ,GAAG,CACjD,GACD;QACJ,MAAM,eAAe;SACnB,GAAG,QAAQ;SACX,GAAI,sBAAsB,SACtB,EAAE,UAAU,mBAAmB,GAC/B,EAAE;SACP;AACD,yBAAiB;SACf,GAAG;SACH,OAAO;SACR;;;AAIL,iBAAW,KAAK,eAAe;AAC/B,kBAAY,KAAK,eAAe;AAChC,uBAAiB,KAAK,eAAe;;KAEvC,eAAe,EAAE,cAAc;AAE7B,UAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;KAGlC,yBAAyB;AAEvB,UAAI,QAAQ,MAAM,UAChB;YAAK,MAAM,WAAW,QAAQ,MAAM,SAClC,KAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;;KAKvC,CAAC;IAEF,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;IAC3D,MAAM,+DAAmC,kBAAkB,EACzD,eAAe,YAAY,iBAAiB,OAC7C,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,SAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAGD,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAI7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;WACrC;IACN,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;IAC3D,MAAM,+DAAmC,kBAAkB,EACzD,eAAe,YAAY,iBAAiB,OAC7C,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,iBAAiB,SAAS,EAC5B,MAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAIH,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAK7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;;;AAK/C,MAAI,YACF,aAAY,UAAU;GACpB,OAAO,MAAM,YAAY,KAAK,EAAE;GAChC,QAAQ,QAAQ,YAAY,MAAM,IAAI;GACtC,gBAAgB;GAGjB,CAAC;AAIJ,YAAU;AAGV,SAAO,WAAW,cAAc;;CAGlC,QAAQ,SAA2D;EACjE,MAAM,oBAAoB,IAAIC,mBAAyB,SAAS;EAGhE,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAG3D,MAAM,oBAAiC,EAAE;AACzC,OAAK,MAAM,OAAO,aAChB,mBAAkB,KAAK,GAAG,IAAI,OAAO;EAIvC,MAAM,mDAAgC,kBAAkB;EAGxD,MAAM,oCAAoB,IAAI,KAAa;AAC3C,OAAK,MAAM,SAAS,iBAAiB;AACnC,qBAAkB,KAAK,MAAM;AAC7B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,mBAAkB,IAAI,MAAM,UAAU;;EAK1C,MAAM,mBAAmB,mBAAmB,IAAI,QAAQ,SAAS;EACjE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AAEnD,MACE,qBACC,SAAS,aAAa,iBAAiB,eAExC,kBAAiB,QAAQ,UAAU;GACjC,OAAO,UAAU;AAEf,QACE,eAAe,SACf,OAAO,MAAM,cAAc,YAC3B,kBAAkB,IAAI,MAAM,UAAU,CAEtC;AAEF,sBAAkB,KAAK,MAAM;;GAE/B,gBAAgB,kBAAkB,UAAU;GAC5C,QAAQ,QAAQ,kBAAkB,MAAM,IAAI;GAC7C,CAAC;MAGF,mBAAkB,UAAU;AAG9B,SAAO,kBAAkB,cAAc;;CAGzC,UAAU,SAAwD;EAChE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AACnD,SAAO,QAAQ,QAAQ,SAAS,UAAU;;CAG5C,KAAK,SAA+D;AAElE,MAAI,CADa,KAAK,YAAY,QAAQ,SAAS,CACrC,UACZ,QAAO,QAAQ,QAAQ,MAAM;EAG/B,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;EAC3D,MAAM,QAAQ,YAAY;AAE1B,MAAI,CAAC,cAAc,CAAC,MAClB,QAAO,QAAQ,QAAQ,MAAM;AAG/B,MAAI,WAAW,cACb,QAAO,QAAQ,QAAQ,MAAM;AAG/B,aAAW,gBAAgB;AAC3B,OAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,MAAI;AACF,SAAM,UAAU;AAChB,UAAO,QAAQ,QAAQ,KAAK;WACrB,OAAO;AACd,WAAQ,MAAM,oCAAoC,MAAM;AACxD,cAAW,gBAAgB;AAC3B,QAAK,YAAY,QAAQ,UAAU,KAAK;AACxC,UAAO,QAAQ,QAAQ,MAAM;;;;;;CAOjC,QAAc;AACZ,MAAI,KAAK,GACP,MAAK,GAAG,OAAO"}
+{"version":3,"file":"sqlite-runner.cjs","names":["AgentRunner","Database","EventType","ReplaySubject"],"sources":["../src/sqlite-runner.ts"],"sourcesContent":["import { AgentRunner, finalizeRunEvents } from \"@copilotkit/runtime/v2\";\nimport type {\n  AgentRunnerConnectRequest,\n  AgentRunnerIsRunningRequest,\n  AgentRunnerRunRequest,\n  AgentRunnerStopRequest,\n} from \"@copilotkit/runtime/v2\";\nimport type { Observable } from \"rxjs\";\nimport { ReplaySubject } from \"rxjs\";\nimport type {\n  AbstractAgent,\n  BaseEvent,\n  RunAgentInput,\n  RunStartedEvent,\n} from \"@ag-ui/client\";\nimport { EventType, compactEvents } from \"@ag-ui/client\";\nimport Database from \"better-sqlite3\";\n\nconst SCHEMA_VERSION = 1;\n\ninterface AgentRunRecord {\n  id: number;\n  thread_id: string;\n  run_id: string;\n  parent_run_id: string | null;\n  events: BaseEvent[];\n  input: RunAgentInput;\n  created_at: number;\n  version: number;\n}\n\nexport interface SqliteAgentRunnerOptions {\n  dbPath?: string;\n}\n\ninterface ActiveConnectionContext {\n  subject: ReplaySubject<BaseEvent>;\n  agent?: AbstractAgent;\n  runSubject?: ReplaySubject<BaseEvent>;\n  currentEvents?: BaseEvent[];\n  stopRequested?: boolean;\n}\n\n// Active connections for streaming events and stop support\nconst ACTIVE_CONNECTIONS = new Map<string, ActiveConnectionContext>();\n\nexport class SqliteAgentRunner extends AgentRunner {\n  private db: any;\n\n  constructor(options: SqliteAgentRunnerOptions = {}) {\n    super();\n    const dbPath = options.dbPath ?? \":memory:\";\n\n    if (!Database) {\n      throw new Error(\n        \"better-sqlite3 is required for SqliteAgentRunner but was not found.\\n\" +\n          \"Please install it in your project:\\n\" +\n          \"  npm install better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  pnpm add better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  yarn add better-sqlite3\\n\\n\" +\n          \"If you don't need persistence, use InMemoryAgentRunner instead.\",\n      );\n    }\n\n    this.db = new Database(dbPath);\n    this.initializeSchema();\n  }\n\n  private initializeSchema(): void {\n    // Create the agent_runs table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS agent_runs (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        thread_id TEXT NOT NULL,\n        run_id TEXT NOT NULL UNIQUE,\n        parent_run_id TEXT,\n        events TEXT NOT NULL,\n        input TEXT NOT NULL,\n        created_at INTEGER NOT NULL,\n        version INTEGER NOT NULL\n      )\n    `);\n\n    // Create run_state table to track active runs\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS run_state (\n        thread_id TEXT PRIMARY KEY,\n        is_running INTEGER DEFAULT 0,\n        current_run_id TEXT,\n        updated_at INTEGER NOT NULL\n      )\n    `);\n\n    // Create indexes for efficient queries\n    this.db.exec(`\n      CREATE INDEX IF NOT EXISTS idx_thread_id ON agent_runs(thread_id);\n      CREATE INDEX IF NOT EXISTS idx_parent_run_id ON agent_runs(parent_run_id);\n    `);\n\n    // Create schema version table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS schema_version (\n        version INTEGER PRIMARY KEY,\n        applied_at INTEGER NOT NULL\n      )\n    `);\n\n    // Check and set schema version\n    const currentVersion = this.db\n      .prepare(\n        \"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\",\n      )\n      .get() as { version: number } | undefined;\n\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      this.db\n        .prepare(\n          \"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\",\n        )\n        .run(SCHEMA_VERSION, Date.now());\n    }\n  }\n\n  private storeRun(\n    threadId: string,\n    runId: string,\n    events: BaseEvent[],\n    input: RunAgentInput,\n    parentRunId?: string | null,\n  ): void {\n    // Compact ONLY the events from this run\n    const compactedEvents = compactEvents(events);\n\n    const stmt = this.db.prepare(`\n      INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version)\n      VALUES (?, ?, ?, ?, ?, ?, ?)\n    `);\n\n    stmt.run(\n      threadId,\n      runId,\n      parentRunId ?? null,\n      JSON.stringify(compactedEvents), // Store only this run's compacted events\n      JSON.stringify(input),\n      Date.now(),\n      SCHEMA_VERSION,\n    );\n  }\n\n  private getHistoricRuns(threadId: string): AgentRunRecord[] {\n    const stmt = this.db.prepare(`\n      WITH RECURSIVE run_chain AS (\n        -- Base case: find the root runs (those without parent)\n        SELECT * FROM agent_runs \n        WHERE thread_id = ? AND parent_run_id IS NULL\n        \n        UNION ALL\n        \n        -- Recursive case: find children of current level\n        SELECT ar.* FROM agent_runs ar\n        INNER JOIN run_chain rc ON ar.parent_run_id = rc.run_id\n        WHERE ar.thread_id = ?\n      )\n      SELECT * FROM run_chain\n      ORDER BY created_at ASC\n    `);\n\n    const rows = stmt.all(threadId, threadId) as any[];\n\n    return rows.map((row) => ({\n      id: row.id,\n      thread_id: row.thread_id,\n      run_id: row.run_id,\n      parent_run_id: row.parent_run_id,\n      events: JSON.parse(row.events),\n      input: JSON.parse(row.input),\n      created_at: row.created_at,\n      version: row.version,\n    }));\n  }\n\n  private getLatestRunId(threadId: string): string | null {\n    const stmt = this.db.prepare(`\n      SELECT run_id FROM agent_runs \n      WHERE thread_id = ? \n      ORDER BY created_at DESC \n      LIMIT 1\n    `);\n\n    const result = stmt.get(threadId) as { run_id: string } | undefined;\n    return result?.run_id ?? null;\n  }\n\n  private setRunState(\n    threadId: string,\n    isRunning: boolean,\n    runId?: string,\n  ): void {\n    const stmt = this.db.prepare(`\n      INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at)\n      VALUES (?, ?, ?, ?)\n    `);\n    stmt.run(threadId, isRunning ? 1 : 0, runId ?? null, Date.now());\n  }\n\n  private getRunState(threadId: string): {\n    isRunning: boolean;\n    currentRunId: string | null;\n  } {\n    const stmt = this.db.prepare(`\n      SELECT is_running, current_run_id FROM run_state WHERE thread_id = ?\n    `);\n    const result = stmt.get(threadId) as\n      | { is_running: number; current_run_id: string | null }\n      | undefined;\n\n    return {\n      isRunning: result?.is_running === 1,\n      currentRunId: result?.current_run_id ?? null,\n    };\n  }\n\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    // Check if thread is already running in database\n    const runState = this.getRunState(request.threadId);\n    if (runState.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n\n    // Mark thread as running in database\n    this.setRunState(request.threadId, true, request.input.runId);\n\n    // Track seen message IDs and current run events in memory for this run\n    const seenMessageIds = new Set<string>();\n    const currentRunEvents: BaseEvent[] = [];\n\n    // Get all previously seen message IDs from historic runs\n    const historicRuns = this.getHistoricRuns(request.threadId);\n    const historicMessageIds = new Set<string>();\n    for (const run of historicRuns) {\n      for (const event of run.events) {\n        if (\"messageId\" in event && typeof event.messageId === \"string\") {\n          historicMessageIds.add(event.messageId);\n        }\n        if (event.type === EventType.RUN_STARTED) {\n          const runStarted = event as RunStartedEvent;\n          const messages = runStarted.input?.messages ?? [];\n          for (const message of messages) {\n            historicMessageIds.add(message.id);\n          }\n        }\n      }\n    }\n\n    // Get or create subject for this thread's connections\n    const nextSubject = new ReplaySubject<BaseEvent>(Infinity);\n    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const prevSubject = prevConnection?.subject;\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Update the active connection for this thread\n    ACTIVE_CONNECTIONS.set(request.threadId, {\n      subject: nextSubject,\n      agent: request.agent,\n      runSubject,\n      currentEvents: currentRunEvents,\n      stopRequested: false,\n    });\n\n    // Helper function to run the agent and handle errors\n    const runAgent = async () => {\n      // Get parent run ID for chaining\n      const parentRunId = this.getLatestRunId(request.threadId);\n\n      const ensureStoppedRunStarted = (stopRequested: boolean) => {\n        if (\n          !stopRequested ||\n          currentRunEvents.some((event) => event.type === EventType.RUN_STARTED)\n        ) {\n          return;\n        }\n        // Stop can arrive before the agent sends its first event.\n        const started: RunStartedEvent = {\n          type: EventType.RUN_STARTED,\n          threadId: request.threadId,\n          runId: request.input.runId,\n          input: {\n            ...request.input,\n            messages: request.input.messages.filter(\n              (message) => !historicMessageIds.has(message.id),\n            ),\n          },\n        };\n        currentRunEvents.unshift(started);\n        runSubject.next(started);\n        nextSubject.next(started);\n      };\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        ensureStoppedRunStarted(connection?.stopRequested ?? false);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n          protocolVersion: request.input.protocolVersion,\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        ensureStoppedRunStarted(connection?.stopRequested ?? false);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n          protocolVersion: request.input.protocolVersion,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run even if it failed (partial events)\n        if (currentRunEvents.length > 0) {\n          this.storeRun(\n            request.threadId,\n            request.input.runId,\n            currentRunEvents,\n            request.input,\n            parentRunId,\n          );\n        }\n\n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Don't emit error to the subject, just complete it\n        // This allows subscribers to get events emitted before the error\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      }\n    };\n\n    // Bridge previous events if they exist\n    if (prevSubject) {\n      prevSubject.subscribe({\n        next: (e) => nextSubject.next(e),\n        error: (err) => nextSubject.error(err),\n        complete: () => {\n          // Don't complete nextSubject here - it needs to stay open for new events\n        },\n      });\n    }\n\n    // Start the agent execution immediately (not lazily)\n    runAgent();\n\n    // Return the run subject (only agent events, no injected messages)\n    return runSubject.asObservable();\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const connectionSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Load historic runs from database\n    const historicRuns = this.getHistoricRuns(request.threadId);\n\n    // Collect all historic events from database\n    const allHistoricEvents: BaseEvent[] = [];\n    for (const run of historicRuns) {\n      allHistoricEvents.push(...run.events);\n    }\n\n    // Compact all events together before emitting\n    const compactedEvents = compactEvents(allHistoricEvents);\n\n    // Emit compacted events and track message IDs\n    const emittedMessageIds = new Set<string>();\n    for (const event of compactedEvents) {\n      connectionSubject.next(event);\n      if (\"messageId\" in event && typeof event.messageId === \"string\") {\n        emittedMessageIds.add(event.messageId);\n      }\n    }\n\n    // Bridge active run to connection if exists\n    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const runState = this.getRunState(request.threadId);\n\n    if (\n      activeConnection &&\n      (runState.isRunning || activeConnection.stopRequested)\n    ) {\n      activeConnection.subject.subscribe({\n        next: (event) => {\n          // Skip message events that we've already emitted from historic\n          if (\n            \"messageId\" in event &&\n            typeof event.messageId === \"string\" &&\n            emittedMessageIds.has(event.messageId)\n          ) {\n            return;\n          }\n          connectionSubject.next(event);\n        },\n        complete: () => connectionSubject.complete(),\n        error: (err) => connectionSubject.error(err),\n      });\n    } else {\n      // No active run, complete after historic events\n      connectionSubject.complete();\n    }\n\n    return connectionSubject.asObservable();\n  }\n\n  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean> {\n    const runState = this.getRunState(request.threadId);\n    return Promise.resolve(runState.isRunning);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const runState = this.getRunState(request.threadId);\n    if (!runState.isRunning) {\n      return Promise.resolve(false);\n    }\n\n    const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const agent = connection?.agent;\n\n    if (!connection || !agent) {\n      return Promise.resolve(false);\n    }\n\n    if (connection.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    connection.stopRequested = true;\n    this.setRunState(request.threadId, false);\n\n    try {\n      agent.abortRun();\n      return Promise.resolve(true);\n    } catch (error) {\n      console.error(\"Failed to abort sqlite agent run\", error);\n      connection.stopRequested = false;\n      this.setRunState(request.threadId, true);\n      return Promise.resolve(false);\n    }\n  }\n\n  /**\n   * Close the database connection (for cleanup)\n   */\n  close(): void {\n    if (this.db) {\n      this.db.close();\n    }\n  }\n}\n"],"mappings":";;;;;;;;AAkBA,MAAM,iBAAiB;AA0BvB,MAAM,qCAAqB,IAAI,KAAsC;AAErE,IAAa,oBAAb,cAAuCA,mCAAY;CACjD,AAAQ;CAER,YAAY,UAAoC,EAAE,EAAE;AAClD,SAAO;EACP,MAAM,SAAS,QAAQ,UAAU;AAEjC,MAAI,CAACC,uBACH,OAAM,IAAI,MACR,6QAQD;AAGH,OAAK,KAAK,IAAIA,uBAAS,OAAO;AAC9B,OAAK,kBAAkB;;CAGzB,AAAQ,mBAAyB;AAE/B,OAAK,GAAG,KAAK;;;;;;;;;;;MAWX;AAGF,OAAK,GAAG,KAAK;;;;;;;MAOX;AAGF,OAAK,GAAG,KAAK;;;MAGX;AAGF,OAAK,GAAG,KAAK;;;;;MAKX;EAGF,MAAM,iBAAiB,KAAK,GACzB,QACC,mEACD,CACA,KAAK;AAER,MAAI,CAAC,kBAAkB,eAAe,UAAU,eAC9C,MAAK,GACF,QACC,4EACD,CACA,IAAI,gBAAgB,KAAK,KAAK,CAAC;;CAItC,AAAQ,SACN,UACA,OACA,QACA,OACA,aACM;EAEN,MAAM,mDAAgC,OAAO;AAO7C,EALa,KAAK,GAAG,QAAQ;;;MAG3B,CAEG,IACH,UACA,OACA,eAAe,MACf,KAAK,UAAU,gBAAgB,EAC/B,KAAK,UAAU,MAAM,EACrB,KAAK,KAAK,EACV,eACD;;CAGH,AAAQ,gBAAgB,UAAoC;AAoB1D,SAnBa,KAAK,GAAG,QAAQ;;;;;;;;;;;;;;;MAe3B,CAEgB,IAAI,UAAU,SAAS,CAE7B,KAAK,SAAS;GACxB,IAAI,IAAI;GACR,WAAW,IAAI;GACf,QAAQ,IAAI;GACZ,eAAe,IAAI;GACnB,QAAQ,KAAK,MAAM,IAAI,OAAO;GAC9B,OAAO,KAAK,MAAM,IAAI,MAAM;GAC5B,YAAY,IAAI;GAChB,SAAS,IAAI;GACd,EAAE;;CAGL,AAAQ,eAAe,UAAiC;AAStD,SARa,KAAK,GAAG,QAAQ;;;;;MAK3B,CAEkB,IAAI,SAAS,EAClB,UAAU;;CAG3B,AAAQ,YACN,UACA,WACA,OACM;AAKN,EAJa,KAAK,GAAG,QAAQ;;;MAG3B,CACG,IAAI,UAAU,YAAY,IAAI,GAAG,SAAS,MAAM,KAAK,KAAK,CAAC;;CAGlE,AAAQ,YAAY,UAGlB;EAIA,MAAM,SAHO,KAAK,GAAG,QAAQ;;MAE3B,CACkB,IAAI,SAAS;AAIjC,SAAO;GACL,WAAW,QAAQ,eAAe;GAClC,cAAc,QAAQ,kBAAkB;GACzC;;CAGH,IAAI,SAAuD;AAGzD,MADiB,KAAK,YAAY,QAAQ,SAAS,CACtC,UACX,OAAM,IAAI,MAAM,yBAAyB;AAI3C,OAAK,YAAY,QAAQ,UAAU,MAAM,QAAQ,MAAM,MAAM;EAG7D,MAAM,iCAAiB,IAAI,KAAa;EACxC,MAAM,mBAAgC,EAAE;EAGxC,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAC3D,MAAM,qCAAqB,IAAI,KAAa;AAC5C,OAAK,MAAM,OAAO,aAChB,MAAK,MAAM,SAAS,IAAI,QAAQ;AAC9B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,oBAAmB,IAAI,MAAM,UAAU;AAEzC,OAAI,MAAM,SAASC,wBAAU,aAAa;IAExC,MAAM,WADa,MACS,OAAO,YAAY,EAAE;AACjD,SAAK,MAAM,WAAW,SACpB,oBAAmB,IAAI,QAAQ,GAAG;;;EAO1C,MAAM,cAAc,IAAIC,mBAAyB,SAAS;EAE1D,MAAM,cADiB,mBAAmB,IAAI,QAAQ,SAAS,EAC3B;EAGpC,MAAM,aAAa,IAAIA,mBAAyB,SAAS;AAGzD,qBAAmB,IAAI,QAAQ,UAAU;GACvC,SAAS;GACT,OAAO,QAAQ;GACf;GACA,eAAe;GACf,eAAe;GAChB,CAAC;EAGF,MAAM,WAAW,YAAY;GAE3B,MAAM,cAAc,KAAK,eAAe,QAAQ,SAAS;GAEzD,MAAM,2BAA2B,kBAA2B;AAC1D,QACE,CAAC,iBACD,iBAAiB,MAAM,UAAU,MAAM,SAASD,wBAAU,YAAY,CAEtE;IAGF,MAAM,UAA2B;KAC/B,MAAMA,wBAAU;KAChB,UAAU,QAAQ;KAClB,OAAO,QAAQ,MAAM;KACrB,OAAO;MACL,GAAG,QAAQ;MACX,UAAU,QAAQ,MAAM,SAAS,QAC9B,YAAY,CAAC,mBAAmB,IAAI,QAAQ,GAAG,CACjD;MACF;KACF;AACD,qBAAiB,QAAQ,QAAQ;AACjC,eAAW,KAAK,QAAQ;AACxB,gBAAY,KAAK,QAAQ;;AAG3B,OAAI;AACF,UAAM,QAAQ,MAAM,SAAS,QAAQ,OAAO;KAC1C,UAAU,EAAE,YAAY;MACtB,IAAI,iBAA4B;AAChC,UAAI,MAAM,SAASA,wBAAU,aAAa;OACxC,MAAM,kBAAkB;AACxB,WAAI,CAAC,gBAAgB,OAAO;QAC1B,MAAM,oBAAoB,QAAQ,MAAM,WACpC,QAAQ,MAAM,SAAS,QACpB,YAAY,CAAC,mBAAmB,IAAI,QAAQ,GAAG,CACjD,GACD;QACJ,MAAM,eAAe;SACnB,GAAG,QAAQ;SACX,GAAI,sBAAsB,SACtB,EAAE,UAAU,mBAAmB,GAC/B,EAAE;SACP;AACD,yBAAiB;SACf,GAAG;SACH,OAAO;SACR;;;AAIL,iBAAW,KAAK,eAAe;AAC/B,kBAAY,KAAK,eAAe;AAChC,uBAAiB,KAAK,eAAe;;KAEvC,eAAe,EAAE,cAAc;AAE7B,UAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;KAGlC,yBAAyB;AAEvB,UAAI,QAAQ,MAAM,UAChB;YAAK,MAAM,WAAW,QAAQ,MAAM,SAClC,KAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;;KAKvC,CAAC;IAEF,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;AAC3D,4BAAwB,YAAY,iBAAiB,MAAM;IAC3D,MAAM,+DAAmC,kBAAkB;KACzD,eAAe,YAAY,iBAAiB;KAC5C,iBAAiB,QAAQ,MAAM;KAChC,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,SAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAGD,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAI7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;WACrC;IACN,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;AAC3D,4BAAwB,YAAY,iBAAiB,MAAM;IAC3D,MAAM,+DAAmC,kBAAkB;KACzD,eAAe,YAAY,iBAAiB;KAC5C,iBAAiB,QAAQ,MAAM;KAChC,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,iBAAiB,SAAS,EAC5B,MAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAIH,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAK7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;;;AAK/C,MAAI,YACF,aAAY,UAAU;GACpB,OAAO,MAAM,YAAY,KAAK,EAAE;GAChC,QAAQ,QAAQ,YAAY,MAAM,IAAI;GACtC,gBAAgB;GAGjB,CAAC;AAIJ,YAAU;AAGV,SAAO,WAAW,cAAc;;CAGlC,QAAQ,SAA2D;EACjE,MAAM,oBAAoB,IAAIC,mBAAyB,SAAS;EAGhE,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAG3D,MAAM,oBAAiC,EAAE;AACzC,OAAK,MAAM,OAAO,aAChB,mBAAkB,KAAK,GAAG,IAAI,OAAO;EAIvC,MAAM,mDAAgC,kBAAkB;EAGxD,MAAM,oCAAoB,IAAI,KAAa;AAC3C,OAAK,MAAM,SAAS,iBAAiB;AACnC,qBAAkB,KAAK,MAAM;AAC7B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,mBAAkB,IAAI,MAAM,UAAU;;EAK1C,MAAM,mBAAmB,mBAAmB,IAAI,QAAQ,SAAS;EACjE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AAEnD,MACE,qBACC,SAAS,aAAa,iBAAiB,eAExC,kBAAiB,QAAQ,UAAU;GACjC,OAAO,UAAU;AAEf,QACE,eAAe,SACf,OAAO,MAAM,cAAc,YAC3B,kBAAkB,IAAI,MAAM,UAAU,CAEtC;AAEF,sBAAkB,KAAK,MAAM;;GAE/B,gBAAgB,kBAAkB,UAAU;GAC5C,QAAQ,QAAQ,kBAAkB,MAAM,IAAI;GAC7C,CAAC;MAGF,mBAAkB,UAAU;AAG9B,SAAO,kBAAkB,cAAc;;CAGzC,UAAU,SAAwD;EAChE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AACnD,SAAO,QAAQ,QAAQ,SAAS,UAAU;;CAG5C,KAAK,SAA+D;AAElE,MAAI,CADa,KAAK,YAAY,QAAQ,SAAS,CACrC,UACZ,QAAO,QAAQ,QAAQ,MAAM;EAG/B,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;EAC3D,MAAM,QAAQ,YAAY;AAE1B,MAAI,CAAC,cAAc,CAAC,MAClB,QAAO,QAAQ,QAAQ,MAAM;AAG/B,MAAI,WAAW,cACb,QAAO,QAAQ,QAAQ,MAAM;AAG/B,aAAW,gBAAgB;AAC3B,OAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,MAAI;AACF,SAAM,UAAU;AAChB,UAAO,QAAQ,QAAQ,KAAK;WACrB,OAAO;AACd,WAAQ,MAAM,oCAAoC,MAAM;AACxD,cAAW,gBAAgB;AAC3B,QAAK,YAAY,QAAQ,UAAU,KAAK;AACxC,UAAO,QAAQ,QAAQ,MAAM;;;;;;CAOjC,QAAc;AACZ,MAAI,KAAK,GACP,MAAK,GAAG,OAAO"}

package/dist/sqlite-runner.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"sqlite-runner.d.cts","names":[],"sources":["../src/sqlite-runner.ts"],"mappings":";;;;;UAgCiB,wBAAA;EACf,MAAA;AAAA;AAAA,cAcW,iBAAA,SAA0B,WAAA;EAAA,QAC7B,EAAA;cAEI,OAAA,GAAS,wBAAA;EAAA,QAqBb,gBAAA;EAAA,QAuDA,QAAA;EAAA,QA0BA,eAAA;EAAA,QAgCA,cAAA;EAAA,QAYA,WAAA;EAAA,QAYA,WAAA;EAiBR,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EAgMhD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAuDxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAKjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;EA5DM;;;EA8F7C,KAAA,CAAA;AAAA"}
+{"version":3,"file":"sqlite-runner.d.cts","names":[],"sources":["../src/sqlite-runner.ts"],"mappings":";;;;;UA+BiB,wBAAA;EACf,MAAA;AAAA;AAAA,cAcW,iBAAA,SAA0B,WAAA;EAAA,QAC7B,EAAA;cAEI,OAAA,GAAS,wBAAA;EAAA,QAqBb,gBAAA;EAAA,QAuDA,QAAA;EAAA,QA0BA,eAAA;EAAA,QAgCA,cAAA;EAAA,QAYA,WAAA;EAAA,QAYA,WAAA;EAiBR,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EA4NhD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAuDxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAKjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;EALpB;;;EAuCnB,KAAA,CAAA;AAAA"}

package/dist/sqlite-runner.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"sqlite-runner.d.mts","names":[],"sources":["../src/sqlite-runner.ts"],"mappings":";;;;;UAgCiB,wBAAA;EACf,MAAA;AAAA;AAAA,cAcW,iBAAA,SAA0B,WAAA;EAAA,QAC7B,EAAA;cAEI,OAAA,GAAS,wBAAA;EAAA,QAqBb,gBAAA;EAAA,QAuDA,QAAA;EAAA,QA0BA,eAAA;EAAA,QAgCA,cAAA;EAAA,QAYA,WAAA;EAAA,QAYA,WAAA;EAiBR,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EAgMhD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAuDxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAKjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;EA5DM;;;EA8F7C,KAAA,CAAA;AAAA"}
+{"version":3,"file":"sqlite-runner.d.mts","names":[],"sources":["../src/sqlite-runner.ts"],"mappings":";;;;;UA+BiB,wBAAA;EACf,MAAA;AAAA;AAAA,cAcW,iBAAA,SAA0B,WAAA;EAAA,QAC7B,EAAA;cAEI,OAAA,GAAS,wBAAA;EAAA,QAqBb,gBAAA;EAAA,QAuDA,QAAA;EAAA,QA0BA,eAAA;EAAA,QAgCA,cAAA;EAAA,QAYA,WAAA;EAAA,QAYA,WAAA;EAiBR,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EA4NhD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAuDxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAKjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;EALpB;;;EAuCnB,KAAA,CAAA;AAAA"}

package/dist/sqlite-runner.mjs

@@ -132,6 +132,21 @@ var SqliteAgentRunner = class extends AgentRunner {
 		});
 		const runAgent = async () => {
 			const parentRunId = this.getLatestRunId(request.threadId);
+			const ensureStoppedRunStarted = (stopRequested) => {
+				if (!stopRequested || currentRunEvents.some((event) => event.type === EventType.RUN_STARTED)) return;
+				const started = {
+					type: EventType.RUN_STARTED,
+					threadId: request.threadId,
+					runId: request.input.runId,
+					input: {
+						...request.input,
+						messages: request.input.messages.filter((message) => !historicMessageIds.has(message.id))
+					}
+				};
+				currentRunEvents.unshift(started);
+				runSubject.next(started);
+				nextSubject.next(started);
+			};
 			try {
 				await request.agent.runAgent(request.input, {
 					onEvent: ({ event }) => {
@@ -164,7 +179,11 @@ var SqliteAgentRunner = class extends AgentRunner {
 					}
 				});
 				const connection = ACTIVE_CONNECTIONS.get(request.threadId);
-				const appendedEvents = finalizeRunEvents(currentRunEvents, { stopRequested: connection?.stopRequested ?? false });
+				ensureStoppedRunStarted(connection?.stopRequested ?? false);
+				const appendedEvents = finalizeRunEvents(currentRunEvents, {
+					stopRequested: connection?.stopRequested ?? false,
+					protocolVersion: request.input.protocolVersion
+				});
 				for (const event of appendedEvents) {
 					runSubject.next(event);
 					nextSubject.next(event);
@@ -182,7 +201,11 @@ var SqliteAgentRunner = class extends AgentRunner {
 				ACTIVE_CONNECTIONS.delete(request.threadId);
 			} catch {
 				const connection = ACTIVE_CONNECTIONS.get(request.threadId);
-				const appendedEvents = finalizeRunEvents(currentRunEvents, { stopRequested: connection?.stopRequested ?? false });
+				ensureStoppedRunStarted(connection?.stopRequested ?? false);
+				const appendedEvents = finalizeRunEvents(currentRunEvents, {
+					stopRequested: connection?.stopRequested ?? false,
+					protocolVersion: request.input.protocolVersion
+				});
 				for (const event of appendedEvents) {
 					runSubject.next(event);
 					nextSubject.next(event);

package/dist/sqlite-runner.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"sqlite-runner.mjs","names":[],"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 \"@copilotkit/runtime/v2\";\nimport { Observable, ReplaySubject } from \"rxjs\";\nimport {\n  AbstractAgent,\n  BaseEvent,\n  RunAgentInput,\n  EventType,\n  RunStartedEvent,\n  compactEvents,\n} from \"@ag-ui/client\";\nimport Database from \"better-sqlite3\";\n\nconst SCHEMA_VERSION = 1;\n\ninterface AgentRunRecord {\n  id: number;\n  thread_id: string;\n  run_id: string;\n  parent_run_id: string | null;\n  events: BaseEvent[];\n  input: RunAgentInput;\n  created_at: number;\n  version: number;\n}\n\nexport interface SqliteAgentRunnerOptions {\n  dbPath?: string;\n}\n\ninterface ActiveConnectionContext {\n  subject: ReplaySubject<BaseEvent>;\n  agent?: AbstractAgent;\n  runSubject?: ReplaySubject<BaseEvent>;\n  currentEvents?: BaseEvent[];\n  stopRequested?: boolean;\n}\n\n// Active connections for streaming events and stop support\nconst ACTIVE_CONNECTIONS = new Map<string, ActiveConnectionContext>();\n\nexport class SqliteAgentRunner extends AgentRunner {\n  private db: any;\n\n  constructor(options: SqliteAgentRunnerOptions = {}) {\n    super();\n    const dbPath = options.dbPath ?? \":memory:\";\n\n    if (!Database) {\n      throw new Error(\n        \"better-sqlite3 is required for SqliteAgentRunner but was not found.\\n\" +\n          \"Please install it in your project:\\n\" +\n          \"  npm install better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  pnpm add better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  yarn add better-sqlite3\\n\\n\" +\n          \"If you don't need persistence, use InMemoryAgentRunner instead.\",\n      );\n    }\n\n    this.db = new Database(dbPath);\n    this.initializeSchema();\n  }\n\n  private initializeSchema(): void {\n    // Create the agent_runs table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS agent_runs (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        thread_id TEXT NOT NULL,\n        run_id TEXT NOT NULL UNIQUE,\n        parent_run_id TEXT,\n        events TEXT NOT NULL,\n        input TEXT NOT NULL,\n        created_at INTEGER NOT NULL,\n        version INTEGER NOT NULL\n      )\n    `);\n\n    // Create run_state table to track active runs\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS run_state (\n        thread_id TEXT PRIMARY KEY,\n        is_running INTEGER DEFAULT 0,\n        current_run_id TEXT,\n        updated_at INTEGER NOT NULL\n      )\n    `);\n\n    // Create indexes for efficient queries\n    this.db.exec(`\n      CREATE INDEX IF NOT EXISTS idx_thread_id ON agent_runs(thread_id);\n      CREATE INDEX IF NOT EXISTS idx_parent_run_id ON agent_runs(parent_run_id);\n    `);\n\n    // Create schema version table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS schema_version (\n        version INTEGER PRIMARY KEY,\n        applied_at INTEGER NOT NULL\n      )\n    `);\n\n    // Check and set schema version\n    const currentVersion = this.db\n      .prepare(\n        \"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\",\n      )\n      .get() as { version: number } | undefined;\n\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      this.db\n        .prepare(\n          \"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\",\n        )\n        .run(SCHEMA_VERSION, Date.now());\n    }\n  }\n\n  private storeRun(\n    threadId: string,\n    runId: string,\n    events: BaseEvent[],\n    input: RunAgentInput,\n    parentRunId?: string | null,\n  ): void {\n    // Compact ONLY the events from this run\n    const compactedEvents = compactEvents(events);\n\n    const stmt = this.db.prepare(`\n      INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version)\n      VALUES (?, ?, ?, ?, ?, ?, ?)\n    `);\n\n    stmt.run(\n      threadId,\n      runId,\n      parentRunId ?? null,\n      JSON.stringify(compactedEvents), // Store only this run's compacted events\n      JSON.stringify(input),\n      Date.now(),\n      SCHEMA_VERSION,\n    );\n  }\n\n  private getHistoricRuns(threadId: string): AgentRunRecord[] {\n    const stmt = this.db.prepare(`\n      WITH RECURSIVE run_chain AS (\n        -- Base case: find the root runs (those without parent)\n        SELECT * FROM agent_runs \n        WHERE thread_id = ? AND parent_run_id IS NULL\n        \n        UNION ALL\n        \n        -- Recursive case: find children of current level\n        SELECT ar.* FROM agent_runs ar\n        INNER JOIN run_chain rc ON ar.parent_run_id = rc.run_id\n        WHERE ar.thread_id = ?\n      )\n      SELECT * FROM run_chain\n      ORDER BY created_at ASC\n    `);\n\n    const rows = stmt.all(threadId, threadId) as any[];\n\n    return rows.map((row) => ({\n      id: row.id,\n      thread_id: row.thread_id,\n      run_id: row.run_id,\n      parent_run_id: row.parent_run_id,\n      events: JSON.parse(row.events),\n      input: JSON.parse(row.input),\n      created_at: row.created_at,\n      version: row.version,\n    }));\n  }\n\n  private getLatestRunId(threadId: string): string | null {\n    const stmt = this.db.prepare(`\n      SELECT run_id FROM agent_runs \n      WHERE thread_id = ? \n      ORDER BY created_at DESC \n      LIMIT 1\n    `);\n\n    const result = stmt.get(threadId) as { run_id: string } | undefined;\n    return result?.run_id ?? null;\n  }\n\n  private setRunState(\n    threadId: string,\n    isRunning: boolean,\n    runId?: string,\n  ): void {\n    const stmt = this.db.prepare(`\n      INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at)\n      VALUES (?, ?, ?, ?)\n    `);\n    stmt.run(threadId, isRunning ? 1 : 0, runId ?? null, Date.now());\n  }\n\n  private getRunState(threadId: string): {\n    isRunning: boolean;\n    currentRunId: string | null;\n  } {\n    const stmt = this.db.prepare(`\n      SELECT is_running, current_run_id FROM run_state WHERE thread_id = ?\n    `);\n    const result = stmt.get(threadId) as\n      | { is_running: number; current_run_id: string | null }\n      | undefined;\n\n    return {\n      isRunning: result?.is_running === 1,\n      currentRunId: result?.current_run_id ?? null,\n    };\n  }\n\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    // Check if thread is already running in database\n    const runState = this.getRunState(request.threadId);\n    if (runState.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n\n    // Mark thread as running in database\n    this.setRunState(request.threadId, true, request.input.runId);\n\n    // Track seen message IDs and current run events in memory for this run\n    const seenMessageIds = new Set<string>();\n    const currentRunEvents: BaseEvent[] = [];\n\n    // Get all previously seen message IDs from historic runs\n    const historicRuns = this.getHistoricRuns(request.threadId);\n    const historicMessageIds = new Set<string>();\n    for (const run of historicRuns) {\n      for (const event of run.events) {\n        if (\"messageId\" in event && typeof event.messageId === \"string\") {\n          historicMessageIds.add(event.messageId);\n        }\n        if (event.type === EventType.RUN_STARTED) {\n          const runStarted = event as RunStartedEvent;\n          const messages = runStarted.input?.messages ?? [];\n          for (const message of messages) {\n            historicMessageIds.add(message.id);\n          }\n        }\n      }\n    }\n\n    // Get or create subject for this thread's connections\n    const nextSubject = new ReplaySubject<BaseEvent>(Infinity);\n    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const prevSubject = prevConnection?.subject;\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Update the active connection for this thread\n    ACTIVE_CONNECTIONS.set(request.threadId, {\n      subject: nextSubject,\n      agent: request.agent,\n      runSubject,\n      currentEvents: currentRunEvents,\n      stopRequested: false,\n    });\n\n    // Helper function to run the agent and handle errors\n    const runAgent = async () => {\n      // Get parent run ID for chaining\n      const parentRunId = this.getLatestRunId(request.threadId);\n\n      try {\n        await request.agent.runAgent(request.input, {\n          onEvent: ({ event }) => {\n            let processedEvent: BaseEvent = event;\n            if (event.type === EventType.RUN_STARTED) {\n              const runStartedEvent = event as RunStartedEvent;\n              if (!runStartedEvent.input) {\n                const sanitizedMessages = request.input.messages\n                  ? request.input.messages.filter(\n                      (message) => !historicMessageIds.has(message.id),\n                    )\n                  : undefined;\n                const updatedInput = {\n                  ...request.input,\n                  ...(sanitizedMessages !== undefined\n                    ? { messages: sanitizedMessages }\n                    : {}),\n                };\n                processedEvent = {\n                  ...runStartedEvent,\n                  input: updatedInput,\n                } as RunStartedEvent;\n              }\n            }\n\n            runSubject.next(processedEvent); // For run() return - only agent events\n            nextSubject.next(processedEvent); // For connect() / store - all events\n            currentRunEvents.push(processedEvent); // Accumulate for database storage\n          },\n          onNewMessage: ({ message }) => {\n            // Called for each new message\n            if (!seenMessageIds.has(message.id)) {\n              seenMessageIds.add(message.id);\n            }\n          },\n          onRunStartedEvent: () => {\n            // Mark input messages as seen without emitting duplicates\n            if (request.input.messages) {\n              for (const message of request.input.messages) {\n                if (!seenMessageIds.has(message.id)) {\n                  seenMessageIds.add(message.id);\n                }\n              }\n            }\n          },\n        });\n\n        const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run in database\n        this.storeRun(\n          request.threadId,\n          request.input.runId,\n          currentRunEvents,\n          request.input,\n          parentRunId,\n        );\n\n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Complete the subjects\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      } catch {\n        const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run even if it failed (partial events)\n        if (currentRunEvents.length > 0) {\n          this.storeRun(\n            request.threadId,\n            request.input.runId,\n            currentRunEvents,\n            request.input,\n            parentRunId,\n          );\n        }\n\n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Don't emit error to the subject, just complete it\n        // This allows subscribers to get events emitted before the error\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      }\n    };\n\n    // Bridge previous events if they exist\n    if (prevSubject) {\n      prevSubject.subscribe({\n        next: (e) => nextSubject.next(e),\n        error: (err) => nextSubject.error(err),\n        complete: () => {\n          // Don't complete nextSubject here - it needs to stay open for new events\n        },\n      });\n    }\n\n    // Start the agent execution immediately (not lazily)\n    runAgent();\n\n    // Return the run subject (only agent events, no injected messages)\n    return runSubject.asObservable();\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const connectionSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Load historic runs from database\n    const historicRuns = this.getHistoricRuns(request.threadId);\n\n    // Collect all historic events from database\n    const allHistoricEvents: BaseEvent[] = [];\n    for (const run of historicRuns) {\n      allHistoricEvents.push(...run.events);\n    }\n\n    // Compact all events together before emitting\n    const compactedEvents = compactEvents(allHistoricEvents);\n\n    // Emit compacted events and track message IDs\n    const emittedMessageIds = new Set<string>();\n    for (const event of compactedEvents) {\n      connectionSubject.next(event);\n      if (\"messageId\" in event && typeof event.messageId === \"string\") {\n        emittedMessageIds.add(event.messageId);\n      }\n    }\n\n    // Bridge active run to connection if exists\n    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const runState = this.getRunState(request.threadId);\n\n    if (\n      activeConnection &&\n      (runState.isRunning || activeConnection.stopRequested)\n    ) {\n      activeConnection.subject.subscribe({\n        next: (event) => {\n          // Skip message events that we've already emitted from historic\n          if (\n            \"messageId\" in event &&\n            typeof event.messageId === \"string\" &&\n            emittedMessageIds.has(event.messageId)\n          ) {\n            return;\n          }\n          connectionSubject.next(event);\n        },\n        complete: () => connectionSubject.complete(),\n        error: (err) => connectionSubject.error(err),\n      });\n    } else {\n      // No active run, complete after historic events\n      connectionSubject.complete();\n    }\n\n    return connectionSubject.asObservable();\n  }\n\n  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean> {\n    const runState = this.getRunState(request.threadId);\n    return Promise.resolve(runState.isRunning);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const runState = this.getRunState(request.threadId);\n    if (!runState.isRunning) {\n      return Promise.resolve(false);\n    }\n\n    const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const agent = connection?.agent;\n\n    if (!connection || !agent) {\n      return Promise.resolve(false);\n    }\n\n    if (connection.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    connection.stopRequested = true;\n    this.setRunState(request.threadId, false);\n\n    try {\n      agent.abortRun();\n      return Promise.resolve(true);\n    } catch (error) {\n      console.error(\"Failed to abort sqlite agent run\", error);\n      connection.stopRequested = false;\n      this.setRunState(request.threadId, true);\n      return Promise.resolve(false);\n    }\n  }\n\n  /**\n   * Close the database connection (for cleanup)\n   */\n  close(): void {\n    if (this.db) {\n      this.db.close();\n    }\n  }\n}\n"],"mappings":";;;;;;AAmBA,MAAM,iBAAiB;AA0BvB,MAAM,qCAAqB,IAAI,KAAsC;AAErE,IAAa,oBAAb,cAAuC,YAAY;CACjD,AAAQ;CAER,YAAY,UAAoC,EAAE,EAAE;AAClD,SAAO;EACP,MAAM,SAAS,QAAQ,UAAU;AAEjC,MAAI,CAAC,SACH,OAAM,IAAI,MACR,6QAQD;AAGH,OAAK,KAAK,IAAI,SAAS,OAAO;AAC9B,OAAK,kBAAkB;;CAGzB,AAAQ,mBAAyB;AAE/B,OAAK,GAAG,KAAK;;;;;;;;;;;MAWX;AAGF,OAAK,GAAG,KAAK;;;;;;;MAOX;AAGF,OAAK,GAAG,KAAK;;;MAGX;AAGF,OAAK,GAAG,KAAK;;;;;MAKX;EAGF,MAAM,iBAAiB,KAAK,GACzB,QACC,mEACD,CACA,KAAK;AAER,MAAI,CAAC,kBAAkB,eAAe,UAAU,eAC9C,MAAK,GACF,QACC,4EACD,CACA,IAAI,gBAAgB,KAAK,KAAK,CAAC;;CAItC,AAAQ,SACN,UACA,OACA,QACA,OACA,aACM;EAEN,MAAM,kBAAkB,cAAc,OAAO;AAO7C,EALa,KAAK,GAAG,QAAQ;;;MAG3B,CAEG,IACH,UACA,OACA,eAAe,MACf,KAAK,UAAU,gBAAgB,EAC/B,KAAK,UAAU,MAAM,EACrB,KAAK,KAAK,EACV,eACD;;CAGH,AAAQ,gBAAgB,UAAoC;AAoB1D,SAnBa,KAAK,GAAG,QAAQ;;;;;;;;;;;;;;;MAe3B,CAEgB,IAAI,UAAU,SAAS,CAE7B,KAAK,SAAS;GACxB,IAAI,IAAI;GACR,WAAW,IAAI;GACf,QAAQ,IAAI;GACZ,eAAe,IAAI;GACnB,QAAQ,KAAK,MAAM,IAAI,OAAO;GAC9B,OAAO,KAAK,MAAM,IAAI,MAAM;GAC5B,YAAY,IAAI;GAChB,SAAS,IAAI;GACd,EAAE;;CAGL,AAAQ,eAAe,UAAiC;AAStD,SARa,KAAK,GAAG,QAAQ;;;;;MAK3B,CAEkB,IAAI,SAAS,EAClB,UAAU;;CAG3B,AAAQ,YACN,UACA,WACA,OACM;AAKN,EAJa,KAAK,GAAG,QAAQ;;;MAG3B,CACG,IAAI,UAAU,YAAY,IAAI,GAAG,SAAS,MAAM,KAAK,KAAK,CAAC;;CAGlE,AAAQ,YAAY,UAGlB;EAIA,MAAM,SAHO,KAAK,GAAG,QAAQ;;MAE3B,CACkB,IAAI,SAAS;AAIjC,SAAO;GACL,WAAW,QAAQ,eAAe;GAClC,cAAc,QAAQ,kBAAkB;GACzC;;CAGH,IAAI,SAAuD;AAGzD,MADiB,KAAK,YAAY,QAAQ,SAAS,CACtC,UACX,OAAM,IAAI,MAAM,yBAAyB;AAI3C,OAAK,YAAY,QAAQ,UAAU,MAAM,QAAQ,MAAM,MAAM;EAG7D,MAAM,iCAAiB,IAAI,KAAa;EACxC,MAAM,mBAAgC,EAAE;EAGxC,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAC3D,MAAM,qCAAqB,IAAI,KAAa;AAC5C,OAAK,MAAM,OAAO,aAChB,MAAK,MAAM,SAAS,IAAI,QAAQ;AAC9B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,oBAAmB,IAAI,MAAM,UAAU;AAEzC,OAAI,MAAM,SAAS,UAAU,aAAa;IAExC,MAAM,WADa,MACS,OAAO,YAAY,EAAE;AACjD,SAAK,MAAM,WAAW,SACpB,oBAAmB,IAAI,QAAQ,GAAG;;;EAO1C,MAAM,cAAc,IAAI,cAAyB,SAAS;EAE1D,MAAM,cADiB,mBAAmB,IAAI,QAAQ,SAAS,EAC3B;EAGpC,MAAM,aAAa,IAAI,cAAyB,SAAS;AAGzD,qBAAmB,IAAI,QAAQ,UAAU;GACvC,SAAS;GACT,OAAO,QAAQ;GACf;GACA,eAAe;GACf,eAAe;GAChB,CAAC;EAGF,MAAM,WAAW,YAAY;GAE3B,MAAM,cAAc,KAAK,eAAe,QAAQ,SAAS;AAEzD,OAAI;AACF,UAAM,QAAQ,MAAM,SAAS,QAAQ,OAAO;KAC1C,UAAU,EAAE,YAAY;MACtB,IAAI,iBAA4B;AAChC,UAAI,MAAM,SAAS,UAAU,aAAa;OACxC,MAAM,kBAAkB;AACxB,WAAI,CAAC,gBAAgB,OAAO;QAC1B,MAAM,oBAAoB,QAAQ,MAAM,WACpC,QAAQ,MAAM,SAAS,QACpB,YAAY,CAAC,mBAAmB,IAAI,QAAQ,GAAG,CACjD,GACD;QACJ,MAAM,eAAe;SACnB,GAAG,QAAQ;SACX,GAAI,sBAAsB,SACtB,EAAE,UAAU,mBAAmB,GAC/B,EAAE;SACP;AACD,yBAAiB;SACf,GAAG;SACH,OAAO;SACR;;;AAIL,iBAAW,KAAK,eAAe;AAC/B,kBAAY,KAAK,eAAe;AAChC,uBAAiB,KAAK,eAAe;;KAEvC,eAAe,EAAE,cAAc;AAE7B,UAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;KAGlC,yBAAyB;AAEvB,UAAI,QAAQ,MAAM,UAChB;YAAK,MAAM,WAAW,QAAQ,MAAM,SAClC,KAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;;KAKvC,CAAC;IAEF,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;IAC3D,MAAM,iBAAiB,kBAAkB,kBAAkB,EACzD,eAAe,YAAY,iBAAiB,OAC7C,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,SAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAGD,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAI7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;WACrC;IACN,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;IAC3D,MAAM,iBAAiB,kBAAkB,kBAAkB,EACzD,eAAe,YAAY,iBAAiB,OAC7C,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,iBAAiB,SAAS,EAC5B,MAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAIH,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAK7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;;;AAK/C,MAAI,YACF,aAAY,UAAU;GACpB,OAAO,MAAM,YAAY,KAAK,EAAE;GAChC,QAAQ,QAAQ,YAAY,MAAM,IAAI;GACtC,gBAAgB;GAGjB,CAAC;AAIJ,YAAU;AAGV,SAAO,WAAW,cAAc;;CAGlC,QAAQ,SAA2D;EACjE,MAAM,oBAAoB,IAAI,cAAyB,SAAS;EAGhE,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAG3D,MAAM,oBAAiC,EAAE;AACzC,OAAK,MAAM,OAAO,aAChB,mBAAkB,KAAK,GAAG,IAAI,OAAO;EAIvC,MAAM,kBAAkB,cAAc,kBAAkB;EAGxD,MAAM,oCAAoB,IAAI,KAAa;AAC3C,OAAK,MAAM,SAAS,iBAAiB;AACnC,qBAAkB,KAAK,MAAM;AAC7B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,mBAAkB,IAAI,MAAM,UAAU;;EAK1C,MAAM,mBAAmB,mBAAmB,IAAI,QAAQ,SAAS;EACjE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AAEnD,MACE,qBACC,SAAS,aAAa,iBAAiB,eAExC,kBAAiB,QAAQ,UAAU;GACjC,OAAO,UAAU;AAEf,QACE,eAAe,SACf,OAAO,MAAM,cAAc,YAC3B,kBAAkB,IAAI,MAAM,UAAU,CAEtC;AAEF,sBAAkB,KAAK,MAAM;;GAE/B,gBAAgB,kBAAkB,UAAU;GAC5C,QAAQ,QAAQ,kBAAkB,MAAM,IAAI;GAC7C,CAAC;MAGF,mBAAkB,UAAU;AAG9B,SAAO,kBAAkB,cAAc;;CAGzC,UAAU,SAAwD;EAChE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AACnD,SAAO,QAAQ,QAAQ,SAAS,UAAU;;CAG5C,KAAK,SAA+D;AAElE,MAAI,CADa,KAAK,YAAY,QAAQ,SAAS,CACrC,UACZ,QAAO,QAAQ,QAAQ,MAAM;EAG/B,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;EAC3D,MAAM,QAAQ,YAAY;AAE1B,MAAI,CAAC,cAAc,CAAC,MAClB,QAAO,QAAQ,QAAQ,MAAM;AAG/B,MAAI,WAAW,cACb,QAAO,QAAQ,QAAQ,MAAM;AAG/B,aAAW,gBAAgB;AAC3B,OAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,MAAI;AACF,SAAM,UAAU;AAChB,UAAO,QAAQ,QAAQ,KAAK;WACrB,OAAO;AACd,WAAQ,MAAM,oCAAoC,MAAM;AACxD,cAAW,gBAAgB;AAC3B,QAAK,YAAY,QAAQ,UAAU,KAAK;AACxC,UAAO,QAAQ,QAAQ,MAAM;;;;;;CAOjC,QAAc;AACZ,MAAI,KAAK,GACP,MAAK,GAAG,OAAO"}
+{"version":3,"file":"sqlite-runner.mjs","names":[],"sources":["../src/sqlite-runner.ts"],"sourcesContent":["import { AgentRunner, finalizeRunEvents } from \"@copilotkit/runtime/v2\";\nimport type {\n  AgentRunnerConnectRequest,\n  AgentRunnerIsRunningRequest,\n  AgentRunnerRunRequest,\n  AgentRunnerStopRequest,\n} from \"@copilotkit/runtime/v2\";\nimport type { Observable } from \"rxjs\";\nimport { ReplaySubject } from \"rxjs\";\nimport type {\n  AbstractAgent,\n  BaseEvent,\n  RunAgentInput,\n  RunStartedEvent,\n} from \"@ag-ui/client\";\nimport { EventType, compactEvents } from \"@ag-ui/client\";\nimport Database from \"better-sqlite3\";\n\nconst SCHEMA_VERSION = 1;\n\ninterface AgentRunRecord {\n  id: number;\n  thread_id: string;\n  run_id: string;\n  parent_run_id: string | null;\n  events: BaseEvent[];\n  input: RunAgentInput;\n  created_at: number;\n  version: number;\n}\n\nexport interface SqliteAgentRunnerOptions {\n  dbPath?: string;\n}\n\ninterface ActiveConnectionContext {\n  subject: ReplaySubject<BaseEvent>;\n  agent?: AbstractAgent;\n  runSubject?: ReplaySubject<BaseEvent>;\n  currentEvents?: BaseEvent[];\n  stopRequested?: boolean;\n}\n\n// Active connections for streaming events and stop support\nconst ACTIVE_CONNECTIONS = new Map<string, ActiveConnectionContext>();\n\nexport class SqliteAgentRunner extends AgentRunner {\n  private db: any;\n\n  constructor(options: SqliteAgentRunnerOptions = {}) {\n    super();\n    const dbPath = options.dbPath ?? \":memory:\";\n\n    if (!Database) {\n      throw new Error(\n        \"better-sqlite3 is required for SqliteAgentRunner but was not found.\\n\" +\n          \"Please install it in your project:\\n\" +\n          \"  npm install better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  pnpm add better-sqlite3\\n\" +\n          \"  or\\n\" +\n          \"  yarn add better-sqlite3\\n\\n\" +\n          \"If you don't need persistence, use InMemoryAgentRunner instead.\",\n      );\n    }\n\n    this.db = new Database(dbPath);\n    this.initializeSchema();\n  }\n\n  private initializeSchema(): void {\n    // Create the agent_runs table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS agent_runs (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        thread_id TEXT NOT NULL,\n        run_id TEXT NOT NULL UNIQUE,\n        parent_run_id TEXT,\n        events TEXT NOT NULL,\n        input TEXT NOT NULL,\n        created_at INTEGER NOT NULL,\n        version INTEGER NOT NULL\n      )\n    `);\n\n    // Create run_state table to track active runs\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS run_state (\n        thread_id TEXT PRIMARY KEY,\n        is_running INTEGER DEFAULT 0,\n        current_run_id TEXT,\n        updated_at INTEGER NOT NULL\n      )\n    `);\n\n    // Create indexes for efficient queries\n    this.db.exec(`\n      CREATE INDEX IF NOT EXISTS idx_thread_id ON agent_runs(thread_id);\n      CREATE INDEX IF NOT EXISTS idx_parent_run_id ON agent_runs(parent_run_id);\n    `);\n\n    // Create schema version table\n    this.db.exec(`\n      CREATE TABLE IF NOT EXISTS schema_version (\n        version INTEGER PRIMARY KEY,\n        applied_at INTEGER NOT NULL\n      )\n    `);\n\n    // Check and set schema version\n    const currentVersion = this.db\n      .prepare(\n        \"SELECT version FROM schema_version ORDER BY version DESC LIMIT 1\",\n      )\n      .get() as { version: number } | undefined;\n\n    if (!currentVersion || currentVersion.version < SCHEMA_VERSION) {\n      this.db\n        .prepare(\n          \"INSERT OR REPLACE INTO schema_version (version, applied_at) VALUES (?, ?)\",\n        )\n        .run(SCHEMA_VERSION, Date.now());\n    }\n  }\n\n  private storeRun(\n    threadId: string,\n    runId: string,\n    events: BaseEvent[],\n    input: RunAgentInput,\n    parentRunId?: string | null,\n  ): void {\n    // Compact ONLY the events from this run\n    const compactedEvents = compactEvents(events);\n\n    const stmt = this.db.prepare(`\n      INSERT INTO agent_runs (thread_id, run_id, parent_run_id, events, input, created_at, version)\n      VALUES (?, ?, ?, ?, ?, ?, ?)\n    `);\n\n    stmt.run(\n      threadId,\n      runId,\n      parentRunId ?? null,\n      JSON.stringify(compactedEvents), // Store only this run's compacted events\n      JSON.stringify(input),\n      Date.now(),\n      SCHEMA_VERSION,\n    );\n  }\n\n  private getHistoricRuns(threadId: string): AgentRunRecord[] {\n    const stmt = this.db.prepare(`\n      WITH RECURSIVE run_chain AS (\n        -- Base case: find the root runs (those without parent)\n        SELECT * FROM agent_runs \n        WHERE thread_id = ? AND parent_run_id IS NULL\n        \n        UNION ALL\n        \n        -- Recursive case: find children of current level\n        SELECT ar.* FROM agent_runs ar\n        INNER JOIN run_chain rc ON ar.parent_run_id = rc.run_id\n        WHERE ar.thread_id = ?\n      )\n      SELECT * FROM run_chain\n      ORDER BY created_at ASC\n    `);\n\n    const rows = stmt.all(threadId, threadId) as any[];\n\n    return rows.map((row) => ({\n      id: row.id,\n      thread_id: row.thread_id,\n      run_id: row.run_id,\n      parent_run_id: row.parent_run_id,\n      events: JSON.parse(row.events),\n      input: JSON.parse(row.input),\n      created_at: row.created_at,\n      version: row.version,\n    }));\n  }\n\n  private getLatestRunId(threadId: string): string | null {\n    const stmt = this.db.prepare(`\n      SELECT run_id FROM agent_runs \n      WHERE thread_id = ? \n      ORDER BY created_at DESC \n      LIMIT 1\n    `);\n\n    const result = stmt.get(threadId) as { run_id: string } | undefined;\n    return result?.run_id ?? null;\n  }\n\n  private setRunState(\n    threadId: string,\n    isRunning: boolean,\n    runId?: string,\n  ): void {\n    const stmt = this.db.prepare(`\n      INSERT OR REPLACE INTO run_state (thread_id, is_running, current_run_id, updated_at)\n      VALUES (?, ?, ?, ?)\n    `);\n    stmt.run(threadId, isRunning ? 1 : 0, runId ?? null, Date.now());\n  }\n\n  private getRunState(threadId: string): {\n    isRunning: boolean;\n    currentRunId: string | null;\n  } {\n    const stmt = this.db.prepare(`\n      SELECT is_running, current_run_id FROM run_state WHERE thread_id = ?\n    `);\n    const result = stmt.get(threadId) as\n      | { is_running: number; current_run_id: string | null }\n      | undefined;\n\n    return {\n      isRunning: result?.is_running === 1,\n      currentRunId: result?.current_run_id ?? null,\n    };\n  }\n\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    // Check if thread is already running in database\n    const runState = this.getRunState(request.threadId);\n    if (runState.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n\n    // Mark thread as running in database\n    this.setRunState(request.threadId, true, request.input.runId);\n\n    // Track seen message IDs and current run events in memory for this run\n    const seenMessageIds = new Set<string>();\n    const currentRunEvents: BaseEvent[] = [];\n\n    // Get all previously seen message IDs from historic runs\n    const historicRuns = this.getHistoricRuns(request.threadId);\n    const historicMessageIds = new Set<string>();\n    for (const run of historicRuns) {\n      for (const event of run.events) {\n        if (\"messageId\" in event && typeof event.messageId === \"string\") {\n          historicMessageIds.add(event.messageId);\n        }\n        if (event.type === EventType.RUN_STARTED) {\n          const runStarted = event as RunStartedEvent;\n          const messages = runStarted.input?.messages ?? [];\n          for (const message of messages) {\n            historicMessageIds.add(message.id);\n          }\n        }\n      }\n    }\n\n    // Get or create subject for this thread's connections\n    const nextSubject = new ReplaySubject<BaseEvent>(Infinity);\n    const prevConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const prevSubject = prevConnection?.subject;\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Update the active connection for this thread\n    ACTIVE_CONNECTIONS.set(request.threadId, {\n      subject: nextSubject,\n      agent: request.agent,\n      runSubject,\n      currentEvents: currentRunEvents,\n      stopRequested: false,\n    });\n\n    // Helper function to run the agent and handle errors\n    const runAgent = async () => {\n      // Get parent run ID for chaining\n      const parentRunId = this.getLatestRunId(request.threadId);\n\n      const ensureStoppedRunStarted = (stopRequested: boolean) => {\n        if (\n          !stopRequested ||\n          currentRunEvents.some((event) => event.type === EventType.RUN_STARTED)\n        ) {\n          return;\n        }\n        // Stop can arrive before the agent sends its first event.\n        const started: RunStartedEvent = {\n          type: EventType.RUN_STARTED,\n          threadId: request.threadId,\n          runId: request.input.runId,\n          input: {\n            ...request.input,\n            messages: request.input.messages.filter(\n              (message) => !historicMessageIds.has(message.id),\n            ),\n          },\n        };\n        currentRunEvents.unshift(started);\n        runSubject.next(started);\n        nextSubject.next(started);\n      };\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        ensureStoppedRunStarted(connection?.stopRequested ?? false);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n          protocolVersion: request.input.protocolVersion,\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        ensureStoppedRunStarted(connection?.stopRequested ?? false);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: connection?.stopRequested ?? false,\n          protocolVersion: request.input.protocolVersion,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the run even if it failed (partial events)\n        if (currentRunEvents.length > 0) {\n          this.storeRun(\n            request.threadId,\n            request.input.runId,\n            currentRunEvents,\n            request.input,\n            parentRunId,\n          );\n        }\n\n        // Mark run as complete in database\n        this.setRunState(request.threadId, false);\n\n        if (connection) {\n          connection.agent = undefined;\n          connection.runSubject = undefined;\n          connection.currentEvents = undefined;\n          connection.stopRequested = false;\n        }\n\n        // Don't emit error to the subject, just complete it\n        // This allows subscribers to get events emitted before the error\n        runSubject.complete();\n        nextSubject.complete();\n\n        ACTIVE_CONNECTIONS.delete(request.threadId);\n      }\n    };\n\n    // Bridge previous events if they exist\n    if (prevSubject) {\n      prevSubject.subscribe({\n        next: (e) => nextSubject.next(e),\n        error: (err) => nextSubject.error(err),\n        complete: () => {\n          // Don't complete nextSubject here - it needs to stay open for new events\n        },\n      });\n    }\n\n    // Start the agent execution immediately (not lazily)\n    runAgent();\n\n    // Return the run subject (only agent events, no injected messages)\n    return runSubject.asObservable();\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const connectionSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    // Load historic runs from database\n    const historicRuns = this.getHistoricRuns(request.threadId);\n\n    // Collect all historic events from database\n    const allHistoricEvents: BaseEvent[] = [];\n    for (const run of historicRuns) {\n      allHistoricEvents.push(...run.events);\n    }\n\n    // Compact all events together before emitting\n    const compactedEvents = compactEvents(allHistoricEvents);\n\n    // Emit compacted events and track message IDs\n    const emittedMessageIds = new Set<string>();\n    for (const event of compactedEvents) {\n      connectionSubject.next(event);\n      if (\"messageId\" in event && typeof event.messageId === \"string\") {\n        emittedMessageIds.add(event.messageId);\n      }\n    }\n\n    // Bridge active run to connection if exists\n    const activeConnection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const runState = this.getRunState(request.threadId);\n\n    if (\n      activeConnection &&\n      (runState.isRunning || activeConnection.stopRequested)\n    ) {\n      activeConnection.subject.subscribe({\n        next: (event) => {\n          // Skip message events that we've already emitted from historic\n          if (\n            \"messageId\" in event &&\n            typeof event.messageId === \"string\" &&\n            emittedMessageIds.has(event.messageId)\n          ) {\n            return;\n          }\n          connectionSubject.next(event);\n        },\n        complete: () => connectionSubject.complete(),\n        error: (err) => connectionSubject.error(err),\n      });\n    } else {\n      // No active run, complete after historic events\n      connectionSubject.complete();\n    }\n\n    return connectionSubject.asObservable();\n  }\n\n  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean> {\n    const runState = this.getRunState(request.threadId);\n    return Promise.resolve(runState.isRunning);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const runState = this.getRunState(request.threadId);\n    if (!runState.isRunning) {\n      return Promise.resolve(false);\n    }\n\n    const connection = ACTIVE_CONNECTIONS.get(request.threadId);\n    const agent = connection?.agent;\n\n    if (!connection || !agent) {\n      return Promise.resolve(false);\n    }\n\n    if (connection.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    connection.stopRequested = true;\n    this.setRunState(request.threadId, false);\n\n    try {\n      agent.abortRun();\n      return Promise.resolve(true);\n    } catch (error) {\n      console.error(\"Failed to abort sqlite agent run\", error);\n      connection.stopRequested = false;\n      this.setRunState(request.threadId, true);\n      return Promise.resolve(false);\n    }\n  }\n\n  /**\n   * Close the database connection (for cleanup)\n   */\n  close(): void {\n    if (this.db) {\n      this.db.close();\n    }\n  }\n}\n"],"mappings":";;;;;;AAkBA,MAAM,iBAAiB;AA0BvB,MAAM,qCAAqB,IAAI,KAAsC;AAErE,IAAa,oBAAb,cAAuC,YAAY;CACjD,AAAQ;CAER,YAAY,UAAoC,EAAE,EAAE;AAClD,SAAO;EACP,MAAM,SAAS,QAAQ,UAAU;AAEjC,MAAI,CAAC,SACH,OAAM,IAAI,MACR,6QAQD;AAGH,OAAK,KAAK,IAAI,SAAS,OAAO;AAC9B,OAAK,kBAAkB;;CAGzB,AAAQ,mBAAyB;AAE/B,OAAK,GAAG,KAAK;;;;;;;;;;;MAWX;AAGF,OAAK,GAAG,KAAK;;;;;;;MAOX;AAGF,OAAK,GAAG,KAAK;;;MAGX;AAGF,OAAK,GAAG,KAAK;;;;;MAKX;EAGF,MAAM,iBAAiB,KAAK,GACzB,QACC,mEACD,CACA,KAAK;AAER,MAAI,CAAC,kBAAkB,eAAe,UAAU,eAC9C,MAAK,GACF,QACC,4EACD,CACA,IAAI,gBAAgB,KAAK,KAAK,CAAC;;CAItC,AAAQ,SACN,UACA,OACA,QACA,OACA,aACM;EAEN,MAAM,kBAAkB,cAAc,OAAO;AAO7C,EALa,KAAK,GAAG,QAAQ;;;MAG3B,CAEG,IACH,UACA,OACA,eAAe,MACf,KAAK,UAAU,gBAAgB,EAC/B,KAAK,UAAU,MAAM,EACrB,KAAK,KAAK,EACV,eACD;;CAGH,AAAQ,gBAAgB,UAAoC;AAoB1D,SAnBa,KAAK,GAAG,QAAQ;;;;;;;;;;;;;;;MAe3B,CAEgB,IAAI,UAAU,SAAS,CAE7B,KAAK,SAAS;GACxB,IAAI,IAAI;GACR,WAAW,IAAI;GACf,QAAQ,IAAI;GACZ,eAAe,IAAI;GACnB,QAAQ,KAAK,MAAM,IAAI,OAAO;GAC9B,OAAO,KAAK,MAAM,IAAI,MAAM;GAC5B,YAAY,IAAI;GAChB,SAAS,IAAI;GACd,EAAE;;CAGL,AAAQ,eAAe,UAAiC;AAStD,SARa,KAAK,GAAG,QAAQ;;;;;MAK3B,CAEkB,IAAI,SAAS,EAClB,UAAU;;CAG3B,AAAQ,YACN,UACA,WACA,OACM;AAKN,EAJa,KAAK,GAAG,QAAQ;;;MAG3B,CACG,IAAI,UAAU,YAAY,IAAI,GAAG,SAAS,MAAM,KAAK,KAAK,CAAC;;CAGlE,AAAQ,YAAY,UAGlB;EAIA,MAAM,SAHO,KAAK,GAAG,QAAQ;;MAE3B,CACkB,IAAI,SAAS;AAIjC,SAAO;GACL,WAAW,QAAQ,eAAe;GAClC,cAAc,QAAQ,kBAAkB;GACzC;;CAGH,IAAI,SAAuD;AAGzD,MADiB,KAAK,YAAY,QAAQ,SAAS,CACtC,UACX,OAAM,IAAI,MAAM,yBAAyB;AAI3C,OAAK,YAAY,QAAQ,UAAU,MAAM,QAAQ,MAAM,MAAM;EAG7D,MAAM,iCAAiB,IAAI,KAAa;EACxC,MAAM,mBAAgC,EAAE;EAGxC,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAC3D,MAAM,qCAAqB,IAAI,KAAa;AAC5C,OAAK,MAAM,OAAO,aAChB,MAAK,MAAM,SAAS,IAAI,QAAQ;AAC9B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,oBAAmB,IAAI,MAAM,UAAU;AAEzC,OAAI,MAAM,SAAS,UAAU,aAAa;IAExC,MAAM,WADa,MACS,OAAO,YAAY,EAAE;AACjD,SAAK,MAAM,WAAW,SACpB,oBAAmB,IAAI,QAAQ,GAAG;;;EAO1C,MAAM,cAAc,IAAI,cAAyB,SAAS;EAE1D,MAAM,cADiB,mBAAmB,IAAI,QAAQ,SAAS,EAC3B;EAGpC,MAAM,aAAa,IAAI,cAAyB,SAAS;AAGzD,qBAAmB,IAAI,QAAQ,UAAU;GACvC,SAAS;GACT,OAAO,QAAQ;GACf;GACA,eAAe;GACf,eAAe;GAChB,CAAC;EAGF,MAAM,WAAW,YAAY;GAE3B,MAAM,cAAc,KAAK,eAAe,QAAQ,SAAS;GAEzD,MAAM,2BAA2B,kBAA2B;AAC1D,QACE,CAAC,iBACD,iBAAiB,MAAM,UAAU,MAAM,SAAS,UAAU,YAAY,CAEtE;IAGF,MAAM,UAA2B;KAC/B,MAAM,UAAU;KAChB,UAAU,QAAQ;KAClB,OAAO,QAAQ,MAAM;KACrB,OAAO;MACL,GAAG,QAAQ;MACX,UAAU,QAAQ,MAAM,SAAS,QAC9B,YAAY,CAAC,mBAAmB,IAAI,QAAQ,GAAG,CACjD;MACF;KACF;AACD,qBAAiB,QAAQ,QAAQ;AACjC,eAAW,KAAK,QAAQ;AACxB,gBAAY,KAAK,QAAQ;;AAG3B,OAAI;AACF,UAAM,QAAQ,MAAM,SAAS,QAAQ,OAAO;KAC1C,UAAU,EAAE,YAAY;MACtB,IAAI,iBAA4B;AAChC,UAAI,MAAM,SAAS,UAAU,aAAa;OACxC,MAAM,kBAAkB;AACxB,WAAI,CAAC,gBAAgB,OAAO;QAC1B,MAAM,oBAAoB,QAAQ,MAAM,WACpC,QAAQ,MAAM,SAAS,QACpB,YAAY,CAAC,mBAAmB,IAAI,QAAQ,GAAG,CACjD,GACD;QACJ,MAAM,eAAe;SACnB,GAAG,QAAQ;SACX,GAAI,sBAAsB,SACtB,EAAE,UAAU,mBAAmB,GAC/B,EAAE;SACP;AACD,yBAAiB;SACf,GAAG;SACH,OAAO;SACR;;;AAIL,iBAAW,KAAK,eAAe;AAC/B,kBAAY,KAAK,eAAe;AAChC,uBAAiB,KAAK,eAAe;;KAEvC,eAAe,EAAE,cAAc;AAE7B,UAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;KAGlC,yBAAyB;AAEvB,UAAI,QAAQ,MAAM,UAChB;YAAK,MAAM,WAAW,QAAQ,MAAM,SAClC,KAAI,CAAC,eAAe,IAAI,QAAQ,GAAG,CACjC,gBAAe,IAAI,QAAQ,GAAG;;;KAKvC,CAAC;IAEF,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;AAC3D,4BAAwB,YAAY,iBAAiB,MAAM;IAC3D,MAAM,iBAAiB,kBAAkB,kBAAkB;KACzD,eAAe,YAAY,iBAAiB;KAC5C,iBAAiB,QAAQ,MAAM;KAChC,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,SAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAGD,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAI7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;WACrC;IACN,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;AAC3D,4BAAwB,YAAY,iBAAiB,MAAM;IAC3D,MAAM,iBAAiB,kBAAkB,kBAAkB;KACzD,eAAe,YAAY,iBAAiB;KAC5C,iBAAiB,QAAQ,MAAM;KAChC,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,iBAAiB,SAAS,EAC5B,MAAK,SACH,QAAQ,UACR,QAAQ,MAAM,OACd,kBACA,QAAQ,OACR,YACD;AAIH,SAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,QAAI,YAAY;AACd,gBAAW,QAAQ;AACnB,gBAAW,aAAa;AACxB,gBAAW,gBAAgB;AAC3B,gBAAW,gBAAgB;;AAK7B,eAAW,UAAU;AACrB,gBAAY,UAAU;AAEtB,uBAAmB,OAAO,QAAQ,SAAS;;;AAK/C,MAAI,YACF,aAAY,UAAU;GACpB,OAAO,MAAM,YAAY,KAAK,EAAE;GAChC,QAAQ,QAAQ,YAAY,MAAM,IAAI;GACtC,gBAAgB;GAGjB,CAAC;AAIJ,YAAU;AAGV,SAAO,WAAW,cAAc;;CAGlC,QAAQ,SAA2D;EACjE,MAAM,oBAAoB,IAAI,cAAyB,SAAS;EAGhE,MAAM,eAAe,KAAK,gBAAgB,QAAQ,SAAS;EAG3D,MAAM,oBAAiC,EAAE;AACzC,OAAK,MAAM,OAAO,aAChB,mBAAkB,KAAK,GAAG,IAAI,OAAO;EAIvC,MAAM,kBAAkB,cAAc,kBAAkB;EAGxD,MAAM,oCAAoB,IAAI,KAAa;AAC3C,OAAK,MAAM,SAAS,iBAAiB;AACnC,qBAAkB,KAAK,MAAM;AAC7B,OAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,mBAAkB,IAAI,MAAM,UAAU;;EAK1C,MAAM,mBAAmB,mBAAmB,IAAI,QAAQ,SAAS;EACjE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AAEnD,MACE,qBACC,SAAS,aAAa,iBAAiB,eAExC,kBAAiB,QAAQ,UAAU;GACjC,OAAO,UAAU;AAEf,QACE,eAAe,SACf,OAAO,MAAM,cAAc,YAC3B,kBAAkB,IAAI,MAAM,UAAU,CAEtC;AAEF,sBAAkB,KAAK,MAAM;;GAE/B,gBAAgB,kBAAkB,UAAU;GAC5C,QAAQ,QAAQ,kBAAkB,MAAM,IAAI;GAC7C,CAAC;MAGF,mBAAkB,UAAU;AAG9B,SAAO,kBAAkB,cAAc;;CAGzC,UAAU,SAAwD;EAChE,MAAM,WAAW,KAAK,YAAY,QAAQ,SAAS;AACnD,SAAO,QAAQ,QAAQ,SAAS,UAAU;;CAG5C,KAAK,SAA+D;AAElE,MAAI,CADa,KAAK,YAAY,QAAQ,SAAS,CACrC,UACZ,QAAO,QAAQ,QAAQ,MAAM;EAG/B,MAAM,aAAa,mBAAmB,IAAI,QAAQ,SAAS;EAC3D,MAAM,QAAQ,YAAY;AAE1B,MAAI,CAAC,cAAc,CAAC,MAClB,QAAO,QAAQ,QAAQ,MAAM;AAG/B,MAAI,WAAW,cACb,QAAO,QAAQ,QAAQ,MAAM;AAG/B,aAAW,gBAAgB;AAC3B,OAAK,YAAY,QAAQ,UAAU,MAAM;AAEzC,MAAI;AACF,SAAM,UAAU;AAChB,UAAO,QAAQ,QAAQ,KAAK;WACrB,OAAO;AACd,WAAQ,MAAM,oCAAoC,MAAM;AACxD,cAAW,gBAAgB;AAC3B,QAAK,YAAY,QAAQ,UAAU,KAAK;AACxC,UAAO,QAAQ,QAAQ,MAAM;;;;;;CAOjC,QAAc;AACZ,MAAI,KAAK,GACP,MAAK,GAAG,OAAO"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/sqlite-runner",
-  "version": "1.75.1",
+  "version": "1.76.0",
   "description": "SQLite-backed agent runner for CopilotKit2",
   "license": "MIT",
   "repository": {
@@ -22,9 +22,9 @@
     "access": "public"
   },
   "dependencies": {
-    "@ag-ui/client": "0.0.59",
+    "@ag-ui/client": "1.0.1",
     "rxjs": "7.8.1",
-    "@copilotkit/runtime": "1.75.1"
+    "@copilotkit/runtime": "1.76.0"
   },
   "devDependencies": {
     "@types/better-sqlite3": "^7.6.13",

package/src/__tests__/sqlite-early-stop.test.ts

@@ -0,0 +1,121 @@
+import {
+  AbstractAgent,
+  EventType,
+  HttpAgent,
+  enforceEvents,
+  verifyEvents,
+} from "@ag-ui/client";
+import type { RunAgentInput, RunAgentResult } from "@ag-ui/client";
+import { EMPTY, firstValueFrom, from, toArray } from "rxjs";
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { SqliteAgentRunner } from "../sqlite-runner";
+
+class ResolvingAbortAgent extends AbstractAgent {
+  private finish: (() => void) | undefined;
+
+  run() {
+    return EMPTY;
+  }
+
+  override runAgent(): Promise<RunAgentResult> {
+    return new Promise((resolve) => {
+      this.finish = () => resolve({ result: undefined, newMessages: [] });
+    });
+  }
+
+  override abortRun(): void {
+    this.finish?.();
+  }
+}
+
+describe("SqliteAgentRunner stop before the first event", () => {
+  afterEach(() => {
+    vi.unstubAllGlobals();
+  });
+
+  it.each([
+    ["1.0", "http"],
+    [undefined, "http"],
+    ["1.0", "resolving"],
+    [undefined, "resolving"],
+  ] as const)(
+    "keeps valid live and replay events for %s / %s",
+    async (protocolVersion, transport) => {
+      const runner = new SqliteAgentRunner();
+      const threadId = crypto.randomUUID();
+      const runId = crypto.randomUUID();
+      const input: RunAgentInput = {
+        threadId,
+        runId,
+        messages: [
+          { id: "user-prompt", role: "user", content: "Start a task" },
+        ],
+        tools: [],
+        context: [],
+        state: {},
+        forwardedProps: {},
+        ...(protocolVersion ? { protocolVersion } : {}),
+      };
+      let requestStarted: (() => void) | undefined;
+      const started = new Promise<void>((resolve) => {
+        requestStarted = resolve;
+      });
+      vi.stubGlobal(
+        "fetch",
+        vi.fn((_url: string | URL | Request, init?: RequestInit) => {
+          requestStarted?.();
+          return new Promise<Response>((_resolve, reject) => {
+            const signal = init?.signal;
+            if (!signal) throw new Error("Missing request signal");
+            const abort = () =>
+              reject(new DOMException("Aborted", "AbortError"));
+            if (signal.aborted) abort();
+            else signal.addEventListener("abort", abort, { once: true });
+          });
+        }),
+      );
+      const agent =
+        transport === "http"
+          ? new HttpAgent({ url: "http://agent.test", threadId })
+          : new ResolvingAbortAgent();
+      try {
+        const result = firstValueFrom(
+          runner.run({ threadId, input, agent }).pipe(toArray()),
+        );
+        if (transport === "http") await started;
+        const live = firstValueFrom(
+          runner.connect({ threadId }).pipe(toArray()),
+        );
+        expect(await runner.stop({ threadId })).toBe(true);
+        const events = await result;
+        expect(events.map((event) => event.type)).toEqual([
+          EventType.RUN_STARTED,
+          EventType.RUN_FINISHED,
+        ]);
+        expect(events[0]).toMatchObject({ threadId, runId, input });
+        expect(events[1]).toEqual({
+          type: EventType.RUN_FINISHED,
+          threadId,
+          runId,
+          ...(protocolVersion ? { outcome: { type: "cancelled" } } : {}),
+        });
+        expect(
+          await firstValueFrom(
+            from(events).pipe(enforceEvents(), verifyEvents(), toArray()),
+          ),
+        ).toEqual(events);
+        expect(await live).toEqual(events);
+        expect(
+          await firstValueFrom(
+            runner
+              .connect({ threadId })
+              .pipe(enforceEvents(), verifyEvents(), toArray()),
+          ),
+        ).toEqual(events);
+        expect(await runner.isRunning({ threadId })).toBe(false);
+      } finally {
+        runner.close();
+      }
+    },
+  );
+});

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

@@ -1,16 +1,16 @@
 import { describe, it, expect, afterEach } from "vitest";
 import { SqliteAgentRunner } from "..";
-import {
-  AbstractAgent,
+import type {
   BaseEvent,
-  EventType,
   Message,
   RunAgentInput,
   RunErrorEvent,
   RunFinishedEvent,
   RunStartedEvent,
 } from "@ag-ui/client";
-import { EMPTY, Subscription, firstValueFrom, from } from "rxjs";
+import { AbstractAgent, EventType } from "@ag-ui/client";
+import type { Subscription } from "rxjs";
+import { EMPTY, firstValueFrom, from } from "rxjs";
 import { toArray } from "rxjs/operators";
 
 type RunCallbacks = {
@@ -600,7 +600,7 @@ describe("SqliteAgentRunner e2e", () => {
         type: EventType.RUN_STARTED,
         threadId,
         runId,
-        parentRunId: null,
+        parentRunId: "custom-parent-run",
         input: customInput,
       };
 
@@ -787,4 +787,85 @@ describe("SqliteAgentRunner e2e", () => {
       expect(replayAgent.messages).toEqual([userMessage]);
     });
   });
+  describe("Subagent Attribution", () => {
+    it("replays subagent lifecycle events and keeps attribution on messages", async () => {
+      const runner = createRunner();
+      const threadId = "thread-subagents";
+      const subagentEvents = [
+        {
+          type: EventType.SUBAGENT_STARTED,
+          subagentRunId: "research",
+          name: "researcher",
+          parentToolCallId: "call-1",
+          parentMessageId: "supervisor",
+        },
+        {
+          type: EventType.SUBAGENT_FINISHED,
+          subagentRunId: "research",
+          result: { sources: 3 },
+        },
+      ] as BaseEvent[];
+      const events = [
+        ...createTextMessageEvents({
+          messageId: "supervisor",
+          content: "Delegating",
+        }),
+        {
+          type: EventType.TOOL_CALL_START,
+          toolCallId: "call-1",
+          toolCallName: "research",
+          parentMessageId: "supervisor",
+        },
+        { type: EventType.TOOL_CALL_END, toolCallId: "call-1" },
+        subagentEvents[0],
+        {
+          type: EventType.TEXT_MESSAGE_START,
+          messageId: "research-1",
+          role: "assistant",
+          subagentRunId: "research",
+        },
+        {
+          type: EventType.TEXT_MESSAGE_CONTENT,
+          messageId: "research-1",
+          delta: "Found 3 sources",
+        },
+        { type: EventType.TEXT_MESSAGE_END, messageId: "research-1" },
+        subagentEvents[1],
+        {
+          type: EventType.TOOL_CALL_RESULT,
+          toolCallId: "call-1",
+          messageId: "result-1",
+          content: "3 sources",
+          role: "tool",
+        },
+      ] as BaseEvent[];
+
+      await collectEvents(
+        runner.run({
+          threadId,
+          agent: new EmitAgent({ events }),
+          input: createRunInput({ threadId, runId: "run-0", messages: [] }),
+        }),
+      );
+
+      const replayEvents = await collectEvents(runner.connect({ threadId }));
+      const replayAgent = new ReplayAgent(replayEvents, threadId);
+      await replayAgent.connectAgent({ runId: "replay-run" });
+
+      expect(
+        replayEvents.filter(
+          ({ type }) =>
+            type === EventType.SUBAGENT_STARTED ||
+            type === EventType.SUBAGENT_FINISHED,
+        ),
+      ).toEqual(subagentEvents);
+      expect(
+        replayAgent.messages.find(({ id }) => id === "research-1"),
+      ).toMatchObject({
+        role: "assistant",
+        content: "Found 3 sources",
+        subagentRunId: "research",
+      });
+    });
+  });
 });

package/src/sqlite-runner.ts

@@ -1,20 +1,19 @@
-import {
-  AgentRunner,
-  finalizeRunEvents,
-  type AgentRunnerConnectRequest,
-  type AgentRunnerIsRunningRequest,
-  type AgentRunnerRunRequest,
-  type AgentRunnerStopRequest,
+import { AgentRunner, finalizeRunEvents } from "@copilotkit/runtime/v2";
+import type {
+  AgentRunnerConnectRequest,
+  AgentRunnerIsRunningRequest,
+  AgentRunnerRunRequest,
+  AgentRunnerStopRequest,
 } from "@copilotkit/runtime/v2";
-import { Observable, ReplaySubject } from "rxjs";
-import {
+import type { Observable } from "rxjs";
+import { ReplaySubject } from "rxjs";
+import type {
   AbstractAgent,
   BaseEvent,
   RunAgentInput,
-  EventType,
   RunStartedEvent,
-  compactEvents,
 } from "@ag-ui/client";
+import { EventType, compactEvents } from "@ag-ui/client";
 import Database from "better-sqlite3";
 
 const SCHEMA_VERSION = 1;
@@ -277,6 +276,30 @@ export class SqliteAgentRunner extends AgentRunner {
       // Get parent run ID for chaining
       const parentRunId = this.getLatestRunId(request.threadId);
 
+      const ensureStoppedRunStarted = (stopRequested: boolean) => {
+        if (
+          !stopRequested ||
+          currentRunEvents.some((event) => event.type === EventType.RUN_STARTED)
+        ) {
+          return;
+        }
+        // Stop can arrive before the agent sends its first event.
+        const started: RunStartedEvent = {
+          type: EventType.RUN_STARTED,
+          threadId: request.threadId,
+          runId: request.input.runId,
+          input: {
+            ...request.input,
+            messages: request.input.messages.filter(
+              (message) => !historicMessageIds.has(message.id),
+            ),
+          },
+        };
+        currentRunEvents.unshift(started);
+        runSubject.next(started);
+        nextSubject.next(started);
+      };
+
       try {
         await request.agent.runAgent(request.input, {
           onEvent: ({ event }) => {
@@ -325,8 +348,10 @@ export class SqliteAgentRunner extends AgentRunner {
         });
 
         const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        ensureStoppedRunStarted(connection?.stopRequested ?? false);
         const appendedEvents = finalizeRunEvents(currentRunEvents, {
           stopRequested: connection?.stopRequested ?? false,
+          protocolVersion: request.input.protocolVersion,
         });
         for (const event of appendedEvents) {
           runSubject.next(event);
@@ -359,8 +384,10 @@ export class SqliteAgentRunner extends AgentRunner {
         ACTIVE_CONNECTIONS.delete(request.threadId);
       } catch {
         const connection = ACTIVE_CONNECTIONS.get(request.threadId);
+        ensureStoppedRunStarted(connection?.stopRequested ?? false);
         const appendedEvents = finalizeRunEvents(currentRunEvents, {
           stopRequested: connection?.stopRequested ?? false,
+          protocolVersion: request.input.protocolVersion,
         });
         for (const event of appendedEvents) {
           runSubject.next(event);