diff --git a/api/core/plugin/entities/parameters.py b/api/core/plugin/entities/parameters.py index 868e2fc1e4..4f652a7ffe 100644 --- a/api/core/plugin/entities/parameters.py +++ b/api/core/plugin/entities/parameters.py @@ -1,5 +1,6 @@ import enum from typing import Any, Optional, Union + from pydantic import BaseModel, Field, field_validator from core.entities.parameter_entities import CommonParameterType diff --git a/api/core/workflow/nodes/agent/entities.py b/api/core/workflow/nodes/agent/entities.py index 66fb8773f9..c2cfc3136a 100644 --- a/api/core/workflow/nodes/agent/entities.py +++ b/api/core/workflow/nodes/agent/entities.py @@ -62,9 +62,9 @@ class AgentNodeData(BaseNodeData, AgentEntity): elif typ == "constant": if isinstance(value, list): # convert dict to ToolSelector - if all(isinstance(val, dict) for val in value): - return value - elif all(isinstance(val, ToolSelector) for val in value): + if all(isinstance(val, dict) for val in value) or all( + isinstance(val, ToolSelector) for val in value + ): return value else: raise ValueError("value must be a list of ToolSelector")