From 77840611b909b87abc92ca7d57a2955ce952352f Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Wed, 5 Apr 2023 19:14:08 +0200 Subject: passing prompt as param --- py/complete.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/py/complete.py b/py/complete.py index 2f151d3..7f78661 100644 --- a/py/complete.py +++ b/py/complete.py @@ -12,12 +12,12 @@ prompt = vim.eval("prompt").strip() openai.api_key = load_api_key() -def complete_engine(): +def complete_engine(prompt): response = openai.Completion.create(stream=True, prompt=prompt, **request_options) text_chunks = map(lambda resp: resp['choices'][0].get('text', ''), response) return text_chunks -def chat_engine(): +def chat_engine(prompt): initial_prompt = options.get('initial_prompt', []) initial_prompt = '\n'.join(initial_prompt) chat_content = f"{initial_prompt}\n\n>>> user\n\n{prompt}".strip() @@ -32,7 +32,7 @@ try: if prompt: print('Completing...') vim.command("redraw") - text_chunks = engines[engine]() + text_chunks = engines[engine](prompt) render_text_chunks(text_chunks) except KeyboardInterrupt: vim.command("normal! a Ctrl-C...") -- cgit v1.2.3