| # Copyright (c) OpenMMLab. All rights reserved. | |
| def multi_turn_1_map_fn(example): | |
| messages = example['messages'] | |
| conversation = [] | |
| for msg in messages: | |
| conversation.append({ | |
| 'system': msg['toy_system'], | |
| 'input': msg['toy_input'], | |
| 'output': msg['toy_output'] | |
| }) | |
| return {'conversation': conversation} | |