wavespeed 0.0.1__tar.gz → 0.0.3__tar.gz

{wavespeed-0.0.1 → wavespeed-0.0.3}/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.1 → wavespeed-0.0.3}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.0.1
-Summary: Python client for Wavespeed
-Author: Wavespeed, Inc.
+Version: 0.0.3
+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.1 → wavespeed-0.0.3}/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.1 → wavespeed-0.0.3}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wavespeed"
-version = "0.0.1"
-description = "Python client for Wavespeed"
+version = "0.0.3"
+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",
@@ -37,7 +37,7 @@ asyncio_mode = "auto"
 testpaths = "tests/"
 
 [tool.setuptools]
-packages = ["wavespeed"]
+packages = ["wavespeed", "wavespeed.schemas"]
 
 [tool.setuptools.package-data]
 "wavespeed" = ["py.typed"]

{wavespeed-0.0.1 → wavespeed-0.0.3}/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.1 → wavespeed-0.0.3}/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.1 → wavespeed-0.0.3}/wavespeed/client.py

@@ -6,7 +6,7 @@ from urllib.parse import urljoin
 from wavespeed.schemas.prediction import Prediction
 
 
-class Wavespeed:
+class WaveSpeed:
     """
     A client for interacting with the Wavespeed AI API.
     """
@@ -147,4 +147,4 @@ class Wavespeed:
         
     def __str__(self) -> str:
         """String representation of the client."""
-        return f"WavespeedClient()"
+        return f"WaveSpeed()"

wavespeed-0.0.3/wavespeed/schemas/__init__.py

@@ -0,0 +1,7 @@
+"""
+Schemas for the Wavespeed API.
+"""
+
+from .prediction import PredictionUrls, Prediction, PredictionResponse
+
+__all__ = [ "PredictionUrls", "Prediction", "PredictionResponse"]

wavespeed-0.0.3/wavespeed/schemas/prediction.py

@@ -0,0 +1,76 @@
+import asyncio
+import logging
+import time
+from typing import List, Dict, Any, Optional
+from datetime import datetime
+from pydantic import BaseModel, Field
+import pydantic
+from typing import TYPE_CHECKING
+if TYPE_CHECKING:
+    from wavespeed.client import WaveSpeed
+
+class PredictionUrls(BaseModel):
+    """URLs associated with a prediction."""
+    get: str
+
+
+class Prediction(BaseModel):
+    """Response from a prediction API call."""
+    _client: "WaveSpeed" = pydantic.PrivateAttr()
+
+    id: str
+    model: str
+    input: Dict[str, Any] | None = None
+    outputs: List[str]
+    urls: PredictionUrls
+    has_nsfw_contents: List[bool]
+    status: str
+    created_at: datetime
+    error: str = ""
+    executionTime: int
+
+    def wait(self) -> "Prediction":
+        while self.status not in ['completed', 'failed']:
+            time.sleep(self._client.poll_interval)
+            print('Waiting for prediction to complete: ', self.urls.get, type(self.urls.get))
+            response = self._client.client.get(self.urls.get)
+            response.raise_for_status()
+            data = response.json()['data']
+            self._update_from_dict(data)
+        return self
+    
+    async def async_wait(self) -> "Prediction":
+        while self.status not in ['completed', 'failed']:
+            await asyncio.sleep(self._client.poll_interval)
+            response = await self._client.async_client.get(self.urls.get)
+            response.raise_for_status()
+            data = response.json()['data']
+            self._update_from_dict(data)
+        return self
+    
+    async def async_reload(self) -> "Prediction":
+        response = await self._client.async_client.get(self.urls.get)
+        response.raise_for_status()
+        data = response.json()['data']
+        self._update_from_dict(data)
+        return self
+    
+    def reload(self) -> "Prediction":
+        response = self._client.client.get(self.urls.get)
+        response.raise_for_status()
+        data = response.json()['data']
+        self._update_from_dict(data)
+        return self
+    
+    def _update_from_dict(self, data: Dict[str, Any]) -> None:
+        """Update the object from a dictionary, handling nested objects properly."""
+        for key, value in data.items():
+            if key == 'urls' and isinstance(value, dict):
+                self.urls = PredictionUrls(**value)
+            elif hasattr(self, key):
+                setattr(self, key, value)
+
+class PredictionResponse(BaseModel):
+    code: int
+    message: str
+    data: Prediction

{wavespeed-0.0.1 → wavespeed-0.0.3}/wavespeed.egg-info/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.0.1
-Summary: Python client for Wavespeed
-Author: Wavespeed, Inc.
+Version: 0.0.3
+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.1 → wavespeed-0.0.3}/wavespeed.egg-info/SOURCES.txt

@@ -8,4 +8,6 @@ wavespeed.egg-info/PKG-INFO
 wavespeed.egg-info/SOURCES.txt
 wavespeed.egg-info/dependency_links.txt
 wavespeed.egg-info/requires.txt
-wavespeed.egg-info/top_level.txt
+wavespeed.egg-info/top_level.txt
+wavespeed/schemas/__init__.py
+wavespeed/schemas/prediction.py