agents 0.0.101 → 0.0.102

package/dist/index.d.ts

@@ -1,13 +1,27 @@
-import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import {
-  ServerCapabilities,
-  Tool,
-  Prompt,
-  Resource,
-} from "@modelcontextprotocol/sdk/types.js";
-import { Server, Connection, PartyServerOptions } from "partyserver";
+import "@modelcontextprotocol/sdk/client/index.js";
+import "@modelcontextprotocol/sdk/types.js";
 export { Connection, ConnectionContext, WSMessage } from "partyserver";
-import { MCPClientManager } from "./mcp/client.js";
+import "./mcp/client.js";
+export {
+  A as Agent,
+  a as AgentContext,
+  i as AgentNamespace,
+  j as AgentOptions,
+  C as CallableMetadata,
+  f as MCPServer,
+  e as MCPServerMessage,
+  M as MCPServersState,
+  R as RPCRequest,
+  c as RPCResponse,
+  d as Schedule,
+  S as StateUpdateMessage,
+  m as StreamingResponse,
+  l as getAgentByName,
+  h as getCurrentAgent,
+  k as routeAgentEmail,
+  r as routeAgentRequest,
+  u as unstable_callable,
+} from "./index-CITGJflw.js";
 import "zod";
 import "@modelcontextprotocol/sdk/client/sse.js";
 import "@modelcontextprotocol/sdk/shared/protocol.js";
@@ -15,398 +29,3 @@ import "ai";
 import "./mcp/do-oauth-client-provider.js";
 import "@modelcontextprotocol/sdk/client/auth.js";
 import "@modelcontextprotocol/sdk/shared/auth.js";
-
-/**
- * RPC request message from client
- */
-type RPCRequest = {
-  type: "rpc";
-  id: string;
-  method: string;
-  args: unknown[];
-};
-/**
- * State update message from client
- */
-type StateUpdateMessage = {
-  type: "cf_agent_state";
-  state: unknown;
-};
-/**
- * RPC response message to client
- */
-type RPCResponse = {
-  type: "rpc";
-  id: string;
-} & (
-  | {
-      success: true;
-      result: unknown;
-      done?: false;
-    }
-  | {
-      success: true;
-      result: unknown;
-      done: true;
-    }
-  | {
-      success: false;
-      error: string;
-    }
-);
-/**
- * Metadata for a callable method
- */
-type CallableMetadata = {
-  /** Optional description of what the method does */
-  description?: string;
-  /** Whether the method supports streaming responses */
-  streaming?: boolean;
-};
-/**
- * Decorator that marks a method as callable by clients
- * @param metadata Optional metadata about the callable method
- */
-declare function unstable_callable(
-  metadata?: CallableMetadata
-): <This, Args extends unknown[], Return>(
-  target: (this: This, ...args: Args) => Return,
-  context: ClassMethodDecoratorContext
-) => (this: This, ...args: Args) => Return;
-/**
- * Represents a scheduled task within an Agent
- * @template T Type of the payload data
- */
-type Schedule<T = string> = {
-  /** Unique identifier for the schedule */
-  id: string;
-  /** Name of the method to be called */
-  callback: string;
-  /** Data to be passed to the callback */
-  payload: T;
-} & (
-  | {
-      /** Type of schedule for one-time execution at a specific time */
-      type: "scheduled";
-      /** Timestamp when the task should execute */
-      time: number;
-    }
-  | {
-      /** Type of schedule for delayed execution */
-      type: "delayed";
-      /** Timestamp when the task should execute */
-      time: number;
-      /** Number of seconds to delay execution */
-      delayInSeconds: number;
-    }
-  | {
-      /** Type of schedule for recurring execution based on cron expression */
-      type: "cron";
-      /** Timestamp for the next execution */
-      time: number;
-      /** Cron expression defining the schedule */
-      cron: string;
-    }
-);
-/**
- * MCP Server state update message from server -> Client
- */
-type MCPServerMessage = {
-  type: "cf_agent_mcp_servers";
-  mcp: MCPServersState;
-};
-type MCPServersState = {
-  servers: {
-    [id: string]: MCPServer;
-  };
-  tools: Tool[];
-  prompts: Prompt[];
-  resources: Resource[];
-};
-type MCPServer = {
-  name: string;
-  server_url: string;
-  auth_url: string | null;
-  state: "authenticating" | "connecting" | "ready" | "discovering" | "failed";
-  instructions: string | null;
-  capabilities: ServerCapabilities | null;
-};
-declare function getCurrentAgent<
-  T extends Agent<unknown, unknown> = Agent<unknown, unknown>,
->(): {
-  agent: T | undefined;
-  connection: Connection | undefined;
-  request: Request<unknown, CfProperties<unknown>> | undefined;
-};
-/**
- * Base class for creating Agent implementations
- * @template Env Environment type containing bindings
- * @template State State type to store within the Agent
- */
-declare class Agent<Env, State = unknown> extends Server<Env> {
-  private _state;
-  private _ParentClass;
-  mcp: MCPClientManager;
-  /**
-   * Initial state for the Agent
-   * Override to provide default state values
-   */
-  initialState: State;
-  /**
-   * Current state of the Agent
-   */
-  get state(): State;
-  /**
-   * Agent configuration options
-   */
-  static options: {
-    /** Whether the Agent should hibernate when inactive */
-    hibernate: boolean;
-  };
-  /**
-   * 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<T = Record<string, string | number | boolean | null>>(
-    strings: TemplateStringsArray,
-    ...values: (string | number | boolean | null)[]
-  ): T[];
-  constructor(ctx: AgentContext, env: Env);
-  private _setStateInternal;
-  /**
-   * Update the Agent's state
-   * @param state New state to set
-   */
-  setState(state: State): void;
-  /**
-   * Called when the Agent's state is updated
-   * @param state Updated state
-   * @param source Source of the state update ("server" or a client connection)
-   */
-  onStateUpdate(state: State | undefined, source: Connection | "server"): void;
-  /**
-   * Called when the Agent receives an email
-   * @param email Email message to process
-   */
-  onEmail(email: ForwardableEmailMessage): Promise<void>;
-  private _tryCatch;
-  onError(connection: Connection, error: unknown): void | Promise<void>;
-  onError(error: unknown): void | Promise<void>;
-  /**
-   * Render content (not implemented in base class)
-   */
-  render(): void;
-  /**
-   * Schedule a task to be executed in the future
-   * @template T Type of the payload data
-   * @param when When to execute the task (Date, seconds delay, or cron expression)
-   * @param callback Name of the method to call
-   * @param payload Data to pass to the callback
-   * @returns Schedule object representing the scheduled task
-   */
-  schedule<T = string>(
-    when: Date | string | number,
-    callback: keyof this,
-    payload?: T
-  ): Promise<Schedule<T>>;
-  /**
-   * Get a scheduled task by ID
-   * @template T Type of the payload data
-   * @param id ID of the scheduled task
-   * @returns The Schedule object or undefined if not found
-   */
-  getSchedule<T = string>(id: string): Promise<Schedule<T> | undefined>;
-  /**
-   * Get scheduled tasks matching the given criteria
-   * @template T Type of the payload data
-   * @param criteria Criteria to filter schedules
-   * @returns Array of matching Schedule objects
-   */
-  getSchedules<T = string>(criteria?: {
-    id?: string;
-    type?: "scheduled" | "delayed" | "cron";
-    timeRange?: {
-      start?: Date;
-      end?: Date;
-    };
-  }): Schedule<T>[];
-  /**
-   * Cancel a scheduled task
-   * @param id ID of the task to cancel
-   * @returns true if the task was cancelled, false otherwise
-   */
-  cancelSchedule(id: string): Promise<boolean>;
-  private _scheduleNextAlarm;
-  /**
-   * 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/}
-   */
-  readonly alarm: () => Promise<void>;
-  /**
-   * Destroy the Agent, removing all state and scheduled tasks
-   */
-  destroy(): Promise<void>;
-  /**
-   * Get all methods marked as callable on this Agent
-   * @returns A map of method names to their metadata
-   */
-  private _isCallable;
-  /**
-   * 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
-   */
-  addMcpServer(
-    serverName: string,
-    url: string,
-    callbackHost: string,
-    agentsPrefix?: string,
-    options?: {
-      client?: ConstructorParameters<typeof Client>[1];
-      transport?: {
-        headers: HeadersInit;
-      };
-    }
-  ): Promise<{
-    id: string;
-    authUrl: string | undefined;
-  }>;
-  _connectToMcpServerInternal(
-    _serverName: string,
-    url: string,
-    callbackUrl: string,
-    options?: {
-      client?: ConstructorParameters<typeof Client>[1];
-      /**
-       * We don't expose the normal set of transport options because:
-       * 1) we can't serialize things like the auth provider or a fetch function into the DB for reconnection purposes
-       * 2) We probably want these options to be agnostic to the transport type (SSE vs Streamable)
-       *
-       * This has the limitation that you can't override fetch, but I think headers should handle nearly all cases needed (i.e. non-standard bearer auth).
-       */
-      transport?: {
-        headers?: HeadersInit;
-      };
-    },
-    reconnect?: {
-      id: string;
-      oauthClientId?: string;
-    }
-  ): Promise<{
-    id: string;
-    authUrl: string | undefined;
-    clientId: string | undefined;
-  }>;
-  removeMcpServer(id: string): Promise<void>;
-  getMcpServers(): MCPServersState;
-}
-/**
- * Namespace for creating Agent instances
- * @template Agentic Type of the Agent class
- */
-type AgentNamespace<Agentic extends Agent<unknown>> =
-  DurableObjectNamespace<Agentic>;
-/**
- * Agent's durable context
- */
-type AgentContext = DurableObjectState;
-/**
- * Configuration options for Agent routing
- */
-type AgentOptions<Env> = PartyServerOptions<Env> & {
-  /**
-   * Whether to enable CORS for the Agent
-   */
-  cors?: boolean | HeadersInit | undefined;
-};
-/**
- * Route a request to the appropriate Agent
- * @param request Request to route
- * @param env Environment containing Agent bindings
- * @param options Routing options
- * @returns Response from the Agent or undefined if no route matched
- */
-declare function routeAgentRequest<Env>(
-  request: Request,
-  env: Env,
-  options?: AgentOptions<Env>
-): Promise<Response | null>;
-/**
- * Route an email to the appropriate Agent
- * @param email Email message to route
- * @param env Environment containing Agent bindings
- * @param options Routing options
- */
-declare function routeAgentEmail<Env>(
-  _email: ForwardableEmailMessage,
-  _env: Env,
-  _options?: AgentOptions<Env>
-): Promise<void>;
-/**
- * Get or create an Agent by name
- * @template Env Environment type containing bindings
- * @template T Type of the Agent class
- * @param namespace Agent namespace
- * @param name Name of the Agent instance
- * @param options Options for Agent creation
- * @returns Promise resolving to an Agent instance stub
- */
-declare function getAgentByName<Env, T extends Agent<Env>>(
-  namespace: AgentNamespace<T>,
-  name: string,
-  options?: {
-    jurisdiction?: DurableObjectJurisdiction;
-    locationHint?: DurableObjectLocationHint;
-  }
-): Promise<DurableObjectStub<T>>;
-/**
- * A wrapper for streaming responses in callable methods
- */
-declare class StreamingResponse {
-  private _connection;
-  private _id;
-  private _closed;
-  constructor(connection: Connection, id: string);
-  /**
-   * Send a chunk of data to the client
-   * @param chunk The data to send
-   */
-  send(chunk: unknown): void;
-  /**
-   * End the stream and send the final chunk (if any)
-   * @param finalChunk Optional final chunk of data to send
-   */
-  end(finalChunk?: unknown): void;
-}
-
-export {
-  Agent,
-  type AgentContext,
-  type AgentNamespace,
-  type AgentOptions,
-  type CallableMetadata,
-  type MCPServer,
-  type MCPServerMessage,
-  type MCPServersState,
-  type RPCRequest,
-  type RPCResponse,
-  type Schedule,
-  type StateUpdateMessage,
-  StreamingResponse,
-  getAgentByName,
-  getCurrentAgent,
-  routeAgentEmail,
-  routeAgentRequest,
-  unstable_callable,
-};

