diff options
Diffstat (limited to 'py/complete.py')
| -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 d47eb2a..f469592 100644 --- a/py/complete.py +++ b/py/complete.py @@ -5,6 +5,7 @@ plugin_root = vim.eval("s:plugin_root") vim.command(f"py3file {plugin_root}/py/utils.py") prompt = vim.eval("prompt") +options = vim.eval("options") openai.api_key = load_api_key() @@ -15,12 +16,12 @@ try: vim.command("redraw") response = openai.Completion.create( - model="text-davinci-003", - prompt=prompt, - max_tokens=1000, - temperature=0.1, stream=True, - request_timeout=request_timeout_seconds, + prompt=prompt, + model=options['model'], + max_tokens=int(options['max_tokens']), + temperature=float(options['temperature']), + request_timeout=float(options['request_timeout']), ) generating_text = False |