summaryrefslogtreecommitdiff
path: root/py/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'py/utils.py')
-rw-r--r--py/utils.py15
1 files changed, 9 insertions, 6 deletions
diff --git a/py/utils.py b/py/utils.py
index 814a4f0..02ecfef 100644
--- a/py/utils.py
+++ b/py/utils.py
@@ -11,9 +11,12 @@ def load_api_key():
pass
return api_key.strip()
-def make_options():
- options = {**vim.eval("options")}
- options['request_timeout'] = float(options['request_timeout'])
- options['temperature'] = float(options['temperature'])
- options['max_tokens'] = int(options['max_tokens'])
- return options
+def make_request_options():
+ options = vim.eval("options")
+ request_options = {}
+ request_options['model'] = options['model']
+ request_options['max_tokens'] = int(options['max_tokens'])
+ request_options['temperature'] = float(options['temperature'])
+ request_options['request_timeout'] = float(options['request_timeout'])
+ return request_options
+