agents 0.0.0-1eac06e → 0.0.0-1f8c517

package/dist/chunk-UJVEAURM.js

@@ -0,0 +1,150 @@
+// src/ai-chat-v5-migration.ts
+var STATE_MAP = {
+  "partial-call": "input-streaming",
+  call: "input-available",
+  result: "output-available",
+  error: "output-error"
+};
+function isUIMessage(message) {
+  return typeof message === "object" && message !== null && "parts" in message && Array.isArray(message.parts);
+}
+function isLegacyMessage(message) {
+  return typeof message === "object" && message !== null && "role" in message && "content" in message && typeof message.role === "string" && typeof message.content === "string";
+}
+function isCorruptArrayMessage(message) {
+  return typeof message === "object" && message !== null && "role" in message && "content" in message && typeof message.role === "string" && Array.isArray(message.content) && !("parts" in message);
+}
+function autoTransformMessage(message, index = 0) {
+  if (isUIMessage(message)) {
+    return message;
+  }
+  const parts = [];
+  if (message.reasoning) {
+    parts.push({
+      type: "reasoning",
+      text: message.reasoning
+    });
+  }
+  if (message.toolInvocations && Array.isArray(message.toolInvocations)) {
+    message.toolInvocations.forEach((inv) => {
+      if (typeof inv === "object" && inv !== null && "toolName" in inv) {
+        const invObj = inv;
+        parts.push({
+          type: `tool-${invObj.toolName}`,
+          toolCallId: invObj.toolCallId,
+          state: STATE_MAP[invObj.state] || "input-available",
+          input: invObj.args,
+          output: invObj.result !== void 0 ? invObj.result : null
+        });
+      }
+    });
+  }
+  if (message.parts && Array.isArray(message.parts)) {
+    message.parts.forEach((part) => {
+      if (typeof part === "object" && part !== null && "type" in part) {
+        const partObj = part;
+        if (partObj.type === "file") {
+          parts.push({
+            type: "file",
+            url: partObj.url || (partObj.data ? `data:${partObj.mimeType || partObj.mediaType};base64,${partObj.data}` : void 0),
+            mediaType: partObj.mediaType || partObj.mimeType,
+            filename: partObj.filename
+          });
+        }
+      }
+    });
+  }
+  if (Array.isArray(message.content)) {
+    message.content.forEach((item) => {
+      if (typeof item === "object" && item !== null && "text" in item) {
+        const itemObj = item;
+        parts.push({
+          type: itemObj.type || "text",
+          text: itemObj.text || ""
+        });
+      }
+    });
+  }
+  if (!parts.length && message.content !== void 0) {
+    parts.push({
+      type: "text",
+      text: typeof message.content === "string" ? message.content : JSON.stringify(message.content)
+    });
+  }
+  if (!parts.length) {
+    parts.push({
+      type: "text",
+      text: typeof message === "string" ? message : JSON.stringify(message)
+    });
+  }
+  return {
+    id: message.id || `msg-${index}`,
+    role: message.role === "data" ? "system" : message.role || "user",
+    parts
+  };
+}
+function migrateToUIMessage(message) {
+  return autoTransformMessage(message);
+}
+function autoTransformMessages(messages) {
+  return messages.map((msg, i) => autoTransformMessage(msg, i));
+}
+function migrateMessagesToUIFormat(messages) {
+  return autoTransformMessages(messages);
+}
+function needsMigration(messages) {
+  return messages.some((message) => {
+    if (isUIMessage(message)) {
+      return false;
+    }
+    if (isCorruptArrayMessage(message)) {
+      return true;
+    }
+    if (isLegacyMessage(message)) {
+      return true;
+    }
+    return true;
+  });
+}
+function analyzeCorruption(messages) {
+  const stats = {
+    total: messages.length,
+    clean: 0,
+    legacyString: 0,
+    corruptArray: 0,
+    unknown: 0,
+    examples: {}
+  };
+  for (const message of messages) {
+    if (isUIMessage(message)) {
+      stats.clean++;
+    } else if (isCorruptArrayMessage(message)) {
+      stats.corruptArray++;
+      if (!stats.examples.corruptArray) {
+        stats.examples.corruptArray = message;
+      }
+    } else if (isLegacyMessage(message)) {
+      stats.legacyString++;
+      if (!stats.examples.legacyString) {
+        stats.examples.legacyString = message;
+      }
+    } else {
+      stats.unknown++;
+      if (!stats.examples.unknown) {
+        stats.examples.unknown = message;
+      }
+    }
+  }
+  return stats;
+}
+
+export {
+  isUIMessage,
+  autoTransformMessage,
+  migrateToUIMessage,
+  autoTransformMessages,
+  migrateMessagesToUIFormat,
+  needsMigration,
+  analyzeCorruption
+};
+//# sourceMappingURL=chunk-UJVEAURM.js.map

