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)