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