feat: add `--prompt` option (#454)

5 files changed, 9 insertions, 1 deletions

diff --git a/scripts/completions/aichat.bash b/scripts/completions/aichat.bash
index f0b9bc3..cb9b708 100644
--- a/scripts/completions/aichat.bash
+++ b/scripts/completions/aichat.bash
@@ -19,7 +19,7 @@ _aichat() {
 
     case "${cmd}" in
         aichat)
-            opts="-m -r -s -e -c -f -H -S -w -h -V --model --role --session --save-session --serve --execute --code --file --no-highlight --no-stream --wrap --light-theme --dry-run --info --list-models --list-roles --list-sessions --help --version"
+            opts="-m -r -s -e -c -f -H -S -w -h -V --model --prompt --role --session --save-session --serve --execute --code --file --no-highlight --no-stream --wrap --light-theme --dry-run --info --list-models --list-roles --list-sessions --help --version"
             if [[ ${cur} == -* || ${COMP_CWORD} -eq 1 ]] ; then
                 COMPREPLY=( $(compgen -W "${opts}" -- "${cur}") )
                 return 0
@@ -49,6 +49,10 @@ _aichat() {
                     COMPREPLY=($(compgen -W "$("$1" --list-models)" -- "${cur}"))
                     return 0
                     ;;
+                --prompt)
+                    COMPREPLY=($(compgen -f "${cur}"))
+                    return 0
+                    ;;
                 -r|--role)
                     COMPREPLY=($(compgen -W "$("$1" --list-roles)" -- "${cur}"))
                     return 0
diff --git a/scripts/completions/aichat.fish b/scripts/completions/aichat.fish
index 47f3c6b..9ff6538 100644
--- a/scripts/completions/aichat.fish
+++ b/scripts/completions/aichat.fish
@@ -1,4 +1,5 @@
 complete -c aichat -s m -l model -x -a "(aichat --list-models)" -d 'Select a LLM model' -r
+complete -c aichat -l prompt -d 'Use the system prompt'
 complete -c aichat -s r -l role -x -a "(aichat --list-roles)" -d 'Select a role' -r
 complete -c aichat -s s -l session -x  -a"(aichat --list-sessions)" -d 'Start or join a session' -r
 complete -c aichat -s f -l file -d 'Include files with the message' -r -F
diff --git a/scripts/completions/aichat.nu b/scripts/completions/aichat.nu
index a8ac4bf..3b6ed9a 100644
--- a/scripts/completions/aichat.nu
+++ b/scripts/completions/aichat.nu
@@ -25,6 +25,7 @@ module completions {
   # All-in-one chat and copilot CLI that integrates 10+ AI platforms
   export extern aichat [
     --model(-m): string@"nu-complete aichat model"    # Select a LLM model
+    --prompt                                          # Use the system prompt
     --role(-r): string@"nu-complete aichat role"      # Select a role
     --session(-s): string@"nu-complete aichat role"   # Start or join a session
     --save-session                                    # Forces the session to be saved
diff --git a/scripts/completions/aichat.ps1 b/scripts/completions/aichat.ps1
index c3e8e96..10b986d 100644
--- a/scripts/completions/aichat.ps1
+++ b/scripts/completions/aichat.ps1
@@ -22,6 +22,7 @@ Register-ArgumentCompleter -Native -CommandName 'aichat' -ScriptBlock {
         'aichat' {
             [CompletionResult]::new('-m', '-m', [CompletionResultType]::ParameterName, 'Select a LLM model')
             [CompletionResult]::new('--model', '--model', [CompletionResultType]::ParameterName, 'Select a LLM model')
+            [CompletionResult]::new('--prompt', '--prompt', [CompletionResultType]::ParameterName, 'Use the system prompt')
             [CompletionResult]::new('-r', '-r', [CompletionResultType]::ParameterName, 'Select a role')
             [CompletionResult]::new('--role', '--role', [CompletionResultType]::ParameterName, 'Select a role')
             [CompletionResult]::new('-s', '-s', [CompletionResultType]::ParameterName, 'Start or join a session')
diff --git a/scripts/completions/aichat.zsh b/scripts/completions/aichat.zsh
index 378caf6..bbcc02b 100644
--- a/scripts/completions/aichat.zsh
+++ b/scripts/completions/aichat.zsh
@@ -16,6 +16,7 @@ _aichat() {
     local context curcontext="$curcontext" state line
      local common=(
 '-m+[Select a LLM model]:MODEL:->models' \
+'--prompt=[Use the system prompt]:PROMPT: ' \
 '--model=[Select a LLM model]:MODEL:->models' \
 '-r+[Select a role]:ROLE:->roles' \
 '--role=[Select a role]:ROLE:->roles' \