summaryrefslogtreecommitdiff
path: root/py/complete.py
diff options
context:
space:
mode:
authorMartin Bielik <mx.bielik@gmail.com>2023-04-15 12:47:43 +0200
committerMartin Bielik <mx.bielik@gmail.com>2023-04-15 12:47:43 +0200
commitf69e39a0406999b6460295aa9f850b0444769080 (patch)
tree5d092f9e754cf84c01bd7cb3718b8580a9fbe0db /py/complete.py
parenteb07c3615b926ce667bc273c03e8cbc8b85c6505 (diff)
downloadvim-ai-f69e39a0406999b6460295aa9f850b0444769080.tar.gz
reorganized request options
Diffstat (limited to 'py/complete.py')
-rw-r--r--py/complete.py11
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', '')