From 08036fb76c437b705d916708d903c7c6c2eef0ba Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Wed, 22 Mar 2023 22:14:25 +0100 Subject: completion configuration --- py/chat.py | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) (limited to 'py/chat.py') diff --git a/py/chat.py b/py/chat.py index eb80e92..294ae63 100644 --- a/py/chat.py +++ b/py/chat.py @@ -4,11 +4,11 @@ import openai plugin_root = vim.eval("s:plugin_root") vim.command(f"py3file {plugin_root}/py/utils.py") -openai.api_key = load_api_key() - -options = vim.eval("options") +options = make_options() file_content = vim.eval('trim(join(getline(1, "$"), "\n"))') +openai.api_key = load_api_key() + lines = file_content.splitlines() messages = [] @@ -38,14 +38,7 @@ try: print('Answering...') vim.command("redraw") - response = openai.ChatCompletion.create( - messages=messages, - stream=True, - model=options['model'], - max_tokens=int(options['max_tokens']), - temperature=float(options['temperature']), - request_timeout=float(options['request_timeout']), - ) + response = openai.ChatCompletion.create(messages=messages, stream=True, **options) generating_text = False for resp in response: -- cgit v1.2.3