Try to exit cleanly in CI

Author: abrookins

python-recipes/agents/03_memory_agent.ipynb

@@ -13,7 +13,7 @@
     "2. Long-term memory storage and retrieval using RedisVL\n",
     "3. Manually storing and retrieving long-term memory vs. exposing tool access (AKA function-calling)\n",
     "4. Managing conversation history size with summarization\n",
-    "5. Memory consolidation and decay\n",
+    "5. Memory consolidation\n",
     "\n",
     "## Let's Begin!\n",
     "<a href=\"https://colab.research.google.com/github/redis-developer/redis-ai-resources/blob/main/python-recipes/agents/03_memory_agent.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@@ -1539,36 +1539,9 @@
     }
    ],
    "source": [
-    "%pip install ipywidgets\n",
-    "\n",
     "import threading\n",
     "\n",
     "\n",
-    "import ipywidgets as widgets\n",
-    "from IPython.display import display, Javascript\n",
-    "\n",
-    "\n",
-    "def popup(text):\n",
-    "    display(\n",
-    "        Javascript(f\"\"\"\n",
-    "    require(['base/js/dialog'], function(dialog) {{\n",
-    "        dialog.modal({{\n",
-    "            title: 'Message',\n",
-    "            body: '{text}',\n",
-    "            buttons: {{'OK': {{'class': 'btn-primary'}}}}\n",
-    "        }});\n",
-    "    }});\n",
-    "    \"\"\")\n",
-    "    )\n",
-    "\n",
-    "\n",
-    "# Example usage:\n",
-    "popup(\"This is a modal message!\")\n",
-    "\n",
-    "# To show the modal again with different content, call popup() again\n",
-    "popup(\"Another message!\")\n",
-    "\n",
-    "\n",
     "def main(thread_id: str = \"book_flight\", user_id: str = \"demo_user\"):\n",
     "    \"\"\"Main interaction loop for the travel agent\"\"\"\n",
     "    print(\"Welcome to the Travel Assistant! (Type 'exit' to quit)\")\n",
@@ -1639,8 +1612,12 @@
     "            state[\"messages\"].append(AIMessage(content=error_message))\n",
     "\n",
     "\n",
-    "user_id = input(\"Enter a user ID: \") or \"demo_user\"\n",
-    "thread_id = input(\"Enter a thread ID: \") or \"demo_thread\"\n",
+    "try:\n",
+    "    user_id = input(\"Enter a user ID: \") or \"demo_user\"\n",
+    "    thread_id = input(\"Enter a thread ID: \") or \"demo_thread\"\n",
+    "except Exception:\n",
+    "    # If we're running in CI, we don't have a terminal to input from, so just exit\n",
+    "    exit()\n",
     "\n",
     "main(thread_id, user_id)"
    ]