aixtools 0.3.8__py3-none-any.whl → 0.3.10__py3-none-any.whl

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 = None

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/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/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 = {

{aixtools-0.3.8.dist-info → aixtools-0.3.10.dist-info}/METADATA

@@ -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.dist-info → aixtools-0.3.10.dist-info}/RECORD

@@ -1,5 +1,5 @@
 aixtools/__init__.py,sha256=9NGHm7LjsQmsvjTZvw6QFJexSvAU4bCoN_KBk9SCa00,260
-aixtools/_version.py,sha256=iWEYFk8Ve-UDYmpiH5KfwfwhQTYtkxZlaGPwKE2xDy0,704
+aixtools/_version.py,sha256=IwCXstmG50R88HCesfawu5BpJEPg8WIKehd-k6CRLJs,706
 aixtools/app.py,sha256=JzQ0nrv_bjDQokllIlGHOV0HEb-V8N6k_nGQH-TEsVU,5227
 aixtools/chainlit.md,sha256=yC37Ly57vjKyiIvK4oUvf4DYxZCwH7iocTlx7bLeGLU,761
 aixtools/context.py,sha256=I_MD40ZnvRm5WPKAKqBUAdXIf8YaurkYUUHSVVy-QvU,598
@@ -34,7 +34,7 @@ aixtools/agents/nodes_to_str.py,sha256=UkOu5Nry827J4H_ohQU3tPBfJxtr3p6FfCfWoUy5u
 aixtools/agents/print_nodes.py,sha256=wVTngNfqM0As845WTRz6G3Rei_Gr3HuBlvu-G_eXuig,1665
 aixtools/agents/prompt.py,sha256=p9OYnyJ4-MyGXwHPrQeJBhZ2a3RV2HqhtdUUCrTMsAQ,3361
 aixtools/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aixtools/auth/auth.py,sha256=AUc9Kw8ZrjiIxsp7Vg1bexg8mIuMbRv8XNU3pynfMdU,8487
+aixtools/auth/auth.py,sha256=w8RODphhpBYN3aPQ6x-btILDZ6InQOX1AK5sbDBEtRs,8736
 aixtools/compliance/__init__.py,sha256=vnw0zEdySIJWvDAJ8DCRRaWmY_agEOz1qlpAdhmtiuo,191
 aixtools/compliance/private_data.py,sha256=OOM9mIp3_w0fNgj3VAEWBl7-jrPc19_Ls1pC5dfF5UY,5323
 aixtools/db/__init__.py,sha256=b8vRhme3egV-aUZbAntnOaDkSXB8UT0Xy5oqQhU_z0Q,399
@@ -54,7 +54,7 @@ aixtools/log_view/filters.py,sha256=xbgSzzUnkqHoS_A_p4lRDMFhXqFvXCjwkcIEBQ_L29g,
 aixtools/log_view/log_utils.py,sha256=PGUUB039x6w7nOLg_M6Jk1aDIvcjyKP4f3YF5kavQ88,914
 aixtools/log_view/node_summary.py,sha256=EJjnBqdBWI-_bI-4nfTxwaost3mtiufb5cK7T54cfuQ,9299
 aixtools/logfilters/__init__.py,sha256=pTD8ujCqjPWBCeB7yv7lmCtnA2KXOnkIv0HExDagkXs,129
-aixtools/logfilters/context_filter.py,sha256=zR3Bnv3fCqXLeb7bCFTmlnWhC6dFIvUb-u712tOnUPk,2259
+aixtools/logfilters/context_filter.py,sha256=7Tfus83Ja2CmhIyd4xEAnGNAdAVSC1shb7_cRfilZQc,2415
 aixtools/logging/__init__.py,sha256=b5oYyGQDUHHxhRtzqKUaQPv8hQeWw54rzDXSV8lDY1w,613
 aixtools/logging/log_objects.py,sha256=gohsgcfyr8vsY7G_hfmj973-Ek1_PN-bMMLEUA-4u6U,7708
 aixtools/logging/logging_config.py,sha256=LvxV3C75-I0096PpcCIbgM-Cp998LzWXeMM14HYbU20,4985
@@ -70,7 +70,7 @@ aixtools/mcp/exceptions.py,sha256=9m3hy9fRINbXTSgkE71XW_luklgH6xBRp7VnSRfpyzQ,17
 aixtools/mcp/fast_mcp_log.py,sha256=XYOS406dVjn5YTHyGRsRvVNQ0SKlRObfrKj6EeLFjHg,1057
 aixtools/mcp/faulty_mcp.py,sha256=uU9vlNGCS_i2k20wocVMaDHTlYjMQxuzjILad9O1cjA,12807
 aixtools/mcp/middleware.py,sha256=V-bWhSL6F-tv74EtqaoYjbvxIUCIfG3XKz1khU7ebU4,2268
-aixtools/mcp/server.py,sha256=wpN7tJmUZqGsb_rOVfXbyTFJ9VPnQl7nG5wa1JAz-No,2566
+aixtools/mcp/server.py,sha256=vebY9xTVMJwM1MSHY2evBrtOA9OU78G-xaebRjfI9kI,2711
 aixtools/model_patch/model_patch.py,sha256=JT-oHubIn2LeoSwWbzEQ5vLH7crJmFUecHyQfaAFHa0,1813
 aixtools/server/__init__.py,sha256=rwPx020YpOzCnrxA80Lc4yLLcIp-Mpe9hNqVO9wDPv0,448
 aixtools/server/app_mounter.py,sha256=0tJ0tC140ezAjnYdlhpLJQjY-TO8NVw7D8LseYCCVY8,3336
@@ -96,8 +96,8 @@ aixtools/utils/chainlit/cl_agent_show.py,sha256=vaRuowp4BRvhxEr5hw0zHEJ7iaSF_5bo
 aixtools/utils/chainlit/cl_utils.py,sha256=fxaxdkcZg6uHdM8uztxdPowg3a2f7VR7B26VPY4t-3c,5738
 aixtools/vault/__init__.py,sha256=fsr_NuX3GZ9WZ7dGfe0gp_5-z3URxAfwVRXw7Xyc0dU,141
 aixtools/vault/vault.py,sha256=9dZLWdZQk9qN_Q9Djkofw9LUKnJqnrX5H0fGusVLBhA,6037
-aixtools-0.3.8.dist-info/METADATA,sha256=0pkdCTHZp4FLAwhMxcTKvXA9yQuxhkdOCwzqUO31B6E,27957
-aixtools-0.3.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aixtools-0.3.8.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
-aixtools-0.3.8.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
-aixtools-0.3.8.dist-info/RECORD,,
+aixtools-0.3.10.dist-info/METADATA,sha256=AhCPj7Q3xRPJCpAAAfoB7grM12n4eawP4WbtJfcMY9Q,27958
+aixtools-0.3.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aixtools-0.3.10.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
+aixtools-0.3.10.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
+aixtools-0.3.10.dist-info/RECORD,,