@copilotkit/runtime 1.54.1 → 1.55.0-next.8

package/dist/v2/runtime/runner/in-memory.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory.cjs","names":["AgentRunner","EventType","ReplaySubject"],"sources":["../../../../src/v2/runtime/runner/in-memory.ts"],"sourcesContent":["import {\n  AgentRunner,\n  AgentRunnerConnectRequest,\n  AgentRunnerIsRunningRequest,\n  AgentRunnerRunRequest,\n  type AgentRunnerStopRequest,\n} from \"./agent-runner\";\nimport { Observable, ReplaySubject } from \"rxjs\";\nimport {\n  AbstractAgent,\n  BaseEvent,\n  EventType,\n  MessagesSnapshotEvent,\n  RunStartedEvent,\n  compactEvents,\n} from \"@ag-ui/client\";\nimport { finalizeRunEvents } from \"@copilotkit/shared\";\n\ninterface HistoricRun {\n  threadId: string;\n  runId: string;\n  parentRunId: string | null;\n  events: BaseEvent[];\n  createdAt: number;\n}\n\nclass InMemoryEventStore {\n  constructor(public threadId: string) {}\n\n  /** The subject that current consumers subscribe to. */\n  subject: ReplaySubject<BaseEvent> | null = null;\n\n  /** True while a run is actively producing events. */\n  isRunning = false;\n\n  /** Current run ID */\n  currentRunId: string | null = null;\n\n  /** Historic completed runs */\n  historicRuns: HistoricRun[] = [];\n\n  /** Currently running agent instance (if any). */\n  agent: AbstractAgent | null = null;\n\n  /** Subject returned from run() while the run is active. */\n  runSubject: ReplaySubject<BaseEvent> | null = null;\n\n  /** True once stop() has been requested but the run has not yet finalized. */\n  stopRequested = false;\n\n  /** Reference to the events emitted in the current run. */\n  currentEvents: BaseEvent[] | null = null;\n}\n\n// Use a symbol key on globalThis to survive hot reloads in development\nconst GLOBAL_STORE_KEY = Symbol.for(\"@copilotkit/runtime/in-memory-store\");\n\ninterface GlobalStoreData {\n  stores: Map<string, InMemoryEventStore>;\n  historicRunsBackup: Map<string, HistoricRun[]>;\n}\n\nfunction getGlobalStore(): Map<string, InMemoryEventStore> {\n  const globalAny = globalThis as unknown as Record<symbol, GlobalStoreData>;\n\n  if (!globalAny[GLOBAL_STORE_KEY]) {\n    globalAny[GLOBAL_STORE_KEY] = {\n      stores: new Map<string, InMemoryEventStore>(),\n      historicRunsBackup: new Map<string, HistoricRun[]>(),\n    };\n  }\n\n  const data = globalAny[GLOBAL_STORE_KEY];\n\n  // Restore historic runs from backup after hot reload\n  // (when stores map is empty but backup has data)\n  if (data.stores.size === 0 && data.historicRunsBackup.size > 0) {\n    for (const [threadId, historicRuns] of data.historicRunsBackup) {\n      const store = new InMemoryEventStore(threadId);\n      store.historicRuns = historicRuns;\n      data.stores.set(threadId, store);\n    }\n  }\n\n  return data.stores;\n}\n\nfunction backupHistoricRuns(\n  threadId: string,\n  historicRuns: HistoricRun[],\n): void {\n  const globalAny = globalThis as unknown as Record<symbol, GlobalStoreData>;\n  if (globalAny[GLOBAL_STORE_KEY]) {\n    globalAny[GLOBAL_STORE_KEY].historicRunsBackup.set(threadId, historicRuns);\n  }\n}\n\nconst GLOBAL_STORE = getGlobalStore();\n\nexport class InMemoryAgentRunner extends AgentRunner {\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    let existingStore = GLOBAL_STORE.get(request.threadId);\n    if (!existingStore) {\n      existingStore = new InMemoryEventStore(request.threadId);\n      GLOBAL_STORE.set(request.threadId, existingStore);\n    }\n    const store = existingStore; // Now store is const and non-null\n\n    if (store.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n    store.isRunning = true;\n    store.currentRunId = request.input.runId;\n    store.agent = request.agent;\n    store.stopRequested = false;\n\n    // Track seen message IDs and current run events for this run\n    const seenMessageIds = new Set<string>();\n    const currentRunEvents: BaseEvent[] = [];\n    store.currentEvents = currentRunEvents;\n\n    // Get all previously seen message IDs from historic runs\n    const historicMessageIds = new Set<string>();\n    for (const run of store.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    const nextSubject = new ReplaySubject<BaseEvent>(Infinity);\n    const prevSubject = store.subject;\n\n    // Update the store's subject immediately\n    store.subject = nextSubject;\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\n    store.runSubject = runSubject;\n\n    // Helper function to run the agent and handle errors\n    const runAgent = async () => {\n      // Get parent run ID for chaining\n      const lastRun = store.historicRuns[store.historicRuns.length - 1];\n      const parentRunId = lastRun?.runId ?? null;\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 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 any messages from the input as seen so they aren't emitted twice\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 appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: store.stopRequested,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the completed run in memory with ONLY its events\n        if (store.currentRunId) {\n          // Compact the events before storing (like SQLite does)\n          const compactedEvents = compactEvents(currentRunEvents);\n\n          store.historicRuns.push({\n            threadId: request.threadId,\n            runId: store.currentRunId,\n            parentRunId,\n            events: compactedEvents,\n            createdAt: Date.now(),\n          });\n\n          // Backup for hot reload survival\n          backupHistoricRuns(request.threadId, store.historicRuns);\n        }\n\n        // Complete the run\n        store.currentEvents = null;\n        store.currentRunId = null;\n        store.agent = null;\n        store.runSubject = null;\n        store.stopRequested = false;\n        store.isRunning = false;\n        runSubject.complete();\n        nextSubject.complete();\n      } catch (error) {\n        const interruptionMessage =\n          error instanceof Error ? error.message : String(error);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: store.stopRequested,\n          interruptionMessage,\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 (store.currentRunId && currentRunEvents.length > 0) {\n          // Compact the events before storing (like SQLite does)\n          const compactedEvents = compactEvents(currentRunEvents);\n          store.historicRuns.push({\n            threadId: request.threadId,\n            runId: store.currentRunId,\n            parentRunId,\n            events: compactedEvents,\n            createdAt: Date.now(),\n          });\n\n          // Backup for hot reload survival\n          backupHistoricRuns(request.threadId, store.historicRuns);\n        }\n\n        // Complete the run\n        store.currentEvents = null;\n        store.currentRunId = null;\n        store.agent = null;\n        store.runSubject = null;\n        store.stopRequested = false;\n        store.isRunning = false;\n        runSubject.complete();\n        nextSubject.complete();\n      }\n    };\n\n    // Bridge previous events if they exist\n    if (prevSubject) {\n      prevSubject.subscribe({\n        next: (e) => nextSubject.next(e),\n        error: (err) => nextSubject.error(err),\n        complete: () => {\n          // Don't complete nextSubject here - it needs to stay open for new events\n        },\n      });\n    }\n\n    // Start the agent execution immediately (not lazily)\n    runAgent();\n\n    // Return the run subject (only agent events, no injected messages)\n    return runSubject.asObservable();\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const store = GLOBAL_STORE.get(request.threadId);\n    const connectionSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    if (!store) {\n      // No store means no events\n      connectionSubject.complete();\n      return connectionSubject.asObservable();\n    }\n\n    // Collect all historic events from memory\n    const allHistoricEvents: BaseEvent[] = [];\n    for (const run of store.historicRuns) {\n      allHistoricEvents.push(...run.events);\n    }\n\n    // Apply compaction to all historic events together (like SQLite)\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    if (store.subject && (store.isRunning || store.stopRequested)) {\n      store.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 store = GLOBAL_STORE.get(request.threadId);\n    return Promise.resolve(store?.isRunning ?? false);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const store = GLOBAL_STORE.get(request.threadId);\n    if (!store || !store.isRunning) {\n      return Promise.resolve(false);\n    }\n    if (store.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    store.stopRequested = true;\n    store.isRunning = false;\n\n    const agent = store.agent;\n    if (!agent) {\n      store.stopRequested = false;\n      store.isRunning = false;\n      return Promise.resolve(false);\n    }\n\n    try {\n      agent.abortRun();\n      return Promise.resolve(true);\n    } catch (error) {\n      console.error(\"Failed to abort agent run\", error);\n      store.stopRequested = false;\n      store.isRunning = true;\n      return Promise.resolve(false);\n    }\n  }\n}\n"],"mappings":";;;;;;;;AA0BA,IAAM,qBAAN,MAAyB;CACvB,YAAY,AAAO,UAAkB;EAAlB;iBAGwB;mBAG/B;sBAGkB;sBAGA,EAAE;eAGF;oBAGgB;uBAG9B;uBAGoB;;;AAItC,MAAM,mBAAmB,OAAO,IAAI,sCAAsC;AAO1E,SAAS,iBAAkD;CACzD,MAAM,YAAY;AAElB,KAAI,CAAC,UAAU,kBACb,WAAU,oBAAoB;EAC5B,wBAAQ,IAAI,KAAiC;EAC7C,oCAAoB,IAAI,KAA4B;EACrD;CAGH,MAAM,OAAO,UAAU;AAIvB,KAAI,KAAK,OAAO,SAAS,KAAK,KAAK,mBAAmB,OAAO,EAC3D,MAAK,MAAM,CAAC,UAAU,iBAAiB,KAAK,oBAAoB;EAC9D,MAAM,QAAQ,IAAI,mBAAmB,SAAS;AAC9C,QAAM,eAAe;AACrB,OAAK,OAAO,IAAI,UAAU,MAAM;;AAIpC,QAAO,KAAK;;AAGd,SAAS,mBACP,UACA,cACM;CACN,MAAM,YAAY;AAClB,KAAI,UAAU,kBACZ,WAAU,kBAAkB,mBAAmB,IAAI,UAAU,aAAa;;AAI9E,MAAM,eAAe,gBAAgB;AAErC,IAAa,sBAAb,cAAyCA,iCAAY;CACnD,IAAI,SAAuD;EACzD,IAAI,gBAAgB,aAAa,IAAI,QAAQ,SAAS;AACtD,MAAI,CAAC,eAAe;AAClB,mBAAgB,IAAI,mBAAmB,QAAQ,SAAS;AACxD,gBAAa,IAAI,QAAQ,UAAU,cAAc;;EAEnD,MAAM,QAAQ;AAEd,MAAI,MAAM,UACR,OAAM,IAAI,MAAM,yBAAyB;AAE3C,QAAM,YAAY;AAClB,QAAM,eAAe,QAAQ,MAAM;AACnC,QAAM,QAAQ,QAAQ;AACtB,QAAM,gBAAgB;EAGtB,MAAM,iCAAiB,IAAI,KAAa;EACxC,MAAM,mBAAgC,EAAE;AACxC,QAAM,gBAAgB;EAGtB,MAAM,qCAAqB,IAAI,KAAa;AAC5C,OAAK,MAAM,OAAO,MAAM,aACtB,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;;;EAM1C,MAAM,cAAc,IAAIC,mBAAyB,SAAS;EAC1D,MAAM,cAAc,MAAM;AAG1B,QAAM,UAAU;EAGhB,MAAM,aAAa,IAAIA,mBAAyB,SAAS;AACzD,QAAM,aAAa;EAGnB,MAAM,WAAW,YAAY;GAG3B,MAAM,cADU,MAAM,aAAa,MAAM,aAAa,SAAS,IAClC,SAAS;AAEtC,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,2DAAmC,kBAAkB,EACzD,eAAe,MAAM,eACtB,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,MAAM,cAAc;KAEtB,MAAM,mDAAgC,iBAAiB;AAEvD,WAAM,aAAa,KAAK;MACtB,UAAU,QAAQ;MAClB,OAAO,MAAM;MACb;MACA,QAAQ;MACR,WAAW,KAAK,KAAK;MACtB,CAAC;AAGF,wBAAmB,QAAQ,UAAU,MAAM,aAAa;;AAI1D,UAAM,gBAAgB;AACtB,UAAM,eAAe;AACrB,UAAM,QAAQ;AACd,UAAM,aAAa;AACnB,UAAM,gBAAgB;AACtB,UAAM,YAAY;AAClB,eAAW,UAAU;AACrB,gBAAY,UAAU;YACf,OAAO;IACd,MAAM,sBACJ,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACxD,MAAM,2DAAmC,kBAAkB;KACzD,eAAe,MAAM;KACrB;KACD,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,MAAM,gBAAgB,iBAAiB,SAAS,GAAG;KAErD,MAAM,mDAAgC,iBAAiB;AACvD,WAAM,aAAa,KAAK;MACtB,UAAU,QAAQ;MAClB,OAAO,MAAM;MACb;MACA,QAAQ;MACR,WAAW,KAAK,KAAK;MACtB,CAAC;AAGF,wBAAmB,QAAQ,UAAU,MAAM,aAAa;;AAI1D,UAAM,gBAAgB;AACtB,UAAM,eAAe;AACrB,UAAM,QAAQ;AACd,UAAM,aAAa;AACnB,UAAM,gBAAgB;AACtB,UAAM,YAAY;AAClB,eAAW,UAAU;AACrB,gBAAY,UAAU;;;AAK1B,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,QAAQ,aAAa,IAAI,QAAQ,SAAS;EAChD,MAAM,oBAAoB,IAAIC,mBAAyB,SAAS;AAEhE,MAAI,CAAC,OAAO;AAEV,qBAAkB,UAAU;AAC5B,UAAO,kBAAkB,cAAc;;EAIzC,MAAM,oBAAiC,EAAE;AACzC,OAAK,MAAM,OAAO,MAAM,aACtB,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;;AAK1C,MAAI,MAAM,YAAY,MAAM,aAAa,MAAM,eAC7C,OAAM,QAAQ,UAAU;GACtB,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,QAAQ,aAAa,IAAI,QAAQ,SAAS;AAChD,SAAO,QAAQ,QAAQ,OAAO,aAAa,MAAM;;CAGnD,KAAK,SAA+D;EAClE,MAAM,QAAQ,aAAa,IAAI,QAAQ,SAAS;AAChD,MAAI,CAAC,SAAS,CAAC,MAAM,UACnB,QAAO,QAAQ,QAAQ,MAAM;AAE/B,MAAI,MAAM,cACR,QAAO,QAAQ,QAAQ,MAAM;AAG/B,QAAM,gBAAgB;AACtB,QAAM,YAAY;EAElB,MAAM,QAAQ,MAAM;AACpB,MAAI,CAAC,OAAO;AACV,SAAM,gBAAgB;AACtB,SAAM,YAAY;AAClB,UAAO,QAAQ,QAAQ,MAAM;;AAG/B,MAAI;AACF,SAAM,UAAU;AAChB,UAAO,QAAQ,QAAQ,KAAK;WACrB,OAAO;AACd,WAAQ,MAAM,6BAA6B,MAAM;AACjD,SAAM,gBAAgB;AACtB,SAAM,YAAY;AAClB,UAAO,QAAQ,QAAQ,MAAM"}

