添加 cmd.risk-analyze

cucyuqing/cmd/risk-analyze.py

@@ -10,19 +10,16 @@ from cucyuqing.utils import print
 from cucyuqing.config import OPENAI_RISK_LLM_API_KEY, OPENAI_RISK_LLM_BASE_URL
 from cucyuqing.pg import pool, get_cur
 from cucyuqing.mysql import mysql
+from cucyuqing.dbscan import run_dbscan
 
 
 async def main():
     await pool.open()
-    print(await batch_risk_analyze(["你是老师", "我是初音未来"]))
-
-async def get_docs() -> list[dict]:
-    # [TODO]
-    raise NotImplemented
-    await mysql.execute("""
-    """)
-    return []
-
+    dbscan_result = await run_dbscan()
+    docs = [cluster[0] for cluster in dbscan_result.clusters]
+    analyze_rusult = await batch_risk_analyze([doc.title for doc in docs])
+    for result, doc in zip(analyze_rusult, docs):
+        print(f"风险: {result} 标题: {doc.title}")
 
 
 async def batch_risk_analyze(