From 89c77af2d93c8ed3f3f452fff709d91f9228f9dc Mon Sep 17 00:00:00 2001 From: flu0r1ne Date: Thu, 11 May 2023 02:29:52 -0500 Subject: Fix printing when provided initial messages --- src/gpt_chat_cli/cmd.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'src/gpt_chat_cli/cmd.py') diff --git a/src/gpt_chat_cli/cmd.py b/src/gpt_chat_cli/cmd.py index 899c705..83cd298 100644 --- a/src/gpt_chat_cli/cmd.py +++ b/src/gpt_chat_cli/cmd.py @@ -297,10 +297,10 @@ def interactive(args : Arguments): initial_message = resolve_initial_message(args.initial_message, interactive=True) - if initial_message: - print( PROMPT, initial_message, sep='', flush=True ) - with prompter as prompt: + if initial_message: + print( prompt.prompt, initial_message, sep='', flush=True ) + while True: try: if initial_message: @@ -339,6 +339,7 @@ def interactive(args : Arguments): hist.append( ChatMessage(Role.ASSISTANT, response) ) except KeyboardInterrupt: # Skip to next prompt + print() continue except EOFError: # Exit on Control-D print() -- cgit v1.2.3