use core.variables.types.SegmentType

12 months ago · e9090205d5
parent f772a3acb1
commit e9090205d5
1 changed files with 3 additions and 4 deletions
--- a/api/core/workflow/nodes/variable_aggregator/entities.py
+++ b/api/core/workflow/nodes/variable_aggregator/entities.py
@ -1,8 +1,9 @@
-from typing import Literal, Optional
+from typing import Optional

 from pydantic import BaseModel

 from core.workflow.nodes.base import BaseNodeData
+from core.variables.types import SegmentType


 class AdvancedSettings(BaseModel):
@ -17,9 +18,7 @@ class AdvancedSettings(BaseModel):
        Group.
        """

-        output_type: Literal[
-            "string", "number", "object", "file", "array[string]", "array[number]", "array[object]", "array[file]"
-        ]
+        output_type: SegmentType
        variables: list[list[str]]
        group_name: str