@copilotkitnext/runtime 1.51.4 → 1.51.5-next.1

package/dist/express.mjs

@@ -1,478 +1,4 @@
-import {
-  callAfterRequestMiddleware,
-  callBeforeRequestMiddleware,
-  createJsonRequest,
-  expectString,
-  handleConnectAgent,
-  handleGetRuntimeInfo,
-  handleRunAgent,
-  handleStopAgent,
-  handleTranscribe,
-  parseMethodCall
-} from "./chunk-5GKH3W25.mjs";
+import { createCopilotEndpointExpress } from "./endpoints/express.mjs";
+import { createCopilotEndpointSingleRouteExpress } from "./endpoints/express-single.mjs";
 
-// src/endpoints/express.ts
-import express from "express";
-import cors from "cors";
-import { logger as logger2 } from "@copilotkitnext/shared";
-
-// src/endpoints/express-utils.ts
-import { Readable } from "stream";
-import { pipeline } from "stream";
-import { promisify } from "util";
-import { logger } from "@copilotkitnext/shared";
-var streamPipeline = promisify(pipeline);
-var METHODS_WITHOUT_BODY = /* @__PURE__ */ new Set(["GET", "HEAD"]);
-function createFetchRequestFromExpress(req) {
-  const method = req.method?.toUpperCase() ?? "GET";
-  const origin = buildOrigin(req);
-  const url = `${origin}${req.originalUrl ?? req.url ?? ""}`;
-  const headers = new Headers();
-  for (const [key, value] of Object.entries(req.headers)) {
-    if (value === void 0) continue;
-    if (Array.isArray(value)) {
-      value.forEach((v) => headers.append(key, v));
-    } else {
-      headers.set(key, value);
-    }
-  }
-  const init = {
-    method,
-    headers
-  };
-  const hasParsedBody = req.body !== void 0 && req.body !== null;
-  const streamConsumed = isStreamConsumed(req, hasParsedBody);
-  if (!METHODS_WITHOUT_BODY.has(method)) {
-    const canStreamBody = req.readable !== false && !streamConsumed;
-    if (canStreamBody) {
-      init.body = Readable.toWeb(req);
-      init.duplex = "half";
-    } else if (hasParsedBody) {
-      const { body, contentType } = synthesizeBody(req.body);
-      if (contentType) {
-        headers.set("content-type", contentType);
-      }
-      headers.delete("content-length");
-      if (body !== void 0) {
-        init.body = body;
-      }
-      logger.info(
-        {
-          url,
-          method,
-          readable: req.readable,
-          readableEnded: req.readableEnded,
-          complete: req.complete
-        },
-        "Express request stream already consumed; synthesized body from parsed content"
-      );
-    } else {
-      headers.delete("content-length");
-      logger.warn(
-        { url, method },
-        "Request stream already consumed but no body was available; sending empty body"
-      );
-    }
-  }
-  const controller = new AbortController();
-  const abort = () => controller.abort();
-  req.on("aborted", abort);
-  req.on("error", abort);
-  req.on("close", () => {
-    if (req.aborted) {
-      abort();
-    }
-  });
-  init.signal = controller.signal;
-  try {
-    return new Request(url, init);
-  } catch (error) {
-    if (error instanceof TypeError && /disturbed|locked/i.test(error.message)) {
-      headers.delete("content-length");
-      delete init.duplex;
-      if (hasParsedBody) {
-        const { body, contentType } = synthesizeBody(req.body);
-        if (contentType) {
-          headers.set("content-type", contentType);
-        }
-        init.body = body;
-        logger.info(
-          { url, method },
-          "Request stream disturbed while constructing Request; reused parsed body"
-        );
-      } else {
-        init.body = void 0;
-        logger.warn(
-          { url, method },
-          "Request stream was disturbed; falling back to empty body"
-        );
-      }
-      return new Request(url, init);
-    }
-    throw error;
-  }
-}
-async function sendFetchResponse(res, response) {
-  res.status(response.status);
-  response.headers.forEach((value, key) => {
-    if (key.toLowerCase() === "content-length" && response.body !== null) {
-      return;
-    }
-    res.setHeader(key, value);
-  });
-  if (!response.body) {
-    res.end();
-    return;
-  }
-  const nodeStream = Readable.fromWeb(response.body);
-  try {
-    await streamPipeline(nodeStream, res);
-  } catch (error) {
-    res.destroy(error);
-    throw error;
-  }
-}
-function buildOrigin(req) {
-  const protocol = req.protocol || (req.secure ? "https" : "http");
-  const host = req.get("host") ?? "localhost";
-  return `${protocol}://${host}`;
-}
-function isStreamConsumed(req, hasParsedBody) {
-  const state = req._readableState;
-  return Boolean(
-    hasParsedBody || req.readableEnded || req.complete || state?.ended || state?.endEmitted
-  );
-}
-function synthesizeBody(body) {
-  if (Buffer.isBuffer(body) || body instanceof Uint8Array) {
-    return { body };
-  }
-  if (typeof body === "string") {
-    return { body };
-  }
-  if (typeof body === "object" && body !== void 0) {
-    return { body: JSON.stringify(body), contentType: "application/json" };
-  }
-  return {};
-}
-
-// src/endpoints/express.ts
-function createCopilotEndpointExpress({
-  runtime,
-  basePath
-}) {
-  const router = express.Router();
-  const normalizedBase = normalizeBasePath(basePath);
-  router.use(
-    cors({
-      origin: "*",
-      methods: ["GET", "HEAD", "PUT", "POST", "DELETE", "PATCH", "OPTIONS"],
-      allowedHeaders: ["*"]
-    })
-  );
-  router.post(
-    joinPath(normalizedBase, "/agent/:agentId/run"),
-    createRouteHandler(runtime, async ({ request, req }) => {
-      const agentId = req.params.agentId;
-      return handleRunAgent({ runtime, request, agentId });
-    })
-  );
-  router.post(
-    joinPath(normalizedBase, "/agent/:agentId/connect"),
-    createRouteHandler(runtime, async ({ request, req }) => {
-      const agentId = req.params.agentId;
-      return handleConnectAgent({ runtime, request, agentId });
-    })
-  );
-  router.post(
-    joinPath(normalizedBase, "/agent/:agentId/stop/:threadId"),
-    createRouteHandler(runtime, async ({ request, req }) => {
-      const agentId = req.params.agentId;
-      const threadId = req.params.threadId;
-      return handleStopAgent({ runtime, request, agentId, threadId });
-    })
-  );
-  router.get(
-    joinPath(normalizedBase, "/info"),
-    createRouteHandler(runtime, async ({ request }) => {
-      return handleGetRuntimeInfo({ runtime, request });
-    })
-  );
-  router.post(
-    joinPath(normalizedBase, "/transcribe"),
-    createRouteHandler(runtime, async ({ request }) => {
-      return handleTranscribe({ runtime, request });
-    })
-  );
-  router.use(joinPath(normalizedBase, "*"), (req, res) => {
-    res.status(404).json({ error: "Not found" });
-  });
-  return router;
-}
-function createRouteHandler(runtime, factory) {
-  return async (req, res, next) => {
-    const path = req.originalUrl ?? req.path;
-    let request = createFetchRequestFromExpress(req);
-    try {
-      const maybeModifiedRequest = await callBeforeRequestMiddleware({
-        runtime,
-        request,
-        path
-      });
-      if (maybeModifiedRequest) {
-        request = maybeModifiedRequest;
-      }
-    } catch (error) {
-      logger2.error(
-        { err: error, url: request.url, path },
-        "Error running before request middleware"
-      );
-      if (error instanceof Response) {
-        try {
-          await sendFetchResponse(res, error);
-        } catch (streamError) {
-          next(streamError);
-        }
-        return;
-      }
-      next(error);
-      return;
-    }
-    try {
-      const response = await factory({ request, req });
-      await sendFetchResponse(res, response);
-      callAfterRequestMiddleware({ runtime, response, path }).catch((error) => {
-        logger2.error(
-          { err: error, url: req.originalUrl ?? req.url, path },
-          "Error running after request middleware"
-        );
-      });
-    } catch (error) {
-      if (error instanceof Response) {
-        try {
-          await sendFetchResponse(res, error);
-        } catch (streamError) {
-          next(streamError);
-          return;
-        }
-        callAfterRequestMiddleware({ runtime, response: error, path }).catch(
-          (mwError) => {
-            logger2.error(
-              { err: mwError, url: req.originalUrl ?? req.url, path },
-              "Error running after request middleware"
-            );
-          }
-        );
-        return;
-      }
-      logger2.error(
-        { err: error, url: request.url, path },
-        "Error running request handler"
-      );
-      next(error);
-    }
-  };
-}
-function normalizeBasePath(path) {
-  if (!path) {
-    throw new Error("basePath must be provided for Express endpoint");
-  }
-  if (!path.startsWith("/")) {
-    return `/${path}`;
-  }
-  if (path.length > 1 && path.endsWith("/")) {
-    return path.slice(0, -1);
-  }
-  return path;
-}
-function joinPath(basePath, suffix) {
-  if (basePath === "/") {
-    return suffix.startsWith("/") ? suffix : `/${suffix}`;
-  }
-  if (!suffix) {
-    return basePath;
-  }
-  if (suffix === "*") {
-    return `${basePath}/*`;
-  }
-  return `${basePath}${suffix.startsWith("/") ? suffix : `/${suffix}`}`;
-}
-
-// src/endpoints/express-single.ts
-import express2 from "express";
-import cors2 from "cors";
-import { logger as logger3 } from "@copilotkitnext/shared";
-function createCopilotEndpointSingleRouteExpress({
-  runtime,
-  basePath
-}) {
-  const router = express2.Router();
-  const routePath = normalizeSingleRoutePath(basePath);
-  router.use(
-    cors2({
-      origin: "*",
-      methods: ["GET", "HEAD", "PUT", "POST", "DELETE", "PATCH", "OPTIONS"],
-      allowedHeaders: ["*"]
-    })
-  );
-  router.post(routePath, createSingleRouteHandler(runtime));
-  router.use((req, res) => {
-    res.status(404).json({ error: "Not found" });
-  });
-  return router;
-}
-function createSingleRouteHandler(runtime) {
-  return async (req, res, next) => {
-    const path = req.originalUrl ?? req.path;
-    let request = createFetchRequestFromExpress(req);
-    try {
-      const maybeModifiedRequest = await callBeforeRequestMiddleware({
-        runtime,
-        request,
-        path
-      });
-      if (maybeModifiedRequest) {
-        request = maybeModifiedRequest;
-      }
-    } catch (error) {
-      logger3.error(
-        { err: error, url: request.url, path },
-        "Error running before request middleware"
-      );
-      if (error instanceof Response) {
-        try {
-          await sendFetchResponse(res, error);
-        } catch (streamError) {
-          next(streamError);
-        }
-        return;
-      }
-      next(error);
-      return;
-    }
-    let methodCall;
-    try {
-      methodCall = await parseMethodCall(request);
-    } catch (error) {
-      if (error instanceof Response) {
-        logger3.warn({ url: request.url }, "Invalid single-route payload");
-        try {
-          await sendFetchResponse(res, error);
-        } catch (streamError) {
-          next(streamError);
-        }
-        return;
-      }
-      logger3.warn(
-        { err: error, url: request.url },
-        "Invalid single-route payload"
-      );
-      res.status(400).json({
-        error: "invalid_request",
-        message: error instanceof Error ? error.message : "Invalid request payload"
-      });
-      return;
-    }
-    try {
-      let response;
-      switch (methodCall.method) {
-        case "agent/run": {
-          const agentId = expectString(methodCall.params, "agentId");
-          const handlerRequest = createJsonRequest(request, methodCall.body);
-          response = await handleRunAgent({
-            runtime,
-            request: handlerRequest,
-            agentId
-          });
-          break;
-        }
-        case "agent/connect": {
-          const agentId = expectString(methodCall.params, "agentId");
-          const handlerRequest = createJsonRequest(request, methodCall.body);
-          response = await handleConnectAgent({
-            runtime,
-            request: handlerRequest,
-            agentId
-          });
-          break;
-        }
-        case "agent/stop": {
-          const agentId = expectString(methodCall.params, "agentId");
-          const threadId = expectString(methodCall.params, "threadId");
-          response = await handleStopAgent({
-            runtime,
-            request,
-            agentId,
-            threadId
-          });
-          break;
-        }
-        case "info": {
-          response = await handleGetRuntimeInfo({ runtime, request });
-          break;
-        }
-        case "transcribe": {
-          const handlerRequest = createJsonRequest(request, methodCall.body);
-          response = await handleTranscribe({
-            runtime,
-            request: handlerRequest
-          });
-          break;
-        }
-        default: {
-          const exhaustive = methodCall.method;
-          return exhaustive;
-        }
-      }
-      await sendFetchResponse(res, response);
-      callAfterRequestMiddleware({ runtime, response, path }).catch((error) => {
-        logger3.error(
-          { err: error, url: req.originalUrl ?? req.url, path },
-          "Error running after request middleware"
-        );
-      });
-    } catch (error) {
-      if (error instanceof Response) {
-        try {
-          await sendFetchResponse(res, error);
-        } catch (streamError) {
-          next(streamError);
-          return;
-        }
-        callAfterRequestMiddleware({ runtime, response: error, path }).catch(
-          (mwError) => {
-            logger3.error(
-              { err: mwError, url: req.originalUrl ?? req.url, path },
-              "Error running after request middleware"
-            );
-          }
-        );
-        return;
-      }
-      logger3.error(
-        { err: error, url: request.url, path },
-        "Error running single-route handler"
-      );
-      next(error);
-    }
-  };
-}
-function normalizeSingleRoutePath(path) {
-  if (!path) {
-    throw new Error(
-      "basePath must be provided for Express single-route endpoint"
-    );
-  }
-  if (!path.startsWith("/")) {
-    return `/${path}`;
-  }
-  if (path.length > 1 && path.endsWith("/")) {
-    return path.slice(0, -1);
-  }
-  return path;
-}
-export {
-  createCopilotEndpointExpress,
-  createCopilotEndpointSingleRouteExpress
-};
-//# sourceMappingURL=express.mjs.map
+export { createCopilotEndpointExpress, createCopilotEndpointSingleRouteExpress };

