aixtools 0.2.9__py3-none-any.whl → 0.2.11__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.2.9'
-__version_tuple__ = version_tuple = (0, 2, 9)
+__version__ = version = '0.2.11'
+__version_tuple__ = version_tuple = (0, 2, 11)
 
 __commit_id__ = commit_id = None

aixtools/auth/auth.py

@@ -7,7 +7,11 @@ import logging
 
 import jwt
 from fastapi import HTTPException
+from fastmcp.server.auth.auth import AuthProvider
 from jwt import ExpiredSignatureError, InvalidAudienceError, InvalidIssuerError, InvalidSignatureError, PyJWKClient
+from mcp.server.auth.provider import (
+    AccessToken,
+)
 
 from aixtools.utils import config
 
@@ -147,3 +151,29 @@ class AccessTokenVerifier:
             AuthTokenErrorCode.MISSING_GROUPS_ERROR,
             f"Could not find any group in JWT token, matching: {self.authorized_groups}",
         )
+
+
+class AccessTokenAuthProvider(AuthProvider):
+    """Authentication provider for MCP servers for validating, authorizing and extracting access tokens."""
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.token_verifier = AccessTokenVerifier()
+        self.app_scope = config.APP_DEFAULT_SCOPE
+
+    async def verify_token(self, token: str) -> AccessToken:
+        """Verify the access token and return an AccessToken object."""
+
+        logger.info("Received verify token request")
+        claims = self.token_verifier.verify(token)
+        scopes = claims.get("scp", "")
+        self.token_verifier.authorize_claims(claims, self.app_scope)
+
+        scopes_arr = []
+        if scopes:
+            scopes_arr = scopes.split(" ")
+
+        logger.info("Authorized the token")
+        return AccessToken(
+            token=token, client_id=self.token_verifier.api_id, scopes=scopes_arr, expires_at=claims.get("exp", None)
+        )

aixtools/utils/config.py

@@ -138,3 +138,7 @@ APP_TENANT_ID = get_variable_env("APP_TENANT_ID")
 
 # used for token authorization check
 APP_AUTHORIZED_GROUPS = get_variable_env("APP_AUTHORIZED_GROUPS", allow_empty=True)
+
+# used to skip authorization in local tests if required.
+SKIP_MCP_AUTHORIZATION = str2bool(get_variable_env("SKIP_MCP_AUTHORIZATION", True, False))
+APP_DEFAULT_SCOPE = get_variable_env("APP_DEFAULT_SCOPE", allow_empty=True)

{aixtools-0.2.9.dist-info → aixtools-0.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aixtools
-Version: 0.2.9
+Version: 0.2.11
 Summary: Tools for AI exploration and debugging
 Requires-Python: >=3.11.2
 Description-Content-Type: text/markdown
@@ -9,7 +9,7 @@ Requires-Dist: cachebox>=5.0.1
 Requires-Dist: chainlit>=2.5.5
 Requires-Dist: colorlog>=6.9.0
 Requires-Dist: fasta2a>=0.5.0
-Requires-Dist: fastmcp>=2.10.2
+Requires-Dist: fastmcp>=2.11.0
 Requires-Dist: hvac>=2.3.0
 Requires-Dist: ipykernel>=6.29.5
 Requires-Dist: langchain-chroma>=0.2.3

{aixtools-0.2.9.dist-info → aixtools-0.2.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 aixtools/__init__.py,sha256=9NGHm7LjsQmsvjTZvw6QFJexSvAU4bCoN_KBk9SCa00,260
-aixtools/_version.py,sha256=051on7ZmwGNyKvbO1AXKoElw7RjLuRmeJqVOApytNd4,704
+aixtools/_version.py,sha256=ctqLY7UIopjEtEtkMewUbTAH-exVquX9ojG1BE5gJ1s,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
@@ -30,7 +30,7 @@ aixtools/agents/agent_batch.py,sha256=0Zu9yNCRPAQZPjXQ-dIUAmP1uGTVbxVt7xvnMpoJMj
 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=4vNcljdpFzvIBXqjK7i3xAy-CC8Hc7ii11t-11z6ofY,6041
+aixtools/auth/auth.py,sha256=aCcFTHjTO-ieA5Xo7913-9OCDowRAoCYG_b_gmtsXsU,7100
 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
@@ -78,7 +78,7 @@ aixtools/tools/doctor/mcp_tool_doctor.py,sha256=sX2q5GfNkmUYxnXrqMpeGIwGfeL1LpYJ
 aixtools/tools/doctor/tool_doctor.py,sha256=EY1pshjLGLD0j6cc1ZFtbc0G19I5IbOZwHFDqypE49Q,2661
 aixtools/tools/doctor/tool_recommendation.py,sha256=LYyVOSXdAorWiY4P-ucSA1vLlV5BTEfX4GzBXNE_X0M,1569
 aixtools/utils/__init__.py,sha256=xT6almZBQYMfj4h7Hq9QXDHyVXbOOTxqLsmJsxYYnSw,757
-aixtools/utils/config.py,sha256=fCATakEfeASo6eFMh5Me_idwMQgiITS3LMYc2Z1bGzU,5187
+aixtools/utils/config.py,sha256=fd7BWDqxzMnHke2M4tjKvkhrxwdkUsuZXDTOShiID24,5412
 aixtools/utils/config_util.py,sha256=3Ya4Qqhj1RJ1qtTTykQ6iayf5uxlpigPXgEJlTi1wn4,2229
 aixtools/utils/enum_with_description.py,sha256=zjSzWxG74eR4x7dpmb74pLTYCWNSMvauHd7_9LpDYIw,1088
 aixtools/utils/files.py,sha256=8JnxwHJRJcjWCdFpjzWmo0po2fRg8esj4H7sOxElYXU,517
@@ -88,8 +88,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.2.9.dist-info/METADATA,sha256=jWcF3SX9eos7TLkvD4UaurdbD4x9JEcQApycD2u_kB8,27229
-aixtools-0.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aixtools-0.2.9.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
-aixtools-0.2.9.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
-aixtools-0.2.9.dist-info/RECORD,,
+aixtools-0.2.11.dist-info/METADATA,sha256=E5GS2Ov0M2EdWvwzzxHduZJvSMepJo_0P5j_uEn95-g,27230
+aixtools-0.2.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aixtools-0.2.11.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
+aixtools-0.2.11.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
+aixtools-0.2.11.dist-info/RECORD,,