Add model choice ✨ (#9)

nosia-ai · Jun 27, 2024 · 3783c77 · 3783c77
1 parent bfd983f
commit 3783c77
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/script/production/setup b/script/production/setup
@@ -3,7 +3,7 @@
 source ./script/concerns/loader
 
 start_loader "Setting up environment"
-OLLAMA_URL=$OLLAMA_URL RAILS_MASTER_KEY=$RAILS_MASTER_KEY ./script/production/setup-env &>> ./log/production.log
+OLLAMA_MODEL=$OLLAMA_MODEL OLLAMA_URL=$OLLAMA_URL RAILS_MASTER_KEY=$RAILS_MASTER_KEY ./script/production/setup-env &>> ./log/production.log
 stop_loader
 
 start_loader "Setting up Docker"

diff --git a/script/production/setup-env b/script/production/setup-env
@@ -19,7 +19,12 @@ else
   echo "OLLAMA_URL is not set, using default value"
   OLLAMA_URL=http://localhost:11434
 fi
-OLLAMA_MODEL=phi3:medium
+if [ -n "$OLLAMA_MODEL" ]; then
+  echo "Using OLLAMA_MODEL from environment"
+else
+  echo "OLLAMA_MODEL is not set, using default value"
+  OLLAMA_MODEL=phi3:medium
+fi
 POSTGRES_HOST=postgres-db
 POSTGRES_PORT=5432
 POSTGRES_DB=nosia_production