feat: add health check route

This commit is contained in:
Shubham Naik
2024-10-02 16:13:23 -07:00
parent 01cb79c2d7
commit 188c3dd942
3 changed files with 31 additions and 0 deletions

8
letta/schemas/health.py Normal file
View File

@@ -0,0 +1,8 @@
from pydantic import BaseModel
class Health(BaseModel):
"""
Health check response body
"""
version: str
status: str

View File

@@ -4,6 +4,7 @@ from letta.server.rest_api.routers.v1.jobs import router as jobs_router
from letta.server.rest_api.routers.v1.llms import router as llm_router
from letta.server.rest_api.routers.v1.sources import router as sources_router
from letta.server.rest_api.routers.v1.tools import router as tools_router
from letta.server.rest_api.routers.v1.health import router as health_router
ROUTERS = [
tools_router,
@@ -12,4 +13,5 @@ ROUTERS = [
llm_router,
blocks_router,
jobs_router,
health_router,
]

View File

@@ -0,0 +1,21 @@
from typing import TYPE_CHECKING, List, Optional
from fastapi import APIRouter, Body, Depends, HTTPException, Query
from letta.cli.cli import version
from letta.schemas.health import Health
from letta.server.rest_api.utils import get_letta_server
from letta.server.server import SyncServer
if TYPE_CHECKING:
pass
router = APIRouter(prefix="/health", tags=["health"])
# Health check
@router.get("/", response_model=Health, operation_id="health_check")
def health_check():
return Health(
version=version(),
status="ok",
)