claw-vector 0.1.2

The semantic memory engine for ClawDB — HNSW vector indexing and storage
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

from __future__ import annotations

from collections.abc import Callable

from fastapi import APIRouter
from fastapi.responses import JSONResponse

from claw_vector_svc.embedder import EmbedderService
from claw_vector_svc.models import HealthResponse, ReadyResponse

def make_health_router(
    get_embedder: Callable[[], EmbedderService | None],
    is_warmup_complete: Callable[[], bool],
) -> APIRouter:
    """Return a FastAPI router bound to the provided *embedder* instance."""
    router = APIRouter(tags=["health"])

    @router.get("/health", response_model=HealthResponse)
    async def health() -> HealthResponse:
        """Liveness probe: the process is up even before the model is ready."""
        embedder = get_embedder()
        ready = bool(embedder and embedder.is_ready and is_warmup_complete())
        return HealthResponse(
            ready=ready,
            model_name=embedder.model_name if embedder else "",
            model_load_time_ms=embedder.load_time_ms if embedder else 0,
        )

    @router.get("/ready", response_model=ReadyResponse)
    async def ready() -> JSONResponse:
        """Readiness probe that returns 503 until the model is loaded."""
        embedder = get_embedder()
        if embedder is not None and embedder.is_ready and is_warmup_complete():
            return JSONResponse(ReadyResponse(ready=True).model_dump(), status_code=200)
        return JSONResponse(ReadyResponse(ready=False).model_dump(), status_code=503)

    return router