From 4b643a8b67f74d1653d8a1a8bc41edb291c48138 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 15:03:43 -0300 Subject: [PATCH] refactor: Fix selected type assignment in Output class --- src/backend/base/langflow/template/field/base.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/backend/base/langflow/template/field/base.py b/src/backend/base/langflow/template/field/base.py index e8c99bd70..33e2d991d 100644 --- a/src/backend/base/langflow/template/field/base.py +++ b/src/backend/base/langflow/template/field/base.py @@ -1,7 +1,6 @@ from enum import Enum from typing import Any, Callable, GenericAlias, Optional, Union, _GenericAlias, _UnionGenericAlias - from pydantic import BaseModel, ConfigDict, Field, field_serializer, field_validator, model_serializer, model_validator from langflow.field_typing import Text @@ -185,7 +184,7 @@ class Output(BaseModel): self.types.append(type_) def set_selected(self): - if not self.selected: + if not self.selected and self.types: self.selected = self.types[0] @model_serializer(mode="wrap")