langchain-core 1.0.0a6__py3-none-any.whl → 1.0.0a8__py3-none-any.whl

langchain_core/runnables/graph_png.py

@@ -1,6 +1,6 @@
 """Helper class to draw a state graph into a PNG file."""
 
-from typing import Any, Optional
+from typing import Any
 
 from langchain_core.runnables.graph import Graph, LabelsDict
 
@@ -18,15 +18,14 @@ class PngDrawer:
     It requires ``graphviz`` and ``pygraphviz`` to be installed.
 
     Example:
-
-        .. code-block:: python
-
-            drawer = PngDrawer()
-            drawer.draw(state_graph, "graph.png")
+        ```python
+        drawer = PngDrawer()
+        drawer.draw(state_graph, "graph.png")
+        ```
     """
 
     def __init__(
-        self, fontname: Optional[str] = None, labels: Optional[LabelsDict] = None
+        self, fontname: str | None = None, labels: LabelsDict | None = None
     ) -> None:
         """Initializes the PNG drawer.
 
@@ -96,7 +95,7 @@ class PngDrawer:
         viz: Any,
         source: str,
         target: str,
-        label: Optional[str] = None,
+        label: str | None = None,
         conditional: bool = False,  # noqa: FBT001,FBT002
     ) -> None:
         """Adds an edge to the graph.
@@ -117,7 +116,7 @@ class PngDrawer:
             style="dotted" if conditional else "solid",
         )
 
-    def draw(self, graph: Graph, output_path: Optional[str] = None) -> Optional[bytes]:
+    def draw(self, graph: Graph, output_path: str | None = None) -> bytes | None:
         """Draw the given state graph into a PNG file.
 
         Requires `graphviz` and `pygraphviz` to be installed.

langchain_core/runnables/history.py

@@ -3,14 +3,11 @@
 from __future__ import annotations
 
 import inspect
-from collections.abc import Sequence
+from collections.abc import Callable, Sequence
 from types import GenericAlias
 from typing import (
     TYPE_CHECKING,
     Any,
-    Callable,
-    Optional,
-    Union,
 )
 
 from pydantic import BaseModel
@@ -34,7 +31,7 @@ if TYPE_CHECKING:
     from langchain_core.tracers.schemas import Run
 
 
-MessagesOrDictWithMessages = Union[Sequence["BaseMessage"], dict[str, Any]]
+MessagesOrDictWithMessages = Sequence["BaseMessage"] | dict[str, Any]
 GetSessionHistoryCallable = Callable[..., BaseChatMessageHistory]
 
 
@@ -74,8 +71,7 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
 
     Example: Chat message history with an in-memory implementation for testing.
 
-    .. code-block:: python
-
+        ```python
         from operator import itemgetter
 
         from langchain_openai.chat_models import ChatOpenAI
@@ -119,125 +115,124 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
         history.add_message(AIMessage(content="hello"))
         print(store)  # noqa: T201
 
+        ```
 
     Example where the wrapped Runnable takes a dictionary input:
 
-        .. code-block:: python
-
-            from typing import Optional
+        ```python
+        from typing import Optional
 
-            from langchain_community.chat_models import ChatAnthropic
-            from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
-            from langchain_core.runnables.history import RunnableWithMessageHistory
+        from langchain_community.chat_models import ChatAnthropic
+        from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+        from langchain_core.runnables.history import RunnableWithMessageHistory
 
 
-            prompt = ChatPromptTemplate.from_messages(
-                [
-                    ("system", "You're an assistant who's good at {ability}"),
-                    MessagesPlaceholder(variable_name="history"),
-                    ("human", "{question}"),
-                ]
-            )
+        prompt = ChatPromptTemplate.from_messages(
+            [
+                ("system", "You're an assistant who's good at {ability}"),
+                MessagesPlaceholder(variable_name="history"),
+                ("human", "{question}"),
+            ]
+        )
 
-            chain = prompt | ChatAnthropic(model="claude-2")
+        chain = prompt | ChatAnthropic(model="claude-2")
 
-            chain_with_history = RunnableWithMessageHistory(
-                chain,
-                # Uses the get_by_session_id function defined in the example
-                # above.
-                get_by_session_id,
-                input_messages_key="question",
-                history_messages_key="history",
-            )
+        chain_with_history = RunnableWithMessageHistory(
+            chain,
+            # Uses the get_by_session_id function defined in the example
+            # above.
+            get_by_session_id,
+            input_messages_key="question",
+            history_messages_key="history",
+        )
 
-            print(
-                chain_with_history.invoke(  # noqa: T201
-                    {"ability": "math", "question": "What does cosine mean?"},
-                    config={"configurable": {"session_id": "foo"}},
-                )
+        print(
+            chain_with_history.invoke(  # noqa: T201
+                {"ability": "math", "question": "What does cosine mean?"},
+                config={"configurable": {"session_id": "foo"}},
             )
+        )
 
-            # Uses the store defined in the example above.
-            print(store)  # noqa: T201
+        # Uses the store defined in the example above.
+        print(store)  # noqa: T201
 
-            print(
-                chain_with_history.invoke(  # noqa: T201
-                    {"ability": "math", "question": "What's its inverse"},
-                    config={"configurable": {"session_id": "foo"}},
-                )
+        print(
+            chain_with_history.invoke(  # noqa: T201
+                {"ability": "math", "question": "What's its inverse"},
+                config={"configurable": {"session_id": "foo"}},
             )
+        )
 
-            print(store)  # noqa: T201
-
+        print(store)  # noqa: T201
+        ```
 
     Example where the session factory takes two keys, user_id and conversation id):
 
