From 704047671f8183057d5323f976e2fc7bba426126 Mon Sep 17 00:00:00 2001
From: yhydev <yhydev@outlook.com>
Date: Sat, 13 Dec 2025 23:19:55 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BB=A3=E7=A0=81=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98=EF=BC=9A=E6=8F=90=E9=AB=98=E8=84=9A=E6=9C=AC=E5=81=A5?=
 =?UTF-8?q?=E5=A3=AE=E6=80=A7=E3=80=81=E5=A2=9E=E5=BC=BA=E7=81=B5=E6=B4=BB?=
 =?UTF-8?q?=E6=80=A7=E3=80=81=E6=94=B9=E5=96=84=E5=8F=AF=E8=AF=BB=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pyproject.toml         |  6 +++++-
 scripts/start-agent.sh | 14 ++++++++++----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 5a42bfd..33e2172 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,4 +4,8 @@ version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.10"
-dependencies = []
+dependencies = [
+    "prefect_shell",
+    "freqtrade",
+    "prefect"
+]
diff --git a/scripts/start-agent.sh b/scripts/start-agent.sh
index 0e01e46..2bdebbf 100644
--- a/scripts/start-agent.sh
+++ b/scripts/start-agent.sh
@@ -1,4 +1,5 @@
 #!/bin/bash
+set -e
 
 check_env_var() {
     if [ -z "${!1// /}" ]; then
@@ -12,9 +13,14 @@ check_env_var OSS_ACCESS_KEY
 check_env_var OSS_SECRET_KEY
 check_env_var WORKER_NAME
 check_env_var BINANCE_FUTURES_DIR
-WORKING_DIR=/kaggle/working
+# Set working directory with default value
+WORKING_DIR=${WORKING_DIR:-/kaggle/working}
 
-pip install git+https://gitea.oopsapi.com/yhydev/myscripts.git prefect_shell freqtrade prefect;
+pip install \
+    git+https://gitea.oopsapi.com/yhydev/myscripts.git \
+    prefect_shell \
+    freqtrade \
+    prefect;
 if [ ! -d $WORKING_DIR/user_data ]; then
 freqtrade create-userdir --userdir $WORKING_DIR/user_data;
 fi
@@ -25,9 +31,9 @@ fi;
 cd $WORKING_DIR;
 if [ ! -f ~/.local/bin/mc ]; then
 mkdir -p ~/.local/bin
-wget https://dl.min.io/aistor/mc/release/linux-amd64/mc -O ~/.local/bin/mc
+wget https://dl.min.io/aistor/mc/release/linux-amd64/mc -O $HOME/.local/bin/mc
 chmod +x ~/.local/bin/mc
 fi
-export PATH=$PATH:~/.local/bin
+export PATH=$PATH:$HOME/.local/bin
 mc alias set oss $OSS_ENDPOINT $OSS_ACCESS_KEY $OSS_SECRET_KEY --api S3v4
 prefect worker start --name $WORKER_NAME --pool freqtrade -l 3
\ No newline at end of file