Merge pull request #1030 from caarmen/fix-pydantic2-blueprint

Fix the blueprint for pydantic version 2
tfranzel · Nov 13, 2023 · fbbcab3 · fbbcab3
2 parents edce053 + d1b874a
commit fbbcab3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/blueprints/pydantic2.py b/docs/blueprints/pydantic2.py
@@ -14,10 +14,10 @@ def get_name(self, auto_schema, direction):
     def map_serializer(self, auto_schema, direction):
 
         # let pydantic generate a JSON schema
-        schema = model_json_schema(self.target, ref_template="#/components/schemas")
+        schema = model_json_schema(self.target, ref_template="#/components/schemas/{model}")
 
         # pull out potential sub-schemas and put them into component section
-        for sub_name, sub_schema in schema.pop("definitions", {}).items():
+        for sub_name, sub_schema in schema.pop("$defs", {}).items():
             component = ResolvedComponent(
                 name=sub_name,
                 type=ResolvedComponent.SCHEMA,