File size: 789 Bytes
4d3ef13
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import json

"""
"""

def data_to_json(questions):
    questions_json = []

    for doc in questions:
        question_obj = {
            "question": doc.page_content,
            "question_template": doc.metadata.get("question_template"),
            "function_name": doc.metadata.get("function_name"),
            "data_formats": doc.metadata.get("data_formats", ""),
            "solution": doc.metadata.get("solution", ""),
            "tags": doc.metadata.get("tags", ""),
            "testcases": json.loads(doc.metadata.get("testcases", "[]")),
            "input_type": doc.metadata.get("input_type", ""),
        }
        questions_json.append(question_obj)

    # Convert the list to a JSON string
    json_output = json.dumps(questions_json, indent=2)

    return json_output