feat: 测试脚本支持 --doc-id 指定文档ID

test/test_upload_api.sh

@@ -1096,6 +1096,7 @@ show_help() {
     echo "  --task-list       获取任务列表"
     echo "  --task-stats      获取任务统计"
     echo "  --task-poll       轮询任务进度直到完成"
+    echo "  --doc-id ID       指定文档ID（用于 --task/--task-poll 等）"
     echo ""
     echo "数据源相关参数:"
     echo "  --name NAME       数据源名称"
@@ -1138,6 +1139,9 @@ main() {
     local NODE_TYPE="graph_node"
     local NODE_IDS=""
     
+    # 指定文档ID
+    local SPECIFIED_DOC_ID=""
+    
     # 解析参数
     while [[ $# -gt 0 ]]; do
         case $1 in
@@ -1217,6 +1221,10 @@ main() {
                 MODE="task-poll"
                 shift
                 ;;
+            --doc-id)
+                SPECIFIED_DOC_ID="$2"
+                shift 2
+                ;;
             --name)
                 DS_NAME="$2"
                 shift 2
@@ -1451,14 +1459,16 @@ main() {
         
         task)
             # 获取任务详情
-            if [ -f "${SCRIPT_DIR}/.last_document_id" ]; then
+            if [ -n "$SPECIFIED_DOC_ID" ]; then
+                DOCUMENT_ID="$SPECIFIED_DOC_ID"
+            elif [ -f "${SCRIPT_DIR}/.last_document_id" ]; then
                 DOCUMENT_ID=$(cat "${SCRIPT_DIR}/.last_document_id")
-                print_header "任务详情"
-                test_get_task_detail "$DOCUMENT_ID"
             else
-                print_error "未找到上次上传的文档ID"
+                print_error "未找到文档ID，请使用 --doc-id 指定"
                 exit 1
             fi
+            print_header "任务详情"
+            test_get_task_detail "$DOCUMENT_ID"
             ;;
         
         task-list)
@@ -1475,14 +1485,16 @@ main() {
         
         task-poll)
             # 轮询任务进度
-            if [ -f "${SCRIPT_DIR}/.last_document_id" ]; then
+            if [ -n "$SPECIFIED_DOC_ID" ]; then
+                DOCUMENT_ID="$SPECIFIED_DOC_ID"
+            elif [ -f "${SCRIPT_DIR}/.last_document_id" ]; then
                 DOCUMENT_ID=$(cat "${SCRIPT_DIR}/.last_document_id")
-                print_header "轮询任务进度"
-                poll_task_progress "$DOCUMENT_ID"
             else
-                print_error "未找到上次上传的文档ID"
+                print_error "未找到文档ID，请使用 --doc-id 指定"
                 exit 1
             fi
+            print_header "轮询任务进度"
+            poll_task_progress "$DOCUMENT_ID"
             ;;
         
         upload)