diff --git a/src/app/endpoints/__init__.py b/src/app/endpoints/__init__.py new file mode 100644 index 000000000..04282dfb0 --- /dev/null +++ b/src/app/endpoints/__init__.py @@ -0,0 +1 @@ +"""Implementation of all endpoints.""" diff --git a/src/app/endpoints/health.py b/src/app/endpoints/health.py index 79be0b1e5..cea167c7d 100644 --- a/src/app/endpoints/health.py +++ b/src/app/endpoints/health.py @@ -13,8 +13,8 @@ from models.responses import ReadinessResponse, LivenessResponse, NotAvailableResponse -router = APIRouter(tags=["health"]) logger = logging.getLogger(__name__) +router = APIRouter(tags=["health"]) get_readiness_responses: dict[int | str, dict[str, Any]] = { diff --git a/src/app/endpoints/models.py b/src/app/endpoints/models.py index 546f7ad68..f2bc390fd 100644 --- a/src/app/endpoints/models.py +++ b/src/app/endpoints/models.py @@ -1,11 +1,10 @@ """Handler for REST API call to provide info.""" -import asyncio import logging -from typing import Any, Optional +from typing import Any from fastapi import APIRouter, Request -from llama_stack_client import Agent, AgentEventLogger, RAGDocument, LlamaStackClient +from llama_stack_client import LlamaStackClient from models.responses import ModelsResponse diff --git a/src/app/endpoints/query.py b/src/app/endpoints/query.py index a2b4e3577..f3bde3c8a 100644 --- a/src/app/endpoints/query.py +++ b/src/app/endpoints/query.py @@ -5,7 +5,6 @@ from fastapi import APIRouter, Request -from version import __version__ from models.responses import QueryResponse logger = logging.getLogger(__name__) diff --git a/src/app/endpoints/root.py b/src/app/endpoints/root.py index f3bf6c8d8..8c1b29b96 100644 --- a/src/app/endpoints/root.py +++ b/src/app/endpoints/root.py @@ -8,7 +8,21 @@ logger = logging.getLogger(__name__) router = APIRouter(tags=["root"]) +index_page = """ + +
+