github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-08 14:05:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Lint

Browse Source
This commit is contained in:
Nuno Campos 2023-09-29 12:00:56 +01:00
parent 7f589ebbc2
commit e35ea565d1
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
libs/langchain/langchain/tools/base.py
View File

@ -771,23 +771,25 @@ def tool(
def _make_with_name(tool_name: str) -> Callable: def _make_with_name(tool_name: str) -> Callable:
def _make_tool(dec_func: Union[Callable, Runnable]) -> BaseTool: def _make_tool(dec_func: Union[Callable, Runnable]) -> BaseTool:
if isinstance(dec_func, Runnable): if isinstance(dec_func, Runnable):
if dec_func.input_schema.schema().get("type") != "object": runnable = dec_func
if runnable.input_schema.schema().get("type") != "object":
raise ValueError("Runnable must have an object schema.") raise ValueError("Runnable must have an object schema.")
async def ainvoke_wrapper( async def ainvoke_wrapper(
callbacks: Optional[Callbacks] = None, **kwargs: Any callbacks: Optional[Callbacks] = None, **kwargs: Any
) -> Any: ) -> Any:
return await dec_func.ainvoke(kwargs, {"callbacks": callbacks}) return await runnable.ainvoke(kwargs, {"callbacks": callbacks})
def invoke_wrapper( def invoke_wrapper(
callbacks: Optional[Callbacks] = None, **kwargs: Any callbacks: Optional[Callbacks] = None, **kwargs: Any
) -> Any: ) -> Any:
return dec_func.invoke(kwargs, {"callbacks": callbacks}) return runnable.invoke(kwargs, {"callbacks": callbacks})
coroutine = ainvoke_wrapper coroutine = ainvoke_wrapper
func = invoke_wrapper func = invoke_wrapper
schema = dec_func.input_schema schema: Optional[Type[BaseModel]] = runnable.input_schema
description = repr(dec_func) description = repr(runnable)
elif inspect.iscoroutinefunction(dec_func): elif inspect.iscoroutinefunction(dec_func):
coroutine = dec_func coroutine = dec_func
func = None func = None