github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-07-15 15:40:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: chain decorator typing

Browse Source
This commit is contained in:
Bagatur
2024-02-08 17:24:40 -08:00
parent 65e97c9b53
commit b199ddc54c
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/core/langchain_core/runnables/base.py
View File

@@ -23,6 +23,7 @@ from typing import (
List,
Mapping,
Optional,
Protocol,
Sequence,
Set,
Tuple,
@@ -4373,6 +4374,11 @@ def coerce_to_runnable(thing: RunnableLike) -> Runnable[Input, Output]:
)
class _CallableWithKwargs(Protocol, Generic[Input, Output]):
def __call__(self, input: Input, **kwargs: Any) -> Output:
...
@overload
def chain(
func: Callable[[Input], Coroutine[Any, Any, Output]],
@@ -4407,6 +4413,10 @@ def chain(
Callable[[Input], Iterator[Output]],
Callable[[Input], Coroutine[Any, Any, Output]],
Callable[[Input], AsyncIterator[Output]],
_CallableWithKwargs[Input, Output],
_CallableWithKwargs[Input, Iterator[Output]],
_CallableWithKwargs[Input, Coroutine[Any, Any, Output]],
_CallableWithKwargs[Input, AsyncIterator[Output]],
],
) -> Runnable[Input, Output]:
"""Decorate a function to make it a Runnable.