aixtools 0.3.8__tar.gz → 0.3.10__tar.gz

{aixtools-0.3.8 → aixtools-0.3.10}/PKG-INFO

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

{aixtools-0.3.8 → aixtools-0.3.10}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.8'
-__version_tuple__ = version_tuple = (0, 3, 8)
+__version__ = version = '0.3.10'
+__version_tuple__ = version_tuple = (0, 3, 10)
 
-__commit_id__ = commit_id = 'g94dc56696'
+__commit_id__ = commit_id = 'g5b197998a'

{aixtools-0.3.8 → aixtools-0.3.10}/aixtools/auth/auth.py

@@ -125,6 +125,10 @@ class AccessTokenVerifier:
             raise AuthTokenError(AuthTokenErrorCode.INVALID_ISSUER) from e
         except InvalidSignatureError as e:
             raise AuthTokenError(AuthTokenErrorCode.INVALID_SIGNATURE) from e
+        except jwt.exceptions.DecodeError as e:
+            raise AuthTokenError(AuthTokenErrorCode.INVALID_TOKEN) from e
+        except jwt.exceptions.PyJWKClientError as e:
+            raise AuthTokenError(AuthTokenErrorCode.INVALID_TOKEN) from e
         except jwt.exceptions.PyJWTError as e:
             logger.exception("Unable to check JWT token: %s", token)
             raise AuthTokenError(AuthTokenErrorCode.JWT_ERROR) from e

{aixtools-0.3.8 → aixtools-0.3.10}/aixtools/logfilters/context_filter.py

@@ -53,8 +53,13 @@ class ContextFilter(logging.Filter):  # pylint: disable=too-few-public-methods
         except ImportError:
             pass
 
-        if not user_id and not session_id:
-            user_id, session_id = self._extract_from_mcp_context()
+        mcp_user_id = None
+        mcp_session_id = None
+        if not user_id or not session_id:
+            mcp_user_id, mcp_session_id = self._extract_from_mcp_context()
+
+        user_id = user_id or mcp_user_id
+        session_id = session_id or mcp_session_id
 
         context = ""
         if session_id and not str(session_id).startswith("default"):

{aixtools-0.3.8 → aixtools-0.3.10}/aixtools/mcp/server.py

@@ -10,6 +10,7 @@ from fastmcp.utilities.types import NotSet
 from aixtools.auth.auth import AccessTokenAuthProvider
 from aixtools.logging.logging_config import get_logger
 from aixtools.mcp.middleware import AixErrorHandlingMiddleware
+from aixtools.utils import config
 
 
 def create_mcp_server(
@@ -58,7 +59,10 @@ def create_mcp_server(
             TimingMiddleware(logger=get_logger("middleware.timing")),
         ]
 
-    if auth is NotSet:
+    # env SKIP_MCP_AUTHORIZATION overrides config.
+    if config.SKIP_MCP_AUTHORIZATION:
+        auth = None
+    elif auth is NotSet:
         auth = AccessTokenAuthProvider()
 
     mcp_args = {