Log prediction URLs with env var gate

* Log prediction URLs from the replicate python
SDK if the environment variable
R8_LOG_PREDICTION_URL is set to 1

Author: 8W9aG

replicate/use.py

@@ -3,6 +3,7 @@
 # - [ ] Support file streaming
 import copy
 import hashlib
+import logging
 import os
 import tempfile
 from functools import cached_property
@@ -210,6 +211,12 @@ def _resolve_ref(obj: Any) -> Any:
     return result
 
 
+def _log_prediction_url(id: str) -> None:
+    if os.environ.get("R8_LOG_PREDICTION_URL") != "1":
+        return
+    logging.info("Running prediction https://replicate.com/p/%s", id)
+
+
 T = TypeVar("T")
 
 
@@ -436,6 +443,8 @@ def create(self, *_: Input.args, **inputs: Input.kwargs) -> Run[Output]:
                 model=self._model, input=processed_inputs
             )
 
+        _log_prediction_url(prediction.id)
+
         return Run(
             prediction=prediction,
             schema=self.openapi_schema(),
@@ -649,6 +658,8 @@ async def create(self, *_: Input.args, **inputs: Input.kwargs) -> AsyncRun[Outpu
                 model=model, input=processed_inputs
             )
 
+        _log_prediction_url(prediction.id)
+
         return AsyncRun(
             prediction=prediction,
             schema=await self.openapi_schema(),