wehub-resource-sync
112 lines
3.5 KiB
Bash
Executable File
112 lines
3.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
LOCAL_BASE_URL="${LOCAL_BASE_URL:-http://127.0.0.1:20128}"
|
|
CLOUD_BASE_URL="${CLOUD_BASE_URL:-https://omniroute.com/v1}"
|
|
MODEL="${MODEL:-kr/claude-sonnet-4.5}"
|
|
STREAM_MODE="${STREAM_MODE:-false}"
|
|
MAX_RETRIES="${MAX_RETRIES:-6}"
|
|
RETRY_DELAY_SECONDS="${RETRY_DELAY_SECONDS:-3}"
|
|
TIMEOUT_SECONDS="${TIMEOUT_SECONDS:-45}"
|
|
|
|
LOCAL_KEYS_URL="${LOCAL_BASE_URL%/}/api/keys"
|
|
LOCAL_SYNC_URL="${LOCAL_BASE_URL%/}/api/sync/cloud"
|
|
CLOUD_CHAT_URL="${CLOUD_BASE_URL%/}/chat/completions"
|
|
|
|
echo "[1/5] Creating local API key"
|
|
KEY_NAME="cloud-e2e-$(date +%s)"
|
|
CREATE_RESP="$(curl -sS -m "${TIMEOUT_SECONDS}" -X POST "${LOCAL_KEYS_URL}" \
|
|
-H 'Content-Type: application/json' \
|
|
--data "{\"name\":\"${KEY_NAME}\"}")"
|
|
|
|
API_KEY="$(echo "${CREATE_RESP}" | jq -r '.key // empty')"
|
|
if [[ -z "${API_KEY}" || "${API_KEY}" == "null" ]]; then
|
|
echo "FAIL: could not create local API key"
|
|
echo "Response: ${CREATE_RESP}"
|
|
exit 1
|
|
fi
|
|
echo " Key created: ${API_KEY:0:12}...${API_KEY: -6}"
|
|
|
|
echo "[2/5] Enabling cloud mode"
|
|
ENABLE_CODE="$(curl -sS -m "${TIMEOUT_SECONDS}" -o /tmp/cloud_enable_e2e.json -w '%{http_code}' \
|
|
-X POST "${LOCAL_SYNC_URL}" \
|
|
-H 'Content-Type: application/json' \
|
|
--data '{"action":"enable"}' || true)"
|
|
echo " enable status: ${ENABLE_CODE}"
|
|
cat /tmp/cloud_enable_e2e.json; echo
|
|
if [[ ! "${ENABLE_CODE}" =~ ^2 ]]; then
|
|
echo "FAIL: cloud enable failed"
|
|
exit 1
|
|
fi
|
|
|
|
echo "[3/5] Running explicit sync"
|
|
SYNC_CODE="$(curl -sS -m "${TIMEOUT_SECONDS}" -o /tmp/cloud_sync_e2e.json -w '%{http_code}' \
|
|
-X POST "${LOCAL_SYNC_URL}" \
|
|
-H 'Content-Type: application/json' \
|
|
--data '{"action":"sync"}' || true)"
|
|
echo " sync status: ${SYNC_CODE}"
|
|
cat /tmp/cloud_sync_e2e.json; echo
|
|
if [[ ! "${SYNC_CODE}" =~ ^2 ]]; then
|
|
echo "FAIL: cloud sync failed"
|
|
exit 1
|
|
fi
|
|
|
|
build_payload() {
|
|
local stream_flag="$1"
|
|
cat <<JSON
|
|
{
|
|
"model": "${MODEL}",
|
|
"stream": ${stream_flag},
|
|
"messages": [
|
|
{"role":"user","content":"Reply with exactly: CLOUD_OK"}
|
|
],
|
|
"max_tokens": 64
|
|
}
|
|
JSON
|
|
}
|
|
|
|
PAYLOAD="$(build_payload "${STREAM_MODE}")"
|
|
|
|
echo "[4/5] Testing cloud OpenAI-compatible endpoint with retry (stream=${STREAM_MODE})"
|
|
SUCCESS=0
|
|
for ATTEMPT in $(seq 1 "${MAX_RETRIES}"); do
|
|
CODE="$(curl -sS -m "${TIMEOUT_SECONDS}" -o /tmp/cloud_chat_e2e.json -w '%{http_code}' \
|
|
-X POST "${CLOUD_CHAT_URL}" \
|
|
-H "Authorization: Bearer ${API_KEY}" \
|
|
-H 'Content-Type: application/json' \
|
|
--data "${PAYLOAD}" || true)"
|
|
echo " attempt ${ATTEMPT}/${MAX_RETRIES}: HTTP ${CODE}"
|
|
cat /tmp/cloud_chat_e2e.json; echo
|
|
|
|
if [[ "${CODE}" =~ ^2 ]]; then
|
|
SUCCESS=1
|
|
break
|
|
fi
|
|
|
|
sleep "${RETRY_DELAY_SECONDS}"
|
|
done
|
|
|
|
echo "[5/5] Result"
|
|
if [[ "${SUCCESS}" -eq 1 ]]; then
|
|
echo "PASS: cloud endpoint accepted synced key"
|
|
exit 0
|
|
fi
|
|
|
|
echo "Primary test failed. Trying fallback with stream=true to distinguish key/auth issues..."
|
|
PAYLOAD_STREAM_TRUE="$(build_payload "true")"
|
|
FALLBACK_CODE="$(curl -sS -m "${TIMEOUT_SECONDS}" -o /tmp/cloud_chat_e2e_fallback.json -w '%{http_code}' \
|
|
-X POST "${CLOUD_CHAT_URL}" \
|
|
-H "Authorization: Bearer ${API_KEY}" \
|
|
-H 'Content-Type: application/json' \
|
|
--data "${PAYLOAD_STREAM_TRUE}" || true)"
|
|
echo " fallback stream=true HTTP ${FALLBACK_CODE}"
|
|
cat /tmp/cloud_chat_e2e_fallback.json; echo
|
|
|
|
if [[ "${FALLBACK_CODE}" =~ ^2 ]]; then
|
|
echo "PARTIAL PASS: API key is valid and cloud works with stream=true; stream=false path appears broken upstream."
|
|
exit 2
|
|
fi
|
|
|
|
echo "FAIL: cloud endpoint still failing after sync/retries (including stream=true fallback)"
|
|
exit 1
|