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

indoxrouter/client.py

@@ -399,6 +399,48 @@ class Client:
         # is having issues with JSON formatted model strings
         return model
 
+    def _format_image_size_for_provider(
+        self, size: str, provider: str, model: str
+    ) -> str:
+        """
+        Format the image size parameter based on the provider's requirements.
+
+        Google requires aspect ratios like "1:1", "4:3", etc. while OpenAI uses pixel dimensions
+        like "1024x1024", "512x512", etc.
+
+        Args:
+            size: The size parameter (e.g., "1024x1024")
+            provider: The provider name (e.g., "google", "openai")
+            model: The model name
+
+        Returns:
+            Formatted size parameter appropriate for the provider
+        """
+        if provider.lower() == "google":
+            # Google uses aspect ratios instead of pixel dimensions
+            # Convert common pixel dimensions to aspect ratios
+            size_to_aspect_ratio = {
+                "1024x1024": "1:1",
+                "512x512": "1:1",
+                "256x256": "1:1",
+                "1024x768": "4:3",
+                "768x1024": "3:4",
+                "1024x1536": "2:3",
+                "1536x1024": "3:2",
+                "1792x1024": "16:9",
+                "1024x1792": "9:16",
+            }
+
+            # Check if size is already in aspect ratio format (contains a colon)
+            if ":" in size:
+                return size
+
+            # Convert to aspect ratio if we have a mapping, otherwise use default 1:1
+            return size_to_aspect_ratio.get(size, "1:1")
+
+        # For other providers, return the original size
+        return size
+
     def chat(
         self,
         messages: List[Dict[str, str]],
@@ -531,10 +573,10 @@ class Client:
         self,
         prompt: str,
         model: str = DEFAULT_IMAGE_MODEL,
-        size: str = "1024x1024",
-        n: int = 1,
-        quality: str = "standard",
-        style: str = "vivid",
+        size: Optional[str] = None,
+        n: Optional[int] = None,
+        quality: Optional[str] = None,
+        style: Optional[str] = None,
         # Standard parameters
         response_format: Optional[str] = None,
         user: Optional[str] = None,
@@ -563,10 +605,13 @@ class Client:
         Args:
             prompt: Text prompt
             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")
+
+            # Provider-specific parameters - will only be included if explicitly provided
+            # Note: Different providers support different parameters
+            size: Image size - For OpenAI: "1024x1024", "512x512", etc. For Google: "1:1", "4:3", etc.
             n: Number of images to generate
-            quality: Image quality (e.g., "standard", "hd")
-            style: Image style (e.g., "vivid", "natural")
+            quality: Image quality (e.g., "standard", "hd") - supported by some providers
+            style: Image style (e.g., "vivid", "natural") - supported by some providers
 
             # Standard parameters
             response_format: Format of the response - "url" or "b64_json"
@@ -599,22 +644,40 @@ class Client:
         # Format the model string
         formatted_model = self._format_model_string(model)
 
+        # Extract provider from model string if present
+        provider = "openai"  # Default provider
+        if "/" in model:
+            provider, _ = model.split("/", 1)
+
         # Filter out problematic parameters
         filtered_kwargs = {}
         for key, value in kwargs.items():
             if key not in ["return_generator"]:  # List of parameters to exclude
                 filtered_kwargs[key] = value
 
-        # Create the base request data
+        # Create the base request data with only the required parameters
         data = {
             "prompt": prompt,
             "model": formatted_model,
-            "n": n,
-            "size": size,
-            "quality": quality,
-            "style": style,
         }
 
+        # Add optional parameters only if they are explicitly provided
+        if n is not None:
+            data["n"] = n
+
+        # Handle size parameter with provider-specific formatting
+        if size is not None:
+            formatted_size = self._format_image_size_for_provider(size, provider, model)
+            data["size"] = formatted_size
+        elif provider.lower() == "google":
+            # Default size for Google if not provided
+            data["size"] = "1:1"
+
+        if quality is not None:
+            data["quality"] = quality
+        if style is not None:
+            data["style"] = style
+
         # Add standard parameters if provided
         if response_format is not None:
             data["response_format"] = response_format

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

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

indoxrouter-0.1.19.dist-info/RECORD

@@ -0,0 +1,8 @@
+indoxrouter/__init__.py,sha256=kwGvH8F5oqm2O4kLs-UtPfcY0AYiy5ZDUg-Sh3iYJA4,1627
+indoxrouter/client.py,sha256=5y2MFHYx-ge82K30yPYnDRUe4jS9aCF6n3D1JaB1Crw,37298
+indoxrouter/constants.py,sha256=GezZ9nuwK3A37xpWlcfXCrMsIIlCdP7xnvEkPBMyn5g,1383
+indoxrouter/exceptions.py,sha256=qs7f9AnJ7SkOyf9N5qRaZIKpECE8uBq1Pvcg19Jif-U,1718
+indoxrouter-0.1.19.dist-info/METADATA,sha256=bYZiVHEyt-yKnngnshHSu4fMHuBl1MnJqeWbKByr0aU,6639
+indoxrouter-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+indoxrouter-0.1.19.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
+indoxrouter-0.1.19.dist-info/RECORD,,

indoxrouter-0.1.17.dist-info/RECORD

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