Merge pull request #933 from SuanmoSuanyangTechnology/feature/agent-tool_xjn
feat(workflow)
This commit is contained in:
@@ -201,12 +201,15 @@ class VariablePool:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _extract_field(struct: "VariableStruct", field: str | None) -> Any:
|
def _extract_field(struct: "VariableStruct", field: str | None) -> Any:
|
||||||
"""If field is given, drill into a dict/object variable's value."""
|
"""If field is given, drill into a dict/object/array[file] variable's value."""
|
||||||
if field is None:
|
if field is None:
|
||||||
return struct.instance.get_value()
|
return struct.instance.get_value()
|
||||||
value = struct.instance.get_value()
|
value = struct.instance.get_value()
|
||||||
|
# array[file]: extract the field from every element, return a list
|
||||||
|
if isinstance(value, list):
|
||||||
|
return [item.get(field) if isinstance(item, dict) else getattr(item, field, None) for item in value]
|
||||||
if not isinstance(value, dict):
|
if not isinstance(value, dict):
|
||||||
raise KeyError(f"Variable is not an object, cannot access field '{field}'")
|
raise KeyError(f"Variable is not an object or array, cannot access field '{field}'")
|
||||||
return value.get(field)
|
return value.get(field)
|
||||||
|
|
||||||
def get_instance(
|
def get_instance(
|
||||||
|
|||||||
@@ -6,6 +6,20 @@ from app.core.workflow.nodes.base_config import BaseNodeConfig
|
|||||||
from app.core.workflow.nodes.enums import ComparisonOperator, LogicOperator, ValueInputType
|
from app.core.workflow.nodes.enums import ComparisonOperator, LogicOperator, ValueInputType
|
||||||
|
|
||||||
|
|
||||||
|
class SubVariableConditionItem(BaseModel):
|
||||||
|
"""A single condition on a file object's field, used inside sub_variable_condition."""
|
||||||
|
key: str = Field(..., description="Field name of the file object, e.g. type, size, name")
|
||||||
|
operator: ComparisonOperator = Field(..., description="Comparison operator")
|
||||||
|
value: Any = Field(default=None, description="Value to compare with")
|
||||||
|
var_type: str = Field(default="string", description="Field value type: string or number")
|
||||||
|
|
||||||
|
|
||||||
|
class SubVariableCondition(BaseModel):
|
||||||
|
"""Sub-conditions applied to each file element in an array[file] variable."""
|
||||||
|
logical_operator: LogicOperator = Field(default=LogicOperator.AND)
|
||||||
|
conditions: list[SubVariableConditionItem] = Field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
class ConditionDetail(BaseModel):
|
class ConditionDetail(BaseModel):
|
||||||
operator: ComparisonOperator = Field(
|
operator: ComparisonOperator = Field(
|
||||||
...,
|
...,
|
||||||
@@ -14,12 +28,12 @@ class ConditionDetail(BaseModel):
|
|||||||
|
|
||||||
left: str = Field(
|
left: str = Field(
|
||||||
...,
|
...,
|
||||||
description="Value to compare against"
|
description="Variable selector, e.g. {{sys.files}}"
|
||||||
)
|
)
|
||||||
|
|
||||||
right: Any = Field(
|
right: Any = Field(
|
||||||
default=None,
|
default=None,
|
||||||
description="Value to compare with"
|
description="Value to compare with (unused when sub_variable_condition is set)"
|
||||||
)
|
)
|
||||||
|
|
||||||
input_type: ValueInputType = Field(
|
input_type: ValueInputType = Field(
|
||||||
@@ -27,6 +41,11 @@ class ConditionDetail(BaseModel):
|
|||||||
description="Value input type for comparison"
|
description="Value input type for comparison"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
sub_variable_condition: SubVariableCondition | None = Field(
|
||||||
|
default=None,
|
||||||
|
description="Sub-conditions for array[file] fields. When set, operator must be contains/not_contains."
|
||||||
|
)
|
||||||
|
|
||||||
@field_validator("input_type", mode="before")
|
@field_validator("input_type", mode="before")
|
||||||
@classmethod
|
@classmethod
|
||||||
def lower_input_type(cls, v):
|
def lower_input_type(cls, v):
|
||||||
@@ -39,16 +58,19 @@ class ConditionDetail(BaseModel):
|
|||||||
|
|
||||||
|
|
||||||
class ConditionBranchConfig(BaseModel):
|
class ConditionBranchConfig(BaseModel):
|
||||||
"""Configuration for a conditional branch"""
|
"""Configuration for a conditional branch.
|
||||||
|
|
||||||
|
logical_operator controls how all expressions are combined (AND/OR).
|
||||||
|
"""
|
||||||
|
|
||||||
logical_operator: LogicOperator = Field(
|
logical_operator: LogicOperator = Field(
|
||||||
default=LogicOperator.AND,
|
default=LogicOperator.AND,
|
||||||
description="Logical operator used to combine multiple condition expressions"
|
description="Logical operator used to combine all conditions"
|
||||||
)
|
)
|
||||||
|
|
||||||
expressions: list[ConditionDetail] = Field(
|
expressions: list[ConditionDetail] = Field(
|
||||||
...,
|
default_factory=list,
|
||||||
description="List of condition expressions within this branch"
|
description="List of conditions within this branch"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ from app.core.workflow.engine.variable_pool import VariablePool
|
|||||||
from app.core.workflow.nodes.base_node import BaseNode
|
from app.core.workflow.nodes.base_node import BaseNode
|
||||||
from app.core.workflow.nodes.enums import ComparisonOperator, LogicOperator, ValueInputType
|
from app.core.workflow.nodes.enums import ComparisonOperator, LogicOperator, ValueInputType
|
||||||
from app.core.workflow.nodes.if_else import IfElseNodeConfig
|
from app.core.workflow.nodes.if_else import IfElseNodeConfig
|
||||||
from app.core.workflow.nodes.operators import ConditionExpressionResolver, CompareOperatorInstance
|
from app.core.workflow.nodes.operators import ConditionExpressionResolver, CompareOperatorInstance, ArrayFileContainsOperator
|
||||||
from app.core.workflow.variable.base_variable import VariableType
|
from app.core.workflow.variable.base_variable import VariableType
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
@@ -90,11 +90,9 @@ class IfElseNode(BaseNode):
|
|||||||
list[str]: A list of Python boolean expression strings,
|
list[str]: A list of Python boolean expression strings,
|
||||||
ordered by branch priority.
|
ordered by branch priority.
|
||||||
"""
|
"""
|
||||||
branch_index = 0
|
|
||||||
conditions = []
|
conditions = []
|
||||||
|
|
||||||
for case_branch in self.typed_config.cases:
|
for case_branch in self.typed_config.cases:
|
||||||
branch_index += 1
|
|
||||||
branch_result = []
|
branch_result = []
|
||||||
for expression in case_branch.expressions:
|
for expression in case_branch.expressions:
|
||||||
pattern = r"\{\{\s*(.*?)\s*\}\}"
|
pattern = r"\{\{\s*(.*?)\s*\}\}"
|
||||||
@@ -103,13 +101,18 @@ class IfElseNode(BaseNode):
|
|||||||
left_value = self.get_variable(left_string, variable_pool)
|
left_value = self.get_variable(left_string, variable_pool)
|
||||||
except KeyError:
|
except KeyError:
|
||||||
left_value = None
|
left_value = None
|
||||||
evaluator = ConditionExpressionResolver.resolve_by_value(left_value)(
|
|
||||||
variable_pool,
|
if expression.sub_variable_condition is not None and isinstance(left_value, list):
|
||||||
expression.left,
|
evaluator = ArrayFileContainsOperator(left_value, expression.sub_variable_condition)
|
||||||
expression.right,
|
else:
|
||||||
expression.input_type
|
evaluator = ConditionExpressionResolver.resolve_by_value(left_value)(
|
||||||
)
|
variable_pool,
|
||||||
|
expression.left,
|
||||||
|
expression.right,
|
||||||
|
expression.input_type
|
||||||
|
)
|
||||||
branch_result.append(self._evaluate(expression.operator, evaluator))
|
branch_result.append(self._evaluate(expression.operator, evaluator))
|
||||||
|
|
||||||
if case_branch.logical_operator == LogicOperator.AND:
|
if case_branch.logical_operator == LogicOperator.AND:
|
||||||
conditions.append(all(branch_result))
|
conditions.append(all(branch_result))
|
||||||
else:
|
else:
|
||||||
|
|||||||
@@ -395,11 +395,71 @@ class NoneObjectComparisonOperator:
|
|||||||
return lambda *args, **kwargs: False
|
return lambda *args, **kwargs: False
|
||||||
|
|
||||||
|
|
||||||
|
class ArrayFileContainsOperator:
|
||||||
|
"""Handles contains/not_contains on array[file] with sub_variable_condition.
|
||||||
|
|
||||||
|
Evaluates whether any (contains) or no (not_contains) file element
|
||||||
|
in the array satisfies all sub-conditions.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, left_value: list[dict], sub_variable_condition: Any):
|
||||||
|
self.left_value = left_value
|
||||||
|
self.sub_variable_condition = sub_variable_condition
|
||||||
|
|
||||||
|
def _match_item(self, file_item: dict) -> bool:
|
||||||
|
"""Check if a single file dict satisfies all sub-conditions."""
|
||||||
|
results = []
|
||||||
|
for cond in self.sub_variable_condition.conditions:
|
||||||
|
field_val = file_item.get(cond.key)
|
||||||
|
result = self._eval_sub(field_val, cond)
|
||||||
|
results.append(result)
|
||||||
|
if self.sub_variable_condition.logical_operator.value == "and":
|
||||||
|
return all(results)
|
||||||
|
return any(results)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _eval_sub(field_val: Any, cond: Any) -> bool:
|
||||||
|
op = cond.operator.value
|
||||||
|
expected = cond.value
|
||||||
|
if field_val is None:
|
||||||
|
return op in ("empty", "not_empty") and op == "empty"
|
||||||
|
match op:
|
||||||
|
case "eq": return str(field_val) == str(expected)
|
||||||
|
case "ne": return str(field_val) != str(expected)
|
||||||
|
case "contains": return isinstance(field_val, str) and str(expected) in field_val
|
||||||
|
case "not_contains": return isinstance(field_val, str) and str(expected) not in field_val
|
||||||
|
case "in": return field_val in (expected if isinstance(expected, list) else [expected])
|
||||||
|
case "not_in": return field_val not in (expected if isinstance(expected, list) else [expected])
|
||||||
|
case "gt": return isinstance(field_val, (int, float)) and field_val > float(expected)
|
||||||
|
case "ge": return isinstance(field_val, (int, float)) and field_val >= float(expected)
|
||||||
|
case "lt": return isinstance(field_val, (int, float)) and field_val < float(expected)
|
||||||
|
case "le": return isinstance(field_val, (int, float)) and field_val <= float(expected)
|
||||||
|
case "empty": return field_val in (None, "", 0)
|
||||||
|
case "not_empty": return field_val not in (None, "", 0)
|
||||||
|
case _: return False
|
||||||
|
|
||||||
|
def contains(self) -> bool:
|
||||||
|
return any(self._match_item(f) for f in self.left_value if isinstance(f, dict))
|
||||||
|
|
||||||
|
def not_contains(self) -> bool:
|
||||||
|
return not self.contains()
|
||||||
|
|
||||||
|
def empty(self) -> bool:
|
||||||
|
return not self.left_value
|
||||||
|
|
||||||
|
def not_empty(self) -> bool:
|
||||||
|
return bool(self.left_value)
|
||||||
|
|
||||||
|
def __getattr__(self, name):
|
||||||
|
return lambda *args, **kwargs: False
|
||||||
|
|
||||||
|
|
||||||
CompareOperatorInstance = Union[
|
CompareOperatorInstance = Union[
|
||||||
StringComparisonOperator,
|
StringComparisonOperator,
|
||||||
NumberComparisonOperator,
|
NumberComparisonOperator,
|
||||||
BooleanComparisonOperator,
|
BooleanComparisonOperator,
|
||||||
ArrayComparisonOperator,
|
ArrayComparisonOperator,
|
||||||
|
ArrayFileContainsOperator,
|
||||||
ObjectComparisonOperator
|
ObjectComparisonOperator
|
||||||
]
|
]
|
||||||
CompareOperatorType = Type[CompareOperatorInstance]
|
CompareOperatorType = Type[CompareOperatorInstance]
|
||||||
|
|||||||
Reference in New Issue
Block a user