gered/llm-ls
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

pass api_token option to ollama backend

Browse source
This commit is contained in:
Gered 2024-11-03 17:56:39 -05:00
parent 5abb3c86b4
commit 9748c0fd60
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
crates/llm-ls/src/backend.rs
View file

@ -118,8 +118,8 @@ enum OllamaAPIResponse {
Error(APIError), Error(APIError),
} }
fn build_ollama_headers() -> HeaderMap { fn build_ollama_headers(api_token: Option<&String>, ide: Ide) -> Result<HeaderMap> {
HeaderMap::new() build_api_headers(api_token, ide)
} }
fn parse_ollama_text(text: &str) -> Result<Vec<Generation>> { fn parse_ollama_text(text: &str) -> Result<Vec<Generation>> {
@ -243,7 +243,7 @@ pub(crate) fn build_headers(
match backend { match backend {
Backend::HuggingFace { .. } => build_api_headers(api_token, ide), Backend::HuggingFace { .. } => build_api_headers(api_token, ide),
Backend::LlamaCpp { .. } => Ok(build_llamacpp_headers()), 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::OpenAi { .. } => build_openai_headers(api_token, ide),
Backend::Tgi { .. } => build_tgi_headers(api_token, ide), Backend::Tgi { .. } => build_tgi_headers(api_token, ide),
} }