From 306e593d8ea7515a8b7fcf3a733fec0352e2e318 Mon Sep 17 00:00:00 2001 From: Guilherme Calabria Lopes Date: Wed, 31 Jul 2024 17:51:12 +0200 Subject: [PATCH] fix: Add logic to serialize objects of Send class --- langserve/serialization.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/langserve/serialization.py b/langserve/serialization.py index 186e9ddd..de0a2893 100644 --- a/langserve/serialization.py +++ b/langserve/serialization.py @@ -10,6 +10,7 @@ any information about the exception. This is done to prevent leaking sensitive information from the server to the client. """ + import abc import logging from functools import lru_cache @@ -38,6 +39,7 @@ ) from langchain_core.prompt_values import ChatPromptValueConcrete from langchain_core.prompts.base import StringPromptValue +from langgraph.constants import Send from langserve.pydantic_v1 import BaseModel, ValidationError from langserve.validation import CallbackEvent @@ -87,6 +89,8 @@ def default(obj) -> Any: """Default serialization for well known objects.""" if isinstance(obj, BaseModel): return obj.dict() + if isinstance(obj, Send): + return {"node": obj.node, "arg": obj.arg} return super().default(obj)