core[patch]: export chat prompt classes

2026-02-03 15:55:44 +00:00 · 2023-11-28 09:42:01 -08:00
3 changed files with 18 additions and 2 deletions
--- a/libs/core/langchain_core/prompts/init.py
+++ b/libs/core/langchain_core/prompts/init.py
@@ -28,9 +28,14 @@ from langchain_core.prompts.base import BasePromptTemplate, format_document
 from langchain_core.prompts.chat import (
    AIMessagePromptTemplate,
    BaseChatPromptTemplate,
+    BaseMessagePromptTemplate,
+    BaseStringMessagePromptTemplate,
    ChatMessagePromptTemplate,
    ChatPromptTemplate,
    HumanMessagePromptTemplate,
+    MessageLike,
+    MessageLikeRepresentation,
+    MessagePromptTemplateT,
    MessagesPlaceholder,
    SystemMessagePromptTemplate,
 )
@@ -53,14 +58,19 @@ from langchain_core.prompts.string import (
 __all__ = [
    "AIMessagePromptTemplate",
    "BaseChatPromptTemplate",
+    "BaseMessagePromptTemplate",
    "BasePromptTemplate",
+    "BaseStringMessagePromptTemplate",
    "ChatMessagePromptTemplate",
    "ChatPromptTemplate",
    "FewShotPromptTemplate",
    "FewShotPromptWithTemplates",
    "FewShotChatMessagePromptTemplate",
    "HumanMessagePromptTemplate",
+    "MessageLike",
+    "MessageLikeRepresentation",
    "MessagesPlaceholder",
+    "MessagePromptTemplateT",
    "PipelinePromptTemplate",
    "PromptTemplate",
    "StringPromptTemplate",
--- a/libs/core/langchain_core/runnables/base.py
+++ b/libs/core/langchain_core/runnables/base.py
@@ -2986,7 +2986,8 @@ class RunnableBinding(RunnableBindingBase[Input, Output]):

        .. code-block:: python

-            from langchain.schema.runnable import RunnableBinding
+            from langchain_core.runnables import RunnableBinding
+
            runnable_binding = RunnableBinding(
                bound=model,
                kwargs={'stop': ['-']} # <-- Note the additional kwargs
--- a/libs/core/tests/unit_tests/prompts/test_imports.py
+++ b/libs/core/tests/unit_tests/prompts/test_imports.py
@@ -3,20 +3,25 @@ from langchain_core.prompts import __all__
 EXPECTED_ALL = [
    "AIMessagePromptTemplate",
    "BaseChatPromptTemplate",
+    "BaseMessagePromptTemplate",
    "BasePromptTemplate",
+    "BaseStringMessagePromptTemplate",
    "ChatMessagePromptTemplate",
    "ChatPromptTemplate",
    "FewShotPromptTemplate",
    "FewShotPromptWithTemplates",
    "FewShotChatMessagePromptTemplate",
-    "format_document",
    "HumanMessagePromptTemplate",
+    "MessageLike",
+    "MessageLikeRepresentation",
    "MessagesPlaceholder",
+    "MessagePromptTemplateT",
    "PipelinePromptTemplate",
    "PromptTemplate",
    "StringPromptTemplate",
    "SystemMessagePromptTemplate",
    "load_prompt",
+    "format_document",
    "check_valid_template",
    "get_template_variables",
    "jinja2_formatter",