diff --git a/tasks/workstation/linux/software/ai/llm.yml b/tasks/workstation/linux/software/ai/llm.yml index 7ae5274..5e0dcf0 100644 --- a/tasks/workstation/linux/software/ai/llm.yml +++ b/tasks/workstation/linux/software/ai/llm.yml @@ -986,17 +986,24 @@ path: "{{ user_user.home }}/{{ item }}" marker: "# {mark} ANSIBLE MANAGED BLOCK | ollama | Manual Script" content: | - function ollama_llms() { + function llm_installs() { {% for llm in llm_installs %} echo "{{ llm.model }}" {% endfor %} } + function llm_removals() { + {% for llm in llm_removals %} + + echo "{{ llm.model }}" + + {% endfor %} + } function ollama_pull() { echo "Manually Pulling Models" - for model in `ollama_llms`; do + for model in `llm_installs`; do echo -e "\n\n Starting 'ollama pull $model' \n\n" ollama pull $model status=$? @@ -1009,6 +1016,22 @@ alias pull-models="ollama_pull " alias pull-ai="ollama_pull " + function ollama_clean() { + echo "Cleaning Models" + + for model in `llm_removals`; do + echo -e "\n Removing '$model'" + ollama rm $model + done + + echo "Done!" + } + alias ollama-clean="ollama_clean " + alias clean-ollama="ollama_clean " + alias clean-llms="ollama_clean " + alias clean-models="ollama_clean " + alias clean-ai="ollama_clean " + function ollama_chat() { ollama run {{ 'qwen3.8:27b' if ansible_memtotal_mb > 32000 else 'qwen3.5:4b' }} }