agents 0.1.1 → 0.1.3

package/dist/mcp/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/mcp/index.ts"],"sourcesContent":["import { DurableObject } from \"cloudflare:workers\";\nimport type { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport type { McpServer } from \"@modelcontextprotocol/sdk/server/mcp.js\";\nimport type { Transport } from \"@modelcontextprotocol/sdk/shared/transport.js\";\nimport type { JSONRPCMessage } from \"@modelcontextprotocol/sdk/types.js\";\nimport {\n  InitializeRequestSchema,\n  JSONRPCMessageSchema,\n  isJSONRPCError,\n  isJSONRPCNotification,\n  isJSONRPCRequest,\n  isJSONRPCResponse,\n  type ElicitResult\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type { Connection, WSMessage } from \"../\";\nimport { Agent } from \"../index\";\n\nconst MAXIMUM_MESSAGE_SIZE_BYTES = 4 * 1024 * 1024; // 4MB\n\n// CORS helper functions\nfunction corsHeaders(_request: Request, corsOptions: CORSOptions = {}) {\n  const origin = \"*\";\n  return {\n    \"Access-Control-Allow-Headers\":\n      corsOptions.headers ||\n      \"Content-Type, mcp-session-id, mcp-protocol-version\",\n    \"Access-Control-Allow-Methods\": corsOptions.methods || \"GET, POST, OPTIONS\",\n    \"Access-Control-Allow-Origin\": corsOptions.origin || origin,\n    \"Access-Control-Expose-Headers\":\n      corsOptions.exposeHeaders || \"mcp-session-id\",\n    \"Access-Control-Max-Age\": (corsOptions.maxAge || 86400).toString()\n  };\n}\n\nfunction isDurableObjectNamespace(\n  namespace: unknown\n): namespace is DurableObjectNamespace<McpAgent> {\n  return (\n    typeof namespace === \"object\" &&\n    namespace !== null &&\n    \"newUniqueId\" in namespace &&\n    typeof namespace.newUniqueId === \"function\" &&\n    \"idFromName\" in namespace &&\n    typeof namespace.idFromName === \"function\"\n  );\n}\n\nfunction handleCORS(\n  request: Request,\n  corsOptions?: CORSOptions\n): Response | null {\n  if (request.method === \"OPTIONS\") {\n    return new Response(null, { headers: corsHeaders(request, corsOptions) });\n  }\n\n  return null;\n}\n\ninterface CORSOptions {\n  origin?: string;\n  methods?: string;\n  headers?: string;\n  maxAge?: number;\n  exposeHeaders?: string;\n}\n\nclass McpSSETransport implements Transport {\n  onclose?: () => void;\n  onerror?: (error: Error) => void;\n  onmessage?: (message: JSONRPCMessage) => void;\n  sessionId?: string;\n\n  private _getWebSocket: () => WebSocket | null;\n  private _started = false;\n  constructor(getWebSocket: () => WebSocket | null) {\n    this._getWebSocket = getWebSocket;\n  }\n\n  async start() {\n    // The transport does not manage the WebSocket connection since it's terminated\n    // by the Durable Object in order to allow hibernation. There's nothing to initialize.\n    if (this._started) {\n      throw new Error(\"Transport already started\");\n    }\n    this._started = true;\n  }\n\n  async send(message: JSONRPCMessage) {\n    if (!this._started) {\n      throw new Error(\"Transport not started\");\n    }\n    const websocket = this._getWebSocket();\n    if (!websocket) {\n      throw new Error(\"WebSocket not connected\");\n    }\n    try {\n      websocket.send(JSON.stringify(message));\n    } catch (error) {\n      this.onerror?.(error as Error);\n      throw error;\n    }\n  }\n\n  async close() {\n    // Similar to start, the only thing to do is to pass the event on to the server\n    this.onclose?.();\n  }\n}\n\ntype TransportType = \"sse\" | \"streamable-http\" | \"unset\";\n\nclass McpStreamableHttpTransport implements Transport {\n  onclose?: () => void;\n  onerror?: (error: Error) => void;\n  onmessage?: (message: JSONRPCMessage) => void;\n  sessionId?: string;\n\n  // TODO: If there is an open connection to send server-initiated messages\n  // back, we should use that connection\n  private _getWebSocketForGetRequest: () => WebSocket | null;\n\n  // Get the appropriate websocket connection for a given message id\n  private _getWebSocketForMessageID: (id: string) => WebSocket | null;\n\n  // Notify the server that a response has been sent for a given message id\n  // so that it may clean up it's mapping of message ids to connections\n  // once they are no longer needed\n  private _notifyResponseIdSent: (id: string) => void;\n\n  private _started = false;\n  constructor(\n    getWebSocketForMessageID: (id: string) => WebSocket | null,\n    notifyResponseIdSent: (id: string | number) => void\n  ) {\n    this._getWebSocketForMessageID = getWebSocketForMessageID;\n    this._notifyResponseIdSent = notifyResponseIdSent;\n    // TODO\n    this._getWebSocketForGetRequest = () => null;\n  }\n\n  async start() {\n    // The transport does not manage the WebSocket connection since it's terminated\n    // by the Durable Object in order to allow hibernation. There's nothing to initialize.\n    if (this._started) {\n      throw new Error(\"Transport already started\");\n    }\n    this._started = true;\n  }\n\n  async send(message: JSONRPCMessage) {\n    if (!this._started) {\n      throw new Error(\"Transport not started\");\n    }\n\n    let websocket: WebSocket | null = null;\n\n    if (isJSONRPCResponse(message) || isJSONRPCError(message)) {\n      websocket = this._getWebSocketForMessageID(message.id.toString());\n      if (!websocket) {\n        throw new Error(\n          `Could not find WebSocket for message id: ${message.id}`\n        );\n      }\n    } else if (isJSONRPCRequest(message)) {\n      // requests originating from the server must be sent over the\n      // the connection created by a GET request\n      websocket = this._getWebSocketForGetRequest();\n    } else if (isJSONRPCNotification(message)) {\n      // notifications do not have an id\n      // but do have a relatedRequestId field\n      // so that they can be sent to the correct connection\n      websocket = null;\n    }\n\n    try {\n      websocket?.send(JSON.stringify(message));\n      if (isJSONRPCResponse(message)) {\n        this._notifyResponseIdSent(message.id.toString());\n      }\n    } catch (error) {\n      this.onerror?.(error as Error);\n      throw error;\n    }\n  }\n\n  async close() {\n    // Similar to start, the only thing to do is to pass the event on to the server\n    this.onclose?.();\n  }\n}\n\ntype MaybePromise<T> = T | Promise<T>;\n\nexport abstract class McpAgent<\n  Env = unknown,\n  State = unknown,\n  Props extends Record<string, unknown> = Record<string, unknown>\n> extends DurableObject<Env> {\n  private _status: \"zero\" | \"starting\" | \"started\" = \"zero\";\n  private _transport?: Transport;\n  private _transportType: TransportType = \"unset\";\n  private _requestIdToConnectionId: Map<string | number, string> = new Map();\n\n  // Hibernation-safe elicitation handling\n  // Uses durable storage instead of in-memory handlers\n\n  /**\n   * Since McpAgent's _aren't_ yet real \"Agents\", let's only expose a couple of the methods\n   * to the outer class: initialState/state/setState/onStateUpdate/sql\n   */\n  private _agent: Agent<Env, State>;\n\n  get mcp() {\n    return this._agent.mcp;\n  }\n\n  protected constructor(ctx: DurableObjectState, env: Env) {\n    super(ctx, env);\n    const self = this;\n\n    this._agent = new (class extends Agent<Env, State> {\n      static options = {\n        hibernate: true\n      };\n\n      onStateUpdate(state: State | undefined, source: Connection | \"server\") {\n        return self.onStateUpdate(state, source);\n      }\n\n      async onMessage(\n        connection: Connection,\n        message: WSMessage\n      ): Promise<void> {\n        return self.onMessage(connection, message);\n      }\n    })(ctx, env);\n  }\n\n  /**\n   * Agents API allowlist\n   */\n  initialState!: State;\n  get state() {\n    return this._agent.state;\n  }\n  sql<T = Record<string, string | number | boolean | null>>(\n    strings: TemplateStringsArray,\n    ...values: (string | number | boolean | null)[]\n  ) {\n    return this._agent.sql<T>(strings, ...values);\n  }\n\n  setState(state: State) {\n    return this._agent.setState(state);\n  }\n\n  /**\n   * Elicit user input with a message and schema\n   */\n  async elicitInput(params: {\n    message: string;\n    requestedSchema: unknown;\n  }): Promise<ElicitResult> {\n    const requestId = `elicit_${Math.random().toString(36).substring(2, 11)}`;\n\n    // Store pending request in durable storage\n    await this.ctx.storage.put(`elicitation:${requestId}`, {\n      message: params.message,\n      requestedSchema: params.requestedSchema,\n      timestamp: Date.now()\n    });\n\n    const elicitRequest = {\n      jsonrpc: \"2.0\" as const,\n      id: requestId,\n      method: \"elicitation/create\",\n      params: {\n        message: params.message,\n        requestedSchema: params.requestedSchema\n      }\n    };\n\n    // Send through MCP transport\n    if (this._transport) {\n      await this._transport.send(elicitRequest);\n    } else {\n      const connections = this._agent?.getConnections();\n      if (!connections || Array.from(connections).length === 0) {\n        await this.ctx.storage.delete(`elicitation:${requestId}`);\n        throw new Error(\"No active connections available for elicitation\");\n      }\n\n      const connectionList = Array.from(connections);\n      for (const connection of connectionList) {\n        try {\n          connection.send(JSON.stringify(elicitRequest));\n        } catch (error) {\n          console.error(\"Failed to send elicitation request:\", error);\n        }\n      }\n    }\n\n    // Wait for response through MCP\n    return this._waitForElicitationResponse(requestId);\n  }\n\n  // we leave the variables as unused for autocomplete purposes\n  // biome-ignore lint/correctness/noUnusedFunctionParameters: overriden later\n  onStateUpdate(state: State | undefined, source: Connection | \"server\") {\n    // override this to handle state updates\n  }\n  async onStart() {\n    const self = this;\n\n    this._agent = new (class extends Agent<Env, State> {\n      initialState: State = self.initialState;\n      static options = {\n        hibernate: true\n      };\n\n      onStateUpdate(state: State | undefined, source: Connection | \"server\") {\n        return self.onStateUpdate(state, source);\n      }\n\n      async onMessage(connection: Connection, event: WSMessage) {\n        return self.onMessage(connection, event);\n      }\n    })(this.ctx, this.env);\n\n    this.props = (await this.ctx.storage.get(\"props\")) as Props;\n    this._transportType = (await this.ctx.storage.get(\n      \"transportType\"\n    )) as TransportType;\n    await this._init(this.props);\n\n    const server = await this.server;\n\n    // Connect to the MCP server\n    if (this._transportType === \"sse\") {\n      this._transport = new McpSSETransport(() => this.getWebSocket());\n      await server.connect(this._transport);\n    } else if (this._transportType === \"streamable-http\") {\n      this._transport = new McpStreamableHttpTransport(\n        (id) => this.getWebSocketForResponseID(id),\n        (id) => this._requestIdToConnectionId.delete(id)\n      );\n      await server.connect(this._transport);\n    }\n  }\n\n  /**\n   * McpAgent API\n   */\n  abstract server: MaybePromise<McpServer | Server>;\n  props!: Props;\n  initRun = false;\n\n  abstract init(): Promise<void>;\n\n  /**\n   * Handle errors that occur during initialization or operation.\n   * Override this method to provide custom error handling.\n   * @param error - The error that occurred\n   * @returns An error response object with status code and message\n   */\n  onError(error: Error): { status: number; message: string } {\n    console.error(\"McpAgent error:\", error);\n    return {\n      status: 500,\n      message:\n        error.message || \"An unexpected error occurred during initialization\"\n    };\n  }\n\n  async _init(props: Props) {\n    await this.updateProps(props);\n    if (!this.ctx.storage.get(\"transportType\")) {\n      await this.ctx.storage.put(\"transportType\", \"unset\");\n    }\n    if (!this.initRun) {\n      this.initRun = true;\n      try {\n        await this.init();\n      } catch (error) {\n        const errorResponse = this.onError(error as Error);\n        throw new Error(`Initialization failed: ${errorResponse.message}`);\n      }\n    }\n  }\n\n  async setInitialized() {\n    await this.ctx.storage.put(\"initialized\", true);\n  }\n\n  async isInitialized() {\n    return (await this.ctx.storage.get(\"initialized\")) === true;\n  }\n\n  async updateProps(props: Props) {\n    await this.ctx.storage.put(\"props\", props ?? {});\n    this.props = props;\n  }\n\n  private async _initialize(): Promise<void> {\n    await this.ctx.blockConcurrencyWhile(async () => {\n      this._status = \"starting\";\n      await this.onStart();\n      this._status = \"started\";\n    });\n  }\n\n  // Allow the worker to fetch a websocket connection to the agent\n  async fetch(request: Request): Promise<Response> {\n    if (this._status !== \"started\") {\n      // This means the server \"woke up\" after hibernation\n      // so we need to hydrate it again\n      await this._initialize();\n    }\n\n    // Only handle WebSocket upgrade requests\n    if (request.headers.get(\"Upgrade\") !== \"websocket\") {\n      return new Response(\"Expected WebSocket Upgrade request\", {\n        status: 400\n      });\n    }\n\n    // This request does not come from the user. The worker generates this\n    // request to generate a websocket connection to the agent.\n    const url = new URL(request.url);\n    // This is not the path that the user requested, but the path that the worker\n    // generated. We'll use this path to determine which transport to use.\n    const path = url.pathname;\n    const server = await this.server;\n\n    switch (path) {\n      case \"/sse\": {\n        // For SSE connections, we can only have one open connection per session\n        // If we get an upgrade while already connected, we should error\n        const websockets = this.ctx.getWebSockets();\n        if (websockets.length > 0) {\n          return new Response(\"Websocket already connected\", { status: 400 });\n        }\n\n        // This session must always use the SSE transporo\n        await this.ctx.storage.put(\"transportType\", \"sse\");\n        this._transportType = \"sse\";\n\n        if (!this._transport) {\n          this._transport = new McpSSETransport(() => this.getWebSocket());\n          await server.connect(this._transport);\n        }\n\n        // Defer to the Agent's fetch method to handle the WebSocket connection\n        return this._agent.fetch(request);\n      }\n      case \"/streamable-http\": {\n        if (!this._transport) {\n          this._transport = new McpStreamableHttpTransport(\n            (id) => this.getWebSocketForResponseID(id),\n            (id) => this._requestIdToConnectionId.delete(id)\n          );\n          await server.connect(this._transport);\n        }\n\n        // This session must always use the streamable-http transport\n        await this.ctx.storage.put(\"transportType\", \"streamable-http\");\n        this._transportType = \"streamable-http\";\n\n        return this._agent.fetch(request);\n      }\n      default:\n        return new Response(\n          \"Internal Server Error: Expected /sse or /streamable-http path\",\n          {\n            status: 500\n          }\n        );\n    }\n  }\n\n  getWebSocket() {\n    const websockets = this.ctx.getWebSockets();\n    if (websockets.length === 0) {\n      return null;\n    }\n    return websockets[0];\n  }\n\n  getWebSocketForResponseID(id: string): WebSocket | null {\n    const connectionId = this._requestIdToConnectionId.get(id);\n    if (connectionId === undefined) {\n      return null;\n    }\n    return this._agent.getConnection(connectionId) ?? null;\n  }\n\n  // All messages received here. This is currently never called\n  async onMessage(connection: Connection, event: WSMessage) {\n    // Since we address the DO via both the protocol and the session id,\n    // this should never happen, but let's enforce it just in case\n    if (this._transportType !== \"streamable-http\") {\n      const err = new Error(\n        \"Internal Server Error: Expected streamable-http protocol\"\n      );\n      this._transport?.onerror?.(err);\n      return;\n    }\n\n    let message: JSONRPCMessage;\n    try {\n      // Ensure event is a string\n      const data =\n        typeof event === \"string\" ? event : new TextDecoder().decode(event);\n      message = JSONRPCMessageSchema.parse(JSON.parse(data));\n    } catch (error) {\n      this._transport?.onerror?.(error as Error);\n      return;\n    }\n\n    // Check if this is an elicitation response before passing to transport\n    if (await this._handleElicitationResponse(message)) {\n      return; // Message was handled by elicitation system\n    }\n\n    // We need to map every incoming message to the connection that it came in on\n    // so that we can send relevant responses and notifications back on the same connection\n    if (isJSONRPCRequest(message)) {\n      this._requestIdToConnectionId.set(message.id.toString(), connection.id);\n    }\n\n    this._transport?.onmessage?.(message);\n  }\n\n  /**\n   * Wait for elicitation response through storage polling\n   */\n  private async _waitForElicitationResponse(\n    requestId: string\n  ): Promise<ElicitResult> {\n    const startTime = Date.now();\n    const timeout = 60000; // 60 second timeout\n\n    try {\n      while (Date.now() - startTime < timeout) {\n        // Check if response has been stored\n        const response = await this.ctx.storage.get<ElicitResult>(\n          `elicitation:response:${requestId}`\n        );\n        if (response) {\n          // Immediately clean up both request and response\n          await this.ctx.storage.delete(`elicitation:${requestId}`);\n          await this.ctx.storage.delete(`elicitation:response:${requestId}`);\n          return response;\n        }\n\n        // Sleep briefly before checking again\n        await new Promise((resolve) => setTimeout(resolve, 100));\n      }\n\n      throw new Error(\"Elicitation request timed out\");\n    } finally {\n      // Always clean up on timeout or error\n      await this.ctx.storage.delete(`elicitation:${requestId}`);\n      await this.ctx.storage.delete(`elicitation:response:${requestId}`);\n    }\n  }\n\n  /**\n   * Handle elicitation responses   */\n  private async _handleElicitationResponse(\n    message: JSONRPCMessage\n  ): Promise<boolean> {\n    // Check if this is a response to an elicitation request\n    if (isJSONRPCResponse(message) && message.result) {\n      const requestId = message.id?.toString();\n      if (!requestId || !requestId.startsWith(\"elicit_\")) return false;\n\n      // Check if we have a pending request for this ID\n      const pendingRequest = await this.ctx.storage.get(\n        `elicitation:${requestId}`\n      );\n      if (!pendingRequest) return false;\n\n      // Store the response in durable storage\n      await this.ctx.storage.put(\n        `elicitation:response:${requestId}`,\n        message.result as ElicitResult\n      );\n      return true;\n    }\n\n    // Check if this is an error response to an elicitation request\n    if (isJSONRPCError(message)) {\n      const requestId = message.id?.toString();\n      if (!requestId || !requestId.startsWith(\"elicit_\")) return false;\n\n      // Check if we have a pending request for this ID\n      const pendingRequest = await this.ctx.storage.get(\n        `elicitation:${requestId}`\n      );\n      if (!pendingRequest) return false;\n\n      // Store error response\n      const errorResult: ElicitResult = {\n        action: \"cancel\",\n        content: {\n          error: message.error.message || \"Elicitation request failed\"\n        }\n      };\n      await this.ctx.storage.put(\n        `elicitation:response:${requestId}`,\n        errorResult\n      );\n      return true;\n    }\n\n    return false;\n  }\n\n  // All messages received over SSE after the initial connection has been established\n  // will be passed here\n  async onSSEMcpMessage(\n    _sessionId: string,\n    messageBody: unknown\n  ): Promise<Error | null> {\n    if (this._status !== \"started\") {\n      // This means the server \"woke up\" after hibernation\n      // so we need to hydrate it again\n      await this._initialize();\n    }\n\n    // Since we address the DO via both the protocol and the session id,\n    // this should never happen, but let's enforce it just in case\n    if (this._transportType !== \"sse\") {\n      return new Error(\"Internal Server Error: Expected SSE protocol\");\n    }\n\n    try {\n      let parsedMessage: JSONRPCMessage;\n      try {\n        parsedMessage = JSONRPCMessageSchema.parse(messageBody);\n      } catch (error) {\n        this._transport?.onerror?.(error as Error);\n        throw error;\n      }\n\n      // Check if this is an elicitation response before passing to transport\n      if (await this._handleElicitationResponse(parsedMessage)) {\n        return null; // Message was handled by elicitation system\n      }\n\n      this._transport?.onmessage?.(parsedMessage);\n      return null;\n    } catch (error) {\n      console.error(\"Error forwarding message to SSE:\", error);\n      this._transport?.onerror?.(error as Error);\n      return error as Error;\n    }\n  }\n\n  // Delegate all websocket events to the underlying agent\n  async webSocketMessage(\n    ws: WebSocket,\n    event: ArrayBuffer | string\n  ): Promise<void> {\n    if (this._status !== \"started\") {\n      // This means the server \"woke up\" after hibernation\n      // so we need to hydrate it again\n      await this._initialize();\n    }\n    return await this._agent.webSocketMessage(ws, event);\n  }\n\n  // WebSocket event handlers for hibernation support\n  async webSocketError(ws: WebSocket, error: unknown): Promise<void> {\n    if (this._status !== \"started\") {\n      // This means the server \"woke up\" after hibernation\n      // so we need to hydrate it again\n      await this._initialize();\n    }\n    return await this._agent.webSocketError(ws, error);\n  }\n\n  async webSocketClose(\n    ws: WebSocket,\n    code: number,\n    reason: string,\n    wasClean: boolean\n  ): Promise<void> {\n    if (this._status !== \"started\") {\n      // This means the server \"woke up\" after hibernation\n      // so we need to hydrate it again\n      await this._initialize();\n    }\n    return await this._agent.webSocketClose(ws, code, reason, wasClean);\n  }\n\n  static mount(\n    path: string,\n    {\n      binding = \"MCP_OBJECT\",\n      corsOptions\n    }: {\n      binding?: string;\n      corsOptions?: CORSOptions;\n    } = {}\n  ) {\n    return McpAgent.serveSSE(path, { binding, corsOptions });\n  }\n\n  static serveSSE(\n    path: string,\n    {\n      binding = \"MCP_OBJECT\",\n      corsOptions\n    }: {\n      binding?: string;\n      corsOptions?: CORSOptions;\n    } = {}\n  ) {\n    let pathname = path;\n    if (path === \"/\") {\n      pathname = \"/*\";\n    }\n    const basePattern = new URLPattern({ pathname });\n    const messagePattern = new URLPattern({ pathname: `${pathname}/message` });\n\n    return {\n      async fetch<Env>(\n        this: void,\n        request: Request,\n        env: Env,\n        ctx: ExecutionContext\n      ): Promise<Response> {\n        // Handle CORS preflight\n        const corsResponse = handleCORS(request, corsOptions);\n        if (corsResponse) return corsResponse;\n\n        const url = new URL(request.url);\n        const bindingValue = env[binding as keyof typeof env] as unknown;\n\n        // Ensure we have a binding of some sort\n        if (bindingValue == null || typeof bindingValue !== \"object\") {\n          console.error(\n            `Could not find McpAgent binding for ${binding}. Did you update your wrangler configuration?`\n          );\n          return new Response(\"Invalid binding\", { status: 500 });\n        }\n\n        // Ensure that the binding is to a DurableObject\n        if (!isDurableObjectNamespace(bindingValue)) {\n          return new Response(\"Invalid binding\", { status: 500 });\n        }\n\n        const namespace =\n          bindingValue satisfies DurableObjectNamespace<McpAgent>;\n\n        // Handle initial SSE connection\n        if (request.method === \"GET\" && basePattern.test(url)) {\n          // Use a session ID if one is passed in, or create a unique\n          // session ID for this connection\n          const sessionId =\n            url.searchParams.get(\"sessionId\") ||\n            namespace.newUniqueId().toString();\n\n          // Create a Transform Stream for SSE\n          const { readable, writable } = new TransformStream();\n          const writer = writable.getWriter();\n          const encoder = new TextEncoder();\n\n          // Send the endpoint event\n          const endpointUrl = new URL(request.url);\n          endpointUrl.pathname = encodeURI(`${pathname}/message`);\n          endpointUrl.searchParams.set(\"sessionId\", sessionId);\n          const relativeUrlWithSession =\n            endpointUrl.pathname + endpointUrl.search + endpointUrl.hash;\n          const endpointMessage = `event: endpoint\\ndata: ${relativeUrlWithSession}\\n\\n`;\n          writer.write(encoder.encode(endpointMessage));\n\n          // Get the Durable Object\n          const id = namespace.idFromName(`sse:${sessionId}`);\n          const doStub = namespace.get(id);\n\n          // Initialize the object\n          try {\n            await doStub._init(ctx.props);\n          } catch (error) {\n            console.error(\"Failed to initialize McpAgent:\", error);\n            await writer.close();\n            const errorMessage =\n              error instanceof Error ? error.message : String(error);\n            return new Response(`Initialization failed: ${errorMessage}`, {\n              status: 500\n            });\n          }\n\n          // Connect to the Durable Object via WebSocket\n          const upgradeUrl = new URL(request.url);\n          // enforce that the path that the DO receives is always /sse\n          upgradeUrl.pathname = \"/sse\";\n          const existingHeaders: Record<string, string> = {};\n          request.headers.forEach((value, key) => {\n            existingHeaders[key] = value;\n          });\n          const response = await doStub.fetch(\n            new Request(upgradeUrl, {\n              headers: {\n                ...existingHeaders,\n                Upgrade: \"websocket\",\n                // Required by PartyServer\n                \"x-partykit-room\": sessionId\n              }\n            })\n          );\n\n          // Get the WebSocket\n          const ws = response.webSocket;\n          if (!ws) {\n            console.error(\"Failed to establish WebSocket connection\");\n            await writer.close();\n            return new Response(\"Failed to establish WebSocket connection\", {\n              status: 500\n            });\n          }\n\n          // Accept the WebSocket\n          ws.accept();\n\n          // Handle messages from the Durable Object\n          ws.addEventListener(\"message\", (event) => {\n            async function onMessage(event: MessageEvent) {\n              try {\n                const message = JSON.parse(event.data);\n\n                // validate that the message is a valid JSONRPC message\n                const result = JSONRPCMessageSchema.safeParse(message);\n                if (!result.success) {\n                  // The message was not a valid JSONRPC message, so we will drop it\n                  // PartyKit will broadcast state change messages to all connected clients\n                  // and we need to filter those out so they are not passed to MCP clients\n                  return;\n                }\n\n                // Send the message as an SSE event\n                const messageText = `event: message\\ndata: ${JSON.stringify(result.data)}\\n\\n`;\n                await writer.write(encoder.encode(messageText));\n              } catch (error) {\n                console.error(\"Error forwarding message to SSE:\", error);\n              }\n            }\n            onMessage(event).catch(console.error);\n          });\n\n          // Handle WebSocket errors\n          ws.addEventListener(\"error\", (error) => {\n            async function onError(_error: Event) {\n              try {\n                await writer.close();\n              } catch (_e) {\n                // Ignore errors when closing\n              }\n            }\n            onError(error).catch(console.error);\n          });\n\n          // Handle WebSocket closure\n          ws.addEventListener(\"close\", () => {\n            async function onClose() {\n              try {\n                await writer.close();\n              } catch (error) {\n                console.error(\"Error closing SSE connection:\", error);\n              }\n            }\n            onClose().catch(console.error);\n          });\n\n          // Return the SSE response\n          return new Response(readable, {\n            headers: {\n              \"Cache-Control\": \"no-cache\",\n              Connection: \"keep-alive\",\n              \"Content-Type\": \"text/event-stream\",\n              ...corsHeaders(request, corsOptions)\n            }\n          });\n        }\n\n        // Handle incoming MCP messages. These will be passed to McpAgent\n        // but the response will be sent back via the open SSE connection\n        // so we only need to return a 202 Accepted response for success\n        if (request.method === \"POST\" && messagePattern.test(url)) {\n          const sessionId = url.searchParams.get(\"sessionId\");\n          if (!sessionId) {\n            return new Response(\n              `Missing sessionId. Expected POST to ${pathname} to initiate new one`,\n              { status: 400 }\n            );\n          }\n\n          const contentType = request.headers.get(\"content-type\") || \"\";\n          if (!contentType.includes(\"application/json\")) {\n            return new Response(`Unsupported content-type: ${contentType}`, {\n              status: 400\n            });\n          }\n\n          // check if the request body is too large\n          const contentLength = Number.parseInt(\n            request.headers.get(\"content-length\") || \"0\",\n            10\n          );\n          if (contentLength > MAXIMUM_MESSAGE_SIZE_BYTES) {\n            return new Response(\n              `Request body too large: ${contentLength} bytes`,\n              {\n                status: 400\n              }\n            );\n          }\n\n          // Get the Durable Object\n          const id = namespace.idFromName(`sse:${sessionId}`);\n          const doStub = namespace.get(id);\n\n          const messageBody = await request.json();\n          // Update props with fresh values before processing message\n          await doStub.updateProps(ctx.props);\n          const error = await doStub.onSSEMcpMessage(sessionId, messageBody);\n\n          if (error) {\n            return new Response(error.message, {\n              headers: {\n                \"Cache-Control\": \"no-cache\",\n                Connection: \"keep-alive\",\n                \"Content-Type\": \"text/event-stream\",\n                ...corsHeaders(request, corsOptions)\n              },\n              status: 400\n            });\n          }\n\n          return new Response(\"Accepted\", {\n            headers: {\n              \"Cache-Control\": \"no-cache\",\n              Connection: \"keep-alive\",\n              \"Content-Type\": \"text/event-stream\",\n              ...corsHeaders(request, corsOptions)\n            },\n            status: 202\n          });\n        }\n\n        return new Response(\"Not Found\", { status: 404 });\n      }\n    };\n  }\n\n  static serve(\n    path: string,\n    {\n      binding = \"MCP_OBJECT\",\n      corsOptions\n    }: { binding?: string; corsOptions?: CORSOptions } = {}\n  ) {\n    let pathname = path;\n    if (path === \"/\") {\n      pathname = \"/*\";\n    }\n    const basePattern = new URLPattern({ pathname });\n\n    return {\n      async fetch<Env>(\n        this: void,\n        request: Request,\n        env: Env,\n        ctx: ExecutionContext\n      ): Promise<Response> {\n        // Handle CORS preflight\n        const corsResponse = handleCORS(request, corsOptions);\n        if (corsResponse) {\n          return corsResponse;\n        }\n\n        const url = new URL(request.url);\n        const bindingValue = env[binding as keyof typeof env] as unknown;\n\n        // Ensure we have a binding of some sort\n        if (bindingValue == null || typeof bindingValue !== \"object\") {\n          console.error(\n            `Could not find McpAgent binding for ${binding}. Did you update your wrangler configuration?`\n          );\n          return new Response(\"Invalid binding\", { status: 500 });\n        }\n\n        // Ensure that the binding is to a DurableObject\n        if (!isDurableObjectNamespace(bindingValue)) {\n          return new Response(\"Invalid binding\", { status: 500 });\n        }\n\n        const namespace =\n          bindingValue satisfies DurableObjectNamespace<McpAgent>;\n\n        if (request.method === \"POST\" && basePattern.test(url)) {\n          // validate the Accept header\n          const acceptHeader = request.headers.get(\"accept\");\n          // The client MUST include an Accept header, listing both application/json and text/event-stream as supported content types.\n          if (\n            !acceptHeader?.includes(\"application/json\") ||\n            !acceptHeader.includes(\"text/event-stream\")\n          ) {\n            const body = JSON.stringify({\n              error: {\n                code: -32000,\n                message:\n                  \"Not Acceptable: Client must accept both application/json and text/event-stream\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 406 });\n          }\n\n          const ct = request.headers.get(\"content-type\");\n          if (!ct || !ct.includes(\"application/json\")) {\n            const body = JSON.stringify({\n              error: {\n                code: -32000,\n                message:\n                  \"Unsupported Media Type: Content-Type must be application/json\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 415 });\n          }\n\n          // Check content length against maximum allowed size\n          const contentLength = Number.parseInt(\n            request.headers.get(\"content-length\") ?? \"0\",\n            10\n          );\n          if (contentLength > MAXIMUM_MESSAGE_SIZE_BYTES) {\n            const body = JSON.stringify({\n              error: {\n                code: -32000,\n                message: `Request body too large. Maximum size is ${MAXIMUM_MESSAGE_SIZE_BYTES} bytes`\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 413 });\n          }\n\n          let sessionId = request.headers.get(\"mcp-session-id\");\n          let rawMessage: unknown;\n\n          try {\n            rawMessage = await request.json();\n          } catch (_error) {\n            const body = JSON.stringify({\n              error: {\n                code: -32700,\n                message: \"Parse error: Invalid JSON\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 400 });\n          }\n\n          // Make sure the message is an array to simplify logic\n          let arrayMessage: unknown[];\n          if (Array.isArray(rawMessage)) {\n            arrayMessage = rawMessage;\n          } else {\n            arrayMessage = [rawMessage];\n          }\n\n          let messages: JSONRPCMessage[] = [];\n\n          // Try to parse each message as JSON RPC. Fail if any message is invalid\n          for (const msg of arrayMessage) {\n            if (!JSONRPCMessageSchema.safeParse(msg).success) {\n              const body = JSON.stringify({\n                error: {\n                  code: -32700,\n                  message: \"Parse error: Invalid JSON-RPC message\"\n                },\n                id: null,\n                jsonrpc: \"2.0\"\n              });\n              return new Response(body, { status: 400 });\n            }\n          }\n\n          messages = arrayMessage.map((msg) => JSONRPCMessageSchema.parse(msg));\n\n          // Before we pass the messages to the agent, there's another error condition we need to enforce\n          // Check if this is an initialization request\n          // https://spec.modelcontextprotocol.io/specification/2025-03-26/basic/lifecycle/\n          const isInitializationRequest = messages.some(\n            (msg) => InitializeRequestSchema.safeParse(msg).success\n          );\n\n          if (isInitializationRequest && sessionId) {\n            const body = JSON.stringify({\n              error: {\n                code: -32600,\n                message:\n                  \"Invalid Request: Initialization requests must not include a sessionId\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 400 });\n          }\n\n          // The initialization request must be the only request in the batch\n          if (isInitializationRequest && messages.length > 1) {\n            const body = JSON.stringify({\n              error: {\n                code: -32600,\n                message:\n                  \"Invalid Request: Only one initialization request is allowed\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 400 });\n          }\n\n          // If an Mcp-Session-Id is returned by the server during initialization,\n          // clients using the Streamable HTTP transport MUST include it\n          // in the Mcp-Session-Id header on all of their subsequent HTTP requests.\n          if (!isInitializationRequest && !sessionId) {\n            const body = JSON.stringify({\n              error: {\n                code: -32000,\n                message: \"Bad Request: Mcp-Session-Id header is required\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 400 });\n          }\n\n          // If we don't have a sessionId, we are serving an initialization request\n          // and need to generate a new sessionId\n          sessionId = sessionId ?? namespace.newUniqueId().toString();\n\n          // fetch the agent DO\n          const id = namespace.idFromName(`streamable-http:${sessionId}`);\n          const doStub = namespace.get(id);\n          const isInitialized = await doStub.isInitialized();\n\n          if (isInitializationRequest) {\n            try {\n              await doStub._init(ctx.props);\n              await doStub.setInitialized();\n            } catch (error) {\n              console.error(\"Failed to initialize McpAgent:\", error);\n              const errorMessage =\n                error instanceof Error ? error.message : String(error);\n              const body = JSON.stringify({\n                error: {\n                  code: -32001,\n                  message: `Initialization failed: ${errorMessage}`\n                },\n                id: null,\n                jsonrpc: \"2.0\"\n              });\n              return new Response(body, { status: 500 });\n            }\n          } else if (!isInitialized) {\n            // if we have gotten here, then a session id that was never initialized\n            // was provided\n            const body = JSON.stringify({\n              error: {\n                code: -32001,\n                message: \"Session not found\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 404 });\n          } else {\n            // Update props for existing sessions\n            await doStub.updateProps(ctx.props);\n          }\n\n          // We've evaluated all the error conditions! Now it's time to establish\n          // all the streams\n\n          // Create a Transform Stream for SSE\n          const { readable, writable } = new TransformStream();\n          const writer = writable.getWriter();\n          const encoder = new TextEncoder();\n\n          // Connect to the Durable Object via WebSocket\n          const upgradeUrl = new URL(request.url);\n          upgradeUrl.pathname = \"/streamable-http\";\n          const existingHeaders: Record<string, string> = {};\n          request.headers.forEach((value, key) => {\n            existingHeaders[key] = value;\n          });\n          const response = await doStub.fetch(\n            new Request(upgradeUrl, {\n              headers: {\n                ...existingHeaders,\n                Upgrade: \"websocket\",\n                // Required by PartyServer\n                \"x-partykit-room\": sessionId\n              }\n            })\n          );\n\n          // Get the WebSocket\n          const ws = response.webSocket;\n          if (!ws) {\n            console.error(\"Failed to establish WebSocket connection\");\n\n            await writer.close();\n            const body = JSON.stringify({\n              error: {\n                code: -32001,\n                message: \"Failed to establish WebSocket connection\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 500 });\n          }\n\n          // Keep track of the request ids that we have sent to the server\n          // so that we can close the connection once we have received\n          // all the responses\n          const requestIds: Set<string | number> = new Set();\n\n          // Accept the WebSocket\n          ws.accept();\n\n          // Handle messages from the Durable Object\n          ws.addEventListener(\"message\", (event) => {\n            async function onMessage(event: MessageEvent) {\n              try {\n                const data =\n                  typeof event.data === \"string\"\n                    ? event.data\n                    : new TextDecoder().decode(event.data);\n                const message = JSON.parse(data);\n\n                // validate that the message is a valid JSONRPC message\n                const result = JSONRPCMessageSchema.safeParse(message);\n                if (!result.success) {\n                  // The message was not a valid JSONRPC message, so we will drop it\n                  // PartyKit will broadcast state change messages to all connected clients\n                  // and we need to filter those out so they are not passed to MCP clients\n                  return;\n                }\n\n                // If the message is a response or an error, remove the id from the set of\n                // request ids\n                if (\n                  isJSONRPCResponse(result.data) ||\n                  isJSONRPCError(result.data)\n                ) {\n                  requestIds.delete(result.data.id);\n                }\n\n                // Send the message as an SSE event\n                const messageText = `event: message\\ndata: ${JSON.stringify(result.data)}\\n\\n`;\n                await writer.write(encoder.encode(messageText));\n\n                // If we have received all the responses, close the connection\n                if (requestIds.size === 0) {\n                  ws!.close();\n                }\n              } catch (error) {\n                console.error(\"Error forwarding message to SSE:\", error);\n              }\n            }\n            onMessage(event).catch(console.error);\n          });\n\n          // Handle WebSocket errors\n          ws.addEventListener(\"error\", (error) => {\n            async function onError(_error: Event) {\n              try {\n                await writer.close();\n              } catch (_e) {\n                // Ignore errors when closing\n              }\n            }\n            onError(error).catch(console.error);\n          });\n\n          // Handle WebSocket closure\n          ws.addEventListener(\"close\", () => {\n            async function onClose() {\n              try {\n                await writer.close();\n              } catch (error) {\n                console.error(\"Error closing SSE connection:\", error);\n              }\n            }\n            onClose().catch(console.error);\n          });\n\n          // If there are no requests, we send the messages to the agent and acknowledge the request with a 202\n          // since we don't expect any responses back through this connection\n          const hasOnlyNotificationsOrResponses = messages.every(\n            (msg) => isJSONRPCNotification(msg) || isJSONRPCResponse(msg)\n          );\n          if (hasOnlyNotificationsOrResponses) {\n            for (const message of messages) {\n              ws.send(JSON.stringify(message));\n            }\n\n            // closing the websocket will also close the SSE connection\n            ws.close();\n\n            return new Response(null, {\n              headers: corsHeaders(request, corsOptions),\n              status: 202\n            });\n          }\n\n          for (const message of messages) {\n            if (isJSONRPCRequest(message)) {\n              // add each request id that we send off to a set\n              // so that we can keep track of which requests we\n              // still need a response for\n              requestIds.add(message.id);\n            }\n            ws.send(JSON.stringify(message));\n          }\n\n          // Return the SSE response. We handle closing the stream in the ws \"message\"\n          // handler\n          return new Response(readable, {\n            headers: {\n              \"Cache-Control\": \"no-cache\",\n              Connection: \"keep-alive\",\n              \"Content-Type\": \"text/event-stream\",\n              \"mcp-session-id\": sessionId,\n              ...corsHeaders(request, corsOptions)\n            },\n            status: 200\n          });\n        }\n\n        // We don't yet support GET or DELETE requests\n        const body = JSON.stringify({\n          error: {\n            code: -32000,\n            message: \"Method not allowed\"\n          },\n          id: null,\n          jsonrpc: \"2.0\"\n        });\n        return new Response(body, { status: 405 });\n      }\n    };\n  }\n}\n\n// Export client transport classes\nexport { SSEEdgeClientTransport } from \"./sse-edge\";\nexport { StreamableHTTPEdgeClientTransport } from \"./streamable-http-edge\";\n\n// Export elicitation types and schemas\nexport {\n  ElicitRequestSchema,\n  type ElicitRequest,\n  type ElicitResult\n} from \"@modelcontextprotocol/sdk/types.js\";\n"],"mappings":";;;;;;;;;;;;AAAA,SAAS,qBAAqB;AAK9B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AA+0CP;AAAA,EACE;AAAA,OAGK;AA/0CP,IAAM,6BAA6B,IAAI,OAAO;AAG9C,SAAS,YAAY,UAAmB,cAA2B,CAAC,GAAG;AACrE,QAAM,SAAS;AACf,SAAO;AAAA,IACL,gCACE,YAAY,WACZ;AAAA,IACF,gCAAgC,YAAY,WAAW;AAAA,IACvD,+BAA+B,YAAY,UAAU;AAAA,IACrD,iCACE,YAAY,iBAAiB;AAAA,IAC/B,2BAA2B,YAAY,UAAU,OAAO,SAAS;AAAA,EACnE;AACF;AAEA,SAAS,yBACP,WAC+C;AAC/C,SACE,OAAO,cAAc,YACrB,cAAc,QACd,iBAAiB,aACjB,OAAO,UAAU,gBAAgB,cACjC,gBAAgB,aAChB,OAAO,UAAU,eAAe;AAEpC;AAEA,SAAS,WACP,SACA,aACiB;AACjB,MAAI,QAAQ,WAAW,WAAW;AAChC,WAAO,IAAI,SAAS,MAAM,EAAE,SAAS,YAAY,SAAS,WAAW,EAAE,CAAC;AAAA,EAC1E;AAEA,SAAO;AACT;AAUA,IAAM,kBAAN,MAA2C;AAAA,EAQzC,YAAY,cAAsC;AADlD,SAAQ,WAAW;AAEjB,SAAK,gBAAgB;AAAA,EACvB;AAAA,EAEA,MAAM,QAAQ;AAGZ,QAAI,KAAK,UAAU;AACjB,YAAM,IAAI,MAAM,2BAA2B;AAAA,IAC7C;AACA,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,MAAM,KAAK,SAAyB;AAClC,QAAI,CAAC,KAAK,UAAU;AAClB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AACA,UAAM,YAAY,KAAK,cAAc;AACrC,QAAI,CAAC,WAAW;AACd,YAAM,IAAI,MAAM,yBAAyB;AAAA,IAC3C;AACA,QAAI;AACF,gBAAU,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,IACxC,SAAS,OAAO;AACd,WAAK,UAAU,KAAc;AAC7B,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAM,QAAQ;AAEZ,SAAK,UAAU;AAAA,EACjB;AACF;AAIA,IAAM,6BAAN,MAAsD;AAAA,EAmBpD,YACE,0BACA,sBACA;AAJF,SAAQ,WAAW;AAKjB,SAAK,4BAA4B;AACjC,SAAK,wBAAwB;AAE7B,SAAK,6BAA6B,MAAM;AAAA,EAC1C;AAAA,EAEA,MAAM,QAAQ;AAGZ,QAAI,KAAK,UAAU;AACjB,YAAM,IAAI,MAAM,2BAA2B;AAAA,IAC7C;AACA,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,MAAM,KAAK,SAAyB;AAClC,QAAI,CAAC,KAAK,UAAU;AAClB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AAEA,QAAI,YAA8B;AAElC,QAAI,kBAAkB,OAAO,KAAK,eAAe,OAAO,GAAG;AACzD,kBAAY,KAAK,0BAA0B,QAAQ,GAAG,SAAS,CAAC;AAChE,UAAI,CAAC,WAAW;AACd,cAAM,IAAI;AAAA,UACR,4CAA4C,QAAQ,EAAE;AAAA,QACxD;AAAA,MACF;AAAA,IACF,WAAW,iBAAiB,OAAO,GAAG;AAGpC,kBAAY,KAAK,2BAA2B;AAAA,IAC9C,WAAW,sBAAsB,OAAO,GAAG;AAIzC,kBAAY;AAAA,IACd;AAEA,QAAI;AACF,iBAAW,KAAK,KAAK,UAAU,OAAO,CAAC;AACvC,UAAI,kBAAkB,OAAO,GAAG;AAC9B,aAAK,sBAAsB,QAAQ,GAAG,SAAS,CAAC;AAAA,MAClD;AAAA,IACF,SAAS,OAAO;AACd,WAAK,UAAU,KAAc;AAC7B,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAM,QAAQ;AAEZ,SAAK,UAAU;AAAA,EACjB;AACF;AAIO,IAAe,WAAf,MAAe,kBAIZ,cAAmB;AAAA,EAmBjB,YAAY,KAAyB,KAAU;AAxN3D;AAyNI,UAAM,KAAK,GAAG;AAnBhB,SAAQ,UAA2C;AAEnD,SAAQ,iBAAgC;AACxC,SAAQ,2BAAyD,oBAAI,IAAI;AA0JzE,mBAAU;AAzIR,UAAM,OAAO;AAEb,SAAK,SAAS,KAAK,mBAAc,MAAkB;AAAA,MAKjD,cAAc,OAA0B,QAA+B;AACrE,eAAO,KAAK,cAAc,OAAO,MAAM;AAAA,MACzC;AAAA,MAEA,MAAM,UACJ,YACA,SACe;AACf,eAAO,KAAK,UAAU,YAAY,OAAO;AAAA,MAC3C;AAAA,IACF,GAfmB,GACV,UAAU;AAAA,MACf,WAAW;AAAA,IACb,GAHiB,IAehB,KAAK,GAAG;AAAA,EACb;AAAA,EAxBA,IAAI,MAAM;AACR,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EA4BA,IAAI,QAAQ;AACV,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EACA,IACE,YACG,QACH;AACA,WAAO,KAAK,OAAO,IAAO,SAAS,GAAG,MAAM;AAAA,EAC9C;AAAA,EAEA,SAAS,OAAc;AACrB,WAAO,KAAK,OAAO,SAAS,KAAK;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YAAY,QAGQ;AACxB,UAAM,YAAY,UAAU,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,UAAU,GAAG,EAAE,CAAC;AAGvE,UAAM,KAAK,IAAI,QAAQ,IAAI,eAAe,SAAS,IAAI;AAAA,MACrD,SAAS,OAAO;AAAA,MAChB,iBAAiB,OAAO;AAAA,MACxB,WAAW,KAAK,IAAI;AAAA,IACtB,CAAC;AAED,UAAM,gBAAgB;AAAA,MACpB,SAAS;AAAA,MACT,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR,QAAQ;AAAA,QACN,SAAS,OAAO;AAAA,QAChB,iBAAiB,OAAO;AAAA,MAC1B;AAAA,IACF;AAGA,QAAI,KAAK,YAAY;AACnB,YAAM,KAAK,WAAW,KAAK,aAAa;AAAA,IAC1C,OAAO;AACL,YAAM,cAAc,KAAK,QAAQ,eAAe;AAChD,UAAI,CAAC,eAAe,MAAM,KAAK,WAAW,EAAE,WAAW,GAAG;AACxD,cAAM,KAAK,IAAI,QAAQ,OAAO,eAAe,SAAS,EAAE;AACxD,cAAM,IAAI,MAAM,iDAAiD;AAAA,MACnE;AAEA,YAAM,iBAAiB,MAAM,KAAK,WAAW;AAC7C,iBAAW,cAAc,gBAAgB;AACvC,YAAI;AACF,qBAAW,KAAK,KAAK,UAAU,aAAa,CAAC;AAAA,QAC/C,SAAS,OAAO;AACd,kBAAQ,MAAM,uCAAuC,KAAK;AAAA,QAC5D;AAAA,MACF;AAAA,IACF;AAGA,WAAO,KAAK,4BAA4B,SAAS;AAAA,EACnD;AAAA;AAAA;AAAA,EAIA,cAAc,OAA0B,QAA+B;AAAA,EAEvE;AAAA,EACA,MAAM,UAAU;AAvTlB;AAwTI,UAAM,OAAO;AAEb,SAAK,SAAS,KAAK,mBAAc,MAAkB;AAAA,MAAhC;AAAA;AACjB,4BAAsB,KAAK;AAAA;AAAA,MAK3B,cAAc,OAA0B,QAA+B;AACrE,eAAO,KAAK,cAAc,OAAO,MAAM;AAAA,MACzC;AAAA,MAEA,MAAM,UAAU,YAAwB,OAAkB;AACxD,eAAO,KAAK,UAAU,YAAY,KAAK;AAAA,MACzC;AAAA,IACF,GAbmB,GAEV,UAAU;AAAA,MACf,WAAW;AAAA,IACb,GAJiB,IAahB,KAAK,KAAK,KAAK,GAAG;AAErB,SAAK,QAAS,MAAM,KAAK,IAAI,QAAQ,IAAI,OAAO;AAChD,SAAK,iBAAkB,MAAM,KAAK,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,UAAM,KAAK,MAAM,KAAK,KAAK;AAE3B,UAAM,SAAS,MAAM,KAAK;AAG1B,QAAI,KAAK,mBAAmB,OAAO;AACjC,WAAK,aAAa,IAAI,gBAAgB,MAAM,KAAK,aAAa,CAAC;AAC/D,YAAM,OAAO,QAAQ,KAAK,UAAU;AAAA,IACtC,WAAW,KAAK,mBAAmB,mBAAmB;AACpD,WAAK,aAAa,IAAI;AAAA,QACpB,CAAC,OAAO,KAAK,0BAA0B,EAAE;AAAA,QACzC,CAAC,OAAO,KAAK,yBAAyB,OAAO,EAAE;AAAA,MACjD;AACA,YAAM,OAAO,QAAQ,KAAK,UAAU;AAAA,IACtC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAiBA,QAAQ,OAAmD;AACzD,YAAQ,MAAM,mBAAmB,KAAK;AACtC,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,SACE,MAAM,WAAW;AAAA,IACrB;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,OAAc;AACxB,UAAM,KAAK,YAAY,KAAK;AAC5B,QAAI,CAAC,KAAK,IAAI,QAAQ,IAAI,eAAe,GAAG;AAC1C,YAAM,KAAK,IAAI,QAAQ,IAAI,iBAAiB,OAAO;AAAA,IACrD;AACA,QAAI,CAAC,KAAK,SAAS;AACjB,WAAK,UAAU;AACf,UAAI;AACF,cAAM,KAAK,KAAK;AAAA,MAClB,SAAS,OAAO;AACd,cAAM,gBAAgB,KAAK,QAAQ,KAAc;AACjD,cAAM,IAAI,MAAM,0BAA0B,cAAc,OAAO,EAAE;AAAA,MACnE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,iBAAiB;AACrB,UAAM,KAAK,IAAI,QAAQ,IAAI,eAAe,IAAI;AAAA,EAChD;AAAA,EAEA,MAAM,gBAAgB;AACpB,WAAQ,MAAM,KAAK,IAAI,QAAQ,IAAI,aAAa,MAAO;AAAA,EACzD;AAAA,EAEA,MAAM,YAAY,OAAc;AAC9B,UAAM,KAAK,IAAI,QAAQ,IAAI,SAAS,SAAS,CAAC,CAAC;AAC/C,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,MAAc,cAA6B;AACzC,UAAM,KAAK,IAAI,sBAAsB,YAAY;AAC/C,WAAK,UAAU;AACf,YAAM,KAAK,QAAQ;AACnB,WAAK,UAAU;AAAA,IACjB,CAAC;AAAA,EACH;AAAA;AAAA,EAGA,MAAM,MAAM,SAAqC;AAC/C,QAAI,KAAK,YAAY,WAAW;AAG9B,YAAM,KAAK,YAAY;AAAA,IACzB;AAGA,QAAI,QAAQ,QAAQ,IAAI,SAAS,MAAM,aAAa;AAClD,aAAO,IAAI,SAAS,sCAAsC;AAAA,QACxD,QAAQ;AAAA,MACV,CAAC;AAAA,IACH;AAIA,UAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAG/B,UAAM,OAAO,IAAI;AACjB,UAAM,SAAS,MAAM,KAAK;AAE1B,YAAQ,MAAM;AAAA,MACZ,KAAK,QAAQ;AAGX,cAAM,aAAa,KAAK,IAAI,cAAc;AAC1C,YAAI,WAAW,SAAS,GAAG;AACzB,iBAAO,IAAI,SAAS,+BAA+B,EAAE,QAAQ,IAAI,CAAC;AAAA,QACpE;AAGA,cAAM,KAAK,IAAI,QAAQ,IAAI,iBAAiB,KAAK;AACjD,aAAK,iBAAiB;AAEtB,YAAI,CAAC,KAAK,YAAY;AACpB,eAAK,aAAa,IAAI,gBAAgB,MAAM,KAAK,aAAa,CAAC;AAC/D,gBAAM,OAAO,QAAQ,KAAK,UAAU;AAAA,QACtC;AAGA,eAAO,KAAK,OAAO,MAAM,OAAO;AAAA,MAClC;AAAA,MACA,KAAK,oBAAoB;AACvB,YAAI,CAAC,KAAK,YAAY;AACpB,eAAK,aAAa,IAAI;AAAA,YACpB,CAAC,OAAO,KAAK,0BAA0B,EAAE;AAAA,YACzC,CAAC,OAAO,KAAK,yBAAyB,OAAO,EAAE;AAAA,UACjD;AACA,gBAAM,OAAO,QAAQ,KAAK,UAAU;AAAA,QACtC;AAGA,cAAM,KAAK,IAAI,QAAQ,IAAI,iBAAiB,iBAAiB;AAC7D,aAAK,iBAAiB;AAEtB,eAAO,KAAK,OAAO,MAAM,OAAO;AAAA,MAClC;AAAA,MACA;AACE,eAAO,IAAI;AAAA,UACT;AAAA,UACA;AAAA,YACE,QAAQ;AAAA,UACV;AAAA,QACF;AAAA,IACJ;AAAA,EACF;AAAA,EAEA,eAAe;AACb,UAAM,aAAa,KAAK,IAAI,cAAc;AAC1C,QAAI,WAAW,WAAW,GAAG;AAC3B,aAAO;AAAA,IACT;AACA,WAAO,WAAW,CAAC;AAAA,EACrB;AAAA,EAEA,0BAA0B,IAA8B;AACtD,UAAM,eAAe,KAAK,yBAAyB,IAAI,EAAE;AACzD,QAAI,iBAAiB,QAAW;AAC9B,aAAO;AAAA,IACT;AACA,WAAO,KAAK,OAAO,cAAc,YAAY,KAAK;AAAA,EACpD;AAAA;AAAA,EAGA,MAAM,UAAU,YAAwB,OAAkB;AAGxD,QAAI,KAAK,mBAAmB,mBAAmB;AAC7C,YAAM,MAAM,IAAI;AAAA,QACd;AAAA,MACF;AACA,WAAK,YAAY,UAAU,GAAG;AAC9B;AAAA,IACF;AAEA,QAAI;AACJ,QAAI;AAEF,YAAM,OACJ,OAAO,UAAU,WAAW,QAAQ,IAAI,YAAY,EAAE,OAAO,KAAK;AACpE,gBAAU,qBAAqB,MAAM,KAAK,MAAM,IAAI,CAAC;AAAA,IACvD,SAAS,OAAO;AACd,WAAK,YAAY,UAAU,KAAc;AACzC;AAAA,IACF;AAGA,QAAI,MAAM,KAAK,2BAA2B,OAAO,GAAG;AAClD;AAAA,IACF;AAIA,QAAI,iBAAiB,OAAO,GAAG;AAC7B,WAAK,yBAAyB,IAAI,QAAQ,GAAG,SAAS,GAAG,WAAW,EAAE;AAAA,IACxE;AAEA,SAAK,YAAY,YAAY,OAAO;AAAA,EACtC;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,4BACZ,WACuB;AACvB,UAAM,YAAY,KAAK,IAAI;AAC3B,UAAM,UAAU;AAEhB,QAAI;AACF,aAAO,KAAK,IAAI,IAAI,YAAY,SAAS;AAEvC,cAAM,WAAW,MAAM,KAAK,IAAI,QAAQ;AAAA,UACtC,wBAAwB,SAAS;AAAA,QACnC;AACA,YAAI,UAAU;AAEZ,gBAAM,KAAK,IAAI,QAAQ,OAAO,eAAe,SAAS,EAAE;AACxD,gBAAM,KAAK,IAAI,QAAQ,OAAO,wBAAwB,SAAS,EAAE;AACjE,iBAAO;AAAA,QACT;AAGA,cAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAG,CAAC;AAAA,MACzD;AAEA,YAAM,IAAI,MAAM,+BAA+B;AAAA,IACjD,UAAE;AAEA,YAAM,KAAK,IAAI,QAAQ,OAAO,eAAe,SAAS,EAAE;AACxD,YAAM,KAAK,IAAI,QAAQ,OAAO,wBAAwB,SAAS,EAAE;AAAA,IACnE;AAAA,EACF;AAAA;AAAA;AAAA,EAIA,MAAc,2BACZ,SACkB;AAElB,QAAI,kBAAkB,OAAO,KAAK,QAAQ,QAAQ;AAChD,YAAM,YAAY,QAAQ,IAAI,SAAS;AACvC,UAAI,CAAC,aAAa,CAAC,UAAU,WAAW,SAAS,EAAG,QAAO;AAG3D,YAAM,iBAAiB,MAAM,KAAK,IAAI,QAAQ;AAAA,QAC5C,eAAe,SAAS;AAAA,MAC1B;AACA,UAAI,CAAC,eAAgB,QAAO;AAG5B,YAAM,KAAK,IAAI,QAAQ;AAAA,QACrB,wBAAwB,SAAS;AAAA,QACjC,QAAQ;AAAA,MACV;AACA,aAAO;AAAA,IACT;AAGA,QAAI,eAAe,OAAO,GAAG;AAC3B,YAAM,YAAY,QAAQ,IAAI,SAAS;AACvC,UAAI,CAAC,aAAa,CAAC,UAAU,WAAW,SAAS,EAAG,QAAO;AAG3D,YAAM,iBAAiB,MAAM,KAAK,IAAI,QAAQ;AAAA,QAC5C,eAAe,SAAS;AAAA,MAC1B;AACA,UAAI,CAAC,eAAgB,QAAO;AAG5B,YAAM,cAA4B;AAAA,QAChC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,OAAO,QAAQ,MAAM,WAAW;AAAA,QAClC;AAAA,MACF;AACA,YAAM,KAAK,IAAI,QAAQ;AAAA,QACrB,wBAAwB,SAAS;AAAA,QACjC;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA,EAIA,MAAM,gBACJ,YACA,aACuB;AACvB,QAAI,KAAK,YAAY,WAAW;AAG9B,YAAM,KAAK,YAAY;AAAA,IACzB;AAIA,QAAI,KAAK,mBAAmB,OAAO;AACjC,aAAO,IAAI,MAAM,8CAA8C;AAAA,IACjE;AAEA,QAAI;AACF,UAAI;AACJ,UAAI;AACF,wBAAgB,qBAAqB,MAAM,WAAW;AAAA,MACxD,SAAS,OAAO;AACd,aAAK,YAAY,UAAU,KAAc;AACzC,cAAM;AAAA,MACR;AAGA,UAAI,MAAM,KAAK,2BAA2B,aAAa,GAAG;AACxD,eAAO;AAAA,MACT;AAEA,WAAK,YAAY,YAAY,aAAa;AAC1C,aAAO;AAAA,IACT,SAAS,OAAO;AACd,cAAQ,MAAM,oCAAoC,KAAK;AACvD,WAAK,YAAY,UAAU,KAAc;AACzC,aAAO;AAAA,IACT;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,iBACJ,IACA,OACe;AACf,QAAI,KAAK,YAAY,WAAW;AAG9B,YAAM,KAAK,YAAY;AAAA,IACzB;AACA,WAAO,MAAM,KAAK,OAAO,iBAAiB,IAAI,KAAK;AAAA,EACrD;AAAA;AAAA,EAGA,MAAM,eAAe,IAAe,OAA+B;AACjE,QAAI,KAAK,YAAY,WAAW;AAG9B,YAAM,KAAK,YAAY;AAAA,IACzB;AACA,WAAO,MAAM,KAAK,OAAO,eAAe,IAAI,KAAK;AAAA,EACnD;AAAA,EAEA,MAAM,eACJ,IACA,MACA,QACA,UACe;AACf,QAAI,KAAK,YAAY,WAAW;AAG9B,YAAM,KAAK,YAAY;AAAA,IACzB;AACA,WAAO,MAAM,KAAK,OAAO,eAAe,IAAI,MAAM,QAAQ,QAAQ;AAAA,EACpE;AAAA,EAEA,OAAO,MACL,MACA;AAAA,IACE,UAAU;AAAA,IACV;AAAA,EACF,IAGI,CAAC,GACL;AACA,WAAO,UAAS,SAAS,MAAM,EAAE,SAAS,YAAY,CAAC;AAAA,EACzD;AAAA,EAEA,OAAO,SACL,MACA;AAAA,IACE,UAAU;AAAA,IACV;AAAA,EACF,IAGI,CAAC,GACL;AACA,QAAI,WAAW;AACf,QAAI,SAAS,KAAK;AAChB,iBAAW;AAAA,IACb;AACA,UAAM,cAAc,IAAI,WAAW,EAAE,SAAS,CAAC;AAC/C,UAAM,iBAAiB,IAAI,WAAW,EAAE,UAAU,GAAG,QAAQ,WAAW,CAAC;AAEzE,WAAO;AAAA,MACL,MAAM,MAEJ,SACA,KACA,KACmB;AAEnB,cAAM,eAAe,WAAW,SAAS,WAAW;AACpD,YAAI,aAAc,QAAO;AAEzB,cAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAC/B,cAAM,eAAe,IAAI,OAA2B;AAGpD,YAAI,gBAAgB,QAAQ,OAAO,iBAAiB,UAAU;AAC5D,kBAAQ;AAAA,YACN,uCAAuC,OAAO;AAAA,UAChD;AACA,iBAAO,IAAI,SAAS,mBAAmB,EAAE,QAAQ,IAAI,CAAC;AAAA,QACxD;AAGA,YAAI,CAAC,yBAAyB,YAAY,GAAG;AAC3C,iBAAO,IAAI,SAAS,mBAAmB,EAAE,QAAQ,IAAI,CAAC;AAAA,QACxD;AAEA,cAAM,YACJ;AAGF,YAAI,QAAQ,WAAW,SAAS,YAAY,KAAK,GAAG,GAAG;AAGrD,gBAAM,YACJ,IAAI,aAAa,IAAI,WAAW,KAChC,UAAU,YAAY,EAAE,SAAS;AAGnC,gBAAM,EAAE,UAAU,SAAS,IAAI,IAAI,gBAAgB;AACnD,gBAAM,SAAS,SAAS,UAAU;AAClC,gBAAM,UAAU,IAAI,YAAY;AAGhC,gBAAM,cAAc,IAAI,IAAI,QAAQ,GAAG;AACvC,sBAAY,WAAW,UAAU,GAAG,QAAQ,UAAU;AACtD,sBAAY,aAAa,IAAI,aAAa,SAAS;AACnD,gBAAM,yBACJ,YAAY,WAAW,YAAY,SAAS,YAAY;AAC1D,gBAAM,kBAAkB;AAAA,QAA0B,sBAAsB;AAAA;AAAA;AACxE,iBAAO,MAAM,QAAQ,OAAO,eAAe,CAAC;AAG5C,gBAAM,KAAK,UAAU,WAAW,OAAO,SAAS,EAAE;AAClD,gBAAM,SAAS,UAAU,IAAI,EAAE;AAG/B,cAAI;AACF,kBAAM,OAAO,MAAM,IAAI,KAAK;AAAA,UAC9B,SAAS,OAAO;AACd,oBAAQ,MAAM,kCAAkC,KAAK;AACrD,kBAAM,OAAO,MAAM;AACnB,kBAAM,eACJ,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AACvD,mBAAO,IAAI,SAAS,0BAA0B,YAAY,IAAI;AAAA,cAC5D,QAAQ;AAAA,YACV,CAAC;AAAA,UACH;AAGA,gBAAM,aAAa,IAAI,IAAI,QAAQ,GAAG;AAEtC,qBAAW,WAAW;AACtB,gBAAM,kBAA0C,CAAC;AACjD,kBAAQ,QAAQ,QAAQ,CAAC,OAAO,QAAQ;AACtC,4BAAgB,GAAG,IAAI;AAAA,UACzB,CAAC;AACD,gBAAM,WAAW,MAAM,OAAO;AAAA,YAC5B,IAAI,QAAQ,YAAY;AAAA,cACtB,SAAS;AAAA,gBACP,GAAG;AAAA,gBACH,SAAS;AAAA;AAAA,gBAET,mBAAmB;AAAA,cACrB;AAAA,YACF,CAAC;AAAA,UACH;AAGA,gBAAM,KAAK,SAAS;AACpB,cAAI,CAAC,IAAI;AACP,oBAAQ,MAAM,0CAA0C;AACxD,kBAAM,OAAO,MAAM;AACnB,mBAAO,IAAI,SAAS,4CAA4C;AAAA,cAC9D,QAAQ;AAAA,YACV,CAAC;AAAA,UACH;AAGA,aAAG,OAAO;AAGV,aAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,2BAAe,UAAUA,QAAqB;AAC5C,kBAAI;AACF,sBAAM,UAAU,KAAK,MAAMA,OAAM,IAAI;AAGrC,sBAAM,SAAS,qBAAqB,UAAU,OAAO;AACrD,oBAAI,CAAC,OAAO,SAAS;AAInB;AAAA,gBACF;AAGA,sBAAM,cAAc;AAAA,QAAyB,KAAK,UAAU,OAAO,IAAI,CAAC;AAAA;AAAA;AACxE,sBAAM,OAAO,MAAM,QAAQ,OAAO,WAAW,CAAC;AAAA,cAChD,SAAS,OAAO;AACd,wBAAQ,MAAM,oCAAoC,KAAK;AAAA,cACzD;AAAA,YACF;AACA,sBAAU,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,UACtC,CAAC;AAGD,aAAG,iBAAiB,SAAS,CAAC,UAAU;AACtC,2BAAe,QAAQ,QAAe;AACpC,kBAAI;AACF,sBAAM,OAAO,MAAM;AAAA,cACrB,SAAS,IAAI;AAAA,cAEb;AAAA,YACF;AACA,oBAAQ,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,UACpC,CAAC;AAGD,aAAG,iBAAiB,SAAS,MAAM;AACjC,2BAAe,UAAU;AACvB,kBAAI;AACF,sBAAM,OAAO,MAAM;AAAA,cACrB,SAAS,OAAO;AACd,wBAAQ,MAAM,iCAAiC,KAAK;AAAA,cACtD;AAAA,YACF;AACA,oBAAQ,EAAE,MAAM,QAAQ,KAAK;AAAA,UAC/B,CAAC;AAGD,iBAAO,IAAI,SAAS,UAAU;AAAA,YAC5B,SAAS;AAAA,cACP,iBAAiB;AAAA,cACjB,YAAY;AAAA,cACZ,gBAAgB;AAAA,cAChB,GAAG,YAAY,SAAS,WAAW;AAAA,YACrC;AAAA,UACF,CAAC;AAAA,QACH;AAKA,YAAI,QAAQ,WAAW,UAAU,eAAe,KAAK,GAAG,GAAG;AACzD,gBAAM,YAAY,IAAI,aAAa,IAAI,WAAW;AAClD,cAAI,CAAC,WAAW;AACd,mBAAO,IAAI;AAAA,cACT,uCAAuC,QAAQ;AAAA,cAC/C,EAAE,QAAQ,IAAI;AAAA,YAChB;AAAA,UACF;AAEA,gBAAM,cAAc,QAAQ,QAAQ,IAAI,cAAc,KAAK;AAC3D,cAAI,CAAC,YAAY,SAAS,kBAAkB,GAAG;AAC7C,mBAAO,IAAI,SAAS,6BAA6B,WAAW,IAAI;AAAA,cAC9D,QAAQ;AAAA,YACV,CAAC;AAAA,UACH;AAGA,gBAAM,gBAAgB,OAAO;AAAA,YAC3B,QAAQ,QAAQ,IAAI,gBAAgB,KAAK;AAAA,YACzC;AAAA,UACF;AACA,cAAI,gBAAgB,4BAA4B;AAC9C,mBAAO,IAAI;AAAA,cACT,2BAA2B,aAAa;AAAA,cACxC;AAAA,gBACE,QAAQ;AAAA,cACV;AAAA,YACF;AAAA,UACF;AAGA,gBAAM,KAAK,UAAU,WAAW,OAAO,SAAS,EAAE;AAClD,gBAAM,SAAS,UAAU,IAAI,EAAE;AAE/B,gBAAM,cAAc,MAAM,QAAQ,KAAK;AAEvC,gBAAM,OAAO,YAAY,IAAI,KAAK;AAClC,gBAAM,QAAQ,MAAM,OAAO,gBAAgB,WAAW,WAAW;AAEjE,cAAI,OAAO;AACT,mBAAO,IAAI,SAAS,MAAM,SAAS;AAAA,cACjC,SAAS;AAAA,gBACP,iBAAiB;AAAA,gBACjB,YAAY;AAAA,gBACZ,gBAAgB;AAAA,gBAChB,GAAG,YAAY,SAAS,WAAW;AAAA,cACrC;AAAA,cACA,QAAQ;AAAA,YACV,CAAC;AAAA,UACH;AAEA,iBAAO,IAAI,SAAS,YAAY;AAAA,YAC9B,SAAS;AAAA,cACP,iBAAiB;AAAA,cACjB,YAAY;AAAA,cACZ,gBAAgB;AAAA,cAChB,GAAG,YAAY,SAAS,WAAW;AAAA,YACrC;AAAA,YACA,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAEA,eAAO,IAAI,SAAS,aAAa,EAAE,QAAQ,IAAI,CAAC;AAAA,MAClD;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,MACL,MACA;AAAA,IACE,UAAU;AAAA,IACV;AAAA,EACF,IAAqD,CAAC,GACtD;AACA,QAAI,WAAW;AACf,QAAI,SAAS,KAAK;AAChB,iBAAW;AAAA,IACb;AACA,UAAM,cAAc,IAAI,WAAW,EAAE,SAAS,CAAC;AAE/C,WAAO;AAAA,MACL,MAAM,MAEJ,SACA,KACA,KACmB;AAEnB,cAAM,eAAe,WAAW,SAAS,WAAW;AACpD,YAAI,cAAc;AAChB,iBAAO;AAAA,QACT;AAEA,cAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAC/B,cAAM,eAAe,IAAI,OAA2B;AAGpD,YAAI,gBAAgB,QAAQ,OAAO,iBAAiB,UAAU;AAC5D,kBAAQ;AAAA,YACN,uCAAuC,OAAO;AAAA,UAChD;AACA,iBAAO,IAAI,SAAS,mBAAmB,EAAE,QAAQ,IAAI,CAAC;AAAA,QACxD;AAGA,YAAI,CAAC,yBAAyB,YAAY,GAAG;AAC3C,iBAAO,IAAI,SAAS,mBAAmB,EAAE,QAAQ,IAAI,CAAC;AAAA,QACxD;AAEA,cAAM,YACJ;AAEF,YAAI,QAAQ,WAAW,UAAU,YAAY,KAAK,GAAG,GAAG;AAEtD,gBAAM,eAAe,QAAQ,QAAQ,IAAI,QAAQ;AAEjD,cACE,CAAC,cAAc,SAAS,kBAAkB,KAC1C,CAAC,aAAa,SAAS,mBAAmB,GAC1C;AACA,kBAAMC,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SACE;AAAA,cACJ;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAEA,gBAAM,KAAK,QAAQ,QAAQ,IAAI,cAAc;AAC7C,cAAI,CAAC,MAAM,CAAC,GAAG,SAAS,kBAAkB,GAAG;AAC3C,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SACE;AAAA,cACJ;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAGA,gBAAM,gBAAgB,OAAO;AAAA,YAC3B,QAAQ,QAAQ,IAAI,gBAAgB,KAAK;AAAA,YACzC;AAAA,UACF;AACA,cAAI,gBAAgB,4BAA4B;AAC9C,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SAAS,2CAA2C,0BAA0B;AAAA,cAChF;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAEA,cAAI,YAAY,QAAQ,QAAQ,IAAI,gBAAgB;AACpD,cAAI;AAEJ,cAAI;AACF,yBAAa,MAAM,QAAQ,KAAK;AAAA,UAClC,SAAS,QAAQ;AACf,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SAAS;AAAA,cACX;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAGA,cAAI;AACJ,cAAI,MAAM,QAAQ,UAAU,GAAG;AAC7B,2BAAe;AAAA,UACjB,OAAO;AACL,2BAAe,CAAC,UAAU;AAAA,UAC5B;AAEA,cAAI,WAA6B,CAAC;AAGlC,qBAAW,OAAO,cAAc;AAC9B,gBAAI,CAAC,qBAAqB,UAAU,GAAG,EAAE,SAAS;AAChD,oBAAMA,QAAO,KAAK,UAAU;AAAA,gBAC1B,OAAO;AAAA,kBACL,MAAM;AAAA,kBACN,SAAS;AAAA,gBACX;AAAA,gBACA,IAAI;AAAA,gBACJ,SAAS;AAAA,cACX,CAAC;AACD,qBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,YAC3C;AAAA,UACF;AAEA,qBAAW,aAAa,IAAI,CAAC,QAAQ,qBAAqB,MAAM,GAAG,CAAC;AAKpE,gBAAM,0BAA0B,SAAS;AAAA,YACvC,CAAC,QAAQ,wBAAwB,UAAU,GAAG,EAAE;AAAA,UAClD;AAEA,cAAI,2BAA2B,WAAW;AACxC,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SACE;AAAA,cACJ;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAGA,cAAI,2BAA2B,SAAS,SAAS,GAAG;AAClD,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SACE;AAAA,cACJ;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAKA,cAAI,CAAC,2BAA2B,CAAC,WAAW;AAC1C,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SAAS;AAAA,cACX;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAIA,sBAAY,aAAa,UAAU,YAAY,EAAE,SAAS;AAG1D,gBAAM,KAAK,UAAU,WAAW,mBAAmB,SAAS,EAAE;AAC9D,gBAAM,SAAS,UAAU,IAAI,EAAE;AAC/B,gBAAM,gBAAgB,MAAM,OAAO,cAAc;AAEjD,cAAI,yBAAyB;AAC3B,gBAAI;AACF,oBAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,oBAAM,OAAO,eAAe;AAAA,YAC9B,SAAS,OAAO;AACd,sBAAQ,MAAM,kCAAkC,KAAK;AACrD,oBAAM,eACJ,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AACvD,oBAAMA,QAAO,KAAK,UAAU;AAAA,gBAC1B,OAAO;AAAA,kBACL,MAAM;AAAA,kBACN,SAAS,0BAA0B,YAAY;AAAA,gBACjD;AAAA,gBACA,IAAI;AAAA,gBACJ,SAAS;AAAA,cACX,CAAC;AACD,qBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,YAC3C;AAAA,UACF,WAAW,CAAC,eAAe;AAGzB,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SAAS;AAAA,cACX;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C,OAAO;AAEL,kBAAM,OAAO,YAAY,IAAI,KAAK;AAAA,UACpC;AAMA,gBAAM,EAAE,UAAU,SAAS,IAAI,IAAI,gBAAgB;AACnD,gBAAM,SAAS,SAAS,UAAU;AAClC,gBAAM,UAAU,IAAI,YAAY;AAGhC,gBAAM,aAAa,IAAI,IAAI,QAAQ,GAAG;AACtC,qBAAW,WAAW;AACtB,gBAAM,kBAA0C,CAAC;AACjD,kBAAQ,QAAQ,QAAQ,CAAC,OAAO,QAAQ;AACtC,4BAAgB,GAAG,IAAI;AAAA,UACzB,CAAC;AACD,gBAAM,WAAW,MAAM,OAAO;AAAA,YAC5B,IAAI,QAAQ,YAAY;AAAA,cACtB,SAAS;AAAA,gBACP,GAAG;AAAA,gBACH,SAAS;AAAA;AAAA,gBAET,mBAAmB;AAAA,cACrB;AAAA,YACF,CAAC;AAAA,UACH;AAGA,gBAAM,KAAK,SAAS;AACpB,cAAI,CAAC,IAAI;AACP,oBAAQ,MAAM,0CAA0C;AAExD,kBAAM,OAAO,MAAM;AACnB,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SAAS;AAAA,cACX;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAKA,gBAAM,aAAmC,oBAAI,IAAI;AAGjD,aAAG,OAAO;AAGV,aAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,2BAAe,UAAUD,QAAqB;AAC5C,kBAAI;AACF,sBAAM,OACJ,OAAOA,OAAM,SAAS,WAClBA,OAAM,OACN,IAAI,YAAY,EAAE,OAAOA,OAAM,IAAI;AACzC,sBAAM,UAAU,KAAK,MAAM,IAAI;AAG/B,sBAAM,SAAS,qBAAqB,UAAU,OAAO;AACrD,oBAAI,CAAC,OAAO,SAAS;AAInB;AAAA,gBACF;AAIA,oBACE,kBAAkB,OAAO,IAAI,KAC7B,eAAe,OAAO,IAAI,GAC1B;AACA,6BAAW,OAAO,OAAO,KAAK,EAAE;AAAA,gBAClC;AAGA,sBAAM,cAAc;AAAA,QAAyB,KAAK,UAAU,OAAO,IAAI,CAAC;AAAA;AAAA;AACxE,sBAAM,OAAO,MAAM,QAAQ,OAAO,WAAW,CAAC;AAG9C,oBAAI,WAAW,SAAS,GAAG;AACzB,qBAAI,MAAM;AAAA,gBACZ;AAAA,cACF,SAAS,OAAO;AACd,wBAAQ,MAAM,oCAAoC,KAAK;AAAA,cACzD;AAAA,YACF;AACA,sBAAU,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,UACtC,CAAC;AAGD,aAAG,iBAAiB,SAAS,CAAC,UAAU;AACtC,2BAAe,QAAQ,QAAe;AACpC,kBAAI;AACF,sBAAM,OAAO,MAAM;AAAA,cACrB,SAAS,IAAI;AAAA,cAEb;AAAA,YACF;AACA,oBAAQ,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,UACpC,CAAC;AAGD,aAAG,iBAAiB,SAAS,MAAM;AACjC,2BAAe,UAAU;AACvB,kBAAI;AACF,sBAAM,OAAO,MAAM;AAAA,cACrB,SAAS,OAAO;AACd,wBAAQ,MAAM,iCAAiC,KAAK;AAAA,cACtD;AAAA,YACF;AACA,oBAAQ,EAAE,MAAM,QAAQ,KAAK;AAAA,UAC/B,CAAC;AAID,gBAAM,kCAAkC,SAAS;AAAA,YAC/C,CAAC,QAAQ,sBAAsB,GAAG,KAAK,kBAAkB,GAAG;AAAA,UAC9D;AACA,cAAI,iCAAiC;AACnC,uBAAW,WAAW,UAAU;AAC9B,iBAAG,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,YACjC;AAGA,eAAG,MAAM;AAET,mBAAO,IAAI,SAAS,MAAM;AAAA,cACxB,SAAS,YAAY,SAAS,WAAW;AAAA,cACzC,QAAQ;AAAA,YACV,CAAC;AAAA,UACH;AAEA,qBAAW,WAAW,UAAU;AAC9B,gBAAI,iBAAiB,OAAO,GAAG;AAI7B,yBAAW,IAAI,QAAQ,EAAE;AAAA,YAC3B;AACA,eAAG,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,UACjC;AAIA,iBAAO,IAAI,SAAS,UAAU;AAAA,YAC5B,SAAS;AAAA,cACP,iBAAiB;AAAA,cACjB,YAAY;AAAA,cACZ,gBAAgB;AAAA,cAChB,kBAAkB;AAAA,cAClB,GAAG,YAAY,SAAS,WAAW;AAAA,YACrC;AAAA,YACA,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAGA,cAAM,OAAO,KAAK,UAAU;AAAA,UAC1B,OAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,UACX;AAAA,UACA,IAAI;AAAA,UACJ,SAAS;AAAA,QACX,CAAC;AACD,eAAO,IAAI,SAAS,MAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,MAC3C;AAAA,IACF;AAAA,EACF;AACF;","names":["event","body"]}
+{"version":3,"sources":["../../src/mcp/index.ts","../../src/mcp/utils.ts","../../src/mcp/transport.ts"],"sourcesContent":["import type { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport type { McpServer } from \"@modelcontextprotocol/sdk/server/mcp.js\";\nimport type { Transport } from \"@modelcontextprotocol/sdk/shared/transport.js\";\nimport type { JSONRPCMessage } from \"@modelcontextprotocol/sdk/types.js\";\nimport {\n  JSONRPCMessageSchema,\n  isJSONRPCError,\n  isJSONRPCNotification,\n  isJSONRPCRequest,\n  isJSONRPCResponse,\n  type ElicitResult\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type {\n  AgentContext,\n  Connection,\n  ConnectionContext,\n  WSMessage\n} from \"../\";\nimport { Agent } from \"../index\";\nimport type {\n  MaybeConnectionTag,\n  MaybePromise,\n  ServeOptions,\n  TransportType\n} from \"./types\";\nimport {\n  createLegacySseHandler,\n  createStreamingHttpHandler,\n  handleCORS,\n  isDurableObjectNamespace,\n  STANDALONE_SSE_MARKER,\n  STANDALONE_SSE_METHOD\n} from \"./utils\";\nimport { McpSSETransport, McpStreamableHttpTransport } from \"./transport\";\n\nexport abstract class McpAgent<\n  Env = unknown,\n  State = unknown,\n  Props extends Record<string, unknown> = Record<string, unknown>\n> extends Agent<Env, State, Props> {\n  private _transport?: Transport;\n  private _requestIdToConnectionId: Map<string | number, string> = new Map();\n  // The connection ID for server-sent requests/notifications\n  private _standaloneSseConnectionId?: string;\n  props?: Props;\n\n  abstract server: MaybePromise<McpServer | Server>;\n  abstract init(): Promise<void>;\n\n  constructor(ctx: AgentContext, env: Env) {\n    super(ctx, env);\n\n    // Re-set the standalone SSE connection ID if\n    // coming out of hibernation\n    for (const connection of this.getConnections<MaybeConnectionTag>()) {\n      const meta = connection.state;\n      if (meta?.role === STANDALONE_SSE_MARKER) {\n        this._standaloneSseConnectionId = connection.id;\n        return;\n      }\n    }\n  }\n\n  /*\n   * Helpers\n   */\n\n  async setInitialized() {\n    // TODO: move to sync api once https://github.com/cloudflare/workerd/pull/4895 lands\n    await this.ctx.storage.put(\"initialized\", true);\n  }\n\n  async isInitialized() {\n    return (await this.ctx.storage.get(\"initialized\")) === true;\n  }\n\n  /** Read the transport type for this agent.\n   * This relies on the naming scheme being `sse:${sessionId}`\n   * or `streamable-http:${sessionId}`.\n   */\n  getTransportType(): TransportType {\n    const [t, ..._] = this.name.split(\":\");\n    switch (t) {\n      case \"sse\":\n        return \"sse\";\n      case \"streamable-http\":\n        return \"streamable-http\";\n      default:\n        throw new Error(\n          \"Invalid transport type. McpAgent must be addressed with a valid protocol.\"\n        );\n    }\n  }\n\n  /** Get the WebSocket for the standalone SSE if any. Streamable HTTP only. */\n  private getWebSocketForStandaloneSse(): WebSocket | null {\n    if (!this._standaloneSseConnectionId) return null;\n    return this.getConnection(this._standaloneSseConnectionId) ?? null;\n  }\n\n  /** Get the unique WebSocket. SSE transport only. */\n  private getWebSocket() {\n    const websockets = Array.from(this.getConnections());\n    if (websockets.length === 0) {\n      return null;\n    }\n    return websockets[0];\n  }\n\n  /** Get the corresponding WebSocket for a responseId. Streamable HTTP only. */\n  private getWebSocketForResponseID(id: string): WebSocket | null {\n    const connectionId = this._requestIdToConnectionId.get(id);\n    if (connectionId === undefined) {\n      return null;\n    }\n    return this.getConnection(connectionId) ?? null;\n  }\n\n  /** Returns a new transport matching the type of the Agent. */\n  private initTransport() {\n    switch (this.getTransportType()) {\n      case \"sse\": {\n        return new McpSSETransport(() => this.getWebSocket());\n      }\n      case \"streamable-http\": {\n        return new McpStreamableHttpTransport(\n          (id) => this.getWebSocketForResponseID(id),\n          (id) => this._requestIdToConnectionId.delete(id),\n          () => this.getWebSocketForStandaloneSse()\n        );\n      }\n    }\n  }\n\n  /** Update and store the props */\n  async updateProps(props?: Props) {\n    await this.ctx.storage.put(\"props\", props ?? {});\n    this.props = props;\n  }\n\n  /*\n   * Base Agent / Parykit Server overrides\n   */\n\n  /** Sets up the MCP transport and server every time the Agent is started.*/\n  async onStart(props?: Props) {\n    // If onStart was passed props, save them in storage\n    if (props) await this.updateProps(props);\n    this.props = await this.ctx.storage.get(\"props\");\n\n    await this.init();\n    const server = await this.server;\n    // Connect to the MCP server\n    this._transport = this.initTransport();\n    await server.connect(this._transport);\n  }\n\n  /** Validates new WebSocket connections. */\n  async onConnect(conn: Connection, _: ConnectionContext): Promise<void> {\n    switch (this.getTransportType()) {\n      case \"sse\": {\n        // For SSE connections, we can only have one open connection per session\n        // If we get an upgrade while already connected, we should error\n        const websockets = Array.from(this.getConnections());\n        if (websockets.length > 1) {\n          conn.close(1008, \"Websocket already connected\");\n          return;\n        }\n        break;\n      }\n      case \"streamable-http\":\n        break;\n    }\n  }\n\n  /** Handles MCP Messages for Streamable HTTP. */\n  async onMessage(connection: Connection, event: WSMessage) {\n    // Since we address the DO via both the protocol and the session id,\n    // this should never happen, but let's enforce it just in case\n    if (this.getTransportType() !== \"streamable-http\") {\n      const err = new Error(\n        \"Internal Server Error: Expected streamable-http protocol\"\n      );\n      this._transport?.onerror?.(err);\n      return;\n    }\n\n    let message: JSONRPCMessage;\n    try {\n      // Ensure event is a string\n      const data =\n        typeof event === \"string\" ? event : new TextDecoder().decode(event);\n      message = JSONRPCMessageSchema.parse(JSON.parse(data));\n    } catch (error) {\n      this._transport?.onerror?.(error as Error);\n      return;\n    }\n\n    // Check if message is our control frame for the standalone SSE stream.\n    if (\n      isJSONRPCNotification(message) &&\n      message.method === STANDALONE_SSE_METHOD\n    ) {\n      if (\n        this._standaloneSseConnectionId &&\n        this._standaloneSseConnectionId !== connection.id\n      ) {\n        // If the standalone SSE was already set, we close the old\n        // socket to avoid dangling connections.\n        const standaloneSseSocket = this.getConnection(\n          this._standaloneSseConnectionId\n        );\n        standaloneSseSocket?.close(1000, \"replaced\");\n      }\n      connection.setState({\n        role: STANDALONE_SSE_MARKER\n      });\n\n      this._standaloneSseConnectionId = connection.id;\n      // This is internal, so we don't forward the message to the server.\n      return;\n    }\n\n    // Check if this is an elicitation response before passing to transport\n    if (await this._handleElicitationResponse(message)) {\n      return; // Message was handled by elicitation system\n    }\n\n    // We need to map every incoming message to the connection that it came in on\n    // so that we can send relevant responses and notifications back on the same connection\n    if (isJSONRPCRequest(message)) {\n      this._requestIdToConnectionId.set(message.id.toString(), connection.id);\n    }\n\n    this._transport?.onmessage?.(message);\n  }\n\n  /** Remove clients from our cache when they disconnect */\n  async onClose(\n    conn: Connection,\n    _code: number,\n    _reason: string,\n    _wasClean: boolean\n  ): Promise<void> {\n    // Remove the connection/socket mapping for the socket that just closed\n    for (const [reqId, connId] of this._requestIdToConnectionId) {\n      if (connId === conn.id) this._requestIdToConnectionId.delete(reqId);\n    }\n\n    // Clear the standalone SSE if it just closed\n    if (this._standaloneSseConnectionId === conn.id) {\n      this._standaloneSseConnectionId = undefined;\n    }\n  }\n\n  /*\n   * Transport ingress and routing\n   */\n\n  /** Handles MCP Messages for the legacy SSE transport. */\n  async onSSEMcpMessage(\n    _sessionId: string,\n    messageBody: unknown\n  ): Promise<Error | null> {\n    // Since we address the DO via both the protocol and the session id,\n    // this should never happen, but let's enforce it just in case\n    if (this.getTransportType() !== \"sse\") {\n      return new Error(\"Internal Server Error: Expected SSE transport\");\n    }\n\n    try {\n      let parsedMessage: JSONRPCMessage;\n      try {\n        parsedMessage = JSONRPCMessageSchema.parse(messageBody);\n      } catch (error) {\n        this._transport?.onerror?.(error as Error);\n        throw error;\n      }\n\n      // Check if this is an elicitation response before passing to transport\n      if (await this._handleElicitationResponse(parsedMessage)) {\n        return null; // Message was handled by elicitation system\n      }\n\n      this._transport?.onmessage?.(parsedMessage);\n      return null;\n    } catch (error) {\n      console.error(\"Error forwarding message to SSE:\", error);\n      this._transport?.onerror?.(error as Error);\n      return error as Error;\n    }\n  }\n\n  /** Elicit user input with a message and schema */\n  async elicitInput(params: {\n    message: string;\n    requestedSchema: unknown;\n  }): Promise<ElicitResult> {\n    const requestId = `elicit_${Math.random().toString(36).substring(2, 11)}`;\n\n    // Store pending request in durable storage\n    await this.ctx.storage.put(`elicitation:${requestId}`, {\n      message: params.message,\n      requestedSchema: params.requestedSchema,\n      timestamp: Date.now()\n    });\n\n    const elicitRequest = {\n      jsonrpc: \"2.0\" as const,\n      id: requestId,\n      method: \"elicitation/create\",\n      params: {\n        message: params.message,\n        requestedSchema: params.requestedSchema\n      }\n    };\n\n    // Send through MCP transport\n    if (this._transport) {\n      await this._transport.send(elicitRequest);\n    } else {\n      const connections = this.getConnections();\n      if (!connections || Array.from(connections).length === 0) {\n        await this.ctx.storage.delete(`elicitation:${requestId}`);\n        throw new Error(\"No active connections available for elicitation\");\n      }\n\n      const connectionList = Array.from(connections);\n      for (const connection of connectionList) {\n        try {\n          connection.send(JSON.stringify(elicitRequest));\n        } catch (error) {\n          console.error(\"Failed to send elicitation request:\", error);\n        }\n      }\n    }\n\n    // Wait for response through MCP\n    return this._waitForElicitationResponse(requestId);\n  }\n\n  /** Wait for elicitation response through storage polling */\n  private async _waitForElicitationResponse(\n    requestId: string\n  ): Promise<ElicitResult> {\n    const startTime = Date.now();\n    const timeout = 60000; // 60 second timeout\n\n    try {\n      while (Date.now() - startTime < timeout) {\n        // Check if response has been stored\n        const response = await this.ctx.storage.get<ElicitResult>(\n          `elicitation:response:${requestId}`\n        );\n        if (response) {\n          // Immediately clean up both request and response\n          await this.ctx.storage.delete(`elicitation:${requestId}`);\n          await this.ctx.storage.delete(`elicitation:response:${requestId}`);\n          return response;\n        }\n\n        // Sleep briefly before checking again\n        await new Promise((resolve) => setTimeout(resolve, 100));\n      }\n\n      throw new Error(\"Elicitation request timed out\");\n    } finally {\n      // Always clean up on timeout or error\n      await this.ctx.storage.delete(`elicitation:${requestId}`);\n      await this.ctx.storage.delete(`elicitation:response:${requestId}`);\n    }\n  }\n\n  /** Handle elicitation responses */\n  private async _handleElicitationResponse(\n    message: JSONRPCMessage\n  ): Promise<boolean> {\n    // Check if this is a response to an elicitation request\n    if (isJSONRPCResponse(message) && message.result) {\n      const requestId = message.id?.toString();\n      if (!requestId || !requestId.startsWith(\"elicit_\")) return false;\n\n      // Check if we have a pending request for this ID\n      const pendingRequest = await this.ctx.storage.get(\n        `elicitation:${requestId}`\n      );\n      if (!pendingRequest) return false;\n\n      // Store the response in durable storage\n      await this.ctx.storage.put(\n        `elicitation:response:${requestId}`,\n        message.result as ElicitResult\n      );\n      return true;\n    }\n\n    // Check if this is an error response to an elicitation request\n    if (isJSONRPCError(message)) {\n      const requestId = message.id?.toString();\n      if (!requestId || !requestId.startsWith(\"elicit_\")) return false;\n\n      // Check if we have a pending request for this ID\n      const pendingRequest = await this.ctx.storage.get(\n        `elicitation:${requestId}`\n      );\n      if (!pendingRequest) return false;\n\n      // Store error response\n      const errorResult: ElicitResult = {\n        action: \"cancel\",\n        content: {\n          error: message.error.message || \"Elicitation request failed\"\n        }\n      };\n      await this.ctx.storage.put(\n        `elicitation:response:${requestId}`,\n        errorResult\n      );\n      return true;\n    }\n\n    return false;\n  }\n\n  /** Return a handler for the given path for this MCP.\n   * Defaults to Streamable HTTP transport.\n   */\n  static serve(\n    path: string,\n    {\n      binding = \"MCP_OBJECT\",\n      corsOptions,\n      transport = \"streamable-http\"\n    }: ServeOptions = {}\n  ) {\n    return {\n      async fetch<Env>(\n        this: void,\n        request: Request,\n        env: Env,\n        ctx: ExecutionContext\n      ): Promise<Response> {\n        // Handle CORS preflight\n        const corsResponse = handleCORS(request, corsOptions);\n        if (corsResponse) {\n          return corsResponse;\n        }\n\n        const bindingValue = env[binding as keyof typeof env] as unknown;\n\n        // Ensure we have a binding of some sort\n        if (bindingValue == null || typeof bindingValue !== \"object\") {\n          throw new Error(\n            `Could not find McpAgent binding for ${binding}. Did you update your wrangler configuration?`\n          );\n        }\n\n        // Ensure that the binding is to a DurableObject\n        if (!isDurableObjectNamespace(bindingValue)) {\n          throw new Error(\n            `Invalid McpAgent binding for ${binding}. Make sure it's a Durable Object binding.`\n          );\n        }\n\n        const namespace =\n          bindingValue satisfies DurableObjectNamespace<McpAgent>;\n\n        switch (transport) {\n          case \"streamable-http\": {\n            // Streamable HTTP transport handling\n            const handleStreamableHttp = createStreamingHttpHandler(\n              path,\n              namespace,\n              corsOptions\n            );\n            return handleStreamableHttp(request, ctx);\n          }\n          case \"sse\": {\n            // Legacy SSE transport handling\n            const handleLegacySse = createLegacySseHandler(\n              path,\n              namespace,\n              corsOptions\n            );\n            return handleLegacySse(request, ctx);\n          }\n          default:\n            return new Response(\n              \"Invalid MCP transport mode. Only `streamable-http` or `sse` are allowed.\",\n              { status: 500 }\n            );\n        }\n      }\n    };\n  }\n  /**\n   * Legacy api\n   **/\n  static mount(path: string, opts: Omit<ServeOptions, \"transport\"> = {}) {\n    return McpAgent.serveSSE(path, opts);\n  }\n\n  static serveSSE(path: string, opts: Omit<ServeOptions, \"transport\"> = {}) {\n    return McpAgent.serve(path, { ...opts, transport: \"sse\" });\n  }\n}\n\n// Export client transport classes\nexport { SSEEdgeClientTransport } from \"./sse-edge\";\nexport { StreamableHTTPEdgeClientTransport } from \"./streamable-http-edge\";\n\n// Export elicitation types and schemas\nexport {\n  ElicitRequestSchema,\n  type ElicitRequest,\n  type ElicitResult\n} from \"@modelcontextprotocol/sdk/types.js\";\n","import {\n  JSONRPCMessageSchema,\n  InitializeRequestSchema,\n  isJSONRPCResponse,\n  isJSONRPCError,\n  isJSONRPCNotification,\n  isJSONRPCRequest\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type { JSONRPCMessage } from \"ai\";\nimport type { McpAgent } from \".\";\nimport { getAgentByName } from \"..\";\nimport type { CORSOptions } from \"./types\";\n\n/**\n * Tag placed in `connection.state.role` to mark the WebSocket that carries\n * the standalone SSE stream for a session. We use it to recover the connection\n * after hibernation.\n */\nexport const STANDALONE_SSE_MARKER = \"standalone-sse\";\n\n/**\n * Internal JSON-RPC notif method to attach current a\n * connection as the standalone SSE.\n */\nexport const STANDALONE_SSE_METHOD = \"cf/standalone_sse/attach\";\nconst MAXIMUM_MESSAGE_SIZE_BYTES = 4 * 1024 * 1024; // 4MB\n\nexport const createStreamingHttpHandler = (\n  basePath: string,\n  namespace: DurableObjectNamespace<McpAgent>,\n  corsOptions?: CORSOptions\n) => {\n  let pathname = basePath;\n  if (basePath === \"/\") pathname = \"/*\";\n\n  const basePattern = new URLPattern({ pathname });\n  return async (request: Request, ctx: ExecutionContext) => {\n    const url = new URL(request.url);\n    if (basePattern.test(url)) {\n      if (request.method === \"POST\") {\n        // Validate the Accept header\n        const acceptHeader = request.headers.get(\"accept\");\n        // The client MUST include an Accept header, listing both application/json and text/event-stream as supported content types.\n        if (\n          !acceptHeader?.includes(\"application/json\") ||\n          !acceptHeader.includes(\"text/event-stream\")\n        ) {\n          const body = JSON.stringify({\n            error: {\n              code: -32000,\n              message:\n                \"Not Acceptable: Client must accept both application/json and text/event-stream\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 406 });\n        }\n\n        const ct = request.headers.get(\"content-type\");\n        if (!ct || !ct.includes(\"application/json\")) {\n          const body = JSON.stringify({\n            error: {\n              code: -32000,\n              message:\n                \"Unsupported Media Type: Content-Type must be application/json\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 415 });\n        }\n\n        // Check content length against maximum allowed size\n        const contentLength = Number.parseInt(\n          request.headers.get(\"content-length\") ?? \"0\",\n          10\n        );\n        if (contentLength > MAXIMUM_MESSAGE_SIZE_BYTES) {\n          const body = JSON.stringify({\n            error: {\n              code: -32000,\n              message: `Request body too large. Maximum size is ${MAXIMUM_MESSAGE_SIZE_BYTES} bytes`\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 413 });\n        }\n\n        let sessionId = request.headers.get(\"mcp-session-id\");\n        let rawMessage: unknown;\n\n        try {\n          rawMessage = await request.json();\n        } catch (_error) {\n          const body = JSON.stringify({\n            error: {\n              code: -32700,\n              message: \"Parse error: Invalid JSON\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 400 });\n        }\n\n        // Make sure the message is an array to simplify logic\n        let arrayMessage: unknown[];\n        if (Array.isArray(rawMessage)) {\n          arrayMessage = rawMessage;\n        } else {\n          arrayMessage = [rawMessage];\n        }\n\n        let messages: JSONRPCMessage[] = [];\n\n        // Try to parse each message as JSON RPC. Fail if any message is invalid\n        for (const msg of arrayMessage) {\n          if (!JSONRPCMessageSchema.safeParse(msg).success) {\n            const body = JSON.stringify({\n              error: {\n                code: -32700,\n                message: \"Parse error: Invalid JSON-RPC message\"\n              },\n              id: null,\n              jsonrpc: \"2.0\"\n            });\n            return new Response(body, { status: 400 });\n          }\n        }\n\n        messages = arrayMessage.map((msg) => JSONRPCMessageSchema.parse(msg));\n\n        // Before we pass the messages to the agent, there's another error condition we need to enforce\n        // Check if this is an initialization request\n        // https://spec.modelcontextprotocol.io/specification/2025-03-26/basic/lifecycle/\n        const isInitializationRequest = messages.some(\n          (msg) => InitializeRequestSchema.safeParse(msg).success\n        );\n\n        if (isInitializationRequest && sessionId) {\n          const body = JSON.stringify({\n            error: {\n              code: -32600,\n              message:\n                \"Invalid Request: Initialization requests must not include a sessionId\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 400 });\n        }\n\n        // The initialization request must be the only request in the batch\n        if (isInitializationRequest && messages.length > 1) {\n          const body = JSON.stringify({\n            error: {\n              code: -32600,\n              message:\n                \"Invalid Request: Only one initialization request is allowed\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 400 });\n        }\n\n        // If an Mcp-Session-Id is returned by the server during initialization,\n        // clients using the Streamable HTTP transport MUST include it\n        // in the Mcp-Session-Id header on all of their subsequent HTTP requests.\n        if (!isInitializationRequest && !sessionId) {\n          const body = JSON.stringify({\n            error: {\n              code: -32000,\n              message: \"Bad Request: Mcp-Session-Id header is required\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 400 });\n        }\n\n        // If we don't have a sessionId, we are serving an initialization request\n        // and need to generate a new sessionId\n        sessionId = sessionId ?? namespace.newUniqueId().toString();\n\n        // Get the agent and set props\n        const agent = await getAgentByName(\n          namespace,\n          `streamable-http:${sessionId}`,\n          { props: ctx.props }\n        );\n        const isInitialized = await agent.isInitialized();\n\n        if (isInitializationRequest) {\n          await agent.setInitialized();\n        } else if (!isInitialized) {\n          // if we have gotten here, then a session id that was never initialized\n          // was provided\n          const body = JSON.stringify({\n            error: {\n              code: -32001,\n              message: \"Session not found\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 404 });\n        }\n\n        // We've evaluated all the error conditions! Now it's time to establish\n        // all the streams\n\n        // Create a Transform Stream for SSE\n        const { readable, writable } = new TransformStream();\n        const writer = writable.getWriter();\n        const encoder = new TextEncoder();\n\n        // Connect to the Durable Object via WebSocket\n        const existingHeaders: Record<string, string> = {};\n        request.headers.forEach((value, key) => {\n          existingHeaders[key] = value;\n        });\n        const req = new Request(request.url, {\n          headers: {\n            ...existingHeaders,\n            Upgrade: \"websocket\"\n          }\n        });\n        if (ctx.props) agent.updateProps(ctx.props);\n        const response = await agent.fetch(req);\n\n        // Get the WebSocket\n        const ws = response.webSocket;\n        if (!ws) {\n          console.error(\"Failed to establish WebSocket connection\");\n\n          await writer.close();\n          const body = JSON.stringify({\n            error: {\n              code: -32001,\n              message: \"Failed to establish WebSocket connection\"\n            },\n            id: null,\n            jsonrpc: \"2.0\"\n          });\n          return new Response(body, { status: 500 });\n        }\n\n        // Keep track of the request ids that we have sent to the server\n        // so that we can close the connection once we have received\n        // all the responses\n        const requestIds: Set<string | number> = new Set();\n\n        // Accept the WebSocket\n        ws.accept();\n\n        // Handle messages from the Durable Object\n        ws.addEventListener(\"message\", (event) => {\n          async function onMessage(event: MessageEvent) {\n            try {\n              const data =\n                typeof event.data === \"string\"\n                  ? event.data\n                  : new TextDecoder().decode(event.data);\n              const message = JSON.parse(data);\n\n              // validate that the message is a valid JSONRPC message\n              const result = JSONRPCMessageSchema.safeParse(message);\n              if (!result.success) {\n                // The message was not a valid JSONRPC message, so we will drop it\n                // PartyKit will broadcast state change messages to all connected clients\n                // and we need to filter those out so they are not passed to MCP clients\n                return;\n              }\n\n              // If the message is a response or an error, remove the id from the set of\n              // request ids\n              if (\n                isJSONRPCResponse(result.data) ||\n                isJSONRPCError(result.data)\n              ) {\n                requestIds.delete(result.data.id);\n              }\n\n              // Send the message as an SSE event\n              const messageText = `event: message\\ndata: ${JSON.stringify(result.data)}\\n\\n`;\n              await writer.write(encoder.encode(messageText));\n\n              // If we have received all the responses, close the connection\n              if (requestIds.size === 0) {\n                ws?.close();\n                await writer.close();\n              }\n            } catch (error) {\n              console.error(\"Error forwarding message to SSE:\", error);\n            }\n          }\n          onMessage(event).catch(console.error);\n        });\n\n        // Handle WebSocket errors\n        ws.addEventListener(\"error\", (error) => {\n          async function onError(_error: Event) {\n            try {\n              await writer.close();\n            } catch (_e) {\n              // Ignore errors when closing\n            }\n          }\n          onError(error).catch(console.error);\n        });\n\n        // Handle WebSocket closure\n        ws.addEventListener(\"close\", () => {\n          async function onClose() {\n            try {\n              await writer.close();\n            } catch (error) {\n              console.error(\"Error closing SSE connection:\", error);\n            }\n          }\n          onClose().catch(console.error);\n        });\n\n        // If there are no requests, we send the messages to the agent and acknowledge the request with a 202\n        // since we don't expect any responses back through this connection\n        const hasOnlyNotificationsOrResponses = messages.every(\n          (msg) => isJSONRPCNotification(msg) || isJSONRPCResponse(msg)\n        );\n        if (hasOnlyNotificationsOrResponses) {\n          for (const message of messages) {\n            ws.send(JSON.stringify(message));\n          }\n\n          // closing the websocket will also close the SSE connection\n          ws.close();\n\n          return new Response(null, {\n            headers: corsHeaders(request, corsOptions),\n            status: 202\n          });\n        }\n\n        for (const message of messages) {\n          if (isJSONRPCRequest(message)) {\n            // add each request id that we send off to a set\n            // so that we can keep track of which requests we\n            // still need a response for\n            requestIds.add(message.id);\n          }\n          ws.send(JSON.stringify(message));\n        }\n\n        // Return the SSE response. We handle closing the stream in the ws \"message\"\n        // handler\n        return new Response(readable, {\n          headers: {\n            \"Cache-Control\": \"no-cache\",\n            Connection: \"keep-alive\",\n            \"Content-Type\": \"text/event-stream\",\n            \"mcp-session-id\": sessionId,\n            ...corsHeaders(request, corsOptions)\n          },\n          status: 200\n        });\n      } else if (request.method === \"GET\") {\n        // Validate the Accept header\n        const acceptHeader = request.headers.get(\"accept\");\n        // The client MUST include an Accept header, listing both application/json and text/event-stream as supported content types.\n        if (!acceptHeader?.includes(\"text/event-stream\")) {\n          const body = JSON.stringify({\n            jsonrpc: \"2.0\",\n            error: {\n              code: -32000,\n              message: \"Not Acceptable: Client must accept text/event-stream\"\n            },\n            id: null\n          });\n          return new Response(body, { status: 406 });\n        }\n\n        // Require sessionId\n        const sessionId = url.searchParams.get(\"sessionId\");\n        if (!sessionId)\n          return new Response(\"Missing sessionId\", { status: 400 });\n\n        // Create SSE stream\n        const { readable, writable } = new TransformStream();\n        const writer = writable.getWriter();\n        const encoder = new TextEncoder();\n\n        const agent = await getAgentByName(\n          namespace,\n          `streamable-http:${sessionId}`,\n          { props: ctx.props }\n        );\n        const isInitialized = await agent.isInitialized();\n        if (!isInitialized) {\n          return new Response(\n            JSON.stringify({\n              jsonrpc: \"2.0\",\n              error: { code: -32001, message: \"Session not found\" },\n              id: null\n            }),\n            { status: 404 }\n          );\n        }\n\n        const existingHeaders: Record<string, string> = {};\n        request.headers.forEach((v, k) => {\n          existingHeaders[k] = v;\n        });\n\n        if (ctx.props) agent.updateProps(ctx.props);\n        const response = await agent.fetch(\n          new Request(request.url, {\n            headers: {\n              ...existingHeaders,\n              Upgrade: \"websocket\"\n            }\n          })\n        );\n\n        const ws = response.webSocket;\n        if (!ws) {\n          await writer.close();\n          return new Response(\"Failed to establish WS to DO\", {\n            status: 500\n          });\n        }\n        ws.accept();\n\n        // Signal the DO to use this connection as the standalone SSE stream\n        ws.send(\n          JSON.stringify({\n            jsonrpc: \"2.0\",\n            method: STANDALONE_SSE_METHOD,\n            params: {}\n          })\n        );\n\n        // Forward DO messages as SSE\n        ws.addEventListener(\"message\", (event) => {\n          try {\n            async function onMessage(ev: MessageEvent) {\n              const data =\n                typeof ev.data === \"string\"\n                  ? ev.data\n                  : new TextDecoder().decode(ev.data);\n              const parsed = JSONRPCMessageSchema.safeParse(JSON.parse(data));\n              if (!parsed.success) return;\n              const frame = `event: message\\ndata: ${JSON.stringify(parsed.data)}\\n\\n`;\n              await writer.write(encoder.encode(frame));\n            }\n            onMessage(event).catch(console.error);\n          } catch (e) {\n            console.error(\"Error forwarding message to SSE:\", e);\n          }\n        });\n\n        ws.addEventListener(\"error\", () => {\n          writer.close().catch(() => {});\n        });\n        ws.addEventListener(\"close\", () => {\n          writer.close().catch(() => {});\n        });\n\n        return new Response(readable, {\n          headers: {\n            \"Cache-Control\": \"no-cache\",\n            Connection: \"keep-alive\",\n            \"Content-Type\": \"text/event-stream\",\n            \"mcp-session-id\": sessionId,\n            ...corsHeaders(request, corsOptions)\n          },\n          status: 200\n        });\n      } else if (request.method === \"DELETE\") {\n        const sessionId = request.headers.get(\"mcp-session-id\");\n        if (!sessionId) {\n          return new Response(\n            JSON.stringify({\n              jsonrpc: \"2.0\",\n              error: {\n                code: -32000,\n                message: \"Bad Request: Mcp-Session-Id header is required\"\n              },\n              id: null\n            }),\n            { status: 400, headers: corsHeaders(request, corsOptions) }\n          );\n        }\n        const agent = await getAgentByName(\n          namespace,\n          `streamable-http:${sessionId}`\n        );\n        const isInitialized = await agent.isInitialized();\n        if (!isInitialized) {\n          return new Response(\n            JSON.stringify({\n              jsonrpc: \"2.0\",\n              error: { code: -32001, message: \"Session not found\" },\n              id: null\n            }),\n            { status: 404, headers: corsHeaders(request, corsOptions) }\n          );\n        }\n        // .destroy() passes an uncatchable Error, so we make sure we first return\n        // the response to the client.\n        ctx.waitUntil(\n          agent.destroy().catch(() => {\n            /* This will always throw. We silently catch here */\n          })\n        );\n        return new Response(null, {\n          status: 204,\n          headers: corsHeaders(request, corsOptions)\n        });\n      }\n    }\n\n    // Route not found\n    const body = JSON.stringify({\n      error: {\n        code: -32000,\n        message: \"Not found\"\n      },\n      id: null,\n      jsonrpc: \"2.0\"\n    });\n    return new Response(body, { status: 404 });\n  };\n};\n\nexport const createLegacySseHandler = (\n  basePath: string,\n  namespace: DurableObjectNamespace<McpAgent>,\n  corsOptions?: CORSOptions\n) => {\n  let pathname = basePath;\n  if (basePath === \"/\") pathname = \"/*\";\n\n  const basePattern = new URLPattern({ pathname });\n  const messagePattern = new URLPattern({ pathname: `${basePath}/message` }); // SSE only\n  return async (request: Request, ctx: ExecutionContext) => {\n    const url = new URL(request.url);\n    // Handle initial SSE connection\n    if (request.method === \"GET\" && basePattern.test(url)) {\n      // Use a session ID if one is passed in, or create a unique\n      // session ID for this connection\n      const sessionId =\n        url.searchParams.get(\"sessionId\") || namespace.newUniqueId().toString();\n\n      // Create a Transform Stream for SSE\n      const { readable, writable } = new TransformStream();\n      const writer = writable.getWriter();\n      const encoder = new TextEncoder();\n\n      // Send the endpoint event\n      const endpointUrl = new URL(request.url);\n      endpointUrl.pathname = encodeURI(`${basePath}/message`);\n      endpointUrl.searchParams.set(\"sessionId\", sessionId);\n      const relativeUrlWithSession =\n        endpointUrl.pathname + endpointUrl.search + endpointUrl.hash;\n      const endpointMessage = `event: endpoint\\ndata: ${relativeUrlWithSession}\\n\\n`;\n      writer.write(encoder.encode(endpointMessage));\n\n      // Get the Durable Object\n      const agent = await getAgentByName(namespace, `sse:${sessionId}`, {\n        props: ctx.props\n      });\n\n      // Connect to the Durable Object via WebSocket\n      const existingHeaders: Record<string, string> = {};\n      request.headers.forEach((value, key) => {\n        existingHeaders[key] = value;\n      });\n      if (ctx.props) agent.updateProps(ctx.props);\n      const response = await agent.fetch(\n        new Request(request.url, {\n          headers: {\n            ...existingHeaders,\n            Upgrade: \"websocket\"\n          }\n        })\n      );\n\n      // Get the WebSocket\n      const ws = response.webSocket;\n      if (!ws) {\n        console.error(\"Failed to establish WebSocket connection\");\n        await writer.close();\n        return new Response(\"Failed to establish WebSocket connection\", {\n          status: 500\n        });\n      }\n\n      // Accept the WebSocket\n      ws.accept();\n\n      // Handle messages from the Durable Object\n      ws.addEventListener(\"message\", (event) => {\n        async function onMessage(event: MessageEvent) {\n          try {\n            const message = JSON.parse(event.data);\n\n            // validate that the message is a valid JSONRPC message\n            const result = JSONRPCMessageSchema.safeParse(message);\n            if (!result.success) {\n              // The message was not a valid JSONRPC message, so we will drop it\n              // PartyKit will broadcast state change messages to all connected clients\n              // and we need to filter those out so they are not passed to MCP clients\n              return;\n            }\n\n            // Send the message as an SSE event\n            const messageText = `event: message\\ndata: ${JSON.stringify(result.data)}\\n\\n`;\n            await writer.write(encoder.encode(messageText));\n          } catch (error) {\n            console.error(\"Error forwarding message to SSE:\", error);\n          }\n        }\n        onMessage(event).catch(console.error);\n      });\n\n      // Handle WebSocket errors\n      ws.addEventListener(\"error\", (error) => {\n        async function onError(_error: Event) {\n          try {\n            await writer.close();\n          } catch (_e) {\n            // Ignore errors when closing\n          }\n        }\n        onError(error).catch(console.error);\n      });\n\n      // Handle WebSocket closure\n      ws.addEventListener(\"close\", () => {\n        async function onClose() {\n          try {\n            await writer.close();\n          } catch (error) {\n            console.error(\"Error closing SSE connection:\", error);\n          }\n        }\n        onClose().catch(console.error);\n      });\n\n      // Return the SSE response\n      return new Response(readable, {\n        headers: {\n          \"Cache-Control\": \"no-cache\",\n          Connection: \"keep-alive\",\n          \"Content-Type\": \"text/event-stream\",\n          ...corsHeaders(request, corsOptions)\n        }\n      });\n    }\n\n    // Handle incoming MCP messages. These will be passed to McpAgent\n    // but the response will be sent back via the open SSE connection\n    // so we only need to return a 202 Accepted response for success\n    if (request.method === \"POST\" && messagePattern.test(url)) {\n      const sessionId = url.searchParams.get(\"sessionId\");\n      if (!sessionId) {\n        return new Response(\n          `Missing sessionId. Expected POST to ${basePath} to initiate new one`,\n          { status: 400 }\n        );\n      }\n\n      const contentType = request.headers.get(\"content-type\") || \"\";\n      if (!contentType.includes(\"application/json\")) {\n        return new Response(`Unsupported content-type: ${contentType}`, {\n          status: 400\n        });\n      }\n\n      // check if the request body is too large\n      const contentLength = Number.parseInt(\n        request.headers.get(\"content-length\") || \"0\",\n        10\n      );\n      if (contentLength > MAXIMUM_MESSAGE_SIZE_BYTES) {\n        return new Response(`Request body too large: ${contentLength} bytes`, {\n          status: 400\n        });\n      }\n\n      // Get the Durable Object\n      const agent = await getAgentByName(namespace, `sse:${sessionId}`, {\n        props: ctx.props\n      });\n\n      const messageBody = await request.json();\n      const error = await agent.onSSEMcpMessage(sessionId, messageBody);\n\n      if (error) {\n        return new Response(error.message, {\n          headers: {\n            \"Cache-Control\": \"no-cache\",\n            Connection: \"keep-alive\",\n            \"Content-Type\": \"text/event-stream\",\n            ...corsHeaders(request, corsOptions)\n          },\n          status: 400\n        });\n      }\n\n      return new Response(\"Accepted\", {\n        headers: {\n          \"Cache-Control\": \"no-cache\",\n          Connection: \"keep-alive\",\n          \"Content-Type\": \"text/event-stream\",\n          ...corsHeaders(request, corsOptions)\n        },\n        status: 202\n      });\n    }\n\n    return new Response(\"Not Found\", { status: 404 });\n  };\n};\n\n// CORS helper functions\nfunction corsHeaders(_request: Request, corsOptions: CORSOptions = {}) {\n  const origin = \"*\";\n  return {\n    \"Access-Control-Allow-Headers\":\n      corsOptions.headers ||\n      \"Content-Type, Accept, mcp-session-id, mcp-protocol-version\",\n    \"Access-Control-Allow-Methods\":\n      corsOptions.methods || \"GET, POST, DELETE, OPTIONS\",\n    \"Access-Control-Allow-Origin\": corsOptions.origin || origin,\n    \"Access-Control-Expose-Headers\":\n      corsOptions.exposeHeaders || \"mcp-session-id\",\n    \"Access-Control-Max-Age\": (corsOptions.maxAge || 86400).toString()\n  };\n}\n\nexport function handleCORS(\n  request: Request,\n  corsOptions?: CORSOptions\n): Response | null {\n  if (request.method === \"OPTIONS\") {\n    return new Response(null, { headers: corsHeaders(request, corsOptions) });\n  }\n\n  return null;\n}\n\nexport function isDurableObjectNamespace(\n  namespace: unknown\n): namespace is DurableObjectNamespace<McpAgent> {\n  return (\n    typeof namespace === \"object\" &&\n    namespace !== null &&\n    \"newUniqueId\" in namespace &&\n    typeof namespace.newUniqueId === \"function\" &&\n    \"idFromName\" in namespace &&\n    typeof namespace.idFromName === \"function\"\n  );\n}\n","import type { Transport } from \"@modelcontextprotocol/sdk/shared/transport.js\";\nimport type {\n  JSONRPCMessage,\n  RequestId\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport {\n  isJSONRPCError,\n  isJSONRPCResponse\n} from \"@modelcontextprotocol/sdk/types.js\";\n\nexport class McpSSETransport implements Transport {\n  sessionId?: string;\n  // Set by the server in `server.connect(transport)`\n  onclose?: () => void;\n  onerror?: (error: Error) => void;\n  onmessage?: (message: JSONRPCMessage) => void;\n\n  private _getWebSocket: () => WebSocket | null;\n  private _started = false;\n  constructor(getWebSocket: () => WebSocket | null) {\n    this._getWebSocket = getWebSocket;\n  }\n\n  async start() {\n    // The transport does not manage the WebSocket connection since it's terminated\n    // by the Durable Object in order to allow hibernation. There's nothing to initialize.\n    if (this._started) {\n      throw new Error(\"Transport already started\");\n    }\n    this._started = true;\n  }\n\n  async send(message: JSONRPCMessage) {\n    if (!this._started) {\n      throw new Error(\"Transport not started\");\n    }\n    const websocket = this._getWebSocket();\n    if (!websocket) {\n      throw new Error(\"WebSocket not connected\");\n    }\n    try {\n      websocket.send(JSON.stringify(message));\n    } catch (error) {\n      this.onerror?.(error as Error);\n    }\n  }\n\n  async close() {\n    // Similar to start, the only thing to do is to pass the event on to the server\n    this.onclose?.();\n  }\n}\n\nexport class McpStreamableHttpTransport implements Transport {\n  sessionId?: string;\n  // Set by the server in `server.connect(transport)`\n  onclose?: () => void;\n  onerror?: (error: Error) => void;\n  onmessage?: (message: JSONRPCMessage) => void;\n\n  // The server MAY use a standalone SSE stream to send requests/notifications.\n  private _getWebSocketForStandaloneSse: () => WebSocket | null;\n\n  // Get the appropriate websocket connection for a given message id\n  private _getWebSocketForMessageID: (id: string) => WebSocket | null;\n\n  // Notify the server that a response has been sent for a given message id\n  // so that it may clean up it's mapping of message ids to connections\n  // once they are no longer needed\n  private _notifyResponseIdSent: (id: string) => void;\n\n  private _started = false;\n  constructor(\n    getWebSocketForMessageID: (id: string) => WebSocket | null,\n    notifyResponseIdSent: (id: string) => void,\n    getWebSocketForStandaloneSse: () => WebSocket | null\n  ) {\n    this._getWebSocketForMessageID = getWebSocketForMessageID;\n    this._notifyResponseIdSent = notifyResponseIdSent;\n    this._getWebSocketForStandaloneSse = getWebSocketForStandaloneSse;\n  }\n\n  async start() {\n    // The transport does not manage the WebSocket connection since it's terminated\n    // by the Durable Object in order to allow hibernation. There's nothing to initialize.\n    if (this._started) {\n      throw new Error(\"Transport already started\");\n    }\n    this._started = true;\n  }\n\n  async send(\n    message: JSONRPCMessage,\n    options?: { relatedRequestId?: RequestId }\n  ) {\n    if (!this._started) {\n      throw new Error(\"Transport not started\");\n    }\n    let requestId = options?.relatedRequestId;\n\n    if (isJSONRPCResponse(message) || isJSONRPCError(message)) {\n      // If the message is a response, use the request ID from the message\n      requestId = message.id;\n    }\n\n    // Check if this message should be sent on the standalone SSE stream (no request ID)\n    // Ignore notifications from tools (which have relatedRequestId set)\n    // Those will be sent via dedicated response SSE streams\n    if (requestId === undefined) {\n      // For standalone SSE streams, we can only send requests and notifications\n      if (isJSONRPCResponse(message) || isJSONRPCError(message)) {\n        throw new Error(\n          \"Cannot send a response on a standalone SSE stream unless resuming a previous client request\"\n        );\n      }\n      const standaloneSseSocket = this._getWebSocketForStandaloneSse();\n      if (!standaloneSseSocket) {\n        // The spec says the server MAY send messages on the stream, so it's ok to discard if no stream\n        return;\n      }\n      try {\n        standaloneSseSocket?.send(JSON.stringify(message));\n      } catch (error) {\n        this.onerror?.(error as Error);\n      }\n      return;\n    }\n\n    const websocket = this._getWebSocketForMessageID(requestId.toString());\n    if (!websocket) {\n      throw new Error(`Could not find WebSocket for message id: ${requestId}`);\n    }\n\n    try {\n      websocket?.send(JSON.stringify(message));\n      // Cleanup on response/error\n      if (isJSONRPCResponse(message) || isJSONRPCError(message)) {\n        this._notifyResponseIdSent(message.id.toString());\n      }\n    } catch (error) {\n      this.onerror?.(error as Error);\n    }\n  }\n\n  async close() {\n    // Similar to start, the only thing to do is to pass the event on to the server\n    this.onclose?.();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AAIA;AAAA,EACE,wBAAAA;AAAA,EACA,kBAAAC;AAAA,EACA,yBAAAC;AAAA,EACA,oBAAAC;AAAA,EACA,qBAAAC;AAAA,OAEK;;;ACXP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAWA,IAAM,wBAAwB;AAM9B,IAAM,wBAAwB;AACrC,IAAM,6BAA6B,IAAI,OAAO;AAEvC,IAAM,6BAA6B,CACxC,UACA,WACA,gBACG;AACH,MAAI,WAAW;AACf,MAAI,aAAa,IAAK,YAAW;AAEjC,QAAM,cAAc,IAAI,WAAW,EAAE,SAAS,CAAC;AAC/C,SAAO,OAAO,SAAkB,QAA0B;AACxD,UAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAC/B,QAAI,YAAY,KAAK,GAAG,GAAG;AACzB,UAAI,QAAQ,WAAW,QAAQ;AAE7B,cAAM,eAAe,QAAQ,QAAQ,IAAI,QAAQ;AAEjD,YACE,CAAC,cAAc,SAAS,kBAAkB,KAC1C,CAAC,aAAa,SAAS,mBAAmB,GAC1C;AACA,gBAAMC,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SACE;AAAA,YACJ;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAEA,cAAM,KAAK,QAAQ,QAAQ,IAAI,cAAc;AAC7C,YAAI,CAAC,MAAM,CAAC,GAAG,SAAS,kBAAkB,GAAG;AAC3C,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SACE;AAAA,YACJ;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAGA,cAAM,gBAAgB,OAAO;AAAA,UAC3B,QAAQ,QAAQ,IAAI,gBAAgB,KAAK;AAAA,UACzC;AAAA,QACF;AACA,YAAI,gBAAgB,4BAA4B;AAC9C,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SAAS,2CAA2C,0BAA0B;AAAA,YAChF;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAEA,YAAI,YAAY,QAAQ,QAAQ,IAAI,gBAAgB;AACpD,YAAI;AAEJ,YAAI;AACF,uBAAa,MAAM,QAAQ,KAAK;AAAA,QAClC,SAAS,QAAQ;AACf,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SAAS;AAAA,YACX;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAGA,YAAI;AACJ,YAAI,MAAM,QAAQ,UAAU,GAAG;AAC7B,yBAAe;AAAA,QACjB,OAAO;AACL,yBAAe,CAAC,UAAU;AAAA,QAC5B;AAEA,YAAI,WAA6B,CAAC;AAGlC,mBAAW,OAAO,cAAc;AAC9B,cAAI,CAAC,qBAAqB,UAAU,GAAG,EAAE,SAAS;AAChD,kBAAMA,QAAO,KAAK,UAAU;AAAA,cAC1B,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SAAS;AAAA,cACX;AAAA,cACA,IAAI;AAAA,cACJ,SAAS;AAAA,YACX,CAAC;AACD,mBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,UAC3C;AAAA,QACF;AAEA,mBAAW,aAAa,IAAI,CAAC,QAAQ,qBAAqB,MAAM,GAAG,CAAC;AAKpE,cAAM,0BAA0B,SAAS;AAAA,UACvC,CAAC,QAAQ,wBAAwB,UAAU,GAAG,EAAE;AAAA,QAClD;AAEA,YAAI,2BAA2B,WAAW;AACxC,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SACE;AAAA,YACJ;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAGA,YAAI,2BAA2B,SAAS,SAAS,GAAG;AAClD,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SACE;AAAA,YACJ;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAKA,YAAI,CAAC,2BAA2B,CAAC,WAAW;AAC1C,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SAAS;AAAA,YACX;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAIA,oBAAY,aAAa,UAAU,YAAY,EAAE,SAAS;AAG1D,cAAM,QAAQ,MAAM;AAAA,UAClB;AAAA,UACA,mBAAmB,SAAS;AAAA,UAC5B,EAAE,OAAO,IAAI,MAAM;AAAA,QACrB;AACA,cAAM,gBAAgB,MAAM,MAAM,cAAc;AAEhD,YAAI,yBAAyB;AAC3B,gBAAM,MAAM,eAAe;AAAA,QAC7B,WAAW,CAAC,eAAe;AAGzB,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SAAS;AAAA,YACX;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAMA,cAAM,EAAE,UAAU,SAAS,IAAI,IAAI,gBAAgB;AACnD,cAAM,SAAS,SAAS,UAAU;AAClC,cAAM,UAAU,IAAI,YAAY;AAGhC,cAAM,kBAA0C,CAAC;AACjD,gBAAQ,QAAQ,QAAQ,CAAC,OAAO,QAAQ;AACtC,0BAAgB,GAAG,IAAI;AAAA,QACzB,CAAC;AACD,cAAM,MAAM,IAAI,QAAQ,QAAQ,KAAK;AAAA,UACnC,SAAS;AAAA,YACP,GAAG;AAAA,YACH,SAAS;AAAA,UACX;AAAA,QACF,CAAC;AACD,YAAI,IAAI,MAAO,OAAM,YAAY,IAAI,KAAK;AAC1C,cAAM,WAAW,MAAM,MAAM,MAAM,GAAG;AAGtC,cAAM,KAAK,SAAS;AACpB,YAAI,CAAC,IAAI;AACP,kBAAQ,MAAM,0CAA0C;AAExD,gBAAM,OAAO,MAAM;AACnB,gBAAMA,QAAO,KAAK,UAAU;AAAA,YAC1B,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SAAS;AAAA,YACX;AAAA,YACA,IAAI;AAAA,YACJ,SAAS;AAAA,UACX,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAKA,cAAM,aAAmC,oBAAI,IAAI;AAGjD,WAAG,OAAO;AAGV,WAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,yBAAe,UAAUC,QAAqB;AAC5C,gBAAI;AACF,oBAAM,OACJ,OAAOA,OAAM,SAAS,WAClBA,OAAM,OACN,IAAI,YAAY,EAAE,OAAOA,OAAM,IAAI;AACzC,oBAAM,UAAU,KAAK,MAAM,IAAI;AAG/B,oBAAM,SAAS,qBAAqB,UAAU,OAAO;AACrD,kBAAI,CAAC,OAAO,SAAS;AAInB;AAAA,cACF;AAIA,kBACE,kBAAkB,OAAO,IAAI,KAC7B,eAAe,OAAO,IAAI,GAC1B;AACA,2BAAW,OAAO,OAAO,KAAK,EAAE;AAAA,cAClC;AAGA,oBAAM,cAAc;AAAA,QAAyB,KAAK,UAAU,OAAO,IAAI,CAAC;AAAA;AAAA;AACxE,oBAAM,OAAO,MAAM,QAAQ,OAAO,WAAW,CAAC;AAG9C,kBAAI,WAAW,SAAS,GAAG;AACzB,oBAAI,MAAM;AACV,sBAAM,OAAO,MAAM;AAAA,cACrB;AAAA,YACF,SAAS,OAAO;AACd,sBAAQ,MAAM,oCAAoC,KAAK;AAAA,YACzD;AAAA,UACF;AACA,oBAAU,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,QACtC,CAAC;AAGD,WAAG,iBAAiB,SAAS,CAAC,UAAU;AACtC,yBAAe,QAAQ,QAAe;AACpC,gBAAI;AACF,oBAAM,OAAO,MAAM;AAAA,YACrB,SAAS,IAAI;AAAA,YAEb;AAAA,UACF;AACA,kBAAQ,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,QACpC,CAAC;AAGD,WAAG,iBAAiB,SAAS,MAAM;AACjC,yBAAe,UAAU;AACvB,gBAAI;AACF,oBAAM,OAAO,MAAM;AAAA,YACrB,SAAS,OAAO;AACd,sBAAQ,MAAM,iCAAiC,KAAK;AAAA,YACtD;AAAA,UACF;AACA,kBAAQ,EAAE,MAAM,QAAQ,KAAK;AAAA,QAC/B,CAAC;AAID,cAAM,kCAAkC,SAAS;AAAA,UAC/C,CAAC,QAAQ,sBAAsB,GAAG,KAAK,kBAAkB,GAAG;AAAA,QAC9D;AACA,YAAI,iCAAiC;AACnC,qBAAW,WAAW,UAAU;AAC9B,eAAG,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,UACjC;AAGA,aAAG,MAAM;AAET,iBAAO,IAAI,SAAS,MAAM;AAAA,YACxB,SAAS,YAAY,SAAS,WAAW;AAAA,YACzC,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAEA,mBAAW,WAAW,UAAU;AAC9B,cAAI,iBAAiB,OAAO,GAAG;AAI7B,uBAAW,IAAI,QAAQ,EAAE;AAAA,UAC3B;AACA,aAAG,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,QACjC;AAIA,eAAO,IAAI,SAAS,UAAU;AAAA,UAC5B,SAAS;AAAA,YACP,iBAAiB;AAAA,YACjB,YAAY;AAAA,YACZ,gBAAgB;AAAA,YAChB,kBAAkB;AAAA,YAClB,GAAG,YAAY,SAAS,WAAW;AAAA,UACrC;AAAA,UACA,QAAQ;AAAA,QACV,CAAC;AAAA,MACH,WAAW,QAAQ,WAAW,OAAO;AAEnC,cAAM,eAAe,QAAQ,QAAQ,IAAI,QAAQ;AAEjD,YAAI,CAAC,cAAc,SAAS,mBAAmB,GAAG;AAChD,gBAAMD,QAAO,KAAK,UAAU;AAAA,YAC1B,SAAS;AAAA,YACT,OAAO;AAAA,cACL,MAAM;AAAA,cACN,SAAS;AAAA,YACX;AAAA,YACA,IAAI;AAAA,UACN,CAAC;AACD,iBAAO,IAAI,SAASA,OAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C;AAGA,cAAM,YAAY,IAAI,aAAa,IAAI,WAAW;AAClD,YAAI,CAAC;AACH,iBAAO,IAAI,SAAS,qBAAqB,EAAE,QAAQ,IAAI,CAAC;AAG1D,cAAM,EAAE,UAAU,SAAS,IAAI,IAAI,gBAAgB;AACnD,cAAM,SAAS,SAAS,UAAU;AAClC,cAAM,UAAU,IAAI,YAAY;AAEhC,cAAM,QAAQ,MAAM;AAAA,UAClB;AAAA,UACA,mBAAmB,SAAS;AAAA,UAC5B,EAAE,OAAO,IAAI,MAAM;AAAA,QACrB;AACA,cAAM,gBAAgB,MAAM,MAAM,cAAc;AAChD,YAAI,CAAC,eAAe;AAClB,iBAAO,IAAI;AAAA,YACT,KAAK,UAAU;AAAA,cACb,SAAS;AAAA,cACT,OAAO,EAAE,MAAM,QAAQ,SAAS,oBAAoB;AAAA,cACpD,IAAI;AAAA,YACN,CAAC;AAAA,YACD,EAAE,QAAQ,IAAI;AAAA,UAChB;AAAA,QACF;AAEA,cAAM,kBAA0C,CAAC;AACjD,gBAAQ,QAAQ,QAAQ,CAAC,GAAG,MAAM;AAChC,0BAAgB,CAAC,IAAI;AAAA,QACvB,CAAC;AAED,YAAI,IAAI,MAAO,OAAM,YAAY,IAAI,KAAK;AAC1C,cAAM,WAAW,MAAM,MAAM;AAAA,UAC3B,IAAI,QAAQ,QAAQ,KAAK;AAAA,YACvB,SAAS;AAAA,cACP,GAAG;AAAA,cACH,SAAS;AAAA,YACX;AAAA,UACF,CAAC;AAAA,QACH;AAEA,cAAM,KAAK,SAAS;AACpB,YAAI,CAAC,IAAI;AACP,gBAAM,OAAO,MAAM;AACnB,iBAAO,IAAI,SAAS,gCAAgC;AAAA,YAClD,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AACA,WAAG,OAAO;AAGV,WAAG;AAAA,UACD,KAAK,UAAU;AAAA,YACb,SAAS;AAAA,YACT,QAAQ;AAAA,YACR,QAAQ,CAAC;AAAA,UACX,CAAC;AAAA,QACH;AAGA,WAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,cAAI;AACF,2BAAe,UAAU,IAAkB;AACzC,oBAAM,OACJ,OAAO,GAAG,SAAS,WACf,GAAG,OACH,IAAI,YAAY,EAAE,OAAO,GAAG,IAAI;AACtC,oBAAM,SAAS,qBAAqB,UAAU,KAAK,MAAM,IAAI,CAAC;AAC9D,kBAAI,CAAC,OAAO,QAAS;AACrB,oBAAM,QAAQ;AAAA,QAAyB,KAAK,UAAU,OAAO,IAAI,CAAC;AAAA;AAAA;AAClE,oBAAM,OAAO,MAAM,QAAQ,OAAO,KAAK,CAAC;AAAA,YAC1C;AACA,sBAAU,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,UACtC,SAAS,GAAG;AACV,oBAAQ,MAAM,oCAAoC,CAAC;AAAA,UACrD;AAAA,QACF,CAAC;AAED,WAAG,iBAAiB,SAAS,MAAM;AACjC,iBAAO,MAAM,EAAE,MAAM,MAAM;AAAA,UAAC,CAAC;AAAA,QAC/B,CAAC;AACD,WAAG,iBAAiB,SAAS,MAAM;AACjC,iBAAO,MAAM,EAAE,MAAM,MAAM;AAAA,UAAC,CAAC;AAAA,QAC/B,CAAC;AAED,eAAO,IAAI,SAAS,UAAU;AAAA,UAC5B,SAAS;AAAA,YACP,iBAAiB;AAAA,YACjB,YAAY;AAAA,YACZ,gBAAgB;AAAA,YAChB,kBAAkB;AAAA,YAClB,GAAG,YAAY,SAAS,WAAW;AAAA,UACrC;AAAA,UACA,QAAQ;AAAA,QACV,CAAC;AAAA,MACH,WAAW,QAAQ,WAAW,UAAU;AACtC,cAAM,YAAY,QAAQ,QAAQ,IAAI,gBAAgB;AACtD,YAAI,CAAC,WAAW;AACd,iBAAO,IAAI;AAAA,YACT,KAAK,UAAU;AAAA,cACb,SAAS;AAAA,cACT,OAAO;AAAA,gBACL,MAAM;AAAA,gBACN,SAAS;AAAA,cACX;AAAA,cACA,IAAI;AAAA,YACN,CAAC;AAAA,YACD,EAAE,QAAQ,KAAK,SAAS,YAAY,SAAS,WAAW,EAAE;AAAA,UAC5D;AAAA,QACF;AACA,cAAM,QAAQ,MAAM;AAAA,UAClB;AAAA,UACA,mBAAmB,SAAS;AAAA,QAC9B;AACA,cAAM,gBAAgB,MAAM,MAAM,cAAc;AAChD,YAAI,CAAC,eAAe;AAClB,iBAAO,IAAI;AAAA,YACT,KAAK,UAAU;AAAA,cACb,SAAS;AAAA,cACT,OAAO,EAAE,MAAM,QAAQ,SAAS,oBAAoB;AAAA,cACpD,IAAI;AAAA,YACN,CAAC;AAAA,YACD,EAAE,QAAQ,KAAK,SAAS,YAAY,SAAS,WAAW,EAAE;AAAA,UAC5D;AAAA,QACF;AAGA,YAAI;AAAA,UACF,MAAM,QAAQ,EAAE,MAAM,MAAM;AAAA,UAE5B,CAAC;AAAA,QACH;AACA,eAAO,IAAI,SAAS,MAAM;AAAA,UACxB,QAAQ;AAAA,UACR,SAAS,YAAY,SAAS,WAAW;AAAA,QAC3C,CAAC;AAAA,MACH;AAAA,IACF;AAGA,UAAM,OAAO,KAAK,UAAU;AAAA,MAC1B,OAAO;AAAA,QACL,MAAM;AAAA,QACN,SAAS;AAAA,MACX;AAAA,MACA,IAAI;AAAA,MACJ,SAAS;AAAA,IACX,CAAC;AACD,WAAO,IAAI,SAAS,MAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,EAC3C;AACF;AAEO,IAAM,yBAAyB,CACpC,UACA,WACA,gBACG;AACH,MAAI,WAAW;AACf,MAAI,aAAa,IAAK,YAAW;AAEjC,QAAM,cAAc,IAAI,WAAW,EAAE,SAAS,CAAC;AAC/C,QAAM,iBAAiB,IAAI,WAAW,EAAE,UAAU,GAAG,QAAQ,WAAW,CAAC;AACzE,SAAO,OAAO,SAAkB,QAA0B;AACxD,UAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAE/B,QAAI,QAAQ,WAAW,SAAS,YAAY,KAAK,GAAG,GAAG;AAGrD,YAAM,YACJ,IAAI,aAAa,IAAI,WAAW,KAAK,UAAU,YAAY,EAAE,SAAS;AAGxE,YAAM,EAAE,UAAU,SAAS,IAAI,IAAI,gBAAgB;AACnD,YAAM,SAAS,SAAS,UAAU;AAClC,YAAM,UAAU,IAAI,YAAY;AAGhC,YAAM,cAAc,IAAI,IAAI,QAAQ,GAAG;AACvC,kBAAY,WAAW,UAAU,GAAG,QAAQ,UAAU;AACtD,kBAAY,aAAa,IAAI,aAAa,SAAS;AACnD,YAAM,yBACJ,YAAY,WAAW,YAAY,SAAS,YAAY;AAC1D,YAAM,kBAAkB;AAAA,QAA0B,sBAAsB;AAAA;AAAA;AACxE,aAAO,MAAM,QAAQ,OAAO,eAAe,CAAC;AAG5C,YAAM,QAAQ,MAAM,eAAe,WAAW,OAAO,SAAS,IAAI;AAAA,QAChE,OAAO,IAAI;AAAA,MACb,CAAC;AAGD,YAAM,kBAA0C,CAAC;AACjD,cAAQ,QAAQ,QAAQ,CAAC,OAAO,QAAQ;AACtC,wBAAgB,GAAG,IAAI;AAAA,MACzB,CAAC;AACD,UAAI,IAAI,MAAO,OAAM,YAAY,IAAI,KAAK;AAC1C,YAAM,WAAW,MAAM,MAAM;AAAA,QAC3B,IAAI,QAAQ,QAAQ,KAAK;AAAA,UACvB,SAAS;AAAA,YACP,GAAG;AAAA,YACH,SAAS;AAAA,UACX;AAAA,QACF,CAAC;AAAA,MACH;AAGA,YAAM,KAAK,SAAS;AACpB,UAAI,CAAC,IAAI;AACP,gBAAQ,MAAM,0CAA0C;AACxD,cAAM,OAAO,MAAM;AACnB,eAAO,IAAI,SAAS,4CAA4C;AAAA,UAC9D,QAAQ;AAAA,QACV,CAAC;AAAA,MACH;AAGA,SAAG,OAAO;AAGV,SAAG,iBAAiB,WAAW,CAAC,UAAU;AACxC,uBAAe,UAAUC,QAAqB;AAC5C,cAAI;AACF,kBAAM,UAAU,KAAK,MAAMA,OAAM,IAAI;AAGrC,kBAAM,SAAS,qBAAqB,UAAU,OAAO;AACrD,gBAAI,CAAC,OAAO,SAAS;AAInB;AAAA,YACF;AAGA,kBAAM,cAAc;AAAA,QAAyB,KAAK,UAAU,OAAO,IAAI,CAAC;AAAA;AAAA;AACxE,kBAAM,OAAO,MAAM,QAAQ,OAAO,WAAW,CAAC;AAAA,UAChD,SAAS,OAAO;AACd,oBAAQ,MAAM,oCAAoC,KAAK;AAAA,UACzD;AAAA,QACF;AACA,kBAAU,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,MACtC,CAAC;AAGD,SAAG,iBAAiB,SAAS,CAAC,UAAU;AACtC,uBAAe,QAAQ,QAAe;AACpC,cAAI;AACF,kBAAM,OAAO,MAAM;AAAA,UACrB,SAAS,IAAI;AAAA,UAEb;AAAA,QACF;AACA,gBAAQ,KAAK,EAAE,MAAM,QAAQ,KAAK;AAAA,MACpC,CAAC;AAGD,SAAG,iBAAiB,SAAS,MAAM;AACjC,uBAAe,UAAU;AACvB,cAAI;AACF,kBAAM,OAAO,MAAM;AAAA,UACrB,SAAS,OAAO;AACd,oBAAQ,MAAM,iCAAiC,KAAK;AAAA,UACtD;AAAA,QACF;AACA,gBAAQ,EAAE,MAAM,QAAQ,KAAK;AAAA,MAC/B,CAAC;AAGD,aAAO,IAAI,SAAS,UAAU;AAAA,QAC5B,SAAS;AAAA,UACP,iBAAiB;AAAA,UACjB,YAAY;AAAA,UACZ,gBAAgB;AAAA,UAChB,GAAG,YAAY,SAAS,WAAW;AAAA,QACrC;AAAA,MACF,CAAC;AAAA,IACH;AAKA,QAAI,QAAQ,WAAW,UAAU,eAAe,KAAK,GAAG,GAAG;AACzD,YAAM,YAAY,IAAI,aAAa,IAAI,WAAW;AAClD,UAAI,CAAC,WAAW;AACd,eAAO,IAAI;AAAA,UACT,uCAAuC,QAAQ;AAAA,UAC/C,EAAE,QAAQ,IAAI;AAAA,QAChB;AAAA,MACF;AAEA,YAAM,cAAc,QAAQ,QAAQ,IAAI,cAAc,KAAK;AAC3D,UAAI,CAAC,YAAY,SAAS,kBAAkB,GAAG;AAC7C,eAAO,IAAI,SAAS,6BAA6B,WAAW,IAAI;AAAA,UAC9D,QAAQ;AAAA,QACV,CAAC;AAAA,MACH;AAGA,YAAM,gBAAgB,OAAO;AAAA,QAC3B,QAAQ,QAAQ,IAAI,gBAAgB,KAAK;AAAA,QACzC;AAAA,MACF;AACA,UAAI,gBAAgB,4BAA4B;AAC9C,eAAO,IAAI,SAAS,2BAA2B,aAAa,UAAU;AAAA,UACpE,QAAQ;AAAA,QACV,CAAC;AAAA,MACH;AAGA,YAAM,QAAQ,MAAM,eAAe,WAAW,OAAO,SAAS,IAAI;AAAA,QAChE,OAAO,IAAI;AAAA,MACb,CAAC;AAED,YAAM,cAAc,MAAM,QAAQ,KAAK;AACvC,YAAM,QAAQ,MAAM,MAAM,gBAAgB,WAAW,WAAW;AAEhE,UAAI,OAAO;AACT,eAAO,IAAI,SAAS,MAAM,SAAS;AAAA,UACjC,SAAS;AAAA,YACP,iBAAiB;AAAA,YACjB,YAAY;AAAA,YACZ,gBAAgB;AAAA,YAChB,GAAG,YAAY,SAAS,WAAW;AAAA,UACrC;AAAA,UACA,QAAQ;AAAA,QACV,CAAC;AAAA,MACH;AAEA,aAAO,IAAI,SAAS,YAAY;AAAA,QAC9B,SAAS;AAAA,UACP,iBAAiB;AAAA,UACjB,YAAY;AAAA,UACZ,gBAAgB;AAAA,UAChB,GAAG,YAAY,SAAS,WAAW;AAAA,QACrC;AAAA,QACA,QAAQ;AAAA,MACV,CAAC;AAAA,IACH;AAEA,WAAO,IAAI,SAAS,aAAa,EAAE,QAAQ,IAAI,CAAC;AAAA,EAClD;AACF;AAGA,SAAS,YAAY,UAAmB,cAA2B,CAAC,GAAG;AACrE,QAAM,SAAS;AACf,SAAO;AAAA,IACL,gCACE,YAAY,WACZ;AAAA,IACF,gCACE,YAAY,WAAW;AAAA,IACzB,+BAA+B,YAAY,UAAU;AAAA,IACrD,iCACE,YAAY,iBAAiB;AAAA,IAC/B,2BAA2B,YAAY,UAAU,OAAO,SAAS;AAAA,EACnE;AACF;AAEO,SAAS,WACd,SACA,aACiB;AACjB,MAAI,QAAQ,WAAW,WAAW;AAChC,WAAO,IAAI,SAAS,MAAM,EAAE,SAAS,YAAY,SAAS,WAAW,EAAE,CAAC;AAAA,EAC1E;AAEA,SAAO;AACT;AAEO,SAAS,yBACd,WAC+C;AAC/C,SACE,OAAO,cAAc,YACrB,cAAc,QACd,iBAAiB,aACjB,OAAO,UAAU,gBAAgB,cACjC,gBAAgB,aAChB,OAAO,UAAU,eAAe;AAEpC;;;ACxvBA;AAAA,EACE,kBAAAC;AAAA,EACA,qBAAAC;AAAA,OACK;AAEA,IAAM,kBAAN,MAA2C;AAAA,EAShD,YAAY,cAAsC;AADlD,SAAQ,WAAW;AAEjB,SAAK,gBAAgB;AAAA,EACvB;AAAA,EAEA,MAAM,QAAQ;AAGZ,QAAI,KAAK,UAAU;AACjB,YAAM,IAAI,MAAM,2BAA2B;AAAA,IAC7C;AACA,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,MAAM,KAAK,SAAyB;AAClC,QAAI,CAAC,KAAK,UAAU;AAClB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AACA,UAAM,YAAY,KAAK,cAAc;AACrC,QAAI,CAAC,WAAW;AACd,YAAM,IAAI,MAAM,yBAAyB;AAAA,IAC3C;AACA,QAAI;AACF,gBAAU,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,IACxC,SAAS,OAAO;AACd,WAAK,UAAU,KAAc;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAM,QAAQ;AAEZ,SAAK,UAAU;AAAA,EACjB;AACF;AAEO,IAAM,6BAAN,MAAsD;AAAA,EAmB3D,YACE,0BACA,sBACA,8BACA;AALF,SAAQ,WAAW;AAMjB,SAAK,4BAA4B;AACjC,SAAK,wBAAwB;AAC7B,SAAK,gCAAgC;AAAA,EACvC;AAAA,EAEA,MAAM,QAAQ;AAGZ,QAAI,KAAK,UAAU;AACjB,YAAM,IAAI,MAAM,2BAA2B;AAAA,IAC7C;AACA,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,MAAM,KACJ,SACA,SACA;AACA,QAAI,CAAC,KAAK,UAAU;AAClB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AACA,QAAI,YAAY,SAAS;AAEzB,QAAIA,mBAAkB,OAAO,KAAKD,gBAAe,OAAO,GAAG;AAEzD,kBAAY,QAAQ;AAAA,IACtB;AAKA,QAAI,cAAc,QAAW;AAE3B,UAAIC,mBAAkB,OAAO,KAAKD,gBAAe,OAAO,GAAG;AACzD,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AACA,YAAM,sBAAsB,KAAK,8BAA8B;AAC/D,UAAI,CAAC,qBAAqB;AAExB;AAAA,MACF;AACA,UAAI;AACF,6BAAqB,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,MACnD,SAAS,OAAO;AACd,aAAK,UAAU,KAAc;AAAA,MAC/B;AACA;AAAA,IACF;AAEA,UAAM,YAAY,KAAK,0BAA0B,UAAU,SAAS,CAAC;AACrE,QAAI,CAAC,WAAW;AACd,YAAM,IAAI,MAAM,4CAA4C,SAAS,EAAE;AAAA,IACzE;AAEA,QAAI;AACF,iBAAW,KAAK,KAAK,UAAU,OAAO,CAAC;AAEvC,UAAIC,mBAAkB,OAAO,KAAKD,gBAAe,OAAO,GAAG;AACzD,aAAK,sBAAsB,QAAQ,GAAG,SAAS,CAAC;AAAA,MAClD;AAAA,IACF,SAAS,OAAO;AACd,WAAK,UAAU,KAAc;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAM,QAAQ;AAEZ,SAAK,UAAU;AAAA,EACjB;AACF;;;AF4WA;AAAA,EACE;AAAA,OAGK;AAjeA,IAAe,WAAf,MAAe,kBAIZ,MAAyB;AAAA,EAUjC,YAAY,KAAmB,KAAU;AACvC,UAAM,KAAK,GAAG;AAThB,SAAQ,2BAAyD,oBAAI,IAAI;AAavE,eAAW,cAAc,KAAK,eAAmC,GAAG;AAClE,YAAM,OAAO,WAAW;AACxB,UAAI,MAAM,SAAS,uBAAuB;AACxC,aAAK,6BAA6B,WAAW;AAC7C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,iBAAiB;AAErB,UAAM,KAAK,IAAI,QAAQ,IAAI,eAAe,IAAI;AAAA,EAChD;AAAA,EAEA,MAAM,gBAAgB;AACpB,WAAQ,MAAM,KAAK,IAAI,QAAQ,IAAI,aAAa,MAAO;AAAA,EACzD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,mBAAkC;AAChC,UAAM,CAAC,GAAG,GAAG,CAAC,IAAI,KAAK,KAAK,MAAM,GAAG;AACrC,YAAQ,GAAG;AAAA,MACT,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AACH,eAAO;AAAA,MACT;AACE,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,IACJ;AAAA,EACF;AAAA;AAAA,EAGQ,+BAAiD;AACvD,QAAI,CAAC,KAAK,2BAA4B,QAAO;AAC7C,WAAO,KAAK,cAAc,KAAK,0BAA0B,KAAK;AAAA,EAChE;AAAA;AAAA,EAGQ,eAAe;AACrB,UAAM,aAAa,MAAM,KAAK,KAAK,eAAe,CAAC;AACnD,QAAI,WAAW,WAAW,GAAG;AAC3B,aAAO;AAAA,IACT;AACA,WAAO,WAAW,CAAC;AAAA,EACrB;AAAA;AAAA,EAGQ,0BAA0B,IAA8B;AAC9D,UAAM,eAAe,KAAK,yBAAyB,IAAI,EAAE;AACzD,QAAI,iBAAiB,QAAW;AAC9B,aAAO;AAAA,IACT;AACA,WAAO,KAAK,cAAc,YAAY,KAAK;AAAA,EAC7C;AAAA;AAAA,EAGQ,gBAAgB;AACtB,YAAQ,KAAK,iBAAiB,GAAG;AAAA,MAC/B,KAAK,OAAO;AACV,eAAO,IAAI,gBAAgB,MAAM,KAAK,aAAa,CAAC;AAAA,MACtD;AAAA,MACA,KAAK,mBAAmB;AACtB,eAAO,IAAI;AAAA,UACT,CAAC,OAAO,KAAK,0BAA0B,EAAE;AAAA,UACzC,CAAC,OAAO,KAAK,yBAAyB,OAAO,EAAE;AAAA,UAC/C,MAAM,KAAK,6BAA6B;AAAA,QAC1C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,YAAY,OAAe;AAC/B,UAAM,KAAK,IAAI,QAAQ,IAAI,SAAS,SAAS,CAAC,CAAC;AAC/C,SAAK,QAAQ;AAAA,EACf;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,QAAQ,OAAe;AAE3B,QAAI,MAAO,OAAM,KAAK,YAAY,KAAK;AACvC,SAAK,QAAQ,MAAM,KAAK,IAAI,QAAQ,IAAI,OAAO;AAE/C,UAAM,KAAK,KAAK;AAChB,UAAM,SAAS,MAAM,KAAK;AAE1B,SAAK,aAAa,KAAK,cAAc;AACrC,UAAM,OAAO,QAAQ,KAAK,UAAU;AAAA,EACtC;AAAA;AAAA,EAGA,MAAM,UAAU,MAAkB,GAAqC;AACrE,YAAQ,KAAK,iBAAiB,GAAG;AAAA,MAC/B,KAAK,OAAO;AAGV,cAAM,aAAa,MAAM,KAAK,KAAK,eAAe,CAAC;AACnD,YAAI,WAAW,SAAS,GAAG;AACzB,eAAK,MAAM,MAAM,6BAA6B;AAC9C;AAAA,QACF;AACA;AAAA,MACF;AAAA,MACA,KAAK;AACH;AAAA,IACJ;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,UAAU,YAAwB,OAAkB;AAGxD,QAAI,KAAK,iBAAiB,MAAM,mBAAmB;AACjD,YAAM,MAAM,IAAI;AAAA,QACd;AAAA,MACF;AACA,WAAK,YAAY,UAAU,GAAG;AAC9B;AAAA,IACF;AAEA,QAAI;AACJ,QAAI;AAEF,YAAM,OACJ,OAAO,UAAU,WAAW,QAAQ,IAAI,YAAY,EAAE,OAAO,KAAK;AACpE,gBAAUE,sBAAqB,MAAM,KAAK,MAAM,IAAI,CAAC;AAAA,IACvD,SAAS,OAAO;AACd,WAAK,YAAY,UAAU,KAAc;AACzC;AAAA,IACF;AAGA,QACEC,uBAAsB,OAAO,KAC7B,QAAQ,WAAW,uBACnB;AACA,UACE,KAAK,8BACL,KAAK,+BAA+B,WAAW,IAC/C;AAGA,cAAM,sBAAsB,KAAK;AAAA,UAC/B,KAAK;AAAA,QACP;AACA,6BAAqB,MAAM,KAAM,UAAU;AAAA,MAC7C;AACA,iBAAW,SAAS;AAAA,QAClB,MAAM;AAAA,MACR,CAAC;AAED,WAAK,6BAA6B,WAAW;AAE7C;AAAA,IACF;AAGA,QAAI,MAAM,KAAK,2BAA2B,OAAO,GAAG;AAClD;AAAA,IACF;AAIA,QAAIC,kBAAiB,OAAO,GAAG;AAC7B,WAAK,yBAAyB,IAAI,QAAQ,GAAG,SAAS,GAAG,WAAW,EAAE;AAAA,IACxE;AAEA,SAAK,YAAY,YAAY,OAAO;AAAA,EACtC;AAAA;AAAA,EAGA,MAAM,QACJ,MACA,OACA,SACA,WACe;AAEf,eAAW,CAAC,OAAO,MAAM,KAAK,KAAK,0BAA0B;AAC3D,UAAI,WAAW,KAAK,GAAI,MAAK,yBAAyB,OAAO,KAAK;AAAA,IACpE;AAGA,QAAI,KAAK,+BAA+B,KAAK,IAAI;AAC/C,WAAK,6BAA6B;AAAA,IACpC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,gBACJ,YACA,aACuB;AAGvB,QAAI,KAAK,iBAAiB,MAAM,OAAO;AACrC,aAAO,IAAI,MAAM,+CAA+C;AAAA,IAClE;AAEA,QAAI;AACF,UAAI;AACJ,UAAI;AACF,wBAAgBF,sBAAqB,MAAM,WAAW;AAAA,MACxD,SAAS,OAAO;AACd,aAAK,YAAY,UAAU,KAAc;AACzC,cAAM;AAAA,MACR;AAGA,UAAI,MAAM,KAAK,2BAA2B,aAAa,GAAG;AACxD,eAAO;AAAA,MACT;AAEA,WAAK,YAAY,YAAY,aAAa;AAC1C,aAAO;AAAA,IACT,SAAS,OAAO;AACd,cAAQ,MAAM,oCAAoC,KAAK;AACvD,WAAK,YAAY,UAAU,KAAc;AACzC,aAAO;AAAA,IACT;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,YAAY,QAGQ;AACxB,UAAM,YAAY,UAAU,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,UAAU,GAAG,EAAE,CAAC;AAGvE,UAAM,KAAK,IAAI,QAAQ,IAAI,eAAe,SAAS,IAAI;AAAA,MACrD,SAAS,OAAO;AAAA,MAChB,iBAAiB,OAAO;AAAA,MACxB,WAAW,KAAK,IAAI;AAAA,IACtB,CAAC;AAED,UAAM,gBAAgB;AAAA,MACpB,SAAS;AAAA,MACT,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR,QAAQ;AAAA,QACN,SAAS,OAAO;AAAA,QAChB,iBAAiB,OAAO;AAAA,MAC1B;AAAA,IACF;AAGA,QAAI,KAAK,YAAY;AACnB,YAAM,KAAK,WAAW,KAAK,aAAa;AAAA,IAC1C,OAAO;AACL,YAAM,cAAc,KAAK,eAAe;AACxC,UAAI,CAAC,eAAe,MAAM,KAAK,WAAW,EAAE,WAAW,GAAG;AACxD,cAAM,KAAK,IAAI,QAAQ,OAAO,eAAe,SAAS,EAAE;AACxD,cAAM,IAAI,MAAM,iDAAiD;AAAA,MACnE;AAEA,YAAM,iBAAiB,MAAM,KAAK,WAAW;AAC7C,iBAAW,cAAc,gBAAgB;AACvC,YAAI;AACF,qBAAW,KAAK,KAAK,UAAU,aAAa,CAAC;AAAA,QAC/C,SAAS,OAAO;AACd,kBAAQ,MAAM,uCAAuC,KAAK;AAAA,QAC5D;AAAA,MACF;AAAA,IACF;AAGA,WAAO,KAAK,4BAA4B,SAAS;AAAA,EACnD;AAAA;AAAA,EAGA,MAAc,4BACZ,WACuB;AACvB,UAAM,YAAY,KAAK,IAAI;AAC3B,UAAM,UAAU;AAEhB,QAAI;AACF,aAAO,KAAK,IAAI,IAAI,YAAY,SAAS;AAEvC,cAAM,WAAW,MAAM,KAAK,IAAI,QAAQ;AAAA,UACtC,wBAAwB,SAAS;AAAA,QACnC;AACA,YAAI,UAAU;AAEZ,gBAAM,KAAK,IAAI,QAAQ,OAAO,eAAe,SAAS,EAAE;AACxD,gBAAM,KAAK,IAAI,QAAQ,OAAO,wBAAwB,SAAS,EAAE;AACjE,iBAAO;AAAA,QACT;AAGA,cAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAG,CAAC;AAAA,MACzD;AAEA,YAAM,IAAI,MAAM,+BAA+B;AAAA,IACjD,UAAE;AAEA,YAAM,KAAK,IAAI,QAAQ,OAAO,eAAe,SAAS,EAAE;AACxD,YAAM,KAAK,IAAI,QAAQ,OAAO,wBAAwB,SAAS,EAAE;AAAA,IACnE;AAAA,EACF;AAAA;AAAA,EAGA,MAAc,2BACZ,SACkB;AAElB,QAAIG,mBAAkB,OAAO,KAAK,QAAQ,QAAQ;AAChD,YAAM,YAAY,QAAQ,IAAI,SAAS;AACvC,UAAI,CAAC,aAAa,CAAC,UAAU,WAAW,SAAS,EAAG,QAAO;AAG3D,YAAM,iBAAiB,MAAM,KAAK,IAAI,QAAQ;AAAA,QAC5C,eAAe,SAAS;AAAA,MAC1B;AACA,UAAI,CAAC,eAAgB,QAAO;AAG5B,YAAM,KAAK,IAAI,QAAQ;AAAA,QACrB,wBAAwB,SAAS;AAAA,QACjC,QAAQ;AAAA,MACV;AACA,aAAO;AAAA,IACT;AAGA,QAAIC,gBAAe,OAAO,GAAG;AAC3B,YAAM,YAAY,QAAQ,IAAI,SAAS;AACvC,UAAI,CAAC,aAAa,CAAC,UAAU,WAAW,SAAS,EAAG,QAAO;AAG3D,YAAM,iBAAiB,MAAM,KAAK,IAAI,QAAQ;AAAA,QAC5C,eAAe,SAAS;AAAA,MAC1B;AACA,UAAI,CAAC,eAAgB,QAAO;AAG5B,YAAM,cAA4B;AAAA,QAChC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,OAAO,QAAQ,MAAM,WAAW;AAAA,QAClC;AAAA,MACF;AACA,YAAM,KAAK,IAAI,QAAQ;AAAA,QACrB,wBAAwB,SAAS;AAAA,QACjC;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,MACL,MACA;AAAA,IACE,UAAU;AAAA,IACV;AAAA,IACA,YAAY;AAAA,EACd,IAAkB,CAAC,GACnB;AACA,WAAO;AAAA,MACL,MAAM,MAEJ,SACA,KACA,KACmB;AAEnB,cAAM,eAAe,WAAW,SAAS,WAAW;AACpD,YAAI,cAAc;AAChB,iBAAO;AAAA,QACT;AAEA,cAAM,eAAe,IAAI,OAA2B;AAGpD,YAAI,gBAAgB,QAAQ,OAAO,iBAAiB,UAAU;AAC5D,gBAAM,IAAI;AAAA,YACR,uCAAuC,OAAO;AAAA,UAChD;AAAA,QACF;AAGA,YAAI,CAAC,yBAAyB,YAAY,GAAG;AAC3C,gBAAM,IAAI;AAAA,YACR,gCAAgC,OAAO;AAAA,UACzC;AAAA,QACF;AAEA,cAAM,YACJ;AAEF,gBAAQ,WAAW;AAAA,UACjB,KAAK,mBAAmB;AAEtB,kBAAM,uBAAuB;AAAA,cAC3B;AAAA,cACA;AAAA,cACA;AAAA,YACF;AACA,mBAAO,qBAAqB,SAAS,GAAG;AAAA,UAC1C;AAAA,UACA,KAAK,OAAO;AAEV,kBAAM,kBAAkB;AAAA,cACtB;AAAA,cACA;AAAA,cACA;AAAA,YACF;AACA,mBAAO,gBAAgB,SAAS,GAAG;AAAA,UACrC;AAAA,UACA;AACE,mBAAO,IAAI;AAAA,cACT;AAAA,cACA,EAAE,QAAQ,IAAI;AAAA,YAChB;AAAA,QACJ;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAIA,OAAO,MAAM,MAAc,OAAwC,CAAC,GAAG;AACrE,WAAO,UAAS,SAAS,MAAM,IAAI;AAAA,EACrC;AAAA,EAEA,OAAO,SAAS,MAAc,OAAwC,CAAC,GAAG;AACxE,WAAO,UAAS,MAAM,MAAM,EAAE,GAAG,MAAM,WAAW,MAAM,CAAC;AAAA,EAC3D;AACF;","names":["JSONRPCMessageSchema","isJSONRPCError","isJSONRPCNotification","isJSONRPCRequest","isJSONRPCResponse","body","event","isJSONRPCError","isJSONRPCResponse","JSONRPCMessageSchema","isJSONRPCNotification","isJSONRPCRequest","isJSONRPCResponse","isJSONRPCError"]}

package/dist/observability/index.js

@@ -1,7 +1,7 @@
 import {
   genericObservability
-} from "../chunk-PNF6ZMUA.js";
-import "../chunk-VYENMKFS.js";
+} from "../chunk-YDUDMOL6.js";
+import "../chunk-MH46VMM4.js";
 import "../chunk-LL2AFX7V.js";
 import "../chunk-QEVM4BVL.js";
 import "../chunk-AVYJQSLW.js";

