Function map_role

scripts/data_getsql.py:36–42

(index: int)

        messages = [message["content"] for message in prompt_messages]

        def map_role(index: int):
            if index == 0:
                return "system"
            elif index % 2 == 1:
                return "user"
            else:
                return "assistant"

        convo_messages = [{"role": map_role(i), "content": message} for i, message in enumerate(messages) if "exit code: 1" not in message]