app/services/create_prompt.py

from langchain.prompts import PromptTemplate
from app.services.templates import templates


def create_prompt(template) -> PromptTemplate:
    prompt = PromptTemplate(template=template, input_variables=["context", "question"])
    return prompt


def create_prompt_emotion() -> PromptTemplate:
    prompt = PromptTemplate(
        input_variables=["style", "content"],
        template=templates[4],
    )
    return prompt