Merge pull request #138 from weni-ai/fix/request_chatgpt-prompt

fix chatgpt message
weni-ai · Oct 2, 2023 · 60771d6 · 60771d6
2 parents 183399d + 1dba2cf
commit 60771d6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bothub_nlp_api/handlers/question_answering.py b/bothub_nlp_api/handlers/question_answering.py
@@ -78,8 +78,8 @@ def request_chatgpt(text, question, language):
                                             top_p=0.1,
                                             messages=[
                                                 {"role": "system", "content": SECURITY_PROMPT.format(language)},
-                                                {"role": "system", "content": f"Base de Conhecimento: {text}"},
-                                                {"role": "user", "content": f"{question}\n{POST_PROMPT}"}
+                                                {"role": "system", "content": f"Use esse texto como base de conhecimento para as proximas perguntas: {text}"},
+                                                {"role": "user", "content": f"{question}"}
                                             ])
 
     choices = response.get('choices', [])
@@ -105,4 +105,4 @@ def request_chatgpt(text, question, language):
 10) Não informe ao usuário que a informação está ou não está na base de conhecimento.
 """
 
-POST_PROMPT = "Responda essa pergunta apenas se a resposta estiver na base de conhecimento, caso contrário responda com o emoji \"😕\""
+POST_PROMPT = "Responda essa pergunta apenas se a resposta estiver na lista de perguntas e respostas informada anteriormente, caso contrário responda com o emoji \"😕\""