From 33b0e222cab0c1d4098c580cd2575e7b62e0a309 Mon Sep 17 00:00:00 2001
From: UncleCode <unclecode@kidocode.com>
Date: Sun, 25 May 2025 16:50:56 +0800
Subject: [PATCH] Add Colab utilities and rename setup function for clarity

---
 crawl4ai/__init__.py | 9 ++++++++-
 crawl4ai/utils.py    | 2 +-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/crawl4ai/__init__.py b/crawl4ai/__init__.py
index 3ba22ece..2c6c2f29 100644
--- a/crawl4ai/__init__.py
+++ b/crawl4ai/__init__.py
@@ -66,6 +66,11 @@ from .deep_crawling import (
     DeepCrawlDecorator,
 )
 
+from .utils import (
+    start_colab_display_server,
+    setup_colab_environment
+)
+
 __all__ = [
     "AsyncLoggerBase",
     "AsyncLogger",
@@ -124,7 +129,9 @@ __all__ = [
     "Crawl4aiDockerClient",
     "ProxyRotationStrategy",
     "RoundRobinProxyStrategy",
-    "ProxyConfig"
+    "ProxyConfig",
+    "start_colab_display_server",
+    "setup_colab_environment",
 ]
 
 
diff --git a/crawl4ai/utils.py b/crawl4ai/utils.py
index 47fe165b..03ae659d 100644
--- a/crawl4ai/utils.py
+++ b/crawl4ai/utils.py
@@ -2914,7 +2914,7 @@ def start_colab_display_server():
 
 
 
-def setup_colab_environment_ipython():
+def setup_colab_environment():
     """
     Alternative setup using IPython magic commands
     """