@copilotkit/runtime 1.56.3 → 1.56.4-canary.1777531098

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

@@ -80,12 +80,32 @@ var IntelligenceAgentRunner = class extends AgentRunner {
 		};
 	}
 	run(request) {
-		const { threadId, agent, input, joinCode } = request;
+		return this.createRunObservable(request);
+	}
+	runWithStartupBoundary(request) {
+		let resolveStartup;
+		let rejectStartup;
+		const startup = new Promise((resolve, reject) => {
+			resolveStartup = resolve;
+			rejectStartup = reject;
+		});
+		return {
+			events: this.createRunObservable(request, {
+				resolveStartup: () => resolveStartup?.(),
+				rejectStartup: (error) => rejectStartup?.(error)
+			}),
+			startup
+		};
+	}
+	createRunObservable(request, startupBoundary) {
+		const { threadId, agent, input } = request;
 		if (this.threads.get(threadId)?.isRunning) throw new Error("Thread already running");
 		return new Observable((observer) => {
 			const socket = this.createSocket();
-			const channelTopic = joinCode ?? threadId;
-			const channel = socket.channel(`ingestion:${channelTopic}`, { runId: input.runId });
+			const channel = socket.channel(`ingestion:${input.runId}`, {
+				thread_id: threadId,
+				run_id: input.runId
+			});
 			const state = {
 				socket,
 				channel,
@@ -112,26 +132,31 @@ var IntelligenceAgentRunner = class extends AgentRunner {
 				if (payload.type === EventType.CUSTOM && payload.name === "stop") this.stop({ threadId });
 			});
 			channel.join().receive("ok", () => {
+				startupBoundary?.resolveStartup();
 				this.executeAgentRun(request, state, threadId).subscribe({ complete: () => observer.complete() });
 			}).receive("error", (resp) => {
+				const error = /* @__PURE__ */ new Error(`Failed to join channel: ${JSON.stringify(resp)}`);
 				const errorEvent = {
 					type: EventType.RUN_ERROR,
-					message: `Failed to join channel: ${JSON.stringify(resp)}`,
+					message: error.message,
 					code: "CHANNEL_JOIN_ERROR"
 				};
 				observer.next(errorEvent);
 				state.currentEvents.push(errorEvent);
 				this.removeThread(threadId);
+				startupBoundary?.rejectStartup(error);
 				observer.complete();
 			}).receive("timeout", () => {
+				const error = /* @__PURE__ */ new Error("Timed out joining channel");
 				const errorEvent = {
 					type: EventType.RUN_ERROR,
-					message: "Timed out joining channel",
+					message: error.message,
 					code: "CHANNEL_JOIN_TIMEOUT"
 				};
 				observer.next(errorEvent);
 				state.currentEvents.push(errorEvent);
 				this.removeThread(threadId);
+				startupBoundary?.rejectStartup(error);
 				observer.complete();
 			});
 			return () => {

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

@@ -1 +1 @@
-{"version":3,"file":"intelligence.mjs","names":["randomUUID","catchError","finalize"],"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  /** Max delay (ms) for WebSocket reconnect backoff. @default 10_000 */\n  maxReconnectMs?: number;\n  /** Max delay (ms) for channel rejoin backoff. @default 30_000 */\n  maxRejoinMs?: number;\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 maxReconnectMs (default 10s).\n   *\n   * rejoinAfterMs — delay before Phoenix re-joins a channel that\n   *   entered the \"errored\" state. 1s base, doubling up to maxRejoinMs (default 30s).\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(\n        100,\n        this.options.maxReconnectMs ?? 10_000,\n      ),\n      rejoinAfterMs: phoenixExponentialBackoff(\n        1_000,\n        this.options.maxRejoinMs ?? 30_000,\n      ),\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(\n      this.stampCanonicalRunOwnership(event, request),\n      state,\n    );\n    const payload = {\n      ...(canonicalEvent as Record<string, unknown>),\n    };\n\n    payload.threadId = request.threadId;\n    payload.runId = request.input.runId;\n    payload.thread_id = request.threadId;\n    payload.run_id = request.input.runId;\n\n    return payload;\n  }\n\n  private stampCanonicalRunOwnership(\n    event: BaseEvent,\n    request: AgentRunnerRunRequest,\n  ): BaseEvent {\n    return {\n      ...(event as BaseEvent & Record<string, unknown>),\n      threadId: request.threadId,\n      runId: request.input.runId,\n    } as BaseEvent;\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(\n        this.stampCanonicalRunOwnership(event, request),\n        state,\n      );\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        threadId: request.threadId,\n        runId: request.input.runId,\n        input: {\n          ...baseInput,\n          threadId: request.threadId,\n          runId: request.input.runId,\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":";;;;;;;;;;AA6CA,IAAa,0BAAb,cAA6C,YAAY;CAIvD,YAAY,SAAyC;AACnD,SAAO;iCAHS,IAAI,KAA0B;AAK9C,OAAK,UAAU;;;;;;;;;;;;;;;;;;;;;;;CAwBjB,AAAQ,eAAuB;EAC7B,MAAM,SAAS,IAAI,OAAO,KAAK,QAAQ,KAAK;GAC1C,GAAI,KAAK,QAAQ,YAAY,EAAE,WAAW,KAAK,QAAQ,WAAW,GAAG,EAAE;GACvE,kBAAkB,0BAChB,KACA,KAAK,QAAQ,kBAAkB,IAChC;GACD,eAAe,0BACb,KACA,KAAK,QAAQ,eAAe,IAC7B;GACF,CAAC;AACF,SAAO,SAAS;AAChB,SAAO;;CAGT,AAAQ,yBACN,OACA,SACA,OACyB;EAKzB,MAAM,UAAU,EACd,GALqB,KAAK,oBAC1B,KAAK,2BAA2B,OAAO,QAAQ,EAC/C,MACD,EAGA;AAED,UAAQ,WAAW,QAAQ;AAC3B,UAAQ,QAAQ,QAAQ,MAAM;AAC9B,UAAQ,YAAY,QAAQ;AAC5B,UAAQ,SAAS,QAAQ,MAAM;AAE/B,SAAO;;CAGT,AAAQ,2BACN,OACA,SACW;AACX,SAAO;GACL,GAAI;GACJ,UAAU,QAAQ;GAClB,OAAO,QAAQ,MAAM;GACtB;;CAGH,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,gBACjBA,cAAY;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,IAAI,YAAY,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,GAAG,sBAAsB,YAAuB;AACtD,QACE,QAAQ,SAAS,UAAU,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,MAAM,UAAU;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,MAAM,UAAU;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,IAAI,YAAY,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,SAAS,UAAU,gBAC3B,QAAQ,SAAS,UAAU,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,oBAC1B,KAAK,2BAA2B,OAAO,QAAQ,EAC/C,MACD;AACD,iBAAc,KAAK,eAAe;AAElC,OAAI,eAAe,SAAS,UAAU,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,MAAM,UAAU;KAChB,UAAU,QAAQ;KAClB,OAAO,QAAQ,MAAM;KACtB;IACD,UAAU,QAAQ;IAClB,OAAO,QAAQ,MAAM;IACrB,OAAO;KACL,GAAG;KACH,UAAU,QAAQ;KAClB,OAAO,QAAQ,MAAM;KACrB,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,SAAO,KACL,QAAQ,MAAM,SAAS,QAAQ,OAAO,EACpC,UAAU,EAAE,YAAkC;AAC5C,OAAI,MAAM,SAAS,UAAU,aAAa;AACxC,uBAAmB,qBAAqB,MAAyB,CAAC;AAClE;;AAGF,qBAAkB;AAClB,sBAAmB,MAAM;KAE5B,CAAC,CACH,CAAC,KACAC,cAAY,UAAU;AACpB,qBAAkB;AAKlB,sBAJmB;IACjB,MAAM,UAAU;IAChB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IAChE,CAC6B;AAC9B,UAAO;IACP,EACFC,iBAAe;AACb,qBAAkB;GAClB,MAAM,WAAW,kBAAkB,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"}
+{"version":3,"file":"intelligence.mjs","names":["randomUUID","catchError","finalize"],"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  /** Max delay (ms) for WebSocket reconnect backoff. @default 10_000 */\n  maxReconnectMs?: number;\n  /** Max delay (ms) for channel rejoin backoff. @default 30_000 */\n  maxRejoinMs?: number;\n}\n\nexport interface RunnerStartupBoundary {\n  events: Observable<BaseEvent>;\n  startup: Promise<void>;\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 maxReconnectMs (default 10s).\n   *\n   * rejoinAfterMs — delay before Phoenix re-joins a channel that\n   *   entered the \"errored\" state. 1s base, doubling up to maxRejoinMs (default 30s).\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(\n        100,\n        this.options.maxReconnectMs ?? 10_000,\n      ),\n      rejoinAfterMs: phoenixExponentialBackoff(\n        1_000,\n        this.options.maxRejoinMs ?? 30_000,\n      ),\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(\n      this.stampCanonicalRunOwnership(event, request),\n      state,\n    );\n    const payload = {\n      ...(canonicalEvent as Record<string, unknown>),\n    };\n\n    payload.threadId = request.threadId;\n    payload.runId = request.input.runId;\n    payload.thread_id = request.threadId;\n    payload.run_id = request.input.runId;\n\n    return payload;\n  }\n\n  private stampCanonicalRunOwnership(\n    event: BaseEvent,\n    request: AgentRunnerRunRequest,\n  ): BaseEvent {\n    return {\n      ...(event as BaseEvent & Record<string, unknown>),\n      threadId: request.threadId,\n      runId: request.input.runId,\n    } as BaseEvent;\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    return this.createRunObservable(request);\n  }\n\n  runWithStartupBoundary(\n    request: AgentRunnerRunRequest,\n  ): RunnerStartupBoundary {\n    let resolveStartup: (() => void) | undefined;\n    let rejectStartup: ((reason: Error) => void) | undefined;\n    const startup = new Promise<void>((resolve, reject) => {\n      resolveStartup = resolve;\n      rejectStartup = reject;\n    });\n\n    return {\n      events: this.createRunObservable(request, {\n        resolveStartup: () => resolveStartup?.(),\n        rejectStartup: (error) => rejectStartup?.(error),\n      }),\n      startup,\n    };\n  }\n\n  private createRunObservable(\n    request: AgentRunnerRunRequest,\n    startupBoundary?: {\n      resolveStartup: () => void;\n      rejectStartup: (error: Error) => void;\n    },\n  ): Observable<BaseEvent> {\n    const { threadId, agent, input } = 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 channel = socket.channel(`ingestion:${input.runId}`, {\n        thread_id: threadId,\n        run_id: 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          startupBoundary?.resolveStartup();\n          this.executeAgentRun(request, state, threadId).subscribe({\n            complete: () => observer.complete(),\n          });\n        })\n        .receive(\"error\", (resp) => {\n          const error = new Error(\n            `Failed to join channel: ${JSON.stringify(resp)}`,\n          );\n          const errorEvent = {\n            type: EventType.RUN_ERROR,\n            message: error.message,\n            code: \"CHANNEL_JOIN_ERROR\",\n          } as BaseEvent;\n          observer.next(errorEvent);\n          state.currentEvents.push(errorEvent);\n          this.removeThread(threadId);\n          startupBoundary?.rejectStartup(error);\n          observer.complete();\n        })\n        .receive(\"timeout\", () => {\n          const error = new Error(\"Timed out joining channel\");\n          const errorEvent = {\n            type: EventType.RUN_ERROR,\n            message: error.message,\n            code: \"CHANNEL_JOIN_TIMEOUT\",\n          } as BaseEvent;\n          observer.next(errorEvent);\n          state.currentEvents.push(errorEvent);\n          this.removeThread(threadId);\n          startupBoundary?.rejectStartup(error);\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(\n        this.stampCanonicalRunOwnership(event, request),\n        state,\n      );\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        threadId: request.threadId,\n        runId: request.input.runId,\n        input: {\n          ...baseInput,\n          threadId: request.threadId,\n          runId: request.input.runId,\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":";;;;;;;;;;AAkDA,IAAa,0BAAb,cAA6C,YAAY;CAIvD,YAAY,SAAyC;AACnD,SAAO;iCAHS,IAAI,KAA0B;AAK9C,OAAK,UAAU;;;;;;;;;;;;;;;;;;;;;;;CAwBjB,AAAQ,eAAuB;EAC7B,MAAM,SAAS,IAAI,OAAO,KAAK,QAAQ,KAAK;GAC1C,GAAI,KAAK,QAAQ,YAAY,EAAE,WAAW,KAAK,QAAQ,WAAW,GAAG,EAAE;GACvE,kBAAkB,0BAChB,KACA,KAAK,QAAQ,kBAAkB,IAChC;GACD,eAAe,0BACb,KACA,KAAK,QAAQ,eAAe,IAC7B;GACF,CAAC;AACF,SAAO,SAAS;AAChB,SAAO;;CAGT,AAAQ,yBACN,OACA,SACA,OACyB;EAKzB,MAAM,UAAU,EACd,GALqB,KAAK,oBAC1B,KAAK,2BAA2B,OAAO,QAAQ,EAC/C,MACD,EAGA;AAED,UAAQ,WAAW,QAAQ;AAC3B,UAAQ,QAAQ,QAAQ,MAAM;AAC9B,UAAQ,YAAY,QAAQ;AAC5B,UAAQ,SAAS,QAAQ,MAAM;AAE/B,SAAO;;CAGT,AAAQ,2BACN,OACA,SACW;AACX,SAAO;GACL,GAAI;GACJ,UAAU,QAAQ;GAClB,OAAO,QAAQ,MAAM;GACtB;;CAGH,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,gBACjBA,cAAY;IAClB,gBAAgB;IACjB;GACF;;CAGH,IAAI,SAAuD;AACzD,SAAO,KAAK,oBAAoB,QAAQ;;CAG1C,uBACE,SACuB;EACvB,IAAI;EACJ,IAAI;EACJ,MAAM,UAAU,IAAI,SAAe,SAAS,WAAW;AACrD,oBAAiB;AACjB,mBAAgB;IAChB;AAEF,SAAO;GACL,QAAQ,KAAK,oBAAoB,SAAS;IACxC,sBAAsB,kBAAkB;IACxC,gBAAgB,UAAU,gBAAgB,MAAM;IACjD,CAAC;GACF;GACD;;CAGH,AAAQ,oBACN,SACA,iBAIuB;EACvB,MAAM,EAAE,UAAU,OAAO,UAAU;AAGnC,MADiB,KAAK,QAAQ,IAAI,SAAS,EAC7B,UACZ,OAAM,IAAI,MAAM,yBAAyB;AAG3C,SAAO,IAAI,YAAY,aAAa;GAClC,MAAM,SAAS,KAAK,cAAc;GAElC,MAAM,UAAU,OAAO,QAAQ,aAAa,MAAM,SAAS;IACzD,WAAW;IACX,QAAQ,MAAM;IACf,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,GAAG,sBAAsB,YAAuB;AACtD,QACE,QAAQ,SAAS,UAAU,UAC1B,QAA0C,SAAS,OAEpD,MAAK,KAAK,EAAE,UAAU,CAAC;KAEzB;AAEF,WACG,MAAM,CACN,QAAQ,YAAY;AACnB,qBAAiB,gBAAgB;AACjC,SAAK,gBAAgB,SAAS,OAAO,SAAS,CAAC,UAAU,EACvD,gBAAgB,SAAS,UAAU,EACpC,CAAC;KACF,CACD,QAAQ,UAAU,SAAS;IAC1B,MAAM,wBAAQ,IAAI,MAChB,2BAA2B,KAAK,UAAU,KAAK,GAChD;IACD,MAAM,aAAa;KACjB,MAAM,UAAU;KAChB,SAAS,MAAM;KACf,MAAM;KACP;AACD,aAAS,KAAK,WAAW;AACzB,UAAM,cAAc,KAAK,WAAW;AACpC,SAAK,aAAa,SAAS;AAC3B,qBAAiB,cAAc,MAAM;AACrC,aAAS,UAAU;KACnB,CACD,QAAQ,iBAAiB;IACxB,MAAM,wBAAQ,IAAI,MAAM,4BAA4B;IACpD,MAAM,aAAa;KACjB,MAAM,UAAU;KAChB,SAAS,MAAM;KACf,MAAM;KACP;AACD,aAAS,KAAK,WAAW;AACzB,UAAM,cAAc,KAAK,WAAW;AACpC,SAAK,aAAa,SAAS;AAC3B,qBAAiB,cAAc,MAAM;AACrC,aAAS,UAAU;KACnB;AAEJ,gBAAa;AACX,SAAK,aAAa,SAAS;;IAE7B;;CAGJ,QAAQ,SAA2D;EACjE,MAAM,EAAE,aAAa;AAErB,SAAO,IAAI,YAAY,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,SAAS,UAAU,gBAC3B,QAAQ,SAAS,UAAU,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,oBAC1B,KAAK,2BAA2B,OAAO,QAAQ,EAC/C,MACD;AACD,iBAAc,KAAK,eAAe;AAElC,OAAI,eAAe,SAAS,UAAU,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,MAAM,UAAU;KAChB,UAAU,QAAQ;KAClB,OAAO,QAAQ,MAAM;KACtB;IACD,UAAU,QAAQ;IAClB,OAAO,QAAQ,MAAM;IACrB,OAAO;KACL,GAAG;KACH,UAAU,QAAQ;KAClB,OAAO,QAAQ,MAAM;KACrB,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,SAAO,KACL,QAAQ,MAAM,SAAS,QAAQ,OAAO,EACpC,UAAU,EAAE,YAAkC;AAC5C,OAAI,MAAM,SAAS,UAAU,aAAa;AACxC,uBAAmB,qBAAqB,MAAyB,CAAC;AAClE;;AAGF,qBAAkB;AAClB,sBAAmB,MAAM;KAE5B,CAAC,CACH,CAAC,KACAC,cAAY,UAAU;AACpB,qBAAkB;AAKlB,sBAJmB;IACjB,MAAM,UAAU;IAChB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IAChE,CAC6B;AAC9B,UAAO;IACP,EACFC,iBAAe;AACb,qBAAkB;GAClB,MAAM,WAAW,kBAAkB,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"}

package/dist/v2/runtime/telemetry/instance-created.cjs

@@ -0,0 +1,33 @@
+require("reflect-metadata");
+const require_telemetry_client = require('./telemetry-client.cjs');
+
+//#region src/v2/runtime/telemetry/instance-created.ts
+/**
+* Fire the `oss.runtime.instance_created` telemetry event for a v2 runtime
+* handler. Called once per handler factory invocation (not per request).
+*
+* v2 does not have a concept of remote endpoints or standalone actions, so
+* those counts are 0 / []. `cloud.api_key_provided` is false at this level
+* because in v2 the cloud public key arrives per-request via the
+* `x-copilotcloud-public-api-key` header — not at handler creation time.
+* See `handlers/handle-run.ts` for the per-request event that DOES carry
+* the key when present.
+*
+* Errors resolving agents are swallowed — telemetry must never break
+* runtime setup.
+*/
+function fireInstanceCreatedTelemetry({ runtime }) {
+	(typeof runtime.agents === "function" ? Promise.resolve(null) : Promise.resolve(runtime.agents)).then((agents) => {
+		require_telemetry_client.default.capture("oss.runtime.instance_created", {
+			actionsAmount: 0,
+			endpointTypes: [],
+			endpointsAmount: 0,
+			agentsAmount: agents ? Object.keys(agents).length : null,
+			"cloud.api_key_provided": false
+		});
+	}).catch(() => {});
+}
+
+//#endregion
+exports.fireInstanceCreatedTelemetry = fireInstanceCreatedTelemetry;
+//# sourceMappingURL=instance-created.cjs.map

package/dist/v2/runtime/telemetry/instance-created.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"instance-created.cjs","names":[],"sources":["../../../../src/v2/runtime/telemetry/instance-created.ts"],"sourcesContent":["import { telemetry } from \".\";\nimport type { CopilotRuntimeLike } from \"../core/runtime\";\n\n/**\n * Fire the `oss.runtime.instance_created` telemetry event for a v2 runtime\n * handler. Called once per handler factory invocation (not per request).\n *\n * v2 does not have a concept of remote endpoints or standalone actions, so\n * those counts are 0 / []. `cloud.api_key_provided` is false at this level\n * because in v2 the cloud public key arrives per-request via the\n * `x-copilotcloud-public-api-key` header — not at handler creation time.\n * See `handlers/handle-run.ts` for the per-request event that DOES carry\n * the key when present.\n *\n * Errors resolving agents are swallowed — telemetry must never break\n * runtime setup.\n */\nexport function fireInstanceCreatedTelemetry({\n  runtime,\n}: {\n  runtime: CopilotRuntimeLike;\n}): void {\n  // agents can be a static Record, a Promise, or a per-request factory.\n  // Factory configs cannot be resolved at handler-creation time (no Request\n  // context), so report agentsAmount as null in that case.\n  const agentsPromise =\n    typeof runtime.agents === \"function\"\n      ? Promise.resolve<Record<string, unknown> | null>(null)\n      : Promise.resolve(runtime.agents);\n\n  agentsPromise\n    .then((agents) => {\n      telemetry.capture(\"oss.runtime.instance_created\", {\n        actionsAmount: 0,\n        endpointTypes: [],\n        endpointsAmount: 0,\n        agentsAmount: agents ? Object.keys(agents).length : null,\n        \"cloud.api_key_provided\": false,\n      });\n    })\n    .catch(() => {\n      // Swallow — telemetry must not break runtime creation.\n    });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAiBA,SAAgB,6BAA6B,EAC3C,WAGO;AASP,EAJE,OAAO,QAAQ,WAAW,aACtB,QAAQ,QAAwC,KAAK,GACrD,QAAQ,QAAQ,QAAQ,OAAO,EAGlC,MAAM,WAAW;AAChB,mCAAU,QAAQ,gCAAgC;GAChD,eAAe;GACf,eAAe,EAAE;GACjB,iBAAiB;GACjB,cAAc,SAAS,OAAO,KAAK,OAAO,CAAC,SAAS;GACpD,0BAA0B;GAC3B,CAAC;GACF,CACD,YAAY,GAEX"}

package/dist/v2/runtime/telemetry/instance-created.mjs

@@ -0,0 +1,33 @@
+import "reflect-metadata";
+import telemetry from "./telemetry-client.mjs";
+
+//#region src/v2/runtime/telemetry/instance-created.ts
+/**
+* Fire the `oss.runtime.instance_created` telemetry event for a v2 runtime
+* handler. Called once per handler factory invocation (not per request).
+*
+* v2 does not have a concept of remote endpoints or standalone actions, so
+* those counts are 0 / []. `cloud.api_key_provided` is false at this level
+* because in v2 the cloud public key arrives per-request via the
+* `x-copilotcloud-public-api-key` header — not at handler creation time.
+* See `handlers/handle-run.ts` for the per-request event that DOES carry
+* the key when present.
+*
+* Errors resolving agents are swallowed — telemetry must never break
+* runtime setup.
+*/
+function fireInstanceCreatedTelemetry({ runtime }) {
+	(typeof runtime.agents === "function" ? Promise.resolve(null) : Promise.resolve(runtime.agents)).then((agents) => {
+		telemetry.capture("oss.runtime.instance_created", {
+			actionsAmount: 0,
+			endpointTypes: [],
+			endpointsAmount: 0,
+			agentsAmount: agents ? Object.keys(agents).length : null,
+			"cloud.api_key_provided": false
+		});
+	}).catch(() => {});
+}
+
+//#endregion
+export { fireInstanceCreatedTelemetry };
+//# sourceMappingURL=instance-created.mjs.map

package/dist/v2/runtime/telemetry/instance-created.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"instance-created.mjs","names":[],"sources":["../../../../src/v2/runtime/telemetry/instance-created.ts"],"sourcesContent":["import { telemetry } from \".\";\nimport type { CopilotRuntimeLike } from \"../core/runtime\";\n\n/**\n * Fire the `oss.runtime.instance_created` telemetry event for a v2 runtime\n * handler. Called once per handler factory invocation (not per request).\n *\n * v2 does not have a concept of remote endpoints or standalone actions, so\n * those counts are 0 / []. `cloud.api_key_provided` is false at this level\n * because in v2 the cloud public key arrives per-request via the\n * `x-copilotcloud-public-api-key` header — not at handler creation time.\n * See `handlers/handle-run.ts` for the per-request event that DOES carry\n * the key when present.\n *\n * Errors resolving agents are swallowed — telemetry must never break\n * runtime setup.\n */\nexport function fireInstanceCreatedTelemetry({\n  runtime,\n}: {\n  runtime: CopilotRuntimeLike;\n}): void {\n  // agents can be a static Record, a Promise, or a per-request factory.\n  // Factory configs cannot be resolved at handler-creation time (no Request\n  // context), so report agentsAmount as null in that case.\n  const agentsPromise =\n    typeof runtime.agents === \"function\"\n      ? Promise.resolve<Record<string, unknown> | null>(null)\n      : Promise.resolve(runtime.agents);\n\n  agentsPromise\n    .then((agents) => {\n      telemetry.capture(\"oss.runtime.instance_created\", {\n        actionsAmount: 0,\n        endpointTypes: [],\n        endpointsAmount: 0,\n        agentsAmount: agents ? Object.keys(agents).length : null,\n        \"cloud.api_key_provided\": false,\n      });\n    })\n    .catch(() => {\n      // Swallow — telemetry must not break runtime creation.\n    });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAiBA,SAAgB,6BAA6B,EAC3C,WAGO;AASP,EAJE,OAAO,QAAQ,WAAW,aACtB,QAAQ,QAAwC,KAAK,GACrD,QAAQ,QAAQ,QAAQ,OAAO,EAGlC,MAAM,WAAW;AAChB,YAAU,QAAQ,gCAAgC;GAChD,eAAe;GACf,eAAe,EAAE;GACjB,iBAAiB;GACjB,cAAc,SAAS,OAAO,KAAK,OAAO,CAAC,SAAS;GACpD,0BAA0B;GAC3B,CAAC;GACF,CACD,YAAY,GAEX"}

package/dist/v2/runtime/telemetry/telemetry-client.cjs

@@ -1,9 +1,5 @@
 require("reflect-metadata");
-const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
-const require_utils = require('./utils.cjs');
 const require_scarf_client = require('./scarf-client.cjs');
-let _segment_analytics_node = require("@segment/analytics-node");
-let uuid = require("uuid");
 
 //#region src/v2/runtime/telemetry/telemetry-client.ts
 function isTelemetryDisabled() {
@@ -11,58 +7,25 @@ function isTelemetryDisabled() {
 }
 var TelemetryClient = class {
 	constructor({ telemetryDisabled, sampleRate } = {}) {
-		this.globalProperties = {};
 		this.telemetryDisabled = false;
 		this.sampleRate = .05;
-		this.anonymousId = `anon_${(0, uuid.v4)()}`;
 		this.telemetryDisabled = telemetryDisabled ?? isTelemetryDisabled();
-		if (this.telemetryDisabled) {
-			this.setSampleRate(sampleRate);
-			return;
-		}
 		this.setSampleRate(sampleRate);
-		this.segment = new _segment_analytics_node.Analytics({ writeKey: process.env.COPILOTKIT_SEGMENT_WRITE_KEY || "n7XAZtQCGS2v1vvBy3LgBCv2h3Y8whja" });
 	}
 	shouldSendEvent() {
 		if (this.telemetryDisabled) return false;
 		return Math.random() < this.sampleRate;
 	}
-	async capture(event, properties) {
+	async capture(event, _properties) {
 		if (!this.shouldSendEvent()) return;
-		const flattenedProperties = require_utils.flattenObject(properties);
-		const propertiesWithGlobal = {
-			...this.globalProperties,
-			...flattenedProperties
-		};
-		const orderedPropertiesWithGlobal = Object.keys(propertiesWithGlobal).sort().reduce((obj, key) => {
-			obj[key] = propertiesWithGlobal[key];
-			return obj;
-		}, {});
-		if (this.segment) this.segment.track({
-			anonymousId: this.anonymousId,
-			event,
-			properties: { ...orderedPropertiesWithGlobal }
-		});
 		await require_scarf_client.default.logEvent({ event });
 	}
-	setGlobalProperties(properties) {
-		const flattenedProperties = require_utils.flattenObject(properties);
-		this.globalProperties = {
-			...this.globalProperties,
-			...flattenedProperties
-		};
-	}
 	setSampleRate(sampleRate) {
 		let _sampleRate;
 		_sampleRate = sampleRate ?? .05;
 		if (process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE) _sampleRate = parseFloat(process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE);
 		if (_sampleRate < 0 || _sampleRate > 1) throw new Error("Sample rate must be between 0 and 1");
 		this.sampleRate = _sampleRate;
-		this.setGlobalProperties({
-			sampleRate: this.sampleRate,
-			sampleRateAdjustmentFactor: 1 - this.sampleRate,
-			sampleWeight: 1 / this.sampleRate
-		});
 	}
 };
 const telemetry = new TelemetryClient();

package/dist/v2/runtime/telemetry/telemetry-client.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"telemetry-client.cjs","names":["Analytics","flattenObject","scarfClient"],"sources":["../../../../src/v2/runtime/telemetry/telemetry-client.ts"],"sourcesContent":["import { Analytics } from \"@segment/analytics-node\";\nimport { AnalyticsEvents } from \"./events\";\nimport { flattenObject } from \"./utils\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport scarfClient from \"./scarf-client\";\n\nexport function isTelemetryDisabled(): boolean {\n  return (\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"true\" ||\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"1\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK ===\n      \"true\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK === \"1\"\n  );\n}\n\nexport class TelemetryClient {\n  segment: Analytics | undefined;\n  globalProperties: Record<string, any> = {};\n  private telemetryDisabled: boolean = false;\n  private sampleRate: number = 0.05;\n  private anonymousId = `anon_${uuidv4()}`;\n\n  constructor({\n    telemetryDisabled,\n    sampleRate,\n  }: {\n    telemetryDisabled?: boolean;\n    sampleRate?: number;\n  } = {}) {\n    this.telemetryDisabled = telemetryDisabled ?? isTelemetryDisabled();\n\n    if (this.telemetryDisabled) {\n      this.setSampleRate(sampleRate);\n      return;\n    }\n\n    this.setSampleRate(sampleRate);\n\n    const writeKey =\n      process.env.COPILOTKIT_SEGMENT_WRITE_KEY ||\n      \"n7XAZtQCGS2v1vvBy3LgBCv2h3Y8whja\";\n\n    this.segment = new Analytics({\n      writeKey,\n    });\n  }\n\n  private shouldSendEvent() {\n    if (this.telemetryDisabled) {\n      return false;\n    }\n    const randomNumber = Math.random();\n    return randomNumber < this.sampleRate;\n  }\n\n  async capture<K extends keyof AnalyticsEvents>(\n    event: K,\n    properties: AnalyticsEvents[K],\n  ) {\n    if (!this.shouldSendEvent()) {\n      return;\n    }\n\n    const flattenedProperties = flattenObject(properties);\n    const propertiesWithGlobal = {\n      ...this.globalProperties,\n      ...flattenedProperties,\n    };\n    const orderedPropertiesWithGlobal = Object.keys(propertiesWithGlobal)\n      .sort()\n      .reduce(\n        (obj, key) => {\n          obj[key] = propertiesWithGlobal[key];\n          return obj;\n        },\n        {} as Record<string, any>,\n      );\n\n    if (this.segment) {\n      this.segment.track({\n        anonymousId: this.anonymousId,\n        event,\n        properties: { ...orderedPropertiesWithGlobal },\n      });\n    }\n\n    await scarfClient.logEvent({\n      event,\n    });\n  }\n\n  setGlobalProperties(properties: Record<string, any>) {\n    const flattenedProperties = flattenObject(properties);\n    this.globalProperties = {\n      ...this.globalProperties,\n      ...flattenedProperties,\n    };\n  }\n\n  private setSampleRate(sampleRate: number | undefined) {\n    let _sampleRate: number;\n\n    _sampleRate = sampleRate ?? 0.05;\n\n    if (process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE) {\n      _sampleRate = parseFloat(process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE);\n    }\n\n    if (_sampleRate < 0 || _sampleRate > 1) {\n      throw new Error(\"Sample rate must be between 0 and 1\");\n    }\n\n    this.sampleRate = _sampleRate;\n    this.setGlobalProperties({\n      sampleRate: this.sampleRate,\n      sampleRateAdjustmentFactor: 1 - this.sampleRate,\n      sampleWeight: 1 / this.sampleRate,\n    });\n  }\n}\n\nconst telemetry = new TelemetryClient();\nexport default telemetry;\n"],"mappings":";;;;;;;;AAMA,SAAgB,sBAA+B;AAC7C,QACG,QAAQ,IACN,kCAAkC,UACpC,QAAQ,IACN,kCAAkC,OACpC,QAAQ,IAA2C,iBAClD,UACD,QAAQ,IAA2C,iBAAiB;;AAIzE,IAAa,kBAAb,MAA6B;CAO3B,YAAY,EACV,mBACA,eAIE,EAAE,EAAE;0BAXgC,EAAE;2BACL;oBACR;qBACP,sBAAgB;AASpC,OAAK,oBAAoB,qBAAqB,qBAAqB;AAEnE,MAAI,KAAK,mBAAmB;AAC1B,QAAK,cAAc,WAAW;AAC9B;;AAGF,OAAK,cAAc,WAAW;AAM9B,OAAK,UAAU,IAAIA,kCAAU,EAC3B,UAJA,QAAQ,IAAI,gCACZ,oCAID,CAAC;;CAGJ,AAAQ,kBAAkB;AACxB,MAAI,KAAK,kBACP,QAAO;AAGT,SADqB,KAAK,QAAQ,GACZ,KAAK;;CAG7B,MAAM,QACJ,OACA,YACA;AACA,MAAI,CAAC,KAAK,iBAAiB,CACzB;EAGF,MAAM,sBAAsBC,4BAAc,WAAW;EACrD,MAAM,uBAAuB;GAC3B,GAAG,KAAK;GACR,GAAG;GACJ;EACD,MAAM,8BAA8B,OAAO,KAAK,qBAAqB,CAClE,MAAM,CACN,QACE,KAAK,QAAQ;AACZ,OAAI,OAAO,qBAAqB;AAChC,UAAO;KAET,EAAE,CACH;AAEH,MAAI,KAAK,QACP,MAAK,QAAQ,MAAM;GACjB,aAAa,KAAK;GAClB;GACA,YAAY,EAAE,GAAG,6BAA6B;GAC/C,CAAC;AAGJ,QAAMC,6BAAY,SAAS,EACzB,OACD,CAAC;;CAGJ,oBAAoB,YAAiC;EACnD,MAAM,sBAAsBD,4BAAc,WAAW;AACrD,OAAK,mBAAmB;GACtB,GAAG,KAAK;GACR,GAAG;GACJ;;CAGH,AAAQ,cAAc,YAAgC;EACpD,IAAI;AAEJ,gBAAc,cAAc;AAE5B,MAAI,QAAQ,IAAI,iCACd,eAAc,WAAW,QAAQ,IAAI,iCAAiC;AAGxE,MAAI,cAAc,KAAK,cAAc,EACnC,OAAM,IAAI,MAAM,sCAAsC;AAGxD,OAAK,aAAa;AAClB,OAAK,oBAAoB;GACvB,YAAY,KAAK;GACjB,4BAA4B,IAAI,KAAK;GACrC,cAAc,IAAI,KAAK;GACxB,CAAC;;;AAIN,MAAM,YAAY,IAAI,iBAAiB"}
+{"version":3,"file":"telemetry-client.cjs","names":["scarfClient"],"sources":["../../../../src/v2/runtime/telemetry/telemetry-client.ts"],"sourcesContent":["import { AnalyticsEvents } from \"./events\";\nimport scarfClient from \"./scarf-client\";\n\nexport function isTelemetryDisabled(): boolean {\n  return (\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"true\" ||\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"1\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK ===\n      \"true\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK === \"1\"\n  );\n}\n\nexport class TelemetryClient {\n  private telemetryDisabled: boolean = false;\n  private sampleRate: number = 0.05;\n\n  constructor({\n    telemetryDisabled,\n    sampleRate,\n  }: {\n    telemetryDisabled?: boolean;\n    sampleRate?: number;\n  } = {}) {\n    this.telemetryDisabled = telemetryDisabled ?? isTelemetryDisabled();\n    this.setSampleRate(sampleRate);\n  }\n\n  private shouldSendEvent() {\n    if (this.telemetryDisabled) {\n      return false;\n    }\n    const randomNumber = Math.random();\n    return randomNumber < this.sampleRate;\n  }\n\n  async capture<K extends keyof AnalyticsEvents>(\n    event: K,\n    _properties: AnalyticsEvents[K],\n  ) {\n    if (!this.shouldSendEvent()) {\n      return;\n    }\n\n    await scarfClient.logEvent({\n      event,\n    });\n  }\n\n  private setSampleRate(sampleRate: number | undefined) {\n    let _sampleRate: number;\n\n    _sampleRate = sampleRate ?? 0.05;\n\n    if (process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE) {\n      _sampleRate = parseFloat(process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE);\n    }\n\n    if (_sampleRate < 0 || _sampleRate > 1) {\n      throw new Error(\"Sample rate must be between 0 and 1\");\n    }\n\n    this.sampleRate = _sampleRate;\n  }\n}\n\nconst telemetry = new TelemetryClient();\nexport default telemetry;\n"],"mappings":";;;;AAGA,SAAgB,sBAA+B;AAC7C,QACG,QAAQ,IACN,kCAAkC,UACpC,QAAQ,IACN,kCAAkC,OACpC,QAAQ,IAA2C,iBAClD,UACD,QAAQ,IAA2C,iBAAiB;;AAIzE,IAAa,kBAAb,MAA6B;CAI3B,YAAY,EACV,mBACA,eAIE,EAAE,EAAE;2BAT6B;oBACR;AAS3B,OAAK,oBAAoB,qBAAqB,qBAAqB;AACnE,OAAK,cAAc,WAAW;;CAGhC,AAAQ,kBAAkB;AACxB,MAAI,KAAK,kBACP,QAAO;AAGT,SADqB,KAAK,QAAQ,GACZ,KAAK;;CAG7B,MAAM,QACJ,OACA,aACA;AACA,MAAI,CAAC,KAAK,iBAAiB,CACzB;AAGF,QAAMA,6BAAY,SAAS,EACzB,OACD,CAAC;;CAGJ,AAAQ,cAAc,YAAgC;EACpD,IAAI;AAEJ,gBAAc,cAAc;AAE5B,MAAI,QAAQ,IAAI,iCACd,eAAc,WAAW,QAAQ,IAAI,iCAAiC;AAGxE,MAAI,cAAc,KAAK,cAAc,EACnC,OAAM,IAAI,MAAM,sCAAsC;AAGxD,OAAK,aAAa;;;AAItB,MAAM,YAAY,IAAI,iBAAiB"}

package/dist/v2/runtime/telemetry/telemetry-client.mjs

@@ -1,8 +1,5 @@
 import "reflect-metadata";
-import { flattenObject } from "./utils.mjs";
 import scarf_client_default from "./scarf-client.mjs";
-import { Analytics } from "@segment/analytics-node";
-import { v4 } from "uuid";
 
 //#region src/v2/runtime/telemetry/telemetry-client.ts
 function isTelemetryDisabled() {
@@ -10,58 +7,25 @@ function isTelemetryDisabled() {
 }
 var TelemetryClient = class {
 	constructor({ telemetryDisabled, sampleRate } = {}) {
-		this.globalProperties = {};
 		this.telemetryDisabled = false;
 		this.sampleRate = .05;
-		this.anonymousId = `anon_${v4()}`;
 		this.telemetryDisabled = telemetryDisabled ?? isTelemetryDisabled();
-		if (this.telemetryDisabled) {
-			this.setSampleRate(sampleRate);
-			return;
-		}
 		this.setSampleRate(sampleRate);
-		this.segment = new Analytics({ writeKey: process.env.COPILOTKIT_SEGMENT_WRITE_KEY || "n7XAZtQCGS2v1vvBy3LgBCv2h3Y8whja" });
 	}
 	shouldSendEvent() {
 		if (this.telemetryDisabled) return false;
 		return Math.random() < this.sampleRate;
 	}
-	async capture(event, properties) {
+	async capture(event, _properties) {
 		if (!this.shouldSendEvent()) return;
-		const flattenedProperties = flattenObject(properties);
-		const propertiesWithGlobal = {
-			...this.globalProperties,
-			...flattenedProperties
-		};
-		const orderedPropertiesWithGlobal = Object.keys(propertiesWithGlobal).sort().reduce((obj, key) => {
-			obj[key] = propertiesWithGlobal[key];
-			return obj;
-		}, {});
-		if (this.segment) this.segment.track({
-			anonymousId: this.anonymousId,
-			event,
-			properties: { ...orderedPropertiesWithGlobal }
-		});
 		await scarf_client_default.logEvent({ event });
 	}
-	setGlobalProperties(properties) {
-		const flattenedProperties = flattenObject(properties);
-		this.globalProperties = {
-			...this.globalProperties,
-			...flattenedProperties
-		};
-	}
 	setSampleRate(sampleRate) {
 		let _sampleRate;
 		_sampleRate = sampleRate ?? .05;
 		if (process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE) _sampleRate = parseFloat(process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE);
 		if (_sampleRate < 0 || _sampleRate > 1) throw new Error("Sample rate must be between 0 and 1");
 		this.sampleRate = _sampleRate;
-		this.setGlobalProperties({
-			sampleRate: this.sampleRate,
-			sampleRateAdjustmentFactor: 1 - this.sampleRate,
-			sampleWeight: 1 / this.sampleRate
-		});
 	}
 };
 const telemetry = new TelemetryClient();

package/dist/v2/runtime/telemetry/telemetry-client.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"telemetry-client.mjs","names":["uuidv4","scarfClient"],"sources":["../../../../src/v2/runtime/telemetry/telemetry-client.ts"],"sourcesContent":["import { Analytics } from \"@segment/analytics-node\";\nimport { AnalyticsEvents } from \"./events\";\nimport { flattenObject } from \"./utils\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport scarfClient from \"./scarf-client\";\n\nexport function isTelemetryDisabled(): boolean {\n  return (\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"true\" ||\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"1\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK ===\n      \"true\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK === \"1\"\n  );\n}\n\nexport class TelemetryClient {\n  segment: Analytics | undefined;\n  globalProperties: Record<string, any> = {};\n  private telemetryDisabled: boolean = false;\n  private sampleRate: number = 0.05;\n  private anonymousId = `anon_${uuidv4()}`;\n\n  constructor({\n    telemetryDisabled,\n    sampleRate,\n  }: {\n    telemetryDisabled?: boolean;\n    sampleRate?: number;\n  } = {}) {\n    this.telemetryDisabled = telemetryDisabled ?? isTelemetryDisabled();\n\n    if (this.telemetryDisabled) {\n      this.setSampleRate(sampleRate);\n      return;\n    }\n\n    this.setSampleRate(sampleRate);\n\n    const writeKey =\n      process.env.COPILOTKIT_SEGMENT_WRITE_KEY ||\n      \"n7XAZtQCGS2v1vvBy3LgBCv2h3Y8whja\";\n\n    this.segment = new Analytics({\n      writeKey,\n    });\n  }\n\n  private shouldSendEvent() {\n    if (this.telemetryDisabled) {\n      return false;\n    }\n    const randomNumber = Math.random();\n    return randomNumber < this.sampleRate;\n  }\n\n  async capture<K extends keyof AnalyticsEvents>(\n    event: K,\n    properties: AnalyticsEvents[K],\n  ) {\n    if (!this.shouldSendEvent()) {\n      return;\n    }\n\n    const flattenedProperties = flattenObject(properties);\n    const propertiesWithGlobal = {\n      ...this.globalProperties,\n      ...flattenedProperties,\n    };\n    const orderedPropertiesWithGlobal = Object.keys(propertiesWithGlobal)\n      .sort()\n      .reduce(\n        (obj, key) => {\n          obj[key] = propertiesWithGlobal[key];\n          return obj;\n        },\n        {} as Record<string, any>,\n      );\n\n    if (this.segment) {\n      this.segment.track({\n        anonymousId: this.anonymousId,\n        event,\n        properties: { ...orderedPropertiesWithGlobal },\n      });\n    }\n\n    await scarfClient.logEvent({\n      event,\n    });\n  }\n\n  setGlobalProperties(properties: Record<string, any>) {\n    const flattenedProperties = flattenObject(properties);\n    this.globalProperties = {\n      ...this.globalProperties,\n      ...flattenedProperties,\n    };\n  }\n\n  private setSampleRate(sampleRate: number | undefined) {\n    let _sampleRate: number;\n\n    _sampleRate = sampleRate ?? 0.05;\n\n    if (process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE) {\n      _sampleRate = parseFloat(process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE);\n    }\n\n    if (_sampleRate < 0 || _sampleRate > 1) {\n      throw new Error(\"Sample rate must be between 0 and 1\");\n    }\n\n    this.sampleRate = _sampleRate;\n    this.setGlobalProperties({\n      sampleRate: this.sampleRate,\n      sampleRateAdjustmentFactor: 1 - this.sampleRate,\n      sampleWeight: 1 / this.sampleRate,\n    });\n  }\n}\n\nconst telemetry = new TelemetryClient();\nexport default telemetry;\n"],"mappings":";;;;;;;AAMA,SAAgB,sBAA+B;AAC7C,QACG,QAAQ,IACN,kCAAkC,UACpC,QAAQ,IACN,kCAAkC,OACpC,QAAQ,IAA2C,iBAClD,UACD,QAAQ,IAA2C,iBAAiB;;AAIzE,IAAa,kBAAb,MAA6B;CAO3B,YAAY,EACV,mBACA,eAIE,EAAE,EAAE;0BAXgC,EAAE;2BACL;oBACR;qBACP,QAAQA,IAAQ;AASpC,OAAK,oBAAoB,qBAAqB,qBAAqB;AAEnE,MAAI,KAAK,mBAAmB;AAC1B,QAAK,cAAc,WAAW;AAC9B;;AAGF,OAAK,cAAc,WAAW;AAM9B,OAAK,UAAU,IAAI,UAAU,EAC3B,UAJA,QAAQ,IAAI,gCACZ,oCAID,CAAC;;CAGJ,AAAQ,kBAAkB;AACxB,MAAI,KAAK,kBACP,QAAO;AAGT,SADqB,KAAK,QAAQ,GACZ,KAAK;;CAG7B,MAAM,QACJ,OACA,YACA;AACA,MAAI,CAAC,KAAK,iBAAiB,CACzB;EAGF,MAAM,sBAAsB,cAAc,WAAW;EACrD,MAAM,uBAAuB;GAC3B,GAAG,KAAK;GACR,GAAG;GACJ;EACD,MAAM,8BAA8B,OAAO,KAAK,qBAAqB,CAClE,MAAM,CACN,QACE,KAAK,QAAQ;AACZ,OAAI,OAAO,qBAAqB;AAChC,UAAO;KAET,EAAE,CACH;AAEH,MAAI,KAAK,QACP,MAAK,QAAQ,MAAM;GACjB,aAAa,KAAK;GAClB;GACA,YAAY,EAAE,GAAG,6BAA6B;GAC/C,CAAC;AAGJ,QAAMC,qBAAY,SAAS,EACzB,OACD,CAAC;;CAGJ,oBAAoB,YAAiC;EACnD,MAAM,sBAAsB,cAAc,WAAW;AACrD,OAAK,mBAAmB;GACtB,GAAG,KAAK;GACR,GAAG;GACJ;;CAGH,AAAQ,cAAc,YAAgC;EACpD,IAAI;AAEJ,gBAAc,cAAc;AAE5B,MAAI,QAAQ,IAAI,iCACd,eAAc,WAAW,QAAQ,IAAI,iCAAiC;AAGxE,MAAI,cAAc,KAAK,cAAc,EACnC,OAAM,IAAI,MAAM,sCAAsC;AAGxD,OAAK,aAAa;AAClB,OAAK,oBAAoB;GACvB,YAAY,KAAK;GACjB,4BAA4B,IAAI,KAAK;GACrC,cAAc,IAAI,KAAK;GACxB,CAAC;;;AAIN,MAAM,YAAY,IAAI,iBAAiB"}
+{"version":3,"file":"telemetry-client.mjs","names":["scarfClient"],"sources":["../../../../src/v2/runtime/telemetry/telemetry-client.ts"],"sourcesContent":["import { AnalyticsEvents } from \"./events\";\nimport scarfClient from \"./scarf-client\";\n\nexport function isTelemetryDisabled(): boolean {\n  return (\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"true\" ||\n    (process.env as Record<string, string | undefined>)\n      .COPILOTKIT_TELEMETRY_DISABLED === \"1\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK ===\n      \"true\" ||\n    (process.env as Record<string, string | undefined>).DO_NOT_TRACK === \"1\"\n  );\n}\n\nexport class TelemetryClient {\n  private telemetryDisabled: boolean = false;\n  private sampleRate: number = 0.05;\n\n  constructor({\n    telemetryDisabled,\n    sampleRate,\n  }: {\n    telemetryDisabled?: boolean;\n    sampleRate?: number;\n  } = {}) {\n    this.telemetryDisabled = telemetryDisabled ?? isTelemetryDisabled();\n    this.setSampleRate(sampleRate);\n  }\n\n  private shouldSendEvent() {\n    if (this.telemetryDisabled) {\n      return false;\n    }\n    const randomNumber = Math.random();\n    return randomNumber < this.sampleRate;\n  }\n\n  async capture<K extends keyof AnalyticsEvents>(\n    event: K,\n    _properties: AnalyticsEvents[K],\n  ) {\n    if (!this.shouldSendEvent()) {\n      return;\n    }\n\n    await scarfClient.logEvent({\n      event,\n    });\n  }\n\n  private setSampleRate(sampleRate: number | undefined) {\n    let _sampleRate: number;\n\n    _sampleRate = sampleRate ?? 0.05;\n\n    if (process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE) {\n      _sampleRate = parseFloat(process.env.COPILOTKIT_TELEMETRY_SAMPLE_RATE);\n    }\n\n    if (_sampleRate < 0 || _sampleRate > 1) {\n      throw new Error(\"Sample rate must be between 0 and 1\");\n    }\n\n    this.sampleRate = _sampleRate;\n  }\n}\n\nconst telemetry = new TelemetryClient();\nexport default telemetry;\n"],"mappings":";;;;AAGA,SAAgB,sBAA+B;AAC7C,QACG,QAAQ,IACN,kCAAkC,UACpC,QAAQ,IACN,kCAAkC,OACpC,QAAQ,IAA2C,iBAClD,UACD,QAAQ,IAA2C,iBAAiB;;AAIzE,IAAa,kBAAb,MAA6B;CAI3B,YAAY,EACV,mBACA,eAIE,EAAE,EAAE;2BAT6B;oBACR;AAS3B,OAAK,oBAAoB,qBAAqB,qBAAqB;AACnE,OAAK,cAAc,WAAW;;CAGhC,AAAQ,kBAAkB;AACxB,MAAI,KAAK,kBACP,QAAO;AAGT,SADqB,KAAK,QAAQ,GACZ,KAAK;;CAG7B,MAAM,QACJ,OACA,aACA;AACA,MAAI,CAAC,KAAK,iBAAiB,CACzB;AAGF,QAAMA,qBAAY,SAAS,EACzB,OACD,CAAC;;CAGJ,AAAQ,cAAc,YAAgC;EACpD,IAAI;AAEJ,gBAAc,cAAc;AAE5B,MAAI,QAAQ,IAAI,iCACd,eAAc,WAAW,QAAQ,IAAI,iCAAiC;AAGxE,MAAI,cAAc,KAAK,cAAc,EACnC,OAAM,IAAI,MAAM,sCAAsC;AAGxD,OAAK,aAAa;;;AAItB,MAAM,YAAY,IAAI,iBAAiB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/runtime",
-  "version": "1.56.3",
+  "version": "1.56.4-canary.1777531098",
   "private": false,
   "keywords": [
     "ai",
@@ -77,10 +77,10 @@
   },
   "dependencies": {
     "@ag-ui/a2ui-middleware": "0.0.5",
-    "@ag-ui/client": "0.0.52",
-    "@ag-ui/core": "0.0.52",
-    "@ag-ui/encoder": "0.0.52",
-    "@ag-ui/langgraph": "0.0.29",
+    "@ag-ui/client": "0.0.53",
+    "@ag-ui/core": "0.0.53",
+    "@ag-ui/encoder": "0.0.53",
+    "@ag-ui/langgraph": "0.0.31",
     "@ag-ui/mcp-apps-middleware": "0.0.3",
     "@ai-sdk/anthropic": "^3.0.49",
     "@ai-sdk/google": "^3.0.33",
@@ -115,10 +115,10 @@
     "uuid": "^10.0.0",
     "ws": "^8.18.0",
     "zod": "^3.23.3",
-    "@copilotkit/shared": "1.56.3"
+    "@copilotkit/shared": "1.56.4-canary.1777531098"
   },
   "devDependencies": {
-    "@copilotkit/aimock": "^1.10.0",
+    "@copilotkit/aimock": "latest",
     "@swc/core": "1.5.28",
     "@types/cors": "^2.8.17",
     "@types/express": "^4.17.21",

package/src/agent/__tests__/agent-test-helpers.ts

@@ -5,7 +5,8 @@
  * BuiltInAgent-specific factories, mock stream builders, and assertion helpers.
  */
 
-import { EventType, type BaseEvent, type RunAgentInput } from "@ag-ui/client";
+import { EventType } from "@ag-ui/client";
+import type { BaseEvent, RunAgentInput } from "@ag-ui/client";
 import type { Observable } from "rxjs";
 import { BuiltInAgent } from "../index";
 import type { AgentFactoryContext, BuiltInAgentFactoryConfig } from "../index";
@@ -85,6 +86,35 @@ export function tanstackToolCallEnd(toolCallId: string) {
   return { type: "TOOL_CALL_END", toolCallId } as const;
 }
 
+/** TanStack tool-call result chunk. `content` is what the tool's `execute()` returned. */
+export function tanstackToolCallResult(toolCallId: string, content: unknown) {
+  return { type: "TOOL_CALL_RESULT", toolCallId, content } as const;
+}
+
+/** TanStack reasoning lifecycle chunk builders */
+export function tanstackReasoningStart(messageId: string) {
+  return { type: "REASONING_START", messageId } as const;
+}
+export function tanstackReasoningMessageStart(messageId: string) {
+  return {
+    type: "REASONING_MESSAGE_START",
+    messageId,
+    role: "reasoning",
+  } as const;
+}
+export function tanstackReasoningMessageContent(
+  messageId: string,
+  delta: string,
+) {
+  return { type: "REASONING_MESSAGE_CONTENT", messageId, delta } as const;
+}
+export function tanstackReasoningMessageEnd(messageId: string) {
+  return { type: "REASONING_MESSAGE_END", messageId } as const;
+}
+export function tanstackReasoningEnd(messageId: string) {
+  return { type: "REASONING_END", messageId } as const;
+}
+
 // ---------------------------------------------------------------------------
 // Mock async iterable builders
 // ---------------------------------------------------------------------------