ngpt 2.0.0__py3-none-any.whl → 2.2.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):
+def interactive_chat_session(client, web_search=False, no_stream=False, temperature=0.7, top_p=1.0, max_length=None, log_file=None):
     """Run an interactive chat session with conversation history."""
     # Define ANSI color codes for terminal output
     COLORS = {
@@ -127,6 +127,20 @@ def interactive_chat_session(client, web_search=False, no_stream=False):
     
     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")
@@ -228,6 +242,11 @@ def interactive_chat_session(client, web_search=False, no_stream=False):
             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)
@@ -239,7 +258,10 @@ def interactive_chat_session(client, web_search=False, no_stream=False):
                 prompt=user_input,
                 messages=conversation,
                 stream=not no_stream,
-                web_search=web_search
+                web_search=web_search,
+                temperature=temperature,
+                top_p=top_p,
+                max_length=max_length
             )
             
             # Add AI response to conversation history
@@ -250,6 +272,11 @@ def interactive_chat_session(client, web_search=False, no_stream=False):
                 # 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()
@@ -261,9 +288,14 @@ def interactive_chat_session(client, web_search=False, no_stream=False):
         # 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')
@@ -286,6 +318,14 @@ def main():
                       help='Enable web search capability (Note: Your API endpoint must support this feature)')
     global_group.add_argument('-n', '--no-stream', action='store_true',
                       help='Return the whole response without streaming')
+    global_group.add_argument('--temperature', type=float, default=0.7,
+                      help='Set temperature (controls randomness, default: 0.7)')
+    global_group.add_argument('--top_p', type=float, default=1.0,
+                      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)')
     
     # Mode flags (mutually exclusive)
     mode_group = parser.add_argument_group('Modes (mutually exclusive)')
@@ -444,7 +484,9 @@ def main():
         # Handle modes
         if args.interactive:
             # Interactive chat mode
-            interactive_chat_session(client, web_search=args.web_search, no_stream=args.no_stream)
+            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, log_file=args.log)
         elif args.shell:
             if args.prompt is None:
                 try:
@@ -456,7 +498,9 @@ def main():
             else:
                 prompt = args.prompt
                 
-            command = client.generate_shell_command(prompt, web_search=args.web_search)
+            command = client.generate_shell_command(prompt, web_search=args.web_search, 
+                                                 temperature=args.temperature, top_p=args.top_p,
+                                                 max_length=args.max_length)
             if not command:
                 return  # Error already printed by client
                 
@@ -492,7 +536,9 @@ def main():
             else:
                 prompt = args.prompt
                 
-            generated_code = client.generate_code(prompt, args.language, web_search=args.web_search)
+            generated_code = client.generate_code(prompt, args.language, web_search=args.web_search,
+                                              temperature=args.temperature, top_p=args.top_p,
+                                              max_length=args.max_length)
             if generated_code:
                 print(f"\nGenerated code:\n{generated_code}")
             
@@ -605,7 +651,9 @@ def main():
                     sys.exit(130)
             
             print("\nSubmission successful. Waiting for response...")
-            response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search)
+            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)
             if args.no_stream and response:
                 print(response)
             
@@ -620,7 +668,9 @@ def main():
                     sys.exit(130)
             else:
                 prompt = args.prompt
-            response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search)
+            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)
             if args.no_stream and response:
                 print(response)
     

ngpt/client.py

@@ -30,6 +30,7 @@ class NGPTClient:
         stream: bool = True,
         temperature: float = 0.7,
         max_tokens: Optional[int] = None,
+        top_p: float = 1.0,
         messages: Optional[List[Dict[str, str]]] = None,
         web_search: bool = False,
         **kwargs
@@ -42,6 +43,7 @@ class NGPTClient:
             stream: Whether to stream the response
             temperature: Controls randomness in the response
             max_tokens: Maximum number of tokens to generate
+            top_p: Controls diversity via nucleus sampling
             messages: Optional list of message objects to override default behavior
             web_search: Whether to enable web search capability
             **kwargs: Additional arguments to pass to the API
@@ -62,6 +64,7 @@ class NGPTClient:
             "messages": messages,
             "stream": stream,
             "temperature": temperature,
+            "top_p": top_p,
         }
         
         # Conditionally add web_search
@@ -158,13 +161,23 @@ class NGPTClient:
             print(f"Error: An unexpected error occurred: {e}")
             return ""
 
-    def generate_shell_command(self, prompt: str, web_search: bool = False) -> str:
+    def generate_shell_command(
+        self, 
+        prompt: str, 
+        web_search: bool = False,
+        temperature: float = 0.4,
+        top_p: float = 0.95,
+        max_length: Optional[int] = None
+    ) -> str:
         """
         Generate a shell command based on the prompt.
         
         Args:
             prompt: Description of the command to generate
             web_search: Whether to enable web search capability
+            temperature: Controls randomness in the response
+            top_p: Controls diversity via nucleus sampling
+            max_length: Maximum number of tokens to generate
             
         Returns:
             The generated shell command
@@ -212,13 +225,24 @@ Command:"""
                 prompt=prompt,
                 stream=False,
                 messages=messages,
