Как набирать словари подсказок, которые могут иметь разные пользовательские ключи и/или значения?

В предыдущих версиях нашего приложения люди просто передавали некоторые аргументы в виде простых строк определенным функциям, поскольку для некоторых из них у нас не было конкретных подсказок типов или типов данных. Что-то вроде:

# Hidden function signature:
def dummy(var: str):
    pass

# Users:
dummy("cat")

Но теперь мы хотим реализовать собственные типы данных для этих сигнатур функций, обеспечивая при этом обратную совместимость. Скажите что-нибудь вроде этого:

# Signature:
def dummy(var: Union[NewDataType, Literal["cat"]])

# Backward compatibility:
dummy("cat")

# New feature:
dummy(NewDataType.cat)

Достичь этого для простых сигнатур функций — это нормально, но проблема возникает, когда сигнатуры более сложные.

Как это реализовать, если аргумент dummy — это словарь, который может принимать как Literal["cat"], так и NewDataType в качестве ключей? Кроме того, как этого добиться, если аргументом является словарь с той же самой предыдущей комбинацией типов ключей, но он также может иметь значения str и int (и четыре возможные комбинации)? Все это должно быть совместимо с mypy, pylint и использовать Python 3.9 (без StrEnum или TypeAlias).

Я пробовал много разных комбинаций, например следующие:

from typing import TypedDict, Literal, Dict, Union
from enum import Enum

# For old support:
AnimalsLiteral = Literal[
    "cat",
    "dog",
    "snake",
]

# New datatypes:
class Animals(Enum):
    cat = "cat"
    dog = "dog"
    snake = "snake"

# Union of Animals Enum and Literal types for full support:
DataType = Union[Animals, AnimalsLiteral]

# option 1, which fails:
def dummy(a: Dict[DataType, str]):
    pass

# option 2, which also fails:
# def dummy(a: Union[Dict[DataType, str], Dict[Animals, str], Dict[AnimalsLiteral, str]]):
#    pass

if __name__ == "__main__":
    # Dictionary with keys as Animals Enum
    input_data1 = {
        Animals.dog: "dog",
    }
    dummy(input_data1)

    # Dictionary with keys as Literal["cat", "dog", "snake"]
    input_data2 = {
        "dog": "dog",
    }
    dummy(input_data2)

    # Dictionary with mixed keys: Animals Enum and Literal string
    input_data3 = {
        Animals.dog: "dog",
        "dog": "dog",
    }
    dummy(input_data3)

dummy(input_data1) в порядке, но dummy(input_data2) выдает следующие ошибки mypy с сигнатурой 2 для манекена:

Argument 1 to "dummy" has incompatible type "dict[str, str]"; expected "Union[dict[Union[Animals, Literal['cat', 'dog', 'snake']], str], dict[Animals, str], dict[Literal['cat', 'dog', 'snake'], str]]"Mypyarg-type

Argument 1 to "dummy" has incompatible type "dict[str, str]"; expected "Union[dict[Union[Animals, Literal['cat', 'dog', 'snake']], str], dict[Animals, str], dict[Literal['cat', 'dog', 'snake'], str]]"Mypyarg-type

(variable) input_data2: dict[str, str]

Конечно, делаю что-то вроде:

input_data2: DataTypes = {
    "dog": "dog",
}

могло бы решить эту проблему, но я не могу просить пользователей всегда делать это при создании своих типов данных.

Кроме того, я попробовал другую альтернативу, используя TypedDict, но все равно сталкиваюсь с теми же ошибками mypy.

В конце концов, я хочу иметь возможность создавать подсказки типов словарей, совместимые с mypy и pylint, которые могут принимать пользовательские типы ключей (как в примере) и даже пользовательские типы значений или комбинацию вышеперечисленного.