package/dist/chunk-UJVEAURM.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ai-chat-v5-migration.ts"],"sourcesContent":["import type { UIMessage } from \"ai\";\n\n/**\n * AI SDK v5 Migration following https://jhak.im/blog/ai-sdk-migration-handling-previously-saved-messages\n * Using exact types from the official AI SDK documentation\n */\n\n/**\n * AI SDK v5 Message Part types reference (from official AI SDK documentation)\n *\n * The migration logic below transforms legacy messages to match these official AI SDK v5 formats:\n * - TextUIPart: { type: \"text\", text: string, state?: \"streaming\" | \"done\" }\n * - ReasoningUIPart: { type: \"reasoning\", text: string, state?: \"streaming\" | \"done\", providerMetadata?: Record<string, unknown> }\n * - FileUIPart: { type: \"file\", mediaType: string, filename?: string, url: string }\n * - ToolUIPart: { type: `tool-${string}`, toolCallId: string, state: \"input-streaming\" | \"input-available\" | \"output-available\" | \"output-error\", input?: Record<string, unknown>, output?: unknown, errorText?: string, providerExecuted?: boolean }\n */\n\n/**\n * Tool invocation from v4 format\n */\ntype ToolInvocation = {\n  toolCallId: string;\n  toolName: string;\n  args: Record<string, unknown>;\n  result?: unknown;\n  state: \"partial-call\" | \"call\" | \"result\" | \"error\";\n};\n\n/**\n * Legacy part from v4 format\n */\ntype LegacyPart = {\n  type: string;\n  text?: string;\n  url?: string;\n  data?: string;\n  mimeType?: string;\n  mediaType?: string;\n  filename?: string;\n};\n\n/**\n * Legacy message format from AI SDK v4\n */\nexport type LegacyMessage = {\n  id?: string;\n  role: string;\n  content: string;\n  reasoning?: string;\n  toolInvocations?: ToolInvocation[];\n  parts?: LegacyPart[];\n  [key: string]: unknown;\n};\n\n/**\n * Corrupt content item\n */\ntype CorruptContentItem = {\n  type: string;\n  text: string;\n};\n\n/**\n * Corrupted message format - has content as array instead of parts\n */\nexport type CorruptArrayMessage = {\n  id?: string;\n  role: string;\n  content: CorruptContentItem[];\n  reasoning?: string;\n  toolInvocations?: ToolInvocation[];\n  [key: string]: unknown;\n};\n\n/**\n * Union type for messages that could be in any format\n */\nexport type MigratableMessage = LegacyMessage | CorruptArrayMessage | UIMessage;\n\n/**\n * Tool call state mapping for v4 to v5 migration\n */\nconst STATE_MAP = {\n  \"partial-call\": \"input-streaming\",\n  call: \"input-available\",\n  result: \"output-available\",\n  error: \"output-error\"\n} as const;\n\n/**\n * Checks if a message is already in the UIMessage format (has parts array)\n */\nexport function isUIMessage(message: unknown): message is UIMessage {\n  return (\n    typeof message === \"object\" &&\n    message !== null &&\n    \"parts\" in message &&\n    Array.isArray((message as { parts: unknown }).parts)\n  );\n}\n\n/**\n * Type guard to check if a message is in legacy format (content as string)\n */\nfunction isLegacyMessage(message: unknown): message is LegacyMessage {\n  return (\n    typeof message === \"object\" &&\n    message !== null &&\n    \"role\" in message &&\n    \"content\" in message &&\n    typeof (message as { role: unknown }).role === \"string\" &&\n    typeof (message as { content: unknown }).content === \"string\"\n  );\n}\n\n/**\n * Type guard to check if a message has corrupted array content format\n * Detects: {role: \"user\", content: [{type: \"text\", text: \"...\"}]}\n */\nfunction isCorruptArrayMessage(\n  message: unknown\n): message is CorruptArrayMessage {\n  return (\n    typeof message === \"object\" &&\n    message !== null &&\n    \"role\" in message &&\n    \"content\" in message &&\n    typeof (message as { role: unknown }).role === \"string\" &&\n    Array.isArray((message as { content: unknown }).content) &&\n    !(\"parts\" in message) // Ensure it's not already a UIMessage\n  );\n}\n\n/**\n * Internal message part type for transformation\n */\ntype TransformMessagePart = {\n  type: string;\n  text?: string;\n  toolCallId?: string;\n  state?: string;\n  input?: Record<string, unknown>;\n  output?: unknown;\n  url?: string;\n  mediaType?: string;\n  errorText?: string;\n  filename?: string;\n};\n\n/**\n * Input message that could be in any format - using unknown for flexibility\n */\ntype InputMessage = {\n  id?: string;\n  role?: string;\n  content?: unknown;\n  reasoning?: string;\n  toolInvocations?: unknown[];\n  parts?: unknown[];\n  [key: string]: unknown;\n};\n\n/**\n * Automatic message transformer following the blog post pattern\n * Handles comprehensive migration from AI SDK v4 to v5 format\n * @param message - Message in any legacy format\n * @param index - Index for ID generation fallback\n * @returns UIMessage in v5 format\n */\nexport function autoTransformMessage(\n  message: InputMessage,\n  index = 0\n): UIMessage {\n  // Already in v5 format\n  if (isUIMessage(message)) {\n    return message;\n  }\n\n  const parts: TransformMessagePart[] = [];\n\n  // Handle reasoning transformation\n  if (message.reasoning) {\n    parts.push({\n      type: \"reasoning\",\n      text: message.reasoning\n    });\n  }\n\n  // Handle tool invocations transformation\n  if (message.toolInvocations && Array.isArray(message.toolInvocations)) {\n    message.toolInvocations.forEach((inv: unknown) => {\n      if (typeof inv === \"object\" && inv !== null && \"toolName\" in inv) {\n        const invObj = inv as ToolInvocation;\n        parts.push({\n          type: `tool-${invObj.toolName}`,\n          toolCallId: invObj.toolCallId,\n          state:\n            STATE_MAP[invObj.state as keyof typeof STATE_MAP] ||\n            \"input-available\",\n          input: invObj.args,\n          output: invObj.result !== undefined ? invObj.result : null\n        });\n      }\n    });\n  }\n\n  // Handle file parts transformation\n  if (message.parts && Array.isArray(message.parts)) {\n    message.parts.forEach((part: unknown) => {\n      if (typeof part === \"object\" && part !== null && \"type\" in part) {\n        const partObj = part as LegacyPart;\n        if (partObj.type === \"file\") {\n          parts.push({\n            type: \"file\",\n            url:\n              partObj.url ||\n              (partObj.data\n                ? `data:${partObj.mimeType || partObj.mediaType};base64,${partObj.data}`\n                : undefined),\n            mediaType: partObj.mediaType || partObj.mimeType,\n            filename: partObj.filename\n          });\n        }\n      }\n    });\n  }\n\n  // Handle corrupt array format: {role: \"user\", content: [{type: \"text\", text: \"...\"}]}\n  if (Array.isArray(message.content)) {\n    message.content.forEach((item: unknown) => {\n      if (typeof item === \"object\" && item !== null && \"text\" in item) {\n        const itemObj = item as CorruptContentItem;\n        parts.push({\n          type: itemObj.type || \"text\",\n          text: itemObj.text || \"\"\n        });\n      }\n    });\n  }\n\n  // Fallback: convert plain content to text part\n  if (!parts.length && message.content !== undefined) {\n    parts.push({\n      type: \"text\",\n      text:\n        typeof message.content === \"string\"\n          ? message.content\n          : JSON.stringify(message.content)\n    });\n  }\n\n  // If still no parts, create a default text part\n  if (!parts.length) {\n    parts.push({\n      type: \"text\",\n      text: typeof message === \"string\" ? message : JSON.stringify(message)\n    });\n  }\n\n  return {\n    id: message.id || `msg-${index}`,\n    role:\n      message.role === \"data\"\n        ? \"system\"\n        : (message.role as \"user\" | \"assistant\" | \"system\") || \"user\",\n    parts: parts as UIMessage[\"parts\"]\n  };\n}\n\n/**\n * Legacy single message migration for backward compatibility\n */\nexport function migrateToUIMessage(message: MigratableMessage): UIMessage {\n  return autoTransformMessage(message as InputMessage);\n}\n\n/**\n * Automatic message transformer for arrays following the blog post pattern\n * @param messages - Array of messages in any format\n * @returns Array of UIMessages in v5 format\n */\nexport function autoTransformMessages(messages: unknown[]): UIMessage[] {\n  return messages.map((msg, i) => autoTransformMessage(msg as InputMessage, i));\n}\n\n/**\n * Migrates an array of messages to UIMessage format (legacy compatibility)\n * @param messages - Array of messages in old or new format\n * @returns Array of UIMessages in the new format\n */\nexport function migrateMessagesToUIFormat(\n  messages: MigratableMessage[]\n): UIMessage[] {\n  return autoTransformMessages(messages as InputMessage[]);\n}\n\n/**\n * Checks if any messages in an array need migration\n * @param messages - Array of messages to check\n * @returns true if any messages are not in proper UIMessage format\n */\nexport function needsMigration(messages: unknown[]): boolean {\n  return messages.some((message) => {\n    // If it's already a UIMessage, no migration needed\n    if (isUIMessage(message)) {\n      return false;\n    }\n\n    // Check for corrupt array format specifically\n    if (isCorruptArrayMessage(message)) {\n      return true;\n    }\n\n    // Check for legacy string format\n    if (isLegacyMessage(message)) {\n      return true;\n    }\n\n    // Any other format needs migration\n    return true;\n  });\n}\n\n/**\n * Analyzes the corruption types in a message array for debugging\n * @param messages - Array of messages to analyze\n * @returns Statistics about corruption types found\n */\nexport function analyzeCorruption(messages: unknown[]): {\n  total: number;\n  clean: number;\n  legacyString: number;\n  corruptArray: number;\n  unknown: number;\n  examples: {\n    legacyString?: unknown;\n    corruptArray?: unknown;\n    unknown?: unknown;\n  };\n} {\n  const stats = {\n    total: messages.length,\n    clean: 0,\n    legacyString: 0,\n    corruptArray: 0,\n    unknown: 0,\n    examples: {} as {\n      legacyString?: unknown;\n      corruptArray?: unknown;\n      unknown?: unknown;\n    }\n  };\n\n  for (const message of messages) {\n    if (isUIMessage(message)) {\n      stats.clean++;\n    } else if (isCorruptArrayMessage(message)) {\n      stats.corruptArray++;\n      if (!stats.examples.corruptArray) {\n        stats.examples.corruptArray = message;\n      }\n    } else if (isLegacyMessage(message)) {\n      stats.legacyString++;\n      if (!stats.examples.legacyString) {\n        stats.examples.legacyString = message;\n      }\n    } else {\n      stats.unknown++;\n      if (!stats.examples.unknown) {\n        stats.examples.unknown = message;\n      }\n    }\n  }\n\n  return stats;\n}\n"],"mappings":";AAkFA,IAAM,YAAY;AAAA,EAChB,gBAAgB;AAAA,EAChB,MAAM;AAAA,EACN,QAAQ;AAAA,EACR,OAAO;AACT;AAKO,SAAS,YAAY,SAAwC;AAClE,SACE,OAAO,YAAY,YACnB,YAAY,QACZ,WAAW,WACX,MAAM,QAAS,QAA+B,KAAK;AAEvD;AAKA,SAAS,gBAAgB,SAA4C;AACnE,SACE,OAAO,YAAY,YACnB,YAAY,QACZ,UAAU,WACV,aAAa,WACb,OAAQ,QAA8B,SAAS,YAC/C,OAAQ,QAAiC,YAAY;AAEzD;AAMA,SAAS,sBACP,SACgC;AAChC,SACE,OAAO,YAAY,YACnB,YAAY,QACZ,UAAU,WACV,aAAa,WACb,OAAQ,QAA8B,SAAS,YAC/C,MAAM,QAAS,QAAiC,OAAO,KACvD,EAAE,WAAW;AAEjB;AAsCO,SAAS,qBACd,SACA,QAAQ,GACG;AAEX,MAAI,YAAY,OAAO,GAAG;AACxB,WAAO;AAAA,EACT;AAEA,QAAM,QAAgC,CAAC;AAGvC,MAAI,QAAQ,WAAW;AACrB,UAAM,KAAK;AAAA,MACT,MAAM;AAAA,MACN,MAAM,QAAQ;AAAA,IAChB,CAAC;AAAA,EACH;AAGA,MAAI,QAAQ,mBAAmB,MAAM,QAAQ,QAAQ,eAAe,GAAG;AACrE,YAAQ,gBAAgB,QAAQ,CAAC,QAAiB;AAChD,UAAI,OAAO,QAAQ,YAAY,QAAQ,QAAQ,cAAc,KAAK;AAChE,cAAM,SAAS;AACf,cAAM,KAAK;AAAA,UACT,MAAM,QAAQ,OAAO,QAAQ;AAAA,UAC7B,YAAY,OAAO;AAAA,UACnB,OACE,UAAU,OAAO,KAA+B,KAChD;AAAA,UACF,OAAO,OAAO;AAAA,UACd,QAAQ,OAAO,WAAW,SAAY,OAAO,SAAS;AAAA,QACxD,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AAGA,MAAI,QAAQ,SAAS,MAAM,QAAQ,QAAQ,KAAK,GAAG;AACjD,YAAQ,MAAM,QAAQ,CAAC,SAAkB;AACvC,UAAI,OAAO,SAAS,YAAY,SAAS,QAAQ,UAAU,MAAM;AAC/D,cAAM,UAAU;AAChB,YAAI,QAAQ,SAAS,QAAQ;AAC3B,gBAAM,KAAK;AAAA,YACT,MAAM;AAAA,YACN,KACE,QAAQ,QACP,QAAQ,OACL,QAAQ,QAAQ,YAAY,QAAQ,SAAS,WAAW,QAAQ,IAAI,KACpE;AAAA,YACN,WAAW,QAAQ,aAAa,QAAQ;AAAA,YACxC,UAAU,QAAQ;AAAA,UACpB,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAGA,MAAI,MAAM,QAAQ,QAAQ,OAAO,GAAG;AAClC,YAAQ,QAAQ,QAAQ,CAAC,SAAkB;AACzC,UAAI,OAAO,SAAS,YAAY,SAAS,QAAQ,UAAU,MAAM;AAC/D,cAAM,UAAU;AAChB,cAAM,KAAK;AAAA,UACT,MAAM,QAAQ,QAAQ;AAAA,UACtB,MAAM,QAAQ,QAAQ;AAAA,QACxB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AAGA,MAAI,CAAC,MAAM,UAAU,QAAQ,YAAY,QAAW;AAClD,UAAM,KAAK;AAAA,MACT,MAAM;AAAA,MACN,MACE,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,OAAO;AAAA,IACtC,CAAC;AAAA,EACH;AAGA,MAAI,CAAC,MAAM,QAAQ;AACjB,UAAM,KAAK;AAAA,MACT,MAAM;AAAA,MACN,MAAM,OAAO,YAAY,WAAW,UAAU,KAAK,UAAU,OAAO;AAAA,IACtE,CAAC;AAAA,EACH;AAEA,SAAO;AAAA,IACL,IAAI,QAAQ,MAAM,OAAO,KAAK;AAAA,IAC9B,MACE,QAAQ,SAAS,SACb,WACC,QAAQ,QAA4C;AAAA,IAC3D;AAAA,EACF;AACF;AAKO,SAAS,mBAAmB,SAAuC;AACxE,SAAO,qBAAqB,OAAuB;AACrD;AAOO,SAAS,sBAAsB,UAAkC;AACtE,SAAO,SAAS,IAAI,CAAC,KAAK,MAAM,qBAAqB,KAAqB,CAAC,CAAC;AAC9E;AAOO,SAAS,0BACd,UACa;AACb,SAAO,sBAAsB,QAA0B;AACzD;AAOO,SAAS,eAAe,UAA8B;AAC3D,SAAO,SAAS,KAAK,CAAC,YAAY;AAEhC,QAAI,YAAY,OAAO,GAAG;AACxB,aAAO;AAAA,IACT;AAGA,QAAI,sBAAsB,OAAO,GAAG;AAClC,aAAO;AAAA,IACT;AAGA,QAAI,gBAAgB,OAAO,GAAG;AAC5B,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT,CAAC;AACH;AAOO,SAAS,kBAAkB,UAWhC;AACA,QAAM,QAAQ;AAAA,IACZ,OAAO,SAAS;AAAA,IAChB,OAAO;AAAA,IACP,cAAc;AAAA,IACd,cAAc;AAAA,IACd,SAAS;AAAA,IACT,UAAU,CAAC;AAAA,EAKb;AAEA,aAAW,WAAW,UAAU;AAC9B,QAAI,YAAY,OAAO,GAAG;AACxB,YAAM;AAAA,IACR,WAAW,sBAAsB,OAAO,GAAG;AACzC,YAAM;AACN,UAAI,CAAC,MAAM,SAAS,cAAc;AAChC,cAAM,SAAS,eAAe;AAAA,MAChC;AAAA,IACF,WAAW,gBAAgB,OAAO,GAAG;AACnC,YAAM;AACN,UAAI,CAAC,MAAM,SAAS,cAAc;AAChC,cAAM,SAAS,eAAe;AAAA,MAChC;AAAA,IACF,OAAO;AACL,YAAM;AACN,UAAI,CAAC,MAAM,SAAS,SAAS;AAC3B,cAAM,SAAS,UAAU;AAAA,MAC3B;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;","names":[]}

package/dist/{chunk-2DJ6XAU6.js → chunk-YDUDMOL6.js}

@@ -1,12 +1,12 @@
 import {
   MCPClientManager
-} from "./chunk-XJR75HO7.js";
+} from "./chunk-MH46VMM4.js";
 import {
   DurableObjectOAuthClientProvider
-} from "./chunk-PVQZBKN7.js";
+} from "./chunk-LL2AFX7V.js";
 import {
   camelCaseToKebabCase
-} from "./chunk-KUH345EY.js";
+} from "./chunk-QEVM4BVL.js";
 
 // src/index.ts
 import { AsyncLocalStorage } from "async_hooks";
@@ -19,13 +19,13 @@ import {
   routePartykitRequest
 } from "partyserver";
 function isRPCRequest(msg) {
-  return typeof msg === "object" && msg !== null && "type" in msg && msg.type === "rpc" && "id" in msg && typeof msg.id === "string" && "method" in msg && typeof msg.method === "string" && "args" in msg && Array.isArray(msg.args);
+  return typeof msg === "object" && msg !== null && "type" in msg && msg.type === "rpc" /* RPC */ && "id" in msg && typeof msg.id === "string" && "method" in msg && typeof msg.method === "string" && "args" in msg && Array.isArray(msg.args);
 }
 function isStateUpdateMessage(msg) {
-  return typeof msg === "object" && msg !== null && "type" in msg && msg.type === "cf_agent_state" && "state" in msg;
+  return typeof msg === "object" && msg !== null && "type" in msg && msg.type === "cf_agent_state" /* CF_AGENT_STATE */ && "state" in msg;
 }
 var callableMetadata = /* @__PURE__ */ new Map();
