@copilotkit/runtime 1.54.1 → 1.55.0-next.8

package/dist/v2/runtime/middleware-sse-parser.mjs

@@ -0,0 +1,137 @@
+import "reflect-metadata";
+import { logger } from "@copilotkit/shared";
+
+//#region src/v2/runtime/middleware-sse-parser.ts
+/**
+* Parse a cloned SSE Response body into structured messages.
+* Returns empty results for non-SSE responses.
+*/
+async function parseSSEResponse(response) {
+	if (!(response.headers.get("content-type") ?? "").includes("text/event-stream")) return { messages: [] };
+	let text;
+	try {
+		text = await response.text();
+	} catch {
+		logger.warn("Failed to read SSE response body in afterRequestMiddleware");
+		return { messages: [] };
+	}
+	if (!text.trim()) return { messages: [] };
+	let threadId;
+	let runId;
+	const messagesById = /* @__PURE__ */ new Map();
+	const toolCallsById = /* @__PURE__ */ new Map();
+	const toolCallParent = /* @__PURE__ */ new Map();
+	let snapshotMessages;
+	for (const line of text.split("\n")) {
+		const trimmed = line.trim();
+		if (!trimmed.startsWith("data:")) continue;
+		let event;
+		try {
+			event = JSON.parse(trimmed.slice(5).trim());
+		} catch {
+			continue;
+		}
+		switch (event.type) {
+			case "RUN_STARTED":
+				threadId = event.threadId;
+				runId = event.runId;
+				break;
+			case "MESSAGES_SNAPSHOT":
+				if (Array.isArray(event.messages)) snapshotMessages = event.messages;
+				break;
+			case "TEXT_MESSAGE_START":
+				messagesById.set(event.messageId, {
+					id: event.messageId,
+					role: event.role ?? "assistant",
+					content: ""
+				});
+				break;
+			case "TEXT_MESSAGE_CONTENT": {
+				const msg = messagesById.get(event.messageId);
+				if (msg) msg.content = (msg.content ?? "") + (event.delta ?? "");
+				break;
+			}
+			case "TEXT_MESSAGE_CHUNK":
+				if (event.messageId) {
+					const existing = messagesById.get(event.messageId);
+					if (existing) existing.content = (existing.content ?? "") + (event.delta ?? "");
+					else messagesById.set(event.messageId, {
+						id: event.messageId,
+						role: event.role ?? "assistant",
+						content: event.delta ?? ""
+					});
+				}
+				break;
+			case "TOOL_CALL_START": {
+				const tc = {
+					id: event.toolCallId,
+					name: event.toolCallName,
+					args: ""
+				};
+				toolCallsById.set(event.toolCallId, tc);
+				if (event.parentMessageId) toolCallParent.set(event.toolCallId, event.parentMessageId);
+				break;
+			}
+			case "TOOL_CALL_ARGS": {
+				const tc = toolCallsById.get(event.toolCallId);
+				if (tc) tc.args += event.delta ?? "";
+				break;
+			}
+			case "TOOL_CALL_CHUNK":
+				if (event.toolCallId) {
+					let tc = toolCallsById.get(event.toolCallId);
+					if (!tc) {
+						tc = {
+							id: event.toolCallId,
+							name: event.toolCallName ?? "",
+							args: ""
+						};
+						toolCallsById.set(event.toolCallId, tc);
+						if (event.parentMessageId) toolCallParent.set(event.toolCallId, event.parentMessageId);
+					}
+					if (event.toolCallName) tc.name = event.toolCallName;
+					tc.args += event.delta ?? "";
+				}
+				break;
+			case "TOOL_CALL_END": {
+				const tc = toolCallsById.get(event.toolCallId);
+				const parentId = toolCallParent.get(event.toolCallId);
+				if (tc && parentId) {
+					const parent = messagesById.get(parentId);
+					if (parent) {
+						parent.toolCalls = parent.toolCalls ?? [];
+						parent.toolCalls.push(tc);
+					}
+				}
+				break;
+			}
+			case "TOOL_CALL_RESULT":
+				messagesById.set(event.messageId, {
+					id: event.messageId,
+					role: "tool",
+					content: event.content,
+					toolCallId: event.toolCallId
+				});
+				break;
+		}
+	}
+	for (const [toolCallId, tc] of toolCallsById) {
+		const parentId = toolCallParent.get(toolCallId);
+		if (!parentId) continue;
+		const parent = messagesById.get(parentId);
+		if (!parent) continue;
+		if (!parent.toolCalls?.some((t) => t.id === tc.id)) {
+			parent.toolCalls = parent.toolCalls ?? [];
+			parent.toolCalls.push(tc);
+		}
+	}
+	return {
+		messages: snapshotMessages ?? [...messagesById.values()],
+		threadId,
+		runId
+	};
+}
+
+//#endregion
+export { parseSSEResponse };
+//# sourceMappingURL=middleware-sse-parser.mjs.map

