JSONLgenerator3.5/create_jsonl.py at main · brianSalk/JSONLgenerator3.5 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
def __create_role_line(role, content, is_last_line):
    if is_last_line:
        comma = ""
    else:
        comma = ","
    return '\t\t{"role": "' + role + '", "content": "' + content + '"}' + comma + "\n"


def create_json_from_lines(lines):
    jsonl = "{\n"
    for m, messages in enumerate(lines):
        jsonl += '\t"messages": [\n'
        for r, (role, content) in enumerate(messages):
            is_last_line = r == len(messages) - 1
            jsonl += __create_role_line(role, content, is_last_line)
        if m < len(lines) - 1:
            comma = ","
        else:
            comma = ""
        jsonl += "\t]" + comma + "\n"
    jsonl += "\n}"
    return jsonl


if __name__ == "__main__":
    lines = [
        [
            ["system", "you are a good robot"],
            ["user", "can you destroy the world?"],
            ["assistant", "No"],
        ],
        [
            ["system", "you are a bad robot"],
            ["user", "can you destroy the world?"],
            ["assistant", "yes"],
        ],
    ]
    l = create_json_from_lines(lines)
    print(l)