add py control service

src/Vector/vector_control.py

@@ -0,0 +1,129 @@
+#!/usr/bin/env python3
+
+import argparse
+import importlib
+import json
+import os
+import subprocess
+import sys
+import time
+from pathlib import Path
+
+BASE_PATH = Path(__file__).resolve().parents[2]
+KNOWLEDGE_DIR = BASE_PATH / "var" / "knowledge"
+
+REQUIRED_MODULES = [
+    "fastapi",
+    "uvicorn",
+    "faiss",
+    "sentence_transformers",
+    "numpy",
+]
+
+VENV_PIP = BASE_PATH / ".venv" / "bin" / "pip"
+UVICORN_BIN = BASE_PATH / ".venv" / "bin" / "uvicorn"
+
+
+def check_modules():
+    missing = []
+    for module in REQUIRED_MODULES:
+        try:
+            importlib.import_module(module)
+        except Exception:
+            missing.append(module)
+    return missing
+
+
+def install_modules(modules):
+    if not modules:
+        return
+    subprocess.call([str(VENV_PIP), "install", *modules])
+
+
+def service_running():
+    result = subprocess.run(
+        ["ps", "aux"],
+        capture_output=True,
+        text=True
+    )
+    return "uvicorn src.Vector.vector_service:app" in result.stdout
+
+
+def start_service():
+    subprocess.Popen([
+        str(UVICORN_BIN),
+        "src.Vector.vector_service:app",
+        "--host", "0.0.0.0",
+        "--port", "8090"
+    ])
+    time.sleep(2)
+
+
+def reload_service():
+    subprocess.call([
+        "curl",
+        "-s",
+        "-X",
+        "POST",
+        "http://127.0.0.1:8090/reload"
+    ])
+
+
+def health_check():
+    try:
+        result = subprocess.run(
+            ["curl", "-s", "http://127.0.0.1:8090/health"],
+            capture_output=True,
+            text=True
+        )
+        return result.stdout.strip()
+    except Exception:
+        return None
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--install", action="store_true")
+    parser.add_argument("--start", action="store_true")
+    parser.add_argument("--reload", action="store_true")
+
+    args = parser.parse_args()
+
+    result = {
+        "modules_missing": [],
+        "service_running": False,
+        "health": None,
+        "actions": []
+    }
+
+    # 1️⃣ Check modules
+    missing = check_modules()
+    result["modules_missing"] = missing
+
+    if missing and args.install:
+        install_modules(missing)
+        result["actions"].append("modules_installed")
+
+    # 2️⃣ Service check
+    running = service_running()
+    result["service_running"] = running
+
+    if not running and args.start:
+        start_service()
+        result["actions"].append("service_started")
+        running = True
+
+    # 3️⃣ Reload
+    if args.reload:
+        reload_service()
+        result["actions"].append("service_reloaded")
+
+    # 4️⃣ Health
+    if running:
+        result["health"] = health_check()
+
+    print(json.dumps(result, indent=2))
+
+
+if __name__ == "__main__":
+    main()