Function _format_llama2

llama_cpp/llama_chat_format.py:902–916

Format the prompt with the llama2 style.

(
    system_message: str, messages: List[Tuple[str, Optional[str]]], sep: str, sep2: str
)

def _format_llama2(
    system_message: str, messages: List[Tuple[str, Optional[str]]], sep: str, sep2: str
) -> str:
    """Format the prompt with the llama2 style."""
    seps = [sep, sep2]
    ret = system_message + sep
    for i, (role, message) in enumerate(messages):
        if system_message and i == 0:
            m = message or ""
            ret += m + seps[i % 2]
        elif message:
            ret += role + message + " " + seps[i % 2]
        else:
            ret += role + " "
    return ret


def _format_add_colon_single(