package/dist/v2/runtime/runner/in-memory.d.cts

@@ -0,0 +1,15 @@
+
+import { AgentRunner, AgentRunnerConnectRequest, AgentRunnerIsRunningRequest, AgentRunnerRunRequest, AgentRunnerStopRequest } from "./agent-runner.cjs";
+import { Observable } from "rxjs";
+import { BaseEvent } from "@ag-ui/client";
+
+//#region src/v2/runtime/runner/in-memory.d.ts
+declare class InMemoryAgentRunner extends AgentRunner {
+  run(request: AgentRunnerRunRequest): Observable<BaseEvent>;
+  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;
+  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;
+  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;
+}
+//#endregion
+export { InMemoryAgentRunner };
+//# sourceMappingURL=in-memory.d.cts.map

package/dist/v2/runtime/runner/in-memory.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory.d.cts","names":[],"sources":["../../../../src/v2/runtime/runner/in-memory.ts"],"mappings":";;;;;;cAmGa,mBAAA,SAA4B,WAAA;EACvC,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EAiMhD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAqDxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAKjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;AAAA"}

package/dist/v2/runtime/runner/in-memory.d.mts

@@ -0,0 +1,15 @@
+import "reflect-metadata";
+import { AgentRunner, AgentRunnerConnectRequest, AgentRunnerIsRunningRequest, AgentRunnerRunRequest, AgentRunnerStopRequest } from "./agent-runner.mjs";
+import { Observable } from "rxjs";
+import { BaseEvent } from "@ag-ui/client";
+
+//#region src/v2/runtime/runner/in-memory.d.ts
+declare class InMemoryAgentRunner extends AgentRunner {
+  run(request: AgentRunnerRunRequest): Observable<BaseEvent>;
+  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;
+  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;
+  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;
+}
+//#endregion
+export { InMemoryAgentRunner };
+//# sourceMappingURL=in-memory.d.mts.map

package/dist/v2/runtime/runner/in-memory.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory.d.mts","names":[],"sources":["../../../../src/v2/runtime/runner/in-memory.ts"],"mappings":";;;;;;cAmGa,mBAAA,SAA4B,WAAA;EACvC,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EAiMhD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAqDxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAKjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;AAAA"}

package/dist/v2/runtime/runner/in-memory.mjs

@@ -0,0 +1,222 @@
+import "reflect-metadata";
+import { AgentRunner } from "./agent-runner.mjs";
+import { finalizeRunEvents } from "@copilotkit/shared";
+import { ReplaySubject } from "rxjs";
+import { EventType, compactEvents } from "@ag-ui/client";
+
+//#region src/v2/runtime/runner/in-memory.ts
+var InMemoryEventStore = class {
+	constructor(threadId) {
+		this.threadId = threadId;
+		this.subject = null;
+		this.isRunning = false;
+		this.currentRunId = null;
+		this.historicRuns = [];
+		this.agent = null;
+		this.runSubject = null;
+		this.stopRequested = false;
+		this.currentEvents = null;
+	}
+};
+const GLOBAL_STORE_KEY = Symbol.for("@copilotkit/runtime/in-memory-store");
+function getGlobalStore() {
+	const globalAny = globalThis;
+	if (!globalAny[GLOBAL_STORE_KEY]) globalAny[GLOBAL_STORE_KEY] = {
+		stores: /* @__PURE__ */ new Map(),
+		historicRunsBackup: /* @__PURE__ */ new Map()
+	};
+	const data = globalAny[GLOBAL_STORE_KEY];
+	if (data.stores.size === 0 && data.historicRunsBackup.size > 0) for (const [threadId, historicRuns] of data.historicRunsBackup) {
+		const store = new InMemoryEventStore(threadId);
+		store.historicRuns = historicRuns;
+		data.stores.set(threadId, store);
+	}
+	return data.stores;
+}
+function backupHistoricRuns(threadId, historicRuns) {
+	const globalAny = globalThis;
+	if (globalAny[GLOBAL_STORE_KEY]) globalAny[GLOBAL_STORE_KEY].historicRunsBackup.set(threadId, historicRuns);
+}
+const GLOBAL_STORE = getGlobalStore();
+var InMemoryAgentRunner = class extends AgentRunner {
+	run(request) {
+		let existingStore = GLOBAL_STORE.get(request.threadId);
+		if (!existingStore) {
+			existingStore = new InMemoryEventStore(request.threadId);
+			GLOBAL_STORE.set(request.threadId, existingStore);
+		}
+		const store = existingStore;
+		if (store.isRunning) throw new Error("Thread already running");
+		store.isRunning = true;
+		store.currentRunId = request.input.runId;
+		store.agent = request.agent;
+		store.stopRequested = false;
+		const seenMessageIds = /* @__PURE__ */ new Set();
+		const currentRunEvents = [];
+		store.currentEvents = currentRunEvents;
+		const historicMessageIds = /* @__PURE__ */ new Set();
+		for (const run of store.historicRuns) for (const event of run.events) {
+			if ("messageId" in event && typeof event.messageId === "string") historicMessageIds.add(event.messageId);
+			if (event.type === EventType.RUN_STARTED) {
+				const messages = event.input?.messages ?? [];
+				for (const message of messages) historicMessageIds.add(message.id);
+			}
+		}
+		const nextSubject = new ReplaySubject(Infinity);
+		const prevSubject = store.subject;
+		store.subject = nextSubject;
+		const runSubject = new ReplaySubject(Infinity);
+		store.runSubject = runSubject;
+		const runAgent = async () => {
+			const parentRunId = store.historicRuns[store.historicRuns.length - 1]?.runId ?? null;
+			try {
+				await request.agent.runAgent(request.input, {
+					onEvent: ({ event }) => {
+						let processedEvent = event;
+						if (event.type === EventType.RUN_STARTED) {
+							const runStartedEvent = event;
+							if (!runStartedEvent.input) {
+								const sanitizedMessages = request.input.messages ? request.input.messages.filter((message) => !historicMessageIds.has(message.id)) : void 0;
+								const updatedInput = {
+									...request.input,
+									...sanitizedMessages !== void 0 ? { messages: sanitizedMessages } : {}
+								};
+								processedEvent = {
+									...runStartedEvent,
+									input: updatedInput
+								};
+							}
+						}
+						runSubject.next(processedEvent);
+						nextSubject.next(processedEvent);
+						currentRunEvents.push(processedEvent);
+					},
+					onNewMessage: ({ message }) => {
+						if (!seenMessageIds.has(message.id)) seenMessageIds.add(message.id);
+					},
+					onRunStartedEvent: () => {
+						if (request.input.messages) {
+							for (const message of request.input.messages) if (!seenMessageIds.has(message.id)) seenMessageIds.add(message.id);
+						}
+					}
+				});
+				const appendedEvents = finalizeRunEvents(currentRunEvents, { stopRequested: store.stopRequested });
+				for (const event of appendedEvents) {
+					runSubject.next(event);
+					nextSubject.next(event);
+				}
+				if (store.currentRunId) {
+					const compactedEvents = compactEvents(currentRunEvents);
+					store.historicRuns.push({
+						threadId: request.threadId,
+						runId: store.currentRunId,
+						parentRunId,
+						events: compactedEvents,
+						createdAt: Date.now()
+					});
+					backupHistoricRuns(request.threadId, store.historicRuns);
+				}
+				store.currentEvents = null;
+				store.currentRunId = null;
+				store.agent = null;
+				store.runSubject = null;
+				store.stopRequested = false;
+				store.isRunning = false;
+				runSubject.complete();
+				nextSubject.complete();
+			} catch (error) {
+				const interruptionMessage = error instanceof Error ? error.message : String(error);
+				const appendedEvents = finalizeRunEvents(currentRunEvents, {
+					stopRequested: store.stopRequested,
+					interruptionMessage
+				});
+				for (const event of appendedEvents) {
+					runSubject.next(event);
+					nextSubject.next(event);
+				}
+				if (store.currentRunId && currentRunEvents.length > 0) {
+					const compactedEvents = compactEvents(currentRunEvents);
+					store.historicRuns.push({
+						threadId: request.threadId,
+						runId: store.currentRunId,
+						parentRunId,
+						events: compactedEvents,
+						createdAt: Date.now()
+					});
+					backupHistoricRuns(request.threadId, store.historicRuns);
+				}
+				store.currentEvents = null;
+				store.currentRunId = null;
+				store.agent = null;
+				store.runSubject = null;
+				store.stopRequested = false;
+				store.isRunning = false;
+				runSubject.complete();
+				nextSubject.complete();
+			}
+		};
+		if (prevSubject) prevSubject.subscribe({
+			next: (e) => nextSubject.next(e),
+			error: (err) => nextSubject.error(err),
+			complete: () => {}
+		});
+		runAgent();
+		return runSubject.asObservable();
+	}
+	connect(request) {
+		const store = GLOBAL_STORE.get(request.threadId);
+		const connectionSubject = new ReplaySubject(Infinity);
+		if (!store) {
+			connectionSubject.complete();
+			return connectionSubject.asObservable();
+		}
+		const allHistoricEvents = [];
+		for (const run of store.historicRuns) allHistoricEvents.push(...run.events);
+		const compactedEvents = compactEvents(allHistoricEvents);
+		const emittedMessageIds = /* @__PURE__ */ new Set();
+		for (const event of compactedEvents) {
+			connectionSubject.next(event);
+			if ("messageId" in event && typeof event.messageId === "string") emittedMessageIds.add(event.messageId);
+		}
+		if (store.subject && (store.isRunning || store.stopRequested)) store.subject.subscribe({
+			next: (event) => {
+				if ("messageId" in event && typeof event.messageId === "string" && emittedMessageIds.has(event.messageId)) return;
+				connectionSubject.next(event);
+			},
+			complete: () => connectionSubject.complete(),
+			error: (err) => connectionSubject.error(err)
+		});
+		else connectionSubject.complete();
+		return connectionSubject.asObservable();
+	}
+	isRunning(request) {
+		const store = GLOBAL_STORE.get(request.threadId);
+		return Promise.resolve(store?.isRunning ?? false);
+	}
+	stop(request) {
+		const store = GLOBAL_STORE.get(request.threadId);
+		if (!store || !store.isRunning) return Promise.resolve(false);
+		if (store.stopRequested) return Promise.resolve(false);
+		store.stopRequested = true;
+		store.isRunning = false;
+		const agent = store.agent;
+		if (!agent) {
+			store.stopRequested = false;
+			store.isRunning = false;
+			return Promise.resolve(false);
+		}
+		try {
+			agent.abortRun();
+			return Promise.resolve(true);
+		} catch (error) {
+			console.error("Failed to abort agent run", error);
+			store.stopRequested = false;
+			store.isRunning = true;
+			return Promise.resolve(false);
+		}
+	}
+};
+
+//#endregion
+export { InMemoryAgentRunner };
+//# sourceMappingURL=in-memory.mjs.map

