indoxrouter 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

indoxrouter/__init__.py

@@ -30,7 +30,7 @@ For custom server URLs:
     ```
 """
 
-from .client import Client
+from .client import Client, IndoxRouter
 from .exceptions import (
     IndoxRouterError,
     AuthenticationError,
@@ -39,13 +39,18 @@ from .exceptions import (
     ProviderError,
     ModelNotFoundError,
     ProviderNotFoundError,
+    ModelNotAvailableError,
     InvalidParametersError,
+    RequestError,
     InsufficientCreditsError,
+    ValidationError,
+    APIError,
 )
 
 __version__ = "0.2.1"
 __all__ = [
     "Client",
+    "IndoxRouter",
     "IndoxRouterError",
     "AuthenticationError",
     "NetworkError",
@@ -53,6 +58,10 @@ __all__ = [
     "ProviderError",
     "ModelNotFoundError",
     "ProviderNotFoundError",
+    "ModelNotAvailableError",
     "InvalidParametersError",
+    "RequestError",
     "InsufficientCreditsError",
+    "ValidationError",
+    "APIError",
 ]

indoxrouter/client.py

@@ -57,10 +57,14 @@ from .exceptions import (
     NetworkError,
     ProviderNotFoundError,
     ModelNotFoundError,
+    ModelNotAvailableError,
     InvalidParametersError,
     RateLimitError,
     ProviderError,
+    RequestError,
     InsufficientCreditsError,
+    ValidationError,
+    APIError,
 )
 from .constants import (
     DEFAULT_BASE_URL,
@@ -316,29 +320,57 @@ class Client:
                 if "provider" in error_message.lower():
                     raise ProviderNotFoundError(error_message)
                 elif "model" in error_message.lower():
-                    raise ModelNotFoundError(error_message)
+                    # Check if it's a model not found vs model not available
+                    if (
+                        "not supported" in error_message.lower()
+                        or "disabled" in error_message.lower()
+                        or "unavailable" in error_message.lower()
+                    ):
+                        raise ModelNotAvailableError(error_message)
+                    else:
+                        raise ModelNotFoundError(error_message)
                 else:
-                    raise NetworkError(
-                        f"Resource not found: {error_message} (URL: {url})"
-                    )
+                    raise APIError(f"Resource not found: {error_message} (URL: {url})")
             elif status_code == 429:
                 raise RateLimitError(f"Rate limit exceeded: {error_message}")
             elif status_code == 400:
-                raise InvalidParametersError(f"Invalid parameters: {error_message}")
+                # Check if it's a validation error or invalid parameters
+                if (
+                    "validation" in error_message.lower()
+                    or "invalid format" in error_message.lower()
+                ):
+                    raise ValidationError(f"Request validation failed: {error_message}")
+                else:
+                    raise InvalidParametersError(f"Invalid parameters: {error_message}")
             elif status_code == 402:
                 raise InsufficientCreditsError(f"Insufficient credits: {error_message}")
+            elif status_code == 422:
+                # Unprocessable Entity - typically validation errors
+                raise ValidationError(f"Request validation failed: {error_message}")
+            elif status_code == 503:
+                # Service Unavailable - model might be temporarily unavailable
+                if "model" in error_message.lower():
+                    raise ModelNotAvailableError(
+                        f"Model temporarily unavailable: {error_message}"
+                    )
+                else:
+                    raise APIError(f"Service unavailable: {error_message}")
             elif status_code == 500:
                 # Provide more detailed information for server errors
                 error_detail = error_data.get("detail", "No details provided")
                 # Include the request data in the error message for better debugging
                 request_data_str = json.dumps(data, indent=2) if data else "None"
-                raise ProviderError(
+                raise RequestError(
                     f"Server error (500): {error_detail}. URL: {url}.\n"
                     f"Request data: {request_data_str}\n"
                     f"This may indicate an issue with the server configuration or a problem with the provider service."
                 )
+            elif status_code >= 400 and status_code < 500:
+                # Client errors
+                raise APIError(f"Client error ({status_code}): {error_message}")
             else:
-                raise ProviderError(f"Provider error ({status_code}): {error_message}")
+                # Server errors
+                raise RequestError(f"Server error ({status_code}): {error_message}")
         except requests.RequestException as e:
             logger.error(f"Request exception: {str(e)}")
             raise NetworkError(f"Network error: {str(e)}")
@@ -503,6 +535,26 @@ class Client:
         n: int = 1,
         quality: str = "standard",
         style: str = "vivid",
+        # Standard parameters
+        response_format: Optional[str] = None,
+        user: Optional[str] = None,
+        # OpenAI-specific parameters
+        background: Optional[str] = None,
+        moderation: Optional[str] = None,
+        output_compression: Optional[int] = None,
+        output_format: Optional[str] = None,
+        # Google-specific parameters
+        negative_prompt: Optional[str] = None,
+        guidance_scale: Optional[float] = None,
+        seed: Optional[int] = None,
+        safety_filter_level: Optional[str] = None,
+        person_generation: Optional[str] = None,
+        include_safety_attributes: Optional[bool] = None,
+        include_rai_reason: Optional[bool] = None,
+        language: Optional[str] = None,
+        output_mime_type: Optional[str] = None,
+        add_watermark: Optional[bool] = None,
+        enhance_prompt: Optional[bool] = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """
@@ -510,11 +562,35 @@ class Client:
 
         Args:
             prompt: Text prompt
-            model: Model to use in the format "provider/model" (e.g., "openai/dall-e-3")
+            model: Model to use in the format "provider/model" (e.g., "openai/dall-e-3", "google/imagen-3.0-generate-002")
             size: Image size (e.g., "1024x1024")
             n: Number of images to generate
             quality: Image quality (e.g., "standard", "hd")
             style: Image style (e.g., "vivid", "natural")
+
+            # Standard parameters
+            response_format: Format of the response - "url" or "b64_json"
+            user: A unique identifier for the end-user
+
+            # OpenAI-specific parameters
+            background: Background style - "transparent", "opaque", or "auto"
+            moderation: Moderation level - "low" or "auto"
+            output_compression: Compression quality for output images (0-100)
+            output_format: Output format - "png", "jpeg", or "webp"
+
+            # Google-specific parameters
+            negative_prompt: Description of what to discourage in the generated images
+            guidance_scale: Controls how much the model adheres to the prompt
+            seed: Random seed for image generation
+            safety_filter_level: Filter level for safety filtering
+            person_generation: Controls generation of people ("dont_allow", "allow_adult", "allow_all")
+            include_safety_attributes: Whether to report safety scores of generated images
+            include_rai_reason: Whether to include filter reason if the image is filtered
+            language: Language of the text in the prompt
+            output_mime_type: MIME type of the generated image
+            add_watermark: Whether to add a watermark to the generated images
+            enhance_prompt: Whether to use prompt rewriting logic
+
             **kwargs: Additional parameters to pass to the API
 
         Returns:
@@ -529,6 +605,7 @@ class Client:
             if key not in ["return_generator"]:  # List of parameters to exclude
                 filtered_kwargs[key] = value
 
+        # Create the base request data
         data = {
             "prompt": prompt,
             "model": formatted_model,
@@ -536,9 +613,52 @@ class Client:
             "size": size,
             "quality": quality,
             "style": style,
-            "additional_params": filtered_kwargs,
         }
 
+        # Add standard parameters if provided
+        if response_format is not None:
+            data["response_format"] = response_format
+        if user is not None:
+            data["user"] = user
+
+        # Add OpenAI-specific parameters if provided
+        if background is not None:
+            data["background"] = background
+        if moderation is not None:
+            data["moderation"] = moderation
+        if output_compression is not None:
+            data["output_compression"] = output_compression
+        if output_format is not None:
+            data["output_format"] = output_format
+
+        # Add Google-specific parameters if provided
+        if negative_prompt is not None:
+            data["negative_prompt"] = negative_prompt
+        if guidance_scale is not None:
+            data["guidance_scale"] = guidance_scale
+        if seed is not None:
+            data["seed"] = seed
+        if safety_filter_level is not None:
+            data["safety_filter_level"] = safety_filter_level
+        if person_generation is not None:
+            data["person_generation"] = person_generation
+        if include_safety_attributes is not None:
+            data["include_safety_attributes"] = include_safety_attributes
+        if include_rai_reason is not None:
+            data["include_rai_reason"] = include_rai_reason
+        if language is not None:
+            data["language"] = language
+        if output_mime_type is not None:
+            data["output_mime_type"] = output_mime_type
+        if add_watermark is not None:
+            data["add_watermark"] = add_watermark
+        if enhance_prompt is not None:
+            data["enhance_prompt"] = enhance_prompt
+
+        # Add any remaining parameters
+        if filtered_kwargs:
+            data["additional_params"] = filtered_kwargs
+
         return self._request("POST", IMAGE_ENDPOINT, data)
 
     def models(self, provider: Optional[str] = None) -> Dict[str, Any]:

indoxrouter/constants.py

@@ -13,6 +13,10 @@ USE_COOKIES = True  # Always use cookie-based authentication
 DEFAULT_MODEL = "openai/gpt-4o-mini"
 DEFAULT_EMBEDDING_MODEL = "openai/text-embedding-3-small"
 DEFAULT_IMAGE_MODEL = "openai/dall-e-3"
+GOOGLE_IMAGE_MODEL = "google/imagen-3.0-generate-002"
+XAI_IMAGE_MODEL = "xai/grok-2-image"
+XAI_IMAGE_LATEST_MODEL = "xai/grok-2-image-latest"
+XAI_IMAGE_SPECIFIC_MODEL = "xai/grok-2-image-1212"
 
 # API endpoints
 CHAT_ENDPOINT = "chat/completions"

indoxrouter/exceptions.py

@@ -38,14 +38,20 @@ class ProviderError(IndoxRouterError):
     pass
 
 
+class ProviderNotFoundError(ProviderError):
+    """Raised when a requested provider is not found."""
+
+    pass
+
+
 class ModelNotFoundError(ProviderError):
     """Raised when a requested model is not found."""
 
     pass
 
 
-class ProviderNotFoundError(ProviderError):
-    """Raised when a requested provider is not found."""
+class ModelNotAvailableError(ProviderError):
+    """Raised when a model is disabled or not supported by the provider."""
 
     pass
 
@@ -56,7 +62,25 @@ class InvalidParametersError(IndoxRouterError):
     pass
 
 
+class RequestError(IndoxRouterError):
+    """Raised when a request to a provider fails."""
+
+    pass
+
+
 class InsufficientCreditsError(IndoxRouterError):
     """Raised when the user doesn't have enough credits."""
 
     pass
+
+
+class ValidationError(IndoxRouterError):
+    """Raised when request validation fails."""
+
+    pass
+
+
+class APIError(IndoxRouterError):
+    """Raised when the API returns an error."""
+
+    pass

{indoxrouter-0.1.15.dist-info → indoxrouter-0.1.17.dist-info}/METADATA

@@ -1,9 +1,13 @@
 Metadata-Version: 2.4
 Name: indoxrouter
-Version: 0.1.15
+Version: 0.1.17
 Summary: A unified client for various AI providers
 Author-email: indoxRouter Team <ashkan.eskandari.dev@gmail.com>
+License: MIT
 Project-URL: Homepage, https://github.com/indoxrouter/indoxrouter
+Project-URL: Repository, https://github.com/indoxrouter/indoxrouter
+Project-URL: Issues, https://github.com/indoxrouter/indoxrouter/issues
+Keywords: ai,api,client,openai,anthropic,google,mistral,xai,imagen,grok,image-generation
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3
@@ -129,13 +133,43 @@ print(f"First embedding: {response['data'][0]['embedding'][:5]}...")
 ### Image Generation
 
 ```python
+# OpenAI Image Generation
 response = client.images(
     prompt="A serene landscape with mountains and a lake",
     model="openai/dall-e-3",
-    size="1024x1024"
+    size="1024x1024",
+    quality="standard",  # Options: standard, hd
+    style="vivid"  # Options: vivid, natural
 )
 
 print(f"Image URL: {response['data'][0]['url']}")
+
+# Google Imagen Image Generation
+from indoxrouter.constants import GOOGLE_IMAGE_MODEL
+
+response = client.images(
+    prompt="A robot holding a red skateboard in a futuristic city",
+    model=GOOGLE_IMAGE_MODEL,
+    n=2,  # Generate 2 images
+    negative_prompt="broken, damaged, low quality",
+    guidance_scale=7.5,  # Control adherence to prompt
+    seed=42,  # For reproducible results
+)
+
+# xAI Grok Image Generation
+from indoxrouter.constants import XAI_IMAGE_MODEL
+
+response = client.images(
+    prompt="A cat in a tree",
+    model=XAI_IMAGE_MODEL,
+    n=1,
+    response_format="b64_json"  # Get base64 encoded image
+)
+
+# Access base64 encoded image data
+if "b64_json" in response["data"][0]:
+    b64_data = response["data"][0]["b64_json"]
+    # Use the base64 data (e.g., to display in HTML or save to file)
 ```
 
 ### Streaming Responses

indoxrouter-0.1.17.dist-info/RECORD

@@ -0,0 +1,8 @@
+indoxrouter/__init__.py,sha256=kwGvH8F5oqm2O4kLs-UtPfcY0AYiy5ZDUg-Sh3iYJA4,1627
+indoxrouter/client.py,sha256=0FZ-7mgE_kxbRpmRwJrucpvHme4MQOToX2VqVw9PKQE,34706
+indoxrouter/constants.py,sha256=GezZ9nuwK3A37xpWlcfXCrMsIIlCdP7xnvEkPBMyn5g,1383
+indoxrouter/exceptions.py,sha256=qs7f9AnJ7SkOyf9N5qRaZIKpECE8uBq1Pvcg19Jif-U,1718
+indoxrouter-0.1.17.dist-info/METADATA,sha256=XFpOPo3DdQQuiPzmAOTbQvVc0bJRxPgD6FkNLkEyMFA,6639
+indoxrouter-0.1.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+indoxrouter-0.1.17.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
+indoxrouter-0.1.17.dist-info/RECORD,,

{indoxrouter-0.1.15.dist-info → indoxrouter-0.1.17.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

indoxrouter-0.1.15.dist-info/RECORD

@@ -1,8 +0,0 @@
-indoxrouter/__init__.py,sha256=28pdx482uGFF_S1msov0LTTGsFTvVBKRqMkDmoXWUBY,1416
-indoxrouter/client.py,sha256=ixzCJP28ddYFQxwRiqdY5Fp1CB978DJpI4yCiOI3s4o,28898
-indoxrouter/constants.py,sha256=tEmfhfCpuKVos1TxVhJfPlv1P0ePWnjnwso4ZHJ0VhM,1190
-indoxrouter/exceptions.py,sha256=0ULxtK9va4718PGTO5VoClXYEJeojpiM-7AganeiZZ4,1263
-indoxrouter-0.1.15.dist-info/METADATA,sha256=CeJbcb4OaBKSTWJjsGl4wOulialexUJN6MDLYniQj-g,5428
-indoxrouter-0.1.15.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-indoxrouter-0.1.15.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
-indoxrouter-0.1.15.dist-info/RECORD,,