From f95b20322e8bac9bf982e617580e077b55de1ca2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 25 Mar 2024 12:26:23 -0300 Subject: [PATCH] Refactor build_schema_from_inputs to use display_name and description for field names and descriptions respectively --- src/backend/langflow/helpers/flow.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/helpers/flow.py b/src/backend/langflow/helpers/flow.py index 90d360f57..5194ce05c 100644 --- a/src/backend/langflow/helpers/flow.py +++ b/src/backend/langflow/helpers/flow.py @@ -136,7 +136,7 @@ async def flow_function({func_args}): compiled_func = compile(func_body, "", "exec") local_scope = {} exec(compiled_func, globals(), local_scope) - return local_scope["dynamic_flow_function"] + return local_scope["flow_function"] def build_function_and_schema(flow_record: Record, graph: "Graph") -> Tuple[Callable, BaseModel]: @@ -189,5 +189,7 @@ def build_schema_from_inputs(name: str, inputs: List[tuple[str, str, str]]) -> B """ fields = {} for input_ in inputs: - fields[input_[1]] = (str, Field(default="", description=input_[2])) + field_name = input_.display_name.lower().replace(" ", "_") + description = input_.description + fields[field_name] = (str, Field(default="", description=description)) return create_model(name, **fields)