package/dist/v2/runtime/runner/in-memory.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory.mjs","names":[],"sources":["../../../../src/v2/runtime/runner/in-memory.ts"],"sourcesContent":["import {\n  AgentRunner,\n  AgentRunnerConnectRequest,\n  AgentRunnerIsRunningRequest,\n  AgentRunnerRunRequest,\n  type AgentRunnerStopRequest,\n} from \"./agent-runner\";\nimport { Observable, ReplaySubject } from \"rxjs\";\nimport {\n  AbstractAgent,\n  BaseEvent,\n  EventType,\n  MessagesSnapshotEvent,\n  RunStartedEvent,\n  compactEvents,\n} from \"@ag-ui/client\";\nimport { finalizeRunEvents } from \"@copilotkit/shared\";\n\ninterface HistoricRun {\n  threadId: string;\n  runId: string;\n  parentRunId: string | null;\n  events: BaseEvent[];\n  createdAt: number;\n}\n\nclass InMemoryEventStore {\n  constructor(public threadId: string) {}\n\n  /** The subject that current consumers subscribe to. */\n  subject: ReplaySubject<BaseEvent> | null = null;\n\n  /** True while a run is actively producing events. */\n  isRunning = false;\n\n  /** Current run ID */\n  currentRunId: string | null = null;\n\n  /** Historic completed runs */\n  historicRuns: HistoricRun[] = [];\n\n  /** Currently running agent instance (if any). */\n  agent: AbstractAgent | null = null;\n\n  /** Subject returned from run() while the run is active. */\n  runSubject: ReplaySubject<BaseEvent> | null = null;\n\n  /** True once stop() has been requested but the run has not yet finalized. */\n  stopRequested = false;\n\n  /** Reference to the events emitted in the current run. */\n  currentEvents: BaseEvent[] | null = null;\n}\n\n// Use a symbol key on globalThis to survive hot reloads in development\nconst GLOBAL_STORE_KEY = Symbol.for(\"@copilotkit/runtime/in-memory-store\");\n\ninterface GlobalStoreData {\n  stores: Map<string, InMemoryEventStore>;\n  historicRunsBackup: Map<string, HistoricRun[]>;\n}\n\nfunction getGlobalStore(): Map<string, InMemoryEventStore> {\n  const globalAny = globalThis as unknown as Record<symbol, GlobalStoreData>;\n\n  if (!globalAny[GLOBAL_STORE_KEY]) {\n    globalAny[GLOBAL_STORE_KEY] = {\n      stores: new Map<string, InMemoryEventStore>(),\n      historicRunsBackup: new Map<string, HistoricRun[]>(),\n    };\n  }\n\n  const data = globalAny[GLOBAL_STORE_KEY];\n\n  // Restore historic runs from backup after hot reload\n  // (when stores map is empty but backup has data)\n  if (data.stores.size === 0 && data.historicRunsBackup.size > 0) {\n    for (const [threadId, historicRuns] of data.historicRunsBackup) {\n      const store = new InMemoryEventStore(threadId);\n      store.historicRuns = historicRuns;\n      data.stores.set(threadId, store);\n    }\n  }\n\n  return data.stores;\n}\n\nfunction backupHistoricRuns(\n  threadId: string,\n  historicRuns: HistoricRun[],\n): void {\n  const globalAny = globalThis as unknown as Record<symbol, GlobalStoreData>;\n  if (globalAny[GLOBAL_STORE_KEY]) {\n    globalAny[GLOBAL_STORE_KEY].historicRunsBackup.set(threadId, historicRuns);\n  }\n}\n\nconst GLOBAL_STORE = getGlobalStore();\n\nexport class InMemoryAgentRunner extends AgentRunner {\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    let existingStore = GLOBAL_STORE.get(request.threadId);\n    if (!existingStore) {\n      existingStore = new InMemoryEventStore(request.threadId);\n      GLOBAL_STORE.set(request.threadId, existingStore);\n    }\n    const store = existingStore; // Now store is const and non-null\n\n    if (store.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n    store.isRunning = true;\n    store.currentRunId = request.input.runId;\n    store.agent = request.agent;\n    store.stopRequested = false;\n\n    // Track seen message IDs and current run events for this run\n    const seenMessageIds = new Set<string>();\n    const currentRunEvents: BaseEvent[] = [];\n    store.currentEvents = currentRunEvents;\n\n    // Get all previously seen message IDs from historic runs\n    const historicMessageIds = new Set<string>();\n    for (const run of store.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    const nextSubject = new ReplaySubject<BaseEvent>(Infinity);\n    const prevSubject = store.subject;\n\n    // Update the store's subject immediately\n    store.subject = nextSubject;\n\n    // Create a subject for run() return value\n    const runSubject = new ReplaySubject<BaseEvent>(Infinity);\n    store.runSubject = runSubject;\n\n    // Helper function to run the agent and handle errors\n    const runAgent = async () => {\n      // Get parent run ID for chaining\n      const lastRun = store.historicRuns[store.historicRuns.length - 1];\n      const parentRunId = lastRun?.runId ?? null;\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 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 any messages from the input as seen so they aren't emitted twice\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 appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: store.stopRequested,\n        });\n        for (const event of appendedEvents) {\n          runSubject.next(event);\n          nextSubject.next(event);\n        }\n\n        // Store the completed run in memory with ONLY its events\n        if (store.currentRunId) {\n          // Compact the events before storing (like SQLite does)\n          const compactedEvents = compactEvents(currentRunEvents);\n\n          store.historicRuns.push({\n            threadId: request.threadId,\n            runId: store.currentRunId,\n            parentRunId,\n            events: compactedEvents,\n            createdAt: Date.now(),\n          });\n\n          // Backup for hot reload survival\n          backupHistoricRuns(request.threadId, store.historicRuns);\n        }\n\n        // Complete the run\n        store.currentEvents = null;\n        store.currentRunId = null;\n        store.agent = null;\n        store.runSubject = null;\n        store.stopRequested = false;\n        store.isRunning = false;\n        runSubject.complete();\n        nextSubject.complete();\n      } catch (error) {\n        const interruptionMessage =\n          error instanceof Error ? error.message : String(error);\n        const appendedEvents = finalizeRunEvents(currentRunEvents, {\n          stopRequested: store.stopRequested,\n          interruptionMessage,\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 (store.currentRunId && currentRunEvents.length > 0) {\n          // Compact the events before storing (like SQLite does)\n          const compactedEvents = compactEvents(currentRunEvents);\n          store.historicRuns.push({\n            threadId: request.threadId,\n            runId: store.currentRunId,\n            parentRunId,\n            events: compactedEvents,\n            createdAt: Date.now(),\n          });\n\n          // Backup for hot reload survival\n          backupHistoricRuns(request.threadId, store.historicRuns);\n        }\n\n        // Complete the run\n        store.currentEvents = null;\n        store.currentRunId = null;\n        store.agent = null;\n        store.runSubject = null;\n        store.stopRequested = false;\n        store.isRunning = false;\n        runSubject.complete();\n        nextSubject.complete();\n      }\n    };\n\n    // Bridge previous events if they exist\n    if (prevSubject) {\n      prevSubject.subscribe({\n        next: (e) => nextSubject.next(e),\n        error: (err) => nextSubject.error(err),\n        complete: () => {\n          // Don't complete nextSubject here - it needs to stay open for new events\n        },\n      });\n    }\n\n    // Start the agent execution immediately (not lazily)\n    runAgent();\n\n    // Return the run subject (only agent events, no injected messages)\n    return runSubject.asObservable();\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const store = GLOBAL_STORE.get(request.threadId);\n    const connectionSubject = new ReplaySubject<BaseEvent>(Infinity);\n\n    if (!store) {\n      // No store means no events\n      connectionSubject.complete();\n      return connectionSubject.asObservable();\n    }\n\n    // Collect all historic events from memory\n    const allHistoricEvents: BaseEvent[] = [];\n    for (const run of store.historicRuns) {\n      allHistoricEvents.push(...run.events);\n    }\n\n    // Apply compaction to all historic events together (like SQLite)\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    if (store.subject && (store.isRunning || store.stopRequested)) {\n      store.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 store = GLOBAL_STORE.get(request.threadId);\n    return Promise.resolve(store?.isRunning ?? false);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const store = GLOBAL_STORE.get(request.threadId);\n    if (!store || !store.isRunning) {\n      return Promise.resolve(false);\n    }\n    if (store.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    store.stopRequested = true;\n    store.isRunning = false;\n\n    const agent = store.agent;\n    if (!agent) {\n      store.stopRequested = false;\n      store.isRunning = false;\n      return Promise.resolve(false);\n    }\n\n    try {\n      agent.abortRun();\n      return Promise.resolve(true);\n    } catch (error) {\n      console.error(\"Failed to abort agent run\", error);\n      store.stopRequested = false;\n      store.isRunning = true;\n      return Promise.resolve(false);\n    }\n  }\n}\n"],"mappings":";;;;;;;AA0BA,IAAM,qBAAN,MAAyB;CACvB,YAAY,AAAO,UAAkB;EAAlB;iBAGwB;mBAG/B;sBAGkB;sBAGA,EAAE;eAGF;oBAGgB;uBAG9B;uBAGoB;;;AAItC,MAAM,mBAAmB,OAAO,IAAI,sCAAsC;AAO1E,SAAS,iBAAkD;CACzD,MAAM,YAAY;AAElB,KAAI,CAAC,UAAU,kBACb,WAAU,oBAAoB;EAC5B,wBAAQ,IAAI,KAAiC;EAC7C,oCAAoB,IAAI,KAA4B;EACrD;CAGH,MAAM,OAAO,UAAU;AAIvB,KAAI,KAAK,OAAO,SAAS,KAAK,KAAK,mBAAmB,OAAO,EAC3D,MAAK,MAAM,CAAC,UAAU,iBAAiB,KAAK,oBAAoB;EAC9D,MAAM,QAAQ,IAAI,mBAAmB,SAAS;AAC9C,QAAM,eAAe;AACrB,OAAK,OAAO,IAAI,UAAU,MAAM;;AAIpC,QAAO,KAAK;;AAGd,SAAS,mBACP,UACA,cACM;CACN,MAAM,YAAY;AAClB,KAAI,UAAU,kBACZ,WAAU,kBAAkB,mBAAmB,IAAI,UAAU,aAAa;;AAI9E,MAAM,eAAe,gBAAgB;AAErC,IAAa,sBAAb,cAAyC,YAAY;CACnD,IAAI,SAAuD;EACzD,IAAI,gBAAgB,aAAa,IAAI,QAAQ,SAAS;AACtD,MAAI,CAAC,eAAe;AAClB,mBAAgB,IAAI,mBAAmB,QAAQ,SAAS;AACxD,gBAAa,IAAI,QAAQ,UAAU,cAAc;;EAEnD,MAAM,QAAQ;AAEd,MAAI,MAAM,UACR,OAAM,IAAI,MAAM,yBAAyB;AAE3C,QAAM,YAAY;AAClB,QAAM,eAAe,QAAQ,MAAM;AACnC,QAAM,QAAQ,QAAQ;AACtB,QAAM,gBAAgB;EAGtB,MAAM,iCAAiB,IAAI,KAAa;EACxC,MAAM,mBAAgC,EAAE;AACxC,QAAM,gBAAgB;EAGtB,MAAM,qCAAqB,IAAI,KAAa;AAC5C,OAAK,MAAM,OAAO,MAAM,aACtB,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;;;EAM1C,MAAM,cAAc,IAAI,cAAyB,SAAS;EAC1D,MAAM,cAAc,MAAM;AAG1B,QAAM,UAAU;EAGhB,MAAM,aAAa,IAAI,cAAyB,SAAS;AACzD,QAAM,aAAa;EAGnB,MAAM,WAAW,YAAY;GAG3B,MAAM,cADU,MAAM,aAAa,MAAM,aAAa,SAAS,IAClC,SAAS;AAEtC,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,iBAAiB,kBAAkB,kBAAkB,EACzD,eAAe,MAAM,eACtB,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,MAAM,cAAc;KAEtB,MAAM,kBAAkB,cAAc,iBAAiB;AAEvD,WAAM,aAAa,KAAK;MACtB,UAAU,QAAQ;MAClB,OAAO,MAAM;MACb;MACA,QAAQ;MACR,WAAW,KAAK,KAAK;MACtB,CAAC;AAGF,wBAAmB,QAAQ,UAAU,MAAM,aAAa;;AAI1D,UAAM,gBAAgB;AACtB,UAAM,eAAe;AACrB,UAAM,QAAQ;AACd,UAAM,aAAa;AACnB,UAAM,gBAAgB;AACtB,UAAM,YAAY;AAClB,eAAW,UAAU;AACrB,gBAAY,UAAU;YACf,OAAO;IACd,MAAM,sBACJ,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACxD,MAAM,iBAAiB,kBAAkB,kBAAkB;KACzD,eAAe,MAAM;KACrB;KACD,CAAC;AACF,SAAK,MAAM,SAAS,gBAAgB;AAClC,gBAAW,KAAK,MAAM;AACtB,iBAAY,KAAK,MAAM;;AAIzB,QAAI,MAAM,gBAAgB,iBAAiB,SAAS,GAAG;KAErD,MAAM,kBAAkB,cAAc,iBAAiB;AACvD,WAAM,aAAa,KAAK;MACtB,UAAU,QAAQ;MAClB,OAAO,MAAM;MACb;MACA,QAAQ;MACR,WAAW,KAAK,KAAK;MACtB,CAAC;AAGF,wBAAmB,QAAQ,UAAU,MAAM,aAAa;;AAI1D,UAAM,gBAAgB;AACtB,UAAM,eAAe;AACrB,UAAM,QAAQ;AACd,UAAM,aAAa;AACnB,UAAM,gBAAgB;AACtB,UAAM,YAAY;AAClB,eAAW,UAAU;AACrB,gBAAY,UAAU;;;AAK1B,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,QAAQ,aAAa,IAAI,QAAQ,SAAS;EAChD,MAAM,oBAAoB,IAAI,cAAyB,SAAS;AAEhE,MAAI,CAAC,OAAO;AAEV,qBAAkB,UAAU;AAC5B,UAAO,kBAAkB,cAAc;;EAIzC,MAAM,oBAAiC,EAAE;AACzC,OAAK,MAAM,OAAO,MAAM,aACtB,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;;AAK1C,MAAI,MAAM,YAAY,MAAM,aAAa,MAAM,eAC7C,OAAM,QAAQ,UAAU;GACtB,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,QAAQ,aAAa,IAAI,QAAQ,SAAS;AAChD,SAAO,QAAQ,QAAQ,OAAO,aAAa,MAAM;;CAGnD,KAAK,SAA+D;EAClE,MAAM,QAAQ,aAAa,IAAI,QAAQ,SAAS;AAChD,MAAI,CAAC,SAAS,CAAC,MAAM,UACnB,QAAO,QAAQ,QAAQ,MAAM;AAE/B,MAAI,MAAM,cACR,QAAO,QAAQ,QAAQ,MAAM;AAG/B,QAAM,gBAAgB;AACtB,QAAM,YAAY;EAElB,MAAM,QAAQ,MAAM;AACpB,MAAI,CAAC,OAAO;AACV,SAAM,gBAAgB;AACtB,SAAM,YAAY;AAClB,UAAO,QAAQ,QAAQ,MAAM;;AAG/B,MAAI;AACF,SAAM,UAAU;AAChB,UAAO,QAAQ,QAAQ,KAAK;WACrB,OAAO;AACd,WAAQ,MAAM,6BAA6B,MAAM;AACjD,SAAM,gBAAgB;AACtB,SAAM,YAAY;AAClB,UAAO,QAAQ,QAAQ,MAAM"}