-        .. code-block:: python
-
-            store = {}
+        ```python
+        store = {}
 
 
-            def get_session_history(
-                user_id: str, conversation_id: str
-            ) -> BaseChatMessageHistory:
-                if (user_id, conversation_id) not in store:
-                    store[(user_id, conversation_id)] = InMemoryHistory()
-                return store[(user_id, conversation_id)]
+        def get_session_history(
+            user_id: str, conversation_id: str
+        ) -> BaseChatMessageHistory:
+            if (user_id, conversation_id) not in store:
+                store[(user_id, conversation_id)] = InMemoryHistory()
+            return store[(user_id, conversation_id)]
 
 
-            prompt = ChatPromptTemplate.from_messages(
-                [
-                    ("system", "You're an assistant who's good at {ability}"),
-                    MessagesPlaceholder(variable_name="history"),
-                    ("human", "{question}"),
-                ]
-            )
+        prompt = ChatPromptTemplate.from_messages(
+            [
+                ("system", "You're an assistant who's good at {ability}"),
+                MessagesPlaceholder(variable_name="history"),
+                ("human", "{question}"),
+            ]
+        )
 
-            chain = prompt | ChatAnthropic(model="claude-2")
-
-            with_message_history = RunnableWithMessageHistory(
-                chain,
-                get_session_history=get_session_history,
-                input_messages_key="question",
-                history_messages_key="history",
-                history_factory_config=[
-                    ConfigurableFieldSpec(
-                        id="user_id",
-                        annotation=str,
-                        name="User ID",
-                        description="Unique identifier for the user.",
-                        default="",
-                        is_shared=True,
-                    ),
-                    ConfigurableFieldSpec(
-                        id="conversation_id",
-                        annotation=str,
-                        name="Conversation ID",
-                        description="Unique identifier for the conversation.",
-                        default="",
-                        is_shared=True,
-                    ),
-                ],
-            )
+        chain = prompt | ChatAnthropic(model="claude-2")
 
-            with_message_history.invoke(
-                {"ability": "math", "question": "What does cosine mean?"},
-                config={"configurable": {"user_id": "123", "conversation_id": "1"}},
-            )
+        with_message_history = RunnableWithMessageHistory(
+            chain,
+            get_session_history=get_session_history,
+            input_messages_key="question",
+            history_messages_key="history",
+            history_factory_config=[
+                ConfigurableFieldSpec(
+                    id="user_id",
+                    annotation=str,
+                    name="User ID",
+                    description="Unique identifier for the user.",
+                    default="",
+                    is_shared=True,
+                ),
+                ConfigurableFieldSpec(
+                    id="conversation_id",
+                    annotation=str,
+                    name="Conversation ID",
+                    description="Unique identifier for the conversation.",
+                    default="",
+                    is_shared=True,
+                ),
+            ],
+        )
 
+        with_message_history.invoke(
+            {"ability": "math", "question": "What does cosine mean?"},
+            config={"configurable": {"user_id": "123", "conversation_id": "1"}},
+        )
+        ```
     """
 
     get_session_history: GetSessionHistoryCallable
     """Function that returns a new BaseChatMessageHistory.
     This function should either take a single positional argument ``session_id`` of type
     string and return a corresponding chat message history instance"""
-    input_messages_key: Optional[str] = None
+    input_messages_key: str | None = None
     """Must be specified if the base runnable accepts a dict as input.
     The key in the input dict that contains the messages."""
-    output_messages_key: Optional[str] = None
+    output_messages_key: str | None = None
     """Must be specified if the base Runnable returns a dict as output.
     The key in the output dict that contains the messages."""
-    history_messages_key: Optional[str] = None
+    history_messages_key: str | None = None
     """Must be specified if the base runnable accepts a dict as input and expects a
     separate key for historical messages."""
     history_factory_config: Sequence[ConfigurableFieldSpec]
@@ -246,23 +241,17 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
 
     def __init__(
         self,
-        runnable: Union[
-            Runnable[
-                list[BaseMessage],
-                Union[str, BaseMessage, MessagesOrDictWithMessages],
-            ],
-            Runnable[
-                dict[str, Any],
-                Union[str, BaseMessage, MessagesOrDictWithMessages],
-            ],
-            LanguageModelLike,
-        ],
+        runnable: Runnable[
+            list[BaseMessage], str | BaseMessage | MessagesOrDictWithMessages
+        ]
+        | Runnable[dict[str, Any], str | BaseMessage | MessagesOrDictWithMessages]
+        | LanguageModelLike,
         get_session_history: GetSessionHistoryCallable,
         *,
-        input_messages_key: Optional[str] = None,
-        output_messages_key: Optional[str] = None,
-        history_messages_key: Optional[str] = None,
-        history_factory_config: Optional[Sequence[ConfigurableFieldSpec]] = None,
+        input_messages_key: str | None = None,
+        output_messages_key: str | None = None,
+        history_messages_key: str | None = None,
+        history_factory_config: Sequence[ConfigurableFieldSpec] | None = None,
         **kwargs: Any,
     ) -> None:
         """Initialize RunnableWithMessageHistory.
