summaryrefslogtreecommitdiff
path: root/py/chat.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--py/chat.py15
1 files changed, 4 insertions, 11 deletions
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: