indoxrouter 0.1.22__tar.gz → 0.1.23__tar.gz

{indoxrouter-0.1.22 → indoxrouter-0.1.23}/PKG-INFO

@@ -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.22 → indoxrouter-0.1.23}/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 → indoxrouter-0.1.23}/indoxrouter.egg-info/PKG-INFO

@@ -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.22 → indoxrouter-0.1.23}/indoxrouter.egg-info/SOURCES.txt

@@ -3,11 +3,6 @@ README.md
 pyproject.toml
 cookbook/README.md
 cookbook/indoxRouter_cookbook.ipynb
-examples/google_image_generation.py
-examples/image_generation.py
-examples/openai_image_generation.py
-examples/provider_specific_image_generation.py
-examples/xai_image_generation.py
 indoxrouter/__init__.py
 indoxrouter/client.py
 indoxrouter/constants.py

{indoxrouter-0.1.22 → indoxrouter-0.1.23}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "indoxrouter"
-version = "0.1.22"
+version = "0.1.23"
 authors = [
     {name = "indoxRouter Team", email = "ashkan.eskandari.dev@gmail.com"},
 ]

indoxrouter-0.1.22/examples/google_image_generation.py

@@ -1,89 +0,0 @@
-"""
-Example script demonstrating Google image generation through indoxRouter.
-
-This script shows how to generate images using Google's Imagen models
-with their specific parameters, especially aspect_ratio instead of size.
-
-Requirements:
-- Install indoxRouter: pip install indoxrouter
-- Set INDOX_ROUTER_API_KEY environment variable with your API key
-"""
-
-import os
-from indoxrouter import Client
-
-# Initialize client with API key from environment variable
-api_key = os.environ.get("INDOX_ROUTER_API_KEY")
-if not api_key:
-    print("Please set the INDOX_ROUTER_API_KEY environment variable")
-    exit(1)
-
-client = Client(api_key=api_key)
-
-
-def generate_image(model, prompt, **kwargs):
-    """Generate an image with the specified Google model and parameters."""
-    print(f"\n=== Generating image with {model} ===")
-    print(f"Prompt: {prompt}")
-
-    try:
-        response = client.images(prompt=prompt, model=f"google/{model}", **kwargs)
-
-        # Print the URL of the generated image
-        if "data" in response and response["data"] and "url" in response["data"][0]:
-            print(f"Image URL: {response['data'][0]['url']}")
-
-        # Print usage/cost information if available
-        if "usage" in response and "cost" in response["usage"]:
-            print(f"Cost: ${response['usage']['cost']:.4f}")
-
-        return response
-    except Exception as e:
-        print(f"Error: {str(e)}")
-        return None
-
-
-def main():
-    """Main function demonstrating Google image generation."""
-
-    # Basic example with square aspect ratio
-    imagen_square = generate_image(
-        model="imagen-3.0-generate-002",
-        prompt="A serene Japanese garden with a koi pond",
-        aspect_ratio="1:1",  # Use aspect_ratio instead of size
-        response_format="url",
-    )
-
-    # Example with landscape aspect ratio
-    imagen_landscape = generate_image(
-        model="imagen-3.0-generate-002",
-        prompt="A panoramic view of the Grand Canyon at sunset",
-        aspect_ratio="16:9",  # Landscape orientation
-        response_format="url",
-    )
-
-    # Example with portrait aspect ratio
-    imagen_portrait = generate_image(
-        model="imagen-3.0-generate-002",
-        prompt="A tall redwood tree in a misty forest",
-        aspect_ratio="9:16",  # Portrait orientation
-        response_format="url",
-    )
-
-    # Example with additional parameters
-    imagen_advanced = generate_image(
-        model="imagen-3.0-generate-002",
-        prompt="A cyberpunk cityscape with neon lights and flying vehicles",
-        aspect_ratio="4:3",
-        negative_prompt="cartoon, illustration, drawing, painting",  # What to avoid
-        guidance_scale=12.0,  # Controls adherence to the text prompt (7-20)
-        seed=12345,  # For reproducible results
-        response_format="url",
-    )
-
-    # Close the client when done
-    client.close()
-
-
-if __name__ == "__main__":
-    main()

indoxrouter-0.1.22/examples/image_generation.py

