github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-29 18:08:36 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions
Show Stats 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,
)