agents 0.0.0-569680f → 0.0.0-57c0db2

package/dist/{chunk-RX76B6DL.js → chunk-HS7VEROK.js}

@@ -1,12 +1,12 @@
 import {
   MCPClientManager
-} from "./chunk-TN7QOY4S.js";
+} from "./chunk-SKACXF37.js";
 import {
   DurableObjectOAuthClientProvider
-} from "./chunk-767EASBA.js";
+} from "./chunk-LL2AFX7V.js";
 import {
   camelCaseToKebabCase
-} from "./chunk-NKZZ66QY.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();
@@ -47,12 +57,24 @@ function getCurrentAgent() {
     return {
       agent: void 0,
       connection: void 0,
-      request: void 0
+      request: void 0,
+      email: void 0
     };
   }
   return store;
 }
-var Agent = class extends Server {
+function withAgentContext(method) {
+  return function(...args) {
+    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);
+    });
+  };
+}
+var _Agent = class _Agent extends Server {
   constructor(ctx, env) {
     super(ctx, env);
     this._state = DEFAULT_STATE;
@@ -67,6 +89,7 @@ var Agent = class extends Server {
      * The observability implementation to use for the Agent
      */
     this.observability = genericObservability;
+    this._flushingQueue = false;
     /**
      * Method called when an alarm fires.
      * Executes any scheduled tasks that are due.
@@ -80,52 +103,74 @@ var Agent = class extends Server {
       const result = this.sql`
       SELECT * FROM cf_agents_schedules WHERE time <= ${now}
     `;
-      for (const row of result || []) {
-        const callback = this[row.callback];
-        if (!callback) {
-          console.error(`callback ${row.callback} not found`);
-          continue;
-        }
-        await agentContext.run(
-          { agent: this, connection: void 0, request: void 0 },
-          async () => {
-            try {
-              this.observability?.emit(
-                {
-                  displayMessage: `Schedule ${row.id} executed`,
-                  id: nanoid(),
-                  payload: row,
-                  timestamp: Date.now(),
-                  type: "schedule:execute"
-                },
-                this.ctx
-              );
-              await callback.bind(this)(JSON.parse(row.payload), row);
-            } catch (e) {
-              console.error(`error executing callback "${row.callback}"`, e);
-            }
+      if (result && Array.isArray(result)) {
+        for (const row of result) {
+          const callback = this[row.callback];
+          if (!callback) {
+            console.error(`callback ${row.callback} not found`);
+            continue;
           }
-        );
-        if (row.type === "cron") {
-          const nextExecutionTime = getNextCronTime(row.cron);
-          const nextTimestamp = Math.floor(nextExecutionTime.getTime() / 1e3);
-          this.sql`
+          await agentContext.run(
+            {
+              agent: this,
+              connection: void 0,
+              request: void 0,
+              email: void 0
+            },
+            async () => {
+              try {
+                this.observability?.emit(
+                  {
+                    displayMessage: `Schedule ${row.id} executed`,
+                    id: nanoid(),
+                    payload: {
+                      callback: row.callback,
+                      id: row.id
+                    },
+                    timestamp: Date.now(),
+                    type: "schedule:execute"
+                  },
+                  this.ctx
+                );
+                await callback.bind(this)(JSON.parse(row.payload), row);
+              } catch (e) {
+                console.error(`error executing callback "${row.callback}"`, e);
+              }
+            }
+          );
+          if (row.type === "cron") {
+            const nextExecutionTime = getNextCronTime(row.cron);
+            const nextTimestamp = Math.floor(nextExecutionTime.getTime() / 1e3);
+            this.sql`
           UPDATE cf_agents_schedules SET time = ${nextTimestamp} WHERE id = ${row.id}
         `;
-        } else {
-          this.sql`
+          } else {
+            this.sql`
           DELETE FROM cf_agents_schedules WHERE id = ${row.id}
         `;
+          }
         }
       }
       await this._scheduleNextAlarm();
     };
+    if (!wrappedClasses.has(this.constructor)) {
+      this._autoWrapCustomMethods();
+      wrappedClasses.add(this.constructor);
+    }
     this.sql`
       CREATE TABLE IF NOT EXISTS cf_agents_state (
         id TEXT PRIMARY KEY NOT NULL,
         state TEXT
       )
     `;
+    this.sql`
+      CREATE TABLE IF NOT EXISTS cf_agents_queues (
+        id TEXT PRIMARY KEY NOT NULL,
+        payload TEXT,
+        callback TEXT,
+        created_at INTEGER DEFAULT (unixepoch())
+      )
+    `;
     void this.ctx.blockConcurrencyWhile(async () => {
       return this._tryCatch(async () => {
         this.sql`
@@ -157,14 +202,14 @@ var Agent = class extends Server {
     const _onRequest = this.onRequest.bind(this);
     this.onRequest = (request) => {
       return agentContext.run(
-        { agent: this, connection: void 0, request },
+        { agent: this, connection: void 0, request, email: void 0 },
         async () => {
           if (this.mcp.isCallbackRequest(request)) {
             await this.mcp.handleCallbackRequest(request);
             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>", {
@@ -179,7 +224,7 @@ var Agent = class extends Server {
     const _onMessage = this.onMessage.bind(this);
     this.onMessage = async (connection, message) => {
       return agentContext.run(
-        { agent: this, connection, request: void 0 },
+        { agent: this, connection, request: void 0, email: void 0 },
         async () => {
           if (typeof message !== "string") {
             return this._tryCatch(() => _onMessage(connection, message));
@@ -216,10 +261,8 @@ var Agent = class 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"
@@ -231,7 +274,7 @@ var Agent = class extends Server {
                 id,
                 result,
                 success: true,
-                type: "rpc"
+                type: "rpc" /* RPC */
               };
               connection.send(JSON.stringify(response));
             } catch (e) {
@@ -239,7 +282,7 @@ var Agent = class 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);
@@ -253,70 +296,92 @@ var Agent = class extends Server {
     const _onConnect = this.onConnect.bind(this);
     this.onConnect = (connection, ctx2) => {
       return agentContext.run(
-        { agent: this, connection, request: ctx2.request },
-        async () => {
-          setTimeout(() => {
-            if (this.state) {
-              connection.send(
-                JSON.stringify({
-                  state: this.state,
-                  type: "cf_agent_state"
-                })
-              );
-            }
+        { agent: this, connection, request: ctx2.request, email: void 0 },
+        () => {
+          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, connection: void 0, request: void 0 },
+        {
+          agent: this,
+          connection: void 0,
+          request: void 0,
+          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;
           `;
-          Promise.allSettled(
-            servers.map((server) => {
-              return this._connectToMcpServerInternal(
-                server.name,
-                server.server_url,
-                server.callback_url,
-                server.server_options ? JSON.parse(server.server_options) : void 0,
-                {
-                  id: server.id,
-                  oauthClientId: server.client_id ?? void 0
-                }
-              );
-            })
-          ).then((_results) => {
             this.broadcast(
               JSON.stringify({
                 mcp: this.getMcpServers(),
-                type: "cf_agent_mcp_servers"
+                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,
+                  server.server_options ? JSON.parse(server.server_options) : void 0,
+                  {
+                    id: server.id,
+                    oauthClientId: server.client_id ?? void 0
+                  }
+                ).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);
           });
-          await this._tryCatch(() => _onStart());
         }
       );
     };
@@ -367,7 +432,6 @@ var Agent = class extends Server {
     }
   }
   _setStateInternal(state, source = "server") {
-    const previousState = this._state;
     this._state = state;
     this.sql`
     INSERT OR REPLACE INTO cf_agents_state (id, state)
@@ -380,23 +444,20 @@ var Agent = class extends Server {
     this.broadcast(
       JSON.stringify({
         state,
-        type: "cf_agent_state"
+        type: "cf_agent_state" /* CF_AGENT_STATE */
       }),
       source !== "server" ? [source.id] : []
     );
     return this._tryCatch(() => {
-      const { connection, request } = agentContext.getStore() || {};
+      const { connection, request, email } = agentContext.getStore() || {};
       return agentContext.run(
-        { agent: this, connection, request },
+        { agent: this, connection, request, email },
         async () => {
           this.observability?.emit(
             {
               displayMessage: "State updated",
               id: nanoid(),
-              payload: {
-                previousState,
-                state
-              },
+              payload: {},
               timestamp: Date.now(),
               type: "state:update"
             },
@@ -423,28 +484,65 @@ var Agent = class extends Server {
   onStateUpdate(state, source) {
   }
   /**
-   * Called when the Agent receives an email
+   * Called when the Agent receives an email via routeAgentEmail()
    * Override this method to handle incoming emails
    * @param email Email message to process
    */
-  async onEmail(email) {
+  async _onEmail(email) {
     return agentContext.run(
-      { agent: this, connection: void 0, request: void 0 },
+      { agent: this, connection: void 0, request: void 0, email },
       async () => {
-        console.log("Received email from:", email.from, "to:", email.to);
-        console.log("Subject:", email.headers.get("subject"));
-        console.log("Override onEmail() in your agent to process emails");
+        if ("onEmail" in this && typeof this.onEmail === "function") {
+          return this._tryCatch(
+            () => this.onEmail(email)
+          );
+        } else {
+          console.log("Received email from:", email.from, "to:", email.to);
+          console.log("Subject:", email.headers.get("subject"));
+          console.log(
+            "Implement onEmail(email: AgentEmail): Promise<void> in your agent to process emails"
+          );
+        }
       }
     );
   }
-  async sendEmail(emailBinding, from, fromName, options) {
-    const agentName = camelCaseToKebabCase(this._ParentClass.name);
-    const agentId = this.name;
-    return sendEmailWithRouting(emailBinding, from, fromName, {
-      ...options,
-      agentName,
-      agentId,
-      includeRoutingHeaders: true
+  /**
+   * Reply to an email
+   * @param email The email to reply to
+   * @param options Options for the reply
+   * @returns void
+   */
+  async replyToEmail(email, options) {
+    return this._tryCatch(async () => {
+      const agentName = camelCaseToKebabCase(this._ParentClass.name);
+      const agentId = this.name;
+      const { createMimeMessage } = await import("mimetext");
+      const msg = createMimeMessage();
+      msg.setSender({ addr: email.to, name: options.fromName });
+      msg.setRecipient(email.from);
+      msg.setSubject(
+        options.subject || `Re: ${email.headers.get("subject")}` || "No subject"
+      );
+      msg.addMessage({
+        contentType: options.contentType || "text/plain",
+        data: options.body
+      });
+      const domain = email.from.split("@")[1];
+      const messageId = `<${agentId}@${domain}>`;
+      msg.setHeader("In-Reply-To", email.headers.get("Message-ID"));
+      msg.setHeader("Message-ID", messageId);
+      msg.setHeader("X-Agent-Name", agentName);
+      msg.setHeader("X-Agent-ID", agentId);
+      if (options.headers) {
+        for (const [key, value] of Object.entries(options.headers)) {
+          msg.setHeader(key, value);
+        }
+      }
+      await email.reply({
+        from: email.to,
+        raw: msg.asRaw(),
+        to: email.from
+      });
     });
   }
   async _tryCatch(fn) {
@@ -454,6 +552,49 @@ var Agent = class extends Server {
       throw this.onError(e);
     }
   }
+  /**
+   * Automatically wrap custom methods with agent context
+   * This ensures getCurrentAgent() works in all custom methods without decorators
+   */
+  _autoWrapCustomMethods() {
+    const basePrototypes = [_Agent.prototype, Server.prototype];
+    const baseMethods = /* @__PURE__ */ new Set();
+    for (const baseProto of basePrototypes) {
+      let proto2 = baseProto;
+      while (proto2 && proto2 !== Object.prototype) {
+        const methodNames = Object.getOwnPropertyNames(proto2);
+        for (const methodName of methodNames) {
+          baseMethods.add(methodName);
+        }
+        proto2 = Object.getPrototypeOf(proto2);
+      }
+    }
+    let proto = Object.getPrototypeOf(this);
+    let depth = 0;
+    while (proto && proto !== Object.prototype && depth < 10) {
+      const methodNames = Object.getOwnPropertyNames(proto);
+      for (const methodName of methodNames) {
+        const descriptor = Object.getOwnPropertyDescriptor(proto, methodName);
+        if (baseMethods.has(methodName) || methodName.startsWith("_") || !descriptor || !!descriptor.get || typeof descriptor.value !== "function") {
+          continue;
+        }
+        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++;
+    }
+  }
   onError(connectionOrError, error) {
     let theError;
     if (connectionOrError && error) {
@@ -479,6 +620,108 @@ var Agent = class extends Server {
   render() {
     throw new Error("Not implemented");
   }
+  /**
+   * Queue a task to be executed in the future
+   * @param payload Payload to pass to the callback
+   * @param callback Name of the method to call
+   * @returns The ID of the queued task
+   */
+  async queue(callback, payload) {
+    const id = nanoid(9);
+    if (typeof callback !== "string") {
+      throw new Error("Callback must be a string");
+    }
+    if (typeof this[callback] !== "function") {
+      throw new Error(`this.${callback} is not a function`);
+    }
+    this.sql`
+      INSERT OR REPLACE INTO cf_agents_queues (id, payload, callback)
+      VALUES (${id}, ${JSON.stringify(payload)}, ${callback})
+    `;
+    void this._flushQueue().catch((e) => {
+      console.error("Error flushing queue:", e);
+    });
+    return id;
+  }
+  async _flushQueue() {
+    if (this._flushingQueue) {
+      return;
+    }
+    this._flushingQueue = true;
+    while (true) {
+      const result = this.sql`
+      SELECT * FROM cf_agents_queues
+      ORDER BY created_at ASC
+    `;
+      if (!result || result.length === 0) {
+        break;
+      }
+      for (const row of result || []) {
+        const callback = this[row.callback];
+        if (!callback) {
+          console.error(`callback ${row.callback} not found`);
+          continue;
+        }
+        const { connection, request, email } = agentContext.getStore() || {};
+        await agentContext.run(
+          {
+            agent: this,
+            connection,
+            request,
+            email
+          },
+          async () => {
+            await callback.bind(this)(JSON.parse(row.payload), row);
+            await this.dequeue(row.id);
+          }
+        );
+      }
+    }
+    this._flushingQueue = false;
+  }
+  /**
+   * Dequeue a task by ID
+   * @param id ID of the task to dequeue
+   */
+  async dequeue(id) {
+    this.sql`DELETE FROM cf_agents_queues WHERE id = ${id}`;
+  }
+  /**
+   * Dequeue all tasks
+   */
+  async dequeueAll() {
+    this.sql`DELETE FROM cf_agents_queues`;
+  }
+  /**
+   * Dequeue all tasks by callback
+   * @param callback Name of the callback to dequeue
+   */
+  async dequeueAllByCallback(callback) {
+    this.sql`DELETE FROM cf_agents_queues WHERE callback = ${callback}`;
+  }
+  /**
+   * Get a queued task by ID
+   * @param id ID of the task to get
+   * @returns The task or undefined if not found
+   */
+  async getQueue(id) {
+    const result = this.sql`
+      SELECT * FROM cf_agents_queues WHERE id = ${id}
+    `;
+    return result ? { ...result[0], payload: JSON.parse(result[0].payload) } : void 0;
+  }
+  /**
+   * Get all queues by key and value
+   * @param key Key to filter by
+   * @param value Value to filter by
+   * @returns Array of matching QueueItem objects
+   */
+  async getQueues(key, value) {
+    const result = this.sql`
+      SELECT * FROM cf_agents_queues
+    `;
+    return result.filter((row) => JSON.parse(row.payload)[key] === value);
+  }
   /**
    * Schedule a task to be executed in the future
    * @template T Type of the payload data
@@ -493,7 +736,10 @@ var Agent = class extends Server {
       {
         displayMessage: `Schedule ${schedule.id} created`,
         id: nanoid(),
-        payload: schedule,
+        payload: {
+          callback,
+          id
+        },
         timestamp: Date.now(),
         type: "schedule:create"
       },
@@ -628,7 +874,10 @@ var Agent = class extends Server {
         {
           displayMessage: `Schedule ${id} cancelled`,
           id: nanoid(),
-          payload: schedule,
+          payload: {
+            callback: schedule.callback,
+            id: schedule.id
+          },
           timestamp: Date.now(),
           type: "schedule:cancel"
         },
@@ -641,9 +890,9 @@ var Agent = class extends Server {
   }
   async _scheduleNextAlarm() {
     const result = this.sql`
-      SELECT time FROM cf_agents_schedules 
+      SELECT time FROM cf_agents_schedules
       WHERE time > ${Math.floor(Date.now() / 1e3)}
-      ORDER BY time ASC 
+      ORDER BY time ASC
       LIMIT 1
     `;
     if (!result) return;
@@ -659,6 +908,7 @@ var Agent = class extends Server {
     this.sql`DROP TABLE IF EXISTS cf_agents_state`;
     this.sql`DROP TABLE IF EXISTS cf_agents_schedules`;
     this.sql`DROP TABLE IF EXISTS cf_agents_mcp_servers`;
+    this.sql`DROP TABLE IF EXISTS cf_agents_queues`;
     await this.ctx.storage.deleteAlarm();
     await this.ctx.storage.deleteAll();
     this.ctx.abort("destroyed");
@@ -713,7 +963,7 @@ var Agent = class 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;
@@ -766,7 +1016,7 @@ var Agent = class extends Server {
     this.broadcast(
       JSON.stringify({
         mcp: this.getMcpServers(),
-        type: "cf_agent_mcp_servers"
+        type: "cf_agent_mcp_servers" /* CF_AGENT_MCP_SERVERS */
       })
     );
   }
@@ -780,17 +1030,19 @@ var Agent = class extends Server {
     const servers = this.sql`
       SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
     `;
-    for (const server of servers) {
-      const serverConn = this.mcp.mcpConnections[server.id];
-      mcpState.servers[server.id] = {
-        auth_url: server.auth_url,
-        capabilities: serverConn?.serverCapabilities ?? null,
-        instructions: serverConn?.instructions ?? null,
-        name: server.name,
-        server_url: server.server_url,
-        // mark as "authenticating" because the server isn't automatically connected, so it's pending authenticating
-        state: serverConn?.connectionState ?? "authenticating"
-      };
+    if (servers && Array.isArray(servers) && servers.length > 0) {
+      for (const server of servers) {
+        const serverConn = this.mcp.mcpConnections[server.id];
+        mcpState.servers[server.id] = {
+          auth_url: server.auth_url,
+          capabilities: serverConn?.serverCapabilities ?? null,
+          instructions: serverConn?.instructions ?? null,
+          name: server.name,
+          server_url: server.server_url,
+          // mark as "authenticating" because the server isn't automatically connected, so it's pending authenticating
+          state: serverConn?.connectionState ?? "authenticating"
+        };
+      }
     }
     return mcpState;
   }
@@ -798,11 +1050,13 @@ var Agent = class extends Server {
 /**
  * Agent configuration options
  */
-Agent.options = {
+_Agent.options = {
   /** Whether the Agent should hibernate when inactive */
   hibernate: true
   // default to hibernate
 };
+var Agent = _Agent;
+var wrappedClasses = /* @__PURE__ */ new Set();
 async function routeAgentRequest(request, env, options) {
   const corsHeaders = options?.cors === true ? {
     "Access-Control-Allow-Credentials": "true",
@@ -838,7 +1092,7 @@ async function routeAgentRequest(request, env, options) {
   }
   return response;
 }
-function createHeaderBasedResolver() {
+function createHeaderBasedEmailResolver() {
   return async (email, _env) => {
     const messageId = email.headers.get("message-id");
     if (messageId) {
@@ -869,7 +1123,7 @@ function createHeaderBasedResolver() {
     return null;
   };
 }
-function createAddressBasedResolver(defaultAgentName) {
+function createAddressBasedEmailResolver(defaultAgentName) {
   return async (email, _env) => {
     const emailMatch = email.to.match(/^([^+@]+)(?:\+([^@]+))?@(.+)$/);
     if (!emailMatch) {
@@ -888,55 +1142,76 @@ function createAddressBasedResolver(defaultAgentName) {
     };
   };
 }
-function createCatchAllResolver(agentName, agentId) {
+function createCatchAllEmailResolver(agentName, agentId) {
   return async () => ({ agentName, agentId });
 }
+var agentMapCache = /* @__PURE__ */ new WeakMap();
 async function routeAgentEmail(email, env, options) {
   const routingInfo = await options.resolver(email, env);
   if (!routingInfo) {
     console.warn("No routing information found for email, dropping message");
     return;
   }
-  const namespaceBinding = env[routingInfo.agentName];
-  if (!namespaceBinding) {
-    console.error(
-      `Agent namespace '${routingInfo.agentName}' not found in environment`
-    );
-    return;
+  if (!agentMapCache.has(env)) {
+    const map = {};
+    for (const [key, value] of Object.entries(env)) {
+      if (value && typeof value === "object" && "idFromName" in value && typeof value.idFromName === "function") {
+        map[key] = value;
+        map[camelCaseToKebabCase(key)] = value;
+      }
+    }
+    agentMapCache.set(env, map);
   }
-  if (typeof namespaceBinding !== "object" || !("idFromName" in namespaceBinding) || typeof namespaceBinding.idFromName !== "function") {
-    console.error(
-      `Environment binding '${routingInfo.agentName}' is not an AgentNamespace (found: ${typeof namespaceBinding})`
+  const agentMap = agentMapCache.get(env);
+  const namespace = agentMap[routingInfo.agentName];
+  if (!namespace) {
+    const availableAgents = Object.keys(agentMap).filter((key) => !key.includes("-")).join(", ");
+    throw new Error(
+      `Agent namespace '${routingInfo.agentName}' not found in environment. Available agents: ${availableAgents}`
     );
-    return;
   }
-  const namespace = namespaceBinding;
-  const agent = await getAgentByName(namespace, routingInfo.agentId);
-  await agent.onEmail(email);
-}
-async function sendEmailWithRouting(emailBinding, from, fromName, options) {
-  const { createMimeMessage } = await import("mimetext");
-  const msg = createMimeMessage();
-  msg.setSender({ addr: from, name: fromName });
-  msg.setRecipient(options.to);
-  msg.setSubject(options.subject);
-  msg.addMessage({
-    contentType: options.contentType || "text/plain",
-    data: options.body
-  });
-  if (options.includeRoutingHeaders && options.agentName && options.agentId) {
-    const domain = options.domain || from.split("@")[1];
-    const messageId = `<${options.agentId}@${domain}>`;
-    msg.setHeader("Message-ID", messageId);
-    msg.setHeader("X-Agent-Name", options.agentName);
-    msg.setHeader("X-Agent-ID", options.agentId);
-  }
-  if (options.headers) {
-    for (const [key, value] of Object.entries(options.headers)) {
-      msg.setHeader(key, value);
-    }
-  }
-  await emailBinding.send(new EmailMessage(from, options.to, msg.asRaw()));
+  const agent = await getAgentByName(
+    namespace,
+    routingInfo.agentId
+  );
+  const serialisableEmail = {
+    getRaw: async () => {
+      const reader = email.raw.getReader();
+      const chunks = [];
+      let done = false;
+      while (!done) {
+        const { value, done: readerDone } = await reader.read();
+        done = readerDone;
+        if (value) {
+          chunks.push(value);
+        }
+      }
+      const totalLength = chunks.reduce((sum, chunk) => sum + chunk.length, 0);
+      const combined = new Uint8Array(totalLength);
+      let offset = 0;
+      for (const chunk of chunks) {
+        combined.set(chunk, offset);
+        offset += chunk.length;
+      }
+      return combined;
+    },
+    headers: email.headers,
+    rawSize: email.rawSize,
+    setReject: (reason) => {
+      email.setReject(reason);
+    },
+    forward: (rcptTo, headers) => {
+      return email.forward(rcptTo, headers);
+    },
+    reply: (options2) => {
+      return email.reply(
+        new EmailMessage(options2.from, options2.to, options2.raw)
+      );
+    },
+    from: email.from,
+    to: email.to
+  };
+  await agent._onEmail(serialisableEmail);
 }
 async function getAgentByName(namespace, name, options) {
   return getServerByName(namespace, name, options);
@@ -960,7 +1235,7 @@ var StreamingResponse = class {
       id: this._id,
       result: chunk,
       success: true,
-      type: "rpc"
+      type: "rpc" /* RPC */
     };
     this._connection.send(JSON.stringify(response));
   }
@@ -978,7 +1253,7 @@ var StreamingResponse = class {
       id: this._id,
       result: finalChunk,
       success: true,
-      type: "rpc"
+      type: "rpc" /* RPC */
     };
     this._connection.send(JSON.stringify(response));
   }
@@ -1010,16 +1285,16 @@ function isLocalMode() {
 
 export {
   genericObservability,
+  callable,
   unstable_callable,
   getCurrentAgent,
   Agent,
   routeAgentRequest,
-  createHeaderBasedResolver,
-  createAddressBasedResolver,
-  createCatchAllResolver,
+  createHeaderBasedEmailResolver,
+  createAddressBasedEmailResolver,
+  createCatchAllEmailResolver,
   routeAgentEmail,
-  sendEmailWithRouting,
   getAgentByName,
   StreamingResponse
 };
-//# sourceMappingURL=chunk-RX76B6DL.js.map
+//# sourceMappingURL=chunk-HS7VEROK.js.map