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

langchain_core/runnables/router.py

@@ -2,13 +2,10 @@
 
 from __future__ import annotations
 
-from collections.abc import Mapping
+from collections.abc import Callable, Mapping
 from typing import (
     TYPE_CHECKING,
     Any,
-    Callable,
-    Optional,
-    Union,
     cast,
 )
 
@@ -52,18 +49,16 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
     Returns the output of the selected Runnable.
 
     Example:
+        ```python
+        from langchain_core.runnables.router import RouterRunnable
+        from langchain_core.runnables import RunnableLambda
 
-        .. code-block:: python
-
-            from langchain_core.runnables.router import RouterRunnable
-            from langchain_core.runnables import RunnableLambda
-
-            add = RunnableLambda(func=lambda x: x + 1)
-            square = RunnableLambda(func=lambda x: x**2)
-
-            router = RouterRunnable(runnables={"add": add, "square": square})
-            router.invoke({"key": "square", "input": 3})
+        add = RunnableLambda(func=lambda x: x + 1)
+        square = RunnableLambda(func=lambda x: x**2)
 
+        router = RouterRunnable(runnables={"add": add, "square": square})
+        router.invoke({"key": "square", "input": 3})
+        ```
     """
 
     runnables: Mapping[str, Runnable[Any, Output]]
@@ -77,7 +72,7 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
 
     def __init__(
         self,
-        runnables: Mapping[str, Union[Runnable[Any, Output], Callable[[Any], Output]]],
+        runnables: Mapping[str, Runnable[Any, Output] | Callable[[Any], Output]],
     ) -> None:
         """Create a RouterRunnable.
 
@@ -110,7 +105,7 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
 
     @override
     def invoke(
-        self, input: RouterInput, config: Optional[RunnableConfig] = None, **kwargs: Any
+        self, input: RouterInput, config: RunnableConfig | None = None, **kwargs: Any
     ) -> Output:
         key = input["key"]
         actual_input = input["input"]
@@ -125,8 +120,8 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
     async def ainvoke(
         self,
         input: RouterInput,
-        config: Optional[RunnableConfig] = None,
-        **kwargs: Optional[Any],
+        config: RunnableConfig | None = None,
+        **kwargs: Any | None,
     ) -> Output:
         key = input["key"]
         actual_input = input["input"]
@@ -141,10 +136,10 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
     def batch(
         self,
         inputs: list[RouterInput],
-        config: Optional[Union[RunnableConfig, list[RunnableConfig]]] = None,
+        config: RunnableConfig | list[RunnableConfig] | None = None,
         *,
         return_exceptions: bool = False,
-        **kwargs: Optional[Any],
+        **kwargs: Any | None,
     ) -> list[Output]:
         if not inputs:
             return []
@@ -157,7 +152,7 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
 
         def invoke(
             runnable: Runnable, input_: Input, config: RunnableConfig
-        ) -> Union[Output, Exception]:
+        ) -> Output | Exception:
             if return_exceptions:
                 try:
                     return runnable.invoke(input_, config, **kwargs)
@@ -178,10 +173,10 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
     async def abatch(
         self,
         inputs: list[RouterInput],
-        config: Optional[Union[RunnableConfig, list[RunnableConfig]]] = None,
+        config: RunnableConfig | list[RunnableConfig] | None = None,
         *,
         return_exceptions: bool = False,
-        **kwargs: Optional[Any],
+        **kwargs: Any | None,
     ) -> list[Output]:
         if not inputs:
             return []
@@ -194,7 +189,7 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
 
         async def ainvoke(
             runnable: Runnable, input_: Input, config: RunnableConfig
-        ) -> Union[Output, Exception]:
+        ) -> Output | Exception:
             if return_exceptions:
                 try:
                     return await runnable.ainvoke(input_, config, **kwargs)
