# for magics

model_kwargs["prompt_templates"] = {
    "code": PromptTemplate.from_template(
        "{prompt}\n\nProduce output as source code only, "
        "with no text or explanation before or after it."
    ),
    "html": PromptTemplate.from_template(
        "{prompt}\n\nProduce output in HTML format only, "
        "with no markup before or afterward."
    ),
    "image": PromptTemplate.from_template(
        "{prompt}\n\nProduce output as an image only, "
        "with no text before or after it."
    ),
    "markdown": PromptTemplate.from_template(
        "{prompt}\n\nProduce output in markdown format only."
    ),
    "md": PromptTemplate.from_template(
        "{prompt}\n\nProduce output in markdown format only."
    ),
    "math": PromptTemplate.from_template(
        "{prompt}\n\nProduce output in LaTeX format only, "
        "with $$ at the beginning and end."
    ),
    "json": PromptTemplate.from_template(
        "{prompt}\n\nProduce output in JSON format only, "
        "with nothing before or after it."
    ),
    "text": PromptTemplate.from_template("{prompt}"),  # No customization
}
