fix

Author: cbornet

libs/core/langchain_core/messages/ai.py

@@ -402,7 +402,7 @@ def add_ai_message_chunks(
                 name=rtc.get("name"),
                 args=rtc.get("args"),
                 index=rtc.get("index"),
-                id=rtc.get("chunk_id"),
+                id=rtc.get("id"),
             )
             for rtc in raw_tool_calls
         ]

libs/core/langchain_core/runnables/base.py

@@ -3079,7 +3079,7 @@ async def ainvoke(
                     if i == 0:
                         part = functools.partial(step.ainvoke, input_, config, **kwargs)
                     else:
-                        part = functools.partial(step.ainvoke, input, config)
+                        part = functools.partial(step.ainvoke, input_, config)
                     input_ = await coro_with_context(part(), context, create_task=True)
             # finish the root run
         except BaseException as e:

libs/core/tests/unit_tests/runnables/test_fallbacks.py

@@ -267,6 +267,7 @@ def _generate(_: Iterator) -> Iterator[str]:
 def _generate_immediate_error(_: Iterator) -> Iterator[str]:
     msg = "immmediate error"
     raise ValueError(msg)
+    yield ""
 
 
 def _generate_delayed_error(_: Iterator) -> Iterator[str]: