Ask : Add ask_model.
This commit is contained in:
parent
a374e6b68f
commit
e59772adb9
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue