diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index 5043c87..a80132a 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -266,6 +266,7 @@ test: COMPOSE_PROFILES: | --profile migrations --profile tests + --profile ml script: - apk add --no-cache docker-compose - export PROFILES="$(printf '%s ' $COMPOSE_PROFILES)" diff --git a/pyproject.toml b/pyproject.toml index eb324ca..4c0f03d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,6 +29,7 @@ backend = [ ] ml = [ "sentence-transformers>=5.1.2", + "prometheus-fastapi-instrumentator>=7.1.0", "torch", ] types = [ diff --git a/src/template_project/ml/entry_point.py b/src/template_project/ml/entry_point.py index 70f3d6c..321568c 100644 --- a/src/template_project/ml/entry_point.py +++ b/src/template_project/ml/entry_point.py @@ -13,6 +13,7 @@ from dishka import AsyncContainer from dishka.integrations.fastapi import setup_dishka from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware +from prometheus_fastapi_instrumentator import Instrumentator from sentence_transformers import SentenceTransformer from template_project.ml.configuration import load_configuration @@ -68,6 +69,7 @@ def make_asgi_application( app.include_router(healthcheck.router) app.include_router(embed.router) app.include_router(predict.router) + Instrumentator().instrument(app).expose(app) setup_dishka(container=ioc, app=app)