Expose replicate.use()

Author: dgellow

src/replicate/__init__.py

@@ -243,6 +243,7 @@ def _reset_client() -> None:  # type: ignore[reportUnusedFunction]
 
 from ._module_client import (
     run as run,
+    use as use,
     files as files,
     models as models,
     account as account,

src/replicate/_module_client.py

@@ -77,16 +77,21 @@ def __load__(self) -> PredictionsResource:
 if TYPE_CHECKING:
     from ._client import Replicate
 
-    # get the type checker to infer the run symbol to the same type
+    # get the type checker to infer the run and use symbols to the same type
     # as the method on the client so we don't have to define it twice
     __client: Replicate = cast(Replicate, {})
     run = __client.run
+    use = __client.use
 else:
 
     def _run(*args, **kwargs):
         return _load_client().run(*args, **kwargs)
 
+    def _use(*args, **kwargs):
+        return _load_client().use(*args, **kwargs)
+
     run = _run
+    use = _use
 
 files: FilesResource = FilesResourceProxy().__as_proxied__()
 models: ModelsResource = ModelsResourceProxy().__as_proxied__()