diff options
author | flu0r1ne <flu0r1ne@flu0r1ne.net> | 2023-05-05 01:50:56 -0500 |
---|---|---|
committer | flu0r1ne <flu0r1ne@flu0r1ne.net> | 2023-05-05 01:50:56 -0500 |
commit | 439e935ec583eb7641152c32798898024859e939 (patch) | |
tree | 90850537f254c2d5c23a04ff2ec7ce12790eb79e /src/gpt_chat_cli/gcli.py | |
parent | 69654841fd272a66403031cd6e9311423f9f117b (diff) | |
download | gpt-chat-cli-439e935ec583eb7641152c32798898024859e939.tar.xz gpt-chat-cli-439e935ec583eb7641152c32798898024859e939.zip |
Add stdin as a input source for the inital prompt
Diffstat (limited to 'src/gpt_chat_cli/gcli.py')
-rw-r--r-- | src/gpt_chat_cli/gcli.py | 27 |
1 files changed, 20 insertions, 7 deletions
diff --git a/src/gpt_chat_cli/gcli.py b/src/gpt_chat_cli/gcli.py index dcd7c8b..f0b754e 100644 --- a/src/gpt_chat_cli/gcli.py +++ b/src/gpt_chat_cli/gcli.py @@ -20,8 +20,10 @@ from .argparsing import ( Arguments, DisplayArguments, CompletionArguments, + DebugArguments, ) +from .version import VERSION from .color import get_color_codes ########################### @@ -132,17 +134,28 @@ def print_streamed_response( def main(): args = parse_args() + if args.version: + print(f'version {VERSION}') + sys.exit(0) + completion_args = args.completion_args + COLOR_CODE = get_color_codes(no_color = not args.display_args.color) - if args.debug_args: - debug_args : DebugArguments = args.debug_args + debug_args : DebugArguments = args.debug_args - if debug_args.save_response_to_file: - save_response_and_arguments(args) - return - elif debug_args.load_response_from_file: - completion_args, completion = load_response_and_arguments(args) + if debug_args.save_response_to_file: + save_response_and_arguments(args) + return + elif debug_args.load_response_from_file: + completion_args, completion = load_response_and_arguments(args) else: + if args.completion_args.message is None: + if sys.stdin.isatty(): + print(f'GPT Chat CLI {VERSION}') + print(f'[{COLOR_CODE.WHITE}#{COLOR_CODE.RESET}]', end=' ', flush=True) + + completion_args.message = sys.stdin.read() + completion = create_chat_completion_from_args(completion_args) print_streamed_response( |