若Pydantic字段为另一个类，如何进行正确的类型注解？

我定义了一个MatchModels模型，其中user_model和item_model字段的值为其他Pydantic类（而非实例），以便后续用作构建器。代码如下：

from typing import Optional
from pydantic import BaseModel

class ItemModel(BaseModel):
    id: int

class UserModel(BaseModel):
    name: str

class MatchModels(BaseModel):
    name: str
    user_model: Optional[type(BaseModel)] = None
    item_model: Optional[type(BaseModel)] = None


m = MatchModels(
    name="match only user",
    user_model=UserModel,
)

请问：若后续要在代码中实例化这些类，此处使用type(BaseModel)是否正确？正确的类型注解方式是什么？

首先，type(BaseModel)这种写法是可行的，但并非最规范的方式。它本质上获取的是BaseModel的类类型，而Python类型提示体系中有专门的写法来表示这类场景。

标准的类型注解方式 #

你需要使用Type[BaseModel]来注解类类型（注意首字母大写的Type，需从typing模块导入），具体调整如下：

1. 导入Type类型：

from typing import Optional, Type

2. 修改MatchModels的字段注解：

class MatchModels(BaseModel):
    name: str
    user_model: Optional[Type[BaseModel]] = None
    item_model: Optional[Type[BaseModel]] = None

这么做的原因 #

• Type[BaseModel]是PEP 484规范中定义的、专门用来表示“BaseModel及其子类的类类型”的写法，比type(BaseModel)更清晰，可读性更强。
• 类型检查工具（如mypy）能更准确地识别这种类型注解，帮你提前发现潜在的类型错误。
• 后续实例化类的逻辑完全不受影响，依然可以正常执行：

# 实例化user_model对应的类
user_instance = m.user_model(name="Alice")

进阶：更严格的类型约束 #

如果希望user_model只能接收UserModel及其子类，item_model只能接收ItemModel及其子类，可以把注解改成更具体的类型：

class MatchModels(BaseModel):
    name: str
    user_model: Optional[Type[UserModel]] = None
    item_model: Optional[Type[ItemModel]] = None

这种写法能让类型检查工具自动校验传入的类是否符合要求，进一步提升代码的健壮性。

内容的提问来源于stack exchange，提问作者GopherM