remove bare list, dict, Sequence, None, Any (#25058)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: -LAN- <laipz8200@outlook.com>

api/core/workflow/nodes/agent/agent_node.py

@@ -66,7 +66,7 @@ class AgentNode(BaseNode):
     _node_type = NodeType.AGENT
     _node_data: AgentNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = AgentNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/answer/answer_node.py

@@ -22,7 +22,7 @@ class AnswerNode(BaseNode):
 
     _node_data: AnswerNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = AnswerNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/answer/answer_stream_generate_router.py

@@ -134,7 +134,7 @@ class AnswerStreamGeneratorRouter:
         node_id_config_mapping: dict[str, dict],
         reverse_edge_mapping: dict[str, list["GraphEdge"]],  # type: ignore[name-defined]
         answer_dependencies: dict[str, list[str]],
-    ) -> None:
+    ):
         """
         Recursive fetch answer dependencies
         :param current_node_id: current node id

api/core/workflow/nodes/answer/answer_stream_processor.py

@@ -18,7 +18,7 @@ logger = logging.getLogger(__name__)
 
 
 class AnswerStreamProcessor(StreamProcessor):
-    def __init__(self, graph: Graph, variable_pool: VariablePool) -> None:
+    def __init__(self, graph: Graph, variable_pool: VariablePool):
         super().__init__(graph, variable_pool)
         self.generate_routes = graph.answer_stream_generate_routes
         self.route_position = {}
@@ -66,7 +66,7 @@ class AnswerStreamProcessor(StreamProcessor):
             else:
                 yield event
 
-    def reset(self) -> None:
+    def reset(self):
         self.route_position = {}
         for answer_node_id, _ in self.generate_routes.answer_generate_route.items():
             self.route_position[answer_node_id] = 0

api/core/workflow/nodes/answer/base_stream_processor.py

@@ -11,7 +11,7 @@ logger = logging.getLogger(__name__)
 
 
 class StreamProcessor(ABC):
-    def __init__(self, graph: Graph, variable_pool: VariablePool) -> None:
+    def __init__(self, graph: Graph, variable_pool: VariablePool):
         self.graph = graph
         self.variable_pool = variable_pool
         self.rest_node_ids = graph.node_ids.copy()
@@ -20,7 +20,7 @@ class StreamProcessor(ABC):
     def process(self, generator: Generator[GraphEngineEvent, None, None]) -> Generator[GraphEngineEvent, None, None]:
         raise NotImplementedError
 
-    def _remove_unreachable_nodes(self, event: NodeRunSucceededEvent | NodeRunExceptionEvent) -> None:
+    def _remove_unreachable_nodes(self, event: NodeRunSucceededEvent | NodeRunExceptionEvent):
         finished_node_id = event.route_node_state.node_id
         if finished_node_id not in self.rest_node_ids:
             return
@@ -89,7 +89,7 @@ class StreamProcessor(ABC):
             node_ids.extend(self._fetch_node_ids_in_reachable_branch(edge.target_node_id, branch_identify))
         return node_ids
 
-    def _remove_node_ids_in_unreachable_branch(self, node_id: str, reachable_node_ids: list[str]) -> None:
+    def _remove_node_ids_in_unreachable_branch(self, node_id: str, reachable_node_ids: list[str]):
         """
         remove target node ids until merge
         """

api/core/workflow/nodes/base/entities.py

@@ -28,7 +28,7 @@ class DefaultValue(BaseModel):
     key: str
 
     @staticmethod
-    def _parse_json(value: str) -> Any:
+    def _parse_json(value: str):
         """Unified JSON parsing handler"""
         try:
             return json.loads(value)

api/core/workflow/nodes/base/node.py

@@ -28,7 +28,7 @@ class BaseNode:
         graph_runtime_state: "GraphRuntimeState",
         previous_node_id: Optional[str] = None,
         thread_pool_id: Optional[str] = None,
-    ) -> None:
+    ):
         self.id = id
         self.tenant_id = graph_init_params.tenant_id
         self.app_id = graph_init_params.app_id
@@ -51,7 +51,7 @@ class BaseNode:
         self.node_id = node_id
 
     @abstractmethod
-    def init_node_data(self, data: Mapping[str, Any]) -> None: ...
+    def init_node_data(self, data: Mapping[str, Any]): ...
 
     @abstractmethod
     def _run(self) -> NodeRunResult | Generator[Union[NodeEvent, "InNodeEvent"], None, None]:
@@ -141,7 +141,7 @@ class BaseNode:
         return {}
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict] = None):
         return {}
 
     @property

api/core/workflow/nodes/code/code_node.py

@@ -28,7 +28,7 @@ class CodeNode(BaseNode):
 
     _node_data: CodeNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = CodeNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -50,7 +50,7 @@ class CodeNode(BaseNode):
         return self._node_data
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict] = None):
         """
         Get default config of node.
         :param filters: filter by node config parameters.

