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;