diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2023-04-15 12:47:43 +0200 |
|---|---|---|
| committer | Martin Bielik <mx.bielik@gmail.com> | 2023-04-15 12:47:43 +0200 |
| commit | f69e39a0406999b6460295aa9f850b0444769080 (patch) | |
| tree | 5d092f9e754cf84c01bd7cb3718b8580a9fbe0db /py/complete.py | |
| parent | eb07c3615b926ce667bc273c03e8cbc8b85c6505 (diff) | |
| download | vim-ai-f69e39a0406999b6460295aa9f850b0444769080.tar.gz | |
reorganized request options
Diffstat (limited to '')
| -rw-r--r-- | py/complete.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/py/complete.py b/py/complete.py index 669ef98..1045720 100644 --- a/py/complete.py +++ b/py/complete.py @@ -4,7 +4,8 @@ vim.command(f"py3file {plugin_root}/py/utils.py") engine = vim.eval("l:engine") config_options = vim.eval("l:options") -request_options = make_request_options(config_options) +openai_options = make_openai_options(config_options) +http_options = make_http_options(config_options) prompt = vim.eval("l:prompt").strip() @@ -12,10 +13,10 @@ def complete_engine(prompt): request = { 'stream': True, 'prompt': prompt, - **request_options + **openai_options } printDebug("[engine-complete] request: {}", request) - response = openai_request('https://api.openai.com/v1/completions', request) + response = openai_request('https://api.openai.com/v1/completions', request, http_options) def map_chunk(resp): printDebug("[engine-complete] response: {}", resp) return resp['choices'][0].get('text', '') @@ -30,10 +31,10 @@ def chat_engine(prompt): request = { 'stream': True, 'messages': messages, - **request_options + **openai_options } printDebug("[engine-chat] request: {}", request) - response = openai_request('https://api.openai.com/v1/chat/completions', request) + response = openai_request('https://api.openai.com/v1/chat/completions', request, http_options) def map_chunk(resp): printDebug("[engine-chat] response: {}", resp) return resp['choices'][0]['delta'].get('content', '') |