api/core/workflow/nodes/document_extractor/node.py

@@ -47,7 +47,7 @@ class DocumentExtractorNode(BaseNode):
 
     _node_data: DocumentExtractorNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = DocumentExtractorNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/end/end_node.py

@@ -14,7 +14,7 @@ class EndNode(BaseNode):
 
     _node_data: EndNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = EndNodeData(**data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/end/end_stream_generate_router.py

@@ -121,7 +121,7 @@ class EndStreamGeneratorRouter:
         node_id_config_mapping: dict[str, dict],
         reverse_edge_mapping: dict[str, list["GraphEdge"]],  # type: ignore[name-defined]
         end_dependencies: dict[str, list[str]],
-    ) -> None:
+    ):
         """
         Recursive fetch end dependencies
         :param current_node_id: current node id

api/core/workflow/nodes/end/end_stream_processor.py

@@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
 
 
 class EndStreamProcessor(StreamProcessor):
-    def __init__(self, graph: Graph, variable_pool: VariablePool) -> None:
+    def __init__(self, graph: Graph, variable_pool: VariablePool):
         super().__init__(graph, variable_pool)
         self.end_stream_param = graph.end_stream_param
         self.route_position = {}
@@ -76,7 +76,7 @@ class EndStreamProcessor(StreamProcessor):
             else:
                 yield event
 
-    def reset(self) -> None:
+    def reset(self):
         self.route_position = {}
         for end_node_id, _ in self.end_stream_param.end_stream_variable_selector_mapping.items():
             self.route_position[end_node_id] = 0

api/core/workflow/nodes/http_request/node.py

@@ -38,7 +38,7 @@ class HttpRequestNode(BaseNode):
 
     _node_data: HttpRequestNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = HttpRequestNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -60,7 +60,7 @@ class HttpRequestNode(BaseNode):
         return self._node_data
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict[str, Any]] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict[str, Any]] = None):
         return {
             "type": "http-request",
             "config": {

api/core/workflow/nodes/if_else/if_else_node.py

@@ -19,7 +19,7 @@ class IfElseNode(BaseNode):
 
     _node_data: IfElseNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = IfElseNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/iteration/iteration_node.py

@@ -67,7 +67,7 @@ class IterationNode(BaseNode):
 
     _node_data: IterationNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = IterationNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -89,7 +89,7 @@ class IterationNode(BaseNode):
         return self._node_data
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict] = None):
         return {
             "type": "iteration",
             "config": {

api/core/workflow/nodes/iteration/iteration_start_node.py

@@ -18,7 +18,7 @@ class IterationStartNode(BaseNode):
 
     _node_data: IterationStartNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = IterationStartNodeData(**data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py

@@ -105,7 +105,7 @@ class KnowledgeRetrievalNode(BaseNode):
         thread_pool_id: Optional[str] = None,
         *,
         llm_file_saver: LLMFileSaver | None = None,
-    ) -> None:
+    ):
         super().__init__(
             id=id,
             config=config,
@@ -125,7 +125,7 @@ class KnowledgeRetrievalNode(BaseNode):
             )
         self._llm_file_saver = llm_file_saver
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = KnowledgeRetrievalNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/list_operator/node.py

@@ -41,7 +41,7 @@ class ListOperatorNode(BaseNode):
 
     _node_data: ListOperatorNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = ListOperatorNodeData(**data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/llm/exc.py

@@ -41,5 +41,5 @@ class FileTypeNotSupportError(LLMNodeError):
 
 
 class UnsupportedPromptContentTypeError(LLMNodeError):
-    def __init__(self, *, type_name: str) -> None:
+    def __init__(self, *, type_name: str):
         super().__init__(f"Prompt content type {type_name} is not supported.")

api/core/workflow/nodes/llm/llm_utils.py

@@ -107,7 +107,7 @@ def fetch_memory(
     return memory
 
 
-def deduct_llm_quota(tenant_id: str, model_instance: ModelInstance, usage: LLMUsage) -> None:
+def deduct_llm_quota(tenant_id: str, model_instance: ModelInstance, usage: LLMUsage):
     provider_model_bundle = model_instance.provider_model_bundle
     provider_configuration = provider_model_bundle.configuration
 

api/core/workflow/nodes/llm/node.py

@@ -120,7 +120,7 @@ class LLMNode(BaseNode):
         thread_pool_id: Optional[str] = None,
         *,
         llm_file_saver: LLMFileSaver | None = None,
-    ) -> None:
+    ):
         super().__init__(
             id=id,
             config=config,
@@ -140,7 +140,7 @@ class LLMNode(BaseNode):
             )
         self._llm_file_saver = llm_file_saver
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = LLMNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -951,7 +951,7 @@ class LLMNode(BaseNode):
         return variable_mapping
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict] = None):
         return {
             "type": "llm",
             "config": {

api/core/workflow/nodes/loop/loop_end_node.py

@@ -18,7 +18,7 @@ class LoopEndNode(BaseNode):
 
     _node_data: LoopEndNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = LoopEndNodeData(**data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/loop/loop_node.py

@@ -54,7 +54,7 @@ class LoopNode(BaseNode):
 
     _node_data: LoopNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = LoopNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/loop/loop_start_node.py

@@ -18,7 +18,7 @@ class LoopStartNode(BaseNode):
 
     _node_data: LoopStartNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = LoopStartNodeData(**data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/parameter_extractor/entities.py

@@ -98,7 +98,7 @@ class ParameterExtractorNodeData(BaseNodeData):
     def set_reasoning_mode(cls, v) -> str:
         return v or "function_call"
 
-    def get_parameter_json_schema(self) -> dict:
+    def get_parameter_json_schema(self):
         """
         Get parameter json schema.
 

