Backup automatico script del 2026-08-23 12:04

services/loogle-mcp/scripts/run_gitea_index_once.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""Indicizzazione Gitea one-shot — eseguire con indexer fermo per evitare OOM."""
+
+from __future__ import annotations
+
+import argparse
+import json
+import os
+import sys
+
+ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if ROOT not in sys.path:
+    sys.path.insert(0, ROOT)
+
+
+def _load_dotenv() -> None:
+    env_path = os.path.join(ROOT, ".env")
+    if not os.path.isfile(env_path):
+        return
+    with open(env_path, encoding="utf-8") as fh:
+        for line in fh:
+            line = line.strip()
+            if not line or line.startswith("#") or "=" not in line:
+                continue
+            key, _, value = line.partition("=")
+            if key.strip() and key.strip() not in os.environ:
+                os.environ[key.strip()] = value.strip().strip("'").strip('"')
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description="Indicizza repo Gitea nel vector store")
+    parser.add_argument("--repo", help="owner/name (default: tutti i repo configurati)")
+    parser.add_argument("--user", default="daniele", help="Utente MCP/Gitea token")
+    parser.add_argument("--max-files", type=int, default=0, help="Limite file per repo (0=config)")
+    parser.add_argument("--force", action="store_true", help="Re-indicizza anche file invariati")
+    args = parser.parse_args()
+
+    _load_dotenv()
+    if os.path.isfile("/.dockerenv"):
+        os.environ.setdefault("MCP_DB", "/data/loogle_mcp.db")
+        os.environ.setdefault("MCP_VECTOR_FALLBACK", "/data/vector_fallback.db")
+
+    from app.db import init_db
+    from app.knowledge import gitea_indexer
+
+    init_db()
+    max_files = args.max_files or None
+
+    if args.repo:
+        result = gitea_indexer.index_repo(
+            args.repo,
+            username=args.user,
+            force=args.force,
+            max_files=max_files,
+        )
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+    else:
+        result = gitea_indexer.index_all(max_files_per_repo=max_files)
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+
+    stats = gitea_indexer.index_stats()
+    print("stats", json.dumps(stats, ensure_ascii=False))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())