大活

art-agent/backend/venv/Lib/site-packages/replicate/hardware.py

@@ -0,0 +1,68 @@
+from typing import TYPE_CHECKING, Any, Dict, List
+
+from typing_extensions import deprecated
+
+from replicate.resource import Namespace, Resource
+
+if TYPE_CHECKING:
+    pass
+
+
+class Hardware(Resource):
+    """
+    Hardware for running a model on Replicate.
+    """
+
+    sku: str
+    """
+    The SKU of the hardware.
+    """
+
+    name: str
+    """
+    The name of the hardware.
+    """
+
+    @property
+    @deprecated("Use `sku` instead of `id`")
+    def id(self) -> str:
+        """
+        DEPRECATED: Use `sku` instead.
+        """
+        return self.sku
+
+
+class HardwareNamespace(Namespace):
+    """
+    Namespace for operations related to hardware.
+    """
+
+    def list(self) -> List[Hardware]:
+        """
+        List all hardware available for you to run models on Replicate.
+
+        Returns:
+            List[Hardware]: A list of hardware.
+        """
+
+        resp = self._client._request("GET", "/v1/hardware")
+        obj = resp.json()
+
+        return [_json_to_hardware(entry) for entry in obj]
+
+    async def async_list(self) -> List[Hardware]:
+        """
+        List all hardware available for you to run models on Replicate.
+
+        Returns:
+            List[Hardware]: A list of hardware.
+        """
+
+        resp = await self._client._async_request("GET", "/v1/hardware")
+        obj = resp.json()
+
+        return [_json_to_hardware(entry) for entry in obj]
+
+
+def _json_to_hardware(json: Dict[str, Any]) -> Hardware:
+    return Hardware(**json)