145 lines
5.8 KiB
Bash
Executable File
145 lines
5.8 KiB
Bash
Executable File
#!/bin/bash
|
||
# 验证 ingest Step 1 的 JSON 输出格式
|
||
# 用法：bash validate-step1.sh <json_file>
|
||
# 返回：0 = 格式正确，1 = 格式有问题（触发回退）
|
||
|
||
SCRIPT_DIR="${BASH_SOURCE[0]%/*}"
|
||
[ "$SCRIPT_DIR" = "${BASH_SOURCE[0]}" ] && SCRIPT_DIR="."
|
||
SCRIPT_DIR="$(cd "$SCRIPT_DIR" && pwd)"
|
||
# shellcheck disable=SC1091
|
||
source "$SCRIPT_DIR/shared-config.sh"
|
||
|
||
JSON_FILE="$1"
|
||
|
||
# 参数检查
|
||
[ -z "$1" ] && { echo "ERROR: usage: validate-step1.sh <json_file>"; exit 1; }
|
||
|
||
# 检查 jq 是否可用（必需依赖）
|
||
command -v jq >/dev/null 2>&1 || {
|
||
echo "ERROR: jq is not installed. Install it via:" >&2
|
||
print_install_hint jq
|
||
exit 1
|
||
}
|
||
|
||
# 检查文件是否存在
|
||
[ -f "$JSON_FILE" ] || { echo "ERROR: file not found: $JSON_FILE"; exit 1; }
|
||
|
||
# 检查是否是有效 JSON
|
||
jq empty "$JSON_FILE" 2>/dev/null || { echo "ERROR: invalid JSON format"; exit 1; }
|
||
|
||
# 检查必需字段存在且类型正确
|
||
jq -e '.entities | type == "array"' "$JSON_FILE" >/dev/null 2>&1 || { echo "ERROR: 'entities' must be an array"; exit 1; }
|
||
jq -e '.topics | type == "array"' "$JSON_FILE" >/dev/null 2>&1 || { echo "ERROR: 'topics' must be an array"; exit 1; }
|
||
jq -e '.connections | type == "array"' "$JSON_FILE" >/dev/null 2>&1 || { echo "ERROR: 'connections' must be an array"; exit 1; }
|
||
jq -e '.contradictions | type == "array"' "$JSON_FILE" >/dev/null 2>&1 || { echo "ERROR: 'contradictions' must be an array"; exit 1; }
|
||
jq -e '.new_vs_existing | type == "object"' "$JSON_FILE" >/dev/null 2>&1 || { echo "ERROR: 'new_vs_existing' must be an object"; exit 1; }
|
||
|
||
# 检查每个 entity 的必需子字段
|
||
VALID_CONFIDENCE="EXTRACTED|INFERRED|AMBIGUOUS|UNVERIFIED"
|
||
|
||
ENTITY_COUNT=$(jq '.entities | length' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$ENTITY_COUNT" -gt 0 ] 2>/dev/null; then
|
||
NON_OBJECT_ENTITY_COUNT=$(jq '[.entities[] | select(type != "object")] | length' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$NON_OBJECT_ENTITY_COUNT" -gt 0 ] 2>/dev/null; then
|
||
echo "ERROR: $NON_OBJECT_ENTITY_COUNT entity/entities must be objects"
|
||
exit 1
|
||
fi
|
||
|
||
# name, type, confidence 必须存在且非空
|
||
BAD_ENTITY_COUNT=$(jq '
|
||
[.entities[] | select(
|
||
(.name // "" | length) == 0 or
|
||
(.type // "" | length) == 0 or
|
||
(.confidence // "" | length) == 0
|
||
)] | length
|
||
' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$BAD_ENTITY_COUNT" -gt 0 ] 2>/dev/null; then
|
||
echo "ERROR: $BAD_ENTITY_COUNT entity/entities missing required fields (name/type/confidence)"
|
||
exit 1
|
||
fi
|
||
|
||
# confidence 值必须是四个有效值之一
|
||
INVALID=$(jq -r '.entities[]? | (.confidence // "MISSING")' "$JSON_FILE" 2>/dev/null | \
|
||
grep -v -E "^($VALID_CONFIDENCE)$" | head -3)
|
||
if [ -n "$INVALID" ]; then
|
||
echo "ERROR: invalid entity confidence value(s): $INVALID"
|
||
echo " Valid values: EXTRACTED | INFERRED | AMBIGUOUS | UNVERIFIED"
|
||
exit 1
|
||
fi
|
||
|
||
# EXTRACTED 和 INFERRED 必须提供 evidence 字段
|
||
NO_EVIDENCE_COUNT=$(jq '
|
||
[.entities[] | select(
|
||
(.confidence == "EXTRACTED" or .confidence == "INFERRED") and
|
||
((.evidence // "" | length) == 0)
|
||
)] | length
|
||
' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$NO_EVIDENCE_COUNT" -gt 0 ] 2>/dev/null; then
|
||
echo "WARN: $NO_EVIDENCE_COUNT entity/entities with EXTRACTED/INFERRED confidence missing 'evidence' field"
|
||
fi
|
||
fi
|
||
|
||
# 检查每个 topic 的必需子字段
|
||
TOPIC_COUNT=$(jq '.topics | length' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$TOPIC_COUNT" -gt 0 ] 2>/dev/null; then
|
||
NON_OBJECT_TOPIC_COUNT=$(jq '[.topics[] | select(type != "object")] | length' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$NON_OBJECT_TOPIC_COUNT" -gt 0 ] 2>/dev/null; then
|
||
echo "ERROR: $NON_OBJECT_TOPIC_COUNT topic(s) must be objects"
|
||
exit 1
|
||
fi
|
||
|
||
BAD_TOPIC_COUNT=$(jq '
|
||
[.topics[] | select(
|
||
(.name // "" | length) == 0
|
||
)] | length
|
||
' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$BAD_TOPIC_COUNT" -gt 0 ] 2>/dev/null; then
|
||
echo "ERROR: $BAD_TOPIC_COUNT topic(s) missing required 'name' field"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
# 检查每个 connection 的必需子字段（from, to, confidence）
|
||
CONN_COUNT=$(jq '.connections | length' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$CONN_COUNT" -gt 0 ] 2>/dev/null; then
|
||
NON_OBJECT_CONN_COUNT=$(jq '[.connections[] | select(type != "object")] | length' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$NON_OBJECT_CONN_COUNT" -gt 0 ] 2>/dev/null; then
|
||
echo "ERROR: $NON_OBJECT_CONN_COUNT connection(s) must be objects"
|
||
exit 1
|
||
fi
|
||
|
||
BAD_CONN_COUNT=$(jq '
|
||
[.connections[] | select(
|
||
(.from // "" | length) == 0 or
|
||
(.to // "" | length) == 0 or
|
||
(.confidence // "" | length) == 0
|
||
)] | length
|
||
' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$BAD_CONN_COUNT" -gt 0 ] 2>/dev/null; then
|
||
echo "ERROR: $BAD_CONN_COUNT connection(s) missing required fields (from/to/confidence)"
|
||
exit 1
|
||
fi
|
||
|
||
INVALID_CONN_CONF=$(jq -r '.connections[]? | (.confidence // "MISSING")' "$JSON_FILE" 2>/dev/null | \
|
||
grep -v -E "^($VALID_CONFIDENCE)$" | head -3)
|
||
if [ -n "$INVALID_CONN_CONF" ]; then
|
||
echo "ERROR: invalid connection confidence value(s): $INVALID_CONN_CONF"
|
||
echo " Valid values: EXTRACTED | INFERRED | AMBIGUOUS | UNVERIFIED"
|
||
exit 1
|
||
fi
|
||
|
||
# EXTRACTED 和 INFERRED connections 必须提供 evidence
|
||
NO_CONN_EVIDENCE=$(jq '
|
||
[.connections[] | select(
|
||
(.confidence == "EXTRACTED" or .confidence == "INFERRED") and
|
||
((.evidence // "" | length) == 0)
|
||
)] | length
|
||
' "$JSON_FILE" 2>/dev/null)
|
||
if [ "$NO_CONN_EVIDENCE" -gt 0 ] 2>/dev/null; then
|
||
echo "WARN: $NO_CONN_EVIDENCE connection(s) with EXTRACTED/INFERRED confidence missing 'evidence' field"
|
||
fi
|
||
fi
|
||
|
||
echo "OK: Step 1 JSON validation passed"
|
||
exit 0
|