aixtools 0.2.10__tar.gz → 0.2.12__tar.gz

{aixtools-0.2.10 → aixtools-0.2.12}/PKG-INFO

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

{aixtools-0.2.10 → aixtools-0.2.12}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.10'
-__version_tuple__ = version_tuple = (0, 2, 10)
+__version__ = version = '0.2.12'
+__version_tuple__ = version_tuple = (0, 2, 12)
 
-__commit_id__ = commit_id = 'ge6b757983'
+__commit_id__ = commit_id = 'g57d2aa39c'

{aixtools-0.2.10 → aixtools-0.2.12}/aixtools/auth/auth.py

@@ -16,6 +16,7 @@ from mcp.server.auth.provider import (
 from aixtools.utils import config
 
 logger = logging.getLogger(__name__)
+TEST_CLIENT = "test-client"
 
 
 class AuthTokenErrorCode(str, enum.Enum):
@@ -159,14 +160,22 @@ class AccessTokenAuthProvider(AuthProvider):
     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")
+        test_token = config.AUTH_TEST_TOKEN
+
+        # check if the token is a test token
+        # this is used for integration test run
+        if test_token and token == test_token:
+            logger.info("Using test token:%s", test_token)
+            return AccessToken(token=token, client_id=TEST_CLIENT, scopes=[], expires_at=None)
+
         claims = self.token_verifier.verify(token)
         scopes = claims.get("scp", "")
-        self.token_verifier.authorize_claims(claims, scopes)
+        self.token_verifier.authorize_claims(claims, self.app_scope)
 
         scopes_arr = []
         if scopes:

{aixtools-0.2.10 → aixtools-0.2.12}/aixtools/utils/config.py

@@ -124,7 +124,7 @@ GOOGLE_CLOUD_LOCATION = get_variable_env("GOOGLE_CLOUD_LOCATION", True)
 # vault parameters.
 VAULT_ADDRESS = get_variable_env("VAULT_ADDRESS", default="http://localhost:8200")
 VAULT_TOKEN = get_variable_env("VAULT_TOKEN", allow_empty=True)
-VAULT_ENV = get_variable_env("ENV", allow_empty=True)
+VAULT_ENV = get_variable_env("VAULT_ENV", allow_empty=True)
 VAULT_MOUNT_POINT = get_variable_env("VAULT_MOUNT_POINT", allow_empty=True)
 VAULT_PATH_PREFIX = get_variable_env("VAULT_PATH_PREFIX", allow_empty=True)
 
@@ -141,3 +141,6 @@ APP_AUTHORIZED_GROUPS = get_variable_env("APP_AUTHORIZED_GROUPS", allow_empty=Tr
 
 # 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)
+
+AUTH_TEST_TOKEN = get_variable_env("AUTH_TEST_TOKEN", allow_empty=True)

{aixtools-0.2.10 → aixtools-0.2.12}/pyproject.toml

@@ -71,3 +71,6 @@ ignore = ["_version.py"]
 [tool.setuptools_scm]
 write_to = "aixtools/_version.py"
 version_scheme = "release-branch-semver"
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"