feat: add health check route
This commit is contained in:
8
letta/schemas/health.py
Normal file
8
letta/schemas/health.py
Normal file
@@ -0,0 +1,8 @@
|
||||
from pydantic import BaseModel
|
||||
|
||||
class Health(BaseModel):
|
||||
"""
|
||||
Health check response body
|
||||
"""
|
||||
version: str
|
||||
status: str
|
||||
@@ -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,
|
||||
]
|
||||
|
||||
21
letta/server/rest_api/routers/v1/health.py
Normal file
21
letta/server/rest_api/routers/v1/health.py
Normal 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",
|
||||
)
|
||||
Reference in New Issue
Block a user