Add caches for collecting definitions and invalid schemas from a Core…

…Schema (#7527)
pydantic · Sep 20, 2023 · 2d36952 · 2d36952
1 parent 56f41a4
commit 2d36952
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pydantic/_internal/_core_utils.py b/pydantic/_internal/_core_utils.py
@@ -116,6 +116,12 @@ def collect_definitions(schema: core_schema.CoreSchema) -> dict[str, core_schema
     defs: dict[str, CoreSchema] = {}
 
     def _record_valid_refs(s: core_schema.CoreSchema, recurse: Recurse) -> core_schema.CoreSchema:
+        if 'metadata' in s:
+            definitions_cache: _DefinitionsState | None = s['metadata'].get(_DEFINITIONS_CACHE_METADATA_KEY, None)
+            if definitions_cache is not None:
+                defs.update(definitions_cache['definitions'])
+                return s
+
         ref = get_ref(s)
         if ref:
             defs[ref] = s