aixtools 0.3.6__tar.gz → 0.3.7__tar.gz

{aixtools-0.3.6 → aixtools-0.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aixtools
-Version: 0.3.6
+Version: 0.3.7
 Summary: Tools for AI exploration and debugging
 Requires-Python: >=3.11.2
 Description-Content-Type: text/markdown

{aixtools-0.3.6 → aixtools-0.3.7}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.6'
-__version_tuple__ = version_tuple = (0, 3, 6)
+__version__ = version = '0.3.7'
+__version_tuple__ = version_tuple = (0, 3, 7)
 
-__commit_id__ = commit_id = 'g4f918962b'
+__commit_id__ = commit_id = 'g889028f57'

{aixtools-0.3.6 → aixtools-0.3.7}/aixtools/a2a/auth_middleware.py

@@ -1,14 +1,13 @@
 "Auth module managing user authentication for A2A server"
 
-from logging import getLogger
-
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.requests import Request
 
 from aixtools.auth.auth import AccessTokenAuthProvider, AuthTokenError
+from aixtools.logging.logging_config import get_logger
 from aixtools.utils import config
 
-logger = getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class AuthMiddleware(BaseHTTPMiddleware):  # pylint: disable=too-few-public-methods

{aixtools-0.3.6 → aixtools-0.3.7}/aixtools/auth/auth.py

@@ -3,7 +3,6 @@ Module that manages OAuth2 functions for authentication
 """
 
 import enum
-import logging
 
 import jwt
 from fastapi import HTTPException
@@ -13,9 +12,11 @@ from mcp.server.auth.provider import (
     AccessToken,
 )
 
+from aixtools.context import user_id_var
+from aixtools.logging.logging_config import get_logger
 from aixtools.utils import config
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 TEST_CLIENT = "test-client"
 
 
@@ -108,6 +109,11 @@ class AccessTokenVerifier:
                 # ensure audience verification is carried out
                 options={"verify_aud": True},
             )
+            # set user_id for logging context.
+            user_id = claims.get("email")
+            if user_id:
+                user_id_var.set(user_id)
+
             logger.info("Verified JWT token")
             return claims