From 6791756b41c0da8fa53f396071e7f5cf9ad5160c Mon Sep 17 00:00:00 2001 From: Yuxiang Lin <43810861+theta-lin@users.noreply.github.com> Date: Wed, 26 Jun 2024 00:24:41 +0800 Subject: [PATCH] Update output parsing for tool runner in notebook example of using agent (#14381) * Update output parsing for tool runner in notebook example of using agent * lint --------- Co-authored-by: Andrei Fajardo --- .../agent_around_query_pipeline_with_HyDE_for_PDFs.ipynb | 4 +++- 1 file changed, 3 insertions(+), 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 index 14fbc98308378..63f5549662f4d 100644 --- 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",