feat(streaming): add .request_id getter

anthropics · Jan 21, 2025 · e1fd091 · e1fd091
1 parent 7a68f81
commit e1fd091
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/lib/BetaMessageStream.ts b/src/lib/BetaMessageStream.ts
@@ -55,6 +55,7 @@ export class BetaMessageStream implements AsyncIterable<BetaMessageStreamEvent>
   #errored = false;
   #aborted = false;
   #catchingPromiseCreated = false;
+  #request_id: string | null | undefined;
 
   constructor() {
     this.#connectedPromise = new Promise<Response | null>((resolve, reject) => {
@@ -75,6 +76,10 @@ export class BetaMessageStream implements AsyncIterable<BetaMessageStreamEvent>
     this.#endPromise.catch(() => {});
   }
 
+  get request_id(): string | null | undefined {
+    return this.#request_id;
+  }
+
   /**
    * Returns the `MessageStream` data, the raw `Response` instance and the ID of the request,
    * returned vie the `request-id` header which is useful for debugging requests and resporting
@@ -178,6 +183,7 @@ export class BetaMessageStream implements AsyncIterable<BetaMessageStreamEvent>
 
   protected _connected(response: Response | null) {
     if (this.ended) return;
+    this.#request_id = response?.headers.get('request-id');
     this.#resolveConnectedPromise(response);
     this._emit('connect');
   }

diff --git a/src/lib/MessageStream.ts b/src/lib/MessageStream.ts
@@ -55,6 +55,7 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
   #errored = false;
   #aborted = false;
   #catchingPromiseCreated = false;
+  #request_id: string | null | undefined;
 
   constructor() {
     this.#connectedPromise = new Promise<Response | null>((resolve, reject) => {
@@ -75,6 +76,10 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
     this.#endPromise.catch(() => {});
   }
 
+  get request_id(): string | null | undefined {
+    return this.#request_id;
+  }
+
   /**
    * Returns the `MessageStream` data, the raw `Response` instance and the ID of the request,
    * returned vie the `request-id` header which is useful for debugging requests and resporting
@@ -178,6 +183,7 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
 
   protected _connected(response: Response | null) {
     if (this.ended) return;
+    this.#request_id = response?.headers.get('request-id');
     this.#resolveConnectedPromise(response);
     this._emit('connect');
   }