From a4a88ec7580ce50a8b1707fc2b1c5d0c09b25c89 Mon Sep 17 00:00:00 2001
From: Ayo <ayo@ayco.io>
Date: Sun, 24 Aug 2025 10:14:43 +0200
Subject: [PATCH] feat(ai): indicate models to use in config file

---
 ai-brainstorm.sh | 5 ++++-
 ai-coder.sh      | 5 ++++-
 ai.sh            | 6 ++++--
 example.conf     | 4 +++-
 4 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/ai-brainstorm.sh b/ai-brainstorm.sh
index 49074b0..1180b4c 100644
--- a/ai-brainstorm.sh
+++ b/ai-brainstorm.sh
@@ -1,4 +1,7 @@
-model=brainstorm:8b
+# Load config
+. ${HOME}/ayo.conf
+
+model=$brainstorm
 
 if ! [ "$2" = "" ]; then
   if [ "$2" = "sleep" ]; then
diff --git a/ai-coder.sh b/ai-coder.sh
index d55e2d2..f37d66e 100644
--- a/ai-coder.sh
+++ b/ai-coder.sh
@@ -1,4 +1,7 @@
-model=coder:30b
+# Load config
+. ${HOME}/ayo.conf
+
+model=$coder
 
 if ! [ "$2" = "" ]; then
   if [ "$2" = "sleep" ]; then
diff --git a/ai.sh b/ai.sh
index 5e271da..fd78feb 100644
--- a/ai.sh
+++ b/ai.sh
@@ -1,4 +1,7 @@
-model=helper:8b
+# Load config
+. ${HOME}/ayo.conf
+
+model=$helper
 
 if ! [ "$2" = "" ]; then
   if [ "$2" = "open-webui" ]; then
@@ -6,7 +9,6 @@ if ! [ "$2" = "" ]; then
       open-webui serve
       python --version
       deactivate
-
   elif [ "$2" = "wake" ]; then
     . $HOME/llm_env/bin/activate
 
diff --git a/example.conf b/example.conf
index 1f7c5b5..2872c7d 100644
--- a/example.conf
+++ b/example.conf
@@ -2,5 +2,7 @@ notes_dir="${HOME}/Notes"
 tasks_dir="${HOME}/Notes/Tasks"
 archive_dir="${HOME}/Notes/Archive"
 scripts_dir="${HOME}/Projects/scripts"
-sys_prompt=$(cat ${HOME}/sys-prompt.sh) # shared system prompt for LLMs (i.e., ai.sh & ai-coder.sh)
 editor="vim"
+coder="coder:30b"
+brainstorm="brainstorm:8b"
+helper="helper:8b"