Zsh compdef para o apt-get install ou outros comandos de 2 partes

Question

Como posso usar um compdef para zsh completion de forma que uma função customizada escrita use um comando de 2 partes para completar?

Exemplo:

my_function () {
    sudo apt-get install $@
}

compdef my_function=<WHAT GOES HERE?>
compdef my_function=apt-get install # DOESN'T WORK
compdef my_function='apt-get install' # ALSO NO

Toda e qualquer ajuda seria apreciada, obrigado!