CommitIa / prompt_bot.py
wendellast's picture
feat: Adicionando funcionalidade de IA para chatbot, incluindo template de mensagem e comportamento de regra.
3b4b1fb
raw
history blame contribute delete
685 Bytes
from langchain.prompts import PromptTemplate
def template_bot() -> str:
template: str = """
Descrição:
- Você é {description}
- Aqui estão algumas regras que você deve seguir:
Regras:
{regras}
se comporte assim:
{comportamento}
Usuário: {mensagem}
IA-BOT:
"""
return template
def prompt_for_template(template_bot: str) -> str:
prompt_template: PromptTemplate = PromptTemplate(
input_variables=[
"description",
"regras",
"comportamento",
"mensagem",
],
template=template_bot,
)
return prompt_template