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

Refactor Component class to improve input mapping and validation

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 10:35:31 -03:00
commit 4717ff0767
1 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/base/langflow/custom/custom_component/component.py
View file

@ -4,7 +4,6 @@ from uuid import UUID
import yaml
from git import TYPE_CHECKING
from loguru import logger
from pydantic import BaseModel
from langflow.inputs.inputs import InputTypes
@ -52,18 +51,20 @@ class Component(CustomComponent):
self.map_inputs(self.inputs)
def map_inputs(self, inputs: List[Input]):
self._inputs = {}
self.inputs = inputs
for input_ in inputs:
self._inputs[input_.name] = input_
def _validate_inputs(self, params: dict):
# Params keys are the `name` attribute of the Input objects
for key, value in params.items():
for key, value in params.copy().items():
if key not in self._inputs:
continue
input_ = self._inputs[key]
# BaseInputMixin has a `validate_assignment=True`
input_.value = value
params[input_.name] = input_.value
def set_attributes(self, params: dict):
self._validate_inputs(params)
@ -71,10 +72,6 @@ class Component(CustomComponent):
if key in self.__dict__:
raise ValueError(f"Key {key} already exists in {self.__class__.__name__}")
setattr(self, key, value)
for input_ in self.inputs:
if input_.name not in params:
setattr(self, input_.name, input_.value)
logger.warning(f"Input {input_.name} not found in arguments")
self._arguments = params
def _set_outputs(self, outputs: List[dict]):