ngpt 4.0.1__py3-none-any.whl → 4.1.0__py3-none-any.whl

ngpt/cli/args.py

@@ -144,6 +144,10 @@ def setup_argument_parser():
     mode_exclusive_group.add_argument('-g', '--gitcommsg', action='store_true',
                                       help='Generate AI-powered git commit messages from staged changes or diff file')
     
+    # Add positional argument for the prompt (optional)
+    parser.add_argument('prompt', nargs='?', default=None,
+                        help='The prompt to send to the language model')
+    
     return parser
 
 def parse_args():

ngpt/cli/modes/interactive.py

@@ -4,6 +4,9 @@ import traceback
 import threading
 import sys
 import time
+import json
+from datetime import datetime
+from ...utils.config import get_config_dir
 from ..formatters import COLORS
 from ..renderers import prettify_markdown, prettify_streaming_markdown, TERMINAL_RENDER_LOCK
 from ..ui import spinner, get_multiline_input
@@ -50,21 +53,29 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
     # Create a separator line - use a consistent separator length for all lines
     separator_length = min(40, term_width - 10)
     separator = f"{COLORS['gray']}{'─' * separator_length}{COLORS['reset']}"
-    print(separator)
-    
-    # Group commands into categories with better formatting
-    print(f"\n{COLORS['cyan']}Navigation:{COLORS['reset']}")
-    print(f"  {COLORS['yellow']}↑/↓{COLORS['reset']} : Browse input history")
-    
-    print(f"\n{COLORS['cyan']}Session Commands:{COLORS['reset']}")
-    print(f"  {COLORS['yellow']}history{COLORS['reset']} : Show conversation history")
-    print(f"  {COLORS['yellow']}clear{COLORS['reset']}   : Reset conversation")
-    print(f"  {COLORS['yellow']}exit{COLORS['reset']}    : End session")
-    
-    if multiline_enabled:
-        print(f"  {COLORS['yellow']}ml{COLORS['reset']}      : Open multiline editor")
-    
-    print(f"\n{separator}\n")
+
+    def show_help():
+        """Displays the help menu."""
+        print(separator)
+        # Group commands into categories with better formatting
+        print(f"\n{COLORS['cyan']}Navigation:{COLORS['reset']}")
+        print(f"  {COLORS['yellow']}↑/↓{COLORS['reset']} : Browse input history")
+        
+        print(f"\n{COLORS['cyan']}Session Commands:{COLORS['reset']}")
+        print(f"  {COLORS['yellow']}history{COLORS['reset']} : Show conversation history")
+        print(f"  {COLORS['yellow']}clear{COLORS['reset']}   : Reset conversation")
+        print(f"  {COLORS['yellow']}exit{COLORS['reset']}    : End session")
+        print(f"  {COLORS['yellow']}save{COLORS['reset']}    : Save current session")
+        print(f"  {COLORS['yellow']}load{COLORS['reset']}    : Load a previous session")
+        print(f"  {COLORS['yellow']}sessions{COLORS['reset']}: List saved sessions")
+        print(f"  {COLORS['yellow']}help{COLORS['reset']}    : Show this help message")
+        
+        if multiline_enabled:
+            print(f"  {COLORS['yellow']}ml{COLORS['reset']}      : Open multiline editor")
+        
+        print(f"\n{separator}\n")
+
+    show_help()
     
     # Show logging info if logger is available
     if logger:
@@ -148,6 +159,71 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
             print(f"\n{COLORS['yellow']}Conversation history cleared.{COLORS['reset']}")
             print(separator)  # Add separator for consistency
     
