Method get_single_query

workers/base.py:195–204

(self, datum, use_cot)

    def get_single_query(self, datum, use_cot):
        datum["option_str"] = "\n".join(
            [f"{k}. {v}" for k, v in datum["option"].items() if len(v) > 1]
        )
        if use_cot:
            query = self.query_prompt_1.format_map(datum)
        else:
            query = self.query_prompt_2.format_map(datum)

        return query
    
    def format_fewshot_user_and_gpt(self, item: dict, use_cot):
        user = self.get_single_query(item, use_cot)