diff --git a/.copier-answers.yml b/.copier-answers.yml
index 99665ee..ca88e2a 100644
--- a/.copier-answers.yml
+++ b/.copier-answers.yml
@@ -8,7 +8,7 @@ has_settings: true
 kind: frontend
 labextension_name: @jupyterlite/ai
 project_short_description: Codestral AI code completions for JupyterLab
-python_name: jupyterlab_codestral
+python_name: jupyterlite_ai
 repository: https://github.com/jupyterlite/jupyterlab-codestral
 test: false
 
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index aefe77a..7314efd 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -45,13 +45,13 @@ jobs:
 
         pip install build
         python -m build
-        pip uninstall -y "jupyterlab_codestral" jupyterlab
+        pip uninstall -y "jupyterlite_ai" jupyterlab
 
     - name: Upload extension packages
       uses: actions/upload-artifact@v4
       with:
         name: extension-artifacts
-        path: dist/jupyterlab_codestral*
+        path: dist/jupyterlite_ai*
         if-no-files-found: error
 
   test_isolated:
@@ -74,7 +74,7 @@ jobs:
         sudo rm -rf $(which node)
         sudo rm -rf $(which node)
 
-        pip install "jupyterlab>=4.0.0,<5" jupyterlab_codestral*.whl
+        pip install "jupyterlab>=4.0.0,<5" jupyterlite_ai*.whl
 
 
         jupyter labextension list
diff --git a/.github/workflows/check-release.yml b/.github/workflows/check-release.yml
index 12da1dd..4f65966 100644
--- a/.github/workflows/check-release.yml
+++ b/.github/workflows/check-release.yml
@@ -26,5 +26,5 @@ jobs:
       - name: Upload Distributions
         uses: actions/upload-artifact@v4
         with:
-          name: jupyterlab_codestral-releaser-dist-${{ github.run_number }}
+          name: jupyterlite_ai-releaser-dist-${{ github.run_number }}
           path: .jupyter_releaser_checkout/dist
diff --git a/.gitignore b/.gitignore
index 765885a..d8d4df1 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,9 +7,9 @@ node_modules/
 *.egg-info/
 .ipynb_checkpoints
 *.tsbuildinfo
-jupyterlab_codestral/labextension
+jupyterlite_ai/labextension
 # Version file is handled by hatchling
-jupyterlab_codestral/_version.py
+jupyterlite_ai/_version.py
 
 # Created by https://www.gitignore.io/api/python
 # Edit at https://www.gitignore.io/?templates=python
diff --git a/.prettierignore b/.prettierignore
index 9f5b04e..ca944b9 100644
--- a/.prettierignore
+++ b/.prettierignore
@@ -3,4 +3,4 @@ node_modules
 **/lib
 **/package.json
 !/package.json
