Fix streaming parameter encoding in prediction creation requests (#178

) Resolves #175 --------- Signed-off-by: Mattt Zmuda <mattt@replicate.com>
replicate · Oct 29, 2023 · 6d892b3 · 6d892b3
1 parent 99d6d1e
commit 6d892b3
Show file tree

Hide file tree

Showing 4 changed files with 337 additions and 3 deletions.
diff --git a/replicate/deployment.py b/replicate/deployment.py
@@ -127,7 +127,7 @@ def create(  # type: ignore
         if webhook_events_filter is not None:
             body["webhook_events_filter"] = webhook_events_filter
         if stream is True:
-            body["stream"] = "true"
+            body["stream"] = True
 
         resp = self._client._request(
             "POST",

diff --git a/replicate/prediction.py b/replicate/prediction.py
@@ -194,7 +194,7 @@ def create(  # type: ignore
         """
 
         input = encode_json(input, upload_file=upload_file)
-        body = {
+        body: Dict[str, Any] = {
             "version": version if isinstance(version, str) else version.id,
             "input": input,
         }
@@ -205,7 +205,7 @@ def create(  # type: ignore
         if webhook_events_filter is not None:
             body["webhook_events_filter"] = webhook_events_filter
         if stream is True:
-            body["stream"] = "true"
+            body["stream"] = True
 
         resp = self._client._request(
             "POST",