aixtools 0.3.9__tar.gz → 0.3.11__tar.gz

{aixtools-0.3.9 → aixtools-0.3.11}/PKG-INFO

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

{aixtools-0.3.9 → aixtools-0.3.11}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.9'
-__version_tuple__ = version_tuple = (0, 3, 9)
+__version__ = version = '0.3.11'
+__version_tuple__ = version_tuple = (0, 3, 11)
 
-__commit_id__ = commit_id = 'gf2f460fba'
+__commit_id__ = commit_id = 'g0077788dc'

{aixtools-0.3.9 → aixtools-0.3.11}/aixtools/a2a/app.py

@@ -32,6 +32,7 @@ from starlette.responses import RedirectResponse
 from aixtools.a2a.auth_middleware import AuthMiddleware
 from aixtools.auth.auth import AccessTokenAuthProvider
 from aixtools.context import session_id_var, user_id_var
+from aixtools.utils import config
 
 
 class AgentWorkerWithMetadataParser(AgentWorker):
@@ -140,6 +141,11 @@ def build_a2a_starlette_app(
         http_handler=request_handler,
     )
 
+    if config.SKIP_MCP_AUTHORIZATION:
+        auth_provider = None
+    elif auth_provider is None:
+        auth_provider = AccessTokenAuthProvider()
+
     app = server.build()
     if auth_provider:
         app.add_middleware(AuthMiddleware, provider=auth_provider)

{aixtools-0.3.9 → aixtools-0.3.11}/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.9 → aixtools-0.3.11}/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 = {