diff --git a/cucyuqing/cmd/risk-analyze.py b/cucyuqing/cmd/risk-analyze.py index ac7c5a4..4d2ebe9 100644 --- a/cucyuqing/cmd/risk-analyze.py +++ b/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(