From f87e49e0dca3f328039a4a7e49c32e0eb84127f4 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Thu, 6 Jun 2024 11:52:03 -0300 Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(schema.py):=20refactor=20?= =?UTF-8?q?serialize=5Fmodel=20method=20in=20Record=20class=20to=20use=20s?= =?UTF-8?q?elf=20instead=20of=20cls=20and=20improve=20method=20signature?= =?UTF-8?q?=20for=20better=20readability=20and=20maintainability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/schema/schema.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/schema/schema.py b/src/backend/base/langflow/schema/schema.py index 624ba7dcb..e3394ec5e 100644 --- a/src/backend/base/langflow/schema/schema.py +++ b/src/backend/base/langflow/schema/schema.py @@ -30,9 +30,9 @@ class Record(BaseModel): values["data"][key] = values[key] return values - @model_serializer(mode="json") - def serialize_model(cls, obj): - data = {k: v.to_json() if hasattr(v, "to_json") else v for k, v in obj.data.items()} + @model_serializer(mode="plain", when_used="json") + def serialize_model(self): + data = {k: v.to_json() if hasattr(v, "to_json") else v for k, v in self.data.items()} return data def get_text(self):