package/dist/v2/runtime/middleware-sse-parser.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware-sse-parser.mjs","names":[],"sources":["../../../src/v2/runtime/middleware-sse-parser.ts"],"sourcesContent":["import { logger } from \"@copilotkit/shared\";\n\nexport interface ParsedSSEResult {\n  messages: Message[];\n  threadId?: string;\n  runId?: string;\n}\n\n/** Minimal message shape reconstructed from AG-UI events. */\nexport interface Message {\n  id: string;\n  role: string;\n  content?: string;\n  toolCalls?: ToolCall[];\n  toolCallId?: string;\n}\n\ninterface ToolCall {\n  id: string;\n  name: string;\n  args: string;\n}\n\n/**\n * Parse a cloned SSE Response body into structured messages.\n * Returns empty results for non-SSE responses.\n */\nexport async function parseSSEResponse(\n  response: Response,\n): Promise<ParsedSSEResult> {\n  const contentType = response.headers.get(\"content-type\") ?? \"\";\n  if (!contentType.includes(\"text/event-stream\")) {\n    return { messages: [] };\n  }\n\n  let text: string;\n  try {\n    text = await response.text();\n  } catch {\n    logger.warn(\"Failed to read SSE response body in afterRequestMiddleware\");\n    return { messages: [] };\n  }\n\n  if (!text.trim()) {\n    return { messages: [] };\n  }\n\n  let threadId: string | undefined;\n  let runId: string | undefined;\n  const messagesById = new Map<string, Message>();\n  const toolCallsById = new Map<string, ToolCall>();\n  const toolCallParent = new Map<string, string>(); // toolCallId → messageId\n  let snapshotMessages: Message[] | undefined;\n\n  for (const line of text.split(\"\\n\")) {\n    const trimmed = line.trim();\n    if (!trimmed.startsWith(\"data:\")) continue;\n\n    let event: Record<string, any>;\n    try {\n      event = JSON.parse(trimmed.slice(5).trim());\n    } catch {\n      continue;\n    }\n\n    switch (event.type) {\n      case \"RUN_STARTED\":\n        threadId = event.threadId;\n        runId = event.runId;\n        break;\n\n      case \"MESSAGES_SNAPSHOT\":\n        if (Array.isArray(event.messages)) {\n          snapshotMessages = event.messages;\n        }\n        break;\n\n      case \"TEXT_MESSAGE_START\":\n        messagesById.set(event.messageId, {\n          id: event.messageId,\n          role: event.role ?? \"assistant\",\n          content: \"\",\n        });\n        break;\n\n      case \"TEXT_MESSAGE_CONTENT\": {\n        const msg = messagesById.get(event.messageId);\n        if (msg) {\n          msg.content = (msg.content ?? \"\") + (event.delta ?? \"\");\n        }\n        break;\n      }\n\n      case \"TEXT_MESSAGE_CHUNK\": {\n        // Chunk format: combined start+content. First chunk creates the\n        // message, subsequent chunks append delta to content.\n        if (event.messageId) {\n          const existing = messagesById.get(event.messageId);\n          if (existing) {\n            existing.content = (existing.content ?? \"\") + (event.delta ?? \"\");\n          } else {\n            messagesById.set(event.messageId, {\n              id: event.messageId,\n              role: event.role ?? \"assistant\",\n              content: event.delta ?? \"\",\n            });\n          }\n        }\n        break;\n      }\n\n      case \"TOOL_CALL_START\": {\n        const tc: ToolCall = {\n          id: event.toolCallId,\n          name: event.toolCallName,\n          args: \"\",\n        };\n        toolCallsById.set(event.toolCallId, tc);\n        if (event.parentMessageId) {\n          toolCallParent.set(event.toolCallId, event.parentMessageId);\n        }\n        break;\n      }\n\n      case \"TOOL_CALL_ARGS\": {\n        const tc = toolCallsById.get(event.toolCallId);\n        if (tc) {\n          tc.args += event.delta ?? \"\";\n        }\n        break;\n      }\n\n      case \"TOOL_CALL_CHUNK\": {\n        // Chunk format: combined start+args. First chunk for a given\n        // toolCallId creates the tool call, subsequent chunks append delta.\n        if (event.toolCallId) {\n          let tc = toolCallsById.get(event.toolCallId);\n          if (!tc) {\n            tc = {\n              id: event.toolCallId,\n              name: event.toolCallName ?? \"\",\n              args: \"\",\n            };\n            toolCallsById.set(event.toolCallId, tc);\n            if (event.parentMessageId) {\n              toolCallParent.set(event.toolCallId, event.parentMessageId);\n            }\n          }\n          if (event.toolCallName) {\n            tc.name = event.toolCallName;\n          }\n          tc.args += event.delta ?? \"\";\n        }\n        break;\n      }\n\n      case \"TOOL_CALL_END\": {\n        const tc = toolCallsById.get(event.toolCallId);\n        const parentId = toolCallParent.get(event.toolCallId);\n        if (tc && parentId) {\n          const parent = messagesById.get(parentId);\n          if (parent) {\n            parent.toolCalls = parent.toolCalls ?? [];\n            parent.toolCalls.push(tc);\n          }\n        }\n        break;\n      }\n\n      case \"TOOL_CALL_RESULT\":\n        messagesById.set(event.messageId, {\n          id: event.messageId,\n          role: \"tool\",\n          content: event.content,\n          toolCallId: event.toolCallId,\n        });\n        break;\n    }\n  }\n\n  // Attach any tool calls not yet linked to their parent message.\n  // This handles TOOL_CALL_CHUNK flows which don't emit TOOL_CALL_END.\n  for (const [toolCallId, tc] of toolCallsById) {\n    const parentId = toolCallParent.get(toolCallId);\n    if (!parentId) continue;\n    const parent = messagesById.get(parentId);\n    if (!parent) continue;\n    const alreadyAttached = parent.toolCalls?.some((t) => t.id === tc.id);\n    if (!alreadyAttached) {\n      parent.toolCalls = parent.toolCalls ?? [];\n      parent.toolCalls.push(tc);\n    }\n  }\n\n  // Prefer MESSAGES_SNAPSHOT if present (contains full history).\n  // Otherwise reconstruct from individual events.\n  const messages = snapshotMessages ?? [...messagesById.values()];\n\n  return { messages, threadId, runId };\n}\n"],"mappings":";;;;;;;;AA2BA,eAAsB,iBACpB,UAC0B;AAE1B,KAAI,EADgB,SAAS,QAAQ,IAAI,eAAe,IAAI,IAC3C,SAAS,oBAAoB,CAC5C,QAAO,EAAE,UAAU,EAAE,EAAE;CAGzB,IAAI;AACJ,KAAI;AACF,SAAO,MAAM,SAAS,MAAM;SACtB;AACN,SAAO,KAAK,6DAA6D;AACzE,SAAO,EAAE,UAAU,EAAE,EAAE;;AAGzB,KAAI,CAAC,KAAK,MAAM,CACd,QAAO,EAAE,UAAU,EAAE,EAAE;CAGzB,IAAI;CACJ,IAAI;CACJ,MAAM,+BAAe,IAAI,KAAsB;CAC/C,MAAM,gCAAgB,IAAI,KAAuB;CACjD,MAAM,iCAAiB,IAAI,KAAqB;CAChD,IAAI;AAEJ,MAAK,MAAM,QAAQ,KAAK,MAAM,KAAK,EAAE;EACnC,MAAM,UAAU,KAAK,MAAM;AAC3B,MAAI,CAAC,QAAQ,WAAW,QAAQ,CAAE;EAElC,IAAI;AACJ,MAAI;AACF,WAAQ,KAAK,MAAM,QAAQ,MAAM,EAAE,CAAC,MAAM,CAAC;UACrC;AACN;;AAGF,UAAQ,MAAM,MAAd;GACE,KAAK;AACH,eAAW,MAAM;AACjB,YAAQ,MAAM;AACd;GAEF,KAAK;AACH,QAAI,MAAM,QAAQ,MAAM,SAAS,CAC/B,oBAAmB,MAAM;AAE3B;GAEF,KAAK;AACH,iBAAa,IAAI,MAAM,WAAW;KAChC,IAAI,MAAM;KACV,MAAM,MAAM,QAAQ;KACpB,SAAS;KACV,CAAC;AACF;GAEF,KAAK,wBAAwB;IAC3B,MAAM,MAAM,aAAa,IAAI,MAAM,UAAU;AAC7C,QAAI,IACF,KAAI,WAAW,IAAI,WAAW,OAAO,MAAM,SAAS;AAEtD;;GAGF,KAAK;AAGH,QAAI,MAAM,WAAW;KACnB,MAAM,WAAW,aAAa,IAAI,MAAM,UAAU;AAClD,SAAI,SACF,UAAS,WAAW,SAAS,WAAW,OAAO,MAAM,SAAS;SAE9D,cAAa,IAAI,MAAM,WAAW;MAChC,IAAI,MAAM;MACV,MAAM,MAAM,QAAQ;MACpB,SAAS,MAAM,SAAS;MACzB,CAAC;;AAGN;GAGF,KAAK,mBAAmB;IACtB,MAAM,KAAe;KACnB,IAAI,MAAM;KACV,MAAM,MAAM;KACZ,MAAM;KACP;AACD,kBAAc,IAAI,MAAM,YAAY,GAAG;AACvC,QAAI,MAAM,gBACR,gBAAe,IAAI,MAAM,YAAY,MAAM,gBAAgB;AAE7D;;GAGF,KAAK,kBAAkB;IACrB,MAAM,KAAK,cAAc,IAAI,MAAM,WAAW;AAC9C,QAAI,GACF,IAAG,QAAQ,MAAM,SAAS;AAE5B;;GAGF,KAAK;AAGH,QAAI,MAAM,YAAY;KACpB,IAAI,KAAK,cAAc,IAAI,MAAM,WAAW;AAC5C,SAAI,CAAC,IAAI;AACP,WAAK;OACH,IAAI,MAAM;OACV,MAAM,MAAM,gBAAgB;OAC5B,MAAM;OACP;AACD,oBAAc,IAAI,MAAM,YAAY,GAAG;AACvC,UAAI,MAAM,gBACR,gBAAe,IAAI,MAAM,YAAY,MAAM,gBAAgB;;AAG/D,SAAI,MAAM,aACR,IAAG,OAAO,MAAM;AAElB,QAAG,QAAQ,MAAM,SAAS;;AAE5B;GAGF,KAAK,iBAAiB;IACpB,MAAM,KAAK,cAAc,IAAI,MAAM,WAAW;IAC9C,MAAM,WAAW,eAAe,IAAI,MAAM,WAAW;AACrD,QAAI,MAAM,UAAU;KAClB,MAAM,SAAS,aAAa,IAAI,SAAS;AACzC,SAAI,QAAQ;AACV,aAAO,YAAY,OAAO,aAAa,EAAE;AACzC,aAAO,UAAU,KAAK,GAAG;;;AAG7B;;GAGF,KAAK;AACH,iBAAa,IAAI,MAAM,WAAW;KAChC,IAAI,MAAM;KACV,MAAM;KACN,SAAS,MAAM;KACf,YAAY,MAAM;KACnB,CAAC;AACF;;;AAMN,MAAK,MAAM,CAAC,YAAY,OAAO,eAAe;EAC5C,MAAM,WAAW,eAAe,IAAI,WAAW;AAC/C,MAAI,CAAC,SAAU;EACf,MAAM,SAAS,aAAa,IAAI,SAAS;AACzC,MAAI,CAAC,OAAQ;AAEb,MAAI,CADoB,OAAO,WAAW,MAAM,MAAM,EAAE,OAAO,GAAG,GAAG,EAC/C;AACpB,UAAO,YAAY,OAAO,aAAa,EAAE;AACzC,UAAO,UAAU,KAAK,GAAG;;;AAQ7B,QAAO;EAAE,UAFQ,oBAAoB,CAAC,GAAG,aAAa,QAAQ,CAAC;EAE5C;EAAU;EAAO"}

