fix files uploading

Author: dtmeadows

src/replicate/lib/_files.py

@@ -51,9 +51,8 @@ def encode_json(
         if file_encoding_strategy == "base64":
             return base64_encode_file(obj)
         else:
-            # todo: support files endpoint
-            # return client.files.create(obj).urls["get"]
-            raise NotImplementedError("File upload is not supported yet")
+            response = client.files.create(content=obj.read())
+            return response.urls.get
     if HAS_NUMPY:
         if isinstance(obj, np.integer):  # type: ignore
             return int(obj)
@@ -91,9 +90,8 @@ async def async_encode_json(
             # TODO: This should ideally use an async based file reader path.
             return base64_encode_file(obj)
         else:
-            # todo: support files endpoint
-            # return (await client.files.async_create(obj)).urls["get"]
-            raise NotImplementedError("File upload is not supported yet")
+            response = await client.files.create(content=obj.read())
+            return response.urls.get
     if HAS_NUMPY:
         if isinstance(obj, np.integer):  # type: ignore
             return int(obj)

src/replicate/lib/_predictions.py

@@ -215,6 +215,6 @@ def output_iterator(prediction: Prediction, client: Replicate) -> Iterator[Any]:
     if prediction.status == "failed":
         raise ModelError(prediction=prediction)
 
-    output: list[Any] = prediction.output or []  # type: ignore[union-attr]
+    output = prediction.output or []  # type: ignore
     new_output = output[len(previous_output) :]
     yield from new_output

src/replicate/resources/files.py

@@ -58,7 +58,7 @@ def create(
         self,
         *,
         content: FileTypes,
-        filename: str,
+        filename: str | NotGiven = NOT_GIVEN,
         metadata: object | NotGiven = NOT_GIVEN,
         type: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -336,7 +336,7 @@ async def create(
         self,
         *,
         content: FileTypes,
-        filename: str,
+        filename: str | NotGiven = NOT_GIVEN,
         metadata: object | NotGiven = NOT_GIVEN,
         type: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.