chore: save changes

Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN-
2026-03-17 04:16:01 +08:00
parent 2f65a498b1
commit 7b98463916
259 changed files with 3580 additions and 3537 deletions

View File

@@ -2,7 +2,7 @@ import contextlib
import json
import logging
from collections.abc import Callable, Generator, Mapping, Sequence
from datetime import datetime
from datetime import UTC, datetime
from typing import TYPE_CHECKING, Any, Literal, cast
from dify_graph.entities.graph_config import NodeConfigDictAdapter
@@ -31,7 +31,6 @@ from dify_graph.nodes.base import LLMUsageTrackingMixin
from dify_graph.nodes.base.node import Node
from dify_graph.nodes.loop.entities import LoopCompletedReason, LoopNodeData, LoopVariableData
from dify_graph.utils.condition.processor import ConditionProcessor
from dify_graph.utils.datetime_utils import naive_utc_now
from dify_graph.variables import Segment, SegmentType, TypeMismatchError, build_segment_with_type, segment_to_variable
if TYPE_CHECKING:
@@ -90,7 +89,7 @@ class LoopNode(LLMUsageTrackingMixin, Node[LoopNodeData]):
loop_variable_selectors[loop_variable.label] = variable_selector
inputs[loop_variable.label] = processed_segment.value
start_at = naive_utc_now()
start_at = datetime.now(UTC).replace(tzinfo=None)
condition_processor = ConditionProcessor()
loop_duration_map: dict[str, float] = {}
@@ -123,10 +122,10 @@ class LoopNode(LLMUsageTrackingMixin, Node[LoopNodeData]):
self._clear_loop_subgraph_variables(loop_node_ids)
graph_engine = self._create_graph_engine(start_at=start_at, root_node_id=root_node_id)
loop_start_time = naive_utc_now()
loop_start_time = datetime.now(UTC).replace(tzinfo=None)
reach_break_node = yield from self._run_single_loop(graph_engine=graph_engine, current_index=i)
# Track loop duration
loop_duration_map[str(i)] = (naive_utc_now() - loop_start_time).total_seconds()
loop_duration_map[str(i)] = (datetime.now(UTC).replace(tzinfo=None) - loop_start_time).total_seconds()
# Accumulate outputs from the sub-graph's response nodes
for key, value in graph_engine.graph_runtime_state.outputs.items():
@@ -417,6 +416,7 @@ class LoopNode(LLMUsageTrackingMixin, Node[LoopNodeData]):
graph_config=self.graph_config,
run_context=self.run_context,
call_depth=self.workflow_call_depth,
child_sync_variable_node_ids=self.graph_init_params.child_sync_variable_node_ids,
)
# Create a new GraphRuntimeState for this iteration