package/dist/v2/runtime/runner/index.d.cts

@@ -0,0 +1,6 @@
+
+import { AgentRunner, AgentRunnerConnectRequest, AgentRunnerIsRunningRequest, AgentRunnerRunRequest, AgentRunnerStopRequest } from "./agent-runner.cjs";
+import { InMemoryAgentRunner } from "./in-memory.cjs";
+import { IntelligenceAgentRunner, IntelligenceAgentRunnerOptions } from "./intelligence.cjs";
+import { finalizeRunEvents } from "@copilotkit/shared";
+export { finalizeRunEvents };

package/dist/v2/runtime/runner/index.d.mts

@@ -0,0 +1,6 @@
+import "reflect-metadata";
+import { AgentRunner, AgentRunnerConnectRequest, AgentRunnerIsRunningRequest, AgentRunnerRunRequest, AgentRunnerStopRequest } from "./agent-runner.mjs";
+import { InMemoryAgentRunner } from "./in-memory.mjs";
+import { IntelligenceAgentRunner, IntelligenceAgentRunnerOptions } from "./intelligence.mjs";
+import { finalizeRunEvents as finalizeRunEvents$1 } from "@copilotkit/shared";
+export { finalizeRunEvents$1 as finalizeRunEvents };

package/dist/v2/runtime/runner/index.mjs

@@ -0,0 +1,7 @@
+import "reflect-metadata";
+import { AgentRunner } from "./agent-runner.mjs";
+import { InMemoryAgentRunner } from "./in-memory.mjs";
+import { IntelligenceAgentRunner } from "./intelligence.mjs";
+import { finalizeRunEvents as finalizeRunEvents$1 } from "@copilotkit/shared";
+
+export { finalizeRunEvents$1 as finalizeRunEvents };

package/dist/v2/runtime/runner/intelligence.cjs

