From ce031fcc35c32ce092fddc8e8aa0a0a359d47870 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 22:01:36 -0300 Subject: [PATCH] refactor: Add description to 'name' field in Output class --- src/backend/base/langflow/template/field/base.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/template/field/base.py b/src/backend/base/langflow/template/field/base.py index 74f71dd05..c2b27ab5e 100644 --- a/src/backend/base/langflow/template/field/base.py +++ b/src/backend/base/langflow/template/field/base.py @@ -2,7 +2,8 @@ from enum import Enum from typing import Optional # type: ignore from typing import Any, Callable, Union -from pydantic import BaseModel, ConfigDict, Field, field_serializer, field_validator, model_serializer, model_validator +from pydantic import (BaseModel, ConfigDict, Field, field_serializer, + field_validator, model_serializer, model_validator) from langflow.field_typing import Text from langflow.field_typing.range_spec import RangeSpec @@ -147,7 +148,7 @@ class Output(BaseModel): selected: Optional[str] = Field(default=None) """The selected output type for the field.""" - name: Optional[str] = Field(default=None) + name: str = Field(description="The name of the field.") """The name of the field.""" hidden: Optional[bool] = Field(default=None) @@ -193,3 +194,7 @@ class Output(BaseModel): if self.display_name is None: self.display_name = self.name return self + raise ValueError("name must be set") + if self.display_name is None: + self.display_name = self.name + return self