package/dist/v2/runtime/middleware.cjs

@@ -0,0 +1,35 @@
+require("reflect-metadata");
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
+const require_middleware_sse_parser = require('./middleware-sse-parser.cjs');
+let _copilotkit_shared = require("@copilotkit/shared");
+
+//#region src/v2/runtime/middleware.ts
+async function callBeforeRequestMiddleware({ runtime, request, path }) {
+	const mw = runtime.beforeRequestMiddleware;
+	if (!mw) return;
+	if (typeof mw === "function") return mw({
+		runtime,
+		request,
+		path
+	});
+	_copilotkit_shared.logger.warn({ mw }, "Unsupported beforeRequestMiddleware value – skipped");
+}
+async function callAfterRequestMiddleware({ runtime, response, path }) {
+	const mw = runtime.afterRequestMiddleware;
+	if (!mw) return;
+	const { messages, threadId, runId } = await require_middleware_sse_parser.parseSSEResponse(response);
+	if (typeof mw === "function") return mw({
+		runtime,
+		response,
+		path,
+		messages,
+		threadId,
+		runId
+	});
+	_copilotkit_shared.logger.warn({ mw }, "Unsupported afterRequestMiddleware value – skipped");
+}
+
+//#endregion
+exports.callAfterRequestMiddleware = callAfterRequestMiddleware;
+exports.callBeforeRequestMiddleware = callBeforeRequestMiddleware;
+//# sourceMappingURL=middleware.cjs.map