package/dist/handlers/get-runtime-info.cjs

@@ -0,0 +1,36 @@
+const require_runtime = require('../runtime.cjs');
+
+//#region src/handlers/get-runtime-info.ts
+async function handleGetRuntimeInfo({ runtime }) {
+	try {
+		const agents = await runtime.agents;
+		const runtimeInfo = {
+			version: require_runtime.VERSION,
+			agents: Object.entries(agents).reduce((acc, [name, agent]) => {
+				acc[name] = {
+					name,
+					description: agent.description,
+					className: agent.constructor.name
+				};
+				return acc;
+			}, {}),
+			audioFileTranscriptionEnabled: !!runtime.transcriptionService
+		};
+		return new Response(JSON.stringify(runtimeInfo), {
+			status: 200,
+			headers: { "Content-Type": "application/json" }
+		});
+	} catch (error) {
+		return new Response(JSON.stringify({
+			error: "Failed to retrieve runtime information",
+			message: error instanceof Error ? error.message : "Unknown error"
+		}), {
+			status: 500,
+			headers: { "Content-Type": "application/json" }
+		});
+	}
+}
+
+//#endregion
+exports.handleGetRuntimeInfo = handleGetRuntimeInfo;
+//# sourceMappingURL=get-runtime-info.cjs.map

