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

Refactor build_schema_from_inputs to use display_name and description for field names and descriptions respectively

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-25 12:26:23 -03:00
commit f95b20322e
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/helpers/flow.py
View file

@ -136,7 +136,7 @@ async def flow_function({func_args}):
compiled_func = compile(func_body, "<string>", "exec") compiled_func = compile(func_body, "<string>", "exec")
local_scope = {} local_scope = {}
exec(compiled_func, globals(), 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]: 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 = {} fields = {}
for input_ in inputs: 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) return create_model(name, **fields)