ngpt 1.7.1__py3-none-any.whl → 2.0.0__py3-none-any.whl

ngpt/client.py

@@ -18,65 +18,12 @@ class NGPTClient:
         self.base_url = base_url if base_url.endswith('/') else base_url + '/'
         self.model = model
         
-        # Detect provider based on base_url domain first
-        detected_provider = self._detect_provider_from_url(self.base_url)
-        self.provider = detected_provider if detected_provider else provider
-        
         # Default headers
         self.headers = {
             "Content-Type": "application/json",
             "Authorization": f"Bearer {self.api_key}"
         }
 
-    def _detect_provider_from_url(self, url: str) -> str:
-        """
-        Detect the API provider based on the base URL domain.
-        
-        Args:
-            url: The base URL to analyze
-            
-        Returns:
-            Detected provider name or empty string if not detected
-        """
-        import re
-        from urllib.parse import urlparse
-        
-        # Extract domain from URL
-        try:
-            parsed_url = urlparse(url)
-            domain = parsed_url.netloc.lower()
-            
-            # Check for known API provider domains
-            if 'openai.com' in domain:
-                return 'OpenAI'
-            elif 'anthropic.com' in domain:
-                return 'Anthropic'
-            elif 'generativelanguage.googleapis.com' in domain:
-                return 'Gemini'
-            elif 'groq.com' in domain:
-                return 'Groq'
-            elif 'cohere.com' in domain or 'cohere.ai' in domain:
-                return 'Cohere'
-            elif 'mistral.ai' in domain:
-                return 'Mistral'
-            elif 'claude.ai' in domain:
-                return 'Claude'
-            elif re.match(r'(127\.0\.0\.1|localhost)', domain):
-                # Local APIs, check path components for clues
-                path = parsed_url.path.lower()
-                if '/ollama' in path:
-                    return 'Ollama'
-                elif '/llama' in path:
-                    return 'Llama'
-                # Default to the explicitly provided provider for local APIs
-                return ''
-            
-            # Return empty string if no match found, will fall back to provided provider
-            return ''
-        except Exception:
-            # In case of any parsing error, fall back to provided provider
-            return ''
-
     def chat(
         self,
         prompt: str,
@@ -132,26 +79,11 @@ class NGPTClient:
         endpoint = "chat/completions"
         url = f"{self.base_url}{endpoint}"
         
-        # Set headers based on provider
-        headers = self.headers.copy()
-        provider = self.provider.lower() if hasattr(self, 'provider') else ""
-        
-        # Provider-specific customizations
-        if 'gemini' in provider:
-            # Gemini uses X-Goog-Api-Key instead of Bearer token
-            headers = {
-                "Content-Type": "application/json",
-                "X-Goog-Api-Key": self.api_key
-            }
-            
-            # Gemini may require a different endpoint or payload structure
-            # This is a simplistic approach - may need further customization
-        
         try:
             if not stream:
                 # Regular request
                 try:
-                    response = requests.post(url, headers=headers, json=payload)
+                    response = requests.post(url, headers=self.headers, json=payload)
                     response.raise_for_status()  # Raise exception for HTTP errors
                     result = response.json()
                     
@@ -165,7 +97,7 @@ class NGPTClient:
             else:
                 # Streaming request
                 collected_content = ""
-                with requests.post(url, headers=headers, json=payload, stream=True) as response:
+                with requests.post(url, headers=self.headers, json=payload, stream=True) as response:
                     response.raise_for_status()  # Raise exception for HTTP errors
                     
                     try:
@@ -339,77 +271,36 @@ Code:"""
         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 []
-        
-        provider = getattr(self, 'provider', '').lower()
-        
-        # Handle provider-specific API differences
-        if 'gemini' in provider:
-            # Gemini API specific handling
-            # Base URL typically ends with /v1beta or similar - we want to add /models
-            base_url = self.base_url.rstrip('/')
-            if base_url.endswith('/v1beta'):
-                url = f"{base_url}/models"
-            else:
-                url = f"{base_url}/models"
             
-            # Gemini uses X-Goog-Api-Key instead of Bearer token
-            headers = {
-                "Content-Type": "application/json",
-                "X-Goog-Api-Key": self.api_key
-            }
+        # 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()
             
-            try:
-                response = requests.get(url, headers=headers)
-                response.raise_for_status()
-                result = response.json()
-                
-                # Gemini API returns models in a different format
-                if "models" in result:
-                    # Transform to match our expected format
-                    return [{"id": model.get("name", "").split("/")[-1], 
-                             "owned_by": "Google"} 
-                            for model in result["models"]]
-                else:
-                    print("Error: Unexpected response format when retrieving Gemini models.")
-                    return []
-                    
-            except requests.exceptions.HTTPError as e:
-                print(f"HTTP Error with Gemini API: {e}")
+            if "data" in result:
+                return result["data"]
+            else:
+                print("Error: Unexpected response format when retrieving models.")
                 return []
                 
-            except Exception as e:
-                print(f"Error retrieving Gemini models: {e}")
-                return []
-        else:
-            # Standard OpenAI-compatible endpoint
-            url = f"{self.base_url}models"
+        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 []
             
-            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 [] 
+        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.7.1.dist-info → ngpt-2.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 1.7.1
+Version: 2.0.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,9 +121,12 @@ ngpt --show-config
 # Show all configurations
 ngpt --show-config --all
 
-# List available models for the current configuration
+# List available models for the active configuration
 ngpt --list-models
 
+# List models for a specific configuration
+ngpt --list-models --config-index 1
+
 # With custom options
 ngpt --api-key your-key --base-url http://your-endpoint --model your-model "Hello"
 
@@ -214,7 +217,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 |
+| `--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 |
 | `--config` | Path to a custom configuration file or, when used without a value, enters interactive configuration mode |

ngpt-2.0.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+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,,

ngpt-1.7.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
-ngpt/cli.py,sha256=AyIraZFq7icPot0moqPVJer72iqbtJxKBhy6VH-dwAA,28746
-ngpt/client.py,sha256=qwAmBHSJvadbsvK5YPbY0By1M4bBvxd7j8piP5UdTgE,16553
-ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
-ngpt-1.7.1.dist-info/METADATA,sha256=ffoVoKaC58f-SKImuqxLfGOv9Umlpmou82aCFWpsfIM,10568
-ngpt-1.7.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ngpt-1.7.1.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
-ngpt-1.7.1.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
-ngpt-1.7.1.dist-info/RECORD,,