WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-28 20:45:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

Disable recursing on runnables for task/entrypoint decorated funcs

Browse Source
This commit is contained in:
Nuno Campos
2025-01-21 15:36:09 -08:00
parent 0177565c6b
commit 12ae297194
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/pregel/call.py
+11 -2
View File
@@ -144,7 +144,9 @@ def get_runnable_for_entrypoint(func: Callable[..., Any]) -> RunnableSeq:
return CACHE[key]
else:
if is_async_callable(func):
run = RunnableCallable(None, func, name=func.__name__, trace=False)
run = RunnableCallable(
None, func, name=func.__name__, trace=False, recurse=False
)
else:
afunc = functools.update_wrapper(
functools.partial(run_in_executor, None, func), func
@@ -154,6 +156,7 @@ def get_runnable_for_entrypoint(func: Callable[..., Any]) -> RunnableSeq:
afunc,
name=func.__name__,
trace=False,
recurse=False,
)
seq = RunnableSeq(
run,
@@ -172,7 +175,12 @@ def get_runnable_for_task(func: Callable[..., Any]) -> RunnableSeq:
else:
if is_async_callable(func):
run = RunnableCallable(
None, func, explode_args=True, name=func.__name__, trace=False
None,
func,
explode_args=True,
name=func.__name__,
trace=False,
recurse=False,
)
else:
run = RunnableCallable(
@@ -181,6 +189,7 @@ def get_runnable_for_task(func: Callable[..., Any]) -> RunnableSeq:
explode_args=True,
name=func.__name__,
trace=False,
recurse=False,
)
seq = RunnableSeq(
run,