vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

llama.vim : pass filenames for each chunk

Browse source 
ggml-ci
This commit is contained in:
Georgi Gerganov 2024-10-13 21:36:02 +03:00
parent 916c2ee3fd
commit ae76a092b8
No known key found for this signature in database
GPG key ID: BF970631944C16B7
1 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
examples/llama.vim
View file

@ -178,7 +178,7 @@ function! s:pick_chunk(text, no_mod, do_evict)
let l:chunk = a:text[l:l0:l:l1] let l:chunk = a:text[l:l0:l:l1]
endif endif
let l:chunk_str = join(l:chunk, "\n") let l:chunk_str = join(l:chunk, "\n") . "\n"
" check if this chunk is already added " check if this chunk is already added
let l:exist = v:false let l:exist = v:false
@ -209,12 +209,16 @@ function! s:pick_chunk(text, no_mod, do_evict)
call remove(s:ring_chunks, 0) call remove(s:ring_chunks, 0)
endif endif
call add(s:ring_chunks, {'data': l:chunk, 'str': l:chunk_str, 'time': reltime()}) call add(s:ring_chunks, {'data': l:chunk, 'str': l:chunk_str, 'time': reltime(), 'filename': expand('%')})
" send asynchronous job with the new extra context so that it is ready for the next FIM " send asynchronous job with the new extra context so that it is ready for the next FIM
let l:extra_context = [] let l:extra_context = []
for l:chunk in s:ring_chunks for l:chunk in s:ring_chunks
call add(l:extra_context, l:chunk.str) call add(l:extra_context, {
\ 'text': l:chunk.str,
\ 'time': l:chunk.time,
\ 'filename': l:chunk.filename
\ })
endfor endfor
let l:request = json_encode({ let l:request = json_encode({
@ -277,10 +281,14 @@ function! llama#fim(is_auto) abort
\ . join(l:lines_suffix, "\n") \ . join(l:lines_suffix, "\n")
\ . "\n" \ . "\n"
" array of strings " prepare the extra context data
let l:extra_context = [] let l:extra_context = []
for l:chunk in s:ring_chunks for l:chunk in s:ring_chunks
call add(l:extra_context, l:chunk.str) call add(l:extra_context, {
\ 'text': l:chunk.str,
\ 'time': l:chunk.time,
\ 'filename': l:chunk.filename
\ })
endfor endfor
let l:request = json_encode({ let l:request = json_encode({