chore: apply code style reformat via dev/reformat

pull/21897/head
kimtaewoong 11 months ago
parent ebd27801a1
commit 485e1425d5

@ -101,6 +101,7 @@ logger = logging.getLogger(__name__)
# Environment variable to control thinking tags preservation (default: true to maintain backward compatibility) # Environment variable to control thinking tags preservation (default: true to maintain backward compatibility)
# helper & envflag # helper & envflag
def _env_flag(name: str, default: bool = True) -> bool: def _env_flag(name: str, default: bool = True) -> bool:
"""Return an env var as bool (1/0 | true/false | yes/no).""" """Return an env var as bool (1/0 | true/false | yes/no)."""
val = os.getenv(name) val = os.getenv(name)
@ -108,6 +109,7 @@ def _env_flag(name: str, default: bool = True) -> bool:
return default return default
return val.lower() in {"1", "true", "yes"} return val.lower() in {"1", "true", "yes"}
# keep `<think>` blocks unless explicitly disabled # keep `<think>` blocks unless explicitly disabled
LLM_NODE_THINKING_TAGS_ENABLED = _env_flag("LLM_NODE_THINKING_TAGS_ENABLED") LLM_NODE_THINKING_TAGS_ENABLED = _env_flag("LLM_NODE_THINKING_TAGS_ENABLED")

@ -1,10 +1,10 @@
import base64 import base64
import importlib
import os
import uuid import uuid
from collections.abc import Sequence from collections.abc import Sequence
from typing import Optional from typing import Optional
from unittest import mock from unittest import mock
import os
import importlib
import pytest import pytest
@ -37,7 +37,7 @@ from core.workflow.nodes.llm.entities import (
VisionConfigOptions, VisionConfigOptions,
) )
from core.workflow.nodes.llm.file_saver import LLMFileSaver 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.enums import UserFrom
from models.provider import ProviderType from models.provider import ProviderType
from models.workflow import WorkflowType from models.workflow import WorkflowType

Loading…
Cancel
Save