js lib user agent headers

.github/workflows/publish.yaml

@@ -17,6 +17,8 @@ jobs:
           registry-url: https://registry.npmjs.org
           cache: npm
       - run: npm ci
+      - name: Set version in src/version.js
+        run: echo "export const version = '${GITHUB_REF_NAME#v}';" > src/version.js
       - run: |
           npm version --no-git-tag-version ${GITHUB_REF_NAME#v}
           npm publish

src/utils.ts

@@ -1,3 +1,4 @@
+import { version } from './version'
 import type { Fetch, ErrorResponse } from './interfaces.js'
 
 class ResponseError extends Error {
@@ -40,16 +41,48 @@
   }
 }
 
+function getPlatform() {
+  if (typeof window !== 'undefined' && window.navigator) {
+    return `${window.navigator.platform.toLowerCase()} Browser/${navigator.userAgent};`
+  } else if (typeof process !== 'undefined') {
+    return `${process.arch} ${process.platform} Node.js/${process.version}`
+  }
+  return '' // unknown
+}
+
+const fetchWithHeaders = async (
+  fetch: Fetch,
+  url: string,
+  options: RequestInit = {},
+): Promise<Response> => {
+  const defaultHeaders = {
+    'Content-Type': 'application/json',
+    Accept: 'application/json',
+    'User-Agent': `ollama-js/${version} (${getPlatform()})`,
+  }
+
+  if (!options.headers) {
+    options.headers = {}
+  }
+
+  options.headers = {
+    ...defaultHeaders,
+    ...options.headers,
+  }
+
+  return fetch(url, options)
+}
+
 export const get = async (fetch: Fetch, host: string): Promise<Response> => {
-  const response = await fetch(host)
+  const response = await fetchWithHeaders(fetch, host)
 
   await checkOk(response)
 
   return response
 }
 
 export const head = async (fetch: Fetch, host: string): Promise<Response> => {
-  const response = await fetch(host, {
+  const response = await fetchWithHeaders(fetch, host, {
     method: 'HEAD',
   })
 
@@ -64,13 +97,13 @@
   data?: Record<string, unknown> | BodyInit,
   options?: { signal: AbortSignal },
 ): Promise<Response> => {
   const isRecord = (input: any): input is Record<string, unknown> => {
     return input !== null && typeof input === 'object' && !Array.isArray(input)
   }
 
   const formattedData = isRecord(data) ? JSON.stringify(data) : data
 
-  const response = await fetch(host, {
+  const response = await fetchWithHeaders(fetch, host, {
     method: 'POST',
     body: formattedData,
     signal: options?.signal,
@@ -86,7 +119,7 @@
   host: string,
   data?: Record<string, unknown>,
 ): Promise<Response> => {
-  const response = await fetch(host, {
+  const response = await fetchWithHeaders(fetch, host, {
     method: 'DELETE',
     body: JSON.stringify(data),
   })

src/version.ts

@@ -0,0 +1 @@
+export const version = '0.0.0'