Raise a clear error when a non-seekable upload stream exceeds 200 MB

The stream was read up to the limit plus one byte and that truncated
size was declared to the API, so an oversize stream failed with a raw
HTTP 413 instead of a local error.

Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01EEvHR83ShvnHGR8VgJcsWV

Author: chengzeyi

src/wavespeed/api/client.py

@@ -587,7 +587,8 @@ def upload(self, file: str | BinaryIO, *, timeout: float | None = None) -> str:
             URL of the uploaded file.
 
         Raises:
-            ValueError: If API key is not configured.
+            ValueError: If API key is not configured, or a non-seekable
+                stream exceeds the 200 MB upload limit.
             FileNotFoundError: If file path does not exist.
             RuntimeError: If upload fails.
 
@@ -627,7 +628,10 @@ def upload(self, file: str | BinaryIO, *, timeout: float | None = None) -> str:
                 size = stream.tell() - start
                 stream.seek(start)
             except (AttributeError, OSError, io.UnsupportedOperation):
-                data = stream.read(200 * 1024 * 1024 + 1)
+                max_size = 200 * 1024 * 1024
+                data = stream.read(max_size + 1)
+                if len(data) > max_size:
+                    raise ValueError("File size exceeds the 200 MB limit.")
                 stream = io.BytesIO(data)
                 size = len(data)
 

tests/test_api.py

@@ -582,6 +582,22 @@ def test_upload_file_not_found(self):
         with self.assertRaises(FileNotFoundError):
             client.upload("/nonexistent/path/to/file.png")
 
+    @patch("wavespeed.api.client.requests.post")
+    def test_upload_oversize_stream_raises(self, mock_post):
+        """A non-seekable stream over the limit fails before any request."""
+
+        class Pipe:
+            name = "clip.mp4"
+
+            def read(self, n):
+                return bytes(n)
+
+        client = Client(api_key="test-key")
+        with self.assertRaises(ValueError) as ctx:
+            client.upload(Pipe())
+        self.assertIn("200 MB", str(ctx.exception))
+        mock_post.assert_not_called()
+
     def test_upload_raises_without_api_key(self):
         """Test that upload raises ValueError without API key."""
         client = Client()