From f9847475b8eae99a70f16dc63bac67da2632474a Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 13 Jun 2024 19:41:54 +0800
Subject: feat: add rag and bot related cli options (#595)

---
 scripts/completions/aichat.zsh | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

(limited to 'scripts/completions/aichat.zsh')

diff --git a/scripts/completions/aichat.zsh b/scripts/completions/aichat.zsh
index 043d1db..ae39481 100644
--- a/scripts/completions/aichat.zsh
+++ b/scripts/completions/aichat.zsh
@@ -22,11 +22,14 @@ _aichat() {
 '--role=[Select a role]:ROLE:->roles' \
 '-s+[Start or join a session]:SESSION:->sessions' \
 '--session=[Start or join a session]:SESSION:->sessions' \
+'--save-session[Forces the session to be saved]' \
+'-b+[Start a bot]:BOT:->bots' \
+'--bot=[Start a bot]:BOT:->bots' \
+'--rag=[Start a RAG]:RAG:->rags' \
 '*-f+[Include files with the message]:FILE:_files' \
 '*--file=[Include files with the message]:FILE:_files' \
 '-w+[Control text wrapping (no, auto, <max-width>)]:WRAP: ' \
 '--wrap=[Control text wrapping (no, auto, <max-width>)]:WRAP: ' \
-'--save-session[Forces the session to be saved]' \
 '--serve[Serve the LLM API and WebAPP]' \
 '-e[Execute commands in natural language]' \
 '--execute[Execute commands in natural language]' \
@@ -40,8 +43,10 @@ _aichat() {
 '--dry-run[Display the message without sending it]' \
 '--info[Display information]' \
 '--list-models[List all available models]' \
-'--list-roles[List all available roles]' \
-'--list-sessions[List all available sessions]' \
+'--list-roles[List all roles]' \
+'--list-sessions[List all sessions]' \
+'--list-bots[List all bots]' \
+'--list-rags[List all RAGs]' \
 '-h[Print help]' \
 '--help[Print help]' \
 '-V[Print version]' \
@@ -53,7 +58,7 @@ _aichat() {
     _arguments "${_arguments_options[@]}" $common \
         && ret=0 
     case $state in
-        models|roles|sessions)
+        models|roles|sessions|bots|rags)
             local -a values expl
             values=( ${(f)"$(_call_program values aichat --list-$state)"} )
             _wanted values expl $state compadd -a values && ret=0
-- 
cgit v1.2.3