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

Add SQLModelSerializable class for SQLModel serialization

Browse source
This commit is contained in:
anovazzi1 2024-01-22 15:54:52 -03:00
commit e6b8cf551b
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/services/database/models/base.py
View file

@ -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