Function _construct_prompt

python-impl/evogit/api.py:286–326

A helper function to construct the prompt. Read the code and the related information (e.g. stack_trace) from the a list of commits. Then feed into the prompt constructor. The operation_type is either "mutation" or "crossover". The prompt_constructor is a function with the following s

(config, commits, chunk_size, operation_type)

def _construct_prompt(config, commits, chunk_size, operation_type) -> str:
    """A helper function to construct the prompt.
    Read the code and the related information (e.g. stack_trace) from the a list of commits.
    Then feed into the prompt constructor.
    The operation_type is either "mutation" or "crossover".
    The prompt_constructor is a function with the following signature:
    def prompt_constructor(operation_type: str, code_infos: List[CodeInfo]) -> str
    """
    code_infos = []
    for commit in commits:
        code = git.read_file(config, commit)
        note = git.read_note(config, commit)
        if note is not None:
            note = json.loads(note)
            if note["timeout"]:
                timeout = True
                stack_trace = None
            else:
                timeout = False
                try:
                    stack_trace = json.loads(note["stdout"])["stack_trace"]
                except json.JSONDecodeError:
                    logger.warning(
                        f"Failed to parse the stack trace of {commit} with note: {note}"
                    )
                    stack_trace = None
                    import traceback

                    traceback.print_exc()

        code_infos.append(CodeInfo(code, stack_trace, timeout))

    prompts = []
    chunk = []
    for code_info in code_infos:
        chunk.append(code_info)
        if len(chunk) == chunk_size:
            prompts.append(config.prompt_constructor(operation_type, chunk))
            chunk = []

    return prompts


def llm_mutation(config, llm_backend, seeds, commits) -> list[str]: