feat: Add v2 support for response headers. (#23)

vectara · Jul 11, 2024 · d4c1e85 · d4c1e85
1 parent 5c65173
commit d4c1e85
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 9 deletions.
diff --git a/docs/package-lock.json b/docs/package-lock.json
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/apiV2/client.ts b/src/apiV2/client.ts
@@ -170,11 +170,8 @@ export const streamQueryV2 = async ({
   };
 
   try {
-    const { cancelStream, stream, status } = await generateStream(
-      headers,
-      JSON.stringify(body),
-      url
-    );
+    const { cancelStream, stream, status, responseHeaders } =
+      await generateStream(headers, JSON.stringify(body), url);
 
     const consumeStream = async () => {
       try {
@@ -211,7 +208,7 @@ export const streamQueryV2 = async ({
 
     consumeStream();
 
-    return { cancelStream, request, status };
+    return { cancelStream, request, status, responseHeaders };
   } catch (error) {
     if (error instanceof Error) {
       onStreamEvent({

diff --git a/src/common/generateStream.ts b/src/common/generateStream.ts
@@ -18,6 +18,7 @@ export const generateStream = async (
     stream: getIterableStream(response.body),
     cancelStream: () => controller.abort(),
     status: response.status,
+    responseHeaders: response.headers,
   };
 };