diff --git a/src/backend/langflow/node/nodes.py b/src/backend/langflow/node/nodes.py index c38ad8e2f..a628af510 100644 --- a/src/backend/langflow/node/nodes.py +++ b/src/backend/langflow/node/nodes.py @@ -61,7 +61,7 @@ class PythonFunctionNode(FrontendNode): multiline=True, value=DEFAULT_PYTHON_FUNCTION, name="code", - ), + ) ], ) description: str = "Python function to be executed." diff --git a/src/backend/langflow/node/template.py b/src/backend/langflow/node/template.py index cc0245730..21b23486a 100644 --- a/src/backend/langflow/node/template.py +++ b/src/backend/langflow/node/template.py @@ -16,7 +16,12 @@ class Field(BaseModel): def to_dict(self): result = self.dict() + # Remove key if it is None + for key in list(result.keys()): + if result[key] is None: + del result[key] result["type"] = result.pop("field_type") + result["list"] = result.pop("is_list") return result @@ -26,7 +31,7 @@ class Template(BaseModel): def to_dict(self): result = {field.name: field.to_dict() for field in self.fields} - result["_type"] = self.type_name + result["_type"] = self.type_name # type: ignore return result