From 367eadd6e23f7842e40bb79b00fd59cac768d2c2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 25 Jan 2024 11:44:09 -0300 Subject: [PATCH] Add model_config to SQLModelSerializable --- src/backend/langflow/services/database/models/base.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/services/database/models/base.py b/src/backend/langflow/services/database/models/base.py index 18b56ff90..17dd947d9 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 pydantic import ConfigDict from sqlmodel import SQLModel @@ -19,7 +20,4 @@ def orjson_dumps(v, *, default=None, sort_keys=False, indent_2=True): class SQLModelSerializable(SQLModel): - class Config: - orm_mode = True - json_loads = orjson.loads - json_dumps = orjson_dumps + model_config = ConfigDict(from_attributes=True)