def __call__()

in src/fmeval/transforms/common.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

    def __call__(self, record: Dict[str, Any]) -> Dict[str, Any]:
        """Augment the input record with LLM prompts and returns said record.

        :param record: The input record.
        :returns: The input record with prompts added in.
        """
        if self.placeholder_to_record_key is not None:
            placeholder_data_dict = {
                placeholder_key: record[self.placeholder_to_record_key[placeholder_key]]
                for placeholder_key in self.placeholder_to_record_key
            }
            for prompt_key in self.output_keys:
                record[prompt_key] = self.prompt_composer.compose(placeholder_data_dict=placeholder_data_dict)
        else:
            for input_key, prompt_key in zip(self.input_keys, self.output_keys):
                record[prompt_key] = self.prompt_composer.compose(record[input_key])
        return record