wavespeed 0.0.3__tar.gz → 0.0.4__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.0.3
+Version: 0.0.4
 Summary: Python client for WaveSpeed AI
 Author: WaveSpeed AI
 License: MIT License

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wavespeed"
-version = "0.0.3"
+version = "0.0.4"
 description = "Python client for WaveSpeed AI"
 readme = "README.md"
 license = { file = "LICENSE" }

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

@@ -11,28 +11,29 @@ 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()

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.0.3
+Version: 0.0.4
 Summary: Python client for WaveSpeed AI
 Author: WaveSpeed AI
 License: MIT License