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(),