mirror of
https://github.com/langgenius/dify.git
synced 2026-03-26 00:38:03 +08:00
a28f22e59d
- Update BaseNodeData import path to dify_graph.entities.base_node_data - Change NodeType.COMMAND/FILE_UPLOAD to BuiltinNodeTypes constants - Fix system_oauth_encryption -> system_encryption rename in commands - Remove tests for deleted agent runner modules - Fix Avatar: named import + string size API in collaboration files - Add missing skill feature deps: @monaco-editor/react, react-arborist, @tanstack/react-virtual - Fix frontend test mocks: add useUserProfile, useLeaderRestoreListener, next/navigation mock, and nodeOutputVars to expected payload Made-with: Cursor
88 lines
1.8 KiB
Python
88 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Sequence
|
|
from enum import StrEnum
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, field_validator
|
|
|
|
from dify_graph.entities.base_node_data import BaseNodeData
|
|
|
|
|
|
class VariableSelector(BaseModel):
|
|
"""
|
|
Variable Selector.
|
|
"""
|
|
|
|
variable: str
|
|
value_selector: Sequence[str]
|
|
|
|
|
|
class OutputVariableType(StrEnum):
|
|
STRING = "string"
|
|
NUMBER = "number"
|
|
INTEGER = "integer"
|
|
SECRET = "secret"
|
|
BOOLEAN = "boolean"
|
|
OBJECT = "object"
|
|
FILE = "file"
|
|
ARRAY = "array"
|
|
ARRAY_STRING = "array[string]"
|
|
ARRAY_NUMBER = "array[number]"
|
|
ARRAY_OBJECT = "array[object]"
|
|
ARRAY_BOOLEAN = "array[boolean]"
|
|
ARRAY_FILE = "array[file]"
|
|
ANY = "any"
|
|
ARRAY_ANY = "array[any]"
|
|
|
|
|
|
class OutputVariableEntity(BaseModel):
|
|
"""
|
|
Output Variable Entity.
|
|
"""
|
|
|
|
variable: str
|
|
value_type: OutputVariableType = OutputVariableType.ANY
|
|
value_selector: Sequence[str]
|
|
|
|
@field_validator("value_type", mode="before")
|
|
@classmethod
|
|
def normalize_value_type(cls, v: Any) -> Any:
|
|
"""
|
|
Normalize value_type to handle case-insensitive array types.
|
|
Converts 'Array[...]' to 'array[...]' for backward compatibility.
|
|
"""
|
|
if isinstance(v, str) and v.startswith("Array["):
|
|
return v.lower()
|
|
return v
|
|
|
|
|
|
class BaseIterationNodeData(BaseNodeData):
|
|
start_node_id: str | None = None
|
|
|
|
|
|
class BaseIterationState(BaseModel):
|
|
iteration_node_id: str
|
|
index: int
|
|
inputs: dict
|
|
|
|
class MetaData(BaseModel):
|
|
pass
|
|
|
|
metadata: MetaData
|
|
|
|
|
|
class BaseLoopNodeData(BaseNodeData):
|
|
start_node_id: str | None = None
|
|
|
|
|
|
class BaseLoopState(BaseModel):
|
|
loop_node_id: str
|
|
index: int
|
|
inputs: dict
|
|
|
|
class MetaData(BaseModel):
|
|
pass
|
|
|
|
metadata: MetaData
|