From: Michael Coppola <redacted>
Date: Tue, 23 Jan 2024 06:51:27 +0000 (-0500)
Subject: llama.vim : added api key support (#5090)
X-Git-Tag: upstream/0.0.4488~2532
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=125d03a5036a02a983c8e98c2cdc126e061afb8e;p=pkg%2Fggml%2Fsources%2Fllama.cpp

llama.vim : added api key support (#5090)

Co-authored-by: Michael Coppola <redacted>
---

diff --git a/examples/llama.vim b/examples/llama.vim
index f03fadfb..1b5ad6ba 100644
--- a/examples/llama.vim
+++ b/examples/llama.vim
@@ -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