mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 10:25:23 -04:00
Yeuoly
31 lines
761 B
Python
31 lines
761 B
Python
from enum import Enum
|
|
|
|
|
|
class InvokeType(Enum):
|
|
Tool = "tool"
|
|
LLM = "llm"
|
|
TextEmbedding = "text_embedding"
|
|
Rerank = "rerank"
|
|
TTS = "tts"
|
|
Speech2Text = "speech2text"
|
|
Moderation = "moderation"
|
|
NodeParameterExtractor = "node_parameter_extractor"
|
|
NodeQuestionClassifier = "node_question_classifier"
|
|
App = "app"
|
|
Storage = "storage"
|
|
UploadFile = "upload_file"
|
|
SYSTEM_SUMMARY = "system_summary"
|
|
|
|
@classmethod
|
|
def value_of(cls, value: str) -> "InvokeType":
|
|
"""
|
|
Get value of given mode.
|
|
|
|
:param value: type
|
|
:return: mode
|
|
"""
|
|
for mode in cls:
|
|
if mode.value == value:
|
|
return mode
|
|
raise ValueError(f"invalid type value {value}")
|