agents 0.0.0-0ac89c6 → 0.0.0-0bb74b8

package/dist/{chunk-4ARKO5R4.js → chunk-MGHXAF5T.js}

@@ -1,22 +1,23 @@
+import {
+  MCPClientManager
+} from "./chunk-MW5BQ2FW.js";
 import {
   DurableObjectOAuthClientProvider
-} from "./chunk-BZXOAZUX.js";
+} from "./chunk-PVQZBKN7.js";
 import {
   camelCaseToKebabCase
-} from "./chunk-QSGN3REV.js";
-import {
-  MCPClientManager
-} from "./chunk-Y67CHZBI.js";
+} from "./chunk-KUH345EY.js";
 
 // src/index.ts
+import { AsyncLocalStorage } from "async_hooks";
+import { parseCronExpression } from "cron-schedule";
+import { nanoid } from "nanoid";
+import { EmailMessage } from "cloudflare:email";
 import {
   Server,
-  routePartykitRequest,
-  getServerByName
+  getServerByName,
+  routePartykitRequest
 } from "partyserver";
-import { parseCronExpression } from "cron-schedule";
-import { nanoid } from "nanoid";
-import { AsyncLocalStorage } from "node:async_hooks";
 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);
 }
@@ -62,6 +63,10 @@ var Agent = class extends Server {
      * Override to provide default state values
      */
     this.initialState = DEFAULT_STATE;
+    /**
+     * The observability implementation to use for the Agent
+     */
+    this.observability = genericObservability;
     /**
      * Method called when an alarm fires.
      * Executes any scheduled tasks that are due.
@@ -82,9 +87,24 @@ var Agent = class extends Server {
           continue;
         }
         await agentContext.run(
-          { agent: this, connection: void 0, request: void 0 },
+          {
+            agent: this,
+            connection: void 0,
+            request: void 0,
+            email: 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);
@@ -142,19 +162,19 @@ 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({
-                type: "cf_agent_mcp_servers",
-                mcp: this._getMcpServerStateInternal()
+                mcp: this.getMcpServers(),
+                type: "cf_agent_mcp_servers"
               })
             );
             return new Response("<script>window.close();</script>", {
-              status: 200,
-              headers: { "content-type": "text/html" }
+              headers: { "content-type": "text/html" },
+              status: 200
             });
           }
           return this._tryCatch(() => _onRequest(request));
@@ -164,7 +184,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));
@@ -172,7 +192,7 @@ var Agent = class extends Server {
           let parsed;
           try {
             parsed = JSON.parse(message);
-          } catch (e) {
+          } catch (_e) {
             return this._tryCatch(() => _onMessage(connection, message));
           }
           if (isStateUpdateMessage(parsed)) {
@@ -196,20 +216,35 @@ var Agent = class extends Server {
                 return;
               }
               const result = await methodFn.apply(this, args);
+              this.observability?.emit(
+                {
+                  displayMessage: `RPC call to ${method}`,
+                  id: nanoid(),
+                  payload: {
+                    args,
+                    method,
+                    streaming: metadata?.streaming,
+                    success: true
+                  },
+                  timestamp: Date.now(),
+                  type: "rpc"
+                },
+                this.ctx
+              );
               const response = {
-                type: "rpc",
+                done: true,
                 id,
-                success: true,
                 result,
-                done: true
+                success: true,
+                type: "rpc"
               };
               connection.send(JSON.stringify(response));
             } catch (e) {
               const response = {
-                type: "rpc",
+                error: e instanceof Error ? e.message : "Unknown error occurred",
                 id: parsed.id,
                 success: false,
-                error: e instanceof Error ? e.message : "Unknown error occurred"
+                type: "rpc"
               };
               connection.send(JSON.stringify(response));
               console.error("RPC error:", e);
@@ -223,23 +258,35 @@ var Agent = class extends Server {
     const _onConnect = this.onConnect.bind(this);
     this.onConnect = (connection, ctx2) => {
       return agentContext.run(
-        { agent: this, connection, request: ctx2.request },
+        { agent: this, connection, request: ctx2.request, email: void 0 },
         async () => {
           setTimeout(() => {
             if (this.state) {
               connection.send(
                 JSON.stringify({
-                  type: "cf_agent_state",
-                  state: this.state
+                  state: this.state,
+                  type: "cf_agent_state"
                 })
               );
             }
             connection.send(
               JSON.stringify({
-                type: "cf_agent_mcp_servers",
-                mcp: this._getMcpServerStateInternal()
+                mcp: this.getMcpServers(),
+                type: "cf_agent_mcp_servers"
               })
             );
+            this.observability?.emit(
+              {
+                displayMessage: "Connection established",
+                id: nanoid(),
+                payload: {
+                  connectionId: connection.id
+                },
+                timestamp: Date.now(),
+                type: "connect"
+              },
+              this.ctx
+            );
             return this._tryCatch(() => _onConnect(connection, ctx2));
           }, 20);
         }
@@ -248,12 +295,17 @@ var Agent = class extends Server {
     const _onStart = this.onStart.bind(this);
     this.onStart = async () => {
       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`
             SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
           `;
-          await Promise.allSettled(
+          Promise.allSettled(
             servers.map((server) => {
               return this._connectToMcpServerInternal(
                 server.name,
@@ -266,13 +318,14 @@ var Agent = class extends Server {
                 }
               );
             })
-          );
-          this.broadcast(
-            JSON.stringify({
-              type: "cf_agent_mcp_servers",
-              mcp: this._getMcpServerStateInternal()
-            })
-          );
+          ).then((_results) => {
+            this.broadcast(
+              JSON.stringify({
+                mcp: this.getMcpServers(),
+                type: "cf_agent_mcp_servers"
+              })
+            );
+          });
           await this._tryCatch(() => _onStart());
         }
       );
@@ -324,6 +377,7 @@ 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)
@@ -335,16 +389,29 @@ var Agent = class extends Server {
   `;
     this.broadcast(
       JSON.stringify({
-        type: "cf_agent_state",
-        state
+        state,
+        type: "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
+              },
+              timestamp: Date.now(),
+              type: "state:update"
+            },
+            this.ctx
+          );
           return this.onStateUpdate(state, source);
         }
       );
@@ -362,20 +429,71 @@ var Agent = class extends Server {
    * @param state Updated state
    * @param source Source of the state update ("server" or a client connection)
    */
+  // biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later
   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
    */
-  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.error("onEmail not implemented");
+        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"
+          );
+        }
       }
     );
   }