package/dist/react.d.ts

@@ -7,7 +7,7 @@ import "cloudflare:workers";
 import "@modelcontextprotocol/sdk/client/index.js";
 import "@modelcontextprotocol/sdk/types.js";
 import "partyserver";
-import "./client-CcIORE73.js";
+import "./client-CvaJdLQA.js";
 import "zod";
 import "@modelcontextprotocol/sdk/shared/protocol.js";
 import "ai";

package/package.json

@@ -4,7 +4,7 @@
     "url": "https://github.com/cloudflare/agents/issues"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.17.5",
+    "@modelcontextprotocol/sdk": "^1.18.0",
     "ai": "5.0.39",
     "cron-schedule": "^5.0.4",
     "mimetext": "^3.0.27",
@@ -18,6 +18,8 @@
   },
   "description": "A home for your AI agents",
   "devDependencies": {
+    "@cloudflare/workers-oauth-provider": "^0.0.6",
+    "alchemy": "^0.62.3",
     "react": "*",
     "vitest-browser-react": "^1.0.1"
   },
@@ -106,9 +108,10 @@
     "check:test:workers": "vitest -r src/tests --watch false",
     "evals": "(cd evals; evalite)",
     "test": "vitest -r src/tests",
-    "test:react": "vitest -r src/react-tests"
+    "test:react": "vitest -r src/react-tests",
+    "test:e2e": "vitest run src/e2e/e2e.test.ts --sequence.concurrent"
   },
   "type": "module",
   "types": "dist/index.d.ts",
