rag_demo / prompts.py
valeriylo's picture
Upload prompts.py
bb6ef89
# 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"]
)