ngpt 1.3.0__py3-none-any.whl → 1.5.1__py3-none-any.whl

ngpt/cli.py

@@ -103,6 +103,8 @@ def main():
     global_group.add_argument('--model', help='Model to use')
     global_group.add_argument('--web-search', action='store_true', 
                       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')
     
     # Mode flags (mutually exclusive)
     mode_group = parser.add_argument_group('Modes (mutually exclusive)')
@@ -110,7 +112,7 @@ def main():
     mode_exclusive_group.add_argument('-s', '--shell', action='store_true', help='Generate and execute shell commands')
     mode_exclusive_group.add_argument('-c', '--code', action='store_true', help='Generate code')
     mode_exclusive_group.add_argument('-t', '--text', action='store_true', help='Enter multi-line text input (submit with Ctrl+D)')
-    # Note: --show-config is handled separately and implicitly acts as a mode
+    mode_exclusive_group.add_argument('-i', '--interactive', action='store_true', help='Start an interactive chat session')
     
     # Language option for code mode
     parser.add_argument('--language', default="python", help='Programming language to generate code in (for code mode)')
@@ -228,7 +230,7 @@ def main():
         return
     
     # Check if prompt is required but not provided
-    if not args.prompt and not (args.shell or args.code or args.text):
+    if not args.prompt and not (args.shell or args.code or args.text or args.interactive):
         parser.print_help()
         return
         
@@ -277,6 +279,139 @@ def main():
                 except subprocess.CalledProcessError as e:
                     print(f"\nError:\n{e.stderr}")
                     
+        elif args.interactive:
+            # Interactive chat mode
+            conversation_history = []
+            print("\033[94m\033[1m" + "Interactive Chat Mode" + "\033[0m")
+            print("Type your messages and press Ctrl+D to send. Type 'exit', 'quit', or use Ctrl+C to exit.")
+            print("Type 'clear' to start a new conversation.")
+            if HAS_PROMPT_TOOLKIT:
+                print("Use arrow keys to navigate, Enter for new line in multi-line mode.")
+            print()
+            
+            try:
+                while True:
+                    try:
+                        # Get user input with prompt_toolkit if available
+                        if HAS_PROMPT_TOOLKIT:
+                            # Create key bindings
+                            kb = KeyBindings()
+                            
+                            # Explicitly bind Ctrl+C to exit
+                            @kb.add('c-c')
+                            def _(event):
+                                event.app.exit(result=None)
+                                print("\nExiting interactive chat mode.")
+                                sys.exit(130)
+                            
+                            # Explicitly bind Ctrl+D to submit
+                            @kb.add('c-d')
+                            def _(event):
+                                event.app.exit(result=event.app.current_buffer.text)
+                            
+                            # Get terminal dimensions
+                            term_width, term_height = shutil.get_terminal_size()
+                            
+                            # Create a styled TextArea
+                            text_area = TextArea(
+                                style="class:input-area",
+                                multiline=True,
+                                wrap_lines=True,
+                                width=term_width - 4,
+                                height=min(10, term_height - 8),
+                                prompt=HTML("<ansiblue>>>> </ansiblue>"),
+                                scrollbar=True,
+                                focus_on_click=True,
+                                lexer=None,
+                            )
+                            text_area.window.right_margins = [ScrollbarMargin(display_arrows=True)]
+                            
+                            # Create a title bar
+                            title_bar = FormattedTextControl(
+                                HTML("<style bg='ansiblue' fg='ansiwhite'><b> NGPT Interactive Chat </b></style>")
+                            )
+                            
+                            # Create a status bar with key bindings and commands info
+                            status_bar = FormattedTextControl(
+                                HTML("<ansiblue><b>Ctrl+D</b></ansiblue>: Submit | <ansiblue><b>Ctrl+C</b></ansiblue>: Exit | Type <ansiblue><b>clear</b></ansiblue> to start new conversation")
+                            )
+                            
+                            # Create the layout
+                            layout = Layout(
+                                HSplit([
+                                    Window(title_bar, height=1),
+                                    Window(height=1, char="-", style="class:separator"),
+                                    text_area,
+                                    Window(height=1, char="-", style="class:separator"),
+                                    Window(status_bar, height=1),
+                                ])
+                            )
+                            
+                            # Create a style
+                            style = Style.from_dict({
+                                "separator": "ansigray",
+                                "input-area": "bg:ansiblack fg:ansiwhite",
+                                "cursor": "bg:ansiwhite fg:ansiblack",
+                            })
+                            
+                            # Create and run the application
+                            app = Application(
+                                layout=layout,
+                                full_screen=False,
+                                key_bindings=kb,
+                                style=style,
+                                mouse_support=True,
+                            )
+                            
+                            user_input = app.run()
+                            if user_input is None:
+                                break
+                        else:
+                            # Fallback to standard input
+                            user_input = input("\033[1m\033[94m>>> \033[0m")
+                        
+                        # Handle special commands
+                        if user_input is None:
+                            break
+                        elif user_input.lower() in ['exit', 'quit', 'q']:
+                            print("Exiting interactive chat mode.")
+                            break
+                        elif user_input.lower() == 'clear':
+                            print("Starting a new conversation.")
+                            conversation_history = []
+                            continue
+                        elif not user_input.strip():
+                            continue
+                        
+                        # Add user message to conversation history
+                        conversation_history.append({"role": "user", "content": user_input})
+                        
+                        # Get response from the model
+                        print("\033[90m" + "AI is thinking..." + "\033[0m")
+                        response = client.chat(
+                            prompt=user_input,
+                            stream=not args.no_stream,
+                            messages=conversation_history,
+                            web_search=args.web_search
+                        )
+                        
+                        # Add assistant message to conversation history
+                        conversation_history.append({"role": "assistant", "content": response})
+                        
+                        # If no streaming, print the response
+                        if args.no_stream and response:
+                            print("\033[92m" + response + "\033[0m")
+                        
+                        print()  # Add spacing between exchanges
+                    
+                    except KeyboardInterrupt:
+                        print("\nExiting interactive chat mode.")
+                        break
+                    
+            except Exception as e:
+                print(f"\nError in interactive mode: {e}")
+                return
+                
         elif args.code:
             if args.prompt is None:
                 try:
@@ -401,7 +536,9 @@ def main():
                     sys.exit(130)
             
             print("\nSubmission successful. Waiting for response...")
-            client.chat(prompt, web_search=args.web_search)
+            response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search)
+            if args.no_stream and response:
+                print(response)
             
         else:
             # Default to chat mode
