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

Fix non str values in cond edges

Browse Source
This commit is contained in:
Nuno Campos
2024-04-11 11:52:20 -07:00
parent ce6989daee
commit 0015cb5f3a
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/graph/graph.py
+4 -4
View File
@@ -62,10 +62,10 @@ class Branch(NamedTuple):
config: RunnableConfig,
*,
reader: Optional[Callable[[], Any]],
writer: Callable[[str], Optional[Runnable]],
writer: Callable[[list[str]], Optional[Runnable]],
) -> Runnable:
result = self.condition.invoke(reader(config) if reader else input, config)
if isinstance(result, str):
if not isinstance(result, list):
result = [result]
if self.ends:
destinations = [self.ends[r] for r in result]
@@ -79,12 +79,12 @@ class Branch(NamedTuple):
config: RunnableConfig,
*,
reader: Optional[Callable[[], Any]],
writer: Callable[[str], Optional[Runnable]],
writer: Callable[[list[str]], Optional[Runnable]],
) -> Runnable:
result = await self.condition.ainvoke(
reader(config) if reader else input, config
)
if isinstance(result, str):
if not isinstance(result, list):
result = [result]
if self.ends:
destinations = [self.ends[r] for r in result]