@@ -0,0 +1,246 @@
+require("reflect-metadata");
+const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
+const require_agent_runner = require('./agent-runner.cjs');
+let _copilotkit_shared = require("@copilotkit/shared");
+let rxjs = require("rxjs");
+let node_crypto = require("node:crypto");
+let _ag_ui_client = require("@ag-ui/client");
+let rxjs_operators = require("rxjs/operators");
+let phoenix = require("phoenix");
+
+//#region src/v2/runtime/runner/intelligence.ts
+var IntelligenceAgentRunner = class extends require_agent_runner.AgentRunner {
+	constructor(options) {
+		super();
+		this.threads = /* @__PURE__ */ new Map();
+		this.options = options;
+	}
+	/**
+	* Create a new Phoenix socket with explicit exponential backoff.
+	*
+	* Each run/connect gets its own socket so that:
+	*  - A socket failure only affects a single thread, not all threads.
+	*  - Cleanup is simple: channel.leave() + socket.disconnect() tears
+	*    down everything for that run with no shared-state concerns.
+	*  - Each run gets its own independent retry budget.
+	*
+	* reconnectAfterMs — delay before Phoenix reconnects the WebSocket
+	*   after an unclean close. 100ms base, doubling up to a 10s cap.
+	*
+	* rejoinAfterMs — delay before Phoenix re-joins a channel that
+	*   entered the "errored" state. 1s base, doubling up to 30s cap.
+	*
+	* These are set explicitly because Phoenix's default schedule is a
+	* fixed stepped array (not exponential), and any code that calls
+	* socket.disconnect() in an onError handler will set
+	* closeWasClean = true and reset the reconnect timer — permanently
+	* killing retries.
+	*/
+	createSocket() {
+		const socket = new phoenix.Socket(this.options.url, {
+			...this.options.authToken ? { authToken: this.options.authToken } : {},
+			reconnectAfterMs: (0, _copilotkit_shared.phoenixExponentialBackoff)(100, 1e4),
+			rejoinAfterMs: (0, _copilotkit_shared.phoenixExponentialBackoff)(1e3, 3e4)
+		});
+		socket.connect();
+		return socket;
+	}
+	createRunnerEventPayload(event, request, state) {
+		const payload = { ...this.stampRunnerMetadata(event, state) };
+		payload.thread_id ??= request.threadId;
+		const runId = payload.runId ?? payload.run_id ?? request.input.runId;
+		if (runId) payload.run_id = runId;
+		return payload;
+	}
+	stampRunnerMetadata(event, state) {
+		const eventRecord = event;
+		const existingMetadata = eventRecord.metadata ?? {};
+		const hasEventId = typeof existingMetadata.cpki_event_id === "string";
+		const hasEventSeq = typeof existingMetadata.cpki_event_seq === "number";
+		if (hasEventId && hasEventSeq) {
+			const eventSeq = existingMetadata.cpki_event_seq;
+			state.nextEventSeq = Math.max(state.nextEventSeq, eventSeq + 1);
+			return eventRecord;
+		}
+		const eventSeq = state.nextEventSeq++;
+		return {
+			...eventRecord,
+			metadata: {
+				...existingMetadata,
+				cpki_event_id: typeof existingMetadata.cpki_event_id === "string" ? existingMetadata.cpki_event_id : (0, node_crypto.randomUUID)(),
+				cpki_event_seq: eventSeq
+			}
+		};
+	}
+	run(request) {
+		const { threadId, agent, input, joinCode } = request;
+		if (this.threads.get(threadId)?.isRunning) throw new Error("Thread already running");
+		return new rxjs.Observable((observer) => {
+			const socket = this.createSocket();
+			const channelTopic = joinCode ?? threadId;
+			const channel = socket.channel(`ingestion:${channelTopic}`, { runId: input.runId });
+			const state = {
+				socket,
+				channel,
+				isRunning: true,
+				stopRequested: false,
+				agent,
+				currentEvents: [],
+				nextEventSeq: 1,
+				hasRunStarted: false
+			};
+			this.threads.set(threadId, state);
+			const MAX_CONSECUTIVE_ERRORS = 5;
+			let consecutiveErrors = 0;
+			socket.onError(() => {
+				consecutiveErrors++;
+				if (consecutiveErrors >= MAX_CONSECUTIVE_ERRORS && state.agent) try {
+					state.agent.abortRun();
+				} catch {}
+			});
+			socket.onOpen(() => {
+				consecutiveErrors = 0;
+			});
+			channel.on(_copilotkit_shared.AG_UI_CHANNEL_EVENT, (payload) => {
+				if (payload.type === _ag_ui_client.EventType.CUSTOM && payload.name === "stop") this.stop({ threadId });
+			});
+			channel.join().receive("ok", () => {
+				this.executeAgentRun(request, state, threadId).subscribe({ complete: () => observer.complete() });
+			}).receive("error", (resp) => {
+				const errorEvent = {
+					type: _ag_ui_client.EventType.RUN_ERROR,
+					message: `Failed to join channel: ${JSON.stringify(resp)}`,
+					code: "CHANNEL_JOIN_ERROR"
+				};
+				observer.next(errorEvent);
+				state.currentEvents.push(errorEvent);
+				this.removeThread(threadId);
+				observer.complete();
+			}).receive("timeout", () => {
+				const errorEvent = {
+					type: _ag_ui_client.EventType.RUN_ERROR,
+					message: "Timed out joining channel",
+					code: "CHANNEL_JOIN_TIMEOUT"
+				};
+				observer.next(errorEvent);
+				state.currentEvents.push(errorEvent);
+				this.removeThread(threadId);
+				observer.complete();
+			});
+			return () => {
+				this.removeThread(threadId);
+			};
+		});
+	}
+	connect(request) {
+		const { threadId } = request;
+		return new rxjs.Observable((observer) => {
+			const socket = this.createSocket();
+			const channel = socket.channel(`thread:${threadId}`);
+			channel.on("ag_ui_event", (payload) => {
+				observer.next(payload);
+				if (payload.type === _ag_ui_client.EventType.RUN_FINISHED || payload.type === _ag_ui_client.EventType.RUN_ERROR) observer.complete();
+			});
+			const cleanup = () => {
+				channel.leave();
+				socket.disconnect();
+			};
+			channel.join().receive("ok", () => void 0).receive("error", (resp) => {
+				observer.error(/* @__PURE__ */ new Error(`Failed to join channel: ${JSON.stringify(resp)}`));
+				cleanup();
+			}).receive("timeout", () => {
+				observer.error(/* @__PURE__ */ new Error("Timed out joining channel"));
+				cleanup();
+			});
+			return () => {
+				cleanup();
+			};
+		});
+	}
+	isRunning(request) {
+		const state = this.threads.get(request.threadId);
+		return Promise.resolve(state?.isRunning ?? false);
+	}
+	stop(request) {
+		const state = this.threads.get(request.threadId);
+		if (!state || !state.isRunning || state.stopRequested) return Promise.resolve(false);
+		state.stopRequested = true;
+		if (state.agent) try {
+			state.agent.abortRun();
+		} catch {}
+		return Promise.resolve(true);
+	}
+	executeAgentRun(request, state, threadId) {
+		const { currentEvents, channel } = state;
+		const pushCanonicalEvent = (event) => {
+			const canonicalEvent = this.stampRunnerMetadata(event, state);
+			currentEvents.push(canonicalEvent);
+			if (canonicalEvent.type === _ag_ui_client.EventType.RUN_STARTED) state.hasRunStarted = true;
+			channel.push("event", this.createRunnerEventPayload(canonicalEvent, request, state));
+		};
+		const getPersistedInputMessages = () => request.persistedInputMessages ?? request.input.messages;
+		const buildRunStartedEvent = (source) => {
+			const baseInput = source?.input ?? request.input;
+			const persistedInputMessages = getPersistedInputMessages();
+			return {
+				...source ?? {
+					type: _ag_ui_client.EventType.RUN_STARTED,
+					threadId: request.threadId,
+					runId: request.input.runId
+				},
+				input: {
+					...baseInput,
+					...persistedInputMessages !== void 0 ? { messages: persistedInputMessages } : {}
+				}
+			};
+		};
+		const ensureRunStarted = () => {
+			if (!state.hasRunStarted) {
+				state.hasRunStarted = true;
+				pushCanonicalEvent(buildRunStartedEvent());
+			}
+		};
+		return (0, rxjs.from)(request.agent.runAgent(request.input, { onEvent: ({ event }) => {
+			if (event.type === _ag_ui_client.EventType.RUN_STARTED) {
+				pushCanonicalEvent(buildRunStartedEvent(event));
+				return;
+			}
+			ensureRunStarted();
+			pushCanonicalEvent(event);
+		} })).pipe((0, rxjs_operators.catchError)((error) => {
+			ensureRunStarted();
+			pushCanonicalEvent({
+				type: _ag_ui_client.EventType.RUN_ERROR,
+				message: error instanceof Error ? error.message : String(error)
+			});
+			return rxjs.EMPTY;
+		}), (0, rxjs_operators.finalize)(() => {
+			ensureRunStarted();
+			const appended = (0, _copilotkit_shared.finalizeRunEvents)(currentEvents, { stopRequested: state.stopRequested });
+			for (const event of appended) channel.push("event", this.createRunnerEventPayload(event, request, state));
+			this.removeThread(threadId);
+		}));
+	}
+	/**
+	* Tear down all resources for a thread: leave the channel,
+	* disconnect the per-run socket, and remove the thread state.
+	*
+	* Idempotent — safe to call multiple times for the same threadId
+	* (e.g. from join error handlers, finalize, and Observable teardown).
+	*/
+	removeThread(threadId) {
+		const state = this.threads.get(threadId);
+		if (!state) return;
+		this.threads.delete(threadId);
+		try {
+			state.channel.leave();
+		} catch {}
+		try {
+			state.socket.disconnect();
+		} catch {}
+	}
+};
+
+//#endregion
+exports.IntelligenceAgentRunner = IntelligenceAgentRunner;
+//# sourceMappingURL=intelligence.cjs.map

