class RunwareTextModelSearch:
    """Text model search node: pick or type a text/chat AIR model for Runware Text Inference."""

    TEXT_MODELS = {
        "MiniMax": [
            "minimax:m2.7@0 (MiniMax-M2.7)",
            "minimax:m2.7@highspeed (MiniMax-M2.7 Highspeed)",
            "minimax:m2.5@0 (MiniMax-M2.5)",
        ],
        "OpenAI": [
            "openai:gpt@5.4 (GPT-5.4)",
            "openai:gpt@5.4-pro (GPT-5.4 Pro)",
            "openai:gpt@5.4-mini (GPT-5.4 Mini)",
            "openai:gpt@5.4-nano (GPT-5.4 Nano)",
        ],
        "Anthropic": [
            "anthropic:claude@opus-4.7 (Claude Opus 4.7)",
            "anthropic:claude@sonnet-4.6 (Claude Sonnet 4.6)",
            "anthropic:claude@haiku-4.5 (Claude Haiku 4.5)",
        ],
        "Google": [
            "google:gemini@3-flash (Gemini 3 Flash)",
        ],
        "MoonshotAI": [
            "moonshotai:kimi@k2.6 (Kimi K2.6)",
        ],
        "Zai": [
            "zai-glm-5-1 (ZAI GLM 5.1)",
        ],
    }

    MODEL_PROVIDERS = [
        "All",
        "Anthropic",
        "Google",
        "MiniMax",
        "MoonshotAI",
        "OpenAI",
        "Zai",
    ]

    @classmethod
    def INPUT_TYPES(cls):
        all_models = cls._get_all_models()
        default_model = "minimax:m2.7@0 (MiniMax-M2.7)"

        return {
            "required": {
                "Model Search": ("STRING", {
                    "tooltip": "Search or type a text model AIR code (e.g. minimax:m2.7@0). Used when 'Use Search Value' is enabled.",
                }),
                "Model Provider": (cls.MODEL_PROVIDERS, {
                    "tooltip": "Filter the model list by provider.",
                    "default": "MiniMax",
                }),
                "TextList": (all_models, {
                    "tooltip": "Choose a model from the filtered list (AIR code is taken from text before ' (').",
                    "default": default_model,
                }),
                "Use Search Value": ("BOOLEAN", {
                    "tooltip": "When enabled, 'Model Search' is used as the model AIR instead of the selection in TextList.",
                    "default": False,
                    "label_on": "Enabled",
                    "label_off": "Disabled",
                }),
            }
        }

    @classmethod
    def _get_all_models(cls):
        all_models = []
        for models in cls.TEXT_MODELS.values():
            all_models.extend(models)
        return all_models

    RETURN_TYPES = ("RUNWARETEXTMODEL",)
    RETURN_NAMES = ("model",)
    FUNCTION = "searchModels"
    CATEGORY = "Runware"
    DESCRIPTION = "Search and select text/chat models for Runware Text Inference."

    @classmethod
    def VALIDATE_INPUTS(cls, TextList):
        return True

    def searchModels(self, **kwargs):
        use_search = kwargs.get("Use Search Value", False)
        search_input = (kwargs.get("Model Search") or "").strip()

        if use_search:
            if not search_input:
                raise Exception("Use Search Value is enabled but Model Search is empty.")
            model_air = search_input
        else:
            current = kwargs.get("TextList")
            model_air = current.split(" (")[0] if current else ""

        return (model_air,)