diff --git a/backend/app.py b/backend/app.py index ed591d4..a79e2de 100644 --- a/backend/app.py +++ b/backend/app.py @@ -34,11 +34,9 @@ AI_CHAT_SYSTEM_PROMPT = os.getenv( "You are the Titan Lab assistant for bstein.dev. Be concise and helpful.", ) AI_CHAT_TIMEOUT_SEC = float(os.getenv("AI_CHAT_TIMEOUT_SEC", "20")) -AI_NODE_NAME = os.getenv("AI_NODE_NAME") or os.getenv("NODE_NAME") or "unknown" -try: - AI_NODE_GPU_MAP = json.loads(os.getenv("AI_NODE_GPU_MAP", "{}")) -except json.JSONDecodeError: - AI_NODE_GPU_MAP = {} +AI_NODE_NAME = os.getenv("AI_CHAT_NODE_NAME") or os.getenv("AI_NODE_NAME") or "ai-cluster" +AI_GPU_DESC = os.getenv("AI_CHAT_GPU_DESC") or "local GPU (dynamic)" +AI_PUBLIC_ENDPOINT = os.getenv("AI_PUBLIC_CHAT_ENDPOINT", "https://chat.ai.bstein.dev/api/ai/chat") _LAB_STATUS_CACHE: dict[str, Any] = {"ts": 0.0, "value": None} @@ -188,13 +186,12 @@ def ai_chat() -> Any: @app.route("/api/ai/info", methods=["GET"]) def ai_info() -> Any: - gpu_label = AI_NODE_GPU_MAP.get(AI_NODE_NAME, "local GPU (dynamic)") return jsonify( { "node": AI_NODE_NAME, - "gpu": gpu_label, + "gpu": AI_GPU_DESC, "model": AI_CHAT_MODEL, - "endpoint": "/api/ai/chat", + "endpoint": AI_PUBLIC_ENDPOINT or "/api/ai/chat", } ) diff --git a/frontend/src/views/AiView.vue b/frontend/src/views/AiView.vue index 05e080f..da8c76a 100644 --- a/frontend/src/views/AiView.vue +++ b/frontend/src/views/AiView.vue @@ -20,13 +20,13 @@ {{ meta.gpu }}