Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/app/endpoints/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
"""Implementation of all endpoints."""
2 changes: 1 addition & 1 deletion src/app/endpoints/health.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]] = {
Expand Down
5 changes: 2 additions & 3 deletions src/app/endpoints/models.py
Original file line number Diff line number Diff line change
@@ -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

Expand Down
1 change: 0 additions & 1 deletion src/app/endpoints/query.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@

from fastapi import APIRouter, Request

from version import __version__
from models.responses import QueryResponse

logger = logging.getLogger(__name__)
Expand Down
16 changes: 15 additions & 1 deletion src/app/endpoints/root.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,21 @@
logger = logging.getLogger(__name__)
router = APIRouter(tags=["root"])

index_page = """
<html>
<head>
<title>Lightspeed core service</title>
</head>
<body style='font-family: sans-serif;text-align:center;'>
<h1>Lightspeed core service</h1>
<img src="https://avatars.githubusercontent.com/u/204013222?s=400&u=47337cca0a4abbca5cfcc45fc20c7a2e82ac35e1&v=4" />
<div><a href="docs">Swagger UI</a></div>
<div><a href="redoc">ReDoc</a></div>
</body>
</html>
"""


@router.get("/", response_class=HTMLResponse)
def root_endpoint_handler(request: Request) -> HTMLResponse:
return HTMLResponse("<html>foo</html>")
return HTMLResponse(index_page)