@@ -1,141 +0,0 @@
-"""
-Example script demonstrating image generation with various providers through indoxRouter.
-
-This script shows how to generate images using different providers (OpenAI, xAI, Google)
-and how to handle the responses to display or save the generated images.
-
-Requirements:
-- Install indoxRouter: pip install indoxrouter
-- Set INDOX_ROUTER_API_KEY environment variable with your API key
-"""
-
-import os
-import base64
-from io import BytesIO
-import requests
-from datetime import datetime
-
-from indoxrouter import Client
-
-# For displaying images in notebooks
-try:
-    from IPython.display import Image, display
-
-    in_notebook = True
-except ImportError:
-    in_notebook = False
-
-# Initialize client with API key from environment variable
-client = Client()
-
-
-def save_image_from_url(url, filename):
-    """Download and save an image from a URL."""
-    response = requests.get(url)
-    if response.status_code == 200:
-        with open(filename, "wb") as f:
-            f.write(response.content)
-        print(f"Image saved to {filename}")
-    else:
-        print(f"Failed to download image: {response.status_code}")
-
-
-def save_image_from_b64(b64_data, filename):
-    """Save an image from base64 data."""
-    image_data = base64.b64decode(b64_data)
-    with open(filename, "wb") as f:
-        f.write(image_data)
-    print(f"Image saved to {filename}")
-
-
-def generate_and_save_image(provider, model, prompt):
-    """Generate an image and save it to a file."""
-    print(f"\n=== Generating image with {provider}/{model} ===")
-    print(f"Prompt: {prompt}")
-
-    try:
-        # Generate the image
-        response = client.images(
-            prompt=prompt,
-            model=f"{provider}/{model}",
-            size="1024x1024",
-        )
-
-        print(f"Response received from {provider}:")
-        print(f"- Success: {response['success']}")
-        print(f"- Cost: ${response['usage']['cost']:.4f}")
-
-        # Check if we have image data
-        if "data" in response and response["data"]:
-            image_data = response["data"][0]
-
-            # Create a timestamp for unique filenames
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-
-            if "url" in image_data and image_data["url"]:
-                filename = f"{provider}_{model}_{timestamp}.png"
-                save_image_from_url(image_data["url"], filename)
-
-                # Display in notebook if possible
-                if in_notebook:
-                    display(Image(url=image_data["url"]))
-
-            elif "b64_json" in image_data and image_data["b64_json"]:
-                filename = f"{provider}_{model}_{timestamp}.png"
-                save_image_from_b64(image_data["b64_json"], filename)
-
-                # Display in notebook if possible
-                if in_notebook:
-                    display(Image(data=base64.b64decode(image_data["b64_json"])))
-
-            # Check for revised prompt (DALL-E 3 often revises prompts)
-            if "revised_prompt" in image_data and image_data["revised_prompt"]:
-                print(f"Revised prompt: {image_data['revised_prompt']}")
-        else:
-            print("No image data found in response")
-            print("Response:", response)
-
-    except Exception as e:
-        print(f"Error generating image with {provider}/{model}: {str(e)}")
-
-
-# Example prompts
-prompts = {
-    "landscape": "A beautiful mountain landscape with a lake at sunset",
-    "animal": "A cute puppy playing with a ball in a garden",
-    "abstract": "An abstract digital art piece with vibrant colors and geometric shapes",
-    "space": "A realistic view of Earth from space with the moon in the background",
-}
-
-
-def main():
-    # Test OpenAI models
-    generate_and_save_image("openai", "dall-e-2", prompts["landscape"])
-    generate_and_save_image("openai", "dall-e-3", prompts["animal"])
-
-    # Test with gpt-image-1 if available
-    try:
-        generate_and_save_image("openai", "gpt-image-1", prompts["abstract"])
-    except Exception as e:
-        print(f"gpt-image-1 test skipped: {str(e)}")
-
-    # Test xAI models
-    try:
-        generate_and_save_image("xai", "grok-2-image", prompts["space"])
-    except Exception as e:
-        print(f"xAI test skipped: {str(e)}")
-
-    # Test Google models
-    try:
-        generate_and_save_image(
-            "google", "imagen-3.0-generate-002", prompts["abstract"]
-        )
-    except Exception as e:
-        print(f"Google test skipped: {str(e)}")
-
-    # Close the client
-    client.close()
-
-
-if __name__ == "__main__":
-    main()

indoxrouter-0.1.22/examples/openai_image_generation.py

@@ -1,87 +0,0 @@
-"""
-Example script demonstrating OpenAI image generation through indoxRouter.
-
-This script shows how to generate images using different OpenAI models:
-- DALL-E 2
-- DALL-E 3
-- GPT-image-1
-
-Requirements:
-- Install indoxRouter: pip install indoxrouter
-- Set INDOX_ROUTER_API_KEY environment variable with your API key
-"""
-
-import os
-from indoxrouter import Client
-
-# Initialize client with API key from environment variable
-api_key = os.environ.get("INDOX_ROUTER_API_KEY")
-if not api_key:
-    print("Please set the INDOX_ROUTER_API_KEY environment variable")
-    exit(1)
-
-client = Client(api_key=api_key)
-
-
-def generate_image(model, prompt, **kwargs):
-    """Generate an image with the specified model and parameters."""
-    print(f"\n=== Generating image with {model} ===")
-    print(f"Prompt: {prompt}")
-
-    try:
-        response = client.images(prompt=prompt, model=f"openai/{model}", **kwargs)
-
-        # Print the URL of the generated image
-        if "data" in response and response["data"] and "url" in response["data"][0]:
-            print(f"Image URL: {response['data'][0]['url']}")
-
-            # Check for revised prompt (DALL-E 3 often revises prompts)
-            if "revised_prompt" in response["data"][0]:
-                print(f"Revised prompt: {response['data'][0]['revised_prompt']}")
-
-        # Print usage/cost information if available
-        if "usage" in response and "cost" in response["usage"]:
-            print(f"Cost: ${response['usage']['cost']:.4f}")
-
-        return response
-    except Exception as e:
-        print(f"Error: {str(e)}")
-        return None
-
-
-def main():
-    """Main function demonstrating OpenAI image generation."""
-
-    # Basic DALL-E 2 example
-    dalle2_response = generate_image(
-        model="dall-e-2",
-        prompt="A beautiful mountain landscape at sunset",
-        size="1024x1024",  # Required parameter
-        response_format="url",  # Optional parameter
-    )
-
-    # DALL-E 3 with quality and style options
-    dalle3_response = generate_image(
-        model="dall-e-3",
-        prompt="A futuristic city with flying cars and tall skyscrapers",
-        size="1024x1024",  # Required parameter
-        quality="hd",  # Optional parameter ('standard' or 'hd')
-        style="vivid",  # Optional parameter ('vivid' or 'natural')
-        response_format="url",
-    )
-
-    # GPT-image-1 example
-    # Note: For GPT-image-1, don't include response_format unless you need it
-    gpt_image_response = generate_image(
-        model="gpt-image-1",
-        prompt="A whimsical fantasy castle with dragons flying around it",
-        size="1024x1024",  # Required parameter
-        # response_format is omitted as it can cause errors with gpt-image-1
-    )
-
-    # Close the client when done
-    client.close()
-
-
-if __name__ == "__main__":
-    main()

indoxrouter-0.1.22/examples/provider_specific_image_generation.py

@@ -1,191 +0,0 @@
-"""
-Example script demonstrating image generation with different providers through indoxRouter.
-
-This script shows how to generate images using different providers (OpenAI, Google, xAI)
-with their provider-specific parameters and handling.
-
-Requirements:
-- Install indoxRouter: pip install indoxrouter
-- Set INDOX_ROUTER_API_KEY environment variable with your API key
-"""
-
-import os
-import base64
-from io import BytesIO
-import requests
-from datetime import datetime
-
-from indoxrouter import Client
-
-# For displaying images in notebooks
-try:
-    from IPython.display import Image, display
-
-    in_notebook = True
-except ImportError:
-    in_notebook = False
-
-# Initialize client with API key from environment variable
-api_key = os.environ.get("INDOX_ROUTER_API_KEY")
-if not api_key:
-    print("Please set the INDOX_ROUTER_API_KEY environment variable")
-    exit(1)
-
-client = Client(api_key=api_key)
-
-
-def save_image_from_url(url, filename):
-    """Download and save an image from a URL."""
-    response = requests.get(url)
-    if response.status_code == 200:
-        with open(filename, "wb") as f:
-            f.write(response.content)
-        print(f"Image saved to {filename}")
-    else:
-        print(f"Failed to download image: {response.status_code}")
-
-
-def save_image_from_b64(b64_data, filename):
-    """Save an image from base64 data."""
-    image_data = base64.b64decode(b64_data)
-    with open(filename, "wb") as f:
-        f.write(image_data)
-    print(f"Image saved to {filename}")
-
-
-def display_response_image(response, provider, model):
-    """Display image from response and save it."""
-    try:
-        if "data" in response and response["data"]:
-            image_data = response["data"][0]
-
-            # Create a timestamp for unique filenames
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-
-            if "url" in image_data and image_data["url"]:
-                filename = f"{provider}_{model}_{timestamp}.png"
-                save_image_from_url(image_data["url"], filename)
-
-                # Display in notebook if possible
-                if in_notebook:
-                    display(Image(url=image_data["url"]))
-
-            elif "b64_json" in image_data and image_data["b64_json"]:
-                filename = f"{provider}_{model}_{timestamp}.png"
-                save_image_from_b64(image_data["b64_json"], filename)
-
-                # Display in notebook if possible
-                if in_notebook:
-                    display(Image(data=base64.b64decode(image_data["b64_json"])))
-
-            # Check for revised prompt (DALL-E 3 often revises prompts)
-            if "revised_prompt" in image_data and image_data["revised_prompt"]:
-                print(f"Revised prompt: {image_data['revised_prompt']}")
-        else:
-            print("No image data found in response")
-    except Exception as e:
-        print(f"Error displaying/saving image: {str(e)}")
-
-
-def main():
-    """Main function demonstrating image generation with different providers."""
-
-    # Example prompt for all providers
-    prompt = "A tranquil zen garden with cherry blossoms and a small pond"
-
-    print("\n=== OpenAI (DALL-E 2) ===")
-    print("Uses pixel dimensions (e.g., '1024x1024')")
-    try:
-        openai_response = client.images(
-            prompt=prompt,
-            model="openai/dall-e-2",
-            size="1024x1024",  # OpenAI uses pixel dimensions
-        )
-        print(f"Response received - Success: {openai_response.get('success', False)}")
-        print(f"Cost: ${openai_response.get('usage', {}).get('cost', 0):.4f}")
-        display_response_image(openai_response, "openai", "dall-e-2")
-    except Exception as e:
-        print(f"Error with OpenAI: {str(e)}")
-
-    print("\n=== OpenAI (GPT-image-1) ===")
-    print("Uses pixel dimensions (e.g., '1024x1024')")
-    try:
-        # For GPT-image-1, don't include response_format parameter unless you specify it
-        gpt_image_response = client.images(
-            prompt=prompt,
-            model="openai/gpt-image-1",
-            size="1024x1024",  # OpenAI uses pixel dimensions
-            # Note: don't include response_format parameter unless explicitly needed
-        )
-        print(
-            f"Response received - Success: {gpt_image_response.get('success', False)}"
-        )
-        print(f"Cost: ${gpt_image_response.get('usage', {}).get('cost', 0):.4f}")
-        display_response_image(gpt_image_response, "openai", "gpt-image-1")
-    except Exception as e:
-        print(f"Error with GPT-image-1: {str(e)}")
-
-    print("\n=== Google (Imagen) ===")
-    print("Uses aspect ratios (e.g., '1:1', '16:9') instead of pixel dimensions")
-    try:
-        google_response = client.images(
-            prompt=prompt,
-            model="google/imagen-3.0-generate-002",
-            # Note: Use 'aspect_ratio' instead of 'size' for Google
-            aspect_ratio="1:1",  # Google uses aspect ratios
-            negative_prompt="cartoon, illustration, drawing, painting",
-        )
-        print(f"Response received - Success: {google_response.get('success', False)}")
-        print(f"Cost: ${google_response.get('usage', {}).get('cost', 0):.4f}")
-        display_response_image(google_response, "google", "imagen")
-    except Exception as e:
-        print(f"Error with Google: {str(e)}")
-
-    print("\n=== xAI (Grok) ===")
-    print("Does not use size parameter, only supports specific parameters")
-    try:
-        xai_response = client.images(
-            prompt=prompt,
-            model="xai/grok-2-image",
-            # Note: xAI doesn't support size parameter, so we don't include it
-            # Only include specific parameters that xAI supports
-            response_format="url",  # xAI supports response_format
-        )
-        print(f"Response received - Success: {xai_response.get('success', False)}")
-        print(f"Cost: ${xai_response.get('usage', {}).get('cost', 0):.4f}")
-        display_response_image(xai_response, "xai", "grok-2-image")
-    except Exception as e:
-        print(f"Error with xAI: {str(e)}")
-
-    # Advanced example with DALL-E 3
-    print("\n=== OpenAI (DALL-E 3) with Advanced Parameters ===")
-    try:
-        dalle3_response = client.images(
-            prompt="A beautiful underwater scene with coral reef and tropical fish",
-            model="openai/dall-e-3",
-            size="1024x1024",
-            quality="standard",  # 'standard' or 'hd'
-            style="vivid",  # 'vivid' or 'natural'
-            response_format="url",
-        )
-        print(f"Response received - Success: {dalle3_response.get('success', False)}")
-        print(f"Cost: ${dalle3_response.get('usage', {}).get('cost', 0):.4f}")
-        display_response_image(dalle3_response, "openai", "dall-e-3")
-    except Exception as e:
-        print(f"Error with DALL-E 3: {str(e)}")
-
-    print("\n=== Provider-Specific Parameter Examples ===")
-    print("Each provider supports different parameters:")
-    print("OpenAI: size, quality, style, background, response_format...")
-    print("Google: aspect_ratio, negative_prompt, guidance_scale, seed...")
-    print("xAI: n, response_format (minimal parameters)")
-    print(
-        "\nRefer to the documentation for the complete list of parameters for each provider."
-    )
-
-    # Close the client when done
-    client.close()
-
-
-if __name__ == "__main__":
-    main()

