github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-19 05:13:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: Fix defusedxml import (#26718)

Browse Source 
Fix defusedxml import. Haven't investigated what's actually going on
under the hood -- defusedxml probably does some weird things in __init__
This commit is contained in:
Eugene Yurtsev 2024-09-20 16:53:24 -04:00 committed by GitHub
parent 79b224f6f3
commit 4fc69d61ad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/core/langchain_core/output_parsers/xml.py
View File

@ -189,7 +189,7 @@ class XMLOutputParser(BaseTransformOutputParser):
# likely if you're reading this you can move them to the top of the file
if self.parser == "defusedxml":
try:
import defusedxml # type: ignore
from defusedxml import ElementTree # type: ignore
except ImportError as e:
raise ImportError(
"defusedxml is not installed. "
@ -197,7 +197,7 @@ class XMLOutputParser(BaseTransformOutputParser):
"You can install it with `pip install defusedxml`"
"See https://github.com/tiran/defusedxml for more details"
) from e
_et = defusedxml.ElementTree # Use the defusedxml parser
_et = ElementTree # Use the defusedxml parser
else:
_et = ET # Use the standard library parser
@ -211,10 +211,9 @@ class XMLOutputParser(BaseTransformOutputParser):
text = text.strip()
try:
root = ET.fromstring(text)
root = _et.fromstring(text)
return self._root_to_dict(root)
except ET.ParseError as e:
except _et.ParseError as e:
msg = f"Failed to parse XML format from completion {text}. Got: {e}"
raise OutputParserException(msg, llm_output=text) from e