11#! /bin/bash
2- set -e
2+ set -e
33# ===============================================================================
44# 1. 配置区域 - 用户只需修改这里
55# ===============================================================================
@@ -22,7 +22,7 @@ TRAIN_BATCH_SIZE=32 # 训练batchsize
2222NUM_STEPS=6 # 每个样本step轮数
2323DEEPFINANCE_TOOL_RESULT_MAX_CHARS=10000
2424
25- # 主目录 (按需更改)
25+ # 主目录(需要更改)
2626export AJET_ROOT=" /mnt/data_cpfs/taoshuchang.tsc/deepresearch/AgentJet_new"
2727
2828NNODES=${WORLD_SIZE}
@@ -105,7 +105,7 @@ export DEEPFINANCE_MCP_CONFIG DEEPFINANCE_TOOL_RESULT_MAX_CHARS
105105# 其他服务配置
106106HF_ENDPOINT=" https://hf-mirror.com"
107107ES_HOSTS=" http://11.160.132.46:8200"
108- export HF_ENDPOINT ES_HOSTS
108+ export HF_ENDPOINT ES_HOSTS
109109
110110# log 文件位置
111111CURRENT_TIME=$( date " +%Y%m%d_%H%M%S" )
@@ -155,6 +155,8 @@ export NCCL_ASYNC_ERROR_HANDLING=1
155155
156156export PYTHONPATH=" ${AJET_ROOT} :${PYTHONPATH} "
157157export RAY_CLUSTER_MODE=" multi_node"
158+ export DEEPFINANCE_PATH=" ${ENV_SERVICE_ROOT} " # AgentJet 内部可能使用此路径
159+ export DEEPFINANCE_SCRIPT=" source /mnt/data/taoshuchang.tsc/anaconda3/etc/profile.d/conda.sh && conda activate finworld_1209 && cd ${ENV_SERVICE_ROOT} && DEEPFINANCE_TOOL_RESULT_MAX_CHARS=${DEEPFINANCE_TOOL_RESULT_MAX_CHARS} DEEPFINANCE_MCP_CONFIG=${DEEPFINANCE_MCP_CONFIG} CACHE_TYPE=${CACHE_TYPE} MONGO_URI=${MONGO_URI} MONGO_DB_NAME=${MONGO_DB_NAME} MONGO_COLLECTION_NAME=${MONGO_COLLECTION_NAME} python -m env_service.env_service --env finworld --portal 0.0.0.0 --port 8080"
158160
159161
160162# ===============================================================================
@@ -201,11 +203,12 @@ if [[ $HOSTNAME == *"-master-"* ]]; then
201203
202204 # 启动训练任务(最核心)
203205 python ajet/launcher.py \
206+ --with-deepfinance \
204207 --conf ${CONFIG_FILE} \
205208 --backbone=" verl" \
206209 --prefix=${env_log_prefix} \
207210 2>&1 | tee ${TRAIN_LOG}
208-
211+
209212
210213# ===============================================================================
211214# 6.2 Worker 节点启动流程
217220 ray stop || true
218221 ray start --address $MASTER_ADDR :6379 --num-gpus 8
219222 while true ; do sleep 60; done
220- fi
223+ fi
0 commit comments