Merge pull request #22 from tisnik/updated-endpoints

tisnik · web-flow · commit fe9fa01f19d1 · 2025-05-12T09:58:21.000+02:00
Updated REST API endpoints
diff --git a/src/app/endpoints/__init__.py 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
@@ -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
@@ -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
@@ -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
@@ -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)