package/dist/v2/runtime/middleware.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.cjs","names":["parseSSEResponse"],"sources":["../../../src/v2/runtime/middleware.ts"],"sourcesContent":["/**\n * Middleware support for CopilotKit Runtime.\n *\n * A middleware hook can be provided as either:\n *   1. A **callback function** executed in-process.\n *   2. A **webhook URL** (http/https).  The runtime will `POST` a JSON payload\n *      to the URL and, for *before* hooks, accept an optional modified\n *      `Request` object in the response body.\n *\n * Two lifecycle hooks are available:\n *   • `BEFORE_REQUEST` – runs *before* the request handler.\n *   • `AFTER_REQUEST`  – runs *after* the handler returns a `Response`.\n */\n\nimport type { CopilotRuntimeLike } from \"./runtime\";\nimport type { MaybePromise } from \"@copilotkit/shared\";\nimport { logger } from \"@copilotkit/shared\";\nimport { parseSSEResponse } from \"./middleware-sse-parser\";\nimport type { Message } from \"./middleware-sse-parser\";\n\n/* ------------------------------------------------------------------------------------------------\n * Public types\n * --------------------------------------------------------------------------------------------- */\n\nexport interface BeforeRequestMiddlewareParameters {\n  runtime: CopilotRuntimeLike;\n  request: Request;\n  path: string;\n}\nexport interface AfterRequestMiddlewareParameters {\n  runtime: CopilotRuntimeLike;\n  response: Response;\n  path: string;\n  /** Reconstructed messages from the SSE stream (empty for non-SSE responses). */\n  messages?: Message[];\n  /** Thread ID extracted from the RUN_STARTED event. */\n  threadId?: string;\n  /** Run ID extracted from the RUN_STARTED event. */\n  runId?: string;\n}\n\nexport type BeforeRequestMiddlewareFn = (\n  params: BeforeRequestMiddlewareParameters,\n) => MaybePromise<Request | void>;\nexport type AfterRequestMiddlewareFn = (\n  params: AfterRequestMiddlewareParameters,\n) => MaybePromise<void>;\n\n/**\n * A middleware value can be either a callback function or a webhook URL.\n */\nexport type BeforeRequestMiddleware = BeforeRequestMiddlewareFn;\nexport type AfterRequestMiddleware = AfterRequestMiddlewareFn;\n\n/** Lifecycle events emitted to webhook middleware. */\nexport enum CopilotKitMiddlewareEvent {\n  BeforeRequest = \"BEFORE_REQUEST\",\n  AfterRequest = \"AFTER_REQUEST\",\n}\n\n/** Stages used by the Middleware Webhook Protocol */\n/** Stages used by the CopilotKit webhook protocol */\nexport enum WebhookStage {\n  BeforeRequest = \"before_request\",\n  AfterRequest = \"after_request\",\n}\n\n/* ------------------------------------------------------------------------------------------------\n * Internal helpers – (de)serialisation\n * --------------------------------------------------------------------------------------------- */\n\nexport async function callBeforeRequestMiddleware({\n  runtime,\n  request,\n  path,\n}: BeforeRequestMiddlewareParameters): Promise<Request | void> {\n  const mw = runtime.beforeRequestMiddleware;\n  if (!mw) return;\n\n  // Function-based middleware (in-process)\n  if (typeof mw === \"function\") {\n    return (mw as BeforeRequestMiddlewareFn)({ runtime, request, path });\n  }\n\n  logger.warn({ mw }, \"Unsupported beforeRequestMiddleware value – skipped\");\n  return;\n}\n\nexport async function callAfterRequestMiddleware({\n  runtime,\n  response,\n  path,\n}: {\n  runtime: CopilotRuntimeLike;\n  response: Response;\n  path: string;\n}): Promise<void> {\n  const mw = runtime.afterRequestMiddleware;\n  if (!mw) return;\n\n  const { messages, threadId, runId } = await parseSSEResponse(response);\n\n  if (typeof mw === \"function\") {\n    return (mw as AfterRequestMiddlewareFn)({\n      runtime,\n      response,\n      path,\n      messages,\n      threadId,\n      runId,\n    });\n  }\n\n  logger.warn({ mw }, \"Unsupported afterRequestMiddleware value – skipped\");\n}\n"],"mappings":";;;;;;AAuEA,eAAsB,4BAA4B,EAChD,SACA,SACA,QAC6D;CAC7D,MAAM,KAAK,QAAQ;AACnB,KAAI,CAAC,GAAI;AAGT,KAAI,OAAO,OAAO,WAChB,QAAQ,GAAiC;EAAE;EAAS;EAAS;EAAM,CAAC;AAGtE,2BAAO,KAAK,EAAE,IAAI,EAAE,sDAAsD;;AAI5E,eAAsB,2BAA2B,EAC/C,SACA,UACA,QAKgB;CAChB,MAAM,KAAK,QAAQ;AACnB,KAAI,CAAC,GAAI;CAET,MAAM,EAAE,UAAU,UAAU,UAAU,MAAMA,+CAAiB,SAAS;AAEtE,KAAI,OAAO,OAAO,WAChB,QAAQ,GAAgC;EACtC;EACA;EACA;EACA;EACA;EACA;EACD,CAAC;AAGJ,2BAAO,KAAK,EAAE,IAAI,EAAE,qDAAqD"}

