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

fix: riff

Browse Source
This commit is contained in:
Yeuoly
2024-11-25 16:44:08 +08:00
parent 5ff9cee326
commit 49bd1a7a49
5 changed files with 31 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/tools/entities/tool_entities.py
View File

@ -1,4 +1,5 @@
import base64
import enum
from enum import Enum
from typing import Any, Optional, Union
@ -33,7 +34,7 @@ class ToolLabelEnum(Enum):
OTHER = "other"
class ToolProviderType(str, Enum):
class ToolProviderType(enum.StrEnum):
"""
Enum class for tool provider
"""
@ -205,7 +206,7 @@ class ToolParameterOption(BaseModel):
class ToolParameter(BaseModel):
class ToolParameterType(str, Enum):
class ToolParameterType(enum.StrEnum):
STRING = CommonParameterType.STRING.value
NUMBER = CommonParameterType.NUMBER.value
BOOLEAN = CommonParameterType.BOOLEAN.value