Actually use PathProxy

Author: aron

replicate/use.py

@@ -154,7 +154,7 @@ def _process_output_with_schema(output: Any, openapi_schema: dict) -> Any:
     # Handle direct string with format=uri
     if output_schema.get("type") == "string" and output_schema.get("format") == "uri":
         if isinstance(output, str) and output.startswith(("http://", "https://")):
-            return _download_file(output)
+            return PathProxy(output)
         return output
 
     # Handle array of strings with format=uri
@@ -163,7 +163,7 @@ def _process_output_with_schema(output: Any, openapi_schema: dict) -> Any:
         if items.get("type") == "string" and items.get("format") == "uri":
             if isinstance(output, list):
                 return [
-                    _download_file(url)
+                    PathProxy(url)
                     if isinstance(url, str) and url.startswith(("http://", "https://"))
                     else url
                     for url in output
@@ -187,15 +187,15 @@ def _process_output_with_schema(output: Any, openapi_schema: dict) -> Any:
                     if isinstance(value, str) and value.startswith(
                         ("http://", "https://")
                     ):
-                        result[prop_name] = _download_file(value)
+                        result[prop_name] = PathProxy(value)
 
                 # Array of files property
                 elif prop_schema.get("type") == "array":
                     items = prop_schema.get("items", {})
                     if items.get("type") == "string" and items.get("format") == "uri":
                         if isinstance(value, list):
                             result[prop_name] = [
-                                _download_file(url)
+                                PathProxy(url)
                                 if isinstance(url, str)
                                 and url.startswith(("http://", "https://"))
                                 else url

tests/test_use.py

@@ -9,6 +9,7 @@
 import respx
 
 import replicate
+from replicate.use import PathProxy
 
 
 class ClientMode(str, Enum):
@@ -540,6 +541,7 @@ async def test_use_path_output(client_mode):
     # Call function with prompt="hello world"
     output = hotdog_detector(prompt="hello world")
 
+    assert isinstance(output, PathProxy)
     assert isinstance(output, Path)
     assert output.exists()
     assert output.read_bytes() == b"fake image data"
@@ -598,6 +600,7 @@ async def test_use_list_of_paths_output(client_mode):
 
     assert isinstance(output, list)
     assert len(output) == 2
+    assert all(isinstance(path, PathProxy) for path in output)
     assert all(isinstance(path, Path) for path in output)
     assert all(path.exists() for path in output)
     assert output[0].read_bytes() == b"fake image 1 data"
@@ -663,6 +666,7 @@ async def test_use_iterator_of_paths_output(client_mode):
     # Convert to list to check contents
     output_list = list(output)
     assert len(output_list) == 2
+    assert all(isinstance(path, PathProxy) for path in output_list)
     assert all(isinstance(path, Path) for path in output_list)
     assert all(path.exists() for path in output_list)
     assert output_list[0].read_bytes() == b"fake image 1 data"