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

fix: create data component not saving old values on check and save (#2483)

Browse source 
Fixed Create Data component to maintain old values and use new code structure

Co-authored-by: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
This commit is contained in:
Lucas Oliveira 2024-07-02 16:20:52 -03:00 • committed by GitHub
commit 629df8fef3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 36 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

59
src/backend/base/langflow/components/prototypes/CreateData.py
View file

@ -1,29 +1,46 @@
from typing import Any from typing import Any
from langflow.custom import CustomComponent from langflow.custom import Component
from langflow.inputs.inputs import IntInput, MessageTextInput
from langflow.io import Output
from langflow.field_typing.range_spec import RangeSpec from langflow.field_typing.range_spec import RangeSpec
from langflow.schema import Data from langflow.schema import Data
from langflow.schema.dotdict import dotdict from langflow.schema.dotdict import dotdict
from langflow.template.field.base import Input from langflow.template.field.base import Input
class CreateDataComponent(CustomComponent): class CreateDataComponent(Component):
display_name = "Create Data" display_name: str = "Create Data"
description = "Dynamically create a Data with a specified number of fields." description: str = "Dynamically create a Data with a specified number of fields."
field_order = ["number_of_fields", "text_key"]
inputs = [
IntInput(
name="number_of_fields",
display_name="Number of Fields",
info="Number of fields to be added to the record.",
real_time_refresh=True,
range_spec=RangeSpec(min=1, max=15, step=1, step_type="int"),
),
MessageTextInput(name="text_key", display_name="Text Key", info="Key to be used as text.", advanced=True),
]
outputs = [
Output(display_name="Data", name="data", method="build_data"),
]
def update_build_config(self, build_config: dotdict, field_value: Any, field_name: str | None = None): def update_build_config(self, build_config: dotdict, field_value: Any, field_name: str | None = None):
if field_name == "number_of_fields": if field_name == "number_of_fields":
default_keys = ["code", "_type", "number_of_fields", "text_key"] default_keys = ["code", "_type", "number_of_fields", "text_key"]
try: try:
field_value_int = int(field_value) field_value_int = int(field_value)
except TypeError: except ValueError:
return build_config return build_config
existing_fields = {} existing_fields = {}
if field_value_int > 15: if field_value_int > 15:
build_config["number_of_fields"]["value"] = 15 build_config["number_of_fields"]["value"] = 15
raise ValueError("Number of fields cannot exceed 15. Try using a Component to combine two Data.") raise ValueError("Number of fields cannot exceed 15. Try using a Component to combine two Data.")
if len(build_config) > len(default_keys) + field_value_int: if len(build_config) > len(default_keys):
# back up the existing template fields # back up the existing template fields
for key in build_config.copy(): for key in build_config.copy():
if key not in default_keys: if key not in default_keys:
@ -47,22 +64,7 @@ class CreateDataComponent(CustomComponent):
build_config["number_of_fields"]["value"] = field_value_int build_config["number_of_fields"]["value"] = field_value_int
return build_config return build_config
def build_config(self): async def build_data(
return {
"number_of_fields": {
"display_name": "Number of Fields",
"info": "Number of fields to be added to the record.",
"real_time_refresh": True,
"rangeSpec": RangeSpec(min=1, max=15, step=1, step_type="int"),
},
"text_key": {
"display_name": "Text Key",
"info": "Key to be used as text.",
"advanced": True,
},
}
def build(
self, self,
number_of_fields: int = 0, number_of_fields: int = 0,
text_key: str = "text", text_key: str = "text",
@ -79,3 +81,14 @@ class CreateDataComponent(CustomComponent):
return_data = Data(data=data, text_key=text_key) return_data = Data(data=data, text_key=text_key)
self.status = return_data self.status = return_data
return return_data return return_data
def post_code_processing(self, new_build_config: dict, current_build_config: dict):
"""
This function is called after the code validation is done.
"""
frontend_node = super().post_code_processing(new_build_config, current_build_config)
new_build_config = self.update_build_config(
frontend_node["template"], frontend_node["template"]["number_of_fields"]["value"], "number_of_fields"
)
frontend_node = super().post_code_processing(new_build_config, current_build_config)
return frontend_node