mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 10:25:23 -04:00
Kalo Chin
d8f9286eb3
- Added exception handling for unsupported protocols in the `get_blob` method. - Raises a ValueError with a clear message if the URL is invalid, guiding users to check the `FILES_URL` environment variable. - Maintained existing functionality while improving robustness against HTTP request failures.
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
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.
|
|
|
|
Raises:
|
|
ValueError: If the URL uses an unsupported protocol (e.g., missing 'http://' or 'https://'),
|
|
suggesting configuration of the FILES_URL environment variable.
|
|
httpx.HTTPStatusError: If the request to fetch the file fails.
|
|
"""
|
|
if self._blob is None:
|
|
try:
|
|
response = httpx.get(self.url)
|
|
response.raise_for_status()
|
|
self._blob = response.content
|
|
except httpx.UnsupportedProtocol as e:
|
|
raise ValueError(
|
|
f"Invalid file URL '{self.url}': {e}. "
|
|
"Ensure the `FILES_URL` environment variable is set in your .env file"
|
|
) from e
|
|
|
|
assert self._blob is not None
|
|
return self._blob
|