Ask : Add ask_model.

This commit is contained in:
Dmitry Voronin 2024-03-22 20:11:15 +03:00
parent a374e6b68f
commit e59772adb9

View file

@ -1,9 +1,24 @@
export _ask_model="mistral"
# Ask general AI. # Ask general AI.
# Usage: ask <QUERY> # Usage: ask <QUERY>
function ask() { function ask() {
curl http://localhost:11434/api/generate -d "{ curl http://localhost:11434/api/generate -d "{
\"model\": \"mistral\", \"model\": \"${_ask_model}\",
\"prompt\":\"${*}\" \"prompt\":\"${*}\"
}" 2> /dev/null | parallel -j1 -- "echo {} | jq -r .response | tr -d '\n'" }" 2> /dev/null | parallel -j1 -- "echo {} | jq -r .response | tr -d '\n'"
echo echo
} }
# Specify ask model.
function ask_model() {
export _ask_model="${1}"
}
function _complete_ask_model() {
local IFS=$'\n'
local models=($(ollama list | sed -e "1d" | cut -f1))
_autocomplete_first ${models[@]}
}
complete -F _complete_ask_model ask_model