From 5f7771bc474bf2145bea46efb4d3839a7badd8dd Mon Sep 17 00:00:00 2001 From: Novice Lee Date: Mon, 2 Dec 2024 21:13:47 +0800 Subject: [PATCH] fix: iteration node use the main thread pool --- api/core/workflow/nodes/iteration/iteration_node.py | 2 +- api/libs/helper.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/workflow/nodes/iteration/iteration_node.py b/api/core/workflow/nodes/iteration/iteration_node.py index 8aa9811d33..9bc3b7c14b 100644 --- a/api/core/workflow/nodes/iteration/iteration_node.py +++ b/api/core/workflow/nodes/iteration/iteration_node.py @@ -116,7 +116,7 @@ class IterationNode(BaseNode[IterationNodeData]): variable_pool.add([self.node_id, "item"], iterator_list_value[0]) # init graph engine - from core.workflow.graph_engine.graph_engine import GraphEngine, GraphEngineThreadPool + from core.workflow.graph_engine.graph_engine import GraphEngine graph_engine = GraphEngine( tenant_id=self.tenant_id, diff --git a/api/libs/helper.py b/api/libs/helper.py index 7652d73c8b..3c1b640a20 100644 --- a/api/libs/helper.py +++ b/api/libs/helper.py @@ -10,10 +10,10 @@ from collections.abc import Generator, Mapping from datetime import datetime from hashlib import sha256 from typing import Any, Optional, Union -from zoneinfo import available_timezones from flask import Response, stream_with_context from flask_restful import fields +from zoneinfo import available_timezones from configs import dify_config from core.app.features.rate_limiting.rate_limit import RateLimitGenerator