mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-24 13:06:56 -04:00
Yeuoly
29 lines
692 B
Python
29 lines
692 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"
|
|
|
|
@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}')
|