Skip downloading files passed directly into other models in use()

Author: aron

replicate/use.py

@@ -225,6 +225,10 @@ def __getattribute__(self, name) -> Any:
         if name in ("__path__", "__target__"):
             return object.__getattribute__(self, name)
 
+        # TODO: We should cover other common properties on Path...
+        if name == "__class__":
+            return Path
+
         return getattr(object.__getattribute__(self, "__path__")(), name)
 
     def __setattr__(self, name, value) -> None:
@@ -332,11 +336,13 @@ def create(self, **inputs: Dict[str, Any]) -> Run:
         """
         Start a prediction with the specified inputs.
         """
-        # Process inputs to convert concatenate OutputIterators to strings
+        # Process inputs to convert concatenate OutputIterators to strings and PathProxy to URLs
         processed_inputs = {}
         for key, value in inputs.items():
             if isinstance(value, OutputIterator) and value.is_concatenate:
                 processed_inputs[key] = str(value)
+            elif isinstance(value, PathProxy):
+                processed_inputs[key] = object.__getattribute__(value, "__target__")
             else:
                 processed_inputs[key] = value
 

tests/test_use.py

@@ -608,6 +608,65 @@ async def test_use_iterator_of_paths_output(use_async_client):
     assert output_list[1].read_bytes() == b"fake image 2 data"
 
 
+@pytest.mark.asyncio
+@pytest.mark.parametrize("use_async_client", [False])
+@respx.mock
+async def test_use_pathproxy_input_conversion(use_async_client):
+    """Test that PathProxy instances are converted to URLs when passed to create()."""
+    mock_model_endpoints()
+
+    # Mock the file download - this should NOT be called
+    file_request_mock = respx.get("https://example.com/input.jpg").mock(
+        return_value=httpx.Response(200, content=b"fake input image data")
+    )
+
+    # Create a PathProxy instance
+    from replicate.use import PathProxy
+
+    path_proxy = PathProxy("https://example.com/input.jpg")
+
+    # Set up a mock for the prediction creation to capture the request
+    request_body = None
+
+    def capture_request(request):
+        nonlocal request_body
+        request_body = request.read()
+        return httpx.Response(
+            201,
+            json={
+                "id": "pred789",
+                "model": "acme/hotdog-detector",
+                "version": "xyz123",
+                "urls": {
+                    "get": "https://api.replicate.com/v1/predictions/pred789",
+                    "cancel": "https://api.replicate.com/v1/predictions/pred789/cancel",
+                },
+                "created_at": "2024-01-01T00:00:00Z",
+                "source": "api",
+                "status": "processing",
+                "input": {"image": "https://example.com/input.jpg"},
+                "output": None,
+                "error": None,
+                "logs": "",
+            },
+        )
+
+    respx.post("https://api.replicate.com/v1/predictions").mock(
+        side_effect=capture_request
+    )
+
+    # Call use and create with PathProxy
+    hotdog_detector = replicate.use("acme/hotdog-detector")
+    run = hotdog_detector.create(image=path_proxy)
+
+    # Verify the request body contains the URL, not the downloaded file
+    parsed_body = json.loads(request_body)
+    assert parsed_body["input"]["image"] == "https://example.com/input.jpg"
+
+    # Assert that the file was never downloaded
+    assert file_request_mock.call_count == 0
+
+
 @pytest.mark.asyncio
 @pytest.mark.parametrize("use_async_client", [False])
 @respx.mock