430ab966b2
- config.py: pydantic-settings with all env vars, privacy mode, per-agent overrides - app_state.py: global singletons (pool, master agent, registry, llm_router, sweep) - main.py: FastAPI lifespan startup — DB pool, LLM router, Odoo client, agents, master - routers/dispatch.py: POST /dispatch with rate limiting and webhook secret auth - routers/approval.py: GET /approval/pending, POST /approval/respond - routers/registry.py: GET/POST /registry/agents, POST /registry/backend overrides - routers/sweep.py: POST /sweep trigger, GET /sweep/status - routers/health.py: GET /health, GET /health/detailed (DB/Odoo/Ollama checks) - requirements.txt: pinned deps (fastapi, uvicorn, asyncpg, anthropic, alembic) - Dockerfile: python:3.11-slim, single uvicorn worker - docker-compose.yml: agent-service + postgres:15, bound to 192.168.2.47:8001 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
from __future__ import annotations
|
|
import asyncio
|
|
import logging
|
|
import time
|
|
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
|
|
logger = logging.getLogger(__name__)
|
|
router = APIRouter(prefix='/health', tags=['health'])
|
|
|
|
_start_time = time.time()
|
|
|
|
|
|
class HealthResponse(BaseModel):
|
|
status: str
|
|
uptime_seconds: float
|
|
|
|
|
|
class DetailedHealthResponse(BaseModel):
|
|
status: str
|
|
uptime_seconds: float
|
|
db: str
|
|
odoo: str
|
|
ollama: str
|
|
master_agent: str
|
|
privacy_mode: str
|
|
|
|
|
|
@router.get('', response_model=HealthResponse)
|
|
async def health():
|
|
return HealthResponse(status='ok', uptime_seconds=round(time.time() - _start_time, 1))
|
|
|
|
|
|
@router.get('/detailed', response_model=DetailedHealthResponse)
|
|
async def health_detailed():
|
|
from ..app_state import get_db_pool, get_master_agent, get_llm_router
|
|
from ..config import get_settings
|
|
|
|
uptime = round(time.time() - _start_time, 1)
|
|
settings = get_settings()
|
|
|
|
# DB check
|
|
db_status = 'unavailable'
|
|
pool = get_db_pool()
|
|
if pool:
|
|
try:
|
|
async with pool.acquire(timeout=5) as conn:
|
|
await conn.fetchval('SELECT 1')
|
|
db_status = 'ok'
|
|
except Exception as exc:
|
|
db_status = f'error: {exc}'
|
|
|
|
# Odoo check
|
|
odoo_status = 'unavailable'
|
|
master = get_master_agent()
|
|
if master and hasattr(master, '_odoo'):
|
|
try:
|
|
await asyncio.wait_for(master._odoo.ping(), timeout=5)
|
|
odoo_status = 'ok'
|
|
except Exception as exc:
|
|
odoo_status = f'error: {exc}'
|
|
|
|
# Ollama check
|
|
ollama_status = 'unavailable'
|
|
llm_router = get_llm_router()
|
|
if llm_router and hasattr(llm_router, '_ollama'):
|
|
try:
|
|
await asyncio.wait_for(llm_router._ollama.ping(), timeout=5)
|
|
ollama_status = 'ok'
|
|
except Exception as exc:
|
|
ollama_status = f'error: {exc}'
|
|
|
|
master_status = 'ok' if master is not None else 'unavailable'
|
|
overall = 'ok' if all(s == 'ok' for s in [db_status, master_status]) else 'degraded'
|
|
|
|
return DetailedHealthResponse(
|
|
status=overall,
|
|
uptime_seconds=uptime,
|
|
db=db_status,
|
|
odoo=odoo_status,
|
|
ollama=ollama_status,
|
|
master_agent=master_status,
|
|
privacy_mode=settings.llm_privacy_mode,
|
|
)
|