大活

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

@@ -0,0 +1,177 @@
+import io
+import json
+import mimetypes
+import os
+import pathlib
+from typing import Any, BinaryIO, Dict, List, Optional, TypedDict, Union
+
+from typing_extensions import Literal, NotRequired, Unpack
+
+from replicate.resource import Namespace, Resource
+
+FileEncodingStrategy = Literal["base64", "url"]
+
+
+class File(Resource):
+    """
+    A file uploaded to Replicate that can be used as an input to a model.
+    """
+
+    id: str
+    """The ID of the file."""
+
+    name: str
+    """The name of the file."""
+
+    content_type: str
+    """The content type of the file."""
+
+    size: int
+    """The size of the file in bytes."""
+
+    etag: str
+    """The ETag of the file."""
+
+    checksums: Dict[str, str]
+    """The checksums of the file."""
+
+    metadata: Dict[str, Any]
+    """The metadata of the file."""
+
+    created_at: str
+    """The time the file was created."""
+
+    expires_at: Optional[str]
+    """The time the file will expire."""
+
+    urls: Dict[str, str]
+    """The URLs of the file."""
+
+
+class Files(Namespace):
+    class CreateFileParams(TypedDict):
+        """Parameters for creating a file."""
+
+        filename: NotRequired[str]
+        """The name of the file."""
+
+        content_type: NotRequired[str]
+        """The content type of the file."""
+
+        metadata: NotRequired[Dict[str, Any]]
+        """The file metadata."""
+
+    def create(
+        self,
+        file: Union[str, pathlib.Path, BinaryIO, io.IOBase],
+        **params: Unpack["Files.CreateFileParams"],
+    ) -> File:
+        """
+        Upload a file that can be passed as an input when running a model.
+        """
+
+        if isinstance(file, (str, pathlib.Path)):
+            file_path = pathlib.Path(file)
+            params["filename"] = file_path.name
+            with open(file, "rb") as f:
+                return self.create(f, **params)
+        elif not isinstance(file, (io.IOBase, BinaryIO)):
+            raise ValueError(
+                "Unsupported file type. Must be a file path or file-like object."
+            )
+
+        resp = self._client._request(
+            "POST", "/v1/files", timeout=None, **_create_file_params(file, **params)
+        )
+
+        return _json_to_file(resp.json())
+
+    async def async_create(
+        self,
+        file: Union[str, pathlib.Path, BinaryIO, io.IOBase],
+        **params: Unpack["Files.CreateFileParams"],
+    ) -> File:
+        """Upload a file asynchronously that can be passed as an input when running a model."""
+
+        if isinstance(file, (str, pathlib.Path)):
+            file_path = pathlib.Path(file)
+            params["filename"] = file_path.name
+            with open(file_path, "rb") as f:
+                return await self.async_create(f, **params)
+        elif not isinstance(file, (io.IOBase, BinaryIO)):
+            raise ValueError(
+                "Unsupported file type. Must be a file path or file-like object."
+            )
+
+        resp = await self._client._async_request(
+            "POST", "/v1/files", timeout=None, **_create_file_params(file, **params)
+        )
+
+        return _json_to_file(resp.json())
+
+    def get(self, file_id: str) -> File:
+        """Get an uploaded file by its ID."""
+
+        resp = self._client._request("GET", f"/v1/files/{file_id}")
+        return _json_to_file(resp.json())
+
+    async def async_get(self, file_id: str) -> File:
+        """Get an uploaded file by its ID asynchronously."""
+
+        resp = await self._client._async_request("GET", f"/v1/files/{file_id}")
+        return _json_to_file(resp.json())
+
+    def list(self) -> List[File]:
+        """List all uploaded files."""
+
+        resp = self._client._request("GET", "/v1/files")
+        return [_json_to_file(obj) for obj in resp.json().get("results", [])]
+
+    async def async_list(self) -> List[File]:
+        """List all uploaded files asynchronously."""
+
+        resp = await self._client._async_request("GET", "/v1/files")
+        return [_json_to_file(obj) for obj in resp.json().get("results", [])]
+
+    def delete(self, file_id: str) -> bool:
+        """Delete an uploaded file by its ID."""
+
+        resp = self._client._request("DELETE", f"/v1/files/{file_id}")
+        return resp.status_code == 204
+
+    async def async_delete(self, file_id: str) -> bool:
+        """Delete an uploaded file by its ID asynchronously."""
+
+        resp = await self._client._async_request("DELETE", f"/v1/files/{file_id}")
+        return resp.status_code == 204
+
+
+def _create_file_params(
+    file: Union[BinaryIO, io.IOBase],
+    **params: Unpack["Files.CreateFileParams"],
+) -> Dict[str, Any]:
+    file.seek(0)
+
+    if params is None:
+        params = {}
+
+    filename = params.get("filename", os.path.basename(getattr(file, "name", "file")))
+    content_type = (
+        params.get("content_type")
+        or mimetypes.guess_type(filename)[0]
+        or "application/octet-stream"
+    )
+    metadata = params.get("metadata")
+
+    data = {}
+    if metadata:
+        data["metadata"] = json.dumps(metadata)
+
+    return {
+        "files": {"content": (filename, file, content_type)},
+        "data": data,
+    }
+
+
+def _json_to_file(json: Dict[str, Any]) -> File:  # pylint: disable=redefined-outer-name
+    return File(**json)