blob: d7040a3a82cd5870426a22a079bee6e9fe4f9174 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
import vim
# import utils
plugin_root = vim.eval("s:plugin_root")
vim.command(f"py3file {plugin_root}/py/utils.py")
prompt, config = load_config_and_prompt()
config_options = config['options']
config_ui = config['ui']
engine = config['engine']
is_selection = vim.eval("l:is_selection")
def complete_engine(prompt):
openai_options = make_openai_options(config_options)
http_options = make_http_options(config_options)
request = {
'prompt': prompt,
**openai_options
}
printDebug("[engine-complete] request: {}", request)
url = config_options['endpoint_url']
response = openai_request(url, request, http_options)
def map_chunk(resp):
printDebug("[engine-complete] response: {}", resp)
return resp['choices'][0].get('text', '')
text_chunks = map(map_chunk, response)
return text_chunks
def chat_engine(prompt):
initial_prompt = config_options.get('initial_prompt', [])
initial_prompt = '\n'.join(initial_prompt)
chat_content = f"{initial_prompt}\n\n>>> user\n\n{prompt}".strip()
messages = parse_chat_messages(chat_content)
return make_chat_text_chunks(messages, config_options)
engines = {"chat": chat_engine, "complete": complete_engine}
try:
if prompt:
print('Completing...')
vim.command("redraw")
text_chunks = engines[engine](prompt)
render_text_chunks(text_chunks, is_selection)
clear_echo_message()
except BaseException as error:
handle_completion_error(error)
printDebug("[complete] error: {}", traceback.format_exc())
|
