Fixes run-llama#11389 (agent_react_multimodal_step.py bug) (run-llama…

…#12566)
chrisalexiuk-nvidia · Apr 25, 2024 · e4a9095 · e4a9095
1 parent 677274c
commit e4a9095
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/llama-index-core/llama_index/core/agent/react_multimodal/step.py b/llama-index-core/llama_index/core/agent/react_multimodal/step.py
@@ -388,9 +388,9 @@ def _run_step(
         # an intermediate step in the middle
         if step.input is not None:
             self._add_user_step_to_reasoning(
-                step,
-                task.extra_state["new_memory"],
-                task.extra_state["current_reasoning"],
+                step=step,
+                memory=task.extra_state["new_memory"],
+                current_reasoning=task.extra_state["current_reasoning"],
                 verbose=self._verbose,
             )
         # TODO: see if we want to do step-based inputs
@@ -428,9 +428,9 @@ async def _arun_step(
         """Run step."""
         if step.input is not None:
             self._add_user_step_to_reasoning(
-                step,
-                task.extra_state["new_memory"],
-                task.extra_state["current_reasoning"],
+                step=step,
+                memory=task.extra_state["new_memory"],
+                current_reasoning=task.extra_state["current_reasoning"],
                 verbose=self._verbose,
             )
         # TODO: see if we want to do step-based inputs