package/dist/index.js

@@ -6,7 +6,7 @@ import {
   routeAgentEmail,
   routeAgentRequest,
   unstable_callable
-} from "./chunk-4CIGD73X.js";
+} from "./chunk-JFRK72K3.js";
 import "./chunk-E3LCYPCB.js";
 import "./chunk-767EASBA.js";
 import "./chunk-NKZZ66QY.js";

package/dist/mcp/index.js

@@ -1,6 +1,6 @@
 import {
   Agent
-} from "../chunk-4CIGD73X.js";
+} from "../chunk-JFRK72K3.js";
 import "../chunk-E3LCYPCB.js";
 import "../chunk-767EASBA.js";
 import "../chunk-NKZZ66QY.js";

package/dist/observability/index.d.ts

@@ -0,0 +1,12 @@
+import 'ai';
+export { b as Observability, O as ObservabilityEvent, g as genericObservability } from '../index-CITGJflw.js';
+import '@modelcontextprotocol/sdk/client/index.js';
+import '@modelcontextprotocol/sdk/types.js';
+import 'partyserver';
+import '../mcp/client.js';
+import 'zod';
+import '@modelcontextprotocol/sdk/client/sse.js';
+import '@modelcontextprotocol/sdk/shared/protocol.js';
+import '../mcp/do-oauth-client-provider.js';
+import '@modelcontextprotocol/sdk/client/auth.js';
+import '@modelcontextprotocol/sdk/shared/auth.js';

