mirror of
https://github.com/langgenius/dify.git
synced 2026-03-27 17:19:55 +08:00
56593f20b0
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: WH-2099 <wh2099@pm.me>
31 lines
1009 B
Python
31 lines
1009 B
Python
from pydantic import BaseModel, Field
|
|
|
|
from dify_graph.entities.base_node_data import BaseNodeData
|
|
from dify_graph.enums import BuiltinNodeTypes, NodeType
|
|
from dify_graph.nodes.llm import ModelConfig, VisionConfig
|
|
from dify_graph.prompt_entities import MemoryConfig
|
|
|
|
|
|
class ClassConfig(BaseModel):
|
|
id: str
|
|
name: str
|
|
|
|
|
|
class QuestionClassifierNodeData(BaseNodeData):
|
|
type: NodeType = BuiltinNodeTypes.QUESTION_CLASSIFIER
|
|
query_variable_selector: list[str]
|
|
model: ModelConfig
|
|
classes: list[ClassConfig]
|
|
instruction: str | None = None
|
|
memory: MemoryConfig | None = None
|
|
vision: VisionConfig = Field(default_factory=VisionConfig)
|
|
|
|
@property
|
|
def structured_output_enabled(self) -> bool:
|
|
# NOTE(QuantumGhost): Temporary workaround for issue #20725
|
|
# (https://github.com/langgenius/dify/issues/20725).
|
|
#
|
|
# The proper fix would be to make `QuestionClassifierNode` inherit
|
|
# from `BaseNode` instead of `LLMNode`.
|
|
return False
|