diff --git a/ai.sh b/ai.sh
index 686b353..d8a9054 100755
--- a/ai.sh
+++ b/ai.sh
@@ -29,6 +29,9 @@ done
 
 function main() {
   case $command in
+    "edit")
+      vim "${scripts_dir}/ai.sh"
+      ;;
     "open-webui")
       . $HOME/open-webui/.venv/bin/activate
       open-webui serve