package/dist/v2/runtime/middleware.d.cts

@@ -0,0 +1,32 @@
+
+import { Message } from "./middleware-sse-parser.cjs";
+import { CopilotRuntimeLike } from "./runtime.cjs";
+import { MaybePromise } from "@copilotkit/shared";
+
+//#region src/v2/runtime/middleware.d.ts
+interface BeforeRequestMiddlewareParameters {
+  runtime: CopilotRuntimeLike;
+  request: Request;
+  path: string;
+}
+interface AfterRequestMiddlewareParameters {
+  runtime: CopilotRuntimeLike;
+  response: Response;
+  path: string;
+  /** Reconstructed messages from the SSE stream (empty for non-SSE responses). */
+  messages?: Message[];
+  /** Thread ID extracted from the RUN_STARTED event. */
+  threadId?: string;
+  /** Run ID extracted from the RUN_STARTED event. */
+  runId?: string;
+}
+type BeforeRequestMiddlewareFn = (params: BeforeRequestMiddlewareParameters) => MaybePromise<Request | void>;
+type AfterRequestMiddlewareFn = (params: AfterRequestMiddlewareParameters) => MaybePromise<void>;
+/**
+ * A middleware value can be either a callback function or a webhook URL.
+ */
+type BeforeRequestMiddleware = BeforeRequestMiddlewareFn;
+type AfterRequestMiddleware = AfterRequestMiddlewareFn;
+//#endregion
+export { AfterRequestMiddleware, BeforeRequestMiddleware };
+//# sourceMappingURL=middleware.d.cts.map

package/dist/v2/runtime/middleware.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.cts","names":[],"sources":["../../../src/v2/runtime/middleware.ts"],"mappings":";;;;;;UAwBiB,iCAAA;EACf,OAAA,EAAS,kBAAA;EACT,OAAA,EAAS,OAAA;EACT,IAAA;AAAA;AAAA,UAEe,gCAAA;EACf,OAAA,EAAS,kBAAA;EACT,QAAA,EAAU,QAAA;EACV,IAAA;EAFS;EAIT,QAAA,GAAW,OAAA;EAHD;EAKV,QAAA;EAFA;EAIA,KAAA;AAAA;AAAA,KAGU,yBAAA,IACV,MAAA,EAAQ,iCAAA,KACL,YAAA,CAAa,OAAA;AAAA,KACN,wBAAA,IACV,MAAA,EAAQ,gCAAA,KACL,YAAA;;AALL;;KAUY,uBAAA,GAA0B,yBAAA;AAAA,KAC1B,sBAAA,GAAyB,wBAAA"}

package/dist/v2/runtime/middleware.d.mts

@@ -0,0 +1,32 @@
+import "reflect-metadata";
+import { Message } from "./middleware-sse-parser.mjs";
+import { CopilotRuntimeLike } from "./runtime.mjs";
+import { MaybePromise } from "@copilotkit/shared";
+
+//#region src/v2/runtime/middleware.d.ts
+interface BeforeRequestMiddlewareParameters {
+  runtime: CopilotRuntimeLike;
+  request: Request;
+  path: string;
+}
+interface AfterRequestMiddlewareParameters {
+  runtime: CopilotRuntimeLike;
+  response: Response;
+  path: string;
+  /** Reconstructed messages from the SSE stream (empty for non-SSE responses). */
+  messages?: Message[];
+  /** Thread ID extracted from the RUN_STARTED event. */
+  threadId?: string;
+  /** Run ID extracted from the RUN_STARTED event. */
+  runId?: string;
+}
+type BeforeRequestMiddlewareFn = (params: BeforeRequestMiddlewareParameters) => MaybePromise<Request | void>;
+type AfterRequestMiddlewareFn = (params: AfterRequestMiddlewareParameters) => MaybePromise<void>;
+/**
+ * A middleware value can be either a callback function or a webhook URL.
+ */
+type BeforeRequestMiddleware = BeforeRequestMiddlewareFn;
+type AfterRequestMiddleware = AfterRequestMiddlewareFn;
+//#endregion
+export { AfterRequestMiddleware, BeforeRequestMiddleware };
+//# sourceMappingURL=middleware.d.mts.map

