distillers/raw_prompt_generator.py

import random 
import json
class RawPromptGenerator():
    def __init__(self,args=None):
        self.args = args
        pass 
    
    def generate_from_file(self, file_path, choice_num=1):
        with open(file_path, 'r') as infile:
            data = json.load(infile)
        result = []
        for my_data in data[0]: 
            result.append({'question': my_data['question'], 'answer': my_data['answer']})
        selected_index = random.sample(range(len(result)), choice_num) 
        selected_result = [result[index] for index in selected_index]
        return selected_result