Source code for llmp.components.instruction.generation
from llmp.components.base import BaseInstructionHandler
from llmp.components.instruction.utils import mutate_instruction, generate_instruction_from_models, instruction_from_working_out
[docs]class InstructionGenerator(BaseInstructionHandler):
[docs] def run(self):
"""Run the instruction generation process."""
if len(self.job.example_records) > 0:
if self.debug:
print("Running instruction generation from working out.")
return instruction_from_working_out(
input_object=self.job.example_records[0].input,
output_object=self.job.example_records[0].output,
debug=self.debug,
)
else:
if self.debug:
print("Running instruction generation from scratch.")
return generate_instruction_from_models(
input_model=self.job.input_model,
output_model=self.job.output_model,
debug=self.debug,
)