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

fix: serialize BaseModel subclasses and avoid calling methods on types (#4479)

Browse source 
Fix serialization for BaseModel subclasses by handling type objects separately
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-11-08 22:23:01 -03:00 • committed by GitHub
commit 69a1c16e82
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/base/langflow/schema/serialize.py
View file

@ -8,6 +8,9 @@ from pydantic.v1 import BaseModel as BaseModelV1
def recursive_serialize_or_str(obj): def recursive_serialize_or_str(obj):
try: try:
if isinstance(obj, type) and issubclass(obj, BaseModel):
# This a type BaseModel and not an instance of it
return repr(obj)
if isinstance(obj, str): if isinstance(obj, str):
return obj return obj
if isinstance(obj, datetime): if isinstance(obj, datetime):
@ -30,13 +33,10 @@ def recursive_serialize_or_str(obj):
# return f"{obj}" this generates '<generator object BaseChatModel.stream at 0x33e9ec770>' # return f"{obj}" this generates '<generator object BaseChatModel.stream at 0x33e9ec770>'
# it is not useful # it is not useful
return "Unconsumed Stream" return "Unconsumed Stream"
if hasattr(obj, "dict"): if hasattr(obj, "dict") and not isinstance(obj, type):
return {k: recursive_serialize_or_str(v) for k, v in obj.dict().items()} return {k: recursive_serialize_or_str(v) for k, v in obj.dict().items()}
if hasattr(obj, "model_dump"): if hasattr(obj, "model_dump") and not isinstance(obj, type):
return {k: recursive_serialize_or_str(v) for k, v in obj.model_dump().items()} return {k: recursive_serialize_or_str(v) for k, v in obj.model_dump().items()}
if isinstance(obj, type) and issubclass(obj, BaseModel):
# This a type BaseModel and not an instance of it
return repr(obj)
return str(obj) return str(obj)
except Exception: # noqa: BLE001 except Exception: # noqa: BLE001
logger.debug(f"Cannot serialize object {obj}") logger.debug(f"Cannot serialize object {obj}")