package/dist/handlers/get-runtime-info.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-runtime-info.cjs","names":["VERSION"],"sources":["../../src/handlers/get-runtime-info.ts"],"sourcesContent":["import { CopilotRuntime } from \"../runtime\";\nimport { AgentDescription, RuntimeInfo } from \"@copilotkitnext/shared\";\nimport { VERSION } from \"../runtime\";\n\ninterface HandleGetRuntimeInfoParameters {\n  runtime: CopilotRuntime;\n  request: Request;\n}\n\nexport async function handleGetRuntimeInfo({\n  runtime,\n}: HandleGetRuntimeInfoParameters) {\n  try {\n    const agents = await runtime.agents;\n\n    const agentsDict = Object.entries(agents).reduce(\n      (acc, [name, agent]) => {\n        acc[name] = {\n          name,\n          description: agent.description,\n          className: agent.constructor.name,\n        };\n        return acc;\n      },\n      {} as Record<string, AgentDescription>,\n    );\n\n    const runtimeInfo: RuntimeInfo = {\n      version: VERSION,\n      agents: agentsDict,\n      audioFileTranscriptionEnabled: !!runtime.transcriptionService,\n    };\n\n    return new Response(JSON.stringify(runtimeInfo), {\n      status: 200,\n      headers: { \"Content-Type\": \"application/json\" },\n    });\n  } catch (error) {\n    return new Response(\n      JSON.stringify({\n        error: \"Failed to retrieve runtime information\",\n        message: error instanceof Error ? error.message : \"Unknown error\",\n      }),\n      {\n        status: 500,\n        headers: { \"Content-Type\": \"application/json\" },\n      },\n    );\n  }\n}\n"],"mappings":";;;AASA,eAAsB,qBAAqB,EACzC,WACiC;AACjC,KAAI;EACF,MAAM,SAAS,MAAM,QAAQ;EAc7B,MAAM,cAA2B;GAC/B,SAASA;GACT,QAdiB,OAAO,QAAQ,OAAO,CAAC,QACvC,KAAK,CAAC,MAAM,WAAW;AACtB,QAAI,QAAQ;KACV;KACA,aAAa,MAAM;KACnB,WAAW,MAAM,YAAY;KAC9B;AACD,WAAO;MAET,EAAE,CACH;GAKC,+BAA+B,CAAC,CAAC,QAAQ;GAC1C;AAED,SAAO,IAAI,SAAS,KAAK,UAAU,YAAY,EAAE;GAC/C,QAAQ;GACR,SAAS,EAAE,gBAAgB,oBAAoB;GAChD,CAAC;UACK,OAAO;AACd,SAAO,IAAI,SACT,KAAK,UAAU;GACb,OAAO;GACP,SAAS,iBAAiB,QAAQ,MAAM,UAAU;GACnD,CAAC,EACF;GACE,QAAQ;GACR,SAAS,EAAE,gBAAgB,oBAAoB;GAChD,CACF"}