indoxrouter-0.1.22/examples/xai_image_generation.py

@@ -1,90 +0,0 @@
-"""
-Example script demonstrating xAI image generation through indoxRouter.
-
-This script shows how to generate images using xAI's Grok model
-with the correct parameters (avoiding 'size' which is not supported).
-
-Requirements:
-- Install indoxRouter: pip install indoxrouter
-- Set INDOX_ROUTER_API_KEY environment variable with your API key
-"""
-
-import os
-from indoxrouter import Client
-
-# Initialize client with API key from environment variable
-api_key = os.environ.get("INDOX_ROUTER_API_KEY")
-if not api_key:
-    print("Please set the INDOX_ROUTER_API_KEY environment variable")
-    exit(1)
-
-client = Client(api_key=api_key)
-
-
-def generate_image(prompt, **kwargs):
-    """Generate an image with xAI's Grok model."""
-    print(f"\n=== Generating image with xAI Grok ===")
-    print(f"Prompt: {prompt}")
-
-    try:
-        response = client.images(
-            prompt=prompt,
-            model="xai/grok-2-image",
-            # DO NOT include 'size' parameter - it will cause an error
-            **kwargs,
-        )
-
-        # Print the URL of the generated image
-        if "data" in response and response["data"] and "url" in response["data"][0]:
-            print(f"Image URL: {response['data'][0]['url']}")
-
-        # Print usage/cost information if available
-        if "usage" in response and "cost" in response["usage"]:
-            print(f"Cost: ${response['usage']['cost']:.4f}")
-
-        return response
-    except Exception as e:
-        print(f"Error: {str(e)}")
-        return None
-
-
-def main():
-    """Main function demonstrating xAI image generation."""
-
-    # Basic example - minimal parameters
-    basic_image = generate_image(
-        prompt="A cybernetic fox in a futuristic city",
-        # No size parameter - xAI doesn't support it
-        response_format="url",  # One of the few parameters xAI supports
-    )
-
-    # Multiple images example
-    multiple_images = generate_image(
-        prompt="A colorful abstract painting of geometric shapes",
-        n=2,  # Generate 2 images
-        response_format="url",
-    )
-
-    # Example with long, detailed prompt
-    detailed_prompt = generate_image(
-        prompt=(
-            "A steampunk-inspired mechanical dragonfly with brass gears, "
-            "copper wings, and glowing blue energy cores, hovering above "
-            "a Victorian-era cityscape with airships in the cloudy sky"
-        ),
-        response_format="url",
-    )
-
-    print("\n=== xAI Parameter Notes ===")
-    print("xAI/Grok supports very few parameters compared to other providers:")
-    print("- prompt: The text prompt (required)")
-    print("- n: Number of images to generate (optional)")
-    print("- response_format: 'url' or 'b64_json' (optional)")
-    print("DO NOT include 'size', 'quality', 'style', or other parameters")
-
-    # Close the client when done
-    client.close()
-
-
-if __name__ == "__main__":
-    main()