From f69e39a0406999b6460295aa9f850b0444769080 Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Sat, 15 Apr 2023 12:47:43 +0200 Subject: reorganized request options --- py/complete.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'py/complete.py') 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', '') -- cgit v1.2.3