api/core/workflow/nodes/parameter_extractor/exc.py

@@ -63,7 +63,7 @@ class InvalidValueTypeError(ParameterExtractorNodeError):
         expected_type: SegmentType,
         actual_type: SegmentType | None,
         value: Any,
-    ) -> None:
+    ):
         message = (
             f"Invalid value for parameter {parameter_name}, expected segment type: {expected_type}, "
             f"actual_type: {actual_type}, python_type: {type(value)}, value: {value}"

api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py

@@ -94,7 +94,7 @@ class ParameterExtractorNode(BaseNode):
 
     _node_data: ParameterExtractorNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = ParameterExtractorNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -119,7 +119,7 @@ class ParameterExtractorNode(BaseNode):
     _model_config: Optional[ModelConfigWithCredentialsEntity] = None
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict] = None):
         return {
             "model": {
                 "prompt_templates": {
@@ -545,7 +545,7 @@ class ParameterExtractorNode(BaseNode):
 
         return prompt_messages
 
-    def _validate_result(self, data: ParameterExtractorNodeData, result: dict) -> dict:
+    def _validate_result(self, data: ParameterExtractorNodeData, result: dict):
         if len(data.parameters) != len(result):
             raise InvalidNumberOfParametersError("Invalid number of parameters")
 
@@ -597,7 +597,7 @@ class ParameterExtractorNode(BaseNode):
             except ValueError:
                 return None
 
-    def _transform_result(self, data: ParameterExtractorNodeData, result: dict) -> dict:
+    def _transform_result(self, data: ParameterExtractorNodeData, result: dict):
         """
         Transform result into standard format.
         """
@@ -690,7 +690,7 @@ class ParameterExtractorNode(BaseNode):
         logger.info("extra error: %s", result)
         return None
 
-    def _generate_default_result(self, data: ParameterExtractorNodeData) -> dict:
+    def _generate_default_result(self, data: ParameterExtractorNodeData):
         """
         Generate default result.
         """

api/core/workflow/nodes/question_classifier/question_classifier_node.py

@@ -63,7 +63,7 @@ class QuestionClassifierNode(BaseNode):
         thread_pool_id: Optional[str] = None,
         *,
         llm_file_saver: LLMFileSaver | None = None,
-    ) -> None:
+    ):
         super().__init__(
             id=id,
             config=config,
@@ -83,7 +83,7 @@ class QuestionClassifierNode(BaseNode):
             )
         self._llm_file_saver = llm_file_saver
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = QuestionClassifierNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -275,7 +275,7 @@ class QuestionClassifierNode(BaseNode):
         return variable_mapping
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict] = None):
         """
         Get default config of node.
         :param filters: filter by node config parameters.

api/core/workflow/nodes/start/start_node.py

@@ -15,7 +15,7 @@ class StartNode(BaseNode):
 
     _node_data: StartNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = StartNodeData(**data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/template_transform/template_transform_node.py

@@ -18,7 +18,7 @@ class TemplateTransformNode(BaseNode):
 
     _node_data: TemplateTransformNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = TemplateTransformNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -40,7 +40,7 @@ class TemplateTransformNode(BaseNode):
         return self._node_data
 
     @classmethod
-    def get_default_config(cls, filters: Optional[dict] = None) -> dict:
+    def get_default_config(cls, filters: Optional[dict] = None):
         """
         Get default config of node.
         :param filters: filter by node config parameters.