package/dist/v2/runtime/middleware.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.mts","names":[],"sources":["../../../src/v2/runtime/middleware.ts"],"mappings":";;;;;;UAwBiB,iCAAA;EACf,OAAA,EAAS,kBAAA;EACT,OAAA,EAAS,OAAA;EACT,IAAA;AAAA;AAAA,UAEe,gCAAA;EACf,OAAA,EAAS,kBAAA;EACT,QAAA,EAAU,QAAA;EACV,IAAA;EAFS;EAIT,QAAA,GAAW,OAAA;EAHD;EAKV,QAAA;EAFA;EAIA,KAAA;AAAA;AAAA,KAGU,yBAAA,IACV,MAAA,EAAQ,iCAAA,KACL,YAAA,CAAa,OAAA;AAAA,KACN,wBAAA,IACV,MAAA,EAAQ,gCAAA,KACL,YAAA;;AALL;;KAUY,uBAAA,GAA0B,yBAAA;AAAA,KAC1B,sBAAA,GAAyB,wBAAA"}

package/dist/v2/runtime/middleware.mjs

@@ -0,0 +1,33 @@
+import "reflect-metadata";
+import { parseSSEResponse } from "./middleware-sse-parser.mjs";
+import { logger } from "@copilotkit/shared";
+
+//#region src/v2/runtime/middleware.ts
+async function callBeforeRequestMiddleware({ runtime, request, path }) {
+	const mw = runtime.beforeRequestMiddleware;
+	if (!mw) return;
+	if (typeof mw === "function") return mw({
+		runtime,
+		request,
+		path
+	});
+	logger.warn({ mw }, "Unsupported beforeRequestMiddleware value – skipped");
+}
+async function callAfterRequestMiddleware({ runtime, response, path }) {
+	const mw = runtime.afterRequestMiddleware;
+	if (!mw) return;
+	const { messages, threadId, runId } = await parseSSEResponse(response);
+	if (typeof mw === "function") return mw({
+		runtime,
+		response,
+		path,
+		messages,
+		threadId,
+		runId
+	});
+	logger.warn({ mw }, "Unsupported afterRequestMiddleware value – skipped");
+}
+
+//#endregion
+export { callAfterRequestMiddleware, callBeforeRequestMiddleware };
+//# sourceMappingURL=middleware.mjs.map

package/dist/v2/runtime/middleware.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.mjs","names":[],"sources":["../../../src/v2/runtime/middleware.ts"],"sourcesContent":["/**\n * Middleware support for CopilotKit Runtime.\n *\n * A middleware hook can be provided as either:\n *   1. A **callback function** executed in-process.\n *   2. A **webhook URL** (http/https).  The runtime will `POST` a JSON payload\n *      to the URL and, for *before* hooks, accept an optional modified\n *      `Request` object in the response body.\n *\n * Two lifecycle hooks are available:\n *   • `BEFORE_REQUEST` – runs *before* the request handler.\n *   • `AFTER_REQUEST`  – runs *after* the handler returns a `Response`.\n */\n\nimport type { CopilotRuntimeLike } from \"./runtime\";\nimport type { MaybePromise } from \"@copilotkit/shared\";\nimport { logger } from \"@copilotkit/shared\";\nimport { parseSSEResponse } from \"./middleware-sse-parser\";\nimport type { Message } from \"./middleware-sse-parser\";\n\n/* ------------------------------------------------------------------------------------------------\n * Public types\n * --------------------------------------------------------------------------------------------- */\n\nexport interface BeforeRequestMiddlewareParameters {\n  runtime: CopilotRuntimeLike;\n  request: Request;\n  path: string;\n}\nexport interface AfterRequestMiddlewareParameters {\n  runtime: CopilotRuntimeLike;\n  response: Response;\n  path: string;\n  /** Reconstructed messages from the SSE stream (empty for non-SSE responses). */\n  messages?: Message[];\n  /** Thread ID extracted from the RUN_STARTED event. */\n  threadId?: string;\n  /** Run ID extracted from the RUN_STARTED event. */\n  runId?: string;\n}\n\nexport type BeforeRequestMiddlewareFn = (\n  params: BeforeRequestMiddlewareParameters,\n) => MaybePromise<Request | void>;\nexport type AfterRequestMiddlewareFn = (\n  params: AfterRequestMiddlewareParameters,\n) => MaybePromise<void>;\n\n/**\n * A middleware value can be either a callback function or a webhook URL.\n */\nexport type BeforeRequestMiddleware = BeforeRequestMiddlewareFn;\nexport type AfterRequestMiddleware = AfterRequestMiddlewareFn;\n\n/** Lifecycle events emitted to webhook middleware. */\nexport enum CopilotKitMiddlewareEvent {\n  BeforeRequest = \"BEFORE_REQUEST\",\n  AfterRequest = \"AFTER_REQUEST\",\n}\n\n/** Stages used by the Middleware Webhook Protocol */\n/** Stages used by the CopilotKit webhook protocol */\nexport enum WebhookStage {\n  BeforeRequest = \"before_request\",\n  AfterRequest = \"after_request\",\n}\n\n/* ------------------------------------------------------------------------------------------------\n * Internal helpers – (de)serialisation\n * --------------------------------------------------------------------------------------------- */\n\nexport async function callBeforeRequestMiddleware({\n  runtime,\n  request,\n  path,\n}: BeforeRequestMiddlewareParameters): Promise<Request | void> {\n  const mw = runtime.beforeRequestMiddleware;\n  if (!mw) return;\n\n  // Function-based middleware (in-process)\n  if (typeof mw === \"function\") {\n    return (mw as BeforeRequestMiddlewareFn)({ runtime, request, path });\n  }\n\n  logger.warn({ mw }, \"Unsupported beforeRequestMiddleware value – skipped\");\n  return;\n}\n\nexport async function callAfterRequestMiddleware({\n  runtime,\n  response,\n  path,\n}: {\n  runtime: CopilotRuntimeLike;\n  response: Response;\n  path: string;\n}): Promise<void> {\n  const mw = runtime.afterRequestMiddleware;\n  if (!mw) return;\n\n  const { messages, threadId, runId } = await parseSSEResponse(response);\n\n  if (typeof mw === \"function\") {\n    return (mw as AfterRequestMiddlewareFn)({\n      runtime,\n      response,\n      path,\n      messages,\n      threadId,\n      runId,\n    });\n  }\n\n  logger.warn({ mw }, \"Unsupported afterRequestMiddleware value – skipped\");\n}\n"],"mappings":";;;;;AAuEA,eAAsB,4BAA4B,EAChD,SACA,SACA,QAC6D;CAC7D,MAAM,KAAK,QAAQ;AACnB,KAAI,CAAC,GAAI;AAGT,KAAI,OAAO,OAAO,WAChB,QAAQ,GAAiC;EAAE;EAAS;EAAS;EAAM,CAAC;AAGtE,QAAO,KAAK,EAAE,IAAI,EAAE,sDAAsD;;AAI5E,eAAsB,2BAA2B,EAC/C,SACA,UACA,QAKgB;CAChB,MAAM,KAAK,QAAQ;AACnB,KAAI,CAAC,GAAI;CAET,MAAM,EAAE,UAAU,UAAU,UAAU,MAAM,iBAAiB,SAAS;AAEtE,KAAI,OAAO,OAAO,WAChB,QAAQ,GAAgC;EACtC;EACA;EACA;EACA;EACA;EACA;EACD,CAAC;AAGJ,QAAO,KAAK,EAAE,IAAI,EAAE,qDAAqD"}

