incompatible_types.py

from typing import Literal, TypedDict

from enum import Enum


class Conn(str, Enum):
    TCP = "TCP"
    UDP = "UDP"


class TcpSettings(TypedDict):
    conn: Literal[Conn.TCP]
    foo: bool


t = TcpSettings(conn="TCP", foo=True)
# Incompatible types (expression has type "Literal['TCP']", TypedDict item "conn" has type "Literal[Conn.TCP]")mypy(error)

Possible solutions.

Use the enum for declaration

t = TcpSettings(conn= Conn.TCP, foo=True)

Use a str literal in the class

class TcpSettings(TypedDict):
    conn: Literal["TCP"]
    foo: bool

But is there a way to have both? An enum + letting users who use TcpSettings to avoid the enum by providing a str literal?