complete -c aimo -s m -l model -x -a "(aimo --list-models)" -d 'Select a LLM model' -r
complete -c aimo -l prompt -d 'Use the system prompt'
complete -c aimo -s r -l role -x -a "(aimo --list-roles)" -d 'Select a role' -r
complete -c aimo -s s -l session -x  -a"(aimo --list-sessions)" -d 'Start or join a session' -r
complete -c aimo -s f -l file -d 'Include files with the message' -r -F
complete -c aimo -s w -l wrap -d 'Control text wrapping (no, auto, <max-width>)'
complete -c aimo -l save-session -d 'Forces the session to be saved'
complete -c aimo -l serve -d 'Serve the LLM API and WebAPP'
complete -c aimo -s e -l execute -d 'Execute commands in natural language'
complete -c aimo -s c -l code -d 'Output code only'
complete -c aimo -s H -l no-highlight -d 'Turn off syntax highlighting'
complete -c aimo -s S -l no-stream -d 'Turns off stream mode'
complete -c aimo -l light-theme -d 'Use light theme'
complete -c aimo -l dry-run -d 'Display the message without sending it'
complete -c aimo -l info -d 'Display information'
complete -c aimo -l list-models -d 'List all available models'
complete -c aimo -l list-roles -d 'List all available roles'
complete -c aimo -l list-sessions -d 'List all available sessions'
complete -c aimo -s h -l help -d 'Print help'
complete -c aimo -s V -l version -d 'Print version'