Function to_chat_text

tests/generate_inputs.py:7–15

(tokenizer: AutoTokenizer, prompt: str)

def to_chat_text(tokenizer: AutoTokenizer, prompt: str) -> str:
    messages = [{"role": "user", "content": prompt}]
    text = tokenizer.apply_chat_template(
        messages,
        tokenize=False,
        add_generation_prompt=True,
        enable_thinking=True,
    )
    return text


def main():