Lint

langchain-ai · Dec 21, 2023 · 314081d · 314081d
1 parent 35094f6
commit 314081d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/libs/langchain/langchain/output_parsers/xml.py b/libs/langchain/langchain/output_parsers/xml.py
@@ -37,7 +37,7 @@ def parse(self, text: str) -> Dict[str, List[Any]]:
 
     def _transform(
         self, input: Iterator[Union[str, BaseMessage]]
-    ) -> Iterator[AddableDict[str, Any]]:
+    ) -> Iterator[AddableDict]:
         parser = ET.XMLPullParser(["start", "end"])
         current_path: List[str] = []
         current_path_has_children = False
@@ -69,7 +69,7 @@ def _transform(
 
     async def _atransform(
         self, input: AsyncIterator[Union[str, BaseMessage]]
-    ) -> AsyncIterator[AddableDict[str, Any]]:
+    ) -> AsyncIterator[AddableDict]:
         parser = ET.XMLPullParser(["start", "end"])
         current_path: List[str] = []
         current_path_has_children = False

diff --git a/libs/langchain/tests/unit_tests/output_parsers/test_xml_parser.py b/libs/langchain/tests/unit_tests/output_parsers/test_xml_parser.py
@@ -31,7 +31,7 @@ def test_xml_output_parser(result: str) -> None:
 
     xml_result = xml_parser.parse(result)
     assert DEF_RESULT_EXPECTED == xml_result
-    assert list(xml_parser.transform(result)) == [
+    assert list(xml_parser.transform(iter(result))) == [
         {"foo": [{"bar": [{"baz": None}]}]},
         {"foo": [{"bar": [{"baz": "slim.shady"}]}]},
         {"foo": [{"baz": "tag"}]},