adds model_name option in ConfigDict

pydantic/_internal/_config.py

@@ -41,6 +41,7 @@ class ConfigWrapper:
     # all annotations are copied directly from ConfigDict, and should be kept up to date, a test will fail if they
     # stop matching
     title: str | None
+    model_name: str | None
     str_to_lower: bool
     str_to_upper: bool
     str_strip_whitespace: bool
@@ -217,6 +218,7 @@ def push(self, config_wrapper: ConfigWrapper | ConfigDict | None):
 
 config_defaults = ConfigDict(
     title=None,
+    model_name=None,
     str_to_lower=False,
     str_to_upper=False,
     str_strip_whitespace=False,

pydantic/_internal/_model_construction.py

@@ -88,6 +88,7 @@ def __new__(
             base_field_names, class_vars, base_private_attributes = mcs._collect_bases_data(bases)
 
             config_wrapper = ConfigWrapper.for_model(bases, namespace, kwargs)
+            cls_name = config_wrapper.config_dict.get('model_name') or cls_name
             namespace['model_config'] = config_wrapper.config_dict
             private_attributes = inspect_namespace(
                 namespace, config_wrapper.ignored_types, class_vars, base_field_names

pydantic/config.py

@@ -33,6 +33,9 @@ class ConfigDict(TypedDict, total=False):
     title: str | None
     """The title for the generated JSON schema, defaults to the model's name"""
 
+    model_name: str | None
+    """If not `None` overrides the model names generated by `pydantic.v1.schema.get_model_name_map()`."""
+
     str_to_lower: bool
     """Whether to convert all characters to lowercase for str types. Defaults to `False`."""
 

pydantic/v1/schema.py

@@ -89,7 +89,8 @@
 default_prefix = '#/definitions/'
 default_ref_template = '#/definitions/{model}'
 
-TypeModelOrEnum = Union[Type['BaseModel'], Type[Enum]]
+T = TypeVar('T', bound='BaseModel')
+TypeModelOrEnum = Union[Type[T], Type[Enum]]
 TypeModelSet = Set[TypeModelOrEnum]
 
 

tests/test_config.py

@@ -29,6 +29,7 @@
 from pydantic.errors import PydanticUserError
 from pydantic.fields import FieldInfo
 from pydantic.type_adapter import TypeAdapter
+from pydantic.v1.schema import get_model_name_map
 from pydantic.warnings import PydanticDeprecationWarning
 
 if sys.version_info < (3, 9):
@@ -492,6 +493,66 @@ def my_function():
         pass
 
 
+def test_config_model_name() -> None:
+    CLIENT_USER_MODEL_NAME = 'ClientUser'
+    BUSINESS_USER_MODEL_NAME = 'BusinessUser'
+
+    def _get_business_user_class():
+        class User(BaseModel):
+            model_config = ConfigDict(model_name=BUSINESS_USER_MODEL_NAME)
+
+        return User
+
+    def _get_client_user_class():
+        class User(BaseModel):
+            model_config = ConfigDict(model_name=CLIENT_USER_MODEL_NAME)
+
+        return User
+
+    BusinessUser = _get_business_user_class()
+    ClientUser = _get_client_user_class()
+
+    name_map = get_model_name_map({BusinessUser, ClientUser})
+    assert name_map[BusinessUser] == BUSINESS_USER_MODEL_NAME
+    assert name_map[ClientUser] == CLIENT_USER_MODEL_NAME
+
+    assert BusinessUser().model_json_schema()['title'] == BUSINESS_USER_MODEL_NAME
+    assert ClientUser().model_json_schema()['title'] == CLIENT_USER_MODEL_NAME
+
+
+def test_config_model_name__long_model_name_on_conflict() -> None:
+    CLIENT_USER_MODEL_NAME = 'ClientUser'
+
+    def _get_client_user_class():
+        class User(BaseModel):
+            model_config = ConfigDict(model_name=CLIENT_USER_MODEL_NAME)
+
+        return User
+
+    def _get_client_2_user_class():
+        class User(BaseModel):
+            model_config = ConfigDict(model_name=CLIENT_USER_MODEL_NAME)
+
+        return User
+
+    ClientUser = _get_client_user_class()
+    ClientUserV2 = _get_client_2_user_class()
+
+    name_map = get_model_name_map({ClientUser, ClientUserV2})
+    assert (
+        name_map[ClientUser]
+        == 'tests__test_config__test_config_model_name__long_model_name_on_conflict__<locals>___get_client_user_class__<locals>__User'
+    )
+    assert (
+        name_map[ClientUserV2]
+        == 'tests__test_config__test_config_model_name__long_model_name_on_conflict__<locals>___get_client_2_user_class__<locals>__User'
+    )
+
+    # FIXME: Maye the model_json_schema should have a long name on conflict?
+    assert ClientUser().model_json_schema()['title'] == CLIENT_USER_MODEL_NAME
+    assert ClientUserV2().model_json_schema()['title'] == CLIENT_USER_MODEL_NAME
+
+
 def test_multiple_inheritance_config():
     class Parent(BaseModel):
         model_config = ConfigDict(frozen=True, extra='forbid')