Update output parsing for tool runner in notebook example of using ag…

…ent (#14381) * Update output parsing for tool runner in notebook example of using agent * lint --------- Co-authored-by: Andrei Fajardo <[email protected]>
run-llama · Jun 25, 2024 · 6791756 · 6791756
1 parent af3e409
commit 6791756
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docs/docs/examples/agent/agent_runner/agent_around_query_pipeline_with_HyDE_for_PDFs.ipynb b/docs/docs/examples/agent/agent_runner/agent_around_query_pipeline_with_HyDE_for_PDFs.ipynb
@@ -459,7 +459,9 @@
     "        tool_name=reasoning_step.action,\n",
     "        tool_input=reasoning_step.action_input,\n",
     "    )\n",
-    "    observation_step = ObservationReasoningStep(observation=str(tool_output))\n",
+    "    observation_step = ObservationReasoningStep(\n",
+    "        observation=str(tool_output[\"output\"])\n",
+    "    )\n",
     "    state[\"current_reasoning\"].append(observation_step)\n",
     "    # TODO: get output\n",
     "\n",