diff options
Diffstat (limited to '')
| -rw-r--r-- | py/chat.py | 24 |
1 files changed, 10 insertions, 14 deletions
@@ -9,24 +9,20 @@ request_options = make_request_options() openai.api_key = load_api_key() -file_content = vim.eval('trim(join(getline(1, "$"), "\n"))') -initial_prompt = '\n'.join(options['initial_prompt']) -prompt = f"{initial_prompt}\n{file_content}" - -lines = prompt.splitlines() -messages = [] - -chat_content = vim.eval('trim(join(getline(1, "$"), "\n"))') -messages = parse_chat_messages(chat_content) - -if not messages: - # roles not found, put whole file content as an user prompt +lines = vim.eval('getline(1, "$")') +contains_user_prompt = any(line == '>>> user' for line in lines) +if not contains_user_prompt: + # user role not found, put whole file content as an user prompt vim.command("normal! ggO>>> user\n") vim.command("normal! G") vim.command("let &ul=&ul") # breaks undo sequence (https://vi.stackexchange.com/a/29087) vim.command("redraw") - chat_content = vim.eval('trim(join(getline(1, "$"), "\n"))') - messages = parse_chat_messages(chat_content) + +initial_prompt = options.get('initial_prompt', []) +initial_prompt = '\n'.join(initial_prompt) +file_content = vim.eval('trim(join(getline(1, "$"), "\n"))') +chat_content = f"{initial_prompt}\n{file_content}" +messages = parse_chat_messages(chat_content) try: if messages[-1]["content"].strip(): |