-                web_search=web_search
+                web_search=web_search,
+                temperature=temperature,
+                top_p=top_p,
+                max_tokens=max_length
             )
         except Exception as e:
             print(f"Error generating shell command: {e}")
             return ""
 
-    def generate_code(self, prompt: str, language: str = "python", web_search: bool = False) -> str:
+    def generate_code(
+        self, 
+        prompt: str, 
+        language: str = "python", 
+        web_search: bool = False,
+        temperature: float = 0.4,
+        top_p: float = 0.95,
+        max_length: Optional[int] = None
+    ) -> str:
         """
         Generate code based on the prompt.
         
@@ -226,6 +250,9 @@ Command:"""
             prompt: Description of the code to generate
             language: Programming language to generate code in
             web_search: Whether to enable web search capability
+            temperature: Controls randomness in the response
+            top_p: Controls diversity via nucleus sampling
+            max_length: Maximum number of tokens to generate
             
         Returns:
             The generated code
@@ -255,7 +282,10 @@ Code:"""
                 prompt=prompt,
                 stream=False,
                 messages=messages,
-                web_search=web_search
+                web_search=web_search,
+                temperature=temperature,
+                top_p=top_p,
+                max_tokens=max_length
             )
         except Exception as e:
             print(f"Error generating code: {e}")

{ngpt-2.0.0.dist-info → ngpt-2.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 2.0.0
+Version: 2.2.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
@@ -37,14 +37,17 @@ Description-Content-Type: text/markdown
 [![PyPI version](https://img.shields.io/pypi/v/ngpt.svg)](https://pypi.org/project/ngpt/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 [![Python Versions](https://img.shields.io/pypi/pyversions/ngpt.svg)](https://pypi.org/project/ngpt/)
+[![Documentation](https://img.shields.io/badge/docs-available-brightgreen.svg)](https://nazdridoy.github.io/ngpt/)
 
 A lightweight Python CLI and library for interacting with OpenAI-compatible APIs, supporting both official and self-hosted LLM endpoints.
 
 ## Table of Contents
 - [Quick Start](#quick-start)
 - [Features](#features)
+- [Documentation](#documentation)
 - [Installation](#installation)
 - [Usage](#usage)
+  - [Documentation](https://nazdridoy.github.io/ngpt/)
   - [CLI Tool](#as-a-cli-tool)
   - [Python Library](#as-a-library)
 - [Configuration](#configuration)
@@ -93,6 +96,12 @@ ngpt --text
 - 🧩 **Clean Code Generation**: Output code without markdown or explanations
 - 📝 **Rich Multiline Editor**: Interactive multiline text input with syntax highlighting and intuitive controls
 
+## Documentation
+
+Comprehensive documentation, including API reference, usage guides, and examples, is available at:
+
+**[https://nazdridoy.github.io/ngpt/](https://nazdridoy.github.io/ngpt/)**
+
 ## Installation
 
 ```bash
@@ -220,6 +229,9 @@ You can configure the client using the following options:
 | `--list-models` | List all available models for the selected configuration (can be combined with --config-index) |
 | `--web-search` | Enable web search capability |
 | `-n, --no-stream` | Return the whole response without streaming |
+| `--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 |
 | `--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) |

ngpt-2.2.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
+ngpt/cli.py,sha256=H_EIwZr-mSpw5JdlSD-CT6zWGWLldCJ4G-BKyWGIoOE,31536
+ngpt/client.py,sha256=75xmzO7e9wQ7y_LzZCacg3mkZdheewcBxB6moPftqYw,13067
+ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
+ngpt-2.2.0.dist-info/METADATA,sha256=seHeG-0SZyJcCN_SXpfpzI3jfwLBs9XrPkcdXt1mMhI,11278
+ngpt-2.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ngpt-2.2.0.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
+ngpt-2.2.0.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
+ngpt-2.2.0.dist-info/RECORD,,

ngpt-2.0.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
-ngpt/cli.py,sha256=AyIraZFq7icPot0moqPVJer72iqbtJxKBhy6VH-dwAA,28746
-ngpt/client.py,sha256=ygtY2xuu-PAFPrz1CUJxcj3hyWw7q2kRG85ClDGClCw,12089
-ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
-ngpt-2.0.0.dist-info/METADATA,sha256=goeEJKkrJeTSrluucsXpuyinzJXbpyTSB5_s6fb4gV8,10686
-ngpt-2.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ngpt-2.0.0.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
-ngpt-2.0.0.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
-ngpt-2.0.0.dist-info/RECORD,,