Update 43_Building_a_Tool_Calling_Agent.ipynb

Author: sjrl

tutorials/43_Building_a_Tool_Calling_Agent.ipynb

@@ -217,29 +217,24 @@
     "search_pipeline.add_component(\"fetcher\", LinkContentFetcher(timeout=3, raise_on_failure=False, retry_attempts=2))\n",
     "search_pipeline.add_component(\"converter\", HTMLToDocument())\n",
     "search_pipeline.add_component(\n",
-    "    \"builder\",\n",
-    "    ChatPromptBuilder(\n",
-    "        template=[\n",
-    "            ChatMessage.from_user(\n",
-    "                \"\"\"\n",
-    "{% for doc in docs %}\n",
-    "  {% if doc.content %}\n",
+    "    \"output_adapter\",\n",
+    "    OutputAdapter(\n",
+    "        template=\"\"\"\n",
+    "{%- for doc in docs -%}\n",
+    "  {%- if doc.content -%}\n",
     "  <search-result url=\"{{ doc.meta.url }}\">\n",
     "  {{ doc.content|truncate(25000) }}\n",
     "  </search-result>\n",
-    "  {% endif %}\n",
-    "{% endfor %}\n",
+    "  {%- endif -%}\n",
+    "{%- endfor -%}\n",
     "\"\"\"\n",
-    "            )\n",
-    "        ],\n",
-    "        variables=[\"docs\"],\n",
-    "        required_variables=[\"docs\"],\n",
+    "        output_type=str,\n",
     "    ),\n",
     ")\n",
     "\n",
     "search_pipeline.connect(\"search.links\", \"fetcher.urls\")\n",
     "search_pipeline.connect(\"fetcher.streams\", \"converter.sources\")\n",
-    "search_pipeline.connect(\"converter.documents\", \"builder.docs\")"
+    "search_pipeline.connect(\"converter.documents\", \"output_adapter.docs\")"
    ]
   },
   {
@@ -268,7 +263,11 @@
     "from haystack.components.agents import Agent\n",
     "from haystack.components.generators.chat import OpenAIChatGenerator\n",
     "\n",
-    "search_component = SuperComponent(pipeline=search_pipeline)\n",
+    "search_component = SuperComponent(\n",
+    "    pipeline=search_pipeline,\n",
+    "    input_mapping={\"query\": [\"search.query\"]},\n",
+    "    output_mapping={\"output_adapter.output\": \"search_result\"}\n",
+    ")\n",
     "search_tool = ComponentTool(\n",
     "    name=\"search\", description=\"Use this tool to search for information on the internet.\", component=search_component\n",
     ")\n",