Fix openapi >=3 versions compatibility

0b01001001 · Dec 10, 2020 · 7a1d15f · 7a1d15f
1 parent 6226af2
commit 7a1d15f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/spectree/spec.py b/spectree/spec.py
@@ -157,12 +157,12 @@ async def async_validate(*args, **kwargs):
             ):
                 if model is not None:
                     assert issubclass(model, BaseModel)
-                    self.models[model.__name__] = model.schema()
+                    self.models[model.__name__] = model.schema(ref_template="#/components/schemas/{model}")
                     setattr(validation, name, model.__name__)
 
             if resp:
                 for model in resp.models:
-                    self.models[model.__name__] = model.schema()
+                    self.models[model.__name__] = model.schema(ref_template="#/components/schemas/{model}")
                 validation.resp = resp
 
             if tags:
@@ -214,8 +214,7 @@ def _generate_spec(self):
             },
             "tags": list(tags.values()),
             "paths": {**routes},
-            "components": {"schemas": {**self.models}},
-            "definitions": self._get_model_definitions(),
+            "components": {"schemas": {**self.models, **self._get_model_definitions()}},
         }
         return spec