-function unstable_callable(metadata = {}) {
+function callable(metadata = {}) {
   return function callableDecorator(target, context) {
     if (!callableMetadata.has(target)) {
       callableMetadata.set(target, metadata);
@@ -33,6 +33,16 @@ function unstable_callable(metadata = {}) {
     return target;
   };
 }
+var didWarnAboutUnstableCallable = false;
+var unstable_callable = (metadata = {}) => {
+  if (!didWarnAboutUnstableCallable) {
+    didWarnAboutUnstableCallable = true;
+    console.warn(
+      "unstable_callable is deprecated, use callable instead. unstable_callable will be removed in the next major version."
+    );
+  }
+  callable(metadata);
+};
 function getNextCronTime(cron) {
   const interval = parseCronExpression(cron);
   return interval.getNextDate();
@@ -55,7 +65,10 @@ function getCurrentAgent() {
 }
 function withAgentContext(method) {
   return function(...args) {
-    const { connection, request, email } = getCurrentAgent();
+    const { connection, request, email, agent } = getCurrentAgent();
+    if (agent === this) {
+      return method.apply(this, args);
+    }
     return agentContext.run({ agent: this, connection, request, email }, () => {
       return method.apply(this, args);
     });
@@ -110,7 +123,10 @@ var _Agent = class _Agent extends Server {
                   {
                     displayMessage: `Schedule ${row.id} executed`,
                     id: nanoid(),
-                    payload: row,
+                    payload: {
+                      callback: row.callback,
+                      id: row.id
+                    },
                     timestamp: Date.now(),
                     type: "schedule:execute"
                   },
@@ -190,7 +206,7 @@ var _Agent = class _Agent extends Server {
             this.broadcast(
               JSON.stringify({
                 mcp: this.getMcpServers(),
-                type: "cf_agent_mcp_servers"
+                type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
               })
             );
             return new Response("<script>window.close();</script>", {
@@ -242,10 +258,8 @@ var _Agent = class _Agent extends Server {
                   displayMessage: `RPC call to ${method}`,
                   id: nanoid(),
                   payload: {
-                    args,
                     method,
-                    streaming: metadata?.streaming,
-                    success: true
+                    streaming: metadata?.streaming
                   },
                   timestamp: Date.now(),
                   type: "rpc"
@@ -257,7 +271,7 @@ var _Agent = class _Agent extends Server {
                 id,
                 result,
                 success: true,
-                type: "rpc"
+                type: "rpc" /* RPC */
               };
               connection.send(JSON.stringify(response));
             } catch (e) {
@@ -265,7 +279,7 @@ var _Agent = class _Agent extends Server {
                 error: e instanceof Error ? e.message : "Unknown error occurred",
                 id: parsed.id,
                 success: false,
-                type: "rpc"
+                type: "rpc" /* RPC */
               };
               connection.send(JSON.stringify(response));
               console.error("RPC error:", e);
@@ -280,41 +294,39 @@ var _Agent = class _Agent extends Server {
     this.onConnect = (connection, ctx2) => {
       return agentContext.run(
         { agent: this, connection, request: ctx2.request, email: void 0 },
-        async () => {
-          setTimeout(() => {
-            if (this.state) {
-              connection.send(
-                JSON.stringify({
-                  state: this.state,
-                  type: "cf_agent_state"
-                })
-              );
-            }
+        () => {
+          if (this.state) {
             connection.send(
               JSON.stringify({
-                mcp: this.getMcpServers(),
-                type: "cf_agent_mcp_servers"
+                state: this.state,
+                type: "cf_agent_state" /* CF_AGENT_STATE */
               })
             );
-            this.observability?.emit(
-              {
-                displayMessage: "Connection established",
-                id: nanoid(),
-                payload: {
-                  connectionId: connection.id
-                },
-                timestamp: Date.now(),
-                type: "connect"
+          }
+          connection.send(
+            JSON.stringify({
+              mcp: this.getMcpServers(),
+              type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
+            })
+          );
+          this.observability?.emit(
+            {
+              displayMessage: "Connection established",
+              id: nanoid(),
+              payload: {
+                connectionId: connection.id
               },
-              this.ctx
-            );
-            return this._tryCatch(() => _onConnect(connection, ctx2));
-          }, 20);
+              timestamp: Date.now(),
+              type: "connect"
+            },
+            this.ctx
+          );
+          return this._tryCatch(() => _onConnect(connection, ctx2));
         }
       );
     };
     const _onStart = this.onStart.bind(this);
-    this.onStart = async () => {
+    this.onStart = async (props) => {
       return agentContext.run(
         {
           agent: this,
@@ -323,13 +335,19 @@ var _Agent = class _Agent extends Server {
           email: void 0
         },
         async () => {
-          const servers = this.sql`
+          await this._tryCatch(() => {
+            const servers = this.sql`
             SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
           `;
-          if (servers && Array.isArray(servers) && servers.length > 0) {
-            Promise.allSettled(
-              servers.map((server) => {
-                return this._connectToMcpServerInternal(
+            this.broadcast(
+              JSON.stringify({
+                mcp: this.getMcpServers(),
+                type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
+              })
+            );
+            if (servers && Array.isArray(servers) && servers.length > 0) {
+              servers.forEach((server) => {
+                this._connectToMcpServerInternal(
                   server.name,
                   server.server_url,
                   server.callback_url,
@@ -338,18 +356,29 @@ var _Agent = class _Agent extends Server {
                     id: server.id,
                     oauthClientId: server.client_id ?? void 0
                   }
-                );
-              })
-            ).then((_results) => {
-              this.broadcast(
-                JSON.stringify({
-                  mcp: this.getMcpServers(),
-                  type: "cf_agent_mcp_servers"
-                })
-              );
-            });
-          }
-          await this._tryCatch(() => _onStart());
+                ).then(() => {
+                  this.broadcast(
+                    JSON.stringify({
+                      mcp: this.getMcpServers(),
+                      type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
+                    })
+                  );
+                }).catch((error) => {
+                  console.error(
+                    `Error connecting to MCP server: ${server.name} (${server.server_url})`,
+                    error
+                  );
+                  this.broadcast(
+                    JSON.stringify({
+                      mcp: this.getMcpServers(),
+                      type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
+                    })
+                  );
+                });
+              });
+            }
+            return _onStart(props);
+          });
         }
       );
     };
@@ -400,7 +429,6 @@ var _Agent = class _Agent extends Server {
     }
   }
   _setStateInternal(state, source = "server") {
-    const previousState = this._state;
     this._state = state;
     this.sql`
     INSERT OR REPLACE INTO cf_agents_state (id, state)
@@ -413,7 +441,7 @@ var _Agent = class _Agent extends Server {
     this.broadcast(
       JSON.stringify({
         state,
-        type: "cf_agent_state"
+        type: "cf_agent_state" /* CF_AGENT_STATE */
       }),
       source !== "server" ? [source.id] : []
     );
@@ -426,10 +454,7 @@ var _Agent = class _Agent extends Server {
             {
               displayMessage: "State updated",
               id: nanoid(),
-              payload: {
-                previousState,
-                state
-              },
+              payload: {},
               timestamp: Date.now(),
               type: "state:update"
             },
@@ -546,28 +571,22 @@ var _Agent = class _Agent extends Server {
     while (proto && proto !== Object.prototype && depth < 10) {
       const methodNames = Object.getOwnPropertyNames(proto);
       for (const methodName of methodNames) {
-        if (baseMethods.has(methodName) || methodName.startsWith("_") || typeof this[methodName] !== "function") {
+        const descriptor = Object.getOwnPropertyDescriptor(proto, methodName);
+        if (baseMethods.has(methodName) || methodName.startsWith("_") || !descriptor || !!descriptor.get || typeof descriptor.value !== "function") {
           continue;
         }
-        if (!baseMethods.has(methodName)) {
-          const descriptor = Object.getOwnPropertyDescriptor(proto, methodName);
-          if (descriptor && typeof descriptor.value === "function") {
-            const wrappedFunction = withAgentContext(
-              // biome-ignore lint/suspicious/noExplicitAny: I can't typescript
-              this[methodName]
-              // biome-ignore lint/suspicious/noExplicitAny: I can't typescript
-            );
-            if (this._isCallable(methodName)) {
-              callableMetadata.set(
-                wrappedFunction,
-                callableMetadata.get(
-                  this[methodName]
-                )
-              );
-            }
-            this.constructor.prototype[methodName] = wrappedFunction;
-          }
+        const wrappedFunction = withAgentContext(
+          // biome-ignore lint/suspicious/noExplicitAny: I can't typescript
+          this[methodName]
+          // biome-ignore lint/suspicious/noExplicitAny: I can't typescript
+        );
+        if (this._isCallable(methodName)) {
+          callableMetadata.set(
+            wrappedFunction,
+            callableMetadata.get(this[methodName])
+          );
         }
+        this.constructor.prototype[methodName] = wrappedFunction;
       }
       proto = Object.getPrototypeOf(proto);
       depth++;
@@ -714,7 +733,10 @@ var _Agent = class _Agent extends Server {
       {
         displayMessage: `Schedule ${schedule.id} created`,
         id: nanoid(),
-        payload: schedule,
+        payload: {
+          callback,
+          id
+        },
         timestamp: Date.now(),
         type: "schedule:create"
       },
@@ -849,7 +871,10 @@ var _Agent = class _Agent extends Server {
         {
           displayMessage: `Schedule ${id} cancelled`,
           id: nanoid(),
-          payload: schedule,
+          payload: {
+            callback: schedule.callback,
+            id: schedule.id
+          },
           timestamp: Date.now(),
           type: "schedule:cancel"
         },
@@ -935,7 +960,7 @@ var _Agent = class _Agent extends Server {
     this.broadcast(
       JSON.stringify({
         mcp: this.getMcpServers(),
-        type: "cf_agent_mcp_servers"
+        type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
       })
     );
     return result;
@@ -988,7 +1013,7 @@ var _Agent = class _Agent extends Server {
     this.broadcast(
       JSON.stringify({
         mcp: this.getMcpServers(),
-        type: "cf_agent_mcp_servers"
+        type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
       })
     );
   }
@@ -1206,7 +1231,7 @@ var StreamingResponse = class {
       id: this._id,
       result: chunk,
       success: true,
-      type: "rpc"
+      type: "rpc" /* RPC */
     };
     this._connection.send(JSON.stringify(response));
   }
@@ -1224,7 +1249,7 @@ var StreamingResponse = class {
       id: this._id,
       result: finalChunk,
       success: true,
-      type: "rpc"
+      type: "rpc" /* RPC */
     };
     this._connection.send(JSON.stringify(response));
   }
@@ -1256,6 +1281,7 @@ function isLocalMode() {
 
 export {
   genericObservability,
+  callable,
   unstable_callable,
   getCurrentAgent,
   Agent,
@@ -1267,4 +1293,4 @@ export {
   getAgentByName,
   StreamingResponse
 };
-//# sourceMappingURL=chunk-2DJ6XAU6.js.map
+//# sourceMappingURL=chunk-YDUDMOL6.js.map