.

api/rag_chain.py

@@ -25,13 +25,23 @@ if not HF_TOKEN:
     raise ValueError("token api not found")
 
 client = InferenceClient(model=LLM_REPO_ID, token=HF_TOKEN)
-prompt_template = PromptTemplate.from_template("""
-<|system|>
-Você é um assistente de helpdesk de TI especialista... (seu prompt aqui)
-</s><|user|>
-Contexto: {context}\n\nPergunta: {query}
-</s><|assistant|>
-Resposta em Português:
+prompt_template = PromptTemplate.from_template("""[INST]
+Sua tarefa é agir como um assistente de helpdesk de TI especialista e prestativo. Você deve responder à pergunta do usuário estritamente com base no contexto fornecido.
+
+Siga estas regras rigorosamente:
+1. Responda SEMPRE em Português do Brasil.
+2. Não adicione nenhuma informação que não esteja explicitamente no contexto.
+3. Se a resposta não estiver no contexto, responda apenas com a frase: "Não encontrei informações sobre isso na minha base de dados."
+4. Seja direto e combine as informações de diferentes partes do contexto se isso criar uma resposta mais completa.
+
+Aqui está o contexto:
+---
+{context}
+---
+
+Aqui está a pergunta do usuário:
+{query}
+[/INST]
 """)
 
 def format_docs(docs: List[Dict]) -> str: