wavespeed 0.0.2__tar.gz → 0.0.4__tar.gz

{wavespeed-0.0.2 → wavespeed-0.0.4}/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2025 Wavespeed AI
+Copyright (c) 2025 WaveSpeed AI
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

{wavespeed-0.0.2 → wavespeed-0.0.4}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.0.2
-Summary: Python client for Wavespeed
-Author: Wavespeed, Inc.
+Version: 0.0.4
+Summary: Python client for WaveSpeed AI
+Author: WaveSpeed AI
 License: MIT License
         
-        Copyright (c) 2025 Wavespeed AI
+        Copyright (c) 2025 WaveSpeed AI
         
         Permission is hereby granted, free of charge, to any person obtaining a copy
         of this software and associated documentation files (the "Software"), to deal
@@ -35,9 +35,9 @@ Requires-Dist: pydantic>1.10.7
 Requires-Dist: typing_extensions>=4.5.0
 Dynamic: license-file
 
-# Wavespeed Python Client
+# WaveSpeed Python Client
 
-A Python client for interacting with the Wavespeed AI API.
+A Python client for interacting with the WaveSpeed AI API.
 
 ## Installation
 
@@ -50,10 +50,10 @@ pip install wavespeed
 ### Synchronous Image Generation
 
 ```python
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-client = Wavespeed(api_key="YOUR_API_KEY")
+client = WaveSpeed(api_key="YOUR_API_KEY")
 
 # Generate an image and wait for the result
 prediction = client.run(
@@ -78,11 +78,11 @@ for i, img_url in enumerate(prediction.outputs):
 
 ```python
 import asyncio
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 async def generate_image():
     # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-    client = Wavespeed(api_key="YOUR_API_KEY")
+    client = WaveSpeed(api_key="YOUR_API_KEY")
     
     try:
         # Generate an image and wait for the result
@@ -115,10 +115,10 @@ asyncio.run(generate_image())
 You can also create a prediction without waiting for it to complete:
 
 ```python
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-client = Wavespeed(api_key="YOUR_API_KEY")
+client = WaveSpeed(api_key="YOUR_API_KEY")
 
 # Create a prediction without waiting
 prediction = client.create(
@@ -189,15 +189,15 @@ python examples/create_generate_image.py --prompt "A futuristic cityscape with f
 
 ## API Reference
 
-### Wavespeed Client
+### WaveSpeed Client
 
 ```python
