From d440905af630d7a8f95836375ec7b71aca3fcb12 Mon Sep 17 00:00:00 2001
From: Eugene Yurtsev <eyurtsev@gmail.com>
Date: Wed, 11 Sep 2024 14:07:43 -0400
Subject: [PATCH] x

---
 langserve/serialization.py             | 2 --
 tests/unit_tests/test_serialization.py | 4 ++--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/langserve/serialization.py b/langserve/serialization.py
index 41433fc9..5093d5de 100644
--- a/langserve/serialization.py
+++ b/langserve/serialization.py
@@ -116,8 +116,6 @@ def _decode_lc_objects(value: Any) -> Any:
         try:
             obj = WellKnownLCObject.model_validate(v)
             parsed = obj.root
-            if set(parsed.model_dump()) != set(value):
-                raise ValueError("Invalid object")
             return parsed
         except (ValidationError, ValueError, TypeError):
             return v
diff --git a/tests/unit_tests/test_serialization.py b/tests/unit_tests/test_serialization.py
index 526c7954..9d3f7a4e 100644
--- a/tests/unit_tests/test_serialization.py
+++ b/tests/unit_tests/test_serialization.py
@@ -6,6 +6,7 @@
 import pytest
 from langchain_core.documents.base import Document
 from langchain_core.messages import HumanMessage, HumanMessageChunk, SystemMessage
+from langchain_core.outputs import ChatGeneration
 from pydantic import BaseModel
 
 from langserve.serialization import (
@@ -48,8 +49,7 @@ def test_document_serialization() -> None:
             "numbers": [1, 2, 3],
             "boom": "Hello, world!",
         },
-        # Requires typing ChatGeneration with Anymessage
-        # [ChatGeneration(message=HumanMessage(content="Hello"))],
+        [ChatGeneration(message=HumanMessage(content="Hello"))],
     ],
 )
 def test_serialization(data: Any) -> None: