indoxrouter 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

indoxrouter/client.py

@@ -531,10 +531,30 @@ 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,
+        # 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]:
         """
@@ -542,11 +562,38 @@ class Client:
 
         Args:
             prompt: Text prompt
-            model: Model to use in the format "provider/model" (e.g., "openai/dall-e-3")
-            size: Image size (e.g., "1024x1024")
+            model: Model to use in the format "provider/model" (e.g., "openai/dall-e-3", "google/imagen-3.0-generate-002")
+
+            # 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"
+            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:
@@ -561,16 +608,66 @@ class Client:
             if key not in ["return_generator"]:  # List of parameters to exclude
                 filtered_kwargs[key] = value
 
+        # Create the base request data with only the required parameters
         data = {
             "prompt": prompt,
             "model": formatted_model,
-            "n": n,
-            "size": size,
-            "quality": quality,
-            "style": style,
-            "additional_params": filtered_kwargs,
         }
 
+        # Add optional parameters only if they are explicitly provided
+        if n is not None:
+            data["n"] = n
+        if size is not None:
+            data["size"] = size
+        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
+        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-0.1.16.dist-info → indoxrouter-0.1.18.dist-info}/METADATA

@@ -1,16 +1,15 @@
 Metadata-Version: 2.4
 Name: indoxrouter
-Version: 0.1.16
+Version: 0.1.18
 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
+Keywords: ai,api,client,openai,anthropic,google,mistral,xai,imagen,grok,image-generation
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
@@ -134,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.18.dist-info/RECORD

@@ -0,0 +1,8 @@
+indoxrouter/__init__.py,sha256=kwGvH8F5oqm2O4kLs-UtPfcY0AYiy5ZDUg-Sh3iYJA4,1627
+indoxrouter/client.py,sha256=j1ESSti83R9_fDiYEop0FJqtho0e2FuYes92XY7iImI,35263
+indoxrouter/constants.py,sha256=GezZ9nuwK3A37xpWlcfXCrMsIIlCdP7xnvEkPBMyn5g,1383
+indoxrouter/exceptions.py,sha256=qs7f9AnJ7SkOyf9N5qRaZIKpECE8uBq1Pvcg19Jif-U,1718
+indoxrouter-0.1.18.dist-info/METADATA,sha256=2K3ms5FkCR1L_PkApVOwzF10AHZWh-nn6yi9GvqDlQE,6639
+indoxrouter-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+indoxrouter-0.1.18.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
+indoxrouter-0.1.18.dist-info/RECORD,,

{indoxrouter-0.1.16.dist-info → indoxrouter-0.1.18.dist-info}/WHEEL

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

indoxrouter-0.1.16.dist-info/RECORD

@@ -1,8 +0,0 @@
-indoxrouter/__init__.py,sha256=kwGvH8F5oqm2O4kLs-UtPfcY0AYiy5ZDUg-Sh3iYJA4,1627
-indoxrouter/client.py,sha256=SQ0ulzkHQZBKFK1dyywjZI_vUQfEd0qVBYQx_8mu7uU,30533
-indoxrouter/constants.py,sha256=tEmfhfCpuKVos1TxVhJfPlv1P0ePWnjnwso4ZHJ0VhM,1190
-indoxrouter/exceptions.py,sha256=qs7f9AnJ7SkOyf9N5qRaZIKpECE8uBq1Pvcg19Jif-U,1718
-indoxrouter-0.1.16.dist-info/METADATA,sha256=wE-sWdHgDNp42hULYxQTx5bO-PB_ydwyvZn4iOTIFYM,5692
-indoxrouter-0.1.16.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-indoxrouter-0.1.16.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
-indoxrouter-0.1.16.dist-info/RECORD,,