Method generate_fewshot_examples

workers/base.py:239–249

Generate a fewshot prompt given a list of data. Note that the roles are already in the fewshot examples. Be careful not to add any extra roles in the final query that is input to an LLM.

(self, data: list[dict], use_cot=False)

        return user, gpt
    
    def generate_fewshot_examples(self, data: list[dict], use_cot=False):
        """
        Generate a fewshot prompt given a list of data.
        Note that the roles are already in the fewshot examples.
        Be careful not to add any extra roles in the final query that is input to an LLM.
        """
        prompt = self.fewshot_prompt
        for item in data:
            user, gpt = self.format_fewshot_user_and_gpt(item, use_cot)
            prompt += self.fewshot_template.format(user=user, gpt=gpt) + self.fewshot_separator
        return prompt