# flake8: noqa
from langchain.prompts.prompt import PromptTemplate

_template = """Учитывая приведенную историю общения и последующий вопрос, перефразируй последующий вопрос так, чтобы он стал обобщенным вопросом в языке оригинала.

История общения:
{chat_history}
Последующий вопрос: {question}
Обобщенный вопрос:"""
CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)

prompt_template = """Используй следующие фрагменты контекста, чтобы ответить на вопрос в конце. Если ты не знаешь ответа, просто скажи, что не знаешь, не пытайся придумать ответ.

{context}

Вопрос: {question}
Полезный ответ:"""
QA_PROMPT = PromptTemplate(
    template=prompt_template, input_variables=["context", "question"]
)