ngpt 1.6.0__tar.gz → 1.7.0__tar.gz

{ngpt-1.6.0 → ngpt-1.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 1.6.0
+Version: 1.7.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
@@ -121,6 +121,9 @@ ngpt --show-config
 # Show all configurations
 ngpt --show-config --all
 
+# List available models for the current configuration
+ngpt --list-models
+
 # With custom options
 ngpt --api-key your-key --base-url http://your-endpoint --model your-model "Hello"
 
@@ -211,6 +214,7 @@ You can configure the client using the following options:
 | `--api-key` | API key for the service |
 | `--base-url` | Base URL for the API |
 | `--model` | Model to use |
+| `--list-models` | List all available models for the current configuration |
 | `--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 |

{ngpt-1.6.0 → ngpt-1.7.0}/README.md

@@ -87,6 +87,9 @@ ngpt --show-config
 # Show all configurations
 ngpt --show-config --all
 
+# List available models for the current configuration
+ngpt --list-models
+
 # With custom options
 ngpt --api-key your-key --base-url http://your-endpoint --model your-model "Hello"
 
@@ -177,6 +180,7 @@ You can configure the client using the following options:
 | `--api-key` | API key for the service |
 | `--base-url` | Base URL for the API |
 | `--model` | Model to use |
+| `--list-models` | List all available models for the current configuration |
 | `--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 |

{ngpt-1.6.0 → ngpt-1.7.0}/ngpt/cli.py

@@ -68,6 +68,8 @@ def show_config_help():
     print("     ngpt --config --config-index 1")
     print("  7. Remove a configuration at a specific index:")
     print("     ngpt --config --remove --config-index 1")
+    print("  8. List available models for the current configuration:")
+    print("     ngpt --list-models")
 
 def check_config(config):
     """Check config for common issues and provide guidance."""
@@ -273,6 +275,7 @@ def main():
     config_group.add_argument('--remove', action='store_true', help='Remove the configuration at the specified index (requires --config and --config-index)')
     config_group.add_argument('--show-config', action='store_true', help='Show the current configuration(s) and exit')
     config_group.add_argument('--all', action='store_true', help='Show details for all configurations (requires --show-config)')
+    config_group.add_argument('--list-models', action='store_true', help='List all available models for the current configuration and exit')
     
     # Global options
     global_group = parser.add_argument_group('Global Options')
@@ -409,18 +412,35 @@ 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):
+    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):
         parser.print_help()
         return
         
     # Check configuration (using the potentially overridden active_config)
-    if not check_config(active_config):
+    if not args.show_config and not args.list_models and not check_config(active_config):
         return
     
     # Initialize client using the potentially overridden active_config
     client = NGPTClient(**active_config)
     
     try:
+        # Handle listing models
+        if args.list_models:
+            print("Retrieving available models...")
+            models = client.list_models()
+            if models:
+                print(f"\nAvailable models for {active_config.get('provider', 'API')}:")
+                print("-" * 50)
+                for model in models:
+                    if "id" in model:
+                        owned_by = f" ({model.get('owned_by', 'Unknown')})" if "owned_by" in model else ""
+                        current = " [active]" if model["id"] == active_config["model"] else ""
+                        print(f"- {model['id']}{owned_by}{current}")
+                print("\nUse --model MODEL_NAME to select a specific model")
+            else:
+                print("No models available or could not retrieve models.")
+            return
+        
         # Handle modes
         if args.interactive:
             # Interactive chat mode

{ngpt-1.6.0 → ngpt-1.7.0}/ngpt/client.py

@@ -259,4 +259,48 @@ Code:"""
             )
         except Exception as e:
             print(f"Error generating code: {e}")
-            return "" 
+            return ""
+
+    def list_models(self) -> list:
+        """
+        Retrieve the list of available models from the API.
+        
+        Returns:
+            List of available model objects or empty list if failed
+        """
+        if not self.api_key:
+            print("Error: API key is not set. Please configure your API key in the config file or provide it with --api-key.")
+            return []
+            
+        # Endpoint for models
+        url = f"{self.base_url}models"
+        
+        try:
+            response = requests.get(url, headers=self.headers)
+            response.raise_for_status()  # Raise exception for HTTP errors
+            result = response.json()
+            
+            if "data" in result:
+                return result["data"]
+            else:
+                print("Error: Unexpected response format when retrieving models.")
+                return []
+                
+        except requests.exceptions.HTTPError as e:
+            if e.response.status_code == 401:
+                print("Error: Authentication failed. Please check your API key.")
+            elif e.response.status_code == 404:
+                print(f"Error: Models endpoint not found at {url}")
+            elif e.response.status_code == 429:
+                print("Error: Rate limit exceeded. Please try again later.")
+            else:
+                print(f"HTTP Error: {e}")
+            return []
+            
+        except requests.exceptions.ConnectionError:
+            print(f"Error: Could not connect to {self.base_url}. Please check your internet connection and base URL.")
+            return []
+            
+        except Exception as e:
+            print(f"Error: An unexpected error occurred while retrieving models: {e}")
+            return [] 

{ngpt-1.6.0 → ngpt-1.7.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ngpt"
-version = "1.6.0"
+version = "1.7.0"
 description = "A lightweight Python CLI and library for interacting with OpenAI-compatible APIs, supporting both official and self-hosted LLM endpoints."
 authors = [
     {name = "nazDridoy", email = "nazdridoy399@gmail.com"},

{ngpt-1.6.0 → ngpt-1.7.0}/uv.lock

@@ -113,7 +113,7 @@ wheels = [
 
 [[package]]
 name = "ngpt"
-version = "1.6.0"
+version = "1.7.0"
 source = { editable = "." }
 dependencies = [
     { name = "prompt-toolkit" },