+  /**
+   * 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) {
     try {
       return await fn();
@@ -418,6 +536,16 @@ var Agent = class extends Server {
    */
   async schedule(when, callback, payload) {
     const id = nanoid(9);
+    const emitScheduleCreate = (schedule) => this.observability?.emit(
+      {
+        displayMessage: `Schedule ${schedule.id} created`,
+        id: nanoid(),
+        payload: schedule,
+        timestamp: Date.now(),
+        type: "schedule:create"
+      },
+      this.ctx
+    );
     if (typeof callback !== "string") {
       throw new Error("Callback must be a string");
     }
@@ -433,13 +561,15 @@ var Agent = class extends Server {
       )}, 'scheduled', ${timestamp})
       `;
       await this._scheduleNextAlarm();
-      return {
-        id,
+      const schedule = {
         callback,
+        id,
         payload,
         time: timestamp,
         type: "scheduled"
       };
+      emitScheduleCreate(schedule);
+      return schedule;
     }
     if (typeof when === "number") {
       const time = new Date(Date.now() + when * 1e3);
@@ -451,14 +581,16 @@ var Agent = class extends Server {
       )}, 'delayed', ${when}, ${timestamp})
       `;
       await this._scheduleNextAlarm();
-      return {
-        id,
+      const schedule = {
         callback,
-        payload,
         delayInSeconds: when,
+        id,
+        payload,
         time: timestamp,
         type: "delayed"
       };
+      emitScheduleCreate(schedule);
+      return schedule;
     }
     if (typeof when === "string") {
       const nextExecutionTime = getNextCronTime(when);
@@ -470,14 +602,16 @@ var Agent = class extends Server {
       )}, 'cron', ${when}, ${timestamp})
       `;
       await this._scheduleNextAlarm();
-      return {
-        id,
+      const schedule = {
         callback,
-        payload,
         cron: when,
+        id,
+        payload,
         time: timestamp,
         type: "cron"
       };
+      emitScheduleCreate(schedule);
+      return schedule;
     }
     throw new Error("Invalid schedule type");
   }
@@ -535,6 +669,19 @@ var Agent = class extends Server {
    * @returns true if the task was cancelled, false otherwise
    */
   async cancelSchedule(id) {
+    const schedule = await this.getSchedule(id);
+    if (schedule) {
+      this.observability?.emit(
+        {
+          displayMessage: `Schedule ${id} cancelled`,
+          id: nanoid(),
+          payload: schedule,
+          timestamp: Date.now(),
+          type: "schedule:cancel"
+        },
+        this.ctx
+      );
+    }
     this.sql`DELETE FROM cf_agents_schedules WHERE id = ${id}`;
     await this._scheduleNextAlarm();
     return true;
@@ -561,6 +708,17 @@ var Agent = class extends Server {
     this.sql`DROP TABLE IF EXISTS cf_agents_mcp_servers`;
     await this.ctx.storage.deleteAlarm();
     await this.ctx.storage.deleteAll();
+    this.ctx.abort("destroyed");
+    this.observability?.emit(
+      {
+        displayMessage: "Agent destroyed",
+        id: nanoid(),
+        payload: {},
+        timestamp: Date.now(),
+        type: "destroy"
+      },
+      this.ctx
+    );
   }
   /**
    * Get all methods marked as callable on this Agent
@@ -586,15 +744,28 @@ var Agent = class extends Server {
       callbackUrl,
       options
     );
+    this.sql`
+        INSERT
+        OR REPLACE INTO cf_agents_mcp_servers (id, name, server_url, client_id, auth_url, callback_url, server_options)
+      VALUES (
+        ${result.id},
+        ${serverName},
+        ${url},
+        ${result.clientId ?? null},
+        ${result.authUrl ?? null},
+        ${callbackUrl},
+        ${options ? JSON.stringify(options) : null}
+        );
+    `;
     this.broadcast(
       JSON.stringify({
-        type: "cf_agent_mcp_servers",
-        mcp: this._getMcpServerStateInternal()
+        mcp: this.getMcpServers(),
+        type: "cf_agent_mcp_servers"
       })
     );
     return result;
   }
-  async _connectToMcpServerInternal(serverName, url, callbackUrl, options, reconnect) {
+  async _connectToMcpServerInternal(_serverName, url, callbackUrl, options, reconnect) {
     const authProvider = new DurableObjectOAuthClientProvider(
       this.ctx.storage,
       this.name,
@@ -621,28 +792,17 @@ var Agent = class extends Server {
       };
     }
     const { id, authUrl, clientId } = await this.mcp.connect(url, {
+      client: options?.client,
       reconnect,
       transport: {
         ...headerTransportOpts,
         authProvider
-      },
-      client: options?.client
+      }
     });
-    this.sql`
-      INSERT OR REPLACE INTO cf_agents_mcp_servers (id, name, server_url, client_id, auth_url, callback_url, server_options)
-      VALUES (
-        ${id},
-        ${serverName},
-        ${url},
-        ${clientId ?? null},
-        ${authUrl ?? null},
-        ${callbackUrl},
-        ${options ? JSON.stringify(options) : null}
-      );
-    `;
     return {
-      id,
-      authUrl
+      authUrl,
+      clientId,
+      id
     };
   }
   async removeMcpServer(id) {
@@ -652,27 +812,31 @@ var Agent = class extends Server {
     `;
     this.broadcast(
       JSON.stringify({
-        type: "cf_agent_mcp_servers",
-        mcp: this._getMcpServerStateInternal()
+        mcp: this.getMcpServers(),
+        type: "cf_agent_mcp_servers"
       })
     );
   }
