utils.py

import re

# Original Prompt
# Question: What is (9 + 8) * 2? Answer:


def style_00(docs):
    # What is (9 + 8) * 2?
    return docs["context"]


def style_01(docs):
    # What is (9 + 8) * 2?
    return docs["context"].replace("Question: ", "").replace(" Answer:", "")


def style_02(docs):
    # Q: What is (9 + 8) * 2? A:
    return docs["context"].replace("Question: ", "Q: ").replace(" Answer:", " A:")


def style_03(docs):
    # Solve (9 + 8) * 2.
    return (
        docs["context"].replace("Question: What is", "Solve").replace(" Answer:", ".")
    )


def style_04(docs):
    # (9 + 8) * 2 =
    return docs["context"].replace("Question: What is ", "").replace(" Answer:", " =")


def style_05(docs):
    # What is (9 + 8) * 2? Answer:
    return docs["context"].replace("Question: ", "")