Added optional headers to Ollama initialisation (#138)

ollama · Sep 3, 2024 · 3a9d651 · 3a9d651
1 parent 218bede
commit 3a9d651
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/src/browser.ts b/src/browser.ts
@@ -74,6 +74,7 @@ export class Ollama {
       const abortController = new AbortController()
       const response = await post(this.fetch, host, request, {
         signal: abortController.signal,
+        headers: this.config.headers
       })
 
       if (!response.body) {
@@ -94,7 +95,9 @@ export class Ollama {
       this.ongoingStreamedRequests.push(abortableAsyncIterator)
       return abortableAsyncIterator
     }
-    const response = await utils.post(this.fetch, host, request)
+    const response = await utils.post(this.fetch, host, request, {
+      headers: this.config.headers
+    })
     return await response.json()
   }
 

diff --git a/src/interfaces.ts b/src/interfaces.ts
@@ -4,6 +4,7 @@ export interface Config {
   host: string
   fetch?: Fetch
   proxy?: boolean
+  headers?: Headers
 }
 
 // request types

diff --git a/src/utils.ts b/src/utils.ts
@@ -169,7 +169,7 @@ export const post = async (
   fetch: Fetch,
   host: string,
   data?: Record<string, unknown> | BodyInit,
-  options?: { signal: AbortSignal },
+  options?: { signal?: AbortSignal, headers?: Headers },
 ): Promise<Response> => {
   const isRecord = (input: any): input is Record<string, unknown> => {
     return input !== null && typeof input === 'object' && !Array.isArray(input)
@@ -181,6 +181,7 @@ export const post = async (
     method: 'POST',
     body: formattedData,
     signal: options?.signal,
+    headers: options?.headers
   })
 
   await checkOk(response)