indoxrouter 0.1.22__py3-none-any.whl → 0.1.23__py3-none-any.whl

indoxrouter/client.py

@@ -673,6 +673,15 @@ class Client:
         if provider.lower() == "google":
             # For Google, use aspect_ratio instead of size
             if aspect_ratio is not None:
+                # Google's imagen-3 has specific supported aspect ratios
+                if model_name == "imagen-3.0-generate-002" and aspect_ratio not in [
+                    "1:1",
+                    "3:4",
+                    "4:3",
+                    "9:16",
+                    "16:9",
+                ]:
+                    aspect_ratio = "1:1"  # Default to 1:1 if not supported
                 data["aspect_ratio"] = aspect_ratio
             elif size is not None:
                 # Convert size to aspect_ratio
@@ -686,7 +695,7 @@ class Client:
         elif provider.lower() == "xai":
             # xAI doesn't support size parameter - do not include it
             pass
-        elif size is not None:
+        elif size is not None and provider.lower() != "xai":
             # For other providers (like OpenAI), use size as is
             data["size"] = size
 
@@ -697,7 +706,9 @@ class Client:
 
         # Add standard parameters if provided
         if response_format is not None:
+            # Only add response_format if explicitly provided by the user
             data["response_format"] = response_format
+
         if user is not None:
             data["user"] = user
 
@@ -739,14 +750,85 @@ class Client:
         if filtered_kwargs:
             data["additional_params"] = filtered_kwargs
 
-        # Special case handling for specific models
+        # Special case handling for specific models and providers
+        # Only include parameters supported by each model based on their JSON definitions
         if provider.lower() == "openai" and "gpt-image" in model_name.lower():
-            # Remove response_format for gpt-image models unless explicitly provided
-            if response_format is None and "response_format" in data:
+            # For OpenAI's gpt-image models, don't automatically add response_format
+            if "response_format" in data and response_format is None:
                 del data["response_format"]
 
+        if provider.lower() == "xai" and "grok-2-image" in model_name.lower():
+            # For xAI's grok-2-image models, ensure size is not included
+            if "size" in data:
+                del data["size"]
+
+        # Clean up any parameters that shouldn't be sent to specific providers
+        # This ensures we only send parameters that each provider supports
+        supported_params = self._get_supported_parameters_for_model(
+            provider, model_name
+        )
+        if supported_params:
+            for param in list(data.keys()):
+                if param not in ["prompt", "model"] and param not in supported_params:
+                    del data[param]
+
         return self._request("POST", IMAGE_ENDPOINT, data)
 
+    def _get_supported_parameters_for_model(
+        self, provider: str, model_name: str
+    ) -> List[str]:
+        """
+        Get the list of supported parameters for a specific model.
+        This helps avoid sending unsupported parameters to providers.
+
+        Args:
+            provider: The provider name (e.g., 'openai', 'google', 'xai')
+            model_name: The model name (e.g., 'gpt-image-1', 'imagen-3.0-generate-002')
+
+        Returns:
+            List of parameter names supported by the model
+        """
+        # Define supported parameters for specific models
+        if provider.lower() == "openai" and "gpt-image" in model_name.lower():
+            return [
+                "prompt",
+                "size",
+                "quality",
+                "n",
+                "user",
+                "background",
+                "moderation",
+                "output_compression",
+                "output_format",
+                "style",
+            ]
+
+        elif provider.lower() == "google" and "imagen" in model_name.lower():
+            return [
+                "prompt",
+                "n",
+                "negative_prompt",
+                "aspect_ratio",
+                "guidance_scale",
+                "seed",
+                "safety_filter_level",
+                "person_generation",
+                "include_safety_attributes",
+                "include_rai_reason",
+                "language",
+                "output_mime_type",
+                "output_compression_quality",
+                "add_watermark",
+                "enhance_prompt",
+                "response_format",
+            ]
+
+        elif provider.lower() == "xai" and "grok-2-image" in model_name.lower():
+            return ["prompt", "n", "response_format"]
+
+        # Default case - allow all parameters
+        return []
+
     def models(self, provider: Optional[str] = None) -> Dict[str, Any]:
         """
         Get available models.

{indoxrouter-0.1.22.dist-info → indoxrouter-0.1.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: indoxrouter
-Version: 0.1.22
+Version: 0.1.23
 Summary: A unified client for various AI providers
 Author-email: indoxRouter Team <ashkan.eskandari.dev@gmail.com>
 License: MIT

indoxrouter-0.1.23.dist-info/RECORD

@@ -0,0 +1,8 @@
+indoxrouter/__init__.py,sha256=kwGvH8F5oqm2O4kLs-UtPfcY0AYiy5ZDUg-Sh3iYJA4,1627
+indoxrouter/client.py,sha256=GwdZJiUARBGcD-A_VI1vFXWtvhQTF-3dDCBz-1qtrhk,41632
+indoxrouter/constants.py,sha256=GezZ9nuwK3A37xpWlcfXCrMsIIlCdP7xnvEkPBMyn5g,1383
+indoxrouter/exceptions.py,sha256=qs7f9AnJ7SkOyf9N5qRaZIKpECE8uBq1Pvcg19Jif-U,1718
+indoxrouter-0.1.23.dist-info/METADATA,sha256=xUem08HFj7X3WWvcbqg70fOW4W4f04-Yb7BSw13zBb0,6639
+indoxrouter-0.1.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+indoxrouter-0.1.23.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
+indoxrouter-0.1.23.dist-info/RECORD,,

indoxrouter-0.1.22.dist-info/RECORD

@@ -1,8 +0,0 @@
-indoxrouter/__init__.py,sha256=kwGvH8F5oqm2O4kLs-UtPfcY0AYiy5ZDUg-Sh3iYJA4,1627
-indoxrouter/client.py,sha256=oAUpx_RDpEIHav_dgb4sc7f_Agji3UXPlvMWLFCnQwE,38437
-indoxrouter/constants.py,sha256=GezZ9nuwK3A37xpWlcfXCrMsIIlCdP7xnvEkPBMyn5g,1383
-indoxrouter/exceptions.py,sha256=qs7f9AnJ7SkOyf9N5qRaZIKpECE8uBq1Pvcg19Jif-U,1718
-indoxrouter-0.1.22.dist-info/METADATA,sha256=aULhnVEH964jtJCoedkHH1SMCGZMkAJl3FK6CYsE-KY,6639
-indoxrouter-0.1.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-indoxrouter-0.1.22.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
-indoxrouter-0.1.22.dist-info/RECORD,,