Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-04 06:00:26 +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( expression_value = condition.invoke(
input, input,
config=patch_config( 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( return runnable.invoke(
input, input,
config=patch_config( 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( expression_value = await condition.ainvoke(
input, input,
config=patch_config( 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( return await runnable.ainvoke(
input, input,
config=patch_config( 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, **kwargs,
) )