Task 010 implement research manifest flow

This commit is contained in:
2026-05-21 21:56:44 +03:00
parent 5b27d1a3ca
commit 7d4a79acb5
23 changed files with 1063 additions and 26 deletions
+1 -1
View File
@@ -112,7 +112,7 @@ edited_plan_id="$(printf '%s' "${edited_json}" | json_eval "json.load(sys.stdin)
request POST "/api/articles/${article_id}/plans/${edited_plan_id}/request-revision" '{"notes":"Smoke revision"}' >/dev/null
request POST "/api/articles/${article_id}/plans/${edited_plan_id}/approve" >/dev/null
research_json="$(request POST "/api/articles/${article_id}/research/start")"
job_type="$(printf '%s' "${research_json}" | json_eval "json.load(sys.stdin)['jobs'][0]['job_type']")"
job_type="$(printf '%s' "${research_json}" | json_eval "json.load(sys.stdin)['job']['job_type']")"
if [[ "${job_type}" != "RESEARCH" ]]; then
echo "Expected RESEARCH job, got ${job_type}" >&2
+112
View File
@@ -0,0 +1,112 @@
#!/usr/bin/env bash
set -euo pipefail
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
COMPOSE_FILE="${ROOT_DIR}/docker-compose.yml"
BACKEND_URL="${BACKEND_URL:-http://localhost:8000}"
WAIT_SECONDS="${WAIT_SECONDS:-60}"
PYTHON_BIN="${PYTHON_BIN:-python3}"
EDITOR_EMAIL="${EDITOR_EMAIL:-editor@example.com}"
cleanup() {
docker compose -f "${COMPOSE_FILE}" down --remove-orphans >/dev/null 2>&1 || true
}
request() {
local method="$1"
local path="$2"
local body="${3:-}"
if [[ -n "${body}" ]]; then
curl -fsS -X "${method}" -H "Content-Type: application/json" \
-H "X-Demo-User-Email: ${EDITOR_EMAIL}" --data "${body}" "${BACKEND_URL}${path}"
return
fi
curl -fsS -X "${method}" -H "X-Demo-User-Email: ${EDITOR_EMAIL}" "${BACKEND_URL}${path}"
}
wait_backend() {
local deadline=$((SECONDS + WAIT_SECONDS))
while (( SECONDS < deadline )); do
if curl -fsS "${BACKEND_URL}/health" >/dev/null 2>&1; then
return 0
fi
sleep 1
done
echo "Timed out waiting for backend at ${BACKEND_URL}" >&2
return 1
}
json_eval() {
local expression="$1"
"${PYTHON_BIN}" -c "import json,sys; print(${expression})"
}
cd "${ROOT_DIR}"
trap cleanup EXIT
docker compose -f "${COMPOSE_FILE}" up --build -d
wait_backend
site_id="$(request GET /api/sites | json_eval "json.load(sys.stdin)[0]['site']['id']")"
article_body="$(
SITE_ID="${site_id}" "${PYTHON_BIN}" - <<'PY'
import json
import os
import uuid
print(json.dumps({
"target_site_id": os.environ["SITE_ID"],
"brief_description": f"Research smoke flow {uuid.uuid4()}",
"working_title": "Research Smoke Flow",
"content_type": "longform_guide",
"primary_keyword": "research smoke flow",
}))
PY
)"
article_id="$(request POST /api/articles "${article_body}" | json_eval "json.load(sys.stdin)['article']['id']")"
questions_json="$(request POST "/api/articles/${article_id}/boundary-questions/generate")"
question_patch_commands="$(
QUESTIONS_JSON="${questions_json}" "${PYTHON_BIN}" - <<'PY'
import json
import os
for question in json.loads(os.environ["QUESTIONS_JSON"])["questions"]:
if question["is_required"]:
print(f"{question['id']}\tAccepted answer for {question['category']}.")
PY
)"
while IFS=$'\t' read -r question_id answer; do
[[ -z "${question_id}" ]] && continue
patch_body="$(ANSWER="${answer}" "${PYTHON_BIN}" - <<'PY'
import json
import os
print(json.dumps({"answer": os.environ["ANSWER"]}))
PY
)"
request PATCH "/api/articles/${article_id}/boundary-questions/${question_id}" "${patch_body}" >/dev/null
done <<< "${question_patch_commands}"
request POST "/api/articles/${article_id}/boundary-questions/submit" >/dev/null
plan_id="$(request POST "/api/articles/${article_id}/plan/generate" | json_eval "json.load(sys.stdin)['plan']['id']")"
request POST "/api/articles/${article_id}/plans/${plan_id}/approve" >/dev/null
research_json="$(request POST "/api/articles/${article_id}/research/start")"
manifest_id="$(printf '%s' "${research_json}" | json_eval "json.load(sys.stdin)['manifest']['id']")"
artifact_count="$(printf '%s' "${research_json}" | json_eval "len(json.load(sys.stdin)['manifest']['artifacts'])")"
if (( artifact_count < 2 )); then
echo "Expected at least 2 research artifacts, got ${artifact_count}" >&2
exit 1
fi
research_list_json="$(request GET "/api/articles/${article_id}/research")"
listed_id="$(printf '%s' "${research_list_json}" | json_eval "json.load(sys.stdin)['manifests'][0]['id']")"
if [[ "${listed_id}" != "${manifest_id}" ]]; then
echo "Expected manifest ${manifest_id}, got ${listed_id}" >&2
exit 1
fi
echo "research flow smoke ok"