From 485e1425d55e7eda6723ddae83dd6146b6e6241f Mon Sep 17 00:00:00 2001 From: kimtaewoong Date: Fri, 4 Jul 2025 04:38:58 +0900 Subject: [PATCH] chore: apply code style reformat via dev/reformat --- api/core/workflow/nodes/llm/node.py | 2 ++ api/tests/unit_tests/core/workflow/nodes/llm/test_node.py | 6 +++--- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/api/core/workflow/nodes/llm/node.py b/api/core/workflow/nodes/llm/node.py index 9e338d87f1..1dff81f870 100644 --- a/api/core/workflow/nodes/llm/node.py +++ b/api/core/workflow/nodes/llm/node.py @@ -101,6 +101,7 @@ logger = logging.getLogger(__name__) # Environment variable to control thinking tags preservation (default: true to maintain backward compatibility) # helper & env‑flag + def _env_flag(name: str, default: bool = True) -> bool: """Return an env var as bool (1/0 | true/false | yes/no).""" val = os.getenv(name) @@ -108,6 +109,7 @@ def _env_flag(name: str, default: bool = True) -> bool: return default return val.lower() in {"1", "true", "yes"} + # keep `` blocks unless explicitly disabled LLM_NODE_THINKING_TAGS_ENABLED = _env_flag("LLM_NODE_THINKING_TAGS_ENABLED") diff --git a/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py b/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py index c79c813e09..519a6cf44e 100644 --- a/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py +++ b/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py @@ -1,10 +1,10 @@ import base64 +import importlib +import os import uuid from collections.abc import Sequence from typing import Optional from unittest import mock -import os -import importlib import pytest @@ -37,7 +37,7 @@ from core.workflow.nodes.llm.entities import ( VisionConfigOptions, ) from core.workflow.nodes.llm.file_saver import LLMFileSaver -from core.workflow.nodes.llm.node import LLMNode, LLM_NODE_THINKING_TAGS_ENABLED +from core.workflow.nodes.llm.node import LLM_NODE_THINKING_TAGS_ENABLED, LLMNode from models.enums import UserFrom from models.provider import ProviderType from models.workflow import WorkflowType