From 7ae58e73b00d7955d9395a6a4a05ef82b6110697 Mon Sep 17 00:00:00 2001
From: aviu16 <162624394+aviu16@users.noreply.github.com>
Date: Mon, 16 Feb 2026 20:46:51 -0500
Subject: [PATCH] fix(git prompt): register async handler on wrapper invocation

---
 lib/git.zsh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/lib/git.zsh b/lib/git.zsh
index 8d38f3268..2bab418e2 100644
--- a/lib/git.zsh
+++ b/lib/git.zsh
@@ -147,12 +147,16 @@ local _style
 if zstyle -t ':omz:alpha:lib:git' async-prompt \
   || { is-at-least 5.0.6 && zstyle -T ':omz:alpha:lib:git' async-prompt }; then
   function git_prompt_info() {
+    # Support wrapper calls like $(_my_git_prompt_info) by lazily registering
+    # the async handler on first invocation.
+    _omz_register_handler _omz_git_prompt_info
     if [[ -n "${_OMZ_ASYNC_OUTPUT[_omz_git_prompt_info]}" ]]; then
       echo -n "${_OMZ_ASYNC_OUTPUT[_omz_git_prompt_info]}"
     fi
   }
 
   function git_prompt_status() {
+    _omz_register_handler _omz_git_prompt_status
     if [[ -n "${_OMZ_ASYNC_OUTPUT[_omz_git_prompt_status]}" ]]; then
       echo -n "${_OMZ_ASYNC_OUTPUT[_omz_git_prompt_status]}"
     fi