From f4118779b31e21b8cfb20ddf64e6fd355d4d02fc Mon Sep 17 00:00:00 2001
From: flu0r1ne <flu0r1ne@flu0r1ne.net>
Date: Sat, 6 May 2023 14:56:08 -0500
Subject: Add alternative file sources, specify system message

---
 src/gpt_chat_cli/argparsing.py | 52 ++++++++++++++++++++++++++++++++++++++----
 1 file changed, 48 insertions(+), 4 deletions(-)

(limited to 'src/gpt_chat_cli/argparsing.py')

diff --git a/src/gpt_chat_cli/argparsing.py b/src/gpt_chat_cli/argparsing.py
index 15826f2..dc7bf25 100644
--- a/src/gpt_chat_cli/argparsing.py
+++ b/src/gpt_chat_cli/argparsing.py
@@ -41,6 +41,9 @@ def validate_args(args: argparse.Namespace, debug : bool = False) -> None:
     if args.interactive and args.n_completions != 1:
         die_validation_err("Only a single completion can be used in interactive mode")
 
+    if (args.prompt_from_fd or args.prompt_from_file) and args.message:
+        die_validation_err("Cannot specify an initial message alongside --prompt_from_fd or --prompt_from_file")
+
     if debug and args.interactive:
 
         if args.interactive and (
@@ -68,6 +71,12 @@ class DebugArguments:
     save_response_to_file: Optional[str]
     load_response_from_file: Optional[str]
 
+@dataclass
+class MessageSource:
+    message: Optional[str] = None
+    prompt_from_fd: Optional[str] = None
+    prompt_from_file: Optional[str] = None
+
 @dataclass
 class Arguments:
     completion_args: CompletionArguments
@@ -75,7 +84,8 @@ class Arguments:
     version: bool
     list_models: bool
     interactive: bool
-    initial_message: Optional[str] = None
+    initial_message: MessageSource
+    system_message: Optional[str] = None
     debug_args: Optional[DebugArguments] = None
 
 def split_arguments(args: argparse.Namespace) -> Arguments:
@@ -89,6 +99,12 @@ def split_arguments(args: argparse.Namespace) -> Arguments:
         top_p=args.top_p,
     )
 
+    msg_src = MessageSource(
+        message = args.message,
+        prompt_from_fd = args.prompt_from_fd,
+        prompt_from_file = args.prompt_from_file,
+    )
+
     display_args = DisplayArguments(
         adornments=(args.adornments == AutoDetectedOption.ON),
         color=(args.color == AutoDetectedOption.ON),
@@ -100,13 +116,14 @@ def split_arguments(args: argparse.Namespace) -> Arguments:
     )
 
     return Arguments(
-           initial_message=args.message,
+           initial_message=msg_src,
            completion_args=completion_args,
            display_args=display_args,
            debug_args=debug_args,
            version=args.version,
            list_models=args.list_models,
-           interactive=args.interactive
+           interactive=args.interactive,
+           system_message=args.system_message
     )
 
 def parse_args() -> Arguments:
@@ -195,6 +212,13 @@ def parse_args() -> Arguments:
         help="How many chat completion choices to generate for each input message.",
     )
 
+    parser.add_argument(
+        "--system-message",
+        type=str,
+        default=os.getenv('f{GCLI_ENV_PREFIX}SYSTEM_MESSAGE'),
+        help="Specify an alternative system message.",
+    )
+
     parser.add_argument(
         "--adornments",
         type=AutoDetectedOption,
@@ -234,6 +258,20 @@ def parse_args() -> Arguments:
         help="Start an interactive session"
     )
 
+    initial_prompt = parser.add_mutually_exclusive_group()
+
+    initial_prompt.add_argument(
+        '--prompt-from-fd',
+        type=int,
+        help="Obtain the initial prompt from the specified file descriptor",
+    )
+
+    initial_prompt.add_argument(
+        '--prompt-from-file',
+        type=str,
+        help="Obtain the initial prompt from the specified file",
+    )
+
     parser.add_argument(
         "message",
         type=str,
@@ -286,7 +324,13 @@ def parse_args() -> Arguments:
         else:
             args.adornments = AutoDetectedOption.OFF
 
-    if args.message is None:
+    initial_message_specified = (
+        args.message or
+        args.prompt_from_fd or
+        args.prompt_from_file
+    )
+
+    if not initial_message_specified:
         if debug and args.load_response_from_file:
             args.interactive = False
         elif sys.stdin.isatty():
-- 
cgit v1.2.3