Simplify concurrent flow

Author: onlined

examples/How_to_use_guardrails.ipynb

@@ -134,21 +134,13 @@
     "    topical_guardrail_task = asyncio.create_task(topical_guardrail(user_request))\n",
     "    chat_task = asyncio.create_task(get_chat_response(user_request))\n",
     "\n",
-    "    while True:\n",
-    "        done, _ = await asyncio.wait(\n",
-    "            [topical_guardrail_task, chat_task], return_when=asyncio.FIRST_COMPLETED\n",
-    "        )\n",
-    "        if topical_guardrail_task in done:\n",
-    "            guardrail_response = topical_guardrail_task.result()\n",
-    "            if guardrail_response == \"not_allowed\":\n",
-    "                chat_task.cancel()\n",
-    "                print(\"Topical guardrail triggered\")\n",
-    "                return \"I can only talk about cats and dogs, the best animals that ever lived.\"\n",
-    "            elif chat_task in done:\n",
-    "                chat_response = chat_task.result()\n",
-    "                return chat_response\n",
-    "        else:\n",
-    "            await asyncio.sleep(0.1)  # sleep for a bit before checking the tasks again"
+    "    guardrail_response = await topical_guardrail_task\n",
+    "    if guardrail_response == \"not_allowed\":\n",  
+    "        chat_task.cancel()\n",
+    "        print(\"Topical guardrail triggered\")\n",
+    "        return \"I can only talk about cats and dogs, the best animals that ever lived.\"\n",
+    "    chat_response = await chat_task\n",
+    "    return chat_response\n",
    ]
   },
   {