ngpt 2.1.0__py3-none-any.whl → 2.3.0__py3-none-any.whl

ngpt/cli.py

@@ -85,7 +85,7 @@ def check_config(config):
     
     return True
 
-def interactive_chat_session(client, web_search=False, no_stream=False, temperature=0.7, top_p=1.0, max_length=None):
+def interactive_chat_session(client, web_search=False, no_stream=False, temperature=0.7, top_p=1.0, max_length=None, log_file=None, preprompt=None):
     """Run an interactive chat session with conversation history."""
     # Define ANSI color codes for terminal output
     COLORS = {
@@ -127,16 +127,35 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
     
     print(f"\n{separator}\n")
     
+    # Initialize log file if provided
+    log_handle = None
+    if log_file:
+        try:
+            import datetime
+            timestamp = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+            log_handle = open(log_file, 'a', encoding='utf-8')
+            log_handle.write(f"\n--- nGPT Session Log: {sys.argv} ---\n")
+            log_handle.write(f"Started at: {timestamp}\n\n")
+            print(f"{COLORS['green']}Logging conversation to: {log_file}{COLORS['reset']}")
+        except Exception as e:
+            print(f"{COLORS['yellow']}Warning: Could not open log file: {str(e)}{COLORS['reset']}")
+            log_handle = None
+    
     # Custom separator - use the same length for consistency
     def print_separator():
         print(f"\n{separator}\n")
     
     # Initialize conversation history
-    system_prompt = "You are a helpful assistant."
+    system_prompt = preprompt if preprompt else "You are a helpful assistant."
     conversation = []
     system_message = {"role": "system", "content": system_prompt}
     conversation.append(system_message)
     
+    # Log system prompt if logging is enabled
+    if log_handle and preprompt:
+        log_handle.write(f"System: {system_prompt}\n\n")
+        log_handle.flush()
+    
     # Initialize prompt_toolkit history
     prompt_history = InMemoryHistory() if HAS_PROMPT_TOOLKIT else None
     
@@ -228,6 +247,11 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
             user_message = {"role": "user", "content": user_input}
             conversation.append(user_message)
             
+            # Log user message if logging is enabled
+            if log_handle:
+                log_handle.write(f"User: {user_input}\n")
+                log_handle.flush()
+            
             # Print assistant indicator with formatting
             if not no_stream:
                 print(f"\n{ngpt_header()}: {COLORS['reset']}", end="", flush=True)
@@ -253,6 +277,11 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
                 # Print response if not streamed
                 if no_stream:
                     print(response)
+                
+                # Log assistant response if logging is enabled
+                if log_handle:
+                    log_handle.write(f"Assistant: {response}\n\n")
+                    log_handle.flush()
             
             # Print separator between exchanges
             print_separator()
@@ -264,9 +293,14 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
         # Print traceback for debugging if it's a serious error
         import traceback
         traceback.print_exc()
+    finally:
+        # Close log file if it was opened
+        if log_handle:
+            log_handle.write(f"\n--- End of Session ---\n")
+            log_handle.close()
 
 def main():
-    parser = argparse.ArgumentParser(description="nGPT - A CLI tool for interacting with custom OpenAI API endpoints")
+    parser = argparse.ArgumentParser(description="nGPT - A CLI tool for interacting with OpenAI-compatible APIs, supporting both official and self-hosted LLM endpoints")
     
     # Version flag
     parser.add_argument('-v', '--version', action='version', version=f'nGPT {__version__}', help='Show version information and exit')
@@ -295,6 +329,10 @@ def main():
                       help='Set top_p (controls diversity, default: 1.0)')
     global_group.add_argument('--max_length', type=int, 
                       help='Set max response length in tokens')
+    global_group.add_argument('--log', metavar='FILE',
+                      help='Set filepath to log conversation to (For interactive modes)')
+    global_group.add_argument('--preprompt', 
+                      help='Set preprompt')
     
     # Mode flags (mutually exclusive)
     mode_group = parser.add_argument_group('Modes (mutually exclusive)')
@@ -455,7 +493,7 @@ def main():
             # Interactive chat mode
             interactive_chat_session(client, web_search=args.web_search, no_stream=args.no_stream,
                                    temperature=args.temperature, top_p=args.top_p, 
-                                   max_length=args.max_length)
+                                   max_length=args.max_length, log_file=args.log, preprompt=args.preprompt)
         elif args.shell:
             if args.prompt is None:
                 try:
@@ -512,7 +550,6 @@ def main():
                 print(f"\nGenerated code:\n{generated_code}")
             
         elif args.text:
-            # Multi-line text input mode
             if args.prompt is not None:
                 prompt = args.prompt
             else:
@@ -620,9 +657,18 @@ def main():
                     sys.exit(130)
             
             print("\nSubmission successful. Waiting for response...")
+            
+            # Create messages array with preprompt if available
+            messages = None
+            if args.preprompt:
+                messages = [
+                    {"role": "system", "content": args.preprompt},
+                    {"role": "user", "content": prompt}
+                ]
+                
             response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search,
                                temperature=args.temperature, top_p=args.top_p,
-                               max_tokens=args.max_length)
+                               max_tokens=args.max_length, messages=messages)
             if args.no_stream and response:
                 print(response)
             
@@ -637,9 +683,18 @@ def main():
                     sys.exit(130)
             else:
                 prompt = args.prompt
+                
+            # Create messages array with preprompt if available
+            messages = None
+            if args.preprompt:
+                messages = [
+                    {"role": "system", "content": args.preprompt},
+                    {"role": "user", "content": prompt}
+                ]
+                
             response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search,
                                temperature=args.temperature, top_p=args.top_p,
-                               max_tokens=args.max_length)
+                               max_tokens=args.max_length, messages=messages)
             if args.no_stream and response:
                 print(response)
     

