Add connection_timeout and timeout config for API client

- Add connection_timeout (10.0s) and timeout (36000.0s / 10 hours) to config.api
- Apply timeout pattern from wsai.py: (min(connection_timeout, timeout), timeout)
- Update upload to use configured timeouts

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <[email protected]>

Author: chengzeyi

src/wavespeed/api/__init__.py

@@ -73,11 +73,12 @@ def run(
     )
 
 
-def upload(file: str | BinaryIO) -> str:
+def upload(file: str | BinaryIO, *, timeout: float | None = None) -> str:
     """Upload a file to WaveSpeed.
 
     Args:
         file: File path string or file-like object to upload.
+        timeout: Total API call timeout in seconds.
 
     Returns:
         URL of the uploaded file.
@@ -91,4 +92,4 @@ def upload(file: str | BinaryIO) -> str:
         url = wavespeed.upload("/path/to/image.png")
         print(url)
     """
-    return _get_default_client().upload(file)
+    return _get_default_client().upload(file, timeout=timeout)

src/wavespeed/api/client.py

@@ -15,6 +15,7 @@ class Client:
     Args:
         api_key: WaveSpeed API key. If not provided, uses wavespeed.config.api.api_key.
         base_url: Base URL for the API. If not provided, uses wavespeed.config.api.base_url.
+        connection_timeout: Timeout for HTTP requests in seconds.
 
     Example:
         client = Client(api_key="your-api-key")
@@ -25,10 +26,12 @@ def __init__(
         self,
         api_key: str | None = None,
         base_url: str | None = None,
+        connection_timeout: float | None = None,
     ) -> None:
         """Initialize the client."""
         self.api_key = api_key or api_config.api_key
         self.base_url = (base_url or api_config.base_url).rstrip("/")
+        self.connection_timeout = connection_timeout or api_config.connection_timeout
 
     def _get_headers(self) -> dict[str, str]:
         """Get request headers with authentication."""
@@ -58,7 +61,9 @@ def _submit(self, model: str, input: dict[str, Any] | None) -> str:
         url = f"{self.base_url}/api/v3/{model}"
         body = input or {}
 
-        response = requests.post(url, json=body, headers=self._get_headers())
+        response = requests.post(
+            url, json=body, headers=self._get_headers(), timeout=self.connection_timeout
+        )
 
         if response.status_code != 200:
             raise RuntimeError(
@@ -88,7 +93,9 @@ def _get_result(self, request_id: str) -> dict[str, Any]:
         """
         url = f"{self.base_url}/api/v3/predictions/{request_id}/result"
 
-        response = requests.get(url, headers=self._get_headers())
+        response = requests.get(
+            url, headers=self._get_headers(), timeout=self.connection_timeout
+        )
 
         if response.status_code != 200:
             raise RuntimeError(
@@ -166,11 +173,12 @@ def run(
         request_id = self._submit(model, input)
         return self._wait(request_id, timeout, poll_interval)
 
-    def upload(self, file: str | BinaryIO) -> str:
+    def upload(self, file: str | BinaryIO, *, timeout: float | None = None) -> str:
         """Upload a file to WaveSpeed.
 
         Args:
             file: File path string or file-like object to upload.
+            timeout: Total API call timeout in seconds.
 
         Returns:
             URL of the uploaded file.
@@ -192,19 +200,25 @@ def upload(self, file: str | BinaryIO) -> str:
 
         url = f"{self.base_url}/api/v3/media/upload/binary"
         headers = {"Authorization": f"Bearer {self.api_key}"}
+        timeout = timeout or api_config.timeout
+        request_timeout = (min(self.connection_timeout, timeout), timeout)
 
         if isinstance(file, str):
             if not os.path.exists(file):
                 raise FileNotFoundError(f"File not found: {file}")
             with open(file, "rb") as f:
                 files = {"file": (os.path.basename(file), f)}
-                response = requests.post(url, headers=headers, files=files)
+                response = requests.post(
+                    url, headers=headers, files=files, timeout=request_timeout
+                )
         else:
             filename = getattr(file, "name", "upload")
             if isinstance(filename, str) and os.path.sep in filename:
                 filename = os.path.basename(filename)
             files = {"file": (filename, file)}
-            response = requests.post(url, headers=headers, files=files)
+            response = requests.post(
+                url, headers=headers, files=files, timeout=request_timeout
+            )
 
         if response.status_code != 200:
             raise RuntimeError(

src/wavespeed/config.py

@@ -21,6 +21,12 @@ class api:
     # API base URL
     base_url: str = "https://api.wavespeed.ai"
 
+    # Connection timeout in seconds
+    connection_timeout: float = 10.0
+
+    # Total API call timeout in seconds
+    timeout: float = 36000.0
+
 
 class serverless:
     """Serverless configuration options.