chore: save changes

Signed-off-by: -LAN- <laipz8200@outlook.com>

api/dify_graph/runtime/variable_pool.py

@@ -8,18 +8,11 @@ from typing import Annotated, Any, Union, cast
 
 from pydantic import BaseModel, Field
 
-from dify_graph.constants import (
-    CONVERSATION_VARIABLE_NODE_ID,
-    ENVIRONMENT_VARIABLE_NODE_ID,
-    RAG_PIPELINE_VARIABLE_NODE_ID,
-    SYSTEM_VARIABLE_NODE_ID,
-)
 from dify_graph.file import File, FileAttribute, file_manager
-from dify_graph.system_variable import SystemVariable
 from dify_graph.variables import Segment, SegmentGroup, VariableBase, build_segment, segment_to_variable
 from dify_graph.variables.consts import SELECTORS_LENGTH
 from dify_graph.variables.segments import FileSegment, ObjectSegment
-from dify_graph.variables.variables import RAGPipelineVariableInput, Variable
+from dify_graph.variables.variables import Variable
 
 VariableValue = Union[str, int, float, dict[str, object], list[object], File]
 
@@ -36,54 +29,6 @@ class VariablePool(BaseModel):
         default=defaultdict(dict),
     )
 
-    # The `user_inputs` is used only when constructing the inputs for the `StartNode`. It's not used elsewhere.
-    user_inputs: Mapping[str, Any] = Field(
-        description="User inputs",
-        default_factory=dict,
-    )
-    system_variables: SystemVariable = Field(
-        description="System variables",
-        default_factory=SystemVariable.default,
-    )
-    environment_variables: Sequence[Variable] = Field(
-        description="Environment variables.",
-        default_factory=list[Variable],
-    )
-    conversation_variables: Sequence[Variable] = Field(
-        description="Conversation variables.",
-        default_factory=list[Variable],
-    )
-    rag_pipeline_variables: list[RAGPipelineVariableInput] = Field(
-        description="RAG pipeline variables.",
-        default_factory=list,
-    )
-
-    def model_post_init(self, context: Any, /):
-        # Create a mapping from field names to SystemVariableKey enum values
-        self._add_system_variables(self.system_variables)
-        # Add environment variables to the variable pool
-        for var in self.environment_variables:
-            self.add((ENVIRONMENT_VARIABLE_NODE_ID, var.name), var)
-        # Add conversation variables to the variable pool. When restoring from a serialized
-        # snapshot, `variable_dictionary` already carries the latest runtime values.
-        # In that case, keep existing entries instead of overwriting them with the
-        # bootstrap list.
-        for var in self.conversation_variables:
-            selector = (CONVERSATION_VARIABLE_NODE_ID, var.name)
-            if self._has(selector):
-                continue
-            self.add(selector, var)
-        # Add rag pipeline variables to the variable pool
-        if self.rag_pipeline_variables:
-            rag_pipeline_variables_map: defaultdict[Any, dict[Any, Any]] = defaultdict(dict)
-            for rag_var in self.rag_pipeline_variables:
-                node_id = rag_var.variable.belong_to_node_id
-                key = rag_var.variable.variable
-                value = rag_var.value
-                rag_pipeline_variables_map[node_id][key] = value
-            for key, value in rag_pipeline_variables_map.items():
-                self.add((RAG_PIPELINE_VARIABLE_NODE_ID, key), value)
-
     def add(self, selector: Sequence[str], value: Any, /):
         """
         Add a variable to the variable pool.
@@ -261,19 +206,7 @@ class VariablePool(BaseModel):
 
         return result
 
-    def _add_system_variables(self, system_variable: SystemVariable):
-        sys_var_mapping = system_variable.to_dict()
-        for key, value in sys_var_mapping.items():
-            if value is None:
-                continue
-            selector = (SYSTEM_VARIABLE_NODE_ID, key)
-            # If the system variable already exists, do not add it again.
-            # This ensures that we can keep the id of the system variables intact.
-            if self._has(selector):
-                continue
-            self.add(selector, value)
-
     @classmethod
     def empty(cls) -> VariablePool:
         """Create an empty variable pool."""
-        return cls(system_variables=SystemVariable.default())
+        return cls()