youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 01:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: replace Enum with StrEnum

Browse Source
This commit is contained in:
Yeuoly
2024-12-16 13:40:02 +08:00
parent 93c3699128
commit a61da6cf95
4 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/plugin/entities/parameters.py
View File

@ -59,7 +59,7 @@ class PluginParameter(BaseModel):
return v
def as_normal_type(typ: enum.Enum):
def as_normal_type(typ: enum.StrEnum):
if typ.value in {
PluginParameterType.SECRET_INPUT,
PluginParameterType.SELECT,
@ -68,7 +68,7 @@ def as_normal_type(typ: enum.Enum):
return typ.value
def cast_parameter_value(typ: enum.Enum, value: Any, /):
def cast_parameter_value(typ: enum.StrEnum, value: Any, /):
try:
match typ.value:
case PluginParameterType.STRING | PluginParameterType.SECRET_INPUT | PluginParameterType.SELECT:
@ -127,7 +127,7 @@ def cast_parameter_value(typ: enum.Enum, value: Any, /):
raise ValueError(f"The tool parameter value {value} is not in correct type of {as_normal_type(typ)}.")
def init_frontend_parameter(rule: PluginParameter, type: enum.Enum, value: Any):
def init_frontend_parameter(rule: PluginParameter, type: enum.StrEnum, value: Any):
"""
init frontend parameter by rule
"""

7
api/core/plugin/entities/plugin_daemon.py
View File

@ -1,6 +1,5 @@
import enum
from datetime import datetime
from enum import Enum
from enum import StrEnum
from typing import Generic, Optional, TypeVar
from pydantic import BaseModel, ConfigDict, Field
@ -31,7 +30,7 @@ class InstallPluginMessage(BaseModel):
Message for installing a plugin.
"""
class Event(Enum):
class Event(StrEnum):
Info = "info"
Done = "done"
Error = "error"
@ -127,7 +126,7 @@ class PluginDaemonInnerError(Exception):
self.message = message
class PluginInstallTaskStatus(enum.StrEnum):
class PluginInstallTaskStatus(StrEnum):
Pending = "pending"
Running = "running"
Success = "success"