github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-23 03:19:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Start RunnableBranch callback tags with 1 instead of 0 (#10755)

Browse Source 
Changes to match `RunnableSequences`

@eyurtsev
This commit is contained in:
Jacob Lee
2023-09-19 16:38:08 -07:00
committed by GitHub
parent 931b292126
commit bbc3fe259b
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/langchain/langchain/schema/runnable/base.py
View File

@@ -771,7 +771,8 @@ class RunnableBranch(Serializable, Runnable[Input, Output]):
expression_value = condition.invoke(
input,
config=patch_config(
config, callbacks=run_manager.get_child(tag=f"condition:{idx}")
config,
callbacks=run_manager.get_child(tag=f"condition:{idx + 1}"),
),
)
@@ -779,7 +780,8 @@ class RunnableBranch(Serializable, Runnable[Input, Output]):
return runnable.invoke(
input,
config=patch_config(
config, callbacks=run_manager.get_child(tag=f"branch:{idx}")
config,
callbacks=run_manager.get_child(tag=f"branch:{idx + 1}"),
),
)
@@ -813,7 +815,8 @@ class RunnableBranch(Serializable, Runnable[Input, Output]):
expression_value = await condition.ainvoke(
input,
config=patch_config(
config, callbacks=run_manager.get_child(tag=f"condition:{idx}")
config,
callbacks=run_manager.get_child(tag=f"condition:{idx + 1}"),
),
)
@@ -821,7 +824,8 @@ class RunnableBranch(Serializable, Runnable[Input, Output]):
return await runnable.ainvoke(
input,
config=patch_config(
config, callbacks=run_manager.get_child(tag=f"branch:{idx}")
config,
callbacks=run_manager.get_child(tag=f"branch:{idx + 1}"),
),
**kwargs,
)