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),
}
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),
}