Upload 4 files

launch.sh

@@ -8,11 +8,11 @@ DATASET_ID=$(printenv DATASET_ID)
 SYNC_INTERVAL=$(printenv SYNC_INTERVAL)
 
 # 如果没有设置用户名和密码，使用默认值
-if [[ -z "${USERNAME}" ]]; then
+if [ -z "${USERNAME}" ]; then
   USERNAME="admin"
 fi
 
-if [[ -z "${PASSWORD}" ]]; then
+if [ -z "${PASSWORD}" ]; then
   PASSWORD="password"
 fi
 
@@ -43,7 +43,7 @@ echo "配置文件内容:"
 cat ${BASE}/config/config.yaml
 
 # 启动数据同步服务（如果提供了必要的环境变量）
-if [[ ! -z "${HF_TOKEN}" ]] && [[ ! -z "${DATASET_ID}" ]]; then
+if [ ! -z "${HF_TOKEN}" ] && [ ! -z "${DATASET_ID}" ]; then
   echo "启动数据同步服务..."
   nohup ${BASE}/sync_data.sh > ${BASE}/sync_data.log 2>&1 &
   echo "数据同步服务已在后台启动"

sync_data.sh

@@ -83,30 +83,54 @@ else
     log_info "HuggingFace API连接测试成功"
 fi
 
+# 生成唯一的测试文件名
+TEST_FILE_NAME="test_file_$(date +%s)"
+
 # 测试数据集权限
 log_info "正在测试Dataset权限..."
 python3 -c "
 from huggingface_hub import HfApi
 try:
     api = HfApi(token='$HF_TOKEN')
-    # 尝试创建一个测试文件
+    
+    # 创建本地测试文件
+    with open('$TEMP_DIR/test_file', 'w') as f:
+        f.write('test')
+    
+    # 上传测试文件
+    test_file_name = '$TEST_FILE_NAME'
+    print(f'正在上传测试文件: {test_file_name}')
+    
     api.upload_file(
-        path_or_fileobj='$TEMP_DIR/test_file' if not open('$TEMP_DIR/test_file', 'w').write('test') else '$TEMP_DIR/test_file',
-        path_in_repo='test_file_' + '$(date +%s)',
+        path_or_fileobj='$TEMP_DIR/test_file',
+        path_in_repo=test_file_name,
         repo_id='$DATASET_ID',
         repo_type='dataset'
     )
     print('成功上传测试文件到Dataset')
+    
+    # 删除已上传的测试文件
+    print('正在删除测试文件...')
+    api.delete_file(
+        path_in_repo=test_file_name,
+        repo_id='$DATASET_ID',
+        repo_type='dataset'
+    )
+    print('已成功删除测试文件')
+    
 except Exception as e:
-    print(f'上传测试文件到Dataset失败: {str(e)}')
+    print(f'Dataset权限测试失败: {str(e)}')
     exit(1)
 "
 if [ $? -ne 0 ]; then
     log_error "Dataset权限测试失败，请检查DATASET_ID是否正确且有写入权限"
 else
-    log_info "Dataset权限测试成功"
+    log_info "Dataset权限测试成功，测试文件已清理"
 fi
 
+# 确保本地测试文件被删除
+rm -f "$TEMP_DIR/test_file"
+
 # 上传备份
 upload_backup() {
     file_path="$1"