-  "version": "0.1.1"
+  "version": "0.1.3"
 }

package/src/index.ts

@@ -308,7 +308,11 @@ function withAgentContext<T extends (...args: any[]) => any>(
  * @template Env Environment type containing bindings
  * @template State State type to store within the Agent
  */
-export class Agent<Env = typeof env, State = unknown> extends Server<Env> {
+export class Agent<
+  Env = typeof env,
+  State = unknown,
+  Props extends Record<string, unknown> = Record<string, unknown>
+> extends Server<Env, Props> {
   private _state = DEFAULT_STATE as State;
 
   private _ParentClass: typeof Agent<Env, State> =
@@ -618,7 +622,7 @@ export class Agent<Env = typeof env, State = unknown> extends Server<Env> {
     };
 
     const _onStart = this.onStart.bind(this);
-    this.onStart = async () => {
+    this.onStart = async (props?: Props) => {
       return agentContext.run(
         {
           agent: this,
@@ -678,7 +682,7 @@ export class Agent<Env = typeof env, State = unknown> extends Server<Env> {
                   });
               });
             }
-            return _onStart();
+            return _onStart(props);
           });
         }
       );
@@ -1875,12 +1879,17 @@ export type EmailSendOptions = {
  * @param options Options for Agent creation
  * @returns Promise resolving to an Agent instance stub
  */
-export async function getAgentByName<Env, T extends Agent<Env>>(
+export async function getAgentByName<
+  Env,
+  T extends Agent<Env>,
+  Props extends Record<string, unknown> = Record<string, unknown>
+>(
   namespace: AgentNamespace<T>,
   name: string,
   options?: {
     jurisdiction?: DurableObjectJurisdiction;
     locationHint?: DurableObjectLocationHint;
+    props?: Props;
   }
 ) {
   return getServerByName<Env, T>(namespace, name, options);