@@ -214,8 +209,8 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
     def stream(
         self,
         input: RouterInput,
-        config: Optional[RunnableConfig] = None,
-        **kwargs: Optional[Any],
+        config: RunnableConfig | None = None,
+        **kwargs: Any | None,
     ) -> Iterator[Output]:
         key = input["key"]
         actual_input = input["input"]
@@ -230,8 +225,8 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
     async def astream(
         self,
         input: RouterInput,
-        config: Optional[RunnableConfig] = None,
-        **kwargs: Optional[Any],
+        config: RunnableConfig | None = None,
+        **kwargs: Any | None,
     ) -> AsyncIterator[Output]:
         key = input["key"]
         actual_input = input["input"]

langchain_core/runnables/schema.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Literal, Union
+from typing import TYPE_CHECKING, Any, Literal
 
 from typing_extensions import NotRequired, TypedDict
 
@@ -28,7 +28,7 @@ class EventData(TypedDict, total=False):
 
     This field is only available if the Runnable raised an exception.
 
-    .. versionadded:: 1.0.0
+    !!! version-added "Added in version 1.0.0"
     """
     output: Any
     """The output of the Runnable that generated the event.
@@ -53,46 +53,44 @@ class BaseStreamEvent(TypedDict):
     Schema of a streaming event which is produced from the astream_events method.
 
     Example:
-
-        .. code-block:: python
-
-            from langchain_core.runnables import RunnableLambda
-
-
-            async def reverse(s: str) -> str:
-                return s[::-1]
-
-
-            chain = RunnableLambda(func=reverse)
-
-            events = [event async for event in chain.astream_events("hello")]
-
-            # will produce the following events
-            # (where some fields have been omitted for brevity):
-            [
-                {
-                    "data": {"input": "hello"},
-                    "event": "on_chain_start",
-                    "metadata": {},
-                    "name": "reverse",
-                    "tags": [],
-                },
-                {
-                    "data": {"chunk": "olleh"},
-                    "event": "on_chain_stream",
-                    "metadata": {},
-                    "name": "reverse",
-                    "tags": [],
-                },
-                {
-                    "data": {"output": "olleh"},
-                    "event": "on_chain_end",
-                    "metadata": {},
-                    "name": "reverse",
-                    "tags": [],
-                },
-            ]
-
+        ```python
+        from langchain_core.runnables import RunnableLambda
+
+
+        async def reverse(s: str) -> str:
+            return s[::-1]
+
+
+        chain = RunnableLambda(func=reverse)
+
+        events = [event async for event in chain.astream_events("hello")]
+
+        # will produce the following events
+        # (where some fields have been omitted for brevity):
+        [
+            {
+                "data": {"input": "hello"},
+                "event": "on_chain_start",
+                "metadata": {},
+                "name": "reverse",
+                "tags": [],
+            },
+            {
+                "data": {"chunk": "olleh"},
+                "event": "on_chain_stream",
+                "metadata": {},
+                "name": "reverse",
+                "tags": [],
+            },
+            {
+                "data": {"output": "olleh"},
+                "event": "on_chain_end",
+                "metadata": {},
+                "name": "reverse",
+                "tags": [],
+            },
+        ]
+        ```
     """
 
     event: str
@@ -172,7 +170,7 @@ class StandardStreamEvent(BaseStreamEvent):
 class CustomStreamEvent(BaseStreamEvent):
     """Custom stream event created by the user.
 
