x

langchain-ai · Sep 11, 2024 · d440905 · d440905
1 parent ff94f96
commit d440905
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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: