Fixes for runpod test

This commit is contained in:
2024-11-16 21:51:18 +01:00
parent 9dfd5ddc39
commit c44d0ffbdf
8 changed files with 58 additions and 19 deletions

View File

@@ -62,8 +62,13 @@ class ResponseParser:
entry_id = str(uuid.uuid4())
timestamp = datetime.now()
parser = ET.XMLPullParser(events=("start", "end"))
parser.feed(xml)
root = None
try:
root = ET.fromstring(xml.strip())
for event, root in parser.read_events():
if event == "start":
break
except ET.ParseError as e:
return ParseErrorEntry(entry_id, timestamp, xml, f"Invalid XML: {str(e)}")