]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
llama.vim : added api key support (#5090)
authorMichael Coppola <redacted>
Tue, 23 Jan 2024 06:51:27 +0000 (01:51 -0500)
committerGitHub <redacted>
Tue, 23 Jan 2024 06:51:27 +0000 (08:51 +0200)
Co-authored-by: Michael Coppola <redacted>
examples/llama.vim

index f03fadfb7a017319eb6aee4bb3fef4acaec2f303..1b5ad6ba0f32f0c4e1c02019ec424ac9ab84b107 100644 (file)
@@ -6,7 +6,7 @@
 " Similarly, you could add an insert mode keybind with
 " inoremap <C-B> <Cmd>call llama#doLlamaGen()<CR>
 "
-" g:llama_api_url and g:llama_overrides can be configured in your .vimrc
+" g:llama_api_url, g:llama_api_key and g:llama_overrides can be configured in your .vimrc
 " let g:llama_api_url = "192.168.1.10:8080"
 " llama_overrides can also be set through buffer/window scopes. For instance
 " autocmd filetype python let b:llama_overrides = {"temp": 0.2}
@@ -82,6 +82,9 @@ func llama#doLlamaGen()
    endif
    let l:querydata.prompt = join(l:buflines, "\n")
    let l:curlcommand = copy(s:curlcommand)
+   if exists("g:llama_api_key")
+       call extend(l:curlcommand, ['--header', 'Authorization: Bearer ' .. g:llama_api_key])
+   endif
    let l:curlcommand[2] = json_encode(l:querydata)
    let b:job = job_start(l:curlcommand, {"callback": function("s:callbackHandler", [l:cbuffer])})
 endfunction