package/dist/handlers/get-runtime-info.mjs

@@ -0,0 +1,36 @@
+import { VERSION } from "../runtime.mjs";
+
+//#region src/handlers/get-runtime-info.ts
+async function handleGetRuntimeInfo({ runtime }) {
+	try {
+		const agents = await runtime.agents;
+		const runtimeInfo = {
+			version: VERSION,
+			agents: Object.entries(agents).reduce((acc, [name, agent]) => {
+				acc[name] = {
+					name,
+					description: agent.description,
+					className: agent.constructor.name
+				};
+				return acc;
+			}, {}),
+			audioFileTranscriptionEnabled: !!runtime.transcriptionService
+		};
+		return new Response(JSON.stringify(runtimeInfo), {
+			status: 200,
+			headers: { "Content-Type": "application/json" }
+		});
+	} catch (error) {
+		return new Response(JSON.stringify({
+			error: "Failed to retrieve runtime information",
+			message: error instanceof Error ? error.message : "Unknown error"
+		}), {
+			status: 500,
+			headers: { "Content-Type": "application/json" }
+		});
+	}
+}
+
+//#endregion
+export { handleGetRuntimeInfo };
+//# sourceMappingURL=get-runtime-info.mjs.map

package/dist/handlers/get-runtime-info.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-runtime-info.mjs","names":[],"sources":["../../src/handlers/get-runtime-info.ts"],"sourcesContent":["import { CopilotRuntime } from \"../runtime\";\nimport { AgentDescription, RuntimeInfo } from \"@copilotkitnext/shared\";\nimport { VERSION } from \"../runtime\";\n\ninterface HandleGetRuntimeInfoParameters {\n  runtime: CopilotRuntime;\n  request: Request;\n}\n\nexport async function handleGetRuntimeInfo({\n  runtime,\n}: HandleGetRuntimeInfoParameters) {\n  try {\n    const agents = await runtime.agents;\n\n    const agentsDict = Object.entries(agents).reduce(\n      (acc, [name, agent]) => {\n        acc[name] = {\n          name,\n          description: agent.description,\n          className: agent.constructor.name,\n        };\n        return acc;\n      },\n      {} as Record<string, AgentDescription>,\n    );\n\n    const runtimeInfo: RuntimeInfo = {\n      version: VERSION,\n      agents: agentsDict,\n      audioFileTranscriptionEnabled: !!runtime.transcriptionService,\n    };\n\n    return new Response(JSON.stringify(runtimeInfo), {\n      status: 200,\n      headers: { \"Content-Type\": \"application/json\" },\n    });\n  } catch (error) {\n    return new Response(\n      JSON.stringify({\n        error: \"Failed to retrieve runtime information\",\n        message: error instanceof Error ? error.message : \"Unknown error\",\n      }),\n      {\n        status: 500,\n        headers: { \"Content-Type\": \"application/json\" },\n      },\n    );\n  }\n}\n"],"mappings":";;;AASA,eAAsB,qBAAqB,EACzC,WACiC;AACjC,KAAI;EACF,MAAM,SAAS,MAAM,QAAQ;EAc7B,MAAM,cAA2B;GAC/B,SAAS;GACT,QAdiB,OAAO,QAAQ,OAAO,CAAC,QACvC,KAAK,CAAC,MAAM,WAAW;AACtB,QAAI,QAAQ;KACV;KACA,aAAa,MAAM;KACnB,WAAW,MAAM,YAAY;KAC9B;AACD,WAAO;MAET,EAAE,CACH;GAKC,+BAA+B,CAAC,CAAC,QAAQ;GAC1C;AAED,SAAO,IAAI,SAAS,KAAK,UAAU,YAAY,EAAE;GAC/C,QAAQ;GACR,SAAS,EAAE,gBAAgB,oBAAoB;GAChD,CAAC;UACK,OAAO;AACd,SAAO,IAAI,SACT,KAAK,UAAU;GACb,OAAO;GACP,SAAS,iBAAiB,QAAQ,MAAM,UAAU;GACnD,CAAC,EACF;GACE,QAAQ;GACR,SAAS,EAAE,gBAAgB,oBAAoB;GAChD,CACF"}

package/dist/handlers/handle-connect.cjs

@@ -0,0 +1,102 @@
+const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
+const require_header_utils = require('./header-utils.cjs');
+let _ag_ui_client = require("@ag-ui/client");
+let _ag_ui_encoder = require("@ag-ui/encoder");
+
+//#region src/handlers/handle-connect.ts
+async function handleConnectAgent({ runtime, request, agentId }) {
+	try {
+		if (!(await runtime.agents)[agentId]) return new Response(JSON.stringify({
+			error: "Agent not found",
+			message: `Agent '${agentId}' does not exist`
+		}), {
+			status: 404,
+			headers: { "Content-Type": "application/json" }
+		});
+		let input;
+		try {
+			const requestBody = await request.json();
+			input = _ag_ui_client.RunAgentInputSchema.parse(requestBody);
+		} catch (error) {
+			console.error("Invalid connect request body:", error);
+			return new Response(JSON.stringify({
+				error: "Invalid request body",
+				details: error instanceof Error ? error.message : String(error)
+			}), {
+				status: 400,
+				headers: { "Content-Type": "application/json" }
+			});
+		}
+		const stream = new TransformStream();
+		const writer = stream.writable.getWriter();
+		const encoder = new _ag_ui_encoder.EventEncoder();
+		let streamClosed = false;
+		(async () => {
+			const forwardableHeaders = require_header_utils.extractForwardableHeaders(request);
+			runtime.runner.connect({
+				threadId: input.threadId,
+				headers: forwardableHeaders
+			}).subscribe({
+				next: async (event) => {
+					if (!request.signal.aborted && !streamClosed) try {
+						await writer.write(encoder.encode(event));
+					} catch (error) {
+						if (error instanceof Error && error.name === "AbortError") streamClosed = true;
+					}
+				},
+				error: async (error) => {
+					console.error("Error running agent:", error);
+					if (!streamClosed) try {
+						await writer.close();
+						streamClosed = true;
+					} catch {}
+				},
+				complete: async () => {
+					if (!streamClosed) try {
+						await writer.close();
+						streamClosed = true;
+					} catch {}
+				}
+			});
+		})().catch((error) => {
+			console.error("Error running agent:", error);
+			console.error("Error stack:", error instanceof Error ? error.stack : "No stack trace");
+			console.error("Error details:", {
+				name: error instanceof Error ? error.name : "Unknown",
+				message: error instanceof Error ? error.message : String(error),
+				cause: error instanceof Error ? error.cause : void 0
+			});
+			if (!streamClosed) try {
+				writer.close();
+				streamClosed = true;
+			} catch {}
+		});
+		return new Response(stream.readable, {
+			status: 200,
+			headers: {
+				"Content-Type": "text/event-stream",
+				"Cache-Control": "no-cache",
+				Connection: "keep-alive"
+			}
+		});
+	} catch (error) {
+		console.error("Error running agent:", error);
+		console.error("Error stack:", error instanceof Error ? error.stack : "No stack trace");
+		console.error("Error details:", {
+			name: error instanceof Error ? error.name : "Unknown",
+			message: error instanceof Error ? error.message : String(error),
+			cause: error instanceof Error ? error.cause : void 0
+		});
+		return new Response(JSON.stringify({
+			error: "Failed to run agent",
+			message: error instanceof Error ? error.message : "Unknown error"
+		}), {
+			status: 500,
+			headers: { "Content-Type": "application/json" }
+		});
+	}
+}
+
+//#endregion
+exports.handleConnectAgent = handleConnectAgent;
+//# sourceMappingURL=handle-connect.cjs.map

