fmt

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",
 ]

libs/langchain/langchain/schema/runnable/base.py

@@ -1,4 +1,7 @@
 from langchain_core.runnables.base import (
+    Input,
+    Other,
+    Output,
     Runnable,
     RunnableBinding,
     RunnableBindingBase,
@@ -28,4 +31,7 @@ __all__ = [
     "RunnableBinding",
     "RunnableMap",
     "coerce_to_runnable",
+    "Input",
+    "Output",
+    "Other",
 ]

libs/langchain/tests/unit_tests/schema/runnable/test_base.py

@@ -13,6 +13,9 @@ EXPECTED_ALL = [
     "RunnableSequence",
     "RunnableSerializable",
     "coerce_to_runnable",
+    "Input",
+    "Output",
+    "Other",
 ]
 
 

libs/langchain/tests/unit_tests/schema/test_messages.py

@@ -3,6 +3,7 @@ from langchain.schema.messages import __all__
 EXPECTED_ALL = [
     "AIMessage",
     "AIMessageChunk",
+    "AnyMessage",
     "BaseMessage",
     "BaseMessageChunk",
     "ChatMessage",