diff --git a/libs/core/langchain_core/output_parsers/__init__.py b/libs/core/langchain_core/output_parsers/__init__.py index e896193cc15..7a75c667729 100644 --- a/libs/core/langchain_core/output_parsers/__init__.py +++ b/libs/core/langchain_core/output_parsers/__init__.py @@ -24,6 +24,11 @@ from langchain_core.output_parsers.list import ( 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 @@ __all__ = [ "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 f79bac5e283..370afe2195a 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 typing import Any, Dict, List, Optional, Type 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 fb164d41403..40478bf7559 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 @@ EXPECTED_ALL = [ "XMLOutputParser", "JsonOutputParser", "PydanticOutputParser", + "JsonOutputToolsParser", + "JsonOutputKeyToolsParser", + "PydanticToolsParser", ]