Skip to content

Commit f74bf59

Browse files
committed
style: fix import ordering in token_usage_middleware
Signed-off-by: Mihai Criveti <crivetimihai@gmail.com>
1 parent 497e74c commit f74bf59

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

mcpgateway/middleware/token_usage_middleware.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,9 +188,12 @@ async def send_wrapper(message: dict) -> None:
188188
# Without this check, an attacker can craft a JWT with fake
189189
# jti/sub and auth_provider=api_token to pollute usage logs.
190190
try:
191-
from mcpgateway.db import EmailApiToken # pylint: disable=import-outside-toplevel
191+
# Third-Party
192192
from sqlalchemy import select # pylint: disable=import-outside-toplevel
193193

194+
# First-Party
195+
from mcpgateway.db import EmailApiToken # pylint: disable=import-outside-toplevel
196+
194197
with fresh_db_session() as verify_db:
195198
exists = verify_db.execute(select(EmailApiToken.id).where(EmailApiToken.jti == jti)).scalar_one_or_none()
196199
if exists is None:

0 commit comments

Comments
 (0)