diff --git a/src/app/endpoints/health.py b/src/app/endpoints/health.py index 03d7789b5..ca09646d5 100644 --- a/src/app/endpoints/health.py +++ b/src/app/endpoints/health.py @@ -22,7 +22,7 @@ ProviderHealthStatus, ) -logger = logging.getLogger(__name__) +logger = logging.getLogger("app.endpoints.handlers") router = APIRouter(tags=["health"]) auth_dependency = get_auth_dependency() @@ -93,6 +93,8 @@ async def readiness_probe_get_method( # Used only for authorization _ = auth + logger.info("Response to /v1/readiness endpoint") + provider_statuses = await get_providers_health_statuses() # Check if any provider is unhealthy (not counting not_implemented as unhealthy) @@ -135,4 +137,6 @@ async def liveness_probe_get_method( # Used only for authorization _ = auth + logger.info("Response to /v1/liveness endpoint") + return LivenessResponse(alive=True) diff --git a/src/app/endpoints/info.py b/src/app/endpoints/info.py index 00608ffbf..18e7107a7 100644 --- a/src/app/endpoints/info.py +++ b/src/app/endpoints/info.py @@ -16,7 +16,7 @@ from models.responses import InfoResponse from version import __version__ -logger = logging.getLogger(__name__) +logger = logging.getLogger("app.endpoints.handlers") router = APIRouter(tags=["info"]) auth_dependency = get_auth_dependency() @@ -58,12 +58,17 @@ async def info_endpoint_handler( # Nothing interesting in the request _ = request + logger.info("Response to /v1/info endpoint") + try: # try to get Llama Stack client client = AsyncLlamaStackClientHolder().get_client() # retrieve version llama_stack_version_object = await client.inspect.version() llama_stack_version = llama_stack_version_object.version + logger.debug("Service name: %s", configuration.configuration.name) + logger.debug("Service version: %s", __version__) + logger.debug("LLama Stack version: %s", llama_stack_version) return InfoResponse( name=configuration.configuration.name, service_version=__version__,