mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 10:25:23 -04:00
Yeuoly
33 lines
868 B
Python
33 lines
868 B
Python
import httpx
|
|
from pydantic import BaseModel
|
|
|
|
from dify_plugin.file.constants import DIFY_FILE_IDENTITY
|
|
from dify_plugin.file.entities import FileType
|
|
|
|
|
|
class File(BaseModel):
|
|
dify_model_identity: str = DIFY_FILE_IDENTITY
|
|
url: str
|
|
mime_type: str | None = None
|
|
filename: str | None = None
|
|
extension: str | None = None
|
|
size: int | None = None
|
|
type: FileType
|
|
|
|
_blob: bytes | None = None
|
|
|
|
@property
|
|
def blob(self) -> bytes:
|
|
"""
|
|
Get the file content as a bytes object.
|
|
|
|
If the file content is not loaded yet, it will be loaded from the URL and stored in the `_blob` attribute.
|
|
"""
|
|
if self._blob is None:
|
|
response = httpx.get(self.url)
|
|
response.raise_for_status()
|
|
self._blob = response.content
|
|
|
|
assert self._blob is not None
|
|
return self._blob
|