Support nested Pydantic model in MarkdownJsonDictParser. (#471)

modelscope · Oct 28, 2024 · 7befaf8 · 7befaf8
1 parent eeee97a
commit 7befaf8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/agentscope/parsers/json_object_parser.py b/src/agentscope/parsers/json_object_parser.py
@@ -271,7 +271,9 @@ def parse(self, response: ModelResponse) -> ModelResponse:
         # Requirement checking by Pydantic
         if self.pydantic_class is not None:
             try:
-                response.parsed = dict(self.pydantic_class(**response.parsed))
+                response.parsed = self.pydantic_class(
+                    **response.parsed,
+                ).model_dump()
             except Exception as e:
                 raise JsonParsingError(
                     message=str(e),