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

Fix: prevent infinity bug on agent (#4876)

Browse source 
* Fix error handling and logging in build_flow function

* Refactor: Update serialize_field function in utils.py

This commit refactors the serialize_field function in utils.py to handle additional data types. It now properly serializes BaseModel objects and dictionaries by recursively calling the serialize_field function on their values. Additionally, it ensures that all other data types are converted to strings before returning.

#4873

* update serialize
This commit is contained in:
anovazzi1 2024-11-26 18:48:00 -03:00 • committed by GitHub
commit 907d9929e4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/api/v1/chat.py
View file

@ -409,6 +409,10 @@ async def build_flow(
for task in tasks: for task in tasks:
task.cancel() task.cancel()
return return
except Exception as e:
logger.error(f"Error building vertices: {e}")
event_manager.on_error(data={"error": str(e)})
raise
event_manager.on_end(data={}) event_manager.on_end(data={})
await event_manager.queue.put((None, None, time.time)) await event_manager.queue.put((None, None, time.time))

8
src/backend/base/langflow/graph/utils.py
View file

@ -79,14 +79,14 @@ def serialize_field(value):
if isinstance(value, Document): if isinstance(value, Document):
return value.to_json() return value.to_json()
if isinstance(value, BaseModel): if isinstance(value, BaseModel):
return value.model_dump() return serialize_field(value.model_dump())
if isinstance(value, dict):
return {k: serialize_field(v) for k, v in value.items()}
if isinstance(value, V1BaseModel): if isinstance(value, V1BaseModel):
if hasattr(value, "to_json"): if hasattr(value, "to_json"):
return value.to_json() return value.to_json()
return value.dict() return value.dict()
if isinstance(value, str): return str(value)
return {"result": value}
return value
def get_artifact_type(value, build_result) -> str: def get_artifact_type(value, build_result) -> str: