agents 0.0.0-94d9a2e → 0.0.0-9688c15

package/dist/{chunk-J6T74FUS.js → chunk-YFPCCSZO.js}

@@ -1,12 +1,18 @@
 import {
   DurableObjectOAuthClientProvider
-} from "./chunk-BZXOAZUX.js";
+} from "./chunk-D6UOOELW.js";
 import {
   camelCaseToKebabCase
-} from "./chunk-QSGN3REV.js";
+} from "./chunk-RN4SNE73.js";
 import {
   MCPClientManager
-} from "./chunk-Y67CHZBI.js";
+} from "./chunk-25YDMV4H.js";
+import {
+  __privateAdd,
+  __privateGet,
+  __privateMethod,
+  __privateSet
+} from "./chunk-HMLY7DHA.js";
 
 // src/index.ts
 import {
@@ -51,12 +57,14 @@ function getCurrentAgent() {
   }
   return store;
 }
+var _state, _ParentClass, _Agent_instances, setStateInternal_fn, tryCatch_fn, scheduleNextAlarm_fn, isCallable_fn, connectToMcpServerInternal_fn, getMcpServerStateInternal_fn;
 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");
+    __privateAdd(this, _Agent_instances);
+    __privateAdd(this, _state, DEFAULT_STATE);
+    __privateAdd(this, _ParentClass, Object.getPrototypeOf(this).constructor);
+    this.mcp = new MCPClientManager(__privateGet(this, _ParentClass).name, "0.0.1");
     /**
      * Initial state for the Agent
      * Override to provide default state values
@@ -103,7 +111,7 @@ var Agent = class extends Server {
         `;
         }
       }
-      await this._scheduleNextAlarm();
+      await __privateMethod(this, _Agent_instances, scheduleNextAlarm_fn).call(this);
     };
     this.sql`
       CREATE TABLE IF NOT EXISTS cf_agents_state (
@@ -112,7 +120,7 @@ var Agent = class extends Server {
       )
     `;
     void this.ctx.blockConcurrencyWhile(async () => {
-      return this._tryCatch(async () => {
+      return __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, async () => {
         this.sql`
         CREATE TABLE IF NOT EXISTS cf_agents_schedules (
           id TEXT PRIMARY KEY NOT NULL DEFAULT (randomblob(9)),
@@ -149,7 +157,7 @@ var Agent = class extends Server {
             this.broadcast(
               JSON.stringify({
                 type: "cf_agent_mcp_servers",
-                mcp: this.getMcpServers()
+                mcp: __privateMethod(this, _Agent_instances, getMcpServerStateInternal_fn).call(this)
               })
             );
             return new Response("<script>window.close();</script>", {
@@ -157,7 +165,7 @@ var Agent = class extends Server {
               headers: { "content-type": "text/html" }
             });
           }
-          return this._tryCatch(() => _onRequest(request));
+          return __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, () => _onRequest(request));
         }
       );
     };
@@ -167,16 +175,16 @@ var Agent = class extends Server {
         { agent: this, connection, request: void 0 },
         async () => {
           if (typeof message !== "string") {
-            return this._tryCatch(() => _onMessage(connection, message));
+            return __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, () => _onMessage(connection, message));
           }
           let parsed;
           try {
             parsed = JSON.parse(message);
           } catch (e) {
-            return this._tryCatch(() => _onMessage(connection, message));
+            return __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, () => _onMessage(connection, message));
           }
           if (isStateUpdateMessage(parsed)) {
-            this._setStateInternal(parsed.state, connection);
+            __privateMethod(this, _Agent_instances, setStateInternal_fn).call(this, parsed.state, connection);
             return;
           }
           if (isRPCRequest(parsed)) {
@@ -186,7 +194,7 @@ var Agent = class extends Server {
               if (typeof methodFn !== "function") {
                 throw new Error(`Method ${method} does not exist`);
               }
-              if (!this._isCallable(method)) {
+              if (!__privateMethod(this, _Agent_instances, isCallable_fn).call(this, method)) {
                 throw new Error(`Method ${method} is not callable`);
               }
               const metadata = callableMetadata.get(methodFn);
@@ -216,7 +224,7 @@ var Agent = class extends Server {
             }
             return;
           }
-          return this._tryCatch(() => _onMessage(connection, message));
+          return __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, () => _onMessage(connection, message));
         }
       );
     };
@@ -237,10 +245,10 @@ var Agent = class extends Server {
             connection.send(
               JSON.stringify({
                 type: "cf_agent_mcp_servers",
-                mcp: this.getMcpServers()
+                mcp: __privateMethod(this, _Agent_instances, getMcpServerStateInternal_fn).call(this)
               })
             );
-            return this._tryCatch(() => _onConnect(connection, ctx2));
+            return __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, () => _onConnect(connection, ctx2));
           }, 20);
         }
       );
@@ -255,25 +263,19 @@ var Agent = class extends Server {
           `;
           await 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
-                }
-              );
+              return __privateMethod(this, _Agent_instances, connectToMcpServerInternal_fn).call(this, 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()
+              mcp: __privateMethod(this, _Agent_instances, getMcpServerStateInternal_fn).call(this)
             })
           );
-          await this._tryCatch(() => _onStart());
+          await __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, () => _onStart());
         }
       );
     };
@@ -282,8 +284,8 @@ var Agent = class extends Server {
    * Current state of the Agent
    */
   get state() {
-    if (this._state !== DEFAULT_STATE) {
-      return this._state;
+    if (__privateGet(this, _state) !== DEFAULT_STATE) {
+      return __privateGet(this, _state);
     }
     const wasChanged = this.sql`
         SELECT state FROM cf_agents_state WHERE id = ${STATE_WAS_CHANGED}
@@ -294,8 +296,8 @@ var Agent = class extends Server {
     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;
+      __privateSet(this, _state, JSON.parse(state));
+      return __privateGet(this, _state);
     }
     if (this.initialState === DEFAULT_STATE) {
       return void 0;
@@ -323,39 +325,12 @@ var Agent = class extends Server {
       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)})
-  `;
-    this.sql`
-    INSERT OR REPLACE INTO cf_agents_state (id, state)
-    VALUES (${STATE_WAS_CHANGED}, ${JSON.stringify(true)})
-  `;
-    this.broadcast(
-      JSON.stringify({
-        type: "cf_agent_state",
-        state
-      }),
-      source !== "server" ? [source.id] : []
-    );
-    return this._tryCatch(() => {
-      const { connection, request } = agentContext.getStore() || {};
-      return agentContext.run(
-        { agent: this, connection, request },
-        async () => {
-          return this.onStateUpdate(state, source);
-        }
-      );
-    });
-  }
   /**
    * Update the Agent's state
    * @param state New state to set
    */
   setState(state) {
-    this._setStateInternal(state, "server");
+    __privateMethod(this, _Agent_instances, setStateInternal_fn).call(this, state, "server");
   }
   /**
    * Called when the Agent's state is updated
@@ -376,13 +351,6 @@ var Agent = class extends Server {
       }
     );
   }
-  async _tryCatch(fn) {
-    try {
-      return await fn();
-    } catch (e) {
-      throw this.onError(e);
-    }
-  }
   onError(connectionOrError, error) {
     let theError;
     if (connectionOrError && error) {
@@ -432,7 +400,7 @@ var Agent = class extends Server {
         payload
       )}, 'scheduled', ${timestamp})
       `;
-      await this._scheduleNextAlarm();
+      await __privateMethod(this, _Agent_instances, scheduleNextAlarm_fn).call(this);
       return {
         id,
         callback,
@@ -450,7 +418,7 @@ var Agent = class extends Server {
         payload
       )}, 'delayed', ${when}, ${timestamp})
       `;
-      await this._scheduleNextAlarm();
+      await __privateMethod(this, _Agent_instances, scheduleNextAlarm_fn).call(this);
       return {
         id,
         callback,
@@ -469,7 +437,7 @@ var Agent = class extends Server {
         payload
       )}, 'cron', ${when}, ${timestamp})
       `;
-      await this._scheduleNextAlarm();
+      await __privateMethod(this, _Agent_instances, scheduleNextAlarm_fn).call(this);
       return {
         id,
         callback,
@@ -536,22 +504,9 @@ var Agent = class extends Server {
    */
   async cancelSchedule(id) {
     this.sql`DELETE FROM cf_agents_schedules WHERE id = ${id}`;
-    await this._scheduleNextAlarm();
+    await __privateMethod(this, _Agent_instances, scheduleNextAlarm_fn).call(this);
     return true;
   }
-  async _scheduleNextAlarm() {
-    const result = this.sql`
-      SELECT time FROM cf_agents_schedules 
-      WHERE time > ${Math.floor(Date.now() / 1e3)}
-      ORDER BY time ASC 
-      LIMIT 1
-    `;
-    if (!result) return;
-    if (result.length > 0 && "time" in result[0]) {
-      const nextTime = result[0].time * 1e3;
-      await this.ctx.storage.setAlarm(nextTime);
-    }
-  }
   /**
    * Destroy the Agent, removing all state and scheduled tasks
    */
@@ -562,13 +517,6 @@ var Agent = class extends Server {
     await this.ctx.storage.deleteAlarm();
     await this.ctx.storage.deleteAll();
   }
-  /**
-   * Get all methods marked as callable on this Agent
-   * @returns A map of method names to their metadata
-   */
-  _isCallable(method) {
-    return callableMetadata.has(this[method]);
-  }
   /**
    * Connect to a new MCP Server
    *
@@ -579,56 +527,121 @@ var Agent = class extends Server {
    * @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
-    );
+    const callbackUrl = `${callbackHost}/${agentsPrefix}/${camelCaseToKebabCase(__privateGet(this, _ParentClass).name)}/${this.name}/callback`;
+    const result = await __privateMethod(this, _Agent_instances, connectToMcpServerInternal_fn).call(this, serverName, url, callbackUrl, options);
     this.broadcast(
       JSON.stringify({
         type: "cf_agent_mcp_servers",
-        mcp: this.getMcpServers()
+        mcp: __privateMethod(this, _Agent_instances, getMcpServerStateInternal_fn).call(this)
       })
     );
     return result;
   }
-  async _connectToMcpServerInternal(serverName, url, callbackUrl, options, reconnect) {
-    const authProvider = new DurableObjectOAuthClientProvider(
-      this.ctx.storage,
-      this.name,
-      callbackUrl
+  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: __privateMethod(this, _Agent_instances, getMcpServerStateInternal_fn).call(this)
+      })
     );
-    if (reconnect) {
-      authProvider.serverId = reconnect.id;
-      if (reconnect.oauthClientId) {
-        authProvider.clientId = reconnect.oauthClientId;
+  }
+};
+_state = new WeakMap();
+_ParentClass = new WeakMap();
+_Agent_instances = new WeakSet();
+setStateInternal_fn = function(state, source = "server") {
+  __privateSet(this, _state, state);
+  this.sql`
+    INSERT OR REPLACE INTO cf_agents_state (id, state)
+    VALUES (${STATE_ROW_ID}, ${JSON.stringify(state)})
+  `;
+  this.sql`
+    INSERT OR REPLACE INTO cf_agents_state (id, state)
+    VALUES (${STATE_WAS_CHANGED}, ${JSON.stringify(true)})
+  `;
+  this.broadcast(
+    JSON.stringify({
+      type: "cf_agent_state",
+      state
+    }),
+    source !== "server" ? [source.id] : []
+  );
+  return __privateMethod(this, _Agent_instances, tryCatch_fn).call(this, () => {
+    const { connection, request } = agentContext.getStore() || {};
+    return agentContext.run(
+      { agent: this, connection, request },
+      async () => {
+        return this.onStateUpdate(state, source);
       }
+    );
+  });
+};
+tryCatch_fn = async function(fn) {
+  try {
+    return await fn();
+  } catch (e) {
+    throw this.onError(e);
+  }
+};
+scheduleNextAlarm_fn = async function() {
+  const result = this.sql`
+      SELECT time FROM cf_agents_schedules 
+      WHERE time > ${Math.floor(Date.now() / 1e3)}
+      ORDER BY time ASC 
+      LIMIT 1
+    `;
+  if (!result) return;
+  if (result.length > 0 && "time" in result[0]) {
+    const nextTime = result[0].time * 1e3;
+    await this.ctx.storage.setAlarm(nextTime);
+  }
+};
+/**
+ * Get all methods marked as callable on this Agent
+ * @returns A map of method names to their metadata
+ */
+isCallable_fn = function(method) {
+  return callableMetadata.has(this[method]);
+};
+connectToMcpServerInternal_fn = async function(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: {
+  }
+  let headerTransportOpts = {};
+  if (options?.transport?.headers) {
+    headerTransportOpts = {
+      eventSourceInit: {
+        fetch: (url2, init) => fetch(url2, {
+          ...init,
           headers: options?.transport?.headers
-        }
-      };
-    }
-    const { id, authUrl, clientId } = await this.mcp.connect(url, {
-      reconnect,
-      transport: {
-        ...headerTransportOpts,
-        authProvider
+        })
       },
-      client: options?.client
-    });
-    this.sql`
+      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},
@@ -640,45 +653,30 @@ var Agent = class extends Server {
         ${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`
+  return {
+    id,
+    authUrl
+  };
+};
+getMcpServerStateInternal_fn = function() {
+  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) {
-      mcpState.servers[server.id] = {
-        name: server.name,
-        server_url: server.server_url,
-        auth_url: server.auth_url,
-        state: this.mcp.mcpConnections[server.id].connectionState,
-        instructions: this.mcp.mcpConnections[server.id].instructions ?? null,
-        capabilities: this.mcp.mcpConnections[server.id].serverCapabilities ?? null
-      };
-    }
-    return mcpState;
+  for (const server of servers) {
+    mcpState.servers[server.id] = {
+      name: server.name,
+      server_url: server.server_url,
+      auth_url: server.auth_url,
+      state: this.mcp.mcpConnections[server.id].connectionState
+    };
   }
+  return mcpState;
 };
 /**
  * Agent configuration options
@@ -728,48 +726,54 @@ async function routeAgentEmail(email, env, options) {
 async function getAgentByName(namespace, name, options) {
   return getServerByName(namespace, name, options);
 }
+var _connection, _id, _closed;
 var StreamingResponse = class {
   constructor(connection, id) {
-    this._closed = false;
-    this._connection = connection;
-    this._id = id;
+    __privateAdd(this, _connection);
+    __privateAdd(this, _id);
+    __privateAdd(this, _closed, false);
+    __privateSet(this, _connection, connection);
+    __privateSet(this, _id, id);
   }
   /**
    * Send a chunk of data to the client
    * @param chunk The data to send
    */
   send(chunk) {
-    if (this._closed) {
+    if (__privateGet(this, _closed)) {
       throw new Error("StreamingResponse is already closed");
     }
     const response = {
       type: "rpc",
-      id: this._id,
+      id: __privateGet(this, _id),
       success: true,
       result: chunk,
       done: false
     };
-    this._connection.send(JSON.stringify(response));
+    __privateGet(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 (__privateGet(this, _closed)) {
       throw new Error("StreamingResponse is already closed");
     }
-    this._closed = true;
+    __privateSet(this, _closed, true);
     const response = {
       type: "rpc",
-      id: this._id,
+      id: __privateGet(this, _id),
       success: true,
       result: finalChunk,
       done: true
     };
-    this._connection.send(JSON.stringify(response));
+    __privateGet(this, _connection).send(JSON.stringify(response));
   }
 };
+_connection = new WeakMap();
+_id = new WeakMap();
+_closed = new WeakMap();
 
 export {
   unstable_callable,
@@ -780,4 +784,4 @@ export {
   getAgentByName,
   StreamingResponse
 };
-//# sourceMappingURL=chunk-J6T74FUS.js.map
+//# sourceMappingURL=chunk-YFPCCSZO.js.map