Zsh: widget for automatic usage of uv with the pip command

.config/nvim/lua/dictionaries/dictionary_names.txt

@@ -14,6 +14,7 @@ Arrabbiata
 Asiahung
 Aule
 Aules
+Autoencoder
 Babybel
 Baez
 Barros
@@ -120,6 +121,7 @@ Kingdoms
 Kluge
 KohakuHub
 Kratom
+KreativAkademie
 Kreyenhop
 Kubitschek
 Käte
@@ -239,6 +241,7 @@ Tornow
 TrackPoint
 Traefik
 UDEV
+UMAP
 Umbar
 V1del
 VIM
@@ -254,6 +257,7 @@ WinterAkademie
 Writhing
 X.Org
 Years
+Yorma
 Ziegeleiweg
 Zimin
 and

.config/zsh/.zshrc

@@ -152,9 +152,24 @@ uvshim() {
 }
 
 python() { uvshim python "$@"; }
+
 python3() { uvshim python3 "$@"; }
-pip() { uvshim pip "$@"; }
-pytest() { uvshim pytest "$@"; }
+
+pip() {
+    local uv_root
+    uv_root="$(find_python_root)"
+
+    if [[ -n "$uv_root" ]]; then
+        if [[ "$1" == "install" ]]; then
+            shift
+            uv add "$@"
+        else
+            uv pip "$@"
+        fi
+    else
+        command pip "$@"
+    fi
+}
 
 _find_venv_upwards() {
     local dir="$PWD"