-jupyterlab_codestral
+jupyterlite_ai
diff --git a/README.md b/README.md
index a6b4273..94f47c4 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-# jupyterlab-codestral
+# jupyterlite-ai
 
 [![Github Actions Status](https://github.com/jupyterlite/jupyterlab-codestral/workflows/Build/badge.svg)](https://github.com/jupyterlite/jupyterlab-codestral/actions/workflows/build.yml)
 [![lite-badge](https://jupyterlite.rtfd.io/en/latest/_static/badge.svg)](https://jupyterlite.github.io/jupyterlab-codestral/lab/index.html)
@@ -34,7 +34,7 @@ See the [Usage](#usage) section below for more information on how to provide you
 To install the extension, execute:
 
 ```bash
-pip install jupyterlab-codestral
+pip install jupyterlite-ai
 ```
 
 # Usage
@@ -56,7 +56,7 @@ pip install jupyterlab-codestral
 To remove the extension, execute:
 
 ```bash
-pip uninstall jupyterlab-codestral
+pip uninstall jupyterlite-ai
 ```
 
 ## Contributing
@@ -100,7 +100,7 @@ jupyter lab build --minimize=False
 ### Development uninstall
 
 ```bash
-pip uninstall jupyterlab-codestral
+pip uninstall jupyterlite-ai
 ```
 
 In development mode, you will also need to remove the symlink created by `jupyter labextension develop`
diff --git a/RELEASE.md b/RELEASE.md
index a38f1a2..0e2660b 100644
--- a/RELEASE.md
+++ b/RELEASE.md
@@ -1,4 +1,4 @@
-# Making a new release of jupyterlab_codestral
+# Making a new release of jupyterlite_ai
 
 The extension can be published to `PyPI` and `npm` manually or using the [Jupyter Releaser](https://github.com/jupyter-server/jupyter_releaser).
 
diff --git a/install.json b/install.json
index 8a9f0e7..c7bdcb3 100644
--- a/install.json
+++ b/install.json
@@ -1,5 +1,5 @@
 {
   "packageManager": "python",
-  "packageName": "jupyterlab_codestral",
-  "uninstallInstructions": "Use your Python package manager (pip, conda, etc.) to uninstall the package jupyterlab_codestral"
+  "packageName": "jupyterlite_ai",
+  "uninstallInstructions": "Use your Python package manager (pip, conda, etc.) to uninstall the package jupyterlite_ai"
 }
diff --git a/jupyterlab_codestral/__init__.py b/jupyterlite_ai/__init__.py
similarity index 85%
rename from jupyterlab_codestral/__init__.py
rename to jupyterlite_ai/__init__.py
index 62dac37..62a7a63 100644
--- a/jupyterlab_codestral/__init__.py
+++ b/jupyterlite_ai/__init__.py
@@ -5,7 +5,7 @@
     # in editable mode with pip. It is highly recommended to install
     # the package from a stable release or in editable mode: https://pip.pypa.io/en/stable/topics/local-project-installs/#editable-installs
     import warnings
-    warnings.warn("Importing 'jupyterlab_codestral' outside a proper installation.")
+    warnings.warn("Importing 'jupyterlite_ai' outside a proper installation.")
     __version__ = "dev"
 
 
diff --git a/package.json b/package.json
index acb7e7f..70695ab 100644
--- a/package.json
+++ b/package.json
@@ -36,7 +36,7 @@
         "clean": "jlpm clean:lib",
         "clean:lib": "rimraf lib tsconfig.tsbuildinfo",
         "clean:lintcache": "rimraf .eslintcache .stylelintcache",
-        "clean:labextension": "rimraf jupyterlab_codestral/labextension jupyterlab_codestral/_version.py",
+        "clean:labextension": "rimraf jupyterlite_ai/labextension jupyterlite_ai/_version.py",
         "clean:all": "jlpm clean:lib && jlpm clean:labextension && jlpm clean:lintcache",
         "eslint": "jlpm eslint:check --fix",
         "eslint:check": "eslint . --cache --ext .ts,.tsx",
@@ -100,7 +100,7 @@
     },
     "jupyterlab": {
         "extension": true,
-        "outputDir": "jupyterlab_codestral/labextension",
+        "outputDir": "jupyterlite_ai/labextension",
         "schemaDir": "schema"
     },
     "eslintIgnore": [
diff --git a/pyproject.toml b/pyproject.toml
index 724dba0..45f1c50 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["hatchling>=1.5.0", "jupyterlab>=4.0.0,<5", "hatch-nodejs-version>=0
 build-backend = "hatchling.build"
 
 [project]
-name = "jupyterlab_codestral"
+name = "jupyterlite_ai"
 readme = "README.md"
 license = { file = "LICENSE" }
 requires-python = ">=3.8"
@@ -33,24 +33,24 @@ source = "nodejs"
 fields = ["description", "authors", "urls"]
 
 [tool.hatch.build.targets.sdist]
-artifacts = ["jupyterlab_codestral/labextension"]
+artifacts = ["jupyterlite_ai/labextension"]
 exclude = [".github", "binder"]
 
 [tool.hatch.build.targets.wheel.shared-data]
-"jupyterlab_codestral/labextension" = "share/jupyter/labextensions/@jupyterlite/ai"
+"jupyterlite_ai/labextension" = "share/jupyter/labextensions/@jupyterlite/ai"
 "install.json" = "share/jupyter/labextensions/@jupyterlite/ai/install.json"
 
 [tool.hatch.build.hooks.version]
-path = "jupyterlab_codestral/_version.py"
+path = "jupyterlite_ai/_version.py"
 
 [tool.hatch.build.hooks.jupyter-builder]
 dependencies = ["hatch-jupyter-builder>=0.5"]
 build-function = "hatch_jupyter_builder.npm_builder"
 ensured-targets = [
-    "jupyterlab_codestral/labextension/static/style.js",
-    "jupyterlab_codestral/labextension/package.json",
+    "jupyterlite_ai/labextension/static/style.js",
+    "jupyterlite_ai/labextension/package.json",
 ]
-skip-if-exists = ["jupyterlab_codestral/labextension/static/style.js"]
+skip-if-exists = ["jupyterlite_ai/labextension/static/style.js"]
 
 [tool.hatch.build.hooks.jupyter-builder.build-kwargs]
 build_cmd = "build:prod"
@@ -60,7 +60,7 @@ npm = ["jlpm"]
 build_cmd = "install:extension"
 npm = ["jlpm"]
 source_dir = "src"
-build_dir = "jupyterlab_codestral/labextension"
+build_dir = "jupyterlite_ai/labextension"
 
 [tool.jupyter-releaser.options]
 version_cmd = "hatch version"