fix: update core.variables imports to core.workflow.variables after S2 merge

Made-with: Cursor
This commit is contained in:
Novice
2026-03-23 09:05:37 +08:00
parent cccff6768a
commit 66e67caa2b
8 changed files with 127 additions and 21 deletions

View File

@ -360,7 +360,6 @@ class Node(Generic[NodeDataT]):
if not isinstance(event, NodeRunStreamChunkEvent):
yield event
def run(self) -> Generator[GraphNodeEventBase, None, None]:
execution_id = self.ensure_execution_id()
self._start_at = naive_utc_now()

View File

@ -6,8 +6,8 @@ from pathlib import PurePosixPath
from typing import Any, cast
from core.sandbox.bash.session import SANDBOX_READY_TIMEOUT
from core.variables import ArrayFileSegment
from core.variables.segments import ArrayStringSegment, FileSegment
from core.workflow.variables import ArrayFileSegment
from core.workflow.variables.segments import ArrayStringSegment, FileSegment
from core.virtual_environment.__base.command_future import CommandCancelledError, CommandTimeoutError
from core.virtual_environment.__base.helpers import pipeline
from core.workflow.enums import NodeType, WorkflowNodeExecutionStatus

View File

@ -2183,7 +2183,7 @@ class LLMNode(Node[LLMNodeData]):
def _extract_prompt_files(self, variable_pool: VariablePool) -> list[File]:
"""Extract files from prompt template variables."""
from core.variables import ArrayFileVariable, FileVariable
from core.workflow.variables import ArrayFileVariable, FileVariable
files: list[File] = []