include extra headers in langserve adapters

nlkitai · Mar 27, 2024 · 6404d9b · 6404d9b
1 parent 7d22aca
commit 6404d9b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/packages/js/langchain/src/langserve/adapter/fetch.ts b/packages/js/langchain/src/langserve/adapter/fetch.ts
@@ -10,6 +10,10 @@ export class LangServeFetchAdapter extends LangServeAbstractAdapter {
         const body = this.getRequestBody(message, extras.conversationHistory);
         const response = await fetch(this.endpointUrl, {
             method: 'POST',
+            headers: {
+                ...extras.headers,
+                'Content-Type': 'application/json'
+            },
             body,
         });
 

diff --git a/packages/js/langchain/src/langserve/adapter/stream.ts b/packages/js/langchain/src/langserve/adapter/stream.ts
@@ -19,7 +19,10 @@ export class LangServeStreamAdapter extends LangServeAbstractAdapter {
         const body = this.getRequestBody(message, extras.conversationHistory);
         fetch(this.endpointUrl, {
             method: 'POST',
-            headers: {'Content-Type': 'application/json'},
+            headers: {
+                ...extras.headers,
+                'Content-Type': 'application/json'
+            },
             body,
         })
             .then(async (response) => {