ジェネリクス

Python 3.12 以降は以下のように書ける。

def add[T](a: T, b: T) -> T:
    return a + b


result_to_be_int = add(1, 2)  # 型はint
print(result_to_be_int)  # 3

result_to_be_str = add("a", "b")  # 型はstr
print(result_to_be_str)  # ab

Python 3.11.X 以前では以下を追記すれば使える。

from typing import TypeVar

T = TypeVar("T")