feat: request info (#619)

* feat: request info

* remove authInfo

* chore: changeset

Author: mattzcarey

.changeset/grumpy-ducks-bow.md

@@ -0,0 +1,5 @@
+---
+"agents": patch
+---
+
+Adds request info to the extra argument in onmessage. Adds a url parm which we will try push upstream to the MCP SDK as it is useful with OpenAI Apps SDK

packages/agents/src/mcp/worker-transport.ts

@@ -6,7 +6,9 @@ import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
 import type {
   JSONRPCMessage,
   RequestId,
-  InitializeRequest
+  InitializeRequest,
+  RequestInfo,
+  MessageExtraInfo
 } from "@modelcontextprotocol/sdk/types.js";
 import {
   isInitializeRequest,
@@ -54,7 +56,7 @@ export class WorkerTransport implements Transport {
   sessionId?: string;
   onclose?: () => void;
   onerror?: (error: Error) => void;
-  onmessage?: (message: JSONRPCMessage) => void;
+  onmessage?: (message: JSONRPCMessage, extra?: MessageExtraInfo) => void;
 
   constructor(options?: WorkerTransportOptions) {
     this.sessionIdGenerator = options?.sessionIdGenerator;
@@ -390,6 +392,12 @@ export class WorkerTransport implements Transport {
       );
     }
 
+    // TODO: push this upstream to the mcp sdk asap
+    const requestInfo: RequestInfo & { url?: string } = {
+      headers: Object.fromEntries(request.headers.entries()),
+      url: request.url
+    };
+
     const isInitializationRequest = messages.some(isInitializeRequest);
 
     if (isInitializationRequest) {
@@ -476,7 +484,7 @@ export class WorkerTransport implements Transport {
 
     if (!hasRequests) {
       for (const message of messages) {
-        this.onmessage?.(message);
+        this.onmessage?.(message, { requestInfo });
       }
       return new Response(null, {
         status: 202,
@@ -502,7 +510,7 @@ export class WorkerTransport implements Transport {
         }
 
         for (const message of messages) {
-          this.onmessage?.(message);
+          this.onmessage?.(message, { requestInfo });
         }
       });
     }
@@ -538,7 +546,7 @@ export class WorkerTransport implements Transport {
     }
 
     for (const message of messages) {
-      this.onmessage?.(message);
+      this.onmessage?.(message, { requestInfo });
     }
 
     return new Response(readable, { headers });