-Wavespeed(api_key)
+WaveSpeed(api_key)
 ```
 
 #### Parameters:
 
-- `api_key` (str): Your Wavespeed API key
+- `api_key` (str): Your WaveSpeed API key
 
 ### Methods
 
@@ -241,8 +241,10 @@ The `Prediction` object contains information about an image generation job:
 prediction.id           # Unique ID of the prediction
 prediction.model        # Model ID used for the prediction
 prediction.status       # Status of the prediction (processing, completed, failed)
-prediction.outputs      # List of output image URLs
 prediction.input        # Input parameters used for the prediction
+prediction.outputs      # List of output image URLs
+prediction.urls.get    # URL to get the prediction status
+prediction.has_nsfw_contents # List of booleans indicating if each image has NSFW content
 prediction.created_at   # Creation timestamp
 prediction.error        # Error message (if any)
 prediction.executionTime # Time taken to execute the prediction in milliseconds
@@ -261,7 +263,7 @@ await prediction.async_reload() -> Prediction
 
 ## Environment Variables
 
-- `WAVESPEED_API_KEY`: Your Wavespeed API key
+- `WAVESPEED_API_KEY`: Your WaveSpeed API key
 - `WAVESPEED_POLL_INTERVAL`: Interval in seconds for polling prediction status (default: 1)
 - `WAVESPEED_TIMEOUT`: Timeout in seconds for API requests (default: 60)
 

{wavespeed-0.0.2 → wavespeed-0.0.4}/README.md

@@ -1,6 +1,6 @@
-# Wavespeed Python Client
+# WaveSpeed Python Client
 
-A Python client for interacting with the Wavespeed AI API.
+A Python client for interacting with the WaveSpeed AI API.
 
 ## Installation
 
@@ -13,10 +13,10 @@ pip install wavespeed
 ### Synchronous Image Generation
 
 ```python
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-client = Wavespeed(api_key="YOUR_API_KEY")
+client = WaveSpeed(api_key="YOUR_API_KEY")
 
 # Generate an image and wait for the result
 prediction = client.run(
@@ -41,11 +41,11 @@ for i, img_url in enumerate(prediction.outputs):
 
 ```python
 import asyncio
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 async def generate_image():
     # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-    client = Wavespeed(api_key="YOUR_API_KEY")
+    client = WaveSpeed(api_key="YOUR_API_KEY")
     
     try:
         # Generate an image and wait for the result
@@ -78,10 +78,10 @@ asyncio.run(generate_image())
 You can also create a prediction without waiting for it to complete:
 
 ```python
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-client = Wavespeed(api_key="YOUR_API_KEY")
+client = WaveSpeed(api_key="YOUR_API_KEY")
 
 # Create a prediction without waiting
 prediction = client.create(
@@ -152,15 +152,15 @@ python examples/create_generate_image.py --prompt "A futuristic cityscape with f
 
 ## API Reference
 
-### Wavespeed Client
+### WaveSpeed Client
 
 ```python
-Wavespeed(api_key)
+WaveSpeed(api_key)
 ```
 
 #### Parameters:
 
-- `api_key` (str): Your Wavespeed API key
+- `api_key` (str): Your WaveSpeed API key
 
 ### Methods
 
@@ -204,8 +204,10 @@ The `Prediction` object contains information about an image generation job:
 prediction.id           # Unique ID of the prediction
 prediction.model        # Model ID used for the prediction
 prediction.status       # Status of the prediction (processing, completed, failed)
-prediction.outputs      # List of output image URLs
 prediction.input        # Input parameters used for the prediction
+prediction.outputs      # List of output image URLs
+prediction.urls.get    # URL to get the prediction status
+prediction.has_nsfw_contents # List of booleans indicating if each image has NSFW content
 prediction.created_at   # Creation timestamp
 prediction.error        # Error message (if any)
 prediction.executionTime # Time taken to execute the prediction in milliseconds
@@ -224,7 +226,7 @@ await prediction.async_reload() -> Prediction
 
 ## Environment Variables
 
-- `WAVESPEED_API_KEY`: Your Wavespeed API key
+- `WAVESPEED_API_KEY`: Your WaveSpeed API key
 - `WAVESPEED_POLL_INTERVAL`: Interval in seconds for polling prediction status (default: 1)
 - `WAVESPEED_TIMEOUT`: Timeout in seconds for API requests (default: 60)
 

{wavespeed-0.0.2 → wavespeed-0.0.4}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wavespeed"
-version = "0.0.2"
-description = "Python client for Wavespeed"
+version = "0.0.4"
+description = "Python client for WaveSpeed AI"
 readme = "README.md"
 license = { file = "LICENSE" }
-authors = [{ name = "Wavespeed, Inc." }]
+authors = [{ name = "WaveSpeed AI" }]
 requires-python = ">=3.8"
 dependencies = [
     "httpx>=0.21.0,<1",

{wavespeed-0.0.2 → wavespeed-0.0.4}/tests/test_client.py

@@ -5,23 +5,22 @@ Tests for the Wavespeed client.
 import pytest
 import respx
 from httpx import Response
-import asyncio
 from datetime import datetime
 
-from wavespeed.client import Wavespeed
+from wavespeed.client import WaveSpeed
 from wavespeed.schemas.prediction import Prediction, PredictionUrls
 
 
 @pytest.fixture
 def client():
     """Create a test client with a dummy API key."""
-    return Wavespeed(api_key="test_api_key")
+    return WaveSpeed(api_key="test_api_key")
 
 
 @pytest.fixture
 async def async_client():
     """Create a test client with a dummy API key and clean up after test."""
-    client = Wavespeed(api_key="test_api_key")
+    client = WaveSpeed(api_key="test_api_key")
     yield client
     await client.close()
 

{wavespeed-0.0.2 → wavespeed-0.0.4}/wavespeed/__init__.py

@@ -6,11 +6,11 @@ A Python client for interacting with the Wavespeed AI API.
 
 __version__ = "0.1.0"
 
-from .client import Wavespeed
+from .client import WaveSpeed
 from .schemas.prediction import Prediction, PredictionUrls, PredictionResponse
 
 __all__ = [
-    "Wavespeed",
+    "WaveSpeed",
     "Prediction",
     "PredictionUrls",
     "PredictionResponse",

{wavespeed-0.0.2 → wavespeed-0.0.4}/wavespeed/client.py

@@ -6,33 +6,34 @@ from urllib.parse import urljoin
 from wavespeed.schemas.prediction import Prediction
 
 
-class Wavespeed:
+class WaveSpeed:
     """
     A client for interacting with the Wavespeed AI API.
     """
     
-    BASE_URL = "https://api.wavespeed.ai/api/v2/"
-    
-    def __init__(self, api_key: str):
+    def __init__(self, api_key: str="", base_url: str = "https://api.wavespeed.ai/api/v2/", timeout: int | None = None):
         """
-        Initialize the Wavespeed client.
+        Initialize the WaveSpeed client.
         
         Args:
-            api_key: Your Wavespeed API key
+            api_key: Your WaveSpeed API key
+            base_url: Base URL for the API
+            timeout: Timeout in seconds for http client send request
         """
         self.api_key = api_key
-        if not api_key:
-            api_key = os.environ.get("WAVESPEED_API_KEY", '')
-        if not api_key:
+        if not self.api_key:
+            self.api_key = os.environ.get("WAVESPEED_API_KEY", '')
+        if not self.api_key:
             raise ValueError("API key is required.")
         self.headers = {
             "Content-Type": "application/json",
-            "Authorization": f"Bearer {api_key}"
+            "Authorization": f"Bearer {self.api_key}"
         }
-        self.async_client = httpx.AsyncClient(headers=self.headers)
-        self.client = httpx.Client(headers=self.headers)
-        self.poll_interval = float(os.environ.get("WAVESPEED_POLL_INTERVAL", 1)) # seconds
-        self.timeout = int(os.environ.get("WAVESPEED_TIMEOUT", 60)) # seconds
+        self.async_client = httpx.AsyncClient(headers=self.headers, timeout=timeout)
+        self.client = httpx.Client(headers=self.headers, timeout=timeout)
+        self.base_url = base_url
+        self.timeout = timeout
+        self.poll_interval = float(os.environ.get("WAVESPEED_POLL_INTERVAL", 1))
     
     async def async_run(
         self,
@@ -51,7 +52,7 @@ class Wavespeed:
         Returns:
             The API response as a dictionary
         """
-        url = urljoin(self.BASE_URL, modelId)
+        url = urljoin(self.base_url, modelId)
         
         payload = input
         
@@ -63,7 +64,6 @@ class Wavespeed:
             url,
             headers=self.headers,
             json=payload,
-            timeout=self.timeout
         )
         
         # Raise an exception for HTTP errors
@@ -90,7 +90,7 @@ class Wavespeed:
         Returns:
             The API response as a dictionary
         """
-        url = urljoin(self.BASE_URL, modelId)
+        url = urljoin(self.base_url, modelId)
         
         payload = input
         
@@ -98,7 +98,6 @@ class Wavespeed:
             url,
             headers=self.headers,
             json=payload,
-            timeout=self.timeout
         )
 
         # Raise an exception for HTTP errors
@@ -109,13 +108,12 @@ class Wavespeed:
         return prediction.wait()
     
     async def async_create(self, modelId: str, input: Dict[str, Any], **kwargs) -> Prediction:
-        url = urljoin(self.BASE_URL, modelId)
+        url = urljoin(self.base_url, modelId)
         payload = input
         response = await self.async_client.post(
             url,
             headers=self.headers,
             json=payload,
-            timeout=self.timeout
         )
         # Raise an exception for HTTP errors
         response.raise_for_status()
@@ -125,13 +123,12 @@ class Wavespeed:
         return prediction
     
     def create(self, modelId: str, input: Dict[str, Any], **kwargs) -> Prediction:
-        url = urljoin(self.BASE_URL, modelId)
+        url = urljoin(self.base_url, modelId)
         payload = input
         response = self.client.post(
             url,
             headers=self.headers,
             json=payload,
-            timeout=self.timeout
         )
         # Raise an exception for HTTP errors
         response.raise_for_status()
@@ -147,4 +144,4 @@ class Wavespeed:
         
     def __str__(self) -> str:
         """String representation of the client."""
-        return f"WavespeedClient()"
+        return f"WaveSpeed()"

{wavespeed-0.0.2 → wavespeed-0.0.4}/wavespeed/schemas/prediction.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel, Field
 import pydantic
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
-    from wavespeed.client import Wavespeed
+    from wavespeed.client import WaveSpeed
 
 class PredictionUrls(BaseModel):
     """URLs associated with a prediction."""
@@ -16,7 +16,7 @@ class PredictionUrls(BaseModel):
 
 class Prediction(BaseModel):
     """Response from a prediction API call."""
-    _client: "Wavespeed" = pydantic.PrivateAttr()
+    _client: "WaveSpeed" = pydantic.PrivateAttr()
 
     id: str
     model: str

{wavespeed-0.0.2 → wavespeed-0.0.4}/wavespeed.egg-info/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.0.2
-Summary: Python client for Wavespeed
-Author: Wavespeed, Inc.
+Version: 0.0.4
+Summary: Python client for WaveSpeed AI
+Author: WaveSpeed AI
 License: MIT License
         
-        Copyright (c) 2025 Wavespeed AI
+        Copyright (c) 2025 WaveSpeed AI
         
         Permission is hereby granted, free of charge, to any person obtaining a copy
         of this software and associated documentation files (the "Software"), to deal
@@ -35,9 +35,9 @@ Requires-Dist: pydantic>1.10.7
 Requires-Dist: typing_extensions>=4.5.0
 Dynamic: license-file
 
-# Wavespeed Python Client
+# WaveSpeed Python Client
 
-A Python client for interacting with the Wavespeed AI API.
+A Python client for interacting with the WaveSpeed AI API.
 
 ## Installation
 
@@ -50,10 +50,10 @@ pip install wavespeed
 ### Synchronous Image Generation
 
 ```python
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-client = Wavespeed(api_key="YOUR_API_KEY")
+client = WaveSpeed(api_key="YOUR_API_KEY")
 
 # Generate an image and wait for the result
 prediction = client.run(
@@ -78,11 +78,11 @@ for i, img_url in enumerate(prediction.outputs):
 
 ```python
 import asyncio
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 async def generate_image():
     # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-    client = Wavespeed(api_key="YOUR_API_KEY")
+    client = WaveSpeed(api_key="YOUR_API_KEY")
     
     try:
         # Generate an image and wait for the result
@@ -115,10 +115,10 @@ asyncio.run(generate_image())
 You can also create a prediction without waiting for it to complete:
 
 ```python
-from wavespeed import Wavespeed
+from wavespeed import WaveSpeed
 
 # Initialize the client with your API key (or set WAVESPEED_API_KEY environment variable)
-client = Wavespeed(api_key="YOUR_API_KEY")
+client = WaveSpeed(api_key="YOUR_API_KEY")
 
 # Create a prediction without waiting
 prediction = client.create(
@@ -189,15 +189,15 @@ python examples/create_generate_image.py --prompt "A futuristic cityscape with f
 
 ## API Reference
 
-### Wavespeed Client
+### WaveSpeed Client
 
 ```python
-Wavespeed(api_key)
+WaveSpeed(api_key)
 ```
 
 #### Parameters:
 
-- `api_key` (str): Your Wavespeed API key
+- `api_key` (str): Your WaveSpeed API key
 
 ### Methods
 
@@ -241,8 +241,10 @@ The `Prediction` object contains information about an image generation job:
 prediction.id           # Unique ID of the prediction
 prediction.model        # Model ID used for the prediction
 prediction.status       # Status of the prediction (processing, completed, failed)
-prediction.outputs      # List of output image URLs
 prediction.input        # Input parameters used for the prediction
+prediction.outputs      # List of output image URLs
+prediction.urls.get    # URL to get the prediction status
+prediction.has_nsfw_contents # List of booleans indicating if each image has NSFW content
 prediction.created_at   # Creation timestamp
 prediction.error        # Error message (if any)
 prediction.executionTime # Time taken to execute the prediction in milliseconds
@@ -261,7 +263,7 @@ await prediction.async_reload() -> Prediction
 
 ## Environment Variables
 
-- `WAVESPEED_API_KEY`: Your Wavespeed API key
+- `WAVESPEED_API_KEY`: Your WaveSpeed API key
 - `WAVESPEED_POLL_INTERVAL`: Interval in seconds for polling prediction status (default: 1)
 - `WAVESPEED_TIMEOUT`: Timeout in seconds for API requests (default: 60)