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

♻️ (schema.py): refactor serialize_model method in Record class to use self instead of cls and improve method signature for better readability and maintainability

Browse source
This commit is contained in:
ogabrielluiz 2024-06-06 11:52:03 -03:00
commit f87e49e0dc
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/schema/schema.py
View file

@ -30,9 +30,9 @@ class Record(BaseModel):
values["data"][key] = values[key] values["data"][key] = values[key]
return values return values
@model_serializer(mode="json") @model_serializer(mode="plain", when_used="json")
def serialize_model(cls, obj): def serialize_model(self):
data = {k: v.to_json() if hasattr(v, "to_json") else v for k, v in obj.data.items()} data = {k: v.to_json() if hasattr(v, "to_json") else v for k, v in self.data.items()}
return data return data
def get_text(self): def get_text(self):