Shared via mypy Playground

main.py

from typing import Any, Literal, NotRequired, TypedDict, final

# @final
class InputPromptJSON(TypedDict):
    """JSON format representing the input text to an LLM completion.

    This may evolve over time as we add support for tools and other features.
    """

    # New rows should always use an array of blocks, even if the application code just gave a string.
    # Historical rows may still use a plain string.
    system_prompt: str
    type: Literal['I']


@final
class S3InputPromptJSON(TypedDict):
    """JSON format representing an s3 reference for the input text to an LLM completion."""

    s3_key: str
    type: Literal['S']
    
class Container:
    item: InputPromptJSON | S3InputPromptJSON
    
def f(c: Container) -> None:
    item: InputPromptJSON | S3InputPromptJSON = c.item
    # if item["type"] == 'S':
    if 's3_key' in item:
        print(item['s3_key'])