langchain-core 0.4.0.dev0__py3-none-any.whl → 1.0.0__py3-none-any.whl

langchain_core/callbacks/base.py

@@ -3,12 +3,10 @@
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import TYPE_CHECKING, Any
 
 from typing_extensions import Self
 
-from langchain_core.v1.messages import AIMessage, AIMessageChunk, MessageV1
-
 if TYPE_CHECKING:
     from collections.abc import Sequence
     from uuid import UUID
@@ -31,16 +29,16 @@ class RetrieverManagerMixin:
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when Retriever errors.
 
         Args:
-            error (BaseException): The error that occurred.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            error: The error that occurred.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_retriever_end(
@@ -48,16 +46,16 @@ class RetrieverManagerMixin:
         documents: Sequence[Document],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when Retriever ends running.
 
         Args:
-            documents (Sequence[Document]): The documents retrieved.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            documents: The documents retrieved.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
 
@@ -68,39 +66,38 @@ class LLMManagerMixin:
         self,
         token: str,
         *,
-        chunk: Optional[
-            Union[GenerationChunk, ChatGenerationChunk, AIMessageChunk]
-        ] = None,
+        chunk: GenerationChunk | ChatGenerationChunk | None = None,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
-        """Run on new LLM token. Only available when streaming is enabled.
+        """Run on new output token. Only available when streaming is enabled.
+
+        For both chat models and non-chat models (legacy LLMs).
 
         Args:
-            token (str): The new token.
-            chunk (GenerationChunk | ChatGenerationChunk | AIMessageChunk): The new
-              generated chunk, containing content and other information.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            token: The new token.
+            chunk: The new generated chunk, containing content and other information.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_llm_end(
         self,
-        response: Union[LLMResult, AIMessage],
+        response: LLMResult,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when LLM ends running.
 
         Args:
-            response (LLMResult | AIMessage): The response which was generated.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            response: The response which was generated.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_llm_error(
@@ -108,16 +105,16 @@ class LLMManagerMixin:
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when LLM errors.
 
         Args:
-            error (BaseException): The error that occurred.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            error: The error that occurred.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
 
@@ -129,16 +126,16 @@ class ChainManagerMixin:
         outputs: dict[str, Any],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when chain ends running.
 
         Args:
-            outputs (dict[str, Any]): The outputs of the chain.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            outputs: The outputs of the chain.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_chain_error(
@@ -146,16 +143,16 @@ class ChainManagerMixin:
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when chain errors.
 
         Args:
-            error (BaseException): The error that occurred.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            error: The error that occurred.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_agent_action(
@@ -163,16 +160,16 @@ class ChainManagerMixin:
         action: AgentAction,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run on agent action.
 
         Args:
-            action (AgentAction): The agent action.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            action: The agent action.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_agent_finish(
@@ -180,16 +177,16 @@ class ChainManagerMixin:
         finish: AgentFinish,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run on the agent end.
 
         Args:
-            finish (AgentFinish): The agent finish.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            finish: The agent finish.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
 
@@ -201,16 +198,16 @@ class ToolManagerMixin:
         output: Any,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when the tool ends running.
 
         Args:
-            output (Any): The output of the tool.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            output: The output of the tool.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_tool_error(
@@ -218,16 +215,16 @@ class ToolManagerMixin:
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when tool errors.
 
         Args:
-            error (BaseException): The error that occurred.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            error: The error that occurred.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
 
@@ -240,52 +237,53 @@ class CallbackManagerMixin:
         prompts: list[str],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when LLM starts running.
 
-        .. ATTENTION::
+        !!! warning
             This method is called for non-chat models (regular LLMs). If you're
             implementing a handler for a chat model, you should use
-            ``on_chat_model_start`` instead.
+            `on_chat_model_start` instead.
 
         Args:
-            serialized (dict[str, Any]): The serialized LLM.
-            prompts (list[str]): The prompts.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized LLM.
+            prompts: The prompts.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_chat_model_start(
         self,
         serialized: dict[str, Any],
-        messages: Union[list[list[BaseMessage]], list[MessageV1]],
+        messages: list[list[BaseMessage]],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when a chat model starts running.
 
-        **ATTENTION**: This method is called for chat models. If you're implementing
-        a handler for a non-chat model, you should use ``on_llm_start`` instead.
+        !!! warning
+            This method is called for chat models. If you're implementing a handler for
+            a non-chat model, you should use `on_llm_start` instead.
 
         Args:
-            serialized (dict[str, Any]): The serialized chat model.
-            messages (list[list[BaseMessage]]): The messages.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized chat model.
+            messages: The messages.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
         # NotImplementedError is thrown intentionally
         # Callback handler will fall back to on_llm_start if this is exception is thrown
@@ -298,21 +296,21 @@ class CallbackManagerMixin:
         query: str,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when the Retriever starts running.
 
         Args:
-            serialized (dict[str, Any]): The serialized Retriever.
-            query (str): The query.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized Retriever.
+            query: The query.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_chain_start(
@@ -321,21 +319,21 @@ class CallbackManagerMixin:
         inputs: dict[str, Any],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when a chain starts running.
 
         Args:
-            serialized (dict[str, Any]): The serialized chain.
-            inputs (dict[str, Any]): The inputs.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized chain.
+            inputs: The inputs.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_tool_start(
@@ -344,23 +342,23 @@ class CallbackManagerMixin:
         input_str: str,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        inputs: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
+        inputs: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when the tool starts running.
 
         Args:
-            serialized (dict[str, Any]): The serialized tool.
-            input_str (str): The input string.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            inputs (Optional[dict[str, Any]]): The inputs.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized chain.
+            input_str: The input string.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            inputs: The inputs.
+            **kwargs: Additional keyword arguments.
         """
 
 
@@ -372,16 +370,16 @@ class RunManagerMixin:
         text: str,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run on an arbitrary text.
 
         Args:
-            text (str): The text.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            text: The text.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_retry(
@@ -389,16 +387,16 @@ class RunManagerMixin:
         retry_state: RetryCallState,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run on a retry event.
 
         Args:
-            retry_state (RetryCallState): The retry state.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            retry_state: The retry state.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     def on_custom_event(
@@ -407,8 +405,8 @@ class RunManagerMixin:
         data: Any,
         *,
         run_id: UUID,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Any:
         """Override to define a handler for a custom event.
@@ -416,14 +414,14 @@ class RunManagerMixin:
         Args:
             name: The name of the custom event.
             data: The data for the custom event. Format will match
-                  the format specified by the user.
+                the format specified by the user.
             run_id: The ID of the run.
             tags: The tags associated with the custom event
                 (includes inherited tags).
             metadata: The metadata associated with the custom event
                 (includes inherited metadata).
 
-        .. versionadded:: 0.2.15
+        !!! version-added "Added in version 0.2.15"
         """
 
 
@@ -443,9 +441,6 @@ class BaseCallbackHandler(
     run_inline: bool = False
     """Whether to run the callback inline."""
 
-    accepts_new_messages: bool = False
-    """Whether the callback accepts new message format."""
-
     @property
     def ignore_llm(self) -> bool:
         """Whether to ignore LLM callbacks."""
@@ -491,52 +486,53 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         prompts: list[str],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> None:
-        """Run when LLM starts running.
+        """Run when the model starts running.
 
-        .. ATTENTION::
+        !!! warning
             This method is called for non-chat models (regular LLMs). If you're
             implementing a handler for a chat model, you should use
-            ``on_chat_model_start`` instead.
+            `on_chat_model_start` instead.
 
         Args:
-            serialized (dict[str, Any]): The serialized LLM.
-            prompts (list[str]): The prompts.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized LLM.
+            prompts: The prompts.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_chat_model_start(
         self,
         serialized: dict[str, Any],
-        messages: Union[list[list[BaseMessage]], list[MessageV1]],
+        messages: list[list[BaseMessage]],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run when a chat model starts running.
 
-        **ATTENTION**: This method is called for chat models. If you're implementing
-        a handler for a non-chat model, you should use ``on_llm_start`` instead.
+        !!! warning
+            This method is called for chat models. If you're implementing a handler for
+            a non-chat model, you should use `on_llm_start` instead.
 
         Args:
-            serialized (dict[str, Any]): The serialized chat model.
-            messages (list[list[BaseMessage]]): The messages.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized chat model.
+            messages: The messages.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
         # NotImplementedError is thrown intentionally
         # Callback handler will fall back to on_llm_start if this is exception is thrown
@@ -547,43 +543,42 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         self,
         token: str,
         *,
-        chunk: Optional[
-            Union[GenerationChunk, ChatGenerationChunk, AIMessageChunk]
-        ] = None,
+        chunk: GenerationChunk | ChatGenerationChunk | None = None,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
-        """Run on new LLM token. Only available when streaming is enabled.
+        """Run on new output token. Only available when streaming is enabled.
+
+        For both chat models and non-chat models (legacy LLMs).
 
         Args:
-            token (str): The new token.
-            chunk (GenerationChunk | ChatGenerationChunk | AIMessageChunk): The new
-              generated chunk, containing content and other information.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            token: The new token.
+            chunk: The new generated chunk, containing content and other information.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_llm_end(
         self,
-        response: Union[LLMResult, AIMessage],
+        response: LLMResult,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
-        """Run when LLM ends running.
+        """Run when the model ends running.
 
         Args:
-            response (LLMResult | AIMessage): The response which was generated.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            response: The response which was generated.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_llm_error(
@@ -591,8 +586,8 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run when LLM errors.
@@ -602,9 +597,9 @@ class AsyncCallbackHandler(BaseCallbackHandler):
             run_id: The run ID. This is the ID of the current run.
             parent_run_id: The parent run ID. This is the ID of the parent run.
             tags: The tags.
-            kwargs (Any): Additional keyword arguments.
-                - response (LLMResult | AIMessage): The response which was generated
-                    before the error occurred.
+            **kwargs: Additional keyword arguments.
+                - response (LLMResult): The response which was generated before
+                    the error occurred.
         """
 
     async def on_chain_start(
@@ -613,21 +608,21 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         inputs: dict[str, Any],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run when a chain starts running.
 
         Args:
-            serialized (dict[str, Any]): The serialized chain.
-            inputs (dict[str, Any]): The inputs.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized chain.
+            inputs: The inputs.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_chain_end(
@@ -635,18 +630,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         outputs: dict[str, Any],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run when a chain ends running.
 
         Args:
-            outputs (dict[str, Any]): The outputs of the chain.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            outputs: The outputs of the chain.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_chain_error(
@@ -654,18 +649,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run when chain errors.
 
         Args:
-            error (BaseException): The error that occurred.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            error: The error that occurred.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_tool_start(
@@ -674,23 +669,23 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         input_str: str,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        inputs: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
+        inputs: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run when the tool starts running.
 
         Args:
-            serialized (dict[str, Any]): The serialized tool.
-            input_str (str): The input string.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            inputs (Optional[dict[str, Any]]): The inputs.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized tool.
+            input_str: The input string.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            inputs: The inputs.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_tool_end(
@@ -698,18 +693,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         output: Any,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run when the tool ends running.
 
         Args:
-            output (Any): The output of the tool.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            output: The output of the tool.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_tool_error(
@@ -717,18 +712,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run when tool errors.
 
         Args:
-            error (BaseException): The error that occurred.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            error: The error that occurred.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_text(
@@ -736,18 +731,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         text: str,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run on an arbitrary text.
 
         Args:
-            text (str): The text.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            text: The text.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_retry(
@@ -755,16 +750,16 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         retry_state: RetryCallState,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
+        parent_run_id: UUID | None = None,
         **kwargs: Any,
     ) -> Any:
         """Run on a retry event.
 
         Args:
-            retry_state (RetryCallState): The retry state.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            kwargs (Any): Additional keyword arguments.
+            retry_state: The retry state.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_agent_action(
@@ -772,18 +767,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         action: AgentAction,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run on agent action.
 
         Args:
-            action (AgentAction): The agent action.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            action: The agent action.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_agent_finish(
@@ -791,18 +786,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         finish: AgentFinish,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run on the agent end.
 
         Args:
-            finish (AgentFinish): The agent finish.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            finish: The agent finish.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_retriever_start(
@@ -811,21 +806,21 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         query: str,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run on the retriever start.
 
         Args:
-            serialized (dict[str, Any]): The serialized retriever.
-            query (str): The query.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            metadata (Optional[dict[str, Any]]): The metadata.
-            kwargs (Any): Additional keyword arguments.
+            serialized: The serialized retriever.
+            query: The query.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            metadata: The metadata.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_retriever_end(
@@ -833,18 +828,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         documents: Sequence[Document],
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run on the retriever end.
 
         Args:
-            documents (Sequence[Document]): The documents retrieved.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            documents: The documents retrieved.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_retriever_error(
@@ -852,18 +847,18 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         error: BaseException,
         *,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
         **kwargs: Any,
     ) -> None:
         """Run on retriever error.
 
         Args:
-            error (BaseException): The error that occurred.
-            run_id (UUID): The run ID. This is the ID of the current run.
-            parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
-            tags (Optional[list[str]]): The tags.
-            kwargs (Any): Additional keyword arguments.
+            error: The error that occurred.
+            run_id: The run ID. This is the ID of the current run.
+            parent_run_id: The parent run ID. This is the ID of the parent run.
+            tags: The tags.
+            **kwargs: Additional keyword arguments.
         """
 
     async def on_custom_event(
@@ -872,23 +867,23 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         data: Any,
         *,
         run_id: UUID,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> None:
-        """Override to define a handler for a custom event.
+        """Override to define a handler for custom events.
 
         Args:
             name: The name of the custom event.
             data: The data for the custom event. Format will match
-                  the format specified by the user.
+                the format specified by the user.
             run_id: The ID of the run.
             tags: The tags associated with the custom event
                 (includes inherited tags).
             metadata: The metadata associated with the custom event
                 (includes inherited metadata).
 
-        .. versionadded:: 0.2.15
+        !!! version-added "Added in version 0.2.15"
         """
 
 
@@ -898,40 +893,37 @@ class BaseCallbackManager(CallbackManagerMixin):
     def __init__(
         self,
         handlers: list[BaseCallbackHandler],
-        inheritable_handlers: Optional[list[BaseCallbackHandler]] = None,
-        parent_run_id: Optional[UUID] = None,
+        inheritable_handlers: list[BaseCallbackHandler] | None = None,
+        parent_run_id: UUID | None = None,
         *,
-        tags: Optional[list[str]] = None,
-        inheritable_tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        inheritable_metadata: Optional[dict[str, Any]] = None,
+        tags: list[str] | None = None,
+        inheritable_tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
+        inheritable_metadata: dict[str, Any] | None = None,
     ) -> None:
         """Initialize callback manager.
 
         Args:
-            handlers (list[BaseCallbackHandler]): The handlers.
-            inheritable_handlers (Optional[list[BaseCallbackHandler]]):
-              The inheritable handlers. Default is None.
-            parent_run_id (Optional[UUID]): The parent run ID. Default is None.
-            tags (Optional[list[str]]): The tags. Default is None.
-            inheritable_tags (Optional[list[str]]): The inheritable tags.
-                Default is None.
-            metadata (Optional[dict[str, Any]]): The metadata. Default is None.
-            inheritable_metadata (Optional[dict[str, Any]]): The inheritable metadata.
-                Default is None.
+            handlers: The handlers.
+            inheritable_handlers: The inheritable handlers.
+            parent_run_id: The parent run ID.
+            tags: The tags.
+            inheritable_tags: The inheritable tags.
+            metadata: The metadata.
+            inheritable_metadata: The inheritable metadata.
         """
         self.handlers: list[BaseCallbackHandler] = handlers
         self.inheritable_handlers: list[BaseCallbackHandler] = (
             inheritable_handlers or []
         )
-        self.parent_run_id: Optional[UUID] = parent_run_id
+        self.parent_run_id: UUID | None = parent_run_id
         self.tags = tags or []
         self.inheritable_tags = inheritable_tags or []
         self.metadata = metadata or {}
         self.inheritable_metadata = inheritable_metadata or {}
 
     def copy(self) -> Self:
-        """Copy the callback manager."""
+        """Return a copy of the callback manager."""
         return self.__class__(
             handlers=self.handlers.copy(),
             inheritable_handlers=self.inheritable_handlers.copy(),
@@ -949,28 +941,29 @@ class BaseCallbackManager(CallbackManagerMixin):
         within merge_configs.
 
         Returns:
-            BaseCallbackManager: The merged callback manager of the same type
-                as the current object.
+            The merged callback manager of the same type as the current object.
 
         Example: Merging two callback managers.
 
-            .. code-block:: python
-
-                from langchain_core.callbacks.manager import CallbackManager, trace_as_chain_group
-                from langchain_core.callbacks.stdout import StdOutCallbackHandler
-
-                manager = CallbackManager(handlers=[StdOutCallbackHandler()], tags=["tag2"])
-                with trace_as_chain_group("My Group Name", tags=["tag1"]) as group_manager:
-                    merged_manager = group_manager.merge(manager)
-                    print(merged_manager.handlers)
-                    # [
-                    #    <langchain_core.callbacks.stdout.StdOutCallbackHandler object at ...>,
-                    #    <langchain_core.callbacks.streaming_stdout.StreamingStdOutCallbackHandler object at ...>,
-                    # ]
-
-                    print(merged_manager.tags)
-                    #    ['tag2', 'tag1']
-
+            ```python
+            from langchain_core.callbacks.manager import (
+                CallbackManager,
+                trace_as_chain_group,
+            )
+            from langchain_core.callbacks.stdout import StdOutCallbackHandler
+
+            manager = CallbackManager(handlers=[StdOutCallbackHandler()], tags=["tag2"])
+            with trace_as_chain_group("My Group Name", tags=["tag1"]) as group_manager:
+                merged_manager = group_manager.merge(manager)
+                print(merged_manager.handlers)
+                # [
+                #    <langchain_core.callbacks.stdout.StdOutCallbackHandler object at ...>,
+                #    <langchain_core.callbacks.streaming_stdout.StreamingStdOutCallbackHandler object at ...>,
+                # ]
+
+                print(merged_manager.tags)
+                #    ['tag2', 'tag1']
+            ```
         """  # noqa: E501
         manager = self.__class__(
             parent_run_id=self.parent_run_id or other.parent_run_id,
@@ -1007,8 +1000,8 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Add a handler to the callback manager.
 
         Args:
-            handler (BaseCallbackHandler): The handler to add.
-            inherit (bool): Whether to inherit the handler. Default is True.
+            handler: The handler to add.
+            inherit: Whether to inherit the handler.
         """
         if handler not in self.handlers:
             self.handlers.append(handler)
@@ -1019,7 +1012,7 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Remove a handler from the callback manager.
 
         Args:
-            handler (BaseCallbackHandler): The handler to remove.
+            handler: The handler to remove.
         """
         if handler in self.handlers:
             self.handlers.remove(handler)
@@ -1034,8 +1027,8 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Set handlers as the only handlers on the callback manager.
 
         Args:
-            handlers (list[BaseCallbackHandler]): The handlers to set.
-            inherit (bool): Whether to inherit the handlers. Default is True.
+            handlers: The handlers to set.
+            inherit: Whether to inherit the handlers.
         """
         self.handlers = []
         self.inheritable_handlers = []
@@ -1050,8 +1043,8 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Set handler as the only handler on the callback manager.
 
         Args:
-            handler (BaseCallbackHandler): The handler to set.
-            inherit (bool): Whether to inherit the handler. Default is True.
+            handler: The handler to set.
+            inherit: Whether to inherit the handler.
         """
         self.set_handlers([handler], inherit=inherit)
 
@@ -1063,8 +1056,8 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Add tags to the callback manager.
 
         Args:
-            tags (list[str]): The tags to add.
-            inherit (bool): Whether to inherit the tags. Default is True.
+            tags: The tags to add.
+            inherit: Whether to inherit the tags.
         """
         for tag in tags:
             if tag in self.tags:
@@ -1077,7 +1070,7 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Remove tags from the callback manager.
 
         Args:
-            tags (list[str]): The tags to remove.
+            tags: The tags to remove.
         """
         for tag in tags:
             if tag in self.tags:
@@ -1093,8 +1086,8 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Add metadata to the callback manager.
 
         Args:
-            metadata (dict[str, Any]): The metadata to add.
-            inherit (bool): Whether to inherit the metadata. Default is True.
+            metadata: The metadata to add.
+            inherit: Whether to inherit the metadata.
         """
         self.metadata.update(metadata)
         if inherit:
@@ -1104,11 +1097,11 @@ class BaseCallbackManager(CallbackManagerMixin):
         """Remove metadata from the callback manager.
 
         Args:
-            keys (list[str]): The keys to remove.
+            keys: The keys to remove.
         """
         for key in keys:
             self.metadata.pop(key, None)
             self.inheritable_metadata.pop(key, None)
 
 
-Callbacks = Optional[Union[list[BaseCallbackHandler], BaseCallbackManager]]
+Callbacks = list[BaseCallbackHandler] | BaseCallbackManager | None