mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
7ad1168062
* support NPHardEval * add .md file and fix minor bugs * refactor and minor fix --------- Co-authored-by: Leymore <zfz-960727@163.com>
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
import ast
|
|
import xml.etree.ElementTree as ET
|
|
|
|
|
|
def append_root_tags(string):
|
|
if not string.strip().startswith('<root>'):
|
|
string = '<root>\n' + string
|
|
if not string.strip().endswith('</root>'):
|
|
string += '\n</root>'
|
|
return string
|
|
|
|
|
|
def parse_xml_to_dict(xml_string):
|
|
final_answer_element = ''
|
|
reasoning_element = ''
|
|
|
|
try:
|
|
# Parse the XML string
|
|
root = ET.fromstring(xml_string)
|
|
|
|
# Find the 'final_answer' tag
|
|
final_answer_element = root.find('final_answer').text
|
|
|
|
# Find the 'reasoning' tag
|
|
reasoning_element = root.find('reasoning').text
|
|
except Exception:
|
|
try:
|
|
assert '<final_answer>' in xml_string
|
|
assert '</final_answer>' in xml_string
|
|
assert '<reasoning>' in xml_string
|
|
assert '</reasoning>' in xml_string
|
|
final_answer_start = xml_string.index('<final_answer>') + len('<final_answer>')
|
|
final_answer_end = xml_string.index('</final_answer>')
|
|
reasoning_start = xml_string.index('<reasoning>') + len('<reasoning>')
|
|
reasoning_end = xml_string.index('</reasoning>')
|
|
final_answer_element = xml_string[final_answer_start:final_answer_end]
|
|
reasoning_element = xml_string[reasoning_start:reasoning_end]
|
|
except Exception:
|
|
final_answer_element = ''
|
|
reasoning_element = ''
|
|
|
|
final_answer_element = ast.literal_eval(final_answer_element.strip())
|
|
return final_answer_element, reasoning_element
|