gradio-for-aiproqgen / convert_to_json.py
Saranath07
Application added
4d3ef13
raw
history blame contribute delete
789 Bytes
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