jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix return value error reported by mypy

Browse source
This commit is contained in:
italojohnny 2024-06-20 17:12:53 -03:00
commit 2212458bef
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/template/template/base.py
View file

@ -1,4 +1,4 @@
from typing import Callable, Union from typing import Callable, Union, cast
from pydantic import BaseModel, Field, model_serializer from pydantic import BaseModel, Field, model_serializer
@ -9,7 +9,8 @@ from langflow.utils.constants import DIRECT_TYPES
class Template(BaseModel): class Template(BaseModel):
type_name: str = Field(serialization_alias="_type") type_name: str = Field(serialization_alias="_type")
fields: list[Input | InputTypes] fields: list[Union[Input, InputTypes]]
def process_fields( def process_fields(
self, self,
@ -49,7 +50,7 @@ class Template(BaseModel):
field = next((field for field in self.fields if field.name == field_name), None) field = next((field for field in self.fields if field.name == field_name), None)
if field is None: if field is None:
raise ValueError(f"Field {field_name} not found in template {self.type_name}") raise ValueError(f"Field {field_name} not found in template {self.type_name}")
return field return cast(Input, field)
def update_field(self, field_name: str, field: Input) -> None: def update_field(self, field_name: str, field: Input) -> None:
"""Updates the field with the given name.""" """Updates the field with the given name."""