From e34b6f3d8d9536fa5acbeeebf8ef0bed75bee186 Mon Sep 17 00:00:00 2001
From: projectmoon <projectmoon@agnos.is>
Date: Sat, 29 Jun 2024 10:41:30 +0200
Subject: [PATCH] ghetto ollama auth

---
 src/lib/providers.ts | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/lib/providers.ts b/src/lib/providers.ts
index 3223193..3bf23b6 100644
--- a/src/lib/providers.ts
+++ b/src/lib/providers.ts
@@ -95,12 +95,18 @@ export const getAvailableChatModelProviders = async () => {
   }
 
   if (ollamaEndpoint) {
+    let headers;
+    if (typeof process.env['OLLAMA_AUTH_HEADER'] !== undefined) {
+      headers = {
+	'Content-Type': 'application/json',
+	'Authorization': process.env['OLLAMA_AUTH_HEADER']
+      };
+    } else {
+      headers = { 'Content-Type': 'application/json'      };
+    }
+
     try {
-      const response = await fetch(`${ollamaEndpoint}/api/tags`, {
-        headers: {
-          'Content-Type': 'application/json',
-        },
-      });
+      const response = await fetch(`${ollamaEndpoint}/api/tags`, { headers });
 
       const { models: ollamaModels } = (await response.json()) as any;