opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-27 11:06:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

refine: use enum reference to avoid plain text declarations

Browse Source
This commit is contained in:
Yeuoly 2025-10-18 19:14:24 +08:00
parent 86cf7952fb
commit b1ffd2ef2b
1 changed files with 23 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
api/core/trigger/entities/entities.py
View File

@ -10,8 +10,7 @@ from core.plugin.entities.parameters import (
PluginParameterAutoGenerate, PluginParameterAutoGenerate,
PluginParameterOption, PluginParameterOption,
PluginParameterTemplate, PluginParameterTemplate,
as_normal_type, PluginParameterType,
cast_parameter_value,
) )
from core.tools.entities.common_entities import I18nObject from core.tools.entities.common_entities import I18nObject
@ -19,24 +18,18 @@ from core.tools.entities.common_entities import I18nObject
class EventParameterType(StrEnum): class EventParameterType(StrEnum):
"""The type of the parameter""" """The type of the parameter"""
STRING = "string" STRING = PluginParameterType.STRING
NUMBER = "number" NUMBER = PluginParameterType.NUMBER
BOOLEAN = "boolean" BOOLEAN = PluginParameterType.BOOLEAN
SELECT = "select" SELECT = PluginParameterType.SELECT
FILE = "file" FILE = PluginParameterType.FILE
FILES = "files" FILES = PluginParameterType.FILES
MODEL_SELECTOR = "model-selector" MODEL_SELECTOR = PluginParameterType.MODEL_SELECTOR
APP_SELECTOR = "app-selector" APP_SELECTOR = PluginParameterType.APP_SELECTOR
OBJECT = "object" OBJECT = PluginParameterType.OBJECT
ARRAY = "array" ARRAY = PluginParameterType.ARRAY
DYNAMIC_SELECT = "dynamic-select" DYNAMIC_SELECT = PluginParameterType.DYNAMIC_SELECT
CHECKBOX = "checkbox" CHECKBOX = PluginParameterType.CHECKBOX
def as_normal_type(self):
return as_normal_type(self)
def cast_value(self, value: Any):
return cast_parameter_value(self, value)
class EventParameter(BaseModel): class EventParameter(BaseModel):
@ -57,7 +50,7 @@ class EventParameter(BaseModel):
default=False, default=False,
description="Whether the parameter is multiple select, only valid for select or dynamic-select type", description="Whether the parameter is multiple select, only valid for select or dynamic-select type",
) )
default: Union[int, float, str, list, None] = None default: Union[int, float, str, list[Any], None] = None
min: Union[float, int, None] = None min: Union[float, int, None] = None
max: Union[float, int, None] = None max: Union[float, int, None] = None
precision: Optional[int] = None precision: Optional[int] = None
@ -96,7 +89,7 @@ class EventEntity(BaseModel):
""" """
identity: EventIdentity = Field(..., description="The identity of the event") identity: EventIdentity = Field(..., description="The identity of the event")
parameters: list[EventParameter] = Field(default=[], description="The parameters of the event") parameters: list[EventParameter] = Field(default_factory=list, description="The parameters of the event")
description: I18nObject = Field(..., description="The description of the event") description: I18nObject = Field(..., description="The description of the event")
output_schema: Optional[Mapping[str, Any]] = Field( output_schema: Optional[Mapping[str, Any]] = Field(
default=None, description="The output schema that this event produces" default=None, description="The output schema that this event produces"
@ -150,7 +143,7 @@ class TriggerProviderEntity(BaseModel):
default=None, default=None,
description="The subscription constructor of the trigger provider", description="The subscription constructor of the trigger provider",
) )
events: list[EventEntity] = Field(default=[], description="The events of the trigger provider") events: list[EventEntity] = Field(default_factory=list, description="The events of the trigger provider")
class Subscription(BaseModel): class Subscription(BaseModel):
@ -165,13 +158,15 @@ class Subscription(BaseModel):
) )
endpoint: str = Field(..., description="The webhook endpoint URL allocated by Dify for receiving events") endpoint: str = Field(..., description="The webhook endpoint URL allocated by Dify for receiving events")
parameters: Mapping[str, Any] = Field(default={}, description="The parameters of the subscription constructor") parameters: Mapping[str, Any] = Field(
default_factory=dict, description="The parameters of the subscription constructor"
)
properties: Mapping[str, Any] = Field( properties: Mapping[str, Any] = Field(
..., description="Subscription data containing all properties and provider-specific information" ..., description="Subscription data containing all properties and provider-specific information"
) )
class Unsubscription(BaseModel): class UnsubscribeResult(BaseModel):
""" """
Result of a trigger unsubscription operation. Result of a trigger unsubscription operation.
@ -192,8 +187,8 @@ class Unsubscription(BaseModel):
class RequestLog(BaseModel): class RequestLog(BaseModel):
id: str = Field(..., description="The id of the request log") id: str = Field(..., description="The id of the request log")
endpoint: str = Field(..., description="The endpoint of the request log") endpoint: str = Field(..., description="The endpoint of the request log")
request: dict = Field(..., description="The request of the request log") request: dict[str, Any] = Field(..., description="The request of the request log")
response: dict = Field(..., description="The response of the request log") response: dict[str, Any] = Field(..., description="The response of the request log")
created_at: datetime = Field(..., description="The created at of the request log") created_at: datetime = Field(..., description="The created at of the request log")
@ -282,5 +277,5 @@ __all__: list[str] = [
"TriggerEventData", "TriggerEventData",
"TriggerProviderEntity", "TriggerProviderEntity",
"TriggerProviderIdentity", "TriggerProviderIdentity",
"Unsubscription", "UnsubscribeResult",
] ]