|
|
|
from fastapi import FastAPI |
|
from routes import embeddings, search, admin |
|
from fastapi.middleware import Middleware |
|
from fastapi.middleware.cors import CORSMiddleware |
|
from datetime import datetime |
|
from uvicorn.logging import DefaultFormatter |
|
import logging |
|
import uvicorn |
|
from exception import generic_exception_handler |
|
|
|
logger = logging.getLogger(__name__) |
|
handler = logging.StreamHandler() |
|
handler.setFormatter(DefaultFormatter()) |
|
logger.addHandler(handler) |
|
|
|
|
|
app = FastAPI() |
|
app.include_router(embeddings.router) |
|
app.include_router(search.router) |
|
app.include_router(admin.router) |
|
app.exception_handler(generic_exception_handler) |
|
|
|
@app.middleware("http") |
|
async def log_requests(request, call_next): |
|
start_time = datetime.utcnow() |
|
response = await call_next(request) |
|
process_time = (datetime.utcnow() - start_time).total_seconds() * 1000 |
|
print( |
|
f"{request.method} {request.url.path} {response.status_code} {process_time:.2f}ms" |
|
) |
|
return response |
|
|
|
|
|
if __name__ == "__main__": |
|
uvicorn.run(app, host="0.0.0.0", port=8080) |
