chore(api/core): apply ruff reformatting (#7624)

api/core/workflow/entities/base_node_data_entities.py

@@ -8,9 +8,11 @@ class BaseNodeData(ABC, BaseModel):
     title: str
     desc: Optional[str] = None
 
+
 class BaseIterationNodeData(BaseNodeData):
     start_node_id: Optional[str] = None
 
+
 class BaseIterationState(BaseModel):
     iteration_node_id: str
     index: int
@@ -19,4 +21,4 @@ class BaseIterationState(BaseModel):
     class MetaData(BaseModel):
         pass
 
-    metadata: MetaData
+    metadata: MetaData

api/core/workflow/entities/node_entities.py

@@ -12,28 +12,28 @@ class NodeType(Enum):
     Node Types.
     """
 
-    START = 'start'
-    END = 'end'
-    ANSWER = 'answer'
-    LLM = 'llm'
-    KNOWLEDGE_RETRIEVAL = 'knowledge-retrieval'
-    IF_ELSE = 'if-else'
-    CODE = 'code'
-    TEMPLATE_TRANSFORM = 'template-transform'
-    QUESTION_CLASSIFIER = 'question-classifier'
-    HTTP_REQUEST = 'http-request'
-    TOOL = 'tool'
-    VARIABLE_AGGREGATOR = 'variable-aggregator'
+    START = "start"
+    END = "end"
+    ANSWER = "answer"
+    LLM = "llm"
+    KNOWLEDGE_RETRIEVAL = "knowledge-retrieval"
+    IF_ELSE = "if-else"
+    CODE = "code"
+    TEMPLATE_TRANSFORM = "template-transform"
+    QUESTION_CLASSIFIER = "question-classifier"
+    HTTP_REQUEST = "http-request"
+    TOOL = "tool"
+    VARIABLE_AGGREGATOR = "variable-aggregator"
     # TODO: merge this into VARIABLE_AGGREGATOR
-    VARIABLE_ASSIGNER = 'variable-assigner'
-    LOOP = 'loop'
-    ITERATION = 'iteration'
-    ITERATION_START = 'iteration-start'  # fake start node for iteration
-    PARAMETER_EXTRACTOR = 'parameter-extractor'
-    CONVERSATION_VARIABLE_ASSIGNER = 'assigner'
+    VARIABLE_ASSIGNER = "variable-assigner"
+    LOOP = "loop"
+    ITERATION = "iteration"
+    ITERATION_START = "iteration-start"  # fake start node for iteration
+    PARAMETER_EXTRACTOR = "parameter-extractor"
+    CONVERSATION_VARIABLE_ASSIGNER = "assigner"
 
     @classmethod
-    def value_of(cls, value: str) -> 'NodeType':
+    def value_of(cls, value: str) -> "NodeType":
         """
         Get value of given node type.
 
@@ -43,7 +43,7 @@ class NodeType(Enum):
         for node_type in cls:
             if node_type.value == value:
                 return node_type
-        raise ValueError(f'invalid node type value {value}')
+        raise ValueError(f"invalid node type value {value}")
 
 
 class NodeRunMetadataKey(Enum):
@@ -51,16 +51,16 @@ class NodeRunMetadataKey(Enum):
     Node Run Metadata Key.
     """
 
-    TOTAL_TOKENS = 'total_tokens'
-    TOTAL_PRICE = 'total_price'
-    CURRENCY = 'currency'
-    TOOL_INFO = 'tool_info'
-    ITERATION_ID = 'iteration_id'
-    ITERATION_INDEX = 'iteration_index'
-    PARALLEL_ID = 'parallel_id'
-    PARALLEL_START_NODE_ID = 'parallel_start_node_id'
-    PARENT_PARALLEL_ID = 'parent_parallel_id'
-    PARENT_PARALLEL_START_NODE_ID = 'parent_parallel_start_node_id'
+    TOTAL_TOKENS = "total_tokens"
+    TOTAL_PRICE = "total_price"
+    CURRENCY = "currency"
+    TOOL_INFO = "tool_info"
+    ITERATION_ID = "iteration_id"
+    ITERATION_INDEX = "iteration_index"
+    PARALLEL_ID = "parallel_id"
+    PARALLEL_START_NODE_ID = "parallel_start_node_id"
+    PARENT_PARALLEL_ID = "parent_parallel_id"
+    PARENT_PARALLEL_START_NODE_ID = "parent_parallel_start_node_id"
 
 
 class NodeRunResult(BaseModel):
@@ -85,6 +85,7 @@ class UserFrom(Enum):
     """
     User from
     """
+
     ACCOUNT = "account"
     END_USER = "end-user"
 

api/core/workflow/entities/variable_entities.py

@@ -5,5 +5,6 @@ class VariableSelector(BaseModel):
     """
     Variable Selector.
     """
+
     variable: str
     value_selector: list[str]

api/core/workflow/entities/variable_pool.py

@@ -23,23 +23,19 @@ class VariablePool(BaseModel):
     # Other elements of the selector are the keys in the second-level dictionary. To get the key, we hash the
     # elements of the selector except the first one.
     variable_dictionary: dict[str, dict[int, Segment]] = Field(
-        description='Variables mapping',
-        default=defaultdict(dict)
+        description="Variables mapping", default=defaultdict(dict)
     )
 
     # TODO: This user inputs is not used for pool.
     user_inputs: Mapping[str, Any] = Field(
-        description='User inputs',
+        description="User inputs",
     )
 
     system_variables: Mapping[SystemVariableKey, Any] = Field(
-        description='System variables',
+        description="System variables",
     )
 
-    environment_variables: Sequence[Variable] = Field(
-        description="Environment variables.",
-        default_factory=list
-    )
+    environment_variables: Sequence[Variable] = Field(description="Environment variables.", default_factory=list)
 
     conversation_variables: Sequence[Variable] | None = None
 

api/core/workflow/entities/workflow_entities.py

@@ -46,13 +46,16 @@ class WorkflowRunState:
 
     current_iteration_state: Optional[BaseIterationState]
 
-    def __init__(self, workflow: Workflow,
-                 start_at: float,
-                 variable_pool: VariablePool,
-                 user_id: str,
-                 user_from: UserFrom,
-                 invoke_from: InvokeFrom,
-                 workflow_call_depth: int):
+    def __init__(
+        self,
+        workflow: Workflow,
+        start_at: float,
+        variable_pool: VariablePool,
+        user_id: str,
+        user_from: UserFrom,
+        invoke_from: InvokeFrom,
+        workflow_call_depth: int,
+    ):
         self.workflow_id = workflow.id
         self.tenant_id = workflow.tenant_id
         self.app_id = workflow.app_id