feat(stream): expose response property as well

anthropics · Jan 22, 2025 · b0235c7 · b0235c7
1 parent db5fffe
commit b0235c7
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;
+  #response: Response | null | undefined;
   #request_id: string | null | undefined;
 
   constructor() {
@@ -76,6 +77,10 @@ export class BetaMessageStream implements AsyncIterable<BetaMessageStreamEvent>
     this.#endPromise.catch(() => {});
   }
 
+  get response(): Response | null | undefined {
+    return this.#response;
+  }
+
   get request_id(): string | null | undefined {
     return this.#request_id;
   }
@@ -183,6 +188,7 @@ export class BetaMessageStream implements AsyncIterable<BetaMessageStreamEvent>
 
   protected _connected(response: Response | null) {
     if (this.ended) return;
+    this.#response = response;
     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;
+  #response: Response | null | undefined;
   #request_id: string | null | undefined;
 
   constructor() {
@@ -76,6 +77,10 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
     this.#endPromise.catch(() => {});
   }
 
+  get response(): Response | null | undefined {
+    return this.#response;
+  }
+
   get request_id(): string | null | undefined {
     return this.#request_id;
   }
@@ -183,6 +188,7 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
 
   protected _connected(response: Response | null) {
     if (this.ended) return;
+    this.#response = response;
     this.#request_id = response?.headers.get('request-id');
     this.#resolveConnectedPromise(response);
     this._emit('connect');