chore(rls): 新增 manual script gate 與 canary wave1
This commit is contained in:
@@ -22,15 +22,19 @@ VIA_API = "--via-api" in sys.argv
|
||||
DRY_RUN = "--dry-run" in sys.argv
|
||||
DATABASE_URL = os.environ.get("DATABASE_URL", "").replace("postgresql+asyncpg://", "postgresql://")
|
||||
API_BASE = os.environ.get("API_BASE", "http://localhost:8000")
|
||||
PROJECT_ID = os.environ.get("AWOOOP_PROJECT_ID", "awoooi")
|
||||
|
||||
|
||||
async def check_status():
|
||||
"""確認需要向量化的數量"""
|
||||
conn = await asyncpg.connect(DATABASE_URL)
|
||||
total = await conn.fetchval("SELECT count(*) FROM knowledge_entries")
|
||||
null_emb = await conn.fetchval("SELECT count(*) FROM knowledge_entries WHERE embedding IS NULL")
|
||||
await conn.close()
|
||||
return total, null_emb
|
||||
try:
|
||||
await conn.execute("SELECT set_config('app.project_id', $1, FALSE)", PROJECT_ID)
|
||||
total = await conn.fetchval("SELECT count(*) FROM knowledge_entries")
|
||||
null_emb = await conn.fetchval("SELECT count(*) FROM knowledge_entries WHERE embedding IS NULL")
|
||||
return total, null_emb
|
||||
finally:
|
||||
await conn.close()
|
||||
|
||||
|
||||
async def via_api():
|
||||
|
||||
Reference in New Issue
Block a user