package/dist/handlers/handle-connect.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-connect.cjs","names":["RunAgentInputSchema","EventEncoder","extractForwardableHeaders"],"sources":["../../src/handlers/handle-connect.ts"],"sourcesContent":["import { RunAgentInput, RunAgentInputSchema } from \"@ag-ui/client\";\nimport { EventEncoder } from \"@ag-ui/encoder\";\nimport { CopilotRuntime } from \"../runtime\";\nimport { extractForwardableHeaders } from \"./header-utils\";\n\ninterface ConnectAgentParameters {\n  request: Request;\n  runtime: CopilotRuntime;\n  agentId: string;\n}\n\nexport async function handleConnectAgent({\n  runtime,\n  request,\n  agentId,\n}: ConnectAgentParameters) {\n  try {\n    const agents = await runtime.agents;\n\n    // Check if the requested agent exists\n    if (!agents[agentId]) {\n      return new Response(\n        JSON.stringify({\n          error: \"Agent not found\",\n          message: `Agent '${agentId}' does not exist`,\n        }),\n        {\n          status: 404,\n          headers: { \"Content-Type\": \"application/json\" },\n        },\n      );\n    }\n\n    // Parse and validate input BEFORE creating the stream\n    // so we can return a proper error response\n    let input: RunAgentInput;\n    try {\n      const requestBody = await request.json();\n      input = RunAgentInputSchema.parse(requestBody);\n    } catch (error) {\n      console.error(\"Invalid connect request body:\", error);\n      return new Response(\n        JSON.stringify({\n          error: \"Invalid request body\",\n          details: error instanceof Error ? error.message : String(error),\n        }),\n        {\n          status: 400,\n          headers: { \"Content-Type\": \"application/json\" },\n        },\n      );\n    }\n\n    const stream = new TransformStream();\n    const writer = stream.writable.getWriter();\n    const encoder = new EventEncoder();\n    let streamClosed = false;\n\n    // Process the agent connect in the background\n    (async () => {\n      const forwardableHeaders = extractForwardableHeaders(request);\n\n      runtime.runner\n        .connect({\n          threadId: input.threadId,\n          headers: forwardableHeaders,\n        })\n        .subscribe({\n          next: async (event) => {\n            if (!request.signal.aborted && !streamClosed) {\n              try {\n                await writer.write(encoder.encode(event));\n              } catch (error) {\n                if (error instanceof Error && error.name === \"AbortError\") {\n                  streamClosed = true;\n                }\n              }\n            }\n          },\n          error: async (error) => {\n            console.error(\"Error running agent:\", error);\n            if (!streamClosed) {\n              try {\n                await writer.close();\n                streamClosed = true;\n              } catch {\n                // Stream already closed\n              }\n            }\n          },\n          complete: async () => {\n            if (!streamClosed) {\n              try {\n                await writer.close();\n                streamClosed = true;\n              } catch {\n                // Stream already closed\n              }\n            }\n          },\n        });\n    })().catch((error) => {\n      console.error(\"Error running agent:\", error);\n      console.error(\n        \"Error stack:\",\n        error instanceof Error ? error.stack : \"No stack trace\",\n      );\n      console.error(\"Error details:\", {\n        name: error instanceof Error ? error.name : \"Unknown\",\n        message: error instanceof Error ? error.message : String(error),\n        cause: error instanceof Error ? error.cause : undefined,\n      });\n      if (!streamClosed) {\n        try {\n          writer.close();\n          streamClosed = true;\n        } catch {\n          // Stream already closed\n        }\n      }\n    });\n\n    // Return the SSE response\n    return new Response(stream.readable, {\n      status: 200,\n      headers: {\n        \"Content-Type\": \"text/event-stream\",\n        \"Cache-Control\": \"no-cache\",\n        Connection: \"keep-alive\",\n      },\n    });\n  } catch (error) {\n    console.error(\"Error running agent:\", error);\n    console.error(\n      \"Error stack:\",\n      error instanceof Error ? error.stack : \"No stack trace\",\n    );\n    console.error(\"Error details:\", {\n      name: error instanceof Error ? error.name : \"Unknown\",\n      message: error instanceof Error ? error.message : String(error),\n      cause: error instanceof Error ? error.cause : undefined,\n    });\n\n    return new Response(\n      JSON.stringify({\n        error: \"Failed to run agent\",\n        message: error instanceof Error ? error.message : \"Unknown error\",\n      }),\n      {\n        status: 500,\n        headers: { \"Content-Type\": \"application/json\" },\n      },\n    );\n  }\n}\n"],"mappings":";;;;;;AAWA,eAAsB,mBAAmB,EACvC,SACA,SACA,WACyB;AACzB,KAAI;AAIF,MAAI,EAHW,MAAM,QAAQ,QAGjB,SACV,QAAO,IAAI,SACT,KAAK,UAAU;GACb,OAAO;GACP,SAAS,UAAU,QAAQ;GAC5B,CAAC,EACF;GACE,QAAQ;GACR,SAAS,EAAE,gBAAgB,oBAAoB;GAChD,CACF;EAKH,IAAI;AACJ,MAAI;GACF,MAAM,cAAc,MAAM,QAAQ,MAAM;AACxC,WAAQA,kCAAoB,MAAM,YAAY;WACvC,OAAO;AACd,WAAQ,MAAM,iCAAiC,MAAM;AACrD,UAAO,IAAI,SACT,KAAK,UAAU;IACb,OAAO;IACP,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IAChE,CAAC,EACF;IACE,QAAQ;IACR,SAAS,EAAE,gBAAgB,oBAAoB;IAChD,CACF;;EAGH,MAAM,SAAS,IAAI,iBAAiB;EACpC,MAAM,SAAS,OAAO,SAAS,WAAW;EAC1C,MAAM,UAAU,IAAIC,6BAAc;EAClC,IAAI,eAAe;AAGnB,GAAC,YAAY;GACX,MAAM,qBAAqBC,+CAA0B,QAAQ;AAE7D,WAAQ,OACL,QAAQ;IACP,UAAU,MAAM;IAChB,SAAS;IACV,CAAC,CACD,UAAU;IACT,MAAM,OAAO,UAAU;AACrB,SAAI,CAAC,QAAQ,OAAO,WAAW,CAAC,aAC9B,KAAI;AACF,YAAM,OAAO,MAAM,QAAQ,OAAO,MAAM,CAAC;cAClC,OAAO;AACd,UAAI,iBAAiB,SAAS,MAAM,SAAS,aAC3C,gBAAe;;;IAKvB,OAAO,OAAO,UAAU;AACtB,aAAQ,MAAM,wBAAwB,MAAM;AAC5C,SAAI,CAAC,aACH,KAAI;AACF,YAAM,OAAO,OAAO;AACpB,qBAAe;aACT;;IAKZ,UAAU,YAAY;AACpB,SAAI,CAAC,aACH,KAAI;AACF,YAAM,OAAO,OAAO;AACpB,qBAAe;aACT;;IAKb,CAAC;MACF,CAAC,OAAO,UAAU;AACpB,WAAQ,MAAM,wBAAwB,MAAM;AAC5C,WAAQ,MACN,gBACA,iBAAiB,QAAQ,MAAM,QAAQ,iBACxC;AACD,WAAQ,MAAM,kBAAkB;IAC9B,MAAM,iBAAiB,QAAQ,MAAM,OAAO;IAC5C,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IAC/D,OAAO,iBAAiB,QAAQ,MAAM,QAAQ;IAC/C,CAAC;AACF,OAAI,CAAC,aACH,KAAI;AACF,WAAO,OAAO;AACd,mBAAe;WACT;IAIV;AAGF,SAAO,IAAI,SAAS,OAAO,UAAU;GACnC,QAAQ;GACR,SAAS;IACP,gBAAgB;IAChB,iBAAiB;IACjB,YAAY;IACb;GACF,CAAC;UACK,OAAO;AACd,UAAQ,MAAM,wBAAwB,MAAM;AAC5C,UAAQ,MACN,gBACA,iBAAiB,QAAQ,MAAM,QAAQ,iBACxC;AACD,UAAQ,MAAM,kBAAkB;GAC9B,MAAM,iBAAiB,QAAQ,MAAM,OAAO;GAC5C,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;GAC/D,OAAO,iBAAiB,QAAQ,MAAM,QAAQ;GAC/C,CAAC;AAEF,SAAO,IAAI,SACT,KAAK,UAAU;GACb,OAAO;GACP,SAAS,iBAAiB,QAAQ,MAAM,UAAU;GACnD,CAAC,EACF;GACE,QAAQ;GACR,SAAS,EAAE,gBAAgB,oBAAoB;GAChD,CACF"}