| 24 | return prompt |
| 25 | |
| 26 | def cal_function(task_description,cal_function): |
| 27 | prompt = f"Please understand the {task_description} task and write the python code to generate a reward calculation function for {task_description} according to the {cal_function} reward function as an example." |
| 28 | cal_function = query(prompt) |
| 29 | return cal_function |
| 30 | |
| 31 | def verify_correction_function(task_description,reward_function,verify_correction_function): |
| 32 | prompt = f"Please understand the {task_description} task and {reward_function} reward function and generate a python code to verify correction function according to the {verify_correction_function} verify correction function as an example." |