use core.variables.types.SegmentType

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[
+        output_type: SegmentType
-            "string", "number", "object", "file", "array[string]", "array[number]", "array[object]", "array[file]"
-        ]
         variables: list[list[str]]
         group_name: str