from fastapi import APIRouter
from services.ollama_client import ollama_client
from db.vector_store import vector_store

router = APIRouter(tags=["health"])


@router.get("/health")
async def health_check():
    ollama_status = await ollama_client.check_health()
    return {
        "status": "ok",
        "service": "tk-ai-service",
        "ollama": ollama_status,
        "embeddings": vector_store.stats(),
    }


@router.get("/models")
async def list_models():
    return await ollama_client.check_health()