package/dist/v2/runtime/runner/agent-runner.cjs

@@ -0,0 +1,8 @@
+require("reflect-metadata");
+
+//#region src/v2/runtime/runner/agent-runner.ts
+var AgentRunner = class {};
+
+//#endregion
+exports.AgentRunner = AgentRunner;
+//# sourceMappingURL=agent-runner.cjs.map

package/dist/v2/runtime/runner/agent-runner.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-runner.cjs","names":[],"sources":["../../../../src/v2/runtime/runner/agent-runner.ts"],"sourcesContent":["import {\n  AbstractAgent,\n  BaseEvent,\n  Message,\n  RunAgentInput,\n} from \"@ag-ui/client\";\nimport { Observable } from \"rxjs\";\n\nexport interface AgentRunnerRunRequest {\n  threadId: string;\n  agent: AbstractAgent;\n  input: RunAgentInput;\n  joinCode?: string;\n  persistedInputMessages?: Message[];\n}\n\nexport interface AgentRunnerConnectRequest {\n  threadId: string;\n  headers?: Record<string, string>;\n  joinCode?: string;\n}\n\nexport interface AgentRunnerIsRunningRequest {\n  threadId: string;\n}\n\nexport interface AgentRunnerStopRequest {\n  threadId: string;\n}\n\nexport abstract class AgentRunner {\n  abstract run(request: AgentRunnerRunRequest): Observable<BaseEvent>;\n  abstract connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;\n  abstract isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;\n  abstract stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;\n}\n"],"mappings":";;;AA8BA,IAAsB,cAAtB,MAAkC"}

package/dist/v2/runtime/runner/agent-runner.d.cts

@@ -0,0 +1,32 @@
+
+import { Observable } from "rxjs";
+import { AbstractAgent, BaseEvent, Message, RunAgentInput } from "@ag-ui/client";
+
+//#region src/v2/runtime/runner/agent-runner.d.ts
+interface AgentRunnerRunRequest {
+  threadId: string;
+  agent: AbstractAgent;
+  input: RunAgentInput;
+  joinCode?: string;
+  persistedInputMessages?: Message[];
+}
+interface AgentRunnerConnectRequest {
+  threadId: string;
+  headers?: Record<string, string>;
+  joinCode?: string;
+}
+interface AgentRunnerIsRunningRequest {
+  threadId: string;
+}
+interface AgentRunnerStopRequest {
+  threadId: string;
+}
+declare abstract class AgentRunner {
+  abstract run(request: AgentRunnerRunRequest): Observable<BaseEvent>;
+  abstract connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;
+  abstract isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;
+  abstract stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;
+}
+//#endregion
+export { AgentRunner, AgentRunnerConnectRequest, AgentRunnerIsRunningRequest, AgentRunnerRunRequest, AgentRunnerStopRequest };
+//# sourceMappingURL=agent-runner.d.cts.map

package/dist/v2/runtime/runner/agent-runner.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-runner.d.cts","names":[],"sources":["../../../../src/v2/runtime/runner/agent-runner.ts"],"mappings":";;;;;UAQiB,qBAAA;EACf,QAAA;EACA,KAAA,EAAO,aAAA;EACP,KAAA,EAAO,aAAA;EACP,QAAA;EACA,sBAAA,GAAyB,OAAA;AAAA;AAAA,UAGV,yBAAA;EACf,QAAA;EACA,OAAA,GAAU,MAAA;EACV,QAAA;AAAA;AAAA,UAGe,2BAAA;EACf,QAAA;AAAA;AAAA,UAGe,sBAAA;EACf,QAAA;AAAA;AAAA,uBAGoB,WAAA;EAAA,SACX,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EAAA,SAChD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAAA,SACxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAAA,SACjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;AAAA"}

package/dist/v2/runtime/runner/agent-runner.d.mts

