From e6b8cf551b07abef1d868c4024a96ed21bbe51e8 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Mon, 22 Jan 2024 15:54:52 -0300 Subject: [PATCH] Add SQLModelSerializable class for SQLModel serialization --- src/backend/langflow/services/database/models/base.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/backend/langflow/services/database/models/base.py b/src/backend/langflow/services/database/models/base.py index 53ee2c37e..18b56ff90 100644 --- a/src/backend/langflow/services/database/models/base.py +++ b/src/backend/langflow/services/database/models/base.py @@ -1,4 +1,5 @@ import orjson +from sqlmodel import SQLModel def orjson_dumps(v, *, default=None, sort_keys=False, indent_2=True): @@ -15,3 +16,10 @@ def orjson_dumps(v, *, default=None, sort_keys=False, indent_2=True): if default is None: return orjson.dumps(v, option=option).decode() return orjson.dumps(v, default=default, option=option).decode() + + +class SQLModelSerializable(SQLModel): + class Config: + orm_mode = True + json_loads = orjson.loads + json_dumps = orjson_dumps