fix: fix qwen2 vl unstable output

aws-samples · Feb 28, 2025 · 46909fc · 46909fc
1 parent 96e97b7
commit 46909fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/source/lambda/job/dep/dist/llm_bot_dep-0.1.0-py3-none-any.whl b/source/lambda/job/dep/dist/llm_bot_dep-0.1.0-py3-none-any.whl
diff --git a/source/lambda/job/dep/llm_bot_dep/splitter_utils.py b/source/lambda/job/dep/llm_bot_dep/splitter_utils.py
@@ -142,7 +142,8 @@ def find_child(headers: dict, header_id: str):
 
 def parse_string_to_xml_node(xml_string):
     try:
-        xml_node = etree.fromstring(xml_string.replace("&", "&amp;"))
+        parser = etree.XMLParser(recover=True)
+        xml_node = etree.fromstring(xml_string.replace("&", "&amp;"), parser)
         return xml_node
     except etree.XMLSyntaxError as e:
         logger.error(f"Error parsing XML: {e}")