# AUTO-GENERATED. DO NOT EDIT.
# Generated by litai/utils/gen_supported_models.py
# Copyright The Lightning AI team.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Supported public model names and their type literal.

This file is auto-generated by `litai/utils/gen_supported_models.py`
from the authoritative `PUBLIC_MODELS` in `lightning_sdk.llm.public_assistants`.

It contains:
- `MODELS`: a tuple of all supported public model names.
- `ModelLiteral`: a `typing.Literal` of the same names, for autocompletion and type safety.

Do not edit this file by hand. Regenerate it instead.
"""

from typing import Final, Literal

MODELS: Final = (
    "anthropic/claude-3-5-sonnet-20240620",
    "google/gemini-2.5-flash",
    "google/gemini-2.5-flash-lite-preview-06-17",
    "google/gemini-2.5-pro",
    "lightning-ai/gpt-oss-120b",
    "lightning-ai/gpt-oss-20b",
    "openai/gpt-4",
    "openai/gpt-4o",
    "openai/gpt-5-2025-08-07",
    "openai/gpt-5-mini-2025-08-07",
    "openai/gpt-5-nano-2025-08-07",
    "openai/o3-mini",
)
ModelLiteral = Literal[
    "anthropic/claude-3-5-sonnet-20240620",
    "google/gemini-2.5-flash",
    "google/gemini-2.5-flash-lite-preview-06-17",
    "google/gemini-2.5-pro",
    "lightning-ai/gpt-oss-120b",
    "lightning-ai/gpt-oss-20b",
    "openai/gpt-4",
    "openai/gpt-4o",
    "openai/gpt-5-2025-08-07",
    "openai/gpt-5-mini-2025-08-07",
    "openai/gpt-5-nano-2025-08-07",
    "openai/o3-mini",
]
