a2a-lite 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

a2a_lite/__init__.py

@@ -102,7 +102,7 @@ from .auth import (
     require_auth,
 )
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"
 
 __all__ = [
     # Core

a2a_lite/agent.py

@@ -170,11 +170,14 @@ class Agent:
             import typing
             from .tasks import TaskContext as _TaskContext
             from .human_loop import InteractionContext as _InteractionContext
+            from .auth import AuthResult as _AuthResult
 
             needs_task_context = False
             needs_interaction = False
+            needs_auth = False
             task_context_param: str | None = None
             interaction_param: str | None = None
+            auth_param: str | None = None
 
             try:
                 resolved_hints = typing.get_type_hints(func)
@@ -190,6 +193,14 @@ class Agent:
                 elif _is_or_subclass(hint, _InteractionContext):
                     needs_interaction = True
                     interaction_param = param_name
+                elif _is_or_subclass(hint, _AuthResult):
+                    needs_auth = True
+                    auth_param = param_name
+
+            # Also detect require_auth decorator
+            if getattr(func, '__requires_auth__', False) and not needs_auth:
+                needs_auth = True
+                auth_param = auth_param or "auth"
 
             # Extract schemas
             input_schema, output_schema = extract_function_schemas(func)
@@ -205,8 +216,10 @@ class Agent:
                 is_streaming=is_streaming,
                 needs_task_context=needs_task_context,
                 needs_interaction=needs_interaction,
+                needs_auth=needs_auth,
                 task_context_param=task_context_param,
                 interaction_param=interaction_param,
+                auth_param=auth_param,
             )
 
             self._skills[skill_name] = skill_def
@@ -383,11 +396,14 @@ class Agent:
         ))
 
         # Run startup hooks
-        for hook in self._on_startup:
-            if asyncio.iscoroutinefunction(hook):
-                asyncio.get_event_loop().run_until_complete(hook())
-            else:
-                hook()
+        async def _run_startup():
+            for hook in self._on_startup:
+                if asyncio.iscoroutinefunction(hook):
+                    await hook()
+                else:
+                    hook()
+        if self._on_startup:
+            asyncio.run(_run_startup())
 
         # Enable discovery if requested
         if enable_discovery:
@@ -434,11 +450,14 @@ class Agent:
             )
         finally:
             # Run shutdown hooks
-            for hook in self._on_shutdown:
-                if asyncio.iscoroutinefunction(hook):
-                    asyncio.get_event_loop().run_until_complete(hook())
-                else:
-                    hook()
+            async def _run_shutdown():
+                for hook in self._on_shutdown:
+                    if asyncio.iscoroutinefunction(hook):
+                        await hook()
+                    else:
+                        hook()
+            if self._on_shutdown:
+                asyncio.run(_run_shutdown())
 
             # Unregister discovery
             if self._discovery:

a2a_lite/auth.py

@@ -222,7 +222,7 @@ class OAuth2Auth(AuthProvider):
             audience="my-agent",
         )
 
-    Requires: pip install pyjwt[crypto]
+    Requires: pip install a2a-lite[oauth]
     """
 
     def __init__(
@@ -277,7 +277,7 @@ class OAuth2Auth(AuthProvider):
 
         except ImportError:
             return AuthResult.failure(
-                "OAuth2 requires pyjwt: pip install pyjwt[crypto]"
+                "OAuth2 requires pyjwt: pip install a2a-lite[oauth]"
             )
         except Exception as e:
             return AuthResult.failure(f"Token validation failed: {str(e)}")

a2a_lite/cli.py

@@ -75,7 +75,7 @@ version = "0.1.0"
 description = "A2A Agent: {name}"
 requires-python = ">=3.10"
 dependencies = [
-    "a2a-lite>=0.2.1",
+    "a2a-lite>=0.2.2",
 ]
 '''
     (project_path / "pyproject.toml").write_text(pyproject)

a2a_lite/decorators.py

@@ -18,8 +18,10 @@ class SkillDefinition:
     is_streaming: bool = False
     needs_task_context: bool = False
     needs_interaction: bool = False
+    needs_auth: bool = False
     task_context_param: Optional[str] = None
     interaction_param: Optional[str] = None
+    auth_param: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for serialization."""

