from setup.easy_imports import ChatPromptTemplate


class Prompt:
    def __init__(self):
        pass

    def create_prompt_template(self, system_prompt, user_prompt):
        prompt_template = ChatPromptTemplate.from_messages(
            [("system", system_prompt), ("user", user_prompt)]
        )
        return prompt_template

prompt = Prompt()