package/dist/v2/runtime/runner/intelligence.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"intelligence.cjs","names":["AgentRunner","Socket","Observable","AG_UI_CHANNEL_EVENT","EventType","EMPTY"],"sources":["../../../../src/v2/runtime/runner/intelligence.ts"],"sourcesContent":["import {\n  AgentRunner,\n  AgentRunnerConnectRequest,\n  AgentRunnerIsRunningRequest,\n  AgentRunnerRunRequest,\n  type AgentRunnerStopRequest,\n} from \"./agent-runner\";\nimport { EMPTY, Observable, from } from \"rxjs\";\nimport { catchError, finalize } from \"rxjs/operators\";\nimport {\n  AbstractAgent,\n  BaseEvent,\n  EventType,\n  RunStartedEvent,\n} from \"@ag-ui/client\";\nimport {\n  finalizeRunEvents,\n  AG_UI_CHANNEL_EVENT,\n  phoenixExponentialBackoff,\n} from \"@copilotkit/shared\";\nimport { Socket, Channel } from \"phoenix\";\nimport { randomUUID } from \"node:crypto\";\n\nexport interface IntelligenceAgentRunnerOptions {\n  /** Phoenix runner websocket URL, e.g. \"ws://localhost:4000/runner\" */\n  url: string;\n  /** Optional Phoenix socket auth token used during websocket connect. */\n  authToken?: string;\n}\n\ninterface ThreadState {\n  socket: Socket;\n  channel: Channel;\n  isRunning: boolean;\n  stopRequested: boolean;\n  agent: AbstractAgent | null;\n  currentEvents: BaseEvent[];\n  nextEventSeq: number;\n  hasRunStarted: boolean;\n}\n\nexport class IntelligenceAgentRunner extends AgentRunner {\n  private options: IntelligenceAgentRunnerOptions;\n  private threads = new Map<string, ThreadState>();\n\n  constructor(options: IntelligenceAgentRunnerOptions) {\n    super();\n    // Store config — sockets are created per-run, not eagerly.\n    this.options = options;\n  }\n\n  /**\n   * Create a new Phoenix socket with explicit exponential backoff.\n   *\n   * Each run/connect gets its own socket so that:\n   *  - A socket failure only affects a single thread, not all threads.\n   *  - Cleanup is simple: channel.leave() + socket.disconnect() tears\n   *    down everything for that run with no shared-state concerns.\n   *  - Each run gets its own independent retry budget.\n   *\n   * reconnectAfterMs — delay before Phoenix reconnects the WebSocket\n   *   after an unclean close. 100ms base, doubling up to a 10s cap.\n   *\n   * rejoinAfterMs — delay before Phoenix re-joins a channel that\n   *   entered the \"errored\" state. 1s base, doubling up to 30s cap.\n   *\n   * These are set explicitly because Phoenix's default schedule is a\n   * fixed stepped array (not exponential), and any code that calls\n   * socket.disconnect() in an onError handler will set\n   * closeWasClean = true and reset the reconnect timer — permanently\n   * killing retries.\n   */\n  private createSocket(): Socket {\n    const socket = new Socket(this.options.url, {\n      ...(this.options.authToken ? { authToken: this.options.authToken } : {}),\n      reconnectAfterMs: phoenixExponentialBackoff(100, 10_000),\n      rejoinAfterMs: phoenixExponentialBackoff(1_000, 30_000),\n    });\n    socket.connect();\n    return socket;\n  }\n\n  private createRunnerEventPayload(\n    event: BaseEvent,\n    request: AgentRunnerRunRequest,\n    state: ThreadState,\n  ): Record<string, unknown> {\n    const canonicalEvent = this.stampRunnerMetadata(event, state);\n    const payload = {\n      ...(canonicalEvent as Record<string, unknown>),\n    };\n\n    payload.thread_id ??= request.threadId;\n\n    const runId = payload.runId ?? payload.run_id ?? request.input.runId;\n\n    if (runId) {\n      payload.run_id = runId;\n    }\n\n    return payload;\n  }\n\n  private stampRunnerMetadata(event: BaseEvent, state: ThreadState): BaseEvent {\n    const eventRecord = event as BaseEvent & {\n      metadata?: Record<string, unknown>;\n    };\n\n    const existingMetadata = eventRecord.metadata ?? {};\n    const hasEventId = typeof existingMetadata.cpki_event_id === \"string\";\n    const hasEventSeq = typeof existingMetadata.cpki_event_seq === \"number\";\n\n    if (hasEventId && hasEventSeq) {\n      const eventSeq = existingMetadata.cpki_event_seq as number;\n      state.nextEventSeq = Math.max(state.nextEventSeq, eventSeq + 1);\n      return eventRecord;\n    }\n\n    const eventSeq = state.nextEventSeq++;\n\n    return {\n      ...eventRecord,\n      metadata: {\n        ...existingMetadata,\n        cpki_event_id:\n          typeof existingMetadata.cpki_event_id === \"string\"\n            ? existingMetadata.cpki_event_id\n            : randomUUID(),\n        cpki_event_seq: eventSeq,\n      },\n    };\n  }\n\n  run(request: AgentRunnerRunRequest): Observable<BaseEvent> {\n    const { threadId, agent, input, joinCode } = request;\n\n    const existing = this.threads.get(threadId);\n    if (existing?.isRunning) {\n      throw new Error(\"Thread already running\");\n    }\n\n    return new Observable((observer) => {\n      const socket = this.createSocket();\n\n      const channelTopic = joinCode ?? threadId;\n      const channel = socket.channel(`ingestion:${channelTopic}`, {\n        runId: input.runId,\n      });\n\n      const state: ThreadState = {\n        socket,\n        channel,\n        isRunning: true,\n        stopRequested: false,\n        agent,\n        currentEvents: [],\n        nextEventSeq: 1,\n        hasRunStarted: false,\n      };\n      this.threads.set(threadId, state);\n\n      // Track consecutive socket errors for this run. Phoenix retries\n      // automatically via reconnectAfterMs, but if the connection fails\n      // repeatedly we abort the agent — otherwise runAgent() completes\n      // normally, finalization events buffer silently on the dead\n      // channel, and the client never receives them.\n      //\n      // Aborting the agent is the single trigger that cascades through\n      // the existing error pipeline: runAgent() rejects → catchError\n      // pushes RUN_ERROR → finalize calls finalizeRunEvents +\n      // removeThread → channel.leave() + socket.disconnect().\n      const MAX_CONSECUTIVE_ERRORS = 5;\n      let consecutiveErrors = 0;\n\n      socket.onError(() => {\n        consecutiveErrors++;\n        if (consecutiveErrors >= MAX_CONSECUTIVE_ERRORS && state.agent) {\n          try {\n            state.agent.abortRun();\n          } catch {\n            // Ignore abort errors.\n          }\n        }\n        // Otherwise: Phoenix retries automatically using the exponential\n        // backoff schedule configured in createSocket().\n      });\n\n      socket.onOpen(() => {\n        // A successful (re)connection resets the counter so transient\n        // network blips don't accumulate across recoveries.\n        consecutiveErrors = 0;\n      });\n\n      // Listen for custom \"stop\" events pushed by the client over the\n      // channel. This must be registered before channel.join() so the\n      // handler is in place by the time the server starts relaying messages.\n      // The client sends the stop event before leaving the channel, so the\n      // runner is guaranteed to receive it while still joined.\n      channel.on(AG_UI_CHANNEL_EVENT, (payload: BaseEvent) => {\n        if (\n          payload.type === EventType.CUSTOM &&\n          (payload as BaseEvent & { name?: string }).name === \"stop\"\n        ) {\n          this.stop({ threadId });\n        }\n      });\n\n      channel\n        .join()\n        .receive(\"ok\", () => {\n          this.executeAgentRun(request, state, threadId).subscribe({\n            complete: () => observer.complete(),\n          });\n        })\n        .receive(\"error\", (resp) => {\n          const errorEvent = {\n            type: EventType.RUN_ERROR,\n            message: `Failed to join channel: ${JSON.stringify(resp)}`,\n            code: \"CHANNEL_JOIN_ERROR\",\n          } as BaseEvent;\n          observer.next(errorEvent);\n          state.currentEvents.push(errorEvent);\n          this.removeThread(threadId);\n          observer.complete();\n        })\n        .receive(\"timeout\", () => {\n          const errorEvent = {\n            type: EventType.RUN_ERROR,\n            message: \"Timed out joining channel\",\n            code: \"CHANNEL_JOIN_TIMEOUT\",\n          } as BaseEvent;\n          observer.next(errorEvent);\n          state.currentEvents.push(errorEvent);\n          this.removeThread(threadId);\n          observer.complete();\n        });\n\n      return () => {\n        this.removeThread(threadId);\n      };\n    });\n  }\n\n  connect(request: AgentRunnerConnectRequest): Observable<BaseEvent> {\n    const { threadId } = request;\n\n    return new Observable((observer) => {\n      const socket = this.createSocket();\n\n      const channel = socket.channel(`thread:${threadId}`);\n\n      channel.on(\"ag_ui_event\", (payload: BaseEvent) => {\n        observer.next(payload);\n\n        if (\n          payload.type === EventType.RUN_FINISHED ||\n          payload.type === EventType.RUN_ERROR\n        ) {\n          observer.complete();\n        }\n      });\n\n      const cleanup = () => {\n        channel.leave();\n        socket.disconnect();\n      };\n\n      channel\n        .join()\n        .receive(\"ok\", () => undefined)\n        .receive(\"error\", (resp) => {\n          observer.error(\n            new Error(`Failed to join channel: ${JSON.stringify(resp)}`),\n          );\n          cleanup();\n        })\n        .receive(\"timeout\", () => {\n          observer.error(new Error(\"Timed out joining channel\"));\n          cleanup();\n        });\n\n      return () => {\n        cleanup();\n      };\n    });\n  }\n\n  isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean> {\n    const state = this.threads.get(request.threadId);\n    return Promise.resolve(state?.isRunning ?? false);\n  }\n\n  stop(request: AgentRunnerStopRequest): Promise<boolean | undefined> {\n    const state = this.threads.get(request.threadId);\n    if (!state || !state.isRunning || state.stopRequested) {\n      return Promise.resolve(false);\n    }\n\n    state.stopRequested = true;\n\n    // Direct local abort — the runtime is the authority.\n    if (state.agent) {\n      try {\n        state.agent.abortRun();\n      } catch {\n        // Ignore abort errors.\n      }\n    }\n\n    return Promise.resolve(true);\n  }\n\n  private executeAgentRun(\n    request: AgentRunnerRunRequest,\n    state: ThreadState,\n    threadId: string,\n  ): Observable<void> {\n    const { currentEvents, channel } = state;\n    const pushCanonicalEvent = (event: BaseEvent): void => {\n      const canonicalEvent = this.stampRunnerMetadata(event, state);\n      currentEvents.push(canonicalEvent);\n\n      if (canonicalEvent.type === EventType.RUN_STARTED) {\n        state.hasRunStarted = true;\n      }\n\n      channel.push(\n        \"event\",\n        this.createRunnerEventPayload(canonicalEvent, request, state),\n      );\n    };\n\n    const getPersistedInputMessages = () =>\n      request.persistedInputMessages ?? request.input.messages;\n\n    const buildRunStartedEvent = (\n      source?: RunStartedEvent,\n    ): RunStartedEvent => {\n      const baseInput = source?.input ?? request.input;\n      const persistedInputMessages = getPersistedInputMessages();\n\n      return {\n        ...(source ?? {\n          type: EventType.RUN_STARTED,\n          threadId: request.threadId,\n          runId: request.input.runId,\n        }),\n        input: {\n          ...baseInput,\n          ...(persistedInputMessages !== undefined\n            ? { messages: persistedInputMessages }\n            : {}),\n        },\n      } as RunStartedEvent;\n    };\n\n    const ensureRunStarted = (): void => {\n      if (!state.hasRunStarted) {\n        state.hasRunStarted = true;\n        pushCanonicalEvent(buildRunStartedEvent());\n      }\n    };\n\n    return from(\n      request.agent.runAgent(request.input, {\n        onEvent: ({ event }: { event: BaseEvent }) => {\n          if (event.type === EventType.RUN_STARTED) {\n            pushCanonicalEvent(buildRunStartedEvent(event as RunStartedEvent));\n            return;\n          }\n\n          ensureRunStarted();\n          pushCanonicalEvent(event);\n        },\n      }),\n    ).pipe(\n      catchError((error) => {\n        ensureRunStarted();\n        const errorEvent = {\n          type: EventType.RUN_ERROR,\n          message: error instanceof Error ? error.message : String(error),\n        } as BaseEvent;\n        pushCanonicalEvent(errorEvent);\n        return EMPTY;\n      }),\n      finalize(() => {\n        ensureRunStarted();\n        const appended = finalizeRunEvents(currentEvents, {\n          stopRequested: state.stopRequested,\n        });\n        for (const event of appended) {\n          channel.push(\n            \"event\",\n            this.createRunnerEventPayload(event, request, state),\n          );\n        }\n        this.removeThread(threadId);\n      }),\n    );\n  }\n\n  /**\n   * Tear down all resources for a thread: leave the channel,\n   * disconnect the per-run socket, and remove the thread state.\n   *\n   * Idempotent — safe to call multiple times for the same threadId\n   * (e.g. from join error handlers, finalize, and Observable teardown).\n   */\n  private removeThread(threadId: string): void {\n    const state = this.threads.get(threadId);\n    if (!state) {\n      return;\n    }\n\n    // Delete first so concurrent calls see the entry as already removed.\n    this.threads.delete(threadId);\n\n    try {\n      state.channel.leave();\n    } catch {\n      // Channel may already be closed/left.\n    }\n    try {\n      state.socket.disconnect();\n    } catch {\n      // Socket may already be disconnected.\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;AAyCA,IAAa,0BAAb,cAA6CA,iCAAY;CAIvD,YAAY,SAAyC;AACnD,SAAO;iCAHS,IAAI,KAA0B;AAK9C,OAAK,UAAU;;;;;;;;;;;;;;;;;;;;;;;CAwBjB,AAAQ,eAAuB;EAC7B,MAAM,SAAS,IAAIC,eAAO,KAAK,QAAQ,KAAK;GAC1C,GAAI,KAAK,QAAQ,YAAY,EAAE,WAAW,KAAK,QAAQ,WAAW,GAAG,EAAE;GACvE,oEAA4C,KAAK,IAAO;GACxD,iEAAyC,KAAO,IAAO;GACxD,CAAC;AACF,SAAO,SAAS;AAChB,SAAO;;CAGT,AAAQ,yBACN,OACA,SACA,OACyB;EAEzB,MAAM,UAAU,EACd,GAFqB,KAAK,oBAAoB,OAAO,MAAM,EAG5D;AAED,UAAQ,cAAc,QAAQ;EAE9B,MAAM,QAAQ,QAAQ,SAAS,QAAQ,UAAU,QAAQ,MAAM;AAE/D,MAAI,MACF,SAAQ,SAAS;AAGnB,SAAO;;CAGT,AAAQ,oBAAoB,OAAkB,OAA+B;EAC3E,MAAM,cAAc;EAIpB,MAAM,mBAAmB,YAAY,YAAY,EAAE;EACnD,MAAM,aAAa,OAAO,iBAAiB,kBAAkB;EAC7D,MAAM,cAAc,OAAO,iBAAiB,mBAAmB;AAE/D,MAAI,cAAc,aAAa;GAC7B,MAAM,WAAW,iBAAiB;AAClC,SAAM,eAAe,KAAK,IAAI,MAAM,cAAc,WAAW,EAAE;AAC/D,UAAO;;EAGT,MAAM,WAAW,MAAM;AAEvB,SAAO;GACL,GAAG;GACH,UAAU;IACR,GAAG;IACH,eACE,OAAO,iBAAiB,kBAAkB,WACtC,iBAAiB,6CACL;IAClB,gBAAgB;IACjB;GACF;;CAGH,IAAI,SAAuD;EACzD,MAAM,EAAE,UAAU,OAAO,OAAO,aAAa;AAG7C,MADiB,KAAK,QAAQ,IAAI,SAAS,EAC7B,UACZ,OAAM,IAAI,MAAM,yBAAyB;AAG3C,SAAO,IAAIC,iBAAY,aAAa;GAClC,MAAM,SAAS,KAAK,cAAc;GAElC,MAAM,eAAe,YAAY;GACjC,MAAM,UAAU,OAAO,QAAQ,aAAa,gBAAgB,EAC1D,OAAO,MAAM,OACd,CAAC;GAEF,MAAM,QAAqB;IACzB;IACA;IACA,WAAW;IACX,eAAe;IACf;IACA,eAAe,EAAE;IACjB,cAAc;IACd,eAAe;IAChB;AACD,QAAK,QAAQ,IAAI,UAAU,MAAM;GAYjC,MAAM,yBAAyB;GAC/B,IAAI,oBAAoB;AAExB,UAAO,cAAc;AACnB;AACA,QAAI,qBAAqB,0BAA0B,MAAM,MACvD,KAAI;AACF,WAAM,MAAM,UAAU;YAChB;KAMV;AAEF,UAAO,aAAa;AAGlB,wBAAoB;KACpB;AAOF,WAAQ,GAAGC,yCAAsB,YAAuB;AACtD,QACE,QAAQ,SAASC,wBAAU,UAC1B,QAA0C,SAAS,OAEpD,MAAK,KAAK,EAAE,UAAU,CAAC;KAEzB;AAEF,WACG,MAAM,CACN,QAAQ,YAAY;AACnB,SAAK,gBAAgB,SAAS,OAAO,SAAS,CAAC,UAAU,EACvD,gBAAgB,SAAS,UAAU,EACpC,CAAC;KACF,CACD,QAAQ,UAAU,SAAS;IAC1B,MAAM,aAAa;KACjB,MAAMA,wBAAU;KAChB,SAAS,2BAA2B,KAAK,UAAU,KAAK;KACxD,MAAM;KACP;AACD,aAAS,KAAK,WAAW;AACzB,UAAM,cAAc,KAAK,WAAW;AACpC,SAAK,aAAa,SAAS;AAC3B,aAAS,UAAU;KACnB,CACD,QAAQ,iBAAiB;IACxB,MAAM,aAAa;KACjB,MAAMA,wBAAU;KAChB,SAAS;KACT,MAAM;KACP;AACD,aAAS,KAAK,WAAW;AACzB,UAAM,cAAc,KAAK,WAAW;AACpC,SAAK,aAAa,SAAS;AAC3B,aAAS,UAAU;KACnB;AAEJ,gBAAa;AACX,SAAK,aAAa,SAAS;;IAE7B;;CAGJ,QAAQ,SAA2D;EACjE,MAAM,EAAE,aAAa;AAErB,SAAO,IAAIF,iBAAY,aAAa;GAClC,MAAM,SAAS,KAAK,cAAc;GAElC,MAAM,UAAU,OAAO,QAAQ,UAAU,WAAW;AAEpD,WAAQ,GAAG,gBAAgB,YAAuB;AAChD,aAAS,KAAK,QAAQ;AAEtB,QACE,QAAQ,SAASE,wBAAU,gBAC3B,QAAQ,SAASA,wBAAU,UAE3B,UAAS,UAAU;KAErB;GAEF,MAAM,gBAAgB;AACpB,YAAQ,OAAO;AACf,WAAO,YAAY;;AAGrB,WACG,MAAM,CACN,QAAQ,YAAY,OAAU,CAC9B,QAAQ,UAAU,SAAS;AAC1B,aAAS,sBACP,IAAI,MAAM,2BAA2B,KAAK,UAAU,KAAK,GAAG,CAC7D;AACD,aAAS;KACT,CACD,QAAQ,iBAAiB;AACxB,aAAS,sBAAM,IAAI,MAAM,4BAA4B,CAAC;AACtD,aAAS;KACT;AAEJ,gBAAa;AACX,aAAS;;IAEX;;CAGJ,UAAU,SAAwD;EAChE,MAAM,QAAQ,KAAK,QAAQ,IAAI,QAAQ,SAAS;AAChD,SAAO,QAAQ,QAAQ,OAAO,aAAa,MAAM;;CAGnD,KAAK,SAA+D;EAClE,MAAM,QAAQ,KAAK,QAAQ,IAAI,QAAQ,SAAS;AAChD,MAAI,CAAC,SAAS,CAAC,MAAM,aAAa,MAAM,cACtC,QAAO,QAAQ,QAAQ,MAAM;AAG/B,QAAM,gBAAgB;AAGtB,MAAI,MAAM,MACR,KAAI;AACF,SAAM,MAAM,UAAU;UAChB;AAKV,SAAO,QAAQ,QAAQ,KAAK;;CAG9B,AAAQ,gBACN,SACA,OACA,UACkB;EAClB,MAAM,EAAE,eAAe,YAAY;EACnC,MAAM,sBAAsB,UAA2B;GACrD,MAAM,iBAAiB,KAAK,oBAAoB,OAAO,MAAM;AAC7D,iBAAc,KAAK,eAAe;AAElC,OAAI,eAAe,SAASA,wBAAU,YACpC,OAAM,gBAAgB;AAGxB,WAAQ,KACN,SACA,KAAK,yBAAyB,gBAAgB,SAAS,MAAM,CAC9D;;EAGH,MAAM,kCACJ,QAAQ,0BAA0B,QAAQ,MAAM;EAElD,MAAM,wBACJ,WACoB;GACpB,MAAM,YAAY,QAAQ,SAAS,QAAQ;GAC3C,MAAM,yBAAyB,2BAA2B;AAE1D,UAAO;IACL,GAAI,UAAU;KACZ,MAAMA,wBAAU;KAChB,UAAU,QAAQ;KAClB,OAAO,QAAQ,MAAM;KACtB;IACD,OAAO;KACL,GAAG;KACH,GAAI,2BAA2B,SAC3B,EAAE,UAAU,wBAAwB,GACpC,EAAE;KACP;IACF;;EAGH,MAAM,yBAA+B;AACnC,OAAI,CAAC,MAAM,eAAe;AACxB,UAAM,gBAAgB;AACtB,uBAAmB,sBAAsB,CAAC;;;AAI9C,wBACE,QAAQ,MAAM,SAAS,QAAQ,OAAO,EACpC,UAAU,EAAE,YAAkC;AAC5C,OAAI,MAAM,SAASA,wBAAU,aAAa;AACxC,uBAAmB,qBAAqB,MAAyB,CAAC;AAClE;;AAGF,qBAAkB;AAClB,sBAAmB,MAAM;KAE5B,CAAC,CACH,CAAC,qCACY,UAAU;AACpB,qBAAkB;AAKlB,sBAJmB;IACjB,MAAMA,wBAAU;IAChB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IAChE,CAC6B;AAC9B,UAAOC;IACP,qCACa;AACb,qBAAkB;GAClB,MAAM,qDAA6B,eAAe,EAChD,eAAe,MAAM,eACtB,CAAC;AACF,QAAK,MAAM,SAAS,SAClB,SAAQ,KACN,SACA,KAAK,yBAAyB,OAAO,SAAS,MAAM,CACrD;AAEH,QAAK,aAAa,SAAS;IAC3B,CACH;;;;;;;;;CAUH,AAAQ,aAAa,UAAwB;EAC3C,MAAM,QAAQ,KAAK,QAAQ,IAAI,SAAS;AACxC,MAAI,CAAC,MACH;AAIF,OAAK,QAAQ,OAAO,SAAS;AAE7B,MAAI;AACF,SAAM,QAAQ,OAAO;UACf;AAGR,MAAI;AACF,SAAM,OAAO,YAAY;UACnB"}