WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-08 18:57:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Nuno Campos
2024-12-04 15:38:15 -08:00
parent 2e9aea6fc8
commit a443b3b256
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/pregel/call.py
+7 -2
View File
@@ -1,3 +1,4 @@
import asyncio
import sys
import types
from typing import Any, Callable, Optional
@@ -111,7 +112,9 @@ def get_runnable_for_func(func: Callable[..., Any]) -> RunnableSeq:
return CACHE[func]
elif not _lookup_module_and_qualname(func):
return RunnableSeq(
RunnableCallable(func, trace=False),
RunnableCallable(None, func, trace=False)
if asyncio.iscoroutinefunction(func)
else RunnableCallable(func, trace=False),
ChannelWrite([ChannelWriteEntry(RETURN)]),
name=func.__name__,
)
@@ -119,7 +122,9 @@ def get_runnable_for_func(func: Callable[..., Any]) -> RunnableSeq:
return CACHE.setdefault(
func,
RunnableSeq(
RunnableCallable(func, trace=False),
RunnableCallable(None, func, trace=False)
if asyncio.iscoroutinefunction(func)
else RunnableCallable(func, trace=False),
ChannelWrite([ChannelWriteEntry(RETURN)]),
name=func.__name__,
),