diff --git a/src/helper/followup_question_gen.py b/src/helper/followup_question_gen.py index 95cd2db..ffebb1e 100644 --- a/src/helper/followup_question_gen.py +++ b/src/helper/followup_question_gen.py @@ -13,8 +13,9 @@ def __init__(self, model="meta-llama/Llama-3-70b-chat-hf", max_tokens=500): ) self.template = """Imagine yourself as the user, adept at generating follow-up questions based on the provided questions {user_question}. Generally, users inquire about the WaveMaker low-code platform and its technical aspects and Try to generate the questions based on the type of intent. - Aim to create 5 questions that will motivate the user to request a demo, with each question approximately 20 characters long. - Make sure only return the questions in the list with double inverted commas are returned, separated by commas within square brackets without adding any sentences in the beginning. + Aim to create "2 QUESTIONS" that will motivate the user to request a demo, with each question approximately 20 characters long. + Make sure that ONLY return the questions in the list with double inverted commas, separated by commas within square brackets without adding any sentences in the beginning. + I will be fired if you don't return a LIST with DOUBLE inverted commas, and if the list of questions is greater than 2. """ self.prompt = PromptTemplate( template=self.template,