mirror of https://github.com/langgenius/dify.git
33 lines
806 B
Python
33 lines
806 B
Python
from enum import Enum
|
|
|
|
|
|
class NodeType(Enum):
|
|
"""
|
|
Node Types.
|
|
"""
|
|
START = 'start'
|
|
END = 'end'
|
|
DIRECT_ANSWER = 'direct-answer'
|
|
LLM = 'llm'
|
|
KNOWLEDGE_RETRIEVAL = 'knowledge-retrieval'
|
|
IF_ELSE = 'if-else'
|
|
CODE = 'code'
|
|
TEMPLATE_TRANSFORM = 'template-transform'
|
|
QUESTION_CLASSIFIER = 'question-classifier'
|
|
HTTP_REQUEST = 'http-request'
|
|
TOOL = 'tool'
|
|
VARIABLE_ASSIGNER = 'variable-assigner'
|
|
|
|
@classmethod
|
|
def value_of(cls, value: str) -> 'NodeType':
|
|
"""
|
|
Get value of given node type.
|
|
|
|
:param value: node type value
|
|
:return: node type
|
|
"""
|
|
for node_type in cls:
|
|
if node_type.value == value:
|
|
return node_type
|
|
raise ValueError(f'invalid node type value {value}')
|