如何在Python中为重载多方法实现正确的Intellisense？

以下代码可正常运行：

from multimethod import multimethod

@multimethod
def foo(a: int):
    print(f"only {a}")

@multimethod
def foo(a: int, b: int):
    print(f"{a} and {b}")

foo(1)
foo(1, 2)

运行输出：

only 1
1 and 2

但VSCode的IntelliSense无法正确显示该函数的多个参数列表（原因是multimethod装饰器需要接收任意参数来选择重载函数），当前显示效果：

期望实现的效果是显示多个参数列表：

请问该如何实现？

要让VSCode的IntelliSense识别多方法的重载签名，需要结合typing.overload装饰器显式声明所有重载版本，再用multimethod实现实际逻辑：

from multimethod import multimethod
from typing import overload

# 用@overload声明所有重载签名，供IntelliSense识别
@overload
def foo(a: int) -> None:
    ...

@overload
def foo(a: int, b: int) -> None:
    ...

# 用multimethod实现实际的多方法分发逻辑
@multimethod
def foo(a: int):
    print(f"only {a}")

@multimethod
def foo(a: int, b: int):
    print(f"{a} and {b}")

foo(1)
foo(1, 2)

这样VSCode默认的Pyright语言服务会读取@overload声明的所有签名，在IntelliSense中展示多个参数列表，同时multimethod依然能在运行时正常处理重载分发。

另外需确保VSCode的Python插件使用Pyright作为语言服务（默认已启用），可在设置中确认python.analysis.typeCheckingMode设为basic或strict。

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