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

Changes to match `RunnableSequences`

@eyurtsev
pull/10790/head
Jacob Lee 11 months ago committed by GitHub
parent 931b292126
commit bbc3fe259b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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,
)

Loading…
Cancel
Save