agents 0.0.0-74a8c74 → 0.0.0-7856b4d

package/dist/{chunk-SZEXGW6W.js → chunk-6RPGDIE2.js}

@@ -1,13 +1,22 @@
+import {
+  MCPClientManager
+} from "./chunk-OYJXQRRH.js";
+import {
+  DurableObjectOAuthClientProvider
+} from "./chunk-BZXOAZUX.js";
+import {
+  camelCaseToKebabCase
+} from "./chunk-VCSB47AK.js";
+
 // src/index.ts
 import {
   Server,
-  routePartykitRequest,
-  getServerByName
+  getServerByName,
+  routePartykitRequest
 } from "partyserver";
 import { parseCronExpression } from "cron-schedule";
 import { nanoid } from "nanoid";
-import { AsyncLocalStorage } from "node:async_hooks";
-import { WorkflowEntrypoint as CFWorkflowEntrypoint } from "cloudflare:workers";
+import { AsyncLocalStorage } from "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);
 }
@@ -23,8 +32,6 @@ function unstable_callable(metadata = {}) {
     return target;
   };
 }
-var WorkflowEntrypoint = class extends CFWorkflowEntrypoint {
-};
 function getNextCronTime(cron) {
   const interval = parseCronExpression(cron);
   return interval.getNextDate();
@@ -32,69 +39,72 @@ function getNextCronTime(cron) {
 var STATE_ROW_ID = "cf_state_row_id";
 var STATE_WAS_CHANGED = "cf_state_was_changed";
 var DEFAULT_STATE = {};
-var unstable_context = new AsyncLocalStorage();
-var Agent = class extends Server {
-  #state = DEFAULT_STATE;
-  /**
-   * Initial state for the Agent
-   * Override to provide default state values
-   */
-  initialState = DEFAULT_STATE;
-  /**
-   * Current state of the Agent
-   */
-  get state() {
-    if (this.#state !== DEFAULT_STATE) {
-      return this.#state;
-    }
-    const wasChanged = this.sql`
-        SELECT state FROM cf_agents_state WHERE id = ${STATE_WAS_CHANGED}
-      `;
-    const result = this.sql`
-      SELECT state FROM cf_agents_state WHERE id = ${STATE_ROW_ID}
-    `;
-    if (wasChanged[0]?.state === "true" || // we do this check for people who updated their code before we shipped wasChanged
-    result[0]?.state) {
-      const state = result[0]?.state;
-      this.#state = JSON.parse(state);
-      return this.#state;
-    }
-    if (this.initialState === DEFAULT_STATE) {
-      return void 0;
-    }
-    this.setState(this.initialState);
-    return this.initialState;
-  }
-  /**
-   * Agent configuration options
-   */
-  static options = {
-    /** Whether the Agent should hibernate when inactive */
-    hibernate: true
-    // default to hibernate
-  };
-  /**
-   * Execute SQL queries against the Agent's database
-   * @template T Type of the returned rows
-   * @param strings SQL query template strings
-   * @param values Values to be inserted into the query
-   * @returns Array of query results
-   */
-  sql(strings, ...values) {
-    let query = "";
-    try {
-      query = strings.reduce(
-        (acc, str, i) => acc + str + (i < values.length ? "?" : ""),
-        ""
-      );
-      return [...this.ctx.storage.sql.exec(query, ...values)];
-    } catch (e) {
-      console.error(`failed to execute sql query: ${query}`, e);
-      throw this.onError(e);
-    }
+var agentContext = new AsyncLocalStorage();
+function getCurrentAgent() {
+  const store = agentContext.getStore();
+  if (!store) {
+    return {
+      agent: void 0,
+      connection: void 0,
+      request: void 0
+    };
   }
+  return store;
+}
+var Agent = class extends Server {
   constructor(ctx, env) {
     super(ctx, env);
+    this._state = DEFAULT_STATE;
+    this._ParentClass = Object.getPrototypeOf(this).constructor;
+    this.mcp = new MCPClientManager(this._ParentClass.name, "0.0.1");
+    /**
+     * Initial state for the Agent
+     * Override to provide default state values
+     */
+    this.initialState = DEFAULT_STATE;
+    /**
+     * Method called when an alarm fires.
+     * Executes any scheduled tasks that are due.
+     *
+     * @remarks
+     * To schedule a task, please use the `this.schedule` method instead.
+     * See {@link https://developers.cloudflare.com/agents/api-reference/schedule-tasks/}
+     */
+    this.alarm = async () => {
+      const now = Math.floor(Date.now() / 1e3);
+      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 {
+              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`
+          DELETE FROM cf_agents_schedules WHERE id = ${row.id}
+        `;
+        }
+      }
+      await this._scheduleNextAlarm();
+    };
     this.sql`
       CREATE TABLE IF NOT EXISTS cf_agents_state (
         id TEXT PRIMARY KEY NOT NULL,
@@ -102,7 +112,7 @@ var Agent = class extends Server {
       )
     `;
     void this.ctx.blockConcurrencyWhile(async () => {
-      return this.#tryCatch(async () => {
+      return this._tryCatch(async () => {
         this.sql`
         CREATE TABLE IF NOT EXISTS cf_agents_schedules (
           id TEXT PRIMARY KEY NOT NULL DEFAULT (randomblob(9)),
@@ -118,22 +128,55 @@ var Agent = class extends Server {
         await this.alarm();
       });
     });
+    this.sql`
+      CREATE TABLE IF NOT EXISTS cf_agents_mcp_servers (
+        id TEXT PRIMARY KEY NOT NULL,
+        name TEXT NOT NULL,
+        server_url TEXT NOT NULL,
+        callback_url TEXT NOT NULL,
+        client_id TEXT,
+        auth_url TEXT,
+        server_options TEXT
+      )
+    `;
+    const _onRequest = this.onRequest.bind(this);
+    this.onRequest = (request) => {
+      return agentContext.run(
+        { agent: this, connection: void 0, request },
+        async () => {
+          if (this.mcp.isCallbackRequest(request)) {
+            await this.mcp.handleCallbackRequest(request);
+            this.broadcast(
+              JSON.stringify({
+                type: "cf_agent_mcp_servers",
+                mcp: this.getMcpServers()
+              })
+            );
+            return new Response("<script>window.close();</script>", {
+              status: 200,
+              headers: { "content-type": "text/html" }
+            });
+          }
+          return this._tryCatch(() => _onRequest(request));
+        }
+      );
+    };
     const _onMessage = this.onMessage.bind(this);
     this.onMessage = async (connection, message) => {
-      return unstable_context.run(
+      return agentContext.run(
         { agent: this, connection, request: void 0 },
         async () => {
           if (typeof message !== "string") {
-            return this.#tryCatch(() => _onMessage(connection, message));
+            return this._tryCatch(() => _onMessage(connection, message));
           }
           let parsed;
           try {
             parsed = JSON.parse(message);
           } catch (e) {
-            return this.#tryCatch(() => _onMessage(connection, message));
+            return this._tryCatch(() => _onMessage(connection, message));
           }
           if (isStateUpdateMessage(parsed)) {
-            this.#setStateInternal(parsed.state, connection);
+            this._setStateInternal(parsed.state, connection);
             return;
           }
           if (isRPCRequest(parsed)) {
@@ -143,7 +186,7 @@ var Agent = class extends Server {
               if (typeof methodFn !== "function") {
                 throw new Error(`Method ${method} does not exist`);
               }
-              if (!this.#isCallable(method)) {
+              if (!this._isCallable(method)) {
                 throw new Error(`Method ${method} is not callable`);
               }
               const metadata = callableMetadata.get(methodFn);
@@ -173,13 +216,13 @@ var Agent = class extends Server {
             }
             return;
           }
-          return this.#tryCatch(() => _onMessage(connection, message));
+          return this._tryCatch(() => _onMessage(connection, message));
         }
       );
     };
     const _onConnect = this.onConnect.bind(this);
     this.onConnect = (connection, ctx2) => {
-      return unstable_context.run(
+      return agentContext.run(
         { agent: this, connection, request: ctx2.request },
         async () => {
           setTimeout(() => {
@@ -191,14 +234,97 @@ var Agent = class extends Server {
                 })
               );
             }
-            return this.#tryCatch(() => _onConnect(connection, ctx2));
+            connection.send(
+              JSON.stringify({
+                type: "cf_agent_mcp_servers",
+                mcp: this.getMcpServers()
+              })
+            );
+            return this._tryCatch(() => _onConnect(connection, ctx2));
           }, 20);
         }
       );
     };
+    const _onStart = this.onStart.bind(this);
+    this.onStart = async () => {
+      return agentContext.run(
+        { agent: this, connection: void 0, request: 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(
+            servers.filter((server) => server.auth_url === null).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
+                }
+              );
+            })
+          );
+          this.broadcast(
+            JSON.stringify({
+              type: "cf_agent_mcp_servers",
+              mcp: this.getMcpServers()
+            })
+          );
+          await this._tryCatch(() => _onStart());
+        }
+      );
+    };
+  }
+  /**
+   * Current state of the Agent
+   */
+  get state() {
+    if (this._state !== DEFAULT_STATE) {
+      return this._state;
+    }
+    const wasChanged = this.sql`
+        SELECT state FROM cf_agents_state WHERE id = ${STATE_WAS_CHANGED}
+      `;
+    const result = this.sql`
+      SELECT state FROM cf_agents_state WHERE id = ${STATE_ROW_ID}
+    `;
+    if (wasChanged[0]?.state === "true" || // we do this check for people who updated their code before we shipped wasChanged
+    result[0]?.state) {
+      const state = result[0]?.state;
+      this._state = JSON.parse(state);
+      return this._state;
+    }
+    if (this.initialState === DEFAULT_STATE) {
+      return void 0;
+    }
+    this.setState(this.initialState);
+    return this.initialState;
   }
-  #setStateInternal(state, source = "server") {
-    this.#state = state;
+  /**
+   * Execute SQL queries against the Agent's database
+   * @template T Type of the returned rows
+   * @param strings SQL query template strings
+   * @param values Values to be inserted into the query
+   * @returns Array of query results
+   */
+  sql(strings, ...values) {
+    let query = "";
+    try {
+      query = strings.reduce(
+        (acc, str, i) => acc + str + (i < values.length ? "?" : ""),
+        ""
+      );
+      return [...this.ctx.storage.sql.exec(query, ...values)];
+    } catch (e) {
+      console.error(`failed to execute sql query: ${query}`, e);
+      throw this.onError(e);
+    }
+  }
+  _setStateInternal(state, source = "server") {
+    this._state = state;
     this.sql`
     INSERT OR REPLACE INTO cf_agents_state (id, state)
     VALUES (${STATE_ROW_ID}, ${JSON.stringify(state)})
@@ -214,9 +340,9 @@ var Agent = class extends Server {
       }),
       source !== "server" ? [source.id] : []
     );
-    return this.#tryCatch(() => {
-      const { connection, request } = unstable_context.getStore() || {};
-      return unstable_context.run(
+    return this._tryCatch(() => {
+      const { connection, request } = agentContext.getStore() || {};
+      return agentContext.run(
         { agent: this, connection, request },
         async () => {
           return this.onStateUpdate(state, source);
@@ -229,7 +355,7 @@ var Agent = class extends Server {
    * @param state New state to set
    */
   setState(state) {
-    this.#setStateInternal(state, "server");
+    this._setStateInternal(state, "server");
   }
   /**
    * Called when the Agent's state is updated
@@ -243,14 +369,14 @@ var Agent = class extends Server {
    * @param email Email message to process
    */
   onEmail(email) {
-    return unstable_context.run(
+    return agentContext.run(
       { agent: this, connection: void 0, request: void 0 },
       async () => {
         console.error("onEmail not implemented");
       }
     );
   }
-  async #tryCatch(fn) {
+  async _tryCatch(fn) {
     try {
       return await fn();
     } catch (e) {
@@ -306,7 +432,7 @@ var Agent = class extends Server {
         payload
       )}, 'scheduled', ${timestamp})
       `;
-      await this.#scheduleNextAlarm();
+      await this._scheduleNextAlarm();
       return {
         id,
         callback,
@@ -324,7 +450,7 @@ var Agent = class extends Server {
         payload
       )}, 'delayed', ${when}, ${timestamp})
       `;
-      await this.#scheduleNextAlarm();
+      await this._scheduleNextAlarm();
       return {
         id,
         callback,
@@ -343,7 +469,7 @@ var Agent = class extends Server {
         payload
       )}, 'cron', ${when}, ${timestamp})
       `;
-      await this.#scheduleNextAlarm();
+      await this._scheduleNextAlarm();
       return {
         id,
         callback,
@@ -410,10 +536,10 @@ var Agent = class extends Server {
    */
   async cancelSchedule(id) {
     this.sql`DELETE FROM cf_agents_schedules WHERE id = ${id}`;
-    await this.#scheduleNextAlarm();
+    await this._scheduleNextAlarm();
     return true;
   }
-  async #scheduleNextAlarm() {
+  async _scheduleNextAlarm() {
     const result = this.sql`
       SELECT time FROM cf_agents_schedules 
       WHERE time > ${Math.floor(Date.now() / 1e3)}
@@ -426,61 +552,144 @@ var Agent = class extends Server {
       await this.ctx.storage.setAlarm(nextTime);
     }
   }
-  /**
-   * Method called when an alarm fires
-   * Executes any scheduled tasks that are due
-   */
-  async alarm() {
-    const now = Math.floor(Date.now() / 1e3);
-    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 unstable_context.run(
-        { agent: this, connection: void 0, request: void 0 },
-        async () => {
-          try {
-            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`
-          DELETE FROM cf_agents_schedules WHERE id = ${row.id}
-        `;
-      }
-    }
-    await this.#scheduleNextAlarm();
-  }
   /**
    * Destroy the Agent, removing all state and scheduled tasks
    */
   async destroy() {
     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`;
     await this.ctx.storage.deleteAlarm();
     await this.ctx.storage.deleteAll();
+    this.ctx.abort("destroyed");
   }
   /**
    * Get all methods marked as callable on this Agent
    * @returns A map of method names to their metadata
    */
-  #isCallable(method) {
+  _isCallable(method) {
     return callableMetadata.has(this[method]);
   }
+  /**
+   * Connect to a new MCP Server
+   *
+   * @param url MCP Server SSE URL
+   * @param callbackHost Base host for the agent, used for the redirect URI.
+   * @param agentsPrefix agents routing prefix if not using `agents`
+   * @param options MCP client and transport (header) options
+   * @returns authUrl
+   */
+  async addMcpServer(serverName, url, callbackHost, agentsPrefix = "agents", options) {
+    const callbackUrl = `${callbackHost}/${agentsPrefix}/${camelCaseToKebabCase(this._ParentClass.name)}/${this.name}/callback`;
+    const result = await this._connectToMcpServerInternal(
+      serverName,
+      url,
+      callbackUrl,
+      options
+    );
+    this.broadcast(
+      JSON.stringify({
+        type: "cf_agent_mcp_servers",
+        mcp: this.getMcpServers()
+      })
+    );
+    return result;
+  }
+  async _connectToMcpServerInternal(serverName, url, callbackUrl, options, reconnect) {
+    const authProvider = new DurableObjectOAuthClientProvider(
+      this.ctx.storage,
+      this.name,
+      callbackUrl
+    );
+    if (reconnect) {
+      authProvider.serverId = reconnect.id;
+      if (reconnect.oauthClientId) {
+        authProvider.clientId = reconnect.oauthClientId;
+      }
+    }
+    let headerTransportOpts = {};
+    if (options?.transport?.headers) {
+      headerTransportOpts = {
+        eventSourceInit: {
+          fetch: (url2, init) => fetch(url2, {
+            ...init,
+            headers: options?.transport?.headers
+          })
+        },
+        requestInit: {
+          headers: options?.transport?.headers
+        }
+      };
+    }
+    const { id, authUrl, clientId } = await this.mcp.connect(url, {
+      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
+    };
+  }
+  async removeMcpServer(id) {
+    this.mcp.closeConnection(id);
+    this.sql`
+      DELETE FROM cf_agents_mcp_servers WHERE id = ${id};
+    `;
+    this.broadcast(
+      JSON.stringify({
+        type: "cf_agent_mcp_servers",
+        mcp: this.getMcpServers()
+      })
+    );
+  }
+  getMcpServers() {
+    const mcpState = {
+      servers: {},
+      tools: this.mcp.listTools(),
+      prompts: this.mcp.listPrompts(),
+      resources: this.mcp.listResources()
+    };
+    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] = {
+        name: server.name,
+        server_url: server.server_url,
+        auth_url: server.auth_url,
+        // mark as "authenticating" because the server isn't automatically connected, so it's pending authenticating
+        state: serverConn?.connectionState ?? "authenticating",
+        instructions: serverConn?.instructions ?? null,
+        capabilities: serverConn?.serverCapabilities ?? null
+      };
+    }
+    return mcpState;
+  }
+};
+/**
+ * Agent configuration options
+ */
+Agent.options = {
+  /** Whether the Agent should hibernate when inactive */
+  hibernate: true
+  // default to hibernate
 };
 async function routeAgentRequest(request, env, options) {
   const corsHeaders = options?.cors === true ? {
@@ -519,62 +728,59 @@ async function routeAgentRequest(request, env, options) {
 }
 async function routeAgentEmail(email, env, options) {
 }
-function getAgentByName(namespace, name, options) {
+async function getAgentByName(namespace, name, options) {
   return getServerByName(namespace, name, options);
 }
 var StreamingResponse = class {
-  #connection;
-  #id;
-  #closed = false;
   constructor(connection, id) {
-    this.#connection = connection;
-    this.#id = id;
+    this._closed = false;
+    this._connection = connection;
+    this._id = id;
   }
   /**
    * Send a chunk of data to the client
    * @param chunk The data to send
    */
   send(chunk) {
-    if (this.#closed) {
+    if (this._closed) {
       throw new Error("StreamingResponse is already closed");
     }
     const response = {
       type: "rpc",
-      id: this.#id,
+      id: this._id,
       success: true,
       result: chunk,
       done: false
     };
-    this.#connection.send(JSON.stringify(response));
+    this._connection.send(JSON.stringify(response));
   }
   /**
    * End the stream and send the final chunk (if any)
    * @param finalChunk Optional final chunk of data to send
    */
   end(finalChunk) {
-    if (this.#closed) {
+    if (this._closed) {
       throw new Error("StreamingResponse is already closed");
     }
-    this.#closed = true;
+    this._closed = true;
     const response = {
       type: "rpc",
-      id: this.#id,
+      id: this._id,
       success: true,
       result: finalChunk,
       done: true
     };
-    this.#connection.send(JSON.stringify(response));
+    this._connection.send(JSON.stringify(response));
   }
 };
 
 export {
   unstable_callable,
-  WorkflowEntrypoint,
-  unstable_context,
+  getCurrentAgent,
   Agent,
   routeAgentRequest,
   routeAgentEmail,
   getAgentByName,
   StreamingResponse
 };
-//# sourceMappingURL=chunk-SZEXGW6W.js.map
+//# sourceMappingURL=chunk-6RPGDIE2.js.map