1 files changed, 11 insertions, 0 deletions
diff --git a/scripts/shell-integration/integration.zsh b/scripts/shell-integration/integration.zsh
new file mode 100644
index 0000000..2328bd6
--- /dev/null
+++ b/scripts/shell-integration/integration.zsh
@@ -0,0 +1,11 @@
+_aichat_zsh() {
+    if [[ -n "$BUFFER" ]]; then
+        local _old=$BUFFER
+        BUFFER+="⌛"
+        zle -I && zle redisplay
+        BUFFER=$(aichat -e "$_old")
+        zle end-of-line
+    fi
+}
+zle -N _aichat_zsh
+bindkey '\ee' _aichat_zsh
+\ No newline at end of file