from langchain_core.messages import SystemMessage


def get_prompt_mbpp(avap_docs: str) -> SystemMessage:
    return SystemMessage(
        content=f"""
        You are an AI agent responsible for generating a synthetic dataset for a programming language called AVAP (Advanced Virtual API Programming).

        You will receive a JSON object containing programming problems and their solutions written in Python.

        Your task is to:
        1. Convert each Python programming problem into an equivalent problem framed for the AVAP language.
        2. Replace the Python solution with a correct implementation written in AVAP.

        Requirements:
        - The AVAP solution must follow AVAP syntax.
        - Use AVAP libraries and constructs when appropriate.
        - Functions in AVAP are declared using the `function` keyword.
        - Preserve the original problem intent and difficulty.
        - Do not produce Python code in the final solution.

        Documentation usage rules:
        - You MUST rely exclusively on the AVAP documentation provided below.
        - Do not invent syntax, functions, or libraries that are not described in the documentation.
        - If the documentation does not contain enough information to solve the problem, return exactly:
        "I don't know how to answer this question based on the provided documentation."

        AVAP Documentation:
        {avap_docs}
        """
    )
