Pydantic：如何为接收模型类返回实例的函数添加类型提示？

我们需要为接收模型类并返回对应实例的构造函数添加准确的类型提示，避免代码检查工具（linter）报错。先给出基础的Pydantic模型：

from pydantic import BaseModel

class Point(BaseModel):
    x: int
    y: int

方法一：直接使用type[BaseModel]（存在类型不匹配问题） #

这种写法的问题在于，函数返回值被固定标注为BaseModel，当我们把返回值赋值给更具体的Point类型变量时，类型检查工具会判定类型不兼容（父类实例无法直接赋值给子类类型变量）。

def make_model_instance(model: type[BaseModel], **kwargs) -> BaseModel:
    return model(**kwargs)

# 类型检查工具会标记此调用，返回值BaseModel无法匹配Point类型
point: Point = make_model_instance(model=Point, x=1, y=2)

方法二：直接使用绑定BaseModel的TypeVar（参数类型不匹配） #

这里的错误是将ModelType直接作为参数类型，但ModelType绑定的是BaseModel的实例类型，而我们传入的是模型类（比如Point是类对象，不是实例），因此类型检查工具会标记参数不匹配。

from typing import TypeVar

ModelType = TypeVar("ModelType", bound=BaseModel)

def make_model_instance(model: ModelType, **kwargs) -> ModelType:
    return model(**kwargs)  # 此处也会报错，因为model被当成实例而非类

# 类型检查工具会标记model参数，期望传入BaseModel实例，实际传入的是类
point: Point = make_model_instance(model=Point, x=1, y=2)

方法三：使用type[ModelType]结合绑定的TypeVar（正确写法） #

这种写法能让类型检查工具正确推导类型，完全符合需求。

from typing import TypeVar

ModelType = TypeVar("ModelType", bound=BaseModel)

def make_model_instance(model: type[ModelType], **kwargs) -> ModelType:
    return model(**kwargs)

# 类型检查工具无报错，可正确推导返回值为Point类型
point: Point = make_model_instance(model=Point, x=1, y=2)

原理说明 #

• TypeVar("ModelType", bound=BaseModel)定义了一个类型变量ModelType，它只能是BaseModel或其子类的实例类型。
• type[ModelType]表示参数是ModelType对应的类对象（比如当ModelType是Point实例时，type[ModelType]就是Point类本身）。
• 函数返回ModelType，意味着返回的是传入类对应的实例。类型检查工具可以根据传入的model参数（如Point类），自动推导返回值为Point实例，和变量标注的Point类型完全匹配，因此不会报错。

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