From 6339706c687ca32a99b9417db0f4bb9700c2cb31 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Fri, 13 Dec 2024 19:51:09 +0800 Subject: [PATCH] fix: ruff reformatter --- api/core/plugin/entities/parameters.py | 1 + api/core/workflow/nodes/agent/entities.py | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) 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")