-  _getMcpServerStateInternal() {
+  getMcpServers() {
     const mcpState = {
-      servers: {},
-      tools: this.mcp.listTools(),
       prompts: this.mcp.listPrompts(),
-      resources: this.mcp.listResources()
+      resources: this.mcp.listResources(),
+      servers: {},
+      tools: this.mcp.listTools()
     };
     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,
-        auth_url: server.auth_url,
-        state: this.mcp.mcpConnections[server.id].connectionState
+        // mark as "authenticating" because the server isn't automatically connected, so it's pending authenticating
+        state: serverConn?.connectionState ?? "authenticating"
       };
     }
     return mcpState;
@@ -688,9 +852,9 @@ Agent.options = {
 };
 async function routeAgentRequest(request, env, options) {
   const corsHeaders = options?.cors === true ? {
-    "Access-Control-Allow-Origin": "*",
-    "Access-Control-Allow-Methods": "GET, POST, HEAD, OPTIONS",
     "Access-Control-Allow-Credentials": "true",
+    "Access-Control-Allow-Methods": "GET, POST, HEAD, OPTIONS",
+    "Access-Control-Allow-Origin": "*",
     "Access-Control-Max-Age": "86400"
   } : options?.cors;
   if (request.method === "OPTIONS") {
@@ -721,7 +885,116 @@ async function routeAgentRequest(request, env, options) {
   }
   return response;
 }
+function createHeaderBasedEmailResolver() {
+  return async (email, _env) => {
+    const messageId = email.headers.get("message-id");
+    if (messageId) {
+      const messageIdMatch = messageId.match(/<([^@]+)@([^>]+)>/);
+      if (messageIdMatch) {
+        const [, agentId2, domain] = messageIdMatch;
+        const agentName2 = domain.split(".")[0];
+        return { agentName: agentName2, agentId: agentId2 };
+      }
+    }
+    const references = email.headers.get("references");
+    if (references) {
+      const referencesMatch = references.match(
+        /<([A-Za-z0-9+/]{43}=)@([^>]+)>/
+      );
+      if (referencesMatch) {
+        const [, base64Id, domain] = referencesMatch;
+        const agentId2 = Buffer.from(base64Id, "base64").toString("hex");
+        const agentName2 = domain.split(".")[0];
+        return { agentName: agentName2, agentId: agentId2 };
+      }
+    }
+    const agentName = email.headers.get("x-agent-name");
+    const agentId = email.headers.get("x-agent-id");
+    if (agentName && agentId) {
+      return { agentName, agentId };
+    }
+    return null;
+  };
+}
+function createAddressBasedEmailResolver(defaultAgentName) {
+  return async (email, _env) => {
+    const emailMatch = email.to.match(/^([^+@]+)(?:\+([^@]+))?@(.+)$/);
+    if (!emailMatch) {
+      return null;
+    }
+    const [, localPart, subAddress] = emailMatch;
+    if (subAddress) {
+      return {
+        agentName: localPart,
+        agentId: subAddress
+      };
+    }
+    return {
+      agentName: defaultAgentName,
+      agentId: localPart
+    };
+  };
+}
+function createCatchAllEmailResolver(agentName, agentId) {
+  return async () => ({ agentName, agentId });
+}
 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) {
+    throw new Error(
+      `Agent namespace '${routingInfo.agentName}' not found in environment`
+    );
+  }
+  if (typeof namespaceBinding !== "object" || !("idFromName" in namespaceBinding) || typeof namespaceBinding.idFromName !== "function") {
+    throw new Error(
+      `Environment binding '${routingInfo.agentName}' is not an AgentNamespace (found: ${typeof namespaceBinding})`
+    );
+  }
+  const namespace = namespaceBinding;
+  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);
@@ -741,11 +1014,11 @@ var StreamingResponse = class {
       throw new Error("StreamingResponse is already closed");
     }
     const response = {
-      type: "rpc",
+      done: false,
       id: this._id,
-      success: true,
       result: chunk,
-      done: false
+      success: true,
+      type: "rpc"
     };
     this._connection.send(JSON.stringify(response));
   }