-    .. versionadded:: 0.2.15
+    !!! version-added "Added in version 0.2.15"
     """
 
     # Overwrite the event field to be more specific.
@@ -184,4 +182,4 @@ class CustomStreamEvent(BaseStreamEvent):
     """The data associated with the event. Free form and can be anything."""
 
 
-StreamEvent = Union[StandardStreamEvent, CustomStreamEvent]
+StreamEvent = StandardStreamEvent | CustomStreamEvent

langchain_core/runnables/utils.py

@@ -6,7 +6,7 @@ import ast
 import asyncio
 import inspect
 import textwrap
-from collections.abc import Mapping, Sequence
+from collections.abc import Callable, Mapping, Sequence
 from contextvars import Context
 from functools import lru_cache
 from inspect import signature
@@ -14,13 +14,10 @@ from itertools import groupby
 from typing import (
     TYPE_CHECKING,
     Any,
-    Callable,
     NamedTuple,
-    Optional,
     Protocol,
     TypeGuard,
     TypeVar,
-    Union,
 )
 
 from typing_extensions import override
@@ -58,7 +55,7 @@ async def gated_coro(semaphore: asyncio.Semaphore, coro: Coroutine) -> Any:
         return await coro
 
 
-async def gather_with_concurrency(n: Union[int, None], *coros: Coroutine) -> list:
+async def gather_with_concurrency(n: int | None, *coros: Coroutine) -> list:
     """Gather coroutines with a limit on the number of concurrent coroutines.
 
     Args:
@@ -344,7 +341,7 @@ class GetLambdaSource(ast.NodeVisitor):
 
     def __init__(self) -> None:
         """Initialize the visitor."""
-        self.source: Optional[str] = None
+        self.source: str | None = None
         self.count = 0
 
     @override
@@ -359,7 +356,7 @@ class GetLambdaSource(ast.NodeVisitor):
             self.source = ast.unparse(node)
 
 
-def get_function_first_arg_dict_keys(func: Callable) -> Optional[list[str]]:
+def get_function_first_arg_dict_keys(func: Callable) -> list[str] | None:
     """Get the keys of the first argument of a function if it is a dict.
 
     Args:
@@ -379,7 +376,7 @@ def get_function_first_arg_dict_keys(func: Callable) -> Optional[list[str]]:
         return None
 
 
-def get_lambda_source(func: Callable) -> Optional[str]:
+def get_lambda_source(func: Callable) -> str | None:
     """Get the source code of a lambda function.
 
     Args:
@@ -521,7 +518,7 @@ class SupportsAdd(Protocol[_T_contra, _T_co]):
 Addable = TypeVar("Addable", bound=SupportsAdd[Any, Any])
 
 
-def add(addables: Iterable[Addable]) -> Optional[Addable]:
+def add(addables: Iterable[Addable]) -> Addable | None:
     """Add a sequence of addable objects together.
 
     Args:
@@ -530,13 +527,13 @@ def add(addables: Iterable[Addable]) -> Optional[Addable]:
     Returns:
         Optional[Addable]: The result of adding the addable objects.
     """
-    final: Optional[Addable] = None
+    final: Addable | None = None
     for chunk in addables:
         final = chunk if final is None else final + chunk
     return final
 
 
-async def aadd(addables: AsyncIterable[Addable]) -> Optional[Addable]:
+async def aadd(addables: AsyncIterable[Addable]) -> Addable | None:
     """Asynchronously add a sequence of addable objects together.
 
     Args:
@@ -545,7 +542,7 @@ async def aadd(addables: AsyncIterable[Addable]) -> Optional[Addable]:
     Returns:
         Optional[Addable]: The result of adding the addable objects.
     """
-    final: Optional[Addable] = None
+    final: Addable | None = None
     async for chunk in addables:
         final = chunk if final is None else final + chunk
     return final
@@ -556,11 +553,11 @@ class ConfigurableField(NamedTuple):
 
     id: str
     """The unique identifier of the field."""
-    name: Optional[str] = None
+    name: str | None = None
     """The name of the field. Defaults to None."""
-    description: Optional[str] = None
+    description: str | None = None
     """The description of the field. Defaults to None."""
-    annotation: Optional[Any] = None
+    annotation: Any | None = None
     """The annotation of the field. Defaults to None."""
     is_shared: bool = False
     """Whether the field is shared. Defaults to False."""
@@ -579,9 +576,9 @@ class ConfigurableFieldSingleOption(NamedTuple):
     """The options for the field."""
     default: str
     """The default value for the field."""
