Function _map_roles

llama_cpp/llama_chat_format.py:886–899

Map the message roles.

(
    messages: List[llama_types.ChatCompletionRequestMessage],
    role_map: Dict[str, str],
)

def _map_roles(
    messages: List[llama_types.ChatCompletionRequestMessage],
    role_map: Dict[str, str],
) -> List[Tuple[str, Optional[str]]]:
    """Map the message roles."""
    output: List[Tuple[str, Optional[str]]] = []
    for message in messages:
        role = message["role"]
        if role in role_map:
            content: str | None = (
                message["content"] if isinstance(message["content"], str) else None
            )
            output.append((role_map[role], content))
    return output


def _format_llama2(