diff --git a/cucyuqing/cmd/risk-analyze.py b/cucyuqing/cmd/risk-analyze.py index 962e573..d431e01 100644 --- a/cucyuqing/cmd/risk-analyze.py +++ b/cucyuqing/cmd/risk-analyze.py @@ -1,7 +1,5 @@ from dataclasses import dataclass -from os import system -from typing import Iterable, Required -from typing_extensions import Doc +from typing import Iterable import openai import asyncio from openai.types.chat import ChatCompletionMessageParam @@ -16,6 +14,16 @@ from cucyuqing.dbscan import Document, run_dbscan async def main(): + while True: + try: + await do_analyze() + await asyncio.sleep(60 * 30) + except Exception as e: + print(e) + await asyncio.sleep(60*60) + + +async def do_analyze(): await asyncio.gather( pool.open(), mysql.connect(), @@ -54,6 +62,11 @@ async def main(): }, ) + await asyncio.gather( + pool.close(), + mysql.disconnect(), + ) + @dataclass class RiskType: