Initial commit

llamaCpp.Wrapper.app/restart.py

@@ -0,0 +1,51 @@
+import asyncio
+import logging
+import shlex
+from dataclasses import dataclass
+from typing import Optional
+
+import httpx
+
+
+log = logging.getLogger("llamacpp_restart")
+
+
+@dataclass
+class RestartPlan:
+    method: str
+    command: Optional[str]
+    url: Optional[str]
+    allowed_container: Optional[str] = None
+
+
+async def trigger_restart(plan: RestartPlan, payload: Optional[dict] = None) -> None:
+    if plan.method == "none":
+        log.warning("Restart requested but restart method is none")
+        return
+    if plan.method == "http":
+        if not plan.url:
+            raise RuntimeError("restart url is required for http method")
+        async with httpx.AsyncClient(timeout=60) as client:
+            resp = await client.post(plan.url, json=payload or {})
+            resp.raise_for_status()
+        return
+    if plan.method == "docker":
+        if not plan.command:
+            raise RuntimeError("restart command must include container id or name for docker method")
+        if plan.allowed_container and plan.command != plan.allowed_container:
+            raise RuntimeError("docker restart command not allowed for non-target container")
+        async with httpx.AsyncClient(transport=httpx.AsyncHTTPTransport(uds="/var/run/docker.sock"), timeout=30) as client:
+            resp = await client.post(f"http://docker/containers/{plan.command}/restart")
+            resp.raise_for_status()
+        return
+    if plan.method == "shell":
+        if not plan.command:
+            raise RuntimeError("restart command is required for shell method")
+        cmd = plan.command
+        args = shlex.split(cmd)
+        proc = await asyncio.create_subprocess_exec(*args)
+        code = await proc.wait()
+        if code != 0:
+            raise RuntimeError(f"restart command failed with exit code {code}")
+        return
+    raise RuntimeError(f"unknown restart method {plan.method}")