@@ -0,0 +1,32 @@
+import "reflect-metadata";
+import { Observable } from "rxjs";
+import { AbstractAgent, BaseEvent, Message, RunAgentInput } from "@ag-ui/client";
+
+//#region src/v2/runtime/runner/agent-runner.d.ts
+interface AgentRunnerRunRequest {
+  threadId: string;
+  agent: AbstractAgent;
+  input: RunAgentInput;
+  joinCode?: string;
+  persistedInputMessages?: Message[];
+}
+interface AgentRunnerConnectRequest {
+  threadId: string;
+  headers?: Record<string, string>;
+  joinCode?: string;
+}
+interface AgentRunnerIsRunningRequest {
+  threadId: string;
+}
+interface AgentRunnerStopRequest {
+  threadId: string;
+}
+declare abstract class AgentRunner {
+  abstract run(request: AgentRunnerRunRequest): Observable<BaseEvent>;
+  abstract connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;
+  abstract isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;
+  abstract stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;
+}
+//#endregion
+export { AgentRunner, AgentRunnerConnectRequest, AgentRunnerIsRunningRequest, AgentRunnerRunRequest, AgentRunnerStopRequest };
+//# sourceMappingURL=agent-runner.d.mts.map

package/dist/v2/runtime/runner/agent-runner.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-runner.d.mts","names":[],"sources":["../../../../src/v2/runtime/runner/agent-runner.ts"],"mappings":";;;;;UAQiB,qBAAA;EACf,QAAA;EACA,KAAA,EAAO,aAAA;EACP,KAAA,EAAO,aAAA;EACP,QAAA;EACA,sBAAA,GAAyB,OAAA;AAAA;AAAA,UAGV,yBAAA;EACf,QAAA;EACA,OAAA,GAAU,MAAA;EACV,QAAA;AAAA;AAAA,UAGe,2BAAA;EACf,QAAA;AAAA;AAAA,UAGe,sBAAA;EACf,QAAA;AAAA;AAAA,uBAGoB,WAAA;EAAA,SACX,GAAA,CAAI,OAAA,EAAS,qBAAA,GAAwB,UAAA,CAAW,SAAA;EAAA,SAChD,OAAA,CAAQ,OAAA,EAAS,yBAAA,GAA4B,UAAA,CAAW,SAAA;EAAA,SACxD,SAAA,CAAU,OAAA,EAAS,2BAAA,GAA8B,OAAA;EAAA,SACjD,IAAA,CAAK,OAAA,EAAS,sBAAA,GAAyB,OAAA;AAAA"}

package/dist/v2/runtime/runner/agent-runner.mjs

@@ -0,0 +1,7 @@
+import "reflect-metadata";
+//#region src/v2/runtime/runner/agent-runner.ts
+var AgentRunner = class {};
+
+//#endregion
+export { AgentRunner };
+//# sourceMappingURL=agent-runner.mjs.map

package/dist/v2/runtime/runner/agent-runner.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-runner.mjs","names":[],"sources":["../../../../src/v2/runtime/runner/agent-runner.ts"],"sourcesContent":["import {\n  AbstractAgent,\n  BaseEvent,\n  Message,\n  RunAgentInput,\n} from \"@ag-ui/client\";\nimport { Observable } from \"rxjs\";\n\nexport interface AgentRunnerRunRequest {\n  threadId: string;\n  agent: AbstractAgent;\n  input: RunAgentInput;\n  joinCode?: string;\n  persistedInputMessages?: Message[];\n}\n\nexport interface AgentRunnerConnectRequest {\n  threadId: string;\n  headers?: Record<string, string>;\n  joinCode?: string;\n}\n\nexport interface AgentRunnerIsRunningRequest {\n  threadId: string;\n}\n\nexport interface AgentRunnerStopRequest {\n  threadId: string;\n}\n\nexport abstract class AgentRunner {\n  abstract run(request: AgentRunnerRunRequest): Observable<BaseEvent>;\n  abstract connect(request: AgentRunnerConnectRequest): Observable<BaseEvent>;\n  abstract isRunning(request: AgentRunnerIsRunningRequest): Promise<boolean>;\n  abstract stop(request: AgentRunnerStopRequest): Promise<boolean | undefined>;\n}\n"],"mappings":";;AA8BA,IAAsB,cAAtB,MAAkC"}

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

@@ -0,0 +1,223 @@
+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 _ag_ui_client = require("@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 require_agent_runner.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 === _ag_ui_client.EventType.RUN_STARTED) {
+				const messages = event.input?.messages ?? [];
+				for (const message of messages) historicMessageIds.add(message.id);
+			}
+		}
+		const nextSubject = new rxjs.ReplaySubject(Infinity);
+		const prevSubject = store.subject;
+		store.subject = nextSubject;
+		const runSubject = new rxjs.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 === _ag_ui_client.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 = (0, _copilotkit_shared.finalizeRunEvents)(currentRunEvents, { stopRequested: store.stopRequested });
+				for (const event of appendedEvents) {
+					runSubject.next(event);
+					nextSubject.next(event);
+				}
+				if (store.currentRunId) {
+					const compactedEvents = (0, _ag_ui_client.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 = (0, _copilotkit_shared.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 = (0, _ag_ui_client.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 rxjs.ReplaySubject(Infinity);
+		if (!store) {
+			connectionSubject.complete();
+			return connectionSubject.asObservable();
+		}
+		const allHistoricEvents = [];
+		for (const run of store.historicRuns) allHistoricEvents.push(...run.events);
+		const compactedEvents = (0, _ag_ui_client.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
+exports.InMemoryAgentRunner = InMemoryAgentRunner;
+//# sourceMappingURL=in-memory.cjs.map