From f339b27dfa8b2ddf9d7a2cc2a7e326d5e1cc604a Mon Sep 17 00:00:00 2001
From: Daulet Amirkhanov <damirkhanov01@gmail.com>
Date: Wed, 27 Aug 2025 15:37:29 +0100
Subject: [PATCH] refactor: update Jupyter notebooks to prevent kernel exit
 during GitHub Actions

---
 notebooks/cognee_demo.ipynb        | 9 ++++++++-
 notebooks/cognee_simple_demo.ipynb | 9 ++++++++-
 notebooks/ontology_demo.ipynb      | 9 ++++++++-
 3 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/notebooks/cognee_demo.ipynb b/notebooks/cognee_demo.ipynb
index 0e6da5161..51eeab560 100644
--- a/notebooks/cognee_demo.ipynb
+++ b/notebooks/cognee_demo.ipynb
@@ -1808,8 +1808,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# Only exit in interactive mode, not during GitHub Actions\n",
     "import os\n",
-    "os._exit(0)"
+    "\n",
+    "# Skip exit if we're running in GitHub Actions\n",
+    "if not os.environ.get('GITHUB_ACTIONS'):\n",
+    "    print(\"Exiting kernel to clean up resources...\")\n",
+    "    os._exit(0)\n",
+    "else:\n",
+    "    print(\"Skipping kernel exit - running in GitHub Actions\")"
    ]
   },
   {
diff --git a/notebooks/cognee_simple_demo.ipynb b/notebooks/cognee_simple_demo.ipynb
index e203ce165..6ca42ada5 100644
--- a/notebooks/cognee_simple_demo.ipynb
+++ b/notebooks/cognee_simple_demo.ipynb
@@ -896,8 +896,15 @@
     }
    ],
    "source": [
+    "# Only exit in interactive mode, not during GitHub Actions\n",
     "import os\n",
-    "os._exit(0)"
+    "\n",
+    "# Skip exit if we're running in GitHub Actions\n",
+    "if not os.environ.get('GITHUB_ACTIONS'):\n",
+    "    print(\"Exiting kernel to clean up resources...\")\n",
+    "    os._exit(0)\n",
+    "else:\n",
+    "    print(\"Skipping kernel exit - running in GitHub Actions\")"
    ]
   },
   {
diff --git a/notebooks/ontology_demo.ipynb b/notebooks/ontology_demo.ipynb
index a38a3a8bc..e48d8467d 100644
--- a/notebooks/ontology_demo.ipynb
+++ b/notebooks/ontology_demo.ipynb
@@ -1149,8 +1149,15 @@
     }
    ],
    "source": [
+    "# Only exit in interactive mode, not during GitHub Actions\n",
     "import os\n",
-    "os._exit(0)"
+    "\n",
+    "# Skip exit if we're running in GitHub Actions\n",
+    "if not os.environ.get('GITHUB_ACTIONS'):\n",
+    "    print(\"Exiting kernel to clean up resources...\")\n",
+    "    os._exit(0)\n",
+    "else:\n",
+    "    print(\"Skipping kernel exit - running in GitHub Actions\")"
    ]
   }
  ],