File size: 353 Bytes
12d3e1a
 
 
 
 
 
 
 
 
 
 
 
7fa7a9c
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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()