gregory
/
.vim
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

lazygit

dev
Gregory Leeman 7 days ago
parent
fa18fad56d
commit
29af9e0ab9
3 changed files with 179 additions and 166 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 107
      plugin/gpt.vim
  2. 29
      python/gpt.py
  3. 9
      vimrc

107
plugin/gpt.vim
View File

@ -1,10 +1,15 @@
" ChatGPT Vim Plugin " gpt.vim
if !has('python3') if !has('python3')
echo "Python 3 support is required for ChatGPT plugin" echo "Python 3 support is required for ChatGPT plugin"
finish finish
endif endif
if !exists("g:gpt_openai_api_key")
echo "g:gpt_openai_api_key is not set. Please set it to your OpenAI API key."
finish
endif
if !exists("g:gpt_max_tokens") if !exists("g:gpt_max_tokens")
let g:gpt_max_tokens = 2000 let g:gpt_max_tokens = 2000
endif endif
@ -13,10 +18,6 @@ if !exists("g:gpt_temperature")
let g:gpt_temperature = 0.7 let g:gpt_temperature = 0.7
endif endif
if !exists("g:gpt_model")
let g:gpt_model = 'gpt-4o'
endif
if !exists("g:gpt_lang") if !exists("g:gpt_lang")
let g:gpt_lang = 'English' let g:gpt_lang = 'English'
endif endif
@ -33,23 +34,42 @@ if !exists("g:gpt_persona")
let g:gpt_persona = 'default' let g:gpt_persona = 'default'
endif endif
let g:gpt_templates = { let g:gpt_personas = {
\ 'Rewrite': 'Rewrite this more idiomatically', \ "default": '',
\ 'Review': 'Review this code', \ "bob": 'You are a helpful expert programmer we are working together to solve complex coding challenges, and I need your help. Please make sure to wrap all code blocks in ``` annotate the programming language you are using.',
\ 'Document': 'Return documentation following language pattern conventions',
\ 'Explain': 'Explain how this works',
\ 'Test': 'Write a test',
\ 'Fix': 'Fix this error',
\} \}
let g:gpt_keys = keys(g:gpt_templates) if !exists("g:gpt_model")
let g:gpt_model = 'gpt-4o'
endif
let g:gpt_personas = { let g:gpt_models = {
\ "default": '', \ "gpt-3.5-turbo": 4097,
\ "bob": 'You are a helpful expert programmer we are working together to solve complex coding challenges, and I need your help. Please make sure to wrap all code blocks in ``` annotate the programming language you are using.', \ "gpt-3.5-turbo-16k": 16385,
\ "gpt-3.5-turbo-1106": 16385,
\ "gpt-4": 8192,
\ "gpt-4-turbo": 128000,
\ "gpt-4-turbo-preview": 128000,
\ "gpt-4-32k": 32768,
\ "gpt-4o": 128000,
\ "gpt-4o-mini": 128000,
\} \}
function! DisplayChatGPTResponse(response, finish_reason, chat_gpt_session_id) " {{{ function! GptInfo() " {{{
echo 'model: ' . g:gpt_model
echo 'max_tokens: ' . string(g:gpt_max_tokens)
echo 'temperature: ' . string(g:gpt_temperature)
echo 'persona: ' . g:gpt_persona
echo 'language: ' . g:gpt_lang
echo 'split_direction: ' . g:gpt_split_direction
echo 'split_ratio: ' . string(g:gpt_split_ratio)
echo 'personas: ' . join(keys(g:gpt_personas), ', ')
echo 'models: ' . join(keys(g:gpt_models), ', ')
endfunction
" }}}
function! GptDisplay(response, finish_reason, chat_gpt_session_id) " {{{
let response = a:response let response = a:response
let finish_reason = a:finish_reason let finish_reason = a:finish_reason
@ -104,16 +124,15 @@ endfunction
" }}} " }}}
function! ChatGPT(prompt, persist) abort " {{{ function! Gpt(prompt, persist) abort " {{{
" echo 'prompt: ' . a:prompt " echo 'prompt: ' . a:prompt
" echo 'persist: ' . a:persist " echo 'persist: ' . a:persist
silent py3file ~/.vim/python/gpt.py silent py3file ~/.vim/python/gpt.py
endfunction endfunction
" }}} " }}}
function! SendToChatGPT(prompt, bang) abort " {{{ function! GptSend(prompt, bang) abort " {{{
let persist = (a:bang ==# '!') ? 1 : 0 let persist = (a:bang ==# '!') ? 1 : 0
@ -142,15 +161,9 @@ function! SendToChatGPT(prompt, bang) abort " {{{
let snippet = '' let snippet = ''
endif endif
if has_key(g:gpt_templates, a:prompt) let prompt = a:prompt . snippet
let prompt = g:gpt_templates[a:prompt]
else
let prompt = a:prompt
endif
let prompt = prompt . snippet
call ChatGPT(prompt, persist) call Gpt(prompt, persist)
let @@ = save_reg let @@ = save_reg
call setreg('@', save_reg, save_regtype) call setreg('@', save_reg, save_regtype)
@ -159,18 +172,19 @@ function! SendToChatGPT(prompt, bang) abort " {{{
call setpos("'<", curpos) call setpos("'<", curpos)
call setpos("'>", curpos) call setpos("'>", curpos)
call setpos('.', save_cursor) call setpos('.', save_cursor)
endfunction endfunction
" }}} " }}}
command! -range -bang -nargs=1 Gpt call SendToChatGPT(<q-args>, '<bang>') function! GptPersona(persona=0) " {{{
if a:persona == 0
for i in range(len(g:gpt_keys)) echo 'Personas: ' . join(keys(g:gpt_personas), ', ')
execute "command! -range -bang -nargs=0 " . g:gpt_keys[i] . " call SendToChatGPT('" . g:gpt_keys[i] . "', '<bang>')" return
endfor endif
function! SetPersona(persona) " {{{
let personas = keys(g:gpt_personas) let personas = keys(g:gpt_personas)
if index(personas, a:persona) != -1 if index(personas, a:persona) != -1
echo 'Persona set to: ' . a:persona echo 'Persona set to: ' . a:persona
let g:gpt_persona = a:persona let g:gpt_persona = a:persona
@ -178,8 +192,31 @@ function! SetPersona(persona) " {{{
let g:gpt_persona = 'default' let g:gpt_persona = 'default'
echo a:persona . ' is not a valid persona. Defaulting to: ' . g:gpt_persona echo a:persona . ' is not a valid persona. Defaulting to: ' . g:gpt_persona
end end
endfunction
" }}}
function! GptModel(model=0) " {{{
if a:model == 0
echo 'Models: ' . join(keys(g:gpt_models), ', ')
return
endif
let models = keys(g:gpt_models)
if index(models, a:model) != -1
echo 'model set to: ' . a:model
let g:gpt_model = a:model
else
let g:gpt_model = 'gpt-4o'
echo a:model . ' is not a valid model. Defaulting to: ' . g:gpt_model
end
endfunction endfunction
" }}} " }}}
command! -nargs=1 Persona call SetPersona(<q-args>) command! -nargs=0 GptInfo call GptInfo()
command! -nargs=1 -range -bang Gpt call GptSend(<q-args>, '<bang>')
command! -nargs=? GptPersona call GptPersona(<q-args>)
command! -nargs=? GptModel call GptModel(<q-args>)

