From bbc3fe259b01a1d5c53cf94675afbbb40a33bfde Mon Sep 17 00:00:00 2001
From: Jacob Lee <jacoblee93@gmail.com>
Date: Tue, 19 Sep 2023 16:38:08 -0700
Subject: [PATCH] Start RunnableBranch callback tags with 1 instead of 0
 (#10755)

Changes to match `RunnableSequences`

@eyurtsev
---
 libs/langchain/langchain/schema/runnable/base.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/libs/langchain/langchain/schema/runnable/base.py b/libs/langchain/langchain/schema/runnable/base.py
index d969e4ab6e..6a850ddb4e 100644
--- a/libs/langchain/langchain/schema/runnable/base.py
+++ b/libs/langchain/langchain/schema/runnable/base.py
@@ -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,
                     )