api/core/workflow/nodes/tool/tool_node.py

@@ -45,7 +45,7 @@ class ToolNode(BaseNode):
 
     _node_data: ToolNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = ToolNodeData.model_validate(data)
 
     @classmethod

api/core/workflow/nodes/variable_aggregator/variable_aggregator_node.py

@@ -15,7 +15,7 @@ class VariableAggregatorNode(BaseNode):
 
     _node_data: VariableAssignerNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = VariableAssignerNodeData(**data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:

api/core/workflow/nodes/variable_assigner/common/impl.py

@@ -11,7 +11,7 @@ from .exc import VariableOperatorNodeError
 class ConversationVariableUpdaterImpl:
     _engine: Engine | None
 
-    def __init__(self, engine: Engine | None = None) -> None:
+    def __init__(self, engine: Engine | None = None):
         self._engine = engine
 
     def _get_engine(self) -> Engine:

api/core/workflow/nodes/variable_assigner/v1/node.py

@@ -30,7 +30,7 @@ class VariableAssignerNode(BaseNode):
 
     _node_data: VariableAssignerData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = VariableAssignerData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]:
@@ -61,7 +61,7 @@ class VariableAssignerNode(BaseNode):
         previous_node_id: Optional[str] = None,
         thread_pool_id: Optional[str] = None,
         conv_var_updater_factory: _CONV_VAR_UPDATER_FACTORY = conversation_variable_updater_factory,
-    ) -> None:
+    ):
         super().__init__(
             id=id,
             config=config,

api/core/workflow/nodes/variable_assigner/v2/exc.py

@@ -32,5 +32,5 @@ class ConversationIDNotFoundError(VariableOperatorNodeError):
 
 
 class InvalidDataError(VariableOperatorNodeError):
-    def __init__(self, message: str) -> None:
+    def __init__(self, message: str):
         super().__init__(message)

api/core/workflow/nodes/variable_assigner/v2/node.py

@@ -58,7 +58,7 @@ class VariableAssignerNode(BaseNode):
 
     _node_data: VariableAssignerNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]) -> None:
+    def init_node_data(self, data: Mapping[str, Any]):
         self._node_data = VariableAssignerNodeData.model_validate(data)
 
     def _get_error_strategy(self) -> Optional[ErrorStrategy]: