From 9748c0fd605e9e814c6754fd56f8bd8073cd8098 Mon Sep 17 00:00:00 2001
From: gered <gered@blarg.ca>
Date: Sun, 3 Nov 2024 17:56:39 -0500
Subject: [PATCH] pass api_token option to ollama backend

---
 crates/llm-ls/src/backend.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/crates/llm-ls/src/backend.rs b/crates/llm-ls/src/backend.rs
index d1467b5..0da12bd 100644
--- a/crates/llm-ls/src/backend.rs
+++ b/crates/llm-ls/src/backend.rs
@@ -118,8 +118,8 @@ enum OllamaAPIResponse {
     Error(APIError),
 }
 
-fn build_ollama_headers() -> HeaderMap {
-    HeaderMap::new()
+fn build_ollama_headers(api_token: Option<&String>, ide: Ide) -> Result<HeaderMap> {
+    build_api_headers(api_token, ide)
 }
 
 fn parse_ollama_text(text: &str) -> Result<Vec<Generation>> {
@@ -243,7 +243,7 @@ pub(crate) fn build_headers(
     match backend {
         Backend::HuggingFace { .. } => build_api_headers(api_token, ide),
         Backend::LlamaCpp { .. } => Ok(build_llamacpp_headers()),
-        Backend::Ollama { .. } => Ok(build_ollama_headers()),
+        Backend::Ollama { .. } => build_ollama_headers(api_token, ide),
         Backend::OpenAi { .. } => build_openai_headers(api_token, ide),
         Backend::Tgi { .. } => build_tgi_headers(api_token, ide),
     }