29
python/gpt.py
View File

@ -3,7 +3,7 @@ import vim
import os import os
try: try:
from openai import AzureOpenAI, OpenAI from openai import OpenAI
except ImportError: except ImportError:
print("Error: openai module not found. Please install with Pip and ensure equality of the versions given by :!python3 -V, and :python3 import sys; print(sys.version)") print("Error: openai module not found. Please install with Pip and ensure equality of the versions given by :!python3 -V, and :python3 import sys; print(sys.version)")
raise raise
@ -15,23 +15,8 @@ def safe_vim_eval(expression):
return None return None
def create_client(): def create_client():
api_type = safe_vim_eval('g:gpt_api_type')
api_key = os.getenv('OPENAI_API_KEY') or safe_vim_eval('g:gpt_key') or safe_vim_eval('g:gpt_openai_api_key') api_key = os.getenv('OPENAI_API_KEY') or safe_vim_eval('g:gpt_key') or safe_vim_eval('g:gpt_openai_api_key')
openai_base_url = os.getenv('OPENAI_PROXY') or os.getenv('OPENAI_API_BASE') or safe_vim_eval('g:gpt_openai_base_url') openai_base_url = os.getenv('OPENAI_PROXY') or os.getenv('OPENAI_API_BASE') or safe_vim_eval('g:gpt_openai_base_url')
if api_type == 'azure':
azure_endpoint = safe_vim_eval('g:gpt_azure_endpoint')
azure_api_version = safe_vim_eval('g:gpt_azure_api_version')
azure_deployment = safe_vim_eval('g:gpt_azure_deployment')
assert azure_endpoint and azure_api_version and azure_deployment, "azure_endpoint, azure_api_version and azure_deployment not set property, please check your settings in `vimrc` or `enviroment`."
assert api_key, "api_key not set, please configure your `openai_api_key` in your `vimrc` or `enviroment`"
client = AzureOpenAI(
azure_endpoint=azure_endpoint,
azure_deployment=azure_deployment,
api_key=api_key,
api_version=azure_api_version,
)
else:
client = OpenAI( client = OpenAI(
base_url=openai_base_url, base_url=openai_base_url,
api_key=api_key, api_key=api_key,
@ -39,7 +24,7 @@ def create_client():
return client return client
def chat_gpt(prompt, persist=0): def chat_gpt(prompt, persist=0)
token_limits = { token_limits = {
"gpt-3.5-turbo": 4097, "gpt-3.5-turbo": 4097,
"gpt-3.5-turbo-16k": 16385, "gpt-3.5-turbo-16k": 16385,
@ -54,6 +39,7 @@ def chat_gpt(prompt, persist=0):
max_tokens = int(vim.eval('g:gpt_max_tokens')) max_tokens = int(vim.eval('g:gpt_max_tokens'))
model = str(vim.eval('g:gpt_model')) model = str(vim.eval('g:gpt_model'))
token_limit = vim.eval('g:gpt_models[g:gpt_model]')
temperature = float(vim.eval('g:gpt_temperature')) temperature = float(vim.eval('g:gpt_temperature'))
lang = str(vim.eval('g:gpt_lang')) lang = str(vim.eval('g:gpt_lang'))
resp = f" And respond in {lang}." if lang != 'None' else "" resp = f" And respond in {lang}." if lang != 'None' else ""
@ -80,7 +66,7 @@ def chat_gpt(prompt, persist=0):
history.reverse() history.reverse()
# Adding messages to history until token limit is reached # Adding messages to history until token limit is reached
token_count = token_limits.get(model, 4097) - max_tokens - len(prompt) - len(str(systemCtx)) token_count = token_limit - max_tokens - len(prompt) - len(str(systemCtx))
for line in history: for line in history:
if line.startswith("USER\n"): if line.startswith("USER\n"):
@ -104,7 +90,7 @@ def chat_gpt(prompt, persist=0):
content += '# GPT' content += '# GPT'
content += '\n\n>>> USER\n' + prompt + '\n\n>>> ASSISTANT\n'.replace("'", "''") content += '\n\n>>> USER\n' + prompt + '\n\n>>> ASSISTANT\n'.replace("'", "''")
vim.command("call DisplayChatGPTResponse('{0}', '', '{1}')".format(content.replace("'", "''"), session_id)) vim.command("call GptDisplay('{0}', '', '{1}')".format(content.replace("'", "''"), session_id))
vim.command("redraw") vim.command("redraw")
messages.append({"role": "user", "content": prompt}) messages.append({"role": "user", "content": prompt})
@ -131,12 +117,11 @@ def chat_gpt(prompt, persist=0):
choice = chunk.choices[0] choice = chunk.choices[0]
finish_reason = choice.finish_reason finish_reason = choice.finish_reason
# Call DisplayChatGPTResponse with the finish_reason or content
if finish_reason: if finish_reason:
vim.command("call DisplayChatGPTResponse('', '{0}', '{1}')".format(finish_reason.replace("'", "''"), chunk_session_id)) vim.command("call GptDisplay('', '{0}', '{1}')".format(finish_reason.replace("'", "''"), chunk_session_id))
elif choice.delta: elif choice.delta:
content = choice.delta.content content = choice.delta.content
vim.command("call DisplayChatGPTResponse('{0}', '', '{1}')".format(content.replace("'", "''"), chunk_session_id)) vim.command("call GptDisplay('{0}', '', '{1}')".format(content.replace("'", "''"), chunk_session_id))
vim.command("redraw") vim.command("redraw")
except Exception as e: except Exception as e:

9
vimrc
View File

@ -153,13 +153,4 @@ set completeopt=noinsert,menuone,noselect
let g:slime_target = "tmux" let g:slime_target = "tmux"
let g:slime_default_config = {"socket_name": get(split($TMUX, ","), 0), "target_pane": ":.2"} let g:slime_default_config = {"socket_name": get(split($TMUX, ","), 0), "target_pane": ":.2"}
let g:gpt_openai_api_key='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' let g:gpt_openai_api_key='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'
let g:gpt_max_tokens=2000
let g:gpt_model='gpt-4o'
let g:gpt_temperature = 0.7
let g:gpt_lang = 'English'
" let g:chat_gpt_split_direction = 'vertical'
" let g:split_ratio=4
vmap <silent> <leader>0 <Plug>(chatgpt-menu)

Write Preview
Loading…
Cancel
Save