
def prompt_generator(question, system_prompt, with_system_prompt):
    if with_system_prompt==True:
        prompt = [
                    {"role": "system", "content": system_prompt},
                    {"role": "user", "content": question},
                 ]
    
        return prompt

    else:
        prompt = [
                    {"role": "user", "content": question},
                 ]
    
        return prompt
        
def finetune_prompt_generator(question, system_prompt, answer, with_system_prompt):
    if with_system_prompt==True:    
        prompt = [
                    {
                        "role": "system",
                        "content": system_prompt,
                    },
                    {
                        "role": "user",
                        "content": question,
                    },
                    {
                        "role": "assistant",
                        "content": answer,
                    },
                ]
        return prompt