bring back tool calling agent

deepset-ai · Nov 15, 2024 · c3d8afe · c3d8afe
1 parent ce70375
commit c3d8afe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/notebooks/swarm.ipynb b/notebooks/swarm.ipynb
@@ -241,8 +241,8 @@
    "outputs": [],
    "source": [
     "@dataclass\n",
-    "class SwarmAgent:\n",
-    "    name: str = \"SwarmAgent\"\n",
+    "class ToolCallingAgent:\n",
+    "    name: str = \"ToolCallingAgent\"\n",
     "    llm: object = OpenAIChatGenerator()\n",
     "    instructions: str = \"You are a helpful Agent\"\n",
     "    functions: list[Callable] = field(default_factory=list)\n",
@@ -291,7 +291,7 @@
     "    return f\"report: refund succeeded for {item_name} - refund id: {random.randint(0,10000)}\"\n",
     "\n",
     "\n",
-    "refund_agent = SwarmAgent(\n",
+    "refund_agent = ToolCallingAgent(\n",
     "    name=\"Refund Agent\",\n",
     "    instructions=(\n",
     "        \"You are a refund agent. \"\n",