a2a_lite/executor.py

@@ -59,21 +59,22 @@ class LiteAgentExecutor(AgentExecutor):
         from a2a.utils import new_agent_text_message
 
         try:
-            # Authenticate the request
+            # Authenticate the request (always run to produce auth_result for injection)
+            auth_result = None
             if self.auth_provider:
                 from .auth import AuthRequest, NoAuth
-                if not isinstance(self.auth_provider, NoAuth):
-                    headers = {}
-                    if context.call_context and context.call_context.state:
-                        headers = context.call_context.state.get('headers', {})
-                    auth_request = AuthRequest(headers=headers)
-                    auth_result = await self.auth_provider.authenticate(auth_request)
-                    if not auth_result.authenticated:
-                        error_msg = json.dumps({
-                            "error": auth_result.error or "Authentication failed",
-                        })
-                        await event_queue.enqueue_event(new_agent_text_message(error_msg))
-                        return
+                headers = {}
+                if context.call_context and context.call_context.state:
+                    headers = context.call_context.state.get('headers', {})
+                auth_request = AuthRequest(headers=headers)
+                auth_result = await self.auth_provider.authenticate(auth_request)
+                # Reject unauthenticated requests (unless NoAuth)
+                if not isinstance(self.auth_provider, NoAuth) and not auth_result.authenticated:
+                    error_msg = json.dumps({
+                        "error": auth_result.error or "Authentication failed",
+                    })
+                    await event_queue.enqueue_event(new_agent_text_message(error_msg))
+                    return
 
             # Extract message and parts
             message, parts = self._extract_message_and_parts(context)
@@ -88,9 +89,10 @@ class LiteAgentExecutor(AgentExecutor):
                 message=message,
             )
 
-            # Store parts in metadata for skill access
+            # Store parts and auth result in metadata for skill access
             ctx.metadata["parts"] = parts
             ctx.metadata["event_queue"] = event_queue
+            ctx.metadata["auth_result"] = auth_result
 
             # Define final handler
             async def final_handler(ctx: MiddlewareContext) -> Any:
@@ -173,6 +175,10 @@ class LiteAgentExecutor(AgentExecutor):
             param_name = skill_def.interaction_param or "ctx"
             params[param_name] = interaction_ctx
 
+        if skill_def.needs_auth:
+            param_name = skill_def.auth_param or "auth"
+            params[param_name] = metadata.get("auth_result")
+
         # Call the handler
         handler = skill_def.handler
 
@@ -212,7 +218,8 @@ class LiteAgentExecutor(AgentExecutor):
             # Skip special context types
             from .tasks import TaskContext as _TaskContext
             from .human_loop import InteractionContext as _InteractionContext
-            if _is_or_subclass(param_type, _TaskContext) or _is_or_subclass(param_type, _InteractionContext):
+            from .auth import AuthResult as _AuthResult
+            if _is_or_subclass(param_type, _TaskContext) or _is_or_subclass(param_type, _InteractionContext) or _is_or_subclass(param_type, _AuthResult):
                 continue
 
             # Convert FilePart
@@ -345,7 +352,7 @@ class LiteAgentExecutor(AgentExecutor):
         if asyncio.iscoroutinefunction(handler):
             return await handler(*args, **kwargs)
         else:
