feat:新增投顾agent和nl2sqlagent

scripts/run_nl2sql_maintenance.py

@@ -0,0 +1,58 @@
+"""运行可由 Cron 或任务平台调用的 NL2SQL 运维任务。"""
+from __future__ import annotations
+
+import argparse
+import asyncio
+import json
+import sys
+from datetime import datetime, timedelta, timezone
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from config.deps import get_redis
+from nl2sql.jobs import run_consistency_check, run_history_cleanup, run_metadata_sync, run_vector_cleanup
+
+
+async def run(task: str) -> dict:
+    redis = next(get_redis())
+    if task == "metadata_sync":
+        result = await run_metadata_sync(redis=redis)
+    elif task == "vector_cleanup":
+        result = await run_vector_cleanup(redis=redis)
+    elif task == "consistency_check":
+        from scripts.check_nl2sql_consistency import collect_consistency
+
+        result = await run_consistency_check(redis=redis, worker=collect_consistency)
+    elif task == "history_cleanup":
+        from config.database.mysql import get_session_factory
+
+        async with get_session_factory()() as db:
+            result = await run_history_cleanup(
+                db,
+                datetime.now(timezone.utc).replace(tzinfo=None) - timedelta(days=180),
+                redis=redis,
+            )
+    else:
+        raise ValueError("不支持的 NL2SQL 运维任务")
+    return {
+        "name": result.name,
+        "status": result.status,
+        "attempts": result.attempts,
+        "detail": result.detail,
+        "error_type": result.error_type,
+    }
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="NL2SQL 运维任务")
+    parser.add_argument(
+        "task",
+        choices=["metadata_sync", "vector_cleanup", "consistency_check", "history_cleanup"],
+    )
+    args = parser.parse_args()
+    print(json.dumps(asyncio.run(run(args.task)), ensure_ascii=False))
+
+
+if __name__ == "__main__":
+    main()