ayo/scripts
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(ai): use switch-case

Browse source
This commit is contained in:
ayo 2026-03-07 11:25:32 +01:00
parent 2883e00cdf
commit ac14e6eed7
1 changed files with 66 additions and 47 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

113
ai.sh
View file

@ -9,7 +9,7 @@ model=$helper
modelfile=$helper_modelfile
# Initialize variables
other_args=${@:2}
command=${@:2}
typora_flag=false
# parse string args (when used as a function and passed "$@")
@ -27,59 +27,78 @@ while [[ $# -gt 0 ]]; do
esac
done
if ! [ "$other_args" = "" ]; then
if [ "$other_args" = "open-webui" ]; then
function main() {
case $command in
"open-webui")
. $HOME/open-webui/.venv/bin/activate
open-webui serve
python --version
deactivate
elif [ "$other_args" = "remote" ]; then
;;
"remote")
export OLLAMA_HOST=192.168.0.6
elif [ "$other_args" = "list" ]; then
OLLAMA_HOST=$host ollama list
elif [ "$other_args" = "ps" ]; then
OLLAMA_HOST=$host ollama ps
elif [ "$other_args" = "rm" ]; then
OLLAMA_HOST=$host ollama rm "$3"
elif [ "$other_args" = "init" ]; then
OLLAMA_HOST=$host ollama create $model -f $modelfile
elif [ "$other_args" = "wake" ]; then
. $HOME/llm_env/bin/activate
;;
"list")
OLLAMA_HOST=$host ollama list
;;
"ps")
OLLAMA_HOST=$host ollama ps
;;
"rm")
OLLAMA_HOST=$host ollama rm "$3"
;;
"init")
OLLAMA_HOST=$host ollama create $model -f $modelfile
;;
"wake")
. $HOME/llm_env/bin/activate
unset OCL_ICD_VENDORS
export OLLAMA_NUM_GPU=999
export no_proxy=localhost,127.0.0.1
export ZES_ENABLE_SYSMAN=1
source $HOME/intel/oneapi/setvars.sh
export SYCL_CACHE_PERSISTENT=1
export SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=0
unset OCL_ICD_VENDORS
export OLLAMA_NUM_GPU=999
export no_proxy=localhost,127.0.0.1
export ZES_ENABLE_SYSMAN=1
source $HOME/intel/oneapi/setvars.sh
export SYCL_CACHE_PERSISTENT=1
export SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=0
$HOME/llama-cpp/ollama serve
python --version
deactivate
$HOME/llama-cpp/ollama serve
python --version
deactivate
echo $ZES_ENABLE_SYSMAN
echo $SYCL_CACHE_PERSISTENT
elif [ "$other_args" = "update" ]; then
curl -fsSL https://ollama.com/install.sh | sh
echo "See instructions on how to expose ollama in the local network: https://git.ayo.run/ayo/scripts/src/branch/main/expose-ollama.md"
elif [ "$other_args" = "sleep" ]; then
OLLAMA_HOST=$host ollama stop $model
else
# If -t flag is set, use typora to display output
if [ "$typora_flag" = true ]; then
tempfile="$(mktemp)"
OLLAMA_HOST=$host ollama run $model "$other_args" --hidethinking > $tempfile
typora $tempfile > /dev/null 2>/dev/null &
else
# If no -t flag, just run the command normally
OLLAMA_HOST=$host ollama run $model "$other_args" --hidethinking
fi
fi
echo $ZES_ENABLE_SYSMAN
echo $SYCL_CACHE_PERSISTENT
;;
"update")
curl -fsSL https://ollama.com/install.sh | sh
echo "See instructions on how to expose ollama in the local network: https://git.ayo.run/ayo/scripts/src/branch/main/expose-ollama.md"
;;
"sleep")
OLLAMA_HOST=$host ollama stop $model
;;
"")
OLLAMA_HOST=$host ollama run $model --hidethinking
;;
*)
# If -t flag is set, use typora to display output
if [ "$typora_flag" = true ]; then
tempfile="$(mktemp)"
OLLAMA_HOST=$host ollama run $model "$command" --hidethinking > $tempfile
typora $tempfile > /dev/null 2>/dev/null &
else
# If no -t flag, just run the command normally
OLLAMA_HOST=$host ollama run $model "$command" --hidethinking
fi
;;
esac
else
OLLAMA_HOST=$host ollama run $model --hidethinking
fi
# release memory
OLLAMA_HOST=$host ollama stop $model
}
# release memory
OLLAMA_HOST=$host ollama stop $model
start_time=$(date +%s%N)
main $@
end_time=$(date +%s%N)
duration=$((end_time - start_time))
duration_ms=$(echo "scale=3; $duration / 1000000" | bc)
duration_s=$(echo "scale=3; $duration_ms / 1000" | bc)
echo "Took $duration_s s"