From 766fd2b08f0628ff6cf42b0b132112a372b60067 Mon Sep 17 00:00:00 2001 From: Lucas Oliveira Date: Wed, 29 Nov 2023 13:44:41 -0300 Subject: [PATCH] Formatting --- src/backend/langflow/template/frontend_node/base.py | 9 ++------- .../template/frontend_node/formatter/field_formatters.py | 9 ++------- 2 files changed, 4 insertions(+), 14 deletions(-) diff --git a/src/backend/langflow/template/frontend_node/base.py b/src/backend/langflow/template/frontend_node/base.py index 353b3465c..2946f1e9f 100644 --- a/src/backend/langflow/template/frontend_node/base.py +++ b/src/backend/langflow/template/frontend_node/base.py @@ -3,8 +3,7 @@ from collections import defaultdict from typing import ClassVar, DefaultDict, Dict, List, Optional from langflow.template.field.base import TemplateField -from langflow.template.frontend_node.constants import (CLASSES_TO_REMOVE, - FORCE_SHOW_FIELDS) +from langflow.template.frontend_node.constants import CLASSES_TO_REMOVE, FORCE_SHOW_FIELDS from langflow.template.frontend_node.formatter import field_formatters from langflow.template.template.base import Template from langflow.utils import constants @@ -136,11 +135,7 @@ class FrontendNode(BaseModel): if "dict" in _type.lower() and field.name == "dict_": field.field_type = "file" field.file_types = [".json", ".yaml", ".yml"] - elif ( - _type.startswith("Dict") - or _type.startswith("Mapping") - or _type.startswith("dict") - ): + elif _type.startswith("Dict") or _type.startswith("Mapping") or _type.startswith("dict"): field.field_type = "dict" return _type diff --git a/src/backend/langflow/template/frontend_node/formatter/field_formatters.py b/src/backend/langflow/template/frontend_node/formatter/field_formatters.py index 9da1ab4d0..dc57ff04f 100644 --- a/src/backend/langflow/template/frontend_node/formatter/field_formatters.py +++ b/src/backend/langflow/template/frontend_node/formatter/field_formatters.py @@ -4,8 +4,7 @@ from typing import ClassVar, Dict, Optional from langflow.template.field.base import TemplateField from langflow.template.frontend_node.constants import FORCE_SHOW_FIELDS from langflow.template.frontend_node.formatter.base import FieldFormatter -from langflow.utils.constants import (ANTHROPIC_MODELS, CHAT_OPENAI_MODELS, - OPENAI_MODELS) +from langflow.utils.constants import ANTHROPIC_MODELS, CHAT_OPENAI_MODELS, OPENAI_MODELS class OpenAIAPIKeyFormatter(FieldFormatter): @@ -150,9 +149,5 @@ class DictCodeFileFormatter(FieldFormatter): if "dict" in _type.lower() and key == "dict_": field.field_type = "file" field.file_types = [".json", ".yaml", ".yml"] - elif ( - _type.startswith("Dict") - or _type.startswith("Mapping") - or _type.startswith("dict") - ): + elif _type.startswith("Dict") or _type.startswith("Mapping") or _type.startswith("dict"): field.field_type = "dict"