From 4a8eee8c1f350f2bb2fca8d139b53d1e34d09bb4 Mon Sep 17 00:00:00 2001
From: Meng Zhang <meng@tabbyml.com>
Date: Thu, 6 Apr 2023 10:25:50 +0800
Subject: [PATCH] refactor: extract triton start logic into bash function

---
 tabby/scripts/tabby.sh | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/tabby/scripts/tabby.sh b/tabby/scripts/tabby.sh
index 38341a8..96f7688 100755
--- a/tabby/scripts/tabby.sh
+++ b/tabby/scripts/tabby.sh
@@ -30,18 +30,20 @@ python -m tabby.tools.download_models --repo_id=$MODEL_NAME
 }
 
 
-supervisor() {
+program:triton() {
 if [[ "$MODEL_BACKEND" == "triton" ]]
 then
 
-local TRITON_SERVER=$(cat <<EOF
+cat <<EOF
 [program:triton]
 command=./tabby/scripts/triton.sh
 EOF
-)
 
 fi
 
+}
+
+supervisor() {
 # Create logs dir if not exists.
 mkdir -p ${LOGS_DIR}
 
@@ -65,7 +67,7 @@ command=dagu scheduler
 [program:dagu_server]
 command=dagu server --host 0.0.0.0 --port 8080
 
-$TRITON_SERVER
+$(program:triton)
 EOF
 )
 }