add back ai primary buffer query (closes #1462)

2025-06-19 17:31:36 +02:00
parent f844a54a87
commit 4ebd486426
4 changed files with 342 additions and 1 deletions
@@ -0,0 +1,41 @@
+#!/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="Text selection query" --expire-time=10000 \
+        "$content" "$response"
+else
+    notify-send --app-name="Text selection query" --expire-time=10000 \
+        "AI Response" "$response"
+fi