-            loop = asyncio.get_event_loop()
+            loop = asyncio.get_running_loop()
             return await loop.run_in_executor(
                 None,
                 lambda: handler(*args, **kwargs)

a2a_lite/testing.py

@@ -214,7 +214,16 @@ class AgentTestClient:
                 result = await gen
                 results.append(result)
 
-        asyncio.run(run_handler())
+        # Handle both sync and async calling contexts
+        try:
+            asyncio.get_running_loop()
+            # Already in an async context — run in a separate thread
+            import concurrent.futures
+            with concurrent.futures.ThreadPoolExecutor(1) as pool:
+                pool.submit(asyncio.run, run_handler()).result()
+        except RuntimeError:
+            # No running loop — safe to use asyncio.run()
+            asyncio.run(run_handler())
         return results
 
 

a2a_lite/utils.py

@@ -1,6 +1,7 @@
 """
 Helper functions for A2A Lite.
 """
+import typing
 from typing import Any, Dict, Type, get_origin, get_args, Union
 import inspect
 
@@ -103,7 +104,10 @@ def extract_function_schemas(func) -> tuple[Dict[str, Any], Dict[str, Any]]:
         Tuple of (input_schema, output_schema)
     """
     sig = inspect.signature(func)
-    hints = getattr(func, '__annotations__', {})
+    try:
+        hints = typing.get_type_hints(func)
+    except Exception:
+        hints = getattr(func, '__annotations__', {})
 
     # Build input schema from parameters
     properties = {}

a2a_lite/webhooks.py

@@ -9,7 +9,7 @@ Simplifies sending notifications when tasks complete:
 """
 from __future__ import annotations
 
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Any, Callable, Dict, List, Optional
 import asyncio
 import json
@@ -19,15 +19,11 @@ import json
 class WebhookConfig:
     """Configuration for a webhook endpoint."""
     url: str
-    headers: Dict[str, str] = None
+    headers: Dict[str, str] = field(default_factory=dict)
     retry_count: int = 3
     retry_delay: float = 1.0
     timeout: float = 30.0
 
-    def __post_init__(self):
-        if self.headers is None:
-            self.headers = {}
-
 
 class WebhookClient:
     """

{a2a_lite-0.2.1.dist-info → a2a_lite-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: a2a-lite
-Version: 0.2.1
+Version: 0.2.2
 Summary: Simplified wrapper for Google's A2A Protocol SDK
 Author: A2A Lite Contributors
 License-Expression: Apache-2.0
@@ -27,6 +27,8 @@ Provides-Extra: dev
 Requires-Dist: httpx>=0.25; extra == 'dev'
 Requires-Dist: pytest-asyncio>=0.21; extra == 'dev'
 Requires-Dist: pytest>=7.0; extra == 'dev'
+Provides-Extra: oauth
+Requires-Dist: pyjwt[crypto]>=2.0; extra == 'oauth'
 Description-Content-Type: text/markdown
 
 # A2A Lite - Python

a2a_lite-0.2.2.dist-info/RECORD

@@ -0,0 +1,19 @@
+a2a_lite/__init__.py,sha256=-uZfl-CoWKt4xFSmda3TEBmPB7pRA71u9C4HkK8Pavs,3520
+a2a_lite/agent.py,sha256=OrIzIaJLv7O55G1jrSl81pNem9S7kD47QDOzxISAmw0,17793
+a2a_lite/auth.py,sha256=A1AMncM0cuWEAcysjumAhjd0lI_jMLmJpeYWuAPj30A,10181
+a2a_lite/cli.py,sha256=chLmUrWPxXmXU75LO8yhN5HP0rct3yN8W_i_txTvoJA,9043
+a2a_lite/decorators.py,sha256=RDekZdDJQR8124zX0lvTinaU7iJdNjNHaNPop17_gmg,1116
+a2a_lite/discovery.py,sha256=BxpiJAUDxIyI2gvsLhjmHte5c9ax5Qf1hbBQnyAmxLQ,4508
+a2a_lite/executor.py,sha256=gC_9mzoGalUJMZUio-SQeXcoZvOU4LR69eK9ikncJwQ,13628
+a2a_lite/human_loop.py,sha256=XAqxp-k8I7TNyuLqqNmLEqABHqcAUiKYCL8n3W5StaY,8685
+a2a_lite/middleware.py,sha256=c6jb9aFfyTf-JY6KjqaSgFJmpzqbHLC6Q1h9NNteqzo,5545
+a2a_lite/parts.py,sha256=qVRiD-H9_NlMPk-R0gTUiGVQ77E2poiuBWAUyAyAoTI,6177
+a2a_lite/streaming.py,sha256=RFv9EJYnhwkT0h1Wovkj4EXwFzCgHdaA-h7WpPaaONo,2329
+a2a_lite/tasks.py,sha256=UpmDP-VGIQ1LodBNq4zx2pJElQ31gOJOAduHFBVyxOA,7039
+a2a_lite/testing.py,sha256=M9IbLA6oUz1DokJ9Sc_r0gK43NNkU78IVkiBRuDFFCU,9393
+a2a_lite/utils.py,sha256=AFLYQ4J-F7H_HeYWAeg8H3p9EOdDv4dOpju_ebrU5PI,3934
+a2a_lite/webhooks.py,sha256=TNhDlG84rrP_gC2pQ-op5xo01p5Z9sm_ZgQIrJRI7OY,6095
+a2a_lite-0.2.2.dist-info/METADATA,sha256=o74kDuPcSpG1A82qVROqwqMMTTvR95vigXwtx64OeCU,12657
+a2a_lite-0.2.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+a2a_lite-0.2.2.dist-info/entry_points.txt,sha256=BONfFqZbCntNal2iwlTJAE09gCUvurfvqslMYVYh4is,46
+a2a_lite-0.2.2.dist-info/RECORD,,

a2a_lite-0.2.1.dist-info/RECORD

@@ -1,19 +0,0 @@
-a2a_lite/__init__.py,sha256=5bFZw7LapphdqfvMvyKfyEthe_kH4iigeGChd5h3TI4,3520
-a2a_lite/agent.py,sha256=WyHe9aA_BYkzIXRjN5byoRc-vItCyL6YLDiyfX5gfPU,17053
-a2a_lite/auth.py,sha256=bjPu_xnAxmfK5_dkqYz5mHouAjHqJFWlfYMqn-S1-A4,10177
-a2a_lite/cli.py,sha256=D46mGduJb8AcZz8WRczNx5OzvIGxqZ7H-4fLCy2LpX8,9043
-a2a_lite/decorators.py,sha256=VdinkddmF61IS8TkjdSfHGdn3nDKwepcIjWZPZBKg3w,1050
-a2a_lite/discovery.py,sha256=BxpiJAUDxIyI2gvsLhjmHte5c9ax5Qf1hbBQnyAmxLQ,4508
-a2a_lite/executor.py,sha256=zkTotenRNa7wlHh6pnIcS546fx5laSgoVIxZizE7SQI,13218
-a2a_lite/human_loop.py,sha256=XAqxp-k8I7TNyuLqqNmLEqABHqcAUiKYCL8n3W5StaY,8685
-a2a_lite/middleware.py,sha256=c6jb9aFfyTf-JY6KjqaSgFJmpzqbHLC6Q1h9NNteqzo,5545
-a2a_lite/parts.py,sha256=qVRiD-H9_NlMPk-R0gTUiGVQ77E2poiuBWAUyAyAoTI,6177
-a2a_lite/streaming.py,sha256=RFv9EJYnhwkT0h1Wovkj4EXwFzCgHdaA-h7WpPaaONo,2329
-a2a_lite/tasks.py,sha256=UpmDP-VGIQ1LodBNq4zx2pJElQ31gOJOAduHFBVyxOA,7039
-a2a_lite/testing.py,sha256=blugOpPKNThlbFTTSCyBVHb7tgQH79RH8z7Vo45hGbg,8953
-a2a_lite/utils.py,sha256=CnkO6HH9oKEXymbJG2ohdt1ESxldQ3fkmcYVO-o9R_k,3841
-a2a_lite/webhooks.py,sha256=t6ebT3jVBEKFpjhBnPI-nuQWIUKQUbJm24phXOBnNKA,6158
-a2a_lite-0.2.1.dist-info/METADATA,sha256=5QLYvcVJh3qwri1zR3Gg-so5YLaAXXAIr-TvvhAg9UM,12583
-a2a_lite-0.2.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-a2a_lite-0.2.1.dist-info/entry_points.txt,sha256=BONfFqZbCntNal2iwlTJAE09gCUvurfvqslMYVYh4is,46
-a2a_lite-0.2.1.dist-info/RECORD,,