@@ -414,7 +551,9 @@ def main():
                     sys.exit(130)
             else:
                 prompt = args.prompt
-            client.chat(prompt, web_search=args.web_search)
+            response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search)
+            if args.no_stream and response:
+                print(response)
     
     except KeyboardInterrupt:
         print("\nOperation cancelled by user. Exiting gracefully.")

{ngpt-1.3.0.dist-info → ngpt-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 1.3.0
+Version: 1.5.1
 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
@@ -64,6 +64,9 @@ pip install ngpt
 # Chat with default settings
 ngpt "Tell me about quantum computing"
 
+# Return response without streaming
+ngpt -n "Tell me about quantum computing"
+
 # Generate code
 ngpt --code "function to calculate the Fibonacci sequence"
 
@@ -202,6 +205,7 @@ You can configure the client using the following options:
 | `--base-url` | Base URL for the API |
 | `--model` | Model to use |
 | `--web-search` | Enable web search capability |
+| `-n, --no-stream` | Return the whole response without streaming |
 | `--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-1.5.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
+ngpt/cli.py,sha256=ZMBmNJfKiKAO88MecT7CDHH1uyqpupeCqGSX4-5LC_g,27117
+ngpt/client.py,sha256=O0dPYeQCJlpWZWBBsroo-5UxeyBVwqC6o3Pm8lRnDiY,10329
+ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
+ngpt-1.5.1.dist-info/METADATA,sha256=eHORtsw7QD5HwfIhqRzCojOTNtfBqZ6kXCFT_8VBRq8,10086
+ngpt-1.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ngpt-1.5.1.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
+ngpt-1.5.1.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
+ngpt-1.5.1.dist-info/RECORD,,

ngpt-1.3.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
-ngpt/cli.py,sha256=U8A9JKOpp-GorBk9yMs2nPThSdUwm83P_e3sDgfqCYo,19880
-ngpt/client.py,sha256=O0dPYeQCJlpWZWBBsroo-5UxeyBVwqC6o3Pm8lRnDiY,10329
-ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
-ngpt-1.3.0.dist-info/METADATA,sha256=mGKNDQa1ppPnBp8lXILcSGYsS_40J0TQRQWLgpF_ewQ,9939
-ngpt-1.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ngpt-1.3.0.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
-ngpt-1.3.0.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
-ngpt-1.3.0.dist-info/RECORD,,