@@ -759,23 +1032,51 @@ var StreamingResponse = class {
     }
     this._closed = true;
     const response = {
-      type: "rpc",
+      done: true,
       id: this._id,
-      success: true,
       result: finalChunk,
-      done: true
+      success: true,
+      type: "rpc"
     };
     this._connection.send(JSON.stringify(response));
   }
 };
 
+// src/observability/index.ts
+var genericObservability = {
+  emit(event) {
+    if (isLocalMode()) {
+      console.log(event.displayMessage);
+      return;
+    }
+    console.log(event);
+  }
+};
+var localMode = false;
+function isLocalMode() {
+  if (localMode) {
+    return true;
+  }
+  const { request } = getCurrentAgent();
+  if (!request) {
+    return false;
+  }
+  const url = new URL(request.url);
+  localMode = url.hostname === "localhost";
+  return localMode;
+}
+
 export {
+  genericObservability,
   unstable_callable,
   getCurrentAgent,
   Agent,
   routeAgentRequest,
+  createHeaderBasedEmailResolver,
+  createAddressBasedEmailResolver,
+  createCatchAllEmailResolver,
   routeAgentEmail,
   getAgentByName,
   StreamingResponse
 };
-//# sourceMappingURL=chunk-4ARKO5R4.js.map
+//# sourceMappingURL=chunk-MGHXAF5T.js.map