aixtools 0.3.0__tar.gz → 0.3.2__tar.gz

{aixtools-0.3.0 → aixtools-0.3.2}/PKG-INFO

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

{aixtools-0.3.0 → aixtools-0.3.2}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.0'
-__version_tuple__ = version_tuple = (0, 3, 0)
+__version__ = version = '0.3.2'
+__version_tuple__ = version_tuple = (0, 3, 2)
 
-__commit_id__ = commit_id = 'g41920ea34'
+__commit_id__ = commit_id = 'g0ceaec30c'

{aixtools-0.3.0 → aixtools-0.3.2}/aixtools/agents/agent.py

@@ -129,6 +129,7 @@ def get_agent(  # noqa: PLR0913, pylint: disable=too-many-arguments,too-many-pos
     output_type: Any = str,
     deps_type=NoneType,
     http_client=None,
+    retries: int = 1,
 ) -> Agent:
     """Get a PydanticAI agent"""
     if model_settings is None:
@@ -147,6 +148,7 @@ def get_agent(  # noqa: PLR0913, pylint: disable=too-many-arguments,too-many-pos
         tools=tools,
         toolsets=toolsets,
         instrument=True,
+        retries=retries,
     )
     return agent
 

{aixtools-0.3.0 → aixtools-0.3.2}/aixtools/auth/auth.py

@@ -30,6 +30,7 @@ class AuthTokenErrorCode(str, enum.Enum):
     JWT_ERROR = "Generic JWT error"
     MISSING_GROUPS_ERROR = "Missing authorized groups"
     INVALID_TOKEN_SCOPE = "Token scope does not match configured scope"
+    TOKEN_MISSING = "Token missing"
 
 
 class AuthTokenError(Exception):
@@ -185,3 +186,18 @@ class AccessTokenAuthProvider(AuthProvider):
         return AccessToken(
             token=token, client_id=self.token_verifier.api_id, scopes=scopes_arr, expires_at=claims.get("exp", None)
         )
+
+    async def verify_auth_header(self, auth_header: str | None) -> AccessToken:
+        """
+        Splits the authorization header and looks for bearer type token,
+        then verifies the extracted bearer token and returns an AccessToken object.
+        """
+
+        if auth_header and auth_header.lower().startswith("bearer "):
+            auth_token = auth_header.split(" ", 1)[1].strip()
+            return await self.verify_token(auth_token)
+
+        raise AuthTokenError(
+            AuthTokenErrorCode.TOKEN_MISSING,
+            f"Could not find bearer token in authorization header: {auth_header}",
+        )