aboutsummaryrefslogtreecommitdiff
path: root/src/gpt_chat_cli/argparsing.py
diff options
context:
space:
mode:
authorflu0r1ne <flu0r1ne@flu0r1ne.net>2023-05-06 14:56:08 -0500
committerflu0r1ne <flu0r1ne@flu0r1ne.net>2023-05-06 14:56:08 -0500
commitf4118779b31e21b8cfb20ddf64e6fd355d4d02fc (patch)
tree56cc6c236ab38dd6ae35701dad152b14a8448007 /src/gpt_chat_cli/argparsing.py
parent42f7e201af301696066a792832e3b47c5597a6fc (diff)
downloadgpt-chat-cli-f4118779b31e21b8cfb20ddf64e6fd355d4d02fc.tar.xz
gpt-chat-cli-f4118779b31e21b8cfb20ddf64e6fd355d4d02fc.zip
Add alternative file sources, specify system message
Diffstat (limited to 'src/gpt_chat_cli/argparsing.py')
-rw-r--r--src/gpt_chat_cli/argparsing.py52
1 files changed, 48 insertions, 4 deletions
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 (
@@ -69,13 +72,20 @@ class DebugArguments:
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
display_args: DisplayArguments
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:
@@ -196,6 +213,13 @@ def parse_args() -> Arguments:
)
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,
choices=list(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():