+    # --- Session Management Functions ---
+
+    def get_history_dir():
+        """Get the history directory, creating it if it doesn't exist."""
+        history_dir = get_config_dir() / "history"
+        history_dir.mkdir(parents=True, exist_ok=True)
+        return history_dir
+
+    def save_session():
+        """Save the current conversation to a JSON file."""
+        history_dir = get_history_dir()
+        timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+        filename = history_dir / f"session_{timestamp}.json"
+        
+        with open(filename, "w") as f:
+            json.dump(conversation, f, indent=2)
+        
+        print(f"\n{COLORS['green']}Session saved to {filename}{COLORS['reset']}")
+
+    def list_sessions():
+        """List all saved sessions."""
+        history_dir = get_history_dir()
+        sessions = sorted(history_dir.glob("session_*.json"), reverse=True)
+        
+        if not sessions:
+            print(f"\n{COLORS['yellow']}No saved sessions found.{COLORS['reset']}")
+            return
+            
+        print(f"\n{COLORS['cyan']}{COLORS['bold']}Saved Sessions:{COLORS['reset']}")
+        for i, session_file in enumerate(sessions):
+            print(f"  [{i}] {session_file.name}")
+
+    def load_session():
+        """Load a conversation from a saved session file."""
+        nonlocal conversation
+        history_dir = get_history_dir()
+        sessions = sorted(history_dir.glob("session_*.json"), reverse=True)
+
+        if not sessions:
+            print(f"\n{COLORS['yellow']}No saved sessions to load.{COLORS['reset']}")
+            return
+
+        list_sessions()
+        
+        try:
+            choice = input("Enter the number of the session to load: ")
+            choice_index = int(choice)
+            
+            if 0 <= choice_index < len(sessions):
+                session_file = sessions[choice_index]
+                with open(session_file, "r") as f:
+                    loaded_conversation = json.load(f)
+                
+                # Basic validation
+                if isinstance(loaded_conversation, list) and all(isinstance(item, dict) for item in loaded_conversation):
+                    conversation = loaded_conversation
+                    print(f"\n{COLORS['green']}Session loaded from {session_file.name}{COLORS['reset']}")
+                    display_history()
+                else:
+                    print(f"\n{COLORS['red']}Error: Invalid session file format.{COLORS['reset']}")
+            else:
+                print(f"\n{COLORS['red']}Error: Invalid selection.{COLORS['reset']}")
+        except (ValueError, IndexError):
+            print(f"\n{COLORS['red']}Error: Invalid input. Please enter a number from the list.{COLORS['reset']}")
+
     try:
         while True:
             # Get user input
@@ -188,6 +264,22 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
             if user_input.lower() == 'clear':
                 clear_history()
                 continue
+            
+            if user_input.lower() == 'save':
+                save_session()
+                continue
+
+            if user_input.lower() == 'sessions':
+                list_sessions()
+                continue
+
+            if user_input.lower() == 'load':
+                load_session()
+                continue
+
+            if user_input.lower() == 'help':
+                show_help()
+                continue
                 
             if multiline_enabled and user_input.lower() == 'ml':
                 print(f"{COLORS['cyan']}Opening multiline editor. Press Ctrl+D to submit.{COLORS['reset']}")

