diff --git a/src/xai_sdk/types/model.py b/src/xai_sdk/types/model.py index ee42198..4099e17 100644 --- a/src/xai_sdk/types/model.py +++ b/src/xai_sdk/types/model.py @@ -42,9 +42,13 @@ ImageGenerationModel: TypeAlias = Literal[ "grok-imagine-image", "grok-imagine-image-pro", + "grok-imagine-image-quality", ] -VideoGenerationModel: TypeAlias = Literal["grok-imagine-video"] +VideoGenerationModel: TypeAlias = Literal[ + "grok-imagine-video", + "grok-imagine-video-1.5-preview", +] AllModels: TypeAlias = Union[ ChatModel,