Skip to content
Snippets Groups Projects
utils.py 551 B
Newer Older
Jannis Vamvas's avatar
Jannis Vamvas committed
from typing import List
Jannis Vamvas's avatar
Jannis Vamvas committed


Jannis Vamvas's avatar
Jannis Vamvas committed
def format_as_chat(message: str, history: List[List[str]]) -> str:
Jannis Vamvas's avatar
Jannis Vamvas committed
    """
Jannis Vamvas's avatar
Jannis Vamvas committed
    Given a message and a history of previous messages, returns a string that formats the conversation as a chat.
Jannis Vamvas's avatar
Jannis Vamvas committed
    Uses the format expected by Meta Llama 3 Instruct.
Jannis Vamvas's avatar
Jannis Vamvas committed

    :param message: A string containing the user's most recent message
Jannis Vamvas's avatar
Jannis Vamvas committed
    :param history: A list of lists of previous messages, where each sublist is a conversation turn:
Jannis Vamvas's avatar
Jannis Vamvas committed
        [[user_message1, assistant_reply1], [user_message2, assistant_reply2], ...]
Jannis Vamvas's avatar
Jannis Vamvas committed
    """
    ...  # TODO