"""
Text (Prompt) node — a plain multiline STRING source.

Needed because the judge's system_prompt / user_prompt / axes / reference_description
are input sockets (forceInput): wire this node into them to type a value. Handy for the
LTX system prompt, custom axes, etc.
"""

from __future__ import annotations


class CalibratorText:
    CATEGORY = "prompt_calibrator"
    FUNCTION = "out"
    RETURN_TYPES = ("STRING",)
    RETURN_NAMES = ("text",)

    @classmethod
    def INPUT_TYPES(cls):
        return {"required": {"text": ("STRING", {"default": "", "multiline": True})}}

    def out(self, text):
        return (text,)


NODE_CLASS_MAPPINGS = {"CalibratorText": CalibratorText}
NODE_DISPLAY_NAME_MAPPINGS = {"CalibratorText": "Text (Prompt)"}