Can typing.Literal be used with custom type aliases in Claudetools

python
from typing import Literal

class Color(str):
    RED = "red"
    GREEN = "green"
    BLUE = "blue"

ColorLiteral = Literal[Color.RED, Color.GREEN, Color.BLUE]

def print_color(color: ColorLiteral):
    print(f"The color is: {color}")

print_color(Color.RED)   # OK
print_color(Color.GREEN) # OK 
print_color(Color.BLUE)  # OK

print_color("yellow")    # Error: Argument of type "Literal['yellow']" is not compatible with parameter of type "ColorLiteral"

In this example:

1. We define a custom `Color` class that inherits from `str` and has class attributes for the different colors.

2. We create a custom type alias `ColorLiteral` using `typing.Literal` and the class attributes of `Color`.

3. We define a function `print_color` that takes a parameter of type `ColorLiteral`.

4. We can call `print_color` with the class attributes of `Color` as arguments, since they are compatible with the `ColorLiteral` type.

5. If we try to call `print_color` with a string that is not one of the defined colors, the type checker will raise an error.

By using `typing.Literal` with custom type aliases, you can create more specific and expressive types for your function parameters in Claudetools. This helps ensure that only valid values are passed to the functions, and provides better type safety and documentation for your code.

The key points are:

- Define a custom class or type alias
- Use `typing.Literal` with the valid values of the custom type
- Use the custom literal type for your function parameters
- The type checker will enforce the valid values at call sites

This approach allows you to leverage the power of `typing.Literal` with your own domain-specific types in Claudetools.