Files
awoooi/scripts/cron_km_vectorize.py
Your Name a4e9a04982
Some checks failed
Code Review / ai-code-review (push) Successful in 10s
run-migration / migrate (push) Successful in 7s
CD Pipeline / build-and-deploy (push) Has been cancelled
CD Pipeline / post-deploy-checks (push) Has been cancelled
CD Pipeline / tests (push) Has been cancelled
fix(ops): harden cold-start schedule recovery
2026-05-05 22:17:10 +08:00

44 lines
1.3 KiB
Python

#!/usr/bin/env python3
"""
KM Vectorize CronJob 入口腳本 — ADR-073 Phase 4-3
每日由 K8s CronJob 呼叫,對新增 KM 條目執行向量化,
確保 RAG 查詢可存取最新知識(飛輪「學習固化」節點)。
2026-04-12 ogt (ADR-073 Phase 4-3, P2-1 重構)
"""
import asyncio
import os
import sys
import httpx
async def main() -> int:
api_base = os.environ.get(
"INTERNAL_API_URL",
"http://awoooi-api-svc.awoooi-prod.svc.cluster.local:8000",
)
url = f"{api_base}/api/v1/knowledge/embed-all"
async with httpx.AsyncClient(timeout=1800) as client:
try:
resp = await client.post(url)
print(f"embed-all: {resp.status_code} {resp.text[:200]}")
if resp.status_code >= 400:
print(f"ERROR: embed-all returned {resp.status_code}", file=sys.stderr)
return 1
result = resp.json()
if int(result.get("failed", 0)) > 0:
print(f"ERROR: embed-all failed rows: {result}", file=sys.stderr)
return 1
return 0
except httpx.RequestError as exc:
print(f"ERROR: request failed — {exc}", file=sys.stderr)
return 1
if __name__ == "__main__":
sys.exit(asyncio.run(main()))