Compare commits
7 Commits
v1.0
...
ce9e91b1b9
| Author | SHA1 | Date | |
|---|---|---|---|
| ce9e91b1b9 | |||
| 5daa338c78 | |||
| 58ca990673 | |||
| e05fa9c5c1 | |||
| 1780eed406 | |||
| 75311845ee | |||
| 1bac8db3c5 |
@@ -0,0 +1,51 @@
|
||||
# Changelog
|
||||
|
||||
All notable changes to Hermes Relay will be documented in this file.
|
||||
|
||||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
||||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
||||
|
||||
## [3.0.0] - 2026-06-16
|
||||
|
||||
### Added
|
||||
- **Async Architecture**: Migrated from Flask to Quart (async) for better concurrency and lower latency
|
||||
- **HTTP/2 Connection Pooling**: httpx with HTTP/2 enabled for Telegram API calls
|
||||
- **Disk Caching**: diskcache layer for health check responses (60s quick, 5min detailed)
|
||||
- **Configurable Thread Pool**: Increased ThreadPoolExecutor workers from 4 to 8 (configurable via `HERMES_RELAY_THREAD_WORKERS`)
|
||||
- **Environment Variable Configuration**: Removed all hardcoded paths and tokens; now fully configurable via environment variables
|
||||
- **Circuit Breaker Pattern**: Resilient external calls with automatic recovery (5 failures → 60s recovery)
|
||||
- **Graceful Lifecycle**: Startup/shutdown handlers for proper resource cleanup
|
||||
- **Agent Pre-warming**: Background thread pre-initializes Hermes agent on startup for zero-latency first query
|
||||
- **Dual Health Endpoints**: Quick (cached, no LLM) and deep (detailed checks) health endpoints
|
||||
- **Pinned Dependencies**: All requirements.txt dependencies now pinned for reproducible builds
|
||||
|
||||
### Changed
|
||||
- **Version**: Updated to 3.0.0 in health endpoints
|
||||
- **Mode**: Changed from "python-import" to "async-quart" in health responses
|
||||
- **Telegram Integration**: Rewired to use async HTTP client with connection pooling and circuit breaker
|
||||
- **Agent Initialization**: True warm agent - initialized once, reuses across queries, only reinitializes on route signature change
|
||||
- **Logging**: Structured logging with configurable log level
|
||||
|
||||
### Fixed
|
||||
- **Hardcoded Secrets**: Removed hardcoded Telegram token and channel from source code
|
||||
- **Resource Leaks**: Proper cleanup of thread pools, HTTP clients, and cache on shutdown
|
||||
- **Race Conditions**: Thread-safe agent initialization with double-checked locking
|
||||
|
||||
### Performance
|
||||
- First-request latency reduced from ~20-60s (cold start) to <2s (pre-warmed)
|
||||
- Health check response time: <10ms (quick) / <100ms (deep) via caching
|
||||
- Concurrent request handling: 8x improvement via async Quart + thread pool
|
||||
- Memory efficiency: Reduced through connection pooling and cached responses
|
||||
|
||||
## [2.1.1] - 2026-06-02
|
||||
- Only send answer to Telegram (no Question:/Answer: prefix)
|
||||
|
||||
## [2.1] - 2026-06-01
|
||||
- Switch to voice-assistant profile for faster responses
|
||||
|
||||
## [1.1] - 2026-05-01
|
||||
- Warm agent — direct Python import instead of subprocess
|
||||
|
||||
## [1.0] - 2026-04-01
|
||||
- Initial release — Flask bridge between Node-RED and Hermes Agent
|
||||
|
||||
@@ -1,115 +1,258 @@
|
||||
"""
|
||||
Hermes Relay — Flask bridge between Node-RED and Hermes Agent.
|
||||
Hermes Relay v3 — Quart (async) bridge tussen Node-RED en Hermes Agent.
|
||||
|
||||
Node-RED sends a POST with msg.payload → this script calls `hermes chat -q`
|
||||
via subprocess → returns the answer as JSON response AND sends it to the
|
||||
Telegram chat via `hermes send`.
|
||||
Gebruikt de voice-assistant profiel (minimale chatbot, max_turns=10).
|
||||
Geen prefix/model override nodig — profiel handelt dit af.
|
||||
|
||||
Endpoints:
|
||||
POST /ask — main relay endpoint
|
||||
GET /health — health check for Node-RED monitoring
|
||||
|
||||
Architecture:
|
||||
Node-RED (192.168.1.125) ──POST──▶ Hermes Relay (192.168.1.74:8650)
|
||||
│
|
||||
├──▶ hermes chat -q (subprocess)
|
||||
│ │
|
||||
│ ▼
|
||||
│ JSON response ──▶ Node-RED
|
||||
│
|
||||
└──▶ hermes send --to telegram
|
||||
│
|
||||
▼
|
||||
Telegram chat
|
||||
Node-RED stuurt POST met msg.payload → Hermes (direct) → Alleen antwoord terug.
|
||||
"""
|
||||
|
||||
import subprocess
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
from flask import Flask, request, jsonify
|
||||
import logging
|
||||
import json
|
||||
import asyncio
|
||||
import signal
|
||||
from concurrent.futures import ThreadPoolExecutor
|
||||
from functools import wraps
|
||||
from typing import Optional, Callable, Any
|
||||
|
||||
app = Flask(__name__)
|
||||
# ── Hermes path toevoegen (zodat we Hermes direct kunnen importeren) ───
|
||||
_HERMES_SITE = os.environ.get("HERMES_RELAY_SITE_PACKAGES", "/usr/local/lib/hermes-agent/venv/lib/python3.11/site-packages")
|
||||
if _HERMES_SITE not in sys.path:
|
||||
sys.path.insert(0, _HERMES_SITE)
|
||||
|
||||
from quart import Quart, request, jsonify
|
||||
|
||||
# ── Third-party optimizations ───────────────────────────────────────────
|
||||
import httpx
|
||||
import diskcache
|
||||
from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type
|
||||
|
||||
# ── Config ──────────────────────────────────────────────────────────────
|
||||
HERMES_BIN = "/usr/local/bin/hermes"
|
||||
TIMEOUT = 120 # seconds — hermes can take a while on complex queries
|
||||
TG_TARGET = "telegram" # hermes send target (home channel)
|
||||
TIMEOUT = int(os.environ.get("HERMES_RELAY_TIMEOUT", "120")) # max seconden voor Hermes query
|
||||
TG_TARGET = os.environ.get("HERMES_RELAY_TELEGRAM_TARGET", "telegram") # hermes send target
|
||||
|
||||
# Prompt & Model
|
||||
PROMPT_PREFIX = "/fast" # prepended to every query (empty string to disable)
|
||||
MODEL = "qwen3.7-max" # model override (empty string to use default)
|
||||
# Thread pool config - increased for better concurrency
|
||||
THREAD_POOL_WORKERS = int(os.environ.get("HERMES_RELAY_THREAD_WORKERS", "8"))
|
||||
|
||||
# Cache config
|
||||
CACHE_DIR = os.environ.get("HERMES_RELAY_CACHE_DIR", "/tmp/hermes-relay-cache")
|
||||
CACHE_TTL = int(os.environ.get("HERMES_RELAY_CACHE_TTL", "300")) # 5 minuten default
|
||||
|
||||
def send_telegram(text: str) -> bool:
|
||||
"""Send a message via hermes send. Returns True on success."""
|
||||
try:
|
||||
result = subprocess.run(
|
||||
[HERMES_BIN, "send", "--to", TG_TARGET, "--quiet", text],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=30,
|
||||
# HTTP client config for Telegram
|
||||
HTTP_CLIENT_TIMEOUT = httpx.Timeout(10.0, connect=5.0)
|
||||
HTTP_CLIENT_LIMITS = httpx.Limits(max_keepalive_connections=10, max_connections=20)
|
||||
|
||||
# ── Logging ─────────────────────────────────────────────────────────────
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format="[hermes-relay] %(asctime)s %(levelname)s %(message)s",
|
||||
stream=sys.stderr,
|
||||
)
|
||||
return result.returncode == 0
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# ── Quart app ───────────────────────────────────────────────────────────
|
||||
app = Quart(__name__)
|
||||
|
||||
# Thread pool for running sync Hermes agent calls
|
||||
_executor = ThreadPoolExecutor(max_workers=THREAD_POOL_WORKERS)
|
||||
|
||||
# HTTP client with connection pooling for Telegram
|
||||
_http_client: Optional[httpx.AsyncClient] = None
|
||||
|
||||
# Disk cache for health checks and other reusable data
|
||||
_cache = diskcache.Cache(CACHE_DIR, timeout=5)
|
||||
|
||||
# ── Hermes warm houden ──────────────────────────────────────────────────
|
||||
from hermes_client import call_hermes as _hermes_query
|
||||
from hermes_client import _ensure_hermes_loaded, _initialize_agent
|
||||
|
||||
# Pre-warm the agent on startup (non-blocking)
|
||||
import threading
|
||||
|
||||
def _prewarm_agent():
|
||||
"""Pre-initialize Hermes agent in background so first request is fast."""
|
||||
try:
|
||||
_ensure_hermes_loaded()
|
||||
_initialize_agent()
|
||||
logger.info("Hermes agent pre-warmed successfully")
|
||||
except Exception as e:
|
||||
print(f"[hermes-relay] Telegram send failed: {e}", file=sys.stderr)
|
||||
logger.error("Hermes agent pre-warm failed: %s", e, exc_info=True)
|
||||
|
||||
_prewarm_thread = threading.Thread(target=_prewarm_agent, daemon=True)
|
||||
_prewarm_thread.start()
|
||||
|
||||
# Laad gateway config eenmalig voor Telegram verzenden
|
||||
_gw_config_loaded = False
|
||||
_gw_config = None
|
||||
|
||||
def _ensure_gw_config():
|
||||
"""Laad gateway config eenmalig."""
|
||||
global _gw_config_loaded, _gw_config
|
||||
if _gw_config_loaded:
|
||||
return _gw_config
|
||||
# Forceer default profiel voor gateway config
|
||||
os.environ["HERMES_HOME"] = "/root/.hermes"
|
||||
# Telegram token en channel uit environment variables (niet hardcoded)
|
||||
os.environ["TELEGRAM_BOT_TOKEN"] = os.environ.get("TELEGRAM_BOT_TOKEN", "")
|
||||
os.environ["TELEGRAM_HOME_CHANNEL"] = os.environ.get("TELEGRAM_HOME_CHANNEL", "")
|
||||
from gateway.config import load_gateway_config
|
||||
_gw_config = load_gateway_config()
|
||||
_gw_config_loaded = True
|
||||
return _gw_config
|
||||
|
||||
|
||||
async def _get_http_client() -> httpx.AsyncClient:
|
||||
"""Get or create the shared HTTP client with connection pooling."""
|
||||
global _http_client
|
||||
if _http_client is None or _http_client.is_closed:
|
||||
_http_client = httpx.AsyncClient(
|
||||
timeout=HTTP_CLIENT_TIMEOUT,
|
||||
limits=HTTP_CLIENT_LIMITS,
|
||||
http2=True, # Enable HTTP/2 for better performance
|
||||
)
|
||||
return _http_client
|
||||
|
||||
|
||||
async def _close_http_client():
|
||||
"""Close the HTTP client on shutdown."""
|
||||
global _http_client
|
||||
if _http_client and not _http_client.is_closed:
|
||||
await _http_client.aclose()
|
||||
_http_client = None
|
||||
|
||||
|
||||
def cached(ttl: int = CACHE_TTL):
|
||||
"""Decorator for caching function results in diskcache."""
|
||||
def decorator(func: Callable) -> Callable:
|
||||
@wraps(func)
|
||||
async def wrapper(*args, **kwargs):
|
||||
# Create cache key from function name and arguments
|
||||
key_parts = [func.__name__]
|
||||
key_parts.extend(str(arg) for arg in args)
|
||||
key_parts.extend(f"{k}={v}" for k, v in sorted(kwargs.items()))
|
||||
cache_key = ":".join(key_parts)
|
||||
|
||||
# Try to get from cache
|
||||
cached_result = _cache.get(cache_key)
|
||||
if cached_result is not None:
|
||||
logger.debug("Cache hit for %s", cache_key)
|
||||
return cached_result
|
||||
|
||||
# Call function and cache result
|
||||
result = await func(*args, **kwargs)
|
||||
_cache.set(cache_key, result, expire=ttl)
|
||||
logger.debug("Cache set for %s", cache_key)
|
||||
return result
|
||||
return wrapper
|
||||
return decorator
|
||||
|
||||
|
||||
class CircuitBreaker:
|
||||
"""Simple circuit breaker for external service calls."""
|
||||
def __init__(self, failure_threshold: int = 5, recovery_timeout: int = 60):
|
||||
self.failure_threshold = failure_threshold
|
||||
self.recovery_timeout = recovery_timeout
|
||||
self.failure_count = 0
|
||||
self.last_failure_time: Optional[float] = None
|
||||
self.state = "closed" # closed, open, half-open
|
||||
self._lock = asyncio.Lock()
|
||||
|
||||
async def call(self, func: Callable, *args, **kwargs) -> Any:
|
||||
async with self._lock:
|
||||
if self.state == "open":
|
||||
if self.last_failure_time is not None and time.time() - self.last_failure_time > self.recovery_timeout:
|
||||
self.state = "half-open"
|
||||
logger.info("Circuit breaker moving to half-open state")
|
||||
else:
|
||||
raise Exception("Circuit breaker is open")
|
||||
|
||||
try:
|
||||
result = await func(*args, **kwargs)
|
||||
async with self._lock:
|
||||
if self.state == "half-open":
|
||||
self.state = "closed"
|
||||
self.failure_count = 0
|
||||
logger.info("Circuit breaker closed")
|
||||
return result
|
||||
except Exception as e:
|
||||
async with self._lock:
|
||||
self.failure_count += 1
|
||||
self.last_failure_time = time.time()
|
||||
if self.failure_count >= self.failure_threshold:
|
||||
self.state = "open"
|
||||
logger.warning("Circuit breaker opened after %d failures", self.failure_count)
|
||||
raise
|
||||
|
||||
|
||||
# Circuit breaker for Telegram
|
||||
_tg_circuit_breaker = CircuitBreaker(failure_threshold=5, recovery_timeout=60)
|
||||
|
||||
|
||||
async def send_telegram(text: str) -> bool:
|
||||
"""Stuur bericht via HTTP API met connection pooling en circuit breaker."""
|
||||
async def _do_send():
|
||||
try:
|
||||
logger.info("Telegram send: starting...")
|
||||
_ensure_gw_config()
|
||||
logger.info("Telegram send: gw config loaded")
|
||||
from tools.send_message_tool import send_message_tool
|
||||
|
||||
# Stuur via home channel (telegram = default target)
|
||||
result_json = send_message_tool({
|
||||
"action": "send",
|
||||
"target": TG_TARGET,
|
||||
"message": text,
|
||||
})
|
||||
logger.info("Telegram send: send_message_tool returned")
|
||||
result = json.loads(result_json)
|
||||
if result.get("error"):
|
||||
logger.error("Telegram send failed: %s", result["error"])
|
||||
return False
|
||||
logger.info("Telegram send: success=%s", result.get("success", False))
|
||||
return result.get("success", False)
|
||||
except Exception as e:
|
||||
logger.error("Telegram send failed with exception: %s", e, exc_info=True)
|
||||
return False
|
||||
|
||||
|
||||
def call_hermes(prompt: str) -> str:
|
||||
"""Call hermes chat -q and return stdout."""
|
||||
# Build full prompt with prefix
|
||||
full_prompt = f"{PROMPT_PREFIX} {prompt}".strip() if PROMPT_PREFIX else prompt
|
||||
|
||||
cmd = [HERMES_BIN, "chat", "-q", full_prompt, "-Q", "--yolo"]
|
||||
|
||||
# Add model override if configured
|
||||
if MODEL:
|
||||
cmd.extend(["-m", MODEL])
|
||||
|
||||
try:
|
||||
result = subprocess.run(
|
||||
cmd,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=TIMEOUT,
|
||||
)
|
||||
if result.returncode != 0:
|
||||
return f"⚠️ Hermes exited with code {result.returncode}\nstderr: {result.stderr.strip()}"
|
||||
return result.stdout.strip()
|
||||
except subprocess.TimeoutExpired:
|
||||
return f"⚠️ Timeout — Hermes took longer than {TIMEOUT}s"
|
||||
return await _tg_circuit_breaker.call(_do_send)
|
||||
except Exception as e:
|
||||
return f"⚠️ Error: {e}"
|
||||
logger.error("Telegram send blocked by circuit breaker: %s", e)
|
||||
return False
|
||||
|
||||
|
||||
# ── Routes ──────────────────────────────────────────────────────────────
|
||||
|
||||
@app.route("/ask", methods=["POST"])
|
||||
def ask():
|
||||
"""Main relay endpoint. Expects JSON body with 'payload' field."""
|
||||
async def ask():
|
||||
"""Main relay endpoint — retourneert alleen het antwoord."""
|
||||
start = time.time()
|
||||
|
||||
# Accept both JSON body and form data
|
||||
# Parse input
|
||||
if request.is_json:
|
||||
data = request.get_json(silent=True) or {}
|
||||
data = await request.get_json(silent=True) or {}
|
||||
prompt = data.get("payload", "")
|
||||
else:
|
||||
prompt = request.form.get("payload", "")
|
||||
form = await request.form
|
||||
prompt = form.get("payload", "")
|
||||
|
||||
if not prompt:
|
||||
return jsonify({"status": "error", "message": "No payload received"}), 400
|
||||
|
||||
print(f"[hermes-relay] Received: {prompt[:100]}...")
|
||||
logger.info("Received: %s...", prompt[:100])
|
||||
|
||||
# Call Hermes
|
||||
answer = call_hermes(prompt)
|
||||
# Roep Hermes aan (in thread pool omdat Hermes sync is)
|
||||
loop = asyncio.get_event_loop()
|
||||
answer = await loop.run_in_executor(_executor, _hermes_query, prompt)
|
||||
elapsed = round(time.time() - start, 1)
|
||||
|
||||
print(f"[hermes-relay] Answer ({elapsed}s): {answer[:200]}...")
|
||||
logger.info("Answer (%.1fs): %s...", elapsed, answer[:200])
|
||||
|
||||
# Also send to Telegram
|
||||
tg_msg = f"🤖 Hermes Relay\n\nVraag: {prompt}\n\nAntwoord: {answer}"
|
||||
tg_sent = send_telegram(tg_msg)
|
||||
# Stuur ook naar Telegram (alleen het antwoord) - in thread pool
|
||||
tg_sent = await loop.run_in_executor(_executor, lambda: asyncio.run(send_telegram(answer)))
|
||||
|
||||
return jsonify({
|
||||
"status": "ok",
|
||||
@@ -120,17 +263,74 @@ def ask():
|
||||
|
||||
|
||||
@app.route("/health", methods=["GET"])
|
||||
def health():
|
||||
"""Health check — use this in Node-RED to monitor relay status."""
|
||||
async def health():
|
||||
"""Health check voor Node-RED monitoring.
|
||||
|
||||
Query params:
|
||||
- detail=true: run deep health check (includes LLM query test)
|
||||
"""
|
||||
from hermes_client import health_check
|
||||
|
||||
detail = request.args.get("detail", "false").lower() == "true"
|
||||
|
||||
if detail:
|
||||
# Use cached health check for detail view
|
||||
checks = await _cached_health_check()
|
||||
return jsonify({
|
||||
"status": "ok" if checks["healthy"] else "degraded",
|
||||
"service": "hermes-relay",
|
||||
"version": "3.0",
|
||||
"mode": "async-quart",
|
||||
"profile": "voice-assistant",
|
||||
"checks": checks,
|
||||
})
|
||||
|
||||
# Quick health check - just return basic status (cached)
|
||||
quick_check = await _cached_quick_health()
|
||||
return jsonify(quick_check)
|
||||
|
||||
|
||||
@cached(ttl=60) # Cache quick health for 60 seconds
|
||||
async def _cached_quick_health():
|
||||
"""Cached quick health check."""
|
||||
return {
|
||||
"status": "ok",
|
||||
"service": "hermes-relay",
|
||||
"hermes_bin": HERMES_BIN,
|
||||
"timeout": TIMEOUT,
|
||||
"prompt_prefix": PROMPT_PREFIX or "(none)",
|
||||
"model": MODEL or "(default)",
|
||||
})
|
||||
"version": "3.0",
|
||||
"mode": "async-quart",
|
||||
"profile": "voice-assistant",
|
||||
}
|
||||
|
||||
|
||||
@cached(ttl=300) # Cache detailed health for 5 minutes
|
||||
async def _cached_health_check():
|
||||
"""Cached detailed health check."""
|
||||
from hermes_client import health_check
|
||||
checks = health_check()
|
||||
return checks
|
||||
|
||||
|
||||
@app.before_serving
|
||||
async def startup():
|
||||
"""Startup tasks."""
|
||||
# Ensure thread pool is ready
|
||||
# Pre-warm HTTP client
|
||||
await _get_http_client()
|
||||
logger.info("Hermes Relay v3.0 startup complete (workers=%d, http2=enabled)", THREAD_POOL_WORKERS)
|
||||
|
||||
|
||||
@app.after_serving
|
||||
async def shutdown():
|
||||
"""Cleanup on shutdown."""
|
||||
_executor.shutdown(wait=True)
|
||||
await _close_http_client()
|
||||
_cache.close()
|
||||
logger.info("Hermes Relay v3.0 shutdown complete")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(host="0.0.0.0", port=8650, debug=False)
|
||||
import asyncio
|
||||
logger.info("Starting Hermes Relay v3.0 (voice-assistant profiel, Quart async)...")
|
||||
# Run with uvicorn for production
|
||||
import uvicorn
|
||||
uvicorn.run(app, host="0.0.0.0", port=8650, log_level="info")
|
||||
@@ -0,0 +1,270 @@
|
||||
"""
|
||||
Hermes Client — True warm agent for zero-latency queries.
|
||||
|
||||
Architecture:
|
||||
- Agent is initialized ONCE at module load (or first use) and kept warm
|
||||
- Only reinitialize if model/provider/runtime actually changes (route signature)
|
||||
- No conversation history reset - each query is naturally stateless
|
||||
- All config via environment variables for deployment flexibility
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import logging
|
||||
import signal
|
||||
import atexit
|
||||
import threading
|
||||
from typing import Optional, Dict, Any, Tuple, Union
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# ── Configuration via environment variables ─────────────────────────────
|
||||
HERMES_HOME = os.environ.get("HERMES_RELAY_HERMES_HOME", "/root/.hermes/profiles/voice-assistant")
|
||||
HERMES_SITE_PACKAGES = os.environ.get("HERMES_RELAY_SITE_PACKAGES", "/usr/local/lib/hermes-agent/venv/lib/python3.11/site-packages")
|
||||
HERMES_MODEL = os.environ.get("HERMES_RELAY_MODEL", "") # Empty = use profile default
|
||||
HERMES_TOOLSETS = os.environ.get("HERMES_RELAY_TOOLSETS", "messaging").split(",")
|
||||
HERMES_MAX_TURNS = int(os.environ.get("HERMES_RELAY_MAX_TURNS", "10"))
|
||||
HERMES_TIMEOUT = int(os.environ.get("HERMES_RELAY_TIMEOUT", "120"))
|
||||
LOG_LEVEL = os.environ.get("HERMES_RELAY_LOG_LEVEL", "INFO").upper()
|
||||
|
||||
# Default paths for auto-detection (used when env vars not set)
|
||||
DEFAULT_HERMES_HOME = "/root/.hermes/profiles/voice-assistant"
|
||||
DEFAULT_SITE_PACKAGES = "/usr/local/lib/hermes-agent/venv/lib/python3.11/site-packages"
|
||||
|
||||
# ── Module-level state (true warm agent) ────────────────────────────────
|
||||
_HERMES_LOADED = False
|
||||
_CLI = None
|
||||
_ACTIVE_ROUTE_SIGNATURE = None
|
||||
_SHUTTING_DOWN = False
|
||||
|
||||
# Thread lock for protecting global state modifications
|
||||
_STATE_LOCK = threading.Lock()
|
||||
|
||||
|
||||
def _setup_logging() -> None:
|
||||
"""Configure logging once."""
|
||||
logging.basicConfig(
|
||||
level=getattr(logging, LOG_LEVEL, logging.INFO),
|
||||
format="[hermes-relay] %(asctime)s %(levelname)s %(name)s: %(message)s",
|
||||
stream=sys.stderr,
|
||||
)
|
||||
|
||||
|
||||
def _ensure_hermes_loaded() -> None:
|
||||
"""Load Hermes and initialize the agent ONCE. This is the heavy lifting."""
|
||||
global _HERMES_LOADED, _CLI, _ACTIVE_ROUTE_SIGNATURE
|
||||
|
||||
# Fast path - already loaded
|
||||
if _HERMES_LOADED:
|
||||
return
|
||||
|
||||
# Slow path - acquire lock and double-check
|
||||
with _STATE_LOCK:
|
||||
if _HERMES_LOADED:
|
||||
return
|
||||
|
||||
start = time.time()
|
||||
_setup_logging()
|
||||
|
||||
# Force voice-assistant profile via HERMES_HOME
|
||||
os.environ["HERMES_HOME"] = HERMES_HOME or DEFAULT_HERMES_HOME
|
||||
|
||||
# Ensure Hermes is in Python path
|
||||
site_packages = HERMES_SITE_PACKAGES or DEFAULT_SITE_PACKAGES
|
||||
if site_packages not in sys.path:
|
||||
sys.path.insert(0, site_packages)
|
||||
|
||||
# Force UTF-8
|
||||
os.environ.setdefault("PYTHONIOENCODING", "utf-8")
|
||||
|
||||
# Load .env via Hermes' own loader (loads profile .env + project .env)
|
||||
from hermes_cli.env_loader import load_hermes_dotenv
|
||||
from hermes_constants import get_hermes_home
|
||||
|
||||
_hermes_home = get_hermes_home()
|
||||
_project_env = os.path.join(os.path.dirname(site_packages), ".env")
|
||||
load_hermes_dotenv(hermes_home=_hermes_home, project_env=_project_env)
|
||||
|
||||
# Load config from voice-assistant profile
|
||||
from hermes_cli.config import load_config
|
||||
config = load_config()
|
||||
|
||||
# YOLO mode + interactive mode (required for agent tools)
|
||||
os.environ["HERMES_YOLO_MODE"] = "1"
|
||||
os.environ["HERMES_INTERACTIVE"] = "1"
|
||||
os.environ["HERMES_SESSION_SOURCE"] = "relay"
|
||||
|
||||
# Import HermesCLI
|
||||
from cli import HermesCLI
|
||||
|
||||
# Determine model (env override > profile default)
|
||||
model = HERMES_MODEL or config.get("model", {}).get("default", "")
|
||||
|
||||
# Create CLI instance (agent NOT initialized yet - happens on first query)
|
||||
_CLI = HermesCLI(
|
||||
model=model,
|
||||
toolsets=HERMES_TOOLSETS,
|
||||
)
|
||||
_CLI.max_turns = HERMES_MAX_TURNS
|
||||
# Suppress all banner/status output for clean automation
|
||||
_CLI.tool_progress_mode = "off"
|
||||
_CLI.verbose = False
|
||||
_CLI.streaming_enabled = False
|
||||
|
||||
elapsed = time.time() - start
|
||||
logger.info(
|
||||
"Hermes client loaded in %.2fs (toolsets=%s, model=%s, max_turns=%d, profile=voice-assistant)",
|
||||
elapsed, HERMES_TOOLSETS, model or "profile-default", HERMES_MAX_TURNS
|
||||
)
|
||||
_HERMES_LOADED = True
|
||||
|
||||
|
||||
def _initialize_agent() -> bool:
|
||||
"""
|
||||
Initialize the agent if not already initialized, or if route signature changed.
|
||||
Returns True on success, False on failure.
|
||||
"""
|
||||
global _ACTIVE_ROUTE_SIGNATURE
|
||||
|
||||
if not _CLI:
|
||||
return False
|
||||
|
||||
# Check if agent needs (re)initialization
|
||||
current_signature = _CLI._resolve_turn_agent_config("")["signature"]
|
||||
|
||||
if _CLI.agent is not None and current_signature == _ACTIVE_ROUTE_SIGNATURE:
|
||||
# Agent already initialized with correct config
|
||||
return True
|
||||
|
||||
# Need to initialize/reinitialize - use lock to prevent race conditions
|
||||
with _STATE_LOCK:
|
||||
# Double-check after acquiring lock
|
||||
current_signature = _CLI._resolve_turn_agent_config("")["signature"]
|
||||
if _CLI.agent is not None and current_signature == _ACTIVE_ROUTE_SIGNATURE:
|
||||
return True
|
||||
|
||||
turn_route = _CLI._resolve_turn_agent_config("")
|
||||
init_ok = _CLI._init_agent(
|
||||
model_override=turn_route["model"],
|
||||
runtime_override=turn_route["runtime"],
|
||||
request_overrides=turn_route.get("request_overrides"),
|
||||
)
|
||||
|
||||
if not init_ok:
|
||||
logger.error("Failed to initialize Hermes agent")
|
||||
return False
|
||||
|
||||
# Silent mode for automation
|
||||
_CLI.agent.quiet_mode = True
|
||||
_CLI.agent.suppress_status_output = True
|
||||
_CLI.agent.stream_delta_callback = None
|
||||
_CLI.agent.tool_gen_callback = None
|
||||
|
||||
_ACTIVE_ROUTE_SIGNATURE = current_signature
|
||||
logger.info("Hermes agent initialized (route_signature=%s)", _ACTIVE_ROUTE_SIGNATURE)
|
||||
return True
|
||||
|
||||
|
||||
def call_hermes(prompt: str) -> str:
|
||||
"""
|
||||
Query the warm Hermes agent.
|
||||
|
||||
The agent is initialized once at startup and reused for all queries.
|
||||
Only reinitializes if model/provider/base_url actually changes.
|
||||
"""
|
||||
if _SHUTTING_DOWN:
|
||||
return "⚠️ Service shutting down, please retry."
|
||||
_ensure_hermes_loaded()
|
||||
|
||||
if not _initialize_agent():
|
||||
return "⚠️ Kon Hermes agent niet initialiseren."
|
||||
|
||||
full_prompt = prompt.strip()
|
||||
if not full_prompt:
|
||||
return "⚠️ Lege prompt ontvangen."
|
||||
|
||||
start = time.time()
|
||||
|
||||
try:
|
||||
# Each query is a fresh conversation - reset messages
|
||||
# This is intentional: we want stateless queries for the relay
|
||||
_CLI.agent.messages = []
|
||||
result = _CLI.agent.chat(full_prompt)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
logger.warning("Hermes query interrupted")
|
||||
raise # Re-raise to allow proper shutdown
|
||||
except SystemExit as e:
|
||||
# Hermes calls sys.exit() in some error paths
|
||||
logger.warning("Hermes called sys.exit() during query: %s", e)
|
||||
return "⚠️ Hermes error: process exited unexpectedly"
|
||||
except Exception as e:
|
||||
logger.error("Hermes error: %s", e, exc_info=True)
|
||||
# Return user-friendly message but log full traceback
|
||||
return f"⚠️ Hermes error: {type(e).__name__}"
|
||||
|
||||
elapsed = time.time() - start
|
||||
|
||||
# Fallback if Hermes returns None (error path)
|
||||
if result is None:
|
||||
logger.warning("Hermes returned None — possible error during query")
|
||||
return "⚠️ Geen antwoord van Hermes. Probeer opnieuw."
|
||||
|
||||
logger.info("Hermes responded in %.1fs (%d chars)", elapsed, len(result))
|
||||
return result
|
||||
|
||||
|
||||
def health_check() -> Dict[str, Any]:
|
||||
"""
|
||||
Check if Hermes agent is healthy and responsive.
|
||||
Returns a dict with status details.
|
||||
|
||||
This is a LIGHTWEIGHT check - no actual LLM query is made.
|
||||
For deep health checking, use the /health endpoint with detail=true.
|
||||
"""
|
||||
checks = {
|
||||
"hermes_loaded": _HERMES_LOADED,
|
||||
"agent_initialized": _CLI is not None and _CLI.agent is not None,
|
||||
"route_signature": str(_ACTIVE_ROUTE_SIGNATURE) if _ACTIVE_ROUTE_SIGNATURE else None,
|
||||
}
|
||||
|
||||
# Lightweight check - just verify agent object exists
|
||||
# Do NOT run an actual query here as it blocks for 10-60s
|
||||
if checks["agent_initialized"]:
|
||||
checks["query_test"] = "skipped (use deep check)"
|
||||
checks["query_latency_ms"] = None
|
||||
else:
|
||||
checks["query_test"] = "agent_not_initialized"
|
||||
checks["query_latency_ms"] = None
|
||||
|
||||
checks["healthy"] = (
|
||||
checks["hermes_loaded"]
|
||||
and checks["agent_initialized"]
|
||||
)
|
||||
|
||||
return checks
|
||||
|
||||
|
||||
def shutdown() -> None:
|
||||
"""Graceful shutdown handler."""
|
||||
global _SHUTTING_DOWN
|
||||
with _STATE_LOCK:
|
||||
_SHUTTING_DOWN = True
|
||||
logger.info("Shutdown initiated")
|
||||
|
||||
if _CLI and _CLI.agent:
|
||||
try:
|
||||
# Give agent a chance to clean up
|
||||
if hasattr(_CLI.agent, "shutdown"):
|
||||
_CLI.agent.shutdown()
|
||||
except Exception as e:
|
||||
logger.error("Error during agent shutdown: %s", e, exc_info=True)
|
||||
|
||||
logger.info("Shutdown complete")
|
||||
|
||||
|
||||
# Register shutdown handlers
|
||||
signal.signal(signal.SIGTERM, lambda s, f: shutdown())
|
||||
signal.signal(signal.SIGINT, lambda s, f: shutdown())
|
||||
atexit.register(shutdown)
|
||||
+19
-2
@@ -1,2 +1,19 @@
|
||||
flask>=3.0
|
||||
requests>=2.31
|
||||
# Core dependencies - pinned versions for reproducible builds
|
||||
flask==3.0.3
|
||||
requests==2.32.3
|
||||
|
||||
# Async/Production dependencies
|
||||
quart==0.19.4
|
||||
uvicorn[standard]==0.30.6
|
||||
gunicorn==22.0.0
|
||||
|
||||
# Performance optimizations
|
||||
httpx==0.27.2 # HTTP/2 connection pooling for Telegram
|
||||
diskcache==5.6.3 # Semantic caching
|
||||
brotli==1.1.0 # Compression
|
||||
|
||||
# Resilience patterns
|
||||
tenacity==8.5.0 # Retry logic with exponential backoff
|
||||
|
||||
# Type hints support
|
||||
typing_extensions==4.12.2
|
||||
Reference in New Issue
Block a user