mirror of
https://github.com/langgenius/dify.git
synced 2026-03-27 01:00:13 +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>
43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
from collections.abc import Mapping
|
|
from enum import StrEnum, auto
|
|
from typing import Annotated, Any, Literal, TypeAlias
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from dify_graph.nodes.human_input.entities import FormInput, UserAction
|
|
|
|
|
|
class PauseReasonType(StrEnum):
|
|
HUMAN_INPUT_REQUIRED = auto()
|
|
SCHEDULED_PAUSE = auto()
|
|
|
|
|
|
class HumanInputRequired(BaseModel):
|
|
TYPE: Literal[PauseReasonType.HUMAN_INPUT_REQUIRED] = PauseReasonType.HUMAN_INPUT_REQUIRED
|
|
form_id: str
|
|
form_content: str
|
|
inputs: list[FormInput] = Field(default_factory=list)
|
|
actions: list[UserAction] = Field(default_factory=list)
|
|
node_id: str
|
|
node_title: str
|
|
|
|
# The `resolved_default_values` stores the resolved values of variable defaults. It's a mapping from
|
|
# `output_variable_name` to their resolved values.
|
|
#
|
|
# For example, The form contains a input with output variable name `name` and placeholder type `VARIABLE`, its
|
|
# selector is ["start", "name"]. While the HumanInputNode is executed, the correspond value of variable
|
|
# `start.name` in variable pool is `John`. Thus, the resolved value of the output variable `name` is `John`. The
|
|
# `resolved_default_values` is `{"name": "John"}`.
|
|
#
|
|
# Only form inputs with default value type `VARIABLE` will be resolved and stored in `resolved_default_values`.
|
|
resolved_default_values: Mapping[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
class SchedulingPause(BaseModel):
|
|
TYPE: Literal[PauseReasonType.SCHEDULED_PAUSE] = PauseReasonType.SCHEDULED_PAUSE
|
|
|
|
message: str
|
|
|
|
|
|
PauseReason: TypeAlias = Annotated[HumanInputRequired | SchedulingPause, Field(discriminator="TYPE")]
|