Commit69a1c16

authored

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

Fix serialization for BaseModel subclasses by handling type objects separately

1 parent33a42bc commit69a1c16Copy full SHA for 69a1c16

File tree

-5

lines changed

-5

lines changed

Lines changed: 5 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,9 @@`
`8`	`8`
`9`	`9`	`defrecursive_serialize_or_str(obj):`
`10`	`10`	`try:`
	`11`	`+ifisinstance(obj,type)andissubclass(obj,BaseModel):`
	`12`	`+# This a type BaseModel and not an instance of it`
	`13`	`+returnrepr(obj)`
`11`	`14`	`ifisinstance(obj,str):`
`12`	`15`	`returnobj`
`13`	`16`	`ifisinstance(obj,datetime):`
`@@ -30,13 +33,10 @@ def recursive_serialize_or_str(obj):`
`30`	`33`	`# return f"{obj}" this generates '<generator object BaseChatModel.stream at 0x33e9ec770>'`
`31`	`34`	`# it is not useful`
`32`	`35`	`return"Unconsumed Stream"`
`33`		`-ifhasattr(obj,"dict"):`
	`36`	`+ifhasattr(obj,"dict")andnotisinstance(obj,type):`
`34`	`37`	`return {k:recursive_serialize_or_str(v)fork,vinobj.dict().items()}`
`35`		`-ifhasattr(obj,"model_dump"):`
	`38`	`+ifhasattr(obj,"model_dump")andnotisinstance(obj,type):`
`36`	`39`	`return {k:recursive_serialize_or_str(v)fork,vinobj.model_dump().items()}`
`37`		`-ifisinstance(obj,type)andissubclass(obj,BaseModel):`
`38`		`-# This a type BaseModel and not an instance of it`
`39`		`-returnrepr(obj)`
`40`	`40`	`returnstr(obj)`
`41`	`41`	`exceptException:# noqa: BLE001`
`42`	`42`	`logger.debug(f"Cannot serialize object{obj}")`

Comments

(0)