package/dist/observability/index.js

@@ -0,0 +1,10 @@
+import {
+  genericObservability
+} from "../chunk-JFRK72K3.js";
+import "../chunk-E3LCYPCB.js";
+import "../chunk-767EASBA.js";
+import "../chunk-NKZZ66QY.js";
+export {
+  genericObservability
+};
+//# sourceMappingURL=index.js.map

package/dist/observability/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/react.d.ts

@@ -1,6 +1,6 @@
 import { PartySocket } from "partysocket";
 import { usePartySocket } from "partysocket/react";
-import { MCPServersState, Agent } from "./index.js";
+import { M as MCPServersState, A as Agent } from "./index-CITGJflw.js";
 import { StreamOptions } from "./client.js";
 import { Method, RPCMethod } from "./serializable.js";
 import "@modelcontextprotocol/sdk/client/index.js";

package/dist/schedule.d.ts

@@ -16,14 +16,14 @@ declare const unstable_scheduleSchema: z.ZodObject<
       z.ZodTypeAny,
       {
         type: "scheduled" | "delayed" | "cron" | "no-schedule";
-        cron?: string | undefined;
         delayInSeconds?: number | undefined;
+        cron?: string | undefined;
         date?: Date | undefined;
       },
       {
         type: "scheduled" | "delayed" | "cron" | "no-schedule";
-        cron?: string | undefined;
         delayInSeconds?: number | undefined;
+        cron?: string | undefined;
         date?: Date | undefined;
       }
     >;
@@ -34,8 +34,8 @@ declare const unstable_scheduleSchema: z.ZodObject<
     description: string;
     when: {
       type: "scheduled" | "delayed" | "cron" | "no-schedule";
-      cron?: string | undefined;
       delayInSeconds?: number | undefined;
+      cron?: string | undefined;
       date?: Date | undefined;
     };
   },
@@ -43,8 +43,8 @@ declare const unstable_scheduleSchema: z.ZodObject<
     description: string;
     when: {
       type: "scheduled" | "delayed" | "cron" | "no-schedule";
-      cron?: string | undefined;
       delayInSeconds?: number | undefined;
+      cron?: string | undefined;
       date?: Date | undefined;
     };
   }

package/package.json

@@ -58,6 +58,11 @@
       "import": "./dist/mcp/do-oauth-client-provider.js",
       "require": "./dist/mcp/do-oauth-client-provider.js"
     },
+    "./observability": {
+      "import": "./dist/observability/index.js",
+      "require": "./dist/observability/index.js",
+      "types": "./dist/observability/index.d.ts"
+    },
     "./react": {
       "types": "./dist/react.d.ts",
       "import": "./dist/react.js",
@@ -96,5 +101,5 @@
   },
   "type": "module",
   "types": "dist/index.d.ts",
-  "version": "0.0.101"
+  "version": "0.0.102"
 }