From 9da98e6c6cd3dc0918901902ae7dfb699e3d736f Mon Sep 17 00:00:00 2001 From: wangxiaolei Date: Mon, 2 Mar 2026 08:59:53 +0800 Subject: [PATCH] fix: fix import error (#32800) --- api/core/workflow/nodes/llm/llm_utils.py | 2 +- .../nodes/parameter_extractor/parameter_extractor_node.py | 2 +- .../nodes/question_classifier/question_classifier_node.py | 3 ++- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/api/core/workflow/nodes/llm/llm_utils.py b/api/core/workflow/nodes/llm/llm_utils.py index 7e52a1a202..72f150d920 100644 --- a/api/core/workflow/nodes/llm/llm_utils.py +++ b/api/core/workflow/nodes/llm/llm_utils.py @@ -9,13 +9,13 @@ from core.model_runtime.entities.message_entities import ( TextPromptMessageContent, ) from core.model_runtime.entities.model_entities import AIModelEntity +from core.model_runtime.memory import PromptMessageMemory from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel from core.workflow.file.models import File from core.workflow.runtime import VariablePool from core.workflow.variables.segments import ArrayAnySegment, ArrayFileSegment, FileSegment, NoneSegment from .exc import InvalidVariableTypeError -from .protocols import PromptMessageMemory def fetch_model_schema(*, model_instance: ModelInstance) -> AIModelEntity: diff --git a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py index 3353a163ad..4272b98116 100644 --- a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py +++ b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py @@ -17,6 +17,7 @@ from core.model_runtime.entities.message_entities import ( UserPromptMessage, ) from core.model_runtime.entities.model_entities import ModelFeature, ModelPropertyKey +from core.model_runtime.memory import PromptMessageMemory from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel from core.model_runtime.utils.encoders import jsonable_encoder from core.prompt.advanced_prompt_transform import AdvancedPromptTransform @@ -33,7 +34,6 @@ from core.workflow.node_events import NodeRunResult from core.workflow.nodes.base import variable_template_parser from core.workflow.nodes.base.node import Node from core.workflow.nodes.llm import llm_utils -from core.workflow.nodes.llm.protocols import PromptMessageMemory from core.workflow.runtime import VariablePool from core.workflow.variables.types import ArrayValidation, SegmentType from factories.variable_factory import build_segment_with_type diff --git a/api/core/workflow/nodes/question_classifier/question_classifier_node.py b/api/core/workflow/nodes/question_classifier/question_classifier_node.py index 789ff605cc..6005bff1a6 100644 --- a/api/core/workflow/nodes/question_classifier/question_classifier_node.py +++ b/api/core/workflow/nodes/question_classifier/question_classifier_node.py @@ -5,6 +5,7 @@ from typing import TYPE_CHECKING, Any from core.model_manager import ModelInstance from core.model_runtime.entities import LLMUsage, ModelPropertyKey, PromptMessageRole +from core.model_runtime.memory import PromptMessageMemory from core.model_runtime.utils.encoders import jsonable_encoder from core.prompt.simple_prompt_transform import ModelMode from core.prompt.utils.prompt_message_util import PromptMessageUtil @@ -26,7 +27,7 @@ from core.workflow.nodes.llm import ( llm_utils, ) from core.workflow.nodes.llm.file_saver import FileSaverImpl, LLMFileSaver -from core.workflow.nodes.llm.protocols import CredentialsProvider, ModelFactory, PromptMessageMemory +from core.workflow.nodes.llm.protocols import CredentialsProvider, ModelFactory from libs.json_in_md_parser import parse_and_check_json_markdown from .entities import QuestionClassifierNodeData