@@ -288,23 +277,23 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
                 This function should either take a single positional argument
                 `session_id` of type string and return a corresponding
                 chat message history instance.
-                .. code-block:: python
-
-                    def get_session_history(
-                        session_id: str, *, user_id: Optional[str] = None
-                    ) -> BaseChatMessageHistory: ...
+                ```python
+                def get_session_history(
+                    session_id: str, *, user_id: Optional[str] = None
+                ) -> BaseChatMessageHistory: ...
+                ```
 
                 Or it should take keyword arguments that match the keys of
                 `session_history_config_specs` and return a corresponding
                 chat message history instance.
 
-                .. code-block:: python
-
-                    def get_session_history(
-                        *,
-                        user_id: str,
-                        thread_id: str,
-                    ) -> BaseChatMessageHistory: ...
+                ```python
+                def get_session_history(
+                    *,
+                    user_id: str,
+                    thread_id: str,
+                ) -> BaseChatMessageHistory: ...
+                ```
 
             input_messages_key: Must be specified if the base runnable accepts a dict
                 as input. Default is None.
@@ -381,13 +370,11 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
         )
 
     @override
-    def get_input_schema(
-        self, config: Optional[RunnableConfig] = None
-    ) -> type[BaseModel]:
+    def get_input_schema(self, config: RunnableConfig | None = None) -> type[BaseModel]:
         fields: dict = {}
         if self.input_messages_key and self.history_messages_key:
             fields[self.input_messages_key] = (
-                Union[str, BaseMessage, Sequence[BaseMessage]],
+                str | BaseMessage | Sequence[BaseMessage],
                 ...,
             )
         elif self.input_messages_key:
@@ -411,7 +398,7 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
 
     @override
     def get_output_schema(
-        self, config: Optional[RunnableConfig] = None
+        self, config: RunnableConfig | None = None
     ) -> type[BaseModel]:
         """Get a pydantic model that can be used to validate output to the Runnable.
 
@@ -443,7 +430,7 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
         )
 
     def _get_input_messages(
-        self, input_val: Union[str, BaseMessage, Sequence[BaseMessage], dict]
+        self, input_val: str | BaseMessage | Sequence[BaseMessage] | dict
     ) -> list[BaseMessage]:
         # If dictionary, try to pluck the single key representing messages
         if isinstance(input_val, dict):
@@ -481,7 +468,7 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
         raise ValueError(msg)
 
     def _get_output_messages(
-        self, output_val: Union[str, BaseMessage, Sequence[BaseMessage], dict]
+        self, output_val: str | BaseMessage | Sequence[BaseMessage] | dict
     ) -> list[BaseMessage]:
         # If dictionary, try to pluck the single key representing messages
         if isinstance(output_val, dict):
@@ -571,7 +558,7 @@ class RunnableWithMessageHistory(RunnableBindingBase):  # type: ignore[no-redef]
         output_messages = self._get_output_messages(output_val)
         await hist.aadd_messages(input_messages + output_messages)
 
-    def _merge_configs(self, *configs: Optional[RunnableConfig]) -> RunnableConfig:
+    def _merge_configs(self, *configs: RunnableConfig | None) -> RunnableConfig:
         config = super()._merge_configs(*configs)
         expected_keys = [field_spec.id for field_spec in self.history_factory_config]