Files
awoooi/scripts/cron_km_vectorize.py
OG T 0d239838b4
Some checks failed
CD Pipeline / build-and-deploy (push) Has been cancelled
fix(cr): Code Review P2 — 測試覆蓋 + CronJob 腳本重構
P2-1: CronJob inline Python 抽成 scripts/cron_km_vectorize.py
      Dockerfile 加入 COPY scripts/,CronJob YAML 改用腳本路徑
P2-2: 新增 test_classify_alert_early.py — 23 tests 覆蓋 7 條分類規則
      含邊界情況:VeleroBackupFailed(backup優先於k8s)、優先順序驗證

595 unit tests passed

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-04-12 15:14:44 +08:00

40 lines
1.1 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.awoooi-prod.svc.cluster.local:8000",
)
url = f"{api_base}/api/v1/knowledge/embed-all"
async with httpx.AsyncClient(timeout=120) 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
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()))