From 67bcacc551e60d51c6f2c00b94ac04ba84cf1a89 Mon Sep 17 00:00:00 2001 From: yasinfakhar Date: Mon, 17 Mar 2025 15:44:24 +0330 Subject: [PATCH] fix[api]: removed deleted imports --- api/src/query_usage/routers/query_usage_router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/src/query_usage/routers/query_usage_router.py b/api/src/query_usage/routers/query_usage_router.py index 9fbdd80..4385f16 100644 --- a/api/src/query_usage/routers/query_usage_router.py +++ b/api/src/query_usage/routers/query_usage_router.py @@ -7,7 +7,7 @@ from src.db.sql_alchemy import Database from src.auth.utils.get_token import authenticate_user -from src.util.execptions import NotFoundError, NotEnoughBalanceError +from src.util.execptions import NotFoundError, RateLimitError from src.query_usage.services.query_usage_service import QueryUsageService from src.util.response import global_response, GlobalResponse, ExceptionResponse @@ -148,7 +148,7 @@ def get_user_query_usages( "max_query_allowance_per_day": service.get_user_max_query_allowance(db, user_id) } return global_response(data) - except NotEnoughBalanceError as e: + except RateLimitError as e: raise HTTPException( status_code=status.HTTP_429_TOO_MANY_REQUESTS, detail=str(e)) except Exception as e: