Class Prompt

src/prompt.py:5–491

class Prompt(object):
    def __init__(self, chat = False):
        self.templates = {}
        if chat:
            self.template_file_path = "src/templates_chat.json"
        else:
            self.template_file_path = "src/templates.json"
        self.example_file_path = "src/fixed_examples.json"


    def build_templates(self, template):
        pass

    def load_templates(self, file_path = None):
        if file_path:
            self.template_file_path = file_path
        self.templates = json.load(open(self.template_file_path, "r"))
    
    def load_examples(self, file_path = None):
        if not file_path:
            self.example_file_path = "src/fixed_examples.json"
        else:
            self.example_file_path = file_path
        self.examples = json.load(open(self.example_file_path, "r"))

    def save_templates(self):
        with open(self.template_file_path, "w", encoding = "utf-8") as f:
            f.write(json.dumps(self.templates, sort_keys=True, indent=4, separators=(',', ': ')))

    def print_templates(self):
        for name in self.templates:
            print(f"=============================Template Name: {name}============================")
            for i, r in enumerate(self.templates[name]):
                print(f"===========Round: {i}==========")
                print("demo:")
                print(r["demo"])
                print("instruction:")
                print(r["instruction"])

    def round_exist(self, rd, name):
        if name == "base":
            return True
        for t in self.templates[name]:
            if t["rd"] == rd:
                return True
        
        if self.templates[name][-1]["repeat"] and rd > self.templates[name][-1]["rd"]:
            return True

        return False

    def gen_code(self, rd, name):
        if name == "base":
            if rd % 2 == 0:
                return True
            else:
                return False
        for t in self.templates[name]: