Lazy import model name formatter when we need it

turner-townsend · Aug 11, 2024 · 36187dc · 36187dc
1 parent 8d271e3
commit 36187dc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/flask_pydantic_spec/types.py b/flask_pydantic_spec/types.py
@@ -4,7 +4,6 @@
 from pydantic import BaseModel
 
 from flask_pydantic_spec.constants import OPENAPI_SCHEMA_TEMPLATE
-from flask_pydantic_spec.utils import get_model_name
 
 
 class ResponseBase:
@@ -117,6 +116,7 @@ def generate_spec(self) -> Dict[str, Any]:
 
     @staticmethod
     def get_schema(model: Type[BaseModel], is_list: bool = False) -> Mapping[str, Any]:
+        from flask_pydantic_spec.utils import get_model_name
         ref_schema = {"$ref": f"#/components/schemas/{get_model_name(model)}"}
         if is_list:
             return {"schema": {"type": "array", "items": ref_schema}}
@@ -172,6 +172,8 @@ def has_model(self) -> bool:
         return self.model is not None
 
     def generate_spec(self) -> Mapping[str, Any]:
+        from flask_pydantic_spec.utils import get_model_name
+
         if self.content_type == "application/octet-stream":
             return {
                 "content": {