ngpt 2.12.0__py3-none-any.whl → 2.14.0__py3-none-any.whl

ngpt/cli/args.py

@@ -88,6 +88,19 @@ def setup_argument_parser():
     global_group.add_argument('--renderer', choices=['auto', 'rich', 'glow'], default='auto',
                       help='Select which markdown renderer to use with --prettify (auto, rich, or glow)')
     
+    # GitCommit message options
+    gitcommsg_group = parser.add_argument_group('Git Commit Message Options')
+    gitcommsg_group.add_argument('-m', '--message-context', 
+                      help='Context to guide AI generation (e.g., file types, commit type)')
+    gitcommsg_group.add_argument('-r', '--recursive-chunk', action='store_true',
+                      help='Process large diffs in chunks with recursive analysis if needed')
+    gitcommsg_group.add_argument('--diff', metavar='FILE',
+                      help='Use diff from specified file instead of staged changes')
+    gitcommsg_group.add_argument('--chunk-size', type=int, default=200,
+                      help='Number of lines per chunk when chunking is enabled (default: 200)')
+    gitcommsg_group.add_argument('--max-depth', type=int, default=3,
+                      help='Maximum recursion depth for recursive chunking (default: 3)')
+    
     # Mode flags (mutually exclusive)
     mode_group = parser.add_argument_group('Modes (mutually exclusive)')
     mode_exclusive_group = mode_group.add_mutually_exclusive_group()
@@ -101,6 +114,10 @@ def setup_argument_parser():
                                       help='Enter multi-line text input (submit with Ctrl+D)')
     mode_exclusive_group.add_argument('--stdin', action='store_true',
                                       help='Read from stdin and use content with prompt. Use {} in prompt as placeholder for stdin content')
+    mode_exclusive_group.add_argument('--rewrite', action='store_true',
+                                      help='Rewrite text from stdin to be more natural while preserving tone and meaning')
+    mode_exclusive_group.add_argument('--gitcommsg', action='store_true',
+                                      help='Generate AI-powered git commit messages from staged changes or diff file')
     
     return parser
 

ngpt/cli/formatters.py

@@ -13,6 +13,7 @@ COLORS = {
     "cyan": "\033[36m",
     "green": "\033[32m", 
     "yellow": "\033[33m",
+    "red": "\033[31m",
     "blue": "\033[34m",
     "magenta": "\033[35m",
     "gray": "\033[90m",

ngpt/cli/main.py

@@ -23,6 +23,8 @@ from .modes.chat import chat_mode
 from .modes.code import code_mode
 from .modes.shell import shell_mode
 from .modes.text import text_mode
+from .modes.rewrite import rewrite_mode
+from .modes.gitcommsg import gitcommsg_mode
 from .args import parse_args, validate_args, handle_cli_config_args, setup_argument_parser, validate_markdown_renderer
 
 def show_cli_config_help():
@@ -226,15 +228,17 @@ def main():
             # Change log to True to create a temp file
             args.log = True
         
-        # If --log is True, it means it was used without a path value
-        log_path = None if args.log is True else args.log
-        logger = create_logger(log_path)
-        if logger:
-            logger.open()
-            print(f"{COLORS['green']}Logging session to: {logger.get_log_path()}{COLORS['reset']}")
-            # If it's a temporary log file, inform the user
-            if logger.is_temporary():
-                print(f"{COLORS['green']}Created temporary log file.{COLORS['reset']}")
+        # Skip logger initialization for gitcommsg mode as it creates its own logger
+        if not args.gitcommsg:
+            # If --log is True, it means it was used without a path value
+            log_path = None if args.log is True else args.log
+            logger = create_logger(log_path)
+            if logger:
+                logger.open()
+                print(f"{COLORS['green']}Logging session to: {logger.get_log_path()}{COLORS['reset']}")
+                # If it's a temporary log file, inform the user
+                if logger.is_temporary():
+                    print(f"{COLORS['green']}Created temporary log file.{COLORS['reset']}")
     
     # Priority order for config selection:
     # 1. Command-line arguments (args.provider, args.config_index)
@@ -460,7 +464,7 @@ def main():
         return
     
     # For interactive mode, we'll allow continuing without a specific prompt
-    if not args.prompt and not (args.shell or args.code or args.text or args.interactive or args.show_config or args.list_models):
+    if not args.prompt and not (args.shell or args.code or args.text or args.interactive or args.show_config or args.list_models or args.rewrite or args.gitcommsg):
         # Simply use the parser's help
         parser = setup_argument_parser()
         parser.print_help()
@@ -549,6 +553,20 @@ def main():
             # Stdin mode (using the chat mode with stdin input)
             chat_mode(client, args, logger=logger)
         
+        elif args.rewrite:
+            # Apply CLI config for rewrite mode
+            args = apply_cli_config(args, "all")
+            
+            # Rewrite mode (process stdin)
+            rewrite_mode(client, args, logger=logger)
+        
+        elif args.gitcommsg:
+            # Apply CLI config for gitcommsg mode
+            args = apply_cli_config(args, "all")
+            
+            # Git commit message generation mode
+            gitcommsg_mode(client, args, logger=logger)
+        
         else:
             # Default to chat mode
             # Apply CLI config for default chat mode

ngpt/cli/modes/__init__.py

@@ -2,5 +2,7 @@ from .chat import chat_mode
 from .code import code_mode
 from .shell import shell_mode
 from .text import text_mode
+from .rewrite import rewrite_mode
+from .gitcommsg import gitcommsg_mode
 
-__all__ = ['chat_mode', 'code_mode', 'shell_mode', 'text_mode']
+__all__ = ['chat_mode', 'code_mode', 'shell_mode', 'text_mode', 'rewrite_mode', 'gitcommsg_mode']