{ngpt-2.1.0.dist-info → ngpt-2.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 2.1.0
+Version: 2.3.0
 Summary: A lightweight Python CLI and library for interacting with OpenAI-compatible APIs, supporting both official and self-hosted LLM endpoints.
 Project-URL: Homepage, https://github.com/nazdridoy/ngpt
 Project-URL: Repository, https://github.com/nazdridoy/ngpt
@@ -81,8 +81,16 @@ ngpt --shell "list all files in the current directory"
 
 # Use multiline editor for complex prompts
 ngpt --text
+
+# Use custom system prompt
+ngpt --preprompt "You are a Linux expert" "How do I find large files?"
+
+# Log your conversation to a file
+ngpt --interactive --log conversation.log
 ```
 
+For more examples and detailed usage, visit the [CLI Usage Guide](https://nazdridoy.github.io/ngpt/usage/cli_usage.html).
+
 ## Features
 
 - ✅ **Dual Mode**: Use as a CLI tool or import as a Python library
@@ -95,6 +103,10 @@ ngpt --text
 - 💻 **Shell Command Generation**: OS-aware command execution
 - 🧩 **Clean Code Generation**: Output code without markdown or explanations
 - 📝 **Rich Multiline Editor**: Interactive multiline text input with syntax highlighting and intuitive controls
+- 🎭 **System Prompts**: Customize model behavior with custom system prompts
+- 📃 **Conversation Logging**: Save your conversations to text files for later reference
+
+See the [Feature Overview](https://nazdridoy.github.io/ngpt/overview.html) for more details.
 
 ## Documentation
 
@@ -102,6 +114,13 @@ Comprehensive documentation, including API reference, usage guides, and examples
 
 **[https://nazdridoy.github.io/ngpt/](https://nazdridoy.github.io/ngpt/)**
 
+Key documentation sections:
+- [Installation Guide](https://nazdridoy.github.io/ngpt/installation.html)
+- [CLI Usage Guide](https://nazdridoy.github.io/ngpt/usage/cli_usage.html)
+- [Library Usage Guide](https://nazdridoy.github.io/ngpt/usage/library_usage.html)
+- [Configuration Guide](https://nazdridoy.github.io/ngpt/configuration.html)
+- [Examples & Tutorials](https://nazdridoy.github.io/ngpt/examples/basic.html)
+
 ## Installation
 
 ```bash
@@ -110,6 +129,8 @@ pip install ngpt
 
 Requires Python 3.8 or newer.
 
+For detailed installation instructions, see the [Installation Guide](https://nazdridoy.github.io/ngpt/installation.html).
+
 ## Usage
 
 ### As a CLI Tool
@@ -121,6 +142,12 @@ ngpt "Hello, how are you?"
 # Interactive chat session with conversation history
 ngpt -i
 
+# Log conversation to a file
+ngpt --interactive --log conversation.log
+
+# Use custom system prompt to guide AI behavior
+ngpt --preprompt "You are a Python programming tutor" "Explain decorators"
+
 # Show version information
 ngpt -v
 
@@ -157,6 +184,8 @@ ngpt -c "create a python function that calculates fibonacci numbers"
 ngpt -t
 ```
 
+For more CLI examples and detailed usage information, see the [CLI Usage Guide](https://nazdridoy.github.io/ngpt/usage/cli_usage.html).
+
 ### As a Library
 
 ```python
@@ -189,6 +218,8 @@ command = client.generate_shell_command("list all files")
 code = client.generate_code("create a python function that calculates fibonacci numbers")
 ```
 
+For more library examples and advanced usage, see the [Library Usage Guide](https://nazdridoy.github.io/ngpt/usage/library_usage.html).
+
 #### Advanced Library Usage
 
 ```python
@@ -215,6 +246,8 @@ code = client.generate_code("function that converts Celsius to Fahrenheit")
 print(code)
 ```
 
+For advanced usage patterns and integrations, check out the [Advanced Examples](https://nazdridoy.github.io/ngpt/examples/advanced.html).
+
 ## Configuration
 
 ### Command Line Options
@@ -232,6 +265,8 @@ You can configure the client using the following options:
 | `--temperature` | Set temperature (controls randomness, default: 0.7) |
 | `--top_p` | Set top_p (controls diversity, default: 1.0) |
 | `--max_length` | Set maximum response length in tokens |
+| `--preprompt` | Set custom system prompt to control AI behavior |
+| `--log` | Set filepath to log conversation to (for interactive modes) |
 | `--config` | Path to a custom configuration file or, when used without a value, enters interactive configuration mode |
 | `--config-index` | Index of the configuration to use (default: 0) |
 | `--remove` | Remove the configuration at the specified index (requires --config and --config-index) |
@@ -243,6 +278,8 @@ You can configure the client using the following options:
 | `-t, --text` | Open interactive multiline editor for complex prompts |
 | `-v, --version` | Show version information |
 
+For a complete reference of all available options, see the [CLI Usage Guide](https://nazdridoy.github.io/ngpt/usage/cli_usage.html).
+
 ### Interactive Configuration
 
 The `--config` option without arguments enters interactive configuration mode, allowing you to add or edit configurations:
@@ -264,6 +301,8 @@ In interactive mode:
 - For security, your API key is not displayed when editing configurations
 - When removing a configuration, you'll be asked to confirm before deletion
 
+For more details on configuring nGPT, see the [Configuration Guide](https://nazdridoy.github.io/ngpt/configuration.html).
+
 ### Configuration File
 
 nGPT uses a configuration file stored in the standard user config directory for your operating system:
@@ -298,6 +337,8 @@ The configuration file uses a JSON list format, allowing you to store multiple c
 ]
 ```
 
+For details on the configuration file format and structure, see the [Configuration Guide](https://nazdridoy.github.io/ngpt/configuration.html).
+
 ### Configuration Priority
 
 nGPT determines configuration values in the following order (highest priority first):

ngpt-2.3.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
+ngpt/cli.py,sha256=MfZ_QDKdylLvhPwUPRgCCBUD3iRDDh8Gy2Xnzf1MbbY,32511
+ngpt/client.py,sha256=75xmzO7e9wQ7y_LzZCacg3mkZdheewcBxB6moPftqYw,13067
+ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
+ngpt-2.3.0.dist-info/METADATA,sha256=TG0WxgRFDG8oWOE3dtJgqRoyrJUfWB-QlgLkEq34Sqg,13535
+ngpt-2.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ngpt-2.3.0.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
+ngpt-2.3.0.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
+ngpt-2.3.0.dist-info/RECORD,,

ngpt-2.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
-ngpt/cli.py,sha256=wNjY-qUuvzODdzffbYqyydJiLfhIsEXXUs9qalqcpvs,30082
-ngpt/client.py,sha256=75xmzO7e9wQ7y_LzZCacg3mkZdheewcBxB6moPftqYw,13067
-ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
-ngpt-2.1.0.dist-info/METADATA,sha256=47qdylepQUHyR-TsJTtB8ezMnRLnZie0uSI1WVbHUNg,11278
-ngpt-2.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ngpt-2.1.0.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
-ngpt-2.1.0.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
-ngpt-2.1.0.dist-info/RECORD,,