forked from Shinonome/dots-hyprland
end-4
42 lines
1.7 KiB
Bash
Executable File
42 lines
1.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# Default system prompt
|
|
SYSTEM_PROMPT="You are a helpful, quick assistant that provides brief and concise explanation \
|
|
to given content in at most 100 characters. If the given content is not in English, translate \
|
|
it to English. If the content is an English word, provide its meaning. If the content is a name, \
|
|
provide some info about it. For a math expression, provide a simplification, \
|
|
each step on a line following this style: \`2x=11 (subtract 7 from both sides)\`. \
|
|
If you do not know the answer, simply say 'No info available'. \
|
|
Only respond for the appropriate case and use as little text as possible.\
|
|
The content:"
|
|
|
|
first_loaded_model=$("$(dirname "$0")/show-loaded-ollama-models.sh" -j | jq -r '.[0].model' 2>/dev/null) || first_loaded_model=""
|
|
model=${first_loaded_model:-"llama3.2"}
|
|
|
|
# Parse command-line arguments
|
|
while [[ "$#" -gt 0 ]]; do
|
|
case $1 in
|
|
--model) model="$2"; shift ;; # Set the model from the flag
|
|
*) echo "Unknown parameter: $1"; exit 1 ;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
# Combine the system prompt with the clipboard content
|
|
content=$(wl-paste -p | tr '\n' ' ')
|
|
prompt="$SYSTEM_PROMPT $content"
|
|
|
|
# Make the API call with the specified or default model
|
|
response=$(curl http://localhost:11434/api/generate -d \
|
|
"{\"model\": \"$model\",\"prompt\": \"$prompt\",\"stream\": false}" \
|
|
| jq -r '.response')
|
|
|
|
# Check if content is a single line and no longer than 30 characters
|
|
if [[ ${#content} -le 30 && "$content" != *$'\n'* ]]; then
|
|
notify-send --app-name="primary-buffer-query" --expire-time=10000 \
|
|
"$content" "$response"
|
|
else
|
|
notify-send --app-name="primary-buffer-query" --expire-time=10000 \
|
|
"AI Response" "$response"
|
|
fi
|