Merge branch 'main' of gitlab.prodcontest.com:team-39/backend

.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)"

pyproject.toml

@@ -29,6 +29,7 @@ backend = [
 ]
 ml = [
     "sentence-transformers>=5.1.2",
+    "prometheus-fastapi-instrumentator>=7.1.0",
     "torch",
 ]
 types = [

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)