refactor: simplify McpSSETransport initialization by using getCurrentAgent util

Author: naji247

packages/agents/src/mcp/index.ts

@@ -80,7 +80,7 @@ export abstract class McpAgent<
   }
 
   /** Get the unique WebSocket. SSE transport only. */
-  private getWebSocket() {
+  getWebSocket() {
     const websockets = Array.from(this.getConnections());
     if (websockets.length === 0) {
       return null;
@@ -110,10 +110,7 @@ export abstract class McpAgent<
   private initTransport() {
     switch (this.getTransportType()) {
       case "sse": {
-        return new McpSSETransport({
-          getWebSocket: () => this.getWebSocket(),
-          sessionId: this.getSessionId()
-        });
+        return new McpSSETransport();
       }
       case "streamable-http": {
         return new StreamableHTTPServerTransport({});

packages/agents/src/mcp/transport.ts

@@ -15,23 +15,22 @@ import type { McpAgent } from ".";
 import { MessageType } from "../ai-types";
 import { MCP_HTTP_METHOD_HEADER, MCP_MESSAGE_HEADER } from "./utils";
 
-export interface McpSSETransportOptions {
-  getWebSocket: () => WebSocket | null;
-  sessionId?: string;
-}
-
 export class McpSSETransport implements Transport {
-  sessionId?: string;
+  sessionId: string;
   // Set by the server in `server.connect(transport)`
   onclose?: () => void;
   onerror?: (error: Error) => void;
   onmessage?: (message: JSONRPCMessage, extra?: MessageExtraInfo) => void;
 
   private _getWebSocket: () => WebSocket | null;
   private _started = false;
-  constructor(options: McpSSETransportOptions) {
-    this._getWebSocket = options.getWebSocket;
-    this.sessionId = options.sessionId;
+  constructor() {
+    const { agent } = getCurrentAgent<McpAgent>();
+    if (!agent)
+      throw new Error("McpAgent was not found in Transport constructor");
+
+    this.sessionId = agent.getSessionId();
+    this._getWebSocket = () => agent.getWebSocket();
   }
 
   async start() {