Mirrors/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-27 06:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

llama.vim : fix info text display [no ci] (#9787)

Browse Source
This commit is contained in:
Georgi Gerganov 2024-10-22 00:35:25 +03:00
parent e01c67affe
commit e94a138d64
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 7 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
examples/llama.vim
View File

@ -482,11 +482,9 @@ function! llama#fim_cancel()
" clear the virtual text " clear the virtual text
let l:bufnr = bufnr('%') let l:bufnr = bufnr('%')
let l:id_vt_fim = nvim_create_namespace('vt_fim') let l:id_vt_fim = nvim_create_namespace('vt_fim')
let l:id_vt_info = nvim_create_namespace('vt_info')
call nvim_buf_clear_namespace(l:bufnr, l:id_vt_fim, 0, -1) call nvim_buf_clear_namespace(l:bufnr, l:id_vt_fim, 0, -1)
call nvim_buf_clear_namespace(l:bufnr, l:id_vt_info, 0, -1)
" remove the mappings " remove the mappings
silent! iunmap <buffer> <Tab> silent! iunmap <buffer> <Tab>
@ -644,13 +642,11 @@ function! s:fim_on_stdout(job_id, data, event) dict
" display virtual text with the suggestion " display virtual text with the suggestion
let l:bufnr = bufnr('%') let l:bufnr = bufnr('%')
let l:id_vt_fim = nvim_create_namespace('vt_fim') let l:id_vt_fim = nvim_create_namespace('vt_fim')
let l:id_vt_info = nvim_create_namespace('vt_info')
" construct the info message " construct the info message
if g:llama_config.show_info > 0 && l:has_info if g:llama_config.show_info > 0 && l:has_info
" prefix the info string with whitespace in order to offset it to the right of the fim overlay let l:prefix = ' '
let l:prefix = repeat(' ', len(s:content[0]) - len(s:line_cur_suffix) + 3)
if l:truncated if l:truncated
let l:info = printf("%s | WARNING: the context is full: %d / %d, increase the server context size or reduce g:llama_config.ring_n_chunks", let l:info = printf("%s | WARNING: the context is full: %d / %d, increase the server context size or reduce g:llama_config.ring_n_chunks",
@ -668,21 +664,15 @@ function! s:fim_on_stdout(job_id, data, event) dict
endif endif
if g:llama_config.show_info == 1 if g:llama_config.show_info == 1
" display it in the statusline " display the info in the statusline
let &statusline = l:info let &statusline = l:info
elseif g:llama_config.show_info == 2 let l:info = ''
" display it to the right of the current line
call nvim_buf_set_extmark(l:bufnr, l:id_vt_info, s:pos_y - 1, s:pos_x - 1, {
\ 'virt_text': [[l:info, 'llama_hl_info']],
"\ 'virt_text_pos': 'eol',
\ 'virt_text_pos': 'right_align',
\ })
endif endif
endif endif
" display the suggestion " display the suggestion and append the info to the end of the first line
call nvim_buf_set_extmark(l:bufnr, l:id_vt_fim, s:pos_y - 1, s:pos_x - 1, { call nvim_buf_set_extmark(l:bufnr, l:id_vt_fim, s:pos_y - 1, s:pos_x - 1, {
\ 'virt_text': [[s:content[0], 'llama_hl_hint']], \ 'virt_text': [[s:content[0], 'llama_hl_hint'], [l:info, 'llama_hl_info']],
\ 'virt_text_win_col': virtcol('.') - 1 \ 'virt_text_win_col': virtcol('.') - 1
\ }) \ })