diff --git a/libs/core/langchain_core/output_parsers/__init__.py b/libs/core/langchain_core/output_parsers/__init__.py index e896193cc15d8..7a75c66772938 100644 --- a/libs/core/langchain_core/output_parsers/__init__.py +++ b/libs/core/langchain_core/output_parsers/__init__.py @@ -24,6 +24,11 @@ MarkdownListOutputParser, NumberedListOutputParser, ) +from langchain_core.output_parsers.openai_tools import ( + JsonOutputKeyToolsParser, + JsonOutputToolsParser, + PydanticToolsParser, +) from langchain_core.output_parsers.pydantic import PydanticOutputParser from langchain_core.output_parsers.string import StrOutputParser from langchain_core.output_parsers.transform import ( @@ -47,4 +52,7 @@ "XMLOutputParser", "JsonOutputParser", "PydanticOutputParser", + "JsonOutputToolsParser", + "JsonOutputKeyToolsParser", + "PydanticToolsParser", ] diff --git a/libs/core/langchain_core/output_parsers/openai_tools.py b/libs/core/langchain_core/output_parsers/openai_tools.py index f79bac5e28369..370afe2195ac8 100644 --- a/libs/core/langchain_core/output_parsers/openai_tools.py +++ b/libs/core/langchain_core/output_parsers/openai_tools.py @@ -5,7 +5,7 @@ from langchain_core.exceptions import OutputParserException from langchain_core.messages import AIMessage, InvalidToolCall -from langchain_core.output_parsers import BaseCumulativeTransformOutputParser +from langchain_core.output_parsers.transform import BaseCumulativeTransformOutputParser from langchain_core.outputs import ChatGeneration, Generation from langchain_core.pydantic_v1 import BaseModel, ValidationError from langchain_core.utils.json import parse_partial_json diff --git a/libs/core/tests/unit_tests/output_parsers/test_imports.py b/libs/core/tests/unit_tests/output_parsers/test_imports.py index fb164d41403c3..40478bf75590e 100644 --- a/libs/core/tests/unit_tests/output_parsers/test_imports.py +++ b/libs/core/tests/unit_tests/output_parsers/test_imports.py @@ -15,6 +15,9 @@ "XMLOutputParser", "JsonOutputParser", "PydanticOutputParser", + "JsonOutputToolsParser", + "JsonOutputKeyToolsParser", + "PydanticToolsParser", ]