-    name: Optional[str] = None
+    name: str | None = None
     """The name of the field. Defaults to None."""
-    description: Optional[str] = None
+    description: str | None = None
     """The description of the field. Defaults to None."""
     is_shared: bool = False
     """Whether the field is shared. Defaults to False."""
@@ -600,9 +597,9 @@ class ConfigurableFieldMultiOption(NamedTuple):
     """The options for the field."""
     default: Sequence[str]
     """The default values for the field."""
-    name: Optional[str] = None
+    name: str | None = None
     """The name of the field. Defaults to None."""
-    description: Optional[str] = None
+    description: str | None = None
     """The description of the field. Defaults to None."""
     is_shared: bool = False
     """Whether the field is shared. Defaults to False."""
@@ -612,9 +609,9 @@ class ConfigurableFieldMultiOption(NamedTuple):
         return hash((self.id, tuple(self.options.keys()), tuple(self.default)))
 
 
-AnyConfigurableField = Union[
-    ConfigurableField, ConfigurableFieldSingleOption, ConfigurableFieldMultiOption
-]
+AnyConfigurableField = (
+    ConfigurableField | ConfigurableFieldSingleOption | ConfigurableFieldMultiOption
+)
 
 
 class ConfigurableFieldSpec(NamedTuple):
@@ -624,15 +621,15 @@ class ConfigurableFieldSpec(NamedTuple):
     """The unique identifier of the field."""
     annotation: Any
     """The annotation of the field."""
-    name: Optional[str] = None
+    name: str | None = None
     """The name of the field. Defaults to None."""
-    description: Optional[str] = None
+    description: str | None = None
     """The description of the field. Defaults to None."""
     default: Any = None
     """The default value for the field. Defaults to None."""
     is_shared: bool = False
     """Whether the field is shared. Defaults to False."""
-    dependencies: Optional[list[str]] = None
+    dependencies: list[str] | None = None
     """The dependencies of the field. Defaults to None."""
 
 
@@ -672,12 +669,12 @@ class _RootEventFilter:
     def __init__(
         self,
         *,
-        include_names: Optional[Sequence[str]] = None,
-        include_types: Optional[Sequence[str]] = None,
-        include_tags: Optional[Sequence[str]] = None,
-        exclude_names: Optional[Sequence[str]] = None,
-        exclude_types: Optional[Sequence[str]] = None,
-        exclude_tags: Optional[Sequence[str]] = None,
+        include_names: Sequence[str] | None = None,
+        include_types: Sequence[str] | None = None,
+        include_tags: Sequence[str] | None = None,
+        exclude_names: Sequence[str] | None = None,
+        exclude_types: Sequence[str] | None = None,
+        exclude_tags: Sequence[str] | None = None,
     ) -> None:
         """Utility to filter the root event in the astream_events implementation.
 

langchain_core/stores.py

@@ -11,9 +11,7 @@ from collections.abc import AsyncIterator, Iterator, Sequence
 from typing import (
     Any,
     Generic,
-    Optional,
     TypeVar,
-    Union,
 )
 
 from typing_extensions import override
@@ -82,7 +80,7 @@ class BaseStore(ABC, Generic[K, V]):
     """
 
     @abstractmethod
-    def mget(self, keys: Sequence[K]) -> list[Optional[V]]:
+    def mget(self, keys: Sequence[K]) -> list[V | None]:
         """Get the values associated with the given keys.
 
         Args:
@@ -93,7 +91,7 @@ class BaseStore(ABC, Generic[K, V]):
             If a key is not found, the corresponding value will be None.
         """
 
