From 860db588de3b3823df08eeab9df709a552185064 Mon Sep 17 00:00:00 2001
From: msramalho <19508417+msramalho@users.noreply.github.com>
Date: Mon, 21 Oct 2024 11:59:48 +0100
Subject: [PATCH] bug fix in metrics

---
 src/utils/metrics.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/utils/metrics.py b/src/utils/metrics.py
index f706154..e63297c 100644
--- a/src/utils/metrics.py
+++ b/src/utils/metrics.py
@@ -4,7 +4,6 @@ import os
 import shutil
 from loguru import logger
 from prometheus_client import Counter, Gauge
-from sqlalchemy.orm import Session
 
 from db import crud
 from db.database import get_db
@@ -54,7 +53,7 @@ async def measure_regular_metrics(sqlite_db_url:str, repeat_in_seconds:int):
         DISK_UTILIZATION.labels(type="database").set(fs.st_size / (2**30))
     except Exception as e: logger.error(e)
 
-    with get_db as db:
+    with get_db() as db:
         count_archives = crud.count_archives(db)
         count_archive_urls = crud.count_archive_urls(db)
         DATABASE_METRICS.labels(query="count_archives", user="-").set(count_archives)