diff --git a/libs/core/langchain_core/language_models/__init__.py b/libs/core/langchain_core/language_models/__init__.py index 16fbe8a676f..4c2e597dddf 100644 --- a/libs/core/langchain_core/language_models/__init__.py +++ b/libs/core/langchain_core/language_models/__init__.py @@ -1,6 +1,7 @@ from langchain_core.language_models.base import ( BaseLanguageModel, LanguageModelInput, + LanguageModelOutput, get_tokenizer, ) from langchain_core.language_models.chat_models import BaseChatModel, SimpleChatModel @@ -14,4 +15,5 @@ __all__ = [ "LLM", "LanguageModelInput", "get_tokenizer", + "LanguageModelOutput", ] diff --git a/libs/core/tests/unit_tests/language_models/test_imports.py b/libs/core/tests/unit_tests/language_models/test_imports.py index 439f940177b..348111f8511 100644 --- a/libs/core/tests/unit_tests/language_models/test_imports.py +++ b/libs/core/tests/unit_tests/language_models/test_imports.py @@ -7,6 +7,7 @@ EXPECTED_ALL = [ "BaseLLM", "LLM", "LanguageModelInput", + "LanguageModelOutput", "get_tokenizer", ] diff --git a/libs/langchain/langchain/schema/cache.py b/libs/langchain/langchain/schema/cache.py index 7a510f06178..8407af18c83 100644 --- a/libs/langchain/langchain/schema/cache.py +++ b/libs/langchain/langchain/schema/cache.py @@ -1,3 +1,3 @@ -from langchain_core.caches import BaseCache +from langchain_core.caches import RETURN_VAL_TYPE, BaseCache -__all__ = ["BaseCache"] +__all__ = ["BaseCache", "RETURN_VAL_TYPE"] diff --git a/libs/langchain/langchain/schema/language_model.py b/libs/langchain/langchain/schema/language_model.py index e10ab8f92ff..63a0292bed8 100644 --- a/libs/langchain/langchain/schema/language_model.py +++ b/libs/langchain/langchain/schema/language_model.py @@ -1,4 +1,15 @@ -from langchain_core.language_models import BaseLanguageModel, get_tokenizer +from langchain_core.language_models import ( + BaseLanguageModel, + LanguageModelInput, + LanguageModelOutput, + get_tokenizer, +) from langchain_core.language_models.base import _get_token_ids_default_method -__all__ = ["get_tokenizer", "BaseLanguageModel", "_get_token_ids_default_method"] +__all__ = [ + "get_tokenizer", + "BaseLanguageModel", + "_get_token_ids_default_method", + "LanguageModelInput", + "LanguageModelOutput", +] diff --git a/libs/langchain/langchain/schema/messages.py b/libs/langchain/langchain/schema/messages.py index de1f3270d6d..2c42e6091e5 100644 --- a/libs/langchain/langchain/schema/messages.py +++ b/libs/langchain/langchain/schema/messages.py @@ -1,6 +1,7 @@ from langchain_core.messages import ( AIMessage, AIMessageChunk, + AnyMessage, BaseMessage, BaseMessageChunk, ChatMessage, @@ -46,4 +47,5 @@ __all__ = [ "_message_to_dict", "_message_from_dict", "message_to_dict", + "AnyMessage", ] diff --git a/libs/langchain/langchain/schema/output_parser.py b/libs/langchain/langchain/schema/output_parser.py index 60f9e488ec8..b0d8e03a7b9 100644 --- a/libs/langchain/langchain/schema/output_parser.py +++ b/libs/langchain/langchain/schema/output_parser.py @@ -7,6 +7,7 @@ from langchain_core.output_parsers import ( BaseTransformOutputParser, StrOutputParser, ) +from langchain_core.output_parsers.base import T # Backwards compatibility. NoOpOutputParser = StrOutputParser @@ -20,4 +21,5 @@ __all__ = [ "NoOpOutputParser", "StrOutputParser", "OutputParserException", + "T", ] diff --git a/libs/langchain/langchain/schema/runnable/base.py b/libs/langchain/langchain/schema/runnable/base.py index 2fc618ec639..c9bde22100f 100644 --- a/libs/langchain/langchain/schema/runnable/base.py +++ b/libs/langchain/langchain/schema/runnable/base.py @@ -1,4 +1,5 @@ from langchain_core.runnables.base import ( + Other, Runnable, RunnableBinding, RunnableBindingBase, @@ -6,16 +7,22 @@ from langchain_core.runnables.base import ( RunnableEachBase, RunnableGenerator, RunnableLambda, + RunnableLike, RunnableParallel, RunnableSequence, RunnableSerializable, coerce_to_runnable, ) +from langchain_core.runnables.utils import Input, Output # Backwards compatibility. RunnableMap = RunnableParallel __all__ = [ + "Input", + "Output", + "RunnableLike", + "Other", "Runnable", "RunnableSerializable", "RunnableSequence", diff --git a/libs/langchain/langchain/schema/runnable/history.py b/libs/langchain/langchain/schema/runnable/history.py index a7cba8f299c..423b192be68 100644 --- a/libs/langchain/langchain/schema/runnable/history.py +++ b/libs/langchain/langchain/schema/runnable/history.py @@ -1,3 +1,11 @@ -from langchain_core.runnables.history import RunnableWithMessageHistory +from langchain_core.runnables.history import ( + GetSessionHistoryCallable, + MessagesOrDictWithMessages, + RunnableWithMessageHistory, +) -__all__ = ["RunnableWithMessageHistory"] +__all__ = [ + "RunnableWithMessageHistory", + "GetSessionHistoryCallable", + "MessagesOrDictWithMessages", +] diff --git a/libs/langchain/langchain/schema/runnable/retry.py b/libs/langchain/langchain/schema/runnable/retry.py index 4e1f4dbcc2e..d146c8ec5c2 100644 --- a/libs/langchain/langchain/schema/runnable/retry.py +++ b/libs/langchain/langchain/schema/runnable/retry.py @@ -1,3 +1,3 @@ -from langchain_core.runnables.retry import RunnableRetry +from langchain_core.runnables.retry import RunnableRetry, U -__all__ = ["RunnableRetry"] +__all__ = ["RunnableRetry", "U"] diff --git a/libs/langchain/langchain/schema/runnable/utils.py b/libs/langchain/langchain/schema/runnable/utils.py index 35bb4365e5a..1a03a6afbc3 100644 --- a/libs/langchain/langchain/schema/runnable/utils.py +++ b/libs/langchain/langchain/schema/runnable/utils.py @@ -1,5 +1,7 @@ from langchain_core.runnables.utils import ( + Addable, AddableDict, + AnyConfigurableField, ConfigurableField, ConfigurableFieldMultiOption, ConfigurableFieldSingleOption, @@ -44,4 +46,6 @@ __all__ = [ "gather_with_concurrency", "Input", "Output", + "Addable", + "AnyConfigurableField", ] diff --git a/libs/langchain/langchain/schema/storage.py b/libs/langchain/langchain/schema/storage.py index 1e64829655b..70d6925a4fd 100644 --- a/libs/langchain/langchain/schema/storage.py +++ b/libs/langchain/langchain/schema/storage.py @@ -1,3 +1,3 @@ -from langchain_core.stores import BaseStore +from langchain_core.stores import BaseStore, K, V -__all__ = ["BaseStore"] +__all__ = ["BaseStore", "K", "V"] diff --git a/libs/langchain/langchain/schema/vectorstore.py b/libs/langchain/langchain/schema/vectorstore.py index 59a719b541e..776dbce00dc 100644 --- a/libs/langchain/langchain/schema/vectorstore.py +++ b/libs/langchain/langchain/schema/vectorstore.py @@ -1,3 +1,3 @@ -from langchain_core.vectorstores import VectorStore, VectorStoreRetriever +from langchain_core.vectorstores import VST, VectorStore, VectorStoreRetriever -__all__ = ["VectorStore", "VectorStoreRetriever"] +__all__ = ["VectorStore", "VectorStoreRetriever", "VST"] diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_base.py b/libs/langchain/tests/unit_tests/schema/runnable/test_base.py index 2f7c84ea5bf..4ee9ba32fed 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_base.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_base.py @@ -13,6 +13,10 @@ EXPECTED_ALL = [ "RunnableSequence", "RunnableSerializable", "coerce_to_runnable", + "Input", + "Output", + "Other", + "RunnableLike", ] diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_history.py b/libs/langchain/tests/unit_tests/schema/runnable/test_history.py index a2e23beb1ff..01937e4f71a 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_history.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_history.py @@ -1,6 +1,10 @@ from langchain.schema.runnable.history import __all__ -EXPECTED_ALL = ["RunnableWithMessageHistory"] +EXPECTED_ALL = [ + "RunnableWithMessageHistory", + "GetSessionHistoryCallable", + "MessagesOrDictWithMessages", +] def test_all_imports() -> None: diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_retry.py b/libs/langchain/tests/unit_tests/schema/runnable/test_retry.py index 581d997c251..5d7bf113e41 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_retry.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_retry.py @@ -1,6 +1,6 @@ from langchain.schema.runnable.retry import __all__ -EXPECTED_ALL = ["RunnableRetry"] +EXPECTED_ALL = ["RunnableRetry", "U"] def test_all_imports() -> None: diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py b/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py index b38a33676cb..d8115a4b664 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py @@ -22,6 +22,8 @@ EXPECTED_ALL = [ "indent_lines_after_first", "Input", "Output", + "Addable", + "AnyConfigurableField", ] diff --git a/libs/langchain/tests/unit_tests/schema/test_cache.py b/libs/langchain/tests/unit_tests/schema/test_cache.py index 73d5559a656..f7946dc848a 100644 --- a/libs/langchain/tests/unit_tests/schema/test_cache.py +++ b/libs/langchain/tests/unit_tests/schema/test_cache.py @@ -1,6 +1,6 @@ from langchain.schema.cache import __all__ -EXPECTED_ALL = ["BaseCache"] +EXPECTED_ALL = ["BaseCache", "RETURN_VAL_TYPE"] def test_all_imports() -> None: diff --git a/libs/langchain/tests/unit_tests/schema/test_language_model.py b/libs/langchain/tests/unit_tests/schema/test_language_model.py index 96ae48c2f56..85c5f24f196 100644 --- a/libs/langchain/tests/unit_tests/schema/test_language_model.py +++ b/libs/langchain/tests/unit_tests/schema/test_language_model.py @@ -1,6 +1,12 @@ from langchain.schema.language_model import __all__ -EXPECTED_ALL = ["BaseLanguageModel", "_get_token_ids_default_method", "get_tokenizer"] +EXPECTED_ALL = [ + "BaseLanguageModel", + "_get_token_ids_default_method", + "get_tokenizer", + "LanguageModelOutput", + "LanguageModelInput", +] def test_all_imports() -> None: diff --git a/libs/langchain/tests/unit_tests/schema/test_messages.py b/libs/langchain/tests/unit_tests/schema/test_messages.py index 3c13488cace..37b6a189e64 100644 --- a/libs/langchain/tests/unit_tests/schema/test_messages.py +++ b/libs/langchain/tests/unit_tests/schema/test_messages.py @@ -22,6 +22,7 @@ EXPECTED_ALL = [ "merge_content", "messages_from_dict", "messages_to_dict", + "AnyMessage", ] diff --git a/libs/langchain/tests/unit_tests/schema/test_output_parser.py b/libs/langchain/tests/unit_tests/schema/test_output_parser.py index d45206f71ab..68316d5a643 100644 --- a/libs/langchain/tests/unit_tests/schema/test_output_parser.py +++ b/libs/langchain/tests/unit_tests/schema/test_output_parser.py @@ -9,6 +9,7 @@ EXPECTED_ALL = [ "NoOpOutputParser", "OutputParserException", "StrOutputParser", + "T", ] diff --git a/libs/langchain/tests/unit_tests/schema/test_storage.py b/libs/langchain/tests/unit_tests/schema/test_storage.py index 91196a2b00c..89b4b0b1818 100644 --- a/libs/langchain/tests/unit_tests/schema/test_storage.py +++ b/libs/langchain/tests/unit_tests/schema/test_storage.py @@ -1,6 +1,6 @@ from langchain.schema.storage import __all__ -EXPECTED_ALL = ["BaseStore"] +EXPECTED_ALL = ["BaseStore", "K", "V"] def test_all_imports() -> None: diff --git a/libs/langchain/tests/unit_tests/schema/test_vectorstore.py b/libs/langchain/tests/unit_tests/schema/test_vectorstore.py index 3e53f84ba1a..adf82506ea2 100644 --- a/libs/langchain/tests/unit_tests/schema/test_vectorstore.py +++ b/libs/langchain/tests/unit_tests/schema/test_vectorstore.py @@ -1,6 +1,6 @@ from langchain.schema.vectorstore import __all__ -EXPECTED_ALL = ["VectorStore", "VectorStoreRetriever"] +EXPECTED_ALL = ["VectorStore", "VectorStoreRetriever", "VST"] def test_all_imports() -> None: