From 86dfeb7eff8b015639f5a703229a16d688fbf8ef Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 19:05:34 -0300 Subject: [PATCH] refactor: Update langflow schema field module to use consistent type hinting for better clarity and consistency --- src/backend/base/langflow/template/field/base.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/template/field/base.py b/src/backend/base/langflow/template/field/base.py index 61224270e..65773cee0 100644 --- a/src/backend/base/langflow/template/field/base.py +++ b/src/backend/base/langflow/template/field/base.py @@ -1,5 +1,13 @@ from enum import Enum -from typing import Any, Callable, GenericAlias, Optional, Union, _GenericAlias, _UnionGenericAlias # type: ignore +from typing import ( + Any, + Callable, + GenericAlias, + Optional, # type: ignore + Union, + _GenericAlias, + _UnionGenericAlias, +) from pydantic import BaseModel, ConfigDict, Field, field_serializer, field_validator, model_serializer, model_validator @@ -181,6 +189,8 @@ class Output(BaseModel): def add_types(self, _type: list[Any]): for type_ in _type: + if self.types is None: + self.types = [] self.types.append(type_) def set_selected(self):