dify/api/enums/__init__.py

65 lines
1.2 KiB
Python

from enum import Enum
from .workflow_nodes import NodeType
class CreatedByRole(str, Enum):
"""
TODO: Need to write docstring
"""
ACCOUNT = "account"
END_USER = "end_user"
class UserFrom(str, Enum):
"""
TODO: Need to write docstring
"""
ACCOUNT = "account"
END_USER = "end-user"
class WorkflowRunTriggeredFrom(str, Enum):
DEBUGGING = "debugging"
APP_RUN = "app-run"
class FileType(str, Enum):
IMAGE = "image"
DOCUMENT = "document"
AUDIO = "audio"
VIDEO = "video"
CUSTOM = "custom"
@staticmethod
def value_of(value):
for member in FileType:
if member.value == value:
return member
raise ValueError(f"No matching enum found for value '{value}'")
class FileTransferMethod(str, Enum):
REMOTE_URL = "remote_url"
LOCAL_FILE = "local_file"
TOOL_FILE = "tool_file"
@staticmethod
def value_of(value):
for member in FileTransferMethod:
if member.value == value:
return member
raise ValueError(f"No matching enum found for value '{value}'")
__all__ = [
"NodeType",
"CreatedByRole",
"UserFrom",
"WorkflowRunTriggeredFrom",
"FileType",
"FileTransferMethod",
]