mirror of
https://github.com/langgenius/dify.git
synced 2026-03-04 23:36:20 +08:00
3f5f893e6c
- Implemented the `exists` method in `SandboxFileSource` and its subclasses to verify the availability of sandbox sources. - Updated `SandboxFileService` to utilize the new `exists` method for improved error handling when listing files and downloading files. - Removed the previous check for storage existence in `archive_source.py` and replaced it with the new method.
33 lines
951 B
Python
33 lines
951 B
Python
from __future__ import annotations
|
|
|
|
import abc
|
|
|
|
from core.sandbox.entities.files import SandboxFileDownloadTicket, SandboxFileNode
|
|
|
|
|
|
class SandboxFileSource(abc.ABC):
|
|
_LIST_TIMEOUT_SECONDS = 30
|
|
_UPLOAD_TIMEOUT_SECONDS = 60 * 10
|
|
_EXPORT_EXPIRES_IN_SECONDS = 60 * 10
|
|
|
|
def __init__(self, *, tenant_id: str, sandbox_id: str):
|
|
self._tenant_id = tenant_id
|
|
self._sandbox_id = sandbox_id
|
|
|
|
@abc.abstractmethod
|
|
def exists(self) -> bool:
|
|
"""Check if the sandbox source exists and is available.
|
|
|
|
Returns:
|
|
True if the sandbox source exists and can be accessed, False otherwise.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
def list_files(self, *, path: str, recursive: bool) -> list[SandboxFileNode]:
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
def download_file(self, *, path: str) -> SandboxFileDownloadTicket:
|
|
raise NotImplementedError
|