{ngpt-4.0.1.dist-info → ngpt-4.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 4.0.1
+Version: 4.1.0
 Summary: A Swiss army knife for LLMs: A fast, lightweight CLI and interactive chat tool that brings the power of any OpenAI-compatible LLM (OpenAI, Ollama, Groq, Claude, Gemini, etc.) straight to your terminal. rewrite texts or refine code, craft git commit messages, generate and run OS-aware shell commands.
 Project-URL: Homepage, https://github.com/nazdridoy/ngpt
 Project-URL: Repository, https://github.com/nazdridoy/ngpt
@@ -81,6 +81,7 @@ Description-Content-Type: text/markdown
 - 🎭 **System Prompts**: Customize model behavior with custom system prompts
 - 🤖 **Custom Roles**: Create and use reusable AI roles for specialized tasks
 - 📃 **Conversation Logging**: Save your conversations to text files for later reference
+- 💾 **Session Management**: Save, load, and list interactive chat sessions.
 - 🔌 **Modular Architecture**: Well-structured codebase with clean separation of concerns
 - 🔄 **Provider Switching**: Easily switch between different LLM providers with a single parameter
 - 🚀 **Performance Optimized**: Fast response times and minimal resource usage
@@ -138,6 +139,13 @@ python -m ngpt "Tell me about quantum computing"
 
 # Start an interactive chat session with conversation memory
 ngpt -i
+# Inside interactive mode, you can use commands like:
+# help     - Show help menu
+# save     - Save the current session
+# load     - Load a previous session
+# sessions - List saved sessions
+# clear    - Clear the conversation
+# exit     - Exit the session
 
 # Return response without streaming
 ngpt --no-stream "Tell me about quantum computing"

{ngpt-4.0.1.dist-info → ngpt-4.1.0.dist-info}/RECORD

@@ -2,7 +2,7 @@ ngpt/__init__.py,sha256=kpKhViLakwMdHZkuLht2vWcjt0uD_5gR33gvMhfXr6w,664
 ngpt/__main__.py,sha256=j3eFYPOtCCFBOGh7NK5IWEnADnTMMSEB9GLyIDoW724,66
 ngpt/client.py,sha256=XjpA2UnvrRvzk6_DzVEddUTzoPlF8koQ-cZURpHoT7c,9041
 ngpt/cli/__init__.py,sha256=hebbDSMGiOd43YNnQP67uzr67Ue6rZPwm2czynr5iZY,43
-ngpt/cli/args.py,sha256=0ClIMzfY_qCuIVApphPDQNb8yDz1zA83mJtJwNrsfCE,15316
+ngpt/cli/args.py,sha256=F3ra7MJYqf0ULBR3C05jZwm9hiTIMpDd5DUC1tg1Fjs,15509
 ngpt/cli/config_manager.py,sha256=NQQcWnjUppAAd0s0p9YAf8EyKS1ex5-0EB4DvKdB4dk,3662
 ngpt/cli/formatters.py,sha256=HBYGlx_7eoAKyzfy0Vq5L0yn8yVKjngqYBukMmXCcz0,9401
 ngpt/cli/main.py,sha256=SERPyiniRDbqcEcyAmAcP1Uo1E4VFXpdaM_1r9cRCiA,31129
@@ -13,7 +13,7 @@ ngpt/cli/modes/__init__.py,sha256=KP7VR6Xw9k1p5Jcu0F38RDxSFvFIzH3j1ThDLNwznUI,36
 ngpt/cli/modes/chat.py,sha256=x1leClKq7UupA_CdW4tym0AivY2o_II123-I5IcAkxQ,7091
 ngpt/cli/modes/code.py,sha256=Qj59xq6fZqgUDw7SbvmPKX_gdpc7DHJhNkn1sB5qgUU,12932
 ngpt/cli/modes/gitcommsg.py,sha256=qFOrll333ebFOkzLP_WD1Qw0VfpphYqeiuHumkP6OB4,54833
-ngpt/cli/modes/interactive.py,sha256=kBs1TIaMOYR6SBWXhgK-etXxI0zYF1-g3_25zwI2DtQ,19072
+ngpt/cli/modes/interactive.py,sha256=ybJ1MFigW_-NBQKe7AOfuWz_XjW3x_9cYzjBvCiVkbs,22790
 ngpt/cli/modes/rewrite.py,sha256=RaN4NOItKrGy5ilRrTa2amK4zmNo59j6K7plUf5SJPM,20290
 ngpt/cli/modes/shell.py,sha256=it1Brq1-LGeNfPKYBeVAwF-a78g9UP-KscofBZQkbr4,41589
 ngpt/cli/modes/text.py,sha256=NOikaU9YVCBgyaCl6pwy9EVt-YY5Q4jBx0l47izpVTA,6986
@@ -23,8 +23,8 @@ ngpt/utils/config.py,sha256=wsArA4osnh8fKqOvtsPqqBxAz3DpdjtaWUFaRtnUdyc,10452
 ngpt/utils/log.py,sha256=f1jg2iFo35PAmsarH8FVL_62plq4VXH0Mu2QiP6RJGw,15934
 ngpt/utils/pipe.py,sha256=qRHF-Ma7bbU0cOcb1Yhe4S-kBavivtnnvLA3EYS4FY4,2162
 ngpt/utils/web_search.py,sha256=w5ke4KJMRxq7r5jtbUXvspja6XhjoPZloVkZ0IvBXIE,30731
-ngpt-4.0.1.dist-info/METADATA,sha256=8wH1Ls5-0R-AgAELLFDGTj24TKX6kTHMyoRj4ruFUH0,31619
-ngpt-4.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ngpt-4.0.1.dist-info/entry_points.txt,sha256=SqAAvLhMrsEpkIr4YFRdUeyuXQ9o0IBCeYgE6AVojoI,44
-ngpt-4.0.1.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
-ngpt-4.0.1.dist-info/RECORD,,
+ngpt-4.1.0.dist-info/METADATA,sha256=5Zl4EPZPe8Kjm2K24fPeXAat340IsEGMFSZ4VZqnVTc,31954
+ngpt-4.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ngpt-4.1.0.dist-info/entry_points.txt,sha256=SqAAvLhMrsEpkIr4YFRdUeyuXQ9o0IBCeYgE6AVojoI,44
+ngpt-4.1.0.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
+ngpt-4.1.0.dist-info/RECORD,,