-    async def amget(self, keys: Sequence[K]) -> list[Optional[V]]:
+    async def amget(self, keys: Sequence[K]) -> list[V | None]:
         """Async get the values associated with the given keys.
 
         Args:
@@ -138,9 +136,7 @@ class BaseStore(ABC, Generic[K, V]):
         return await run_in_executor(None, self.mdelete, keys)
 
     @abstractmethod
-    def yield_keys(
-        self, *, prefix: Optional[str] = None
-    ) -> Union[Iterator[K], Iterator[str]]:
+    def yield_keys(self, *, prefix: str | None = None) -> Iterator[K] | Iterator[str]:
         """Get an iterator over keys that match the given prefix.
 
         Args:
@@ -153,8 +149,8 @@ class BaseStore(ABC, Generic[K, V]):
         """
 
     async def ayield_keys(
-        self, *, prefix: Optional[str] = None
-    ) -> Union[AsyncIterator[K], AsyncIterator[str]]:
+        self, *, prefix: str | None = None
+    ) -> AsyncIterator[K] | AsyncIterator[str]:
         """Async get an iterator over keys that match the given prefix.
 
         Args:
@@ -184,7 +180,7 @@ class InMemoryBaseStore(BaseStore[str, V], Generic[V]):
         """Initialize an empty store."""
         self.store: dict[str, V] = {}
 
-    def mget(self, keys: Sequence[str]) -> list[Optional[V]]:
+    def mget(self, keys: Sequence[str]) -> list[V | None]:
         """Get the values associated with the given keys.
 
         Args:
@@ -196,7 +192,7 @@ class InMemoryBaseStore(BaseStore[str, V], Generic[V]):
         """
         return [self.store.get(key) for key in keys]
 
-    async def amget(self, keys: Sequence[str]) -> list[Optional[V]]:
+    async def amget(self, keys: Sequence[str]) -> list[V | None]:
         """Async get the values associated with the given keys.
 
         Args:
@@ -235,7 +231,7 @@ class InMemoryBaseStore(BaseStore[str, V], Generic[V]):
         """
         self.mdelete(keys)
 
-    def yield_keys(self, prefix: Optional[str] = None) -> Iterator[str]:
+    def yield_keys(self, prefix: str | None = None) -> Iterator[str]:
         """Get an iterator over keys that match the given prefix.
 
         Args:
@@ -251,7 +247,7 @@ class InMemoryBaseStore(BaseStore[str, V], Generic[V]):
                 if key.startswith(prefix):
                     yield key
 
-    async def ayield_keys(self, prefix: Optional[str] = None) -> AsyncIterator[str]:
+    async def ayield_keys(self, prefix: str | None = None) -> AsyncIterator[str]:
         """Async get an async iterator over keys that match the given prefix.
 
         Args:

langchain_core/structured_query.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from abc import ABC, abstractmethod
 from enum import Enum
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel
 
@@ -15,12 +15,12 @@ if TYPE_CHECKING:
 class Visitor(ABC):
     """Defines interface for IR translation using a visitor pattern."""
 
-    allowed_comparators: Optional[Sequence[Comparator]] = None
+    allowed_comparators: Sequence[Comparator] | None = None
     """Allowed comparators for the visitor."""
-    allowed_operators: Optional[Sequence[Operator]] = None
+    allowed_operators: Sequence[Operator] | None = None
     """Allowed operators for the visitor."""
 
-    def _validate_func(self, func: Union[Operator, Comparator]) -> None:
+    def _validate_func(self, func: Operator | Comparator) -> None:
         if (
             isinstance(func, Operator)
             and self.allowed_operators is not None
@@ -174,16 +174,16 @@ class StructuredQuery(Expr):
 
     query: str
     """Query string."""
-    filter: Optional[FilterDirective]
+    filter: FilterDirective | None
     """Filtering expression."""
-    limit: Optional[int]
+    limit: int | None
     """Limit on the number of results."""
 
     def __init__(
         self,
         query: str,
-        filter: Optional[FilterDirective],  # noqa: A002
-        limit: Optional[int] = None,
+        filter: FilterDirective | None,  # noqa: A002
+        limit: int | None = None,
         **kwargs: Any,
     ) -> None:
         """Create a StructuredQuery.