diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 70a9349..a0c718d 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -18,7 +18,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: ['3.9', '3.10', '3.11']
+        python-version: ['3.10', '3.11', '3.12']
 
     steps:
     - name: Checkout
@@ -47,7 +47,7 @@ jobs:
         rm -rf "jupyter_ai_tools"
 
     - name: Upload artifact
-      if: matrix.os == 'ubuntu-latest' && matrix.python-version == '3.9'
+      if: matrix.os == 'ubuntu-latest' && matrix.python-version == '3.10'
       uses: actions/upload-artifact@v4
       with:
         name: my_server_extension-sdist
@@ -99,7 +99,7 @@ jobs:
     - name: Install Python
       uses: actions/setup-python@v5
       with:
-        python-version: '3.9'
+        python-version: '3.10'
         architecture: 'x64'
     - uses: actions/download-artifact@v4
       with:
diff --git a/jupyter_ai_tools/__init__.py b/jupyter_ai_tools/__init__.py
index 987cc59..51f3ce8 100644
--- a/jupyter_ai_tools/__init__.py
+++ b/jupyter_ai_tools/__init__.py
@@ -1,5 +1,17 @@
+from .toolkits.code_execution import toolkit as codeexec_toolkit
+from .toolkits.file_system import toolkit as fs_toolkit
+from .toolkits.git import toolkit as git_toolkit
+from .toolkits.notebook import toolkit as notebook_toolkit
+
 __version__ = "0.2.1"
 
+__all__ = [
+    "fs_toolkit",
+    "codeexec_toolkit",
+    "git_toolkit",
+    "notebook_toolkit",
+]
+
 
 def _jupyter_server_extension_points():
     return [{"module": "jupyter_ai_tools"}]