chore: Update main.py imports and add root endpoint

main.py

@@ -1,11 +1,14 @@
-from typing import Optional, List
+import logging
 from contextlib import asynccontextmanager
+from typing import List, Optional
+
+import chromadb
+from cashews import cache
 from fastapi import FastAPI, HTTPException, Query
 from pydantic import BaseModel
-import chromadb
-import logging
+from starlette.responses import RedirectResponse
+
 from load_data import get_save_path, refresh_data
-from cashews import cache
 
 # Set up logging
 logging.basicConfig(
@@ -51,6 +54,11 @@ async def lifespan(app: FastAPI):
 app = FastAPI(lifespan=lifespan)
 
 
+@app.get("/", include_in_schema=False)
+def root():
+    return RedirectResponse(url="/docs")
+
+
 @app.get("/query", response_model=Optional[QueryResponse])
 @cache(ttl="1h")
 async def api_query_dataset(dataset_id: str, n: int = Query(default=10, ge=1, le=100)):