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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: a2a-lite
-Version: 0.2.1
+Version: 0.2.3
 Summary: Simplified wrapper for Google's A2A Protocol SDK
 Author: A2A Lite Contributors
 License-Expression: Apache-2.0
@@ -21,17 +21,17 @@ Requires-Dist: rich>=13.0
 Requires-Dist: starlette>=0.40.0
 Requires-Dist: typer>=0.9.0
 Requires-Dist: uvicorn>=0.30.0
-Requires-Dist: watchfiles>=0.20.0
-Requires-Dist: zeroconf>=0.80.0
 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
 
-**Build A2A agents in 8 lines. Add enterprise features when you need them.**
+**Build A2A agents in 8 lines. Add features when you need them.**
 
 Wraps the official [A2A Python SDK](https://github.com/a2aproject/a2a-python) with a simple, intuitive API.
 
@@ -147,7 +147,6 @@ class User(BaseModel):
 
 @agent.skill("create_user")
 async def create_user(user: User) -> dict:
-    # 'user' is already a User instance — auto-converted from dict!
     return {"id": 1, "name": user.name}
 ```
 
@@ -186,28 +185,13 @@ Built-in middleware:
 ```python
 from a2a_lite import logging_middleware, timing_middleware, retry_middleware, rate_limit_middleware
 
-agent.use(logging_middleware)
-agent.use(timing_middleware)
-agent.use(rate_limit_middleware(max_per_minute=60))
-agent.use(retry_middleware(max_retries=3))
+agent.add_middleware(logging_middleware)
+agent.add_middleware(timing_middleware)
+agent.add_middleware(rate_limit_middleware(max_per_minute=60))
+agent.add_middleware(retry_middleware(max_retries=3))
 ```
 
-### Level 5: Human-in-the-Loop
-
-```python
-from a2a_lite import InteractionContext
-
-@agent.skill("wizard")
-async def wizard(ctx: InteractionContext) -> dict:
-    name = await ctx.ask("What's your name?")
-    role = await ctx.ask("Role?", options=["Dev", "Manager"])
-
-    if await ctx.confirm(f"Create {name} as {role}?"):
-        return {"created": name}
-    return {"cancelled": True}
-```
-
-### Level 6: File Handling
+### Level 5: File Handling
 
 ```python
 from a2a_lite import FilePart
@@ -218,7 +202,7 @@ async def summarize(doc: FilePart) -> str:
     return f"Summary: {content[:100]}..."
 ```
 
-### Level 7: Task Tracking
+### Level 6: Task Tracking
 
 ```python
 from a2a_lite import TaskContext
@@ -235,7 +219,7 @@ async def process(data: str, task: TaskContext) -> str:
     return "Done!"
 ```
 
-### Level 8: Authentication
+### Level 7: Authentication
 
 ```python
 from a2a_lite import Agent, APIKeyAuth
@@ -260,14 +244,24 @@ agent = Agent(
     auth=BearerAuth(secret="your-jwt-secret"),
 )
 
-# OAuth2
+# OAuth2 (requires: pip install a2a-lite[oauth])
 agent = Agent(
     name="Bot", description="A bot",
     auth=OAuth2Auth(issuer="https://auth.example.com", audience="my-api"),
 )
 ```
 
-### Level 9: CORS and Production Mode
+Skills can receive auth results by type-hinting a parameter as `AuthResult`:
+
+```python
+from a2a_lite.auth import AuthResult
+
+@agent.skill("whoami")
+async def whoami(auth: AuthResult) -> dict:
+    return {"user": auth.identity, "scheme": auth.scheme}
+```
+
+### Level 8: CORS and Production Mode
 
 ```python
 agent = Agent(
@@ -278,11 +272,11 @@ agent = Agent(
 )
 ```
 
-### Level 10: Webhooks
+### Level 9: Completion Hooks
 
 ```python
 @agent.on_complete
-async def notify(skill_name, result):
+async def notify(skill_name, result, ctx):
     print(f"Skill {skill_name} completed with: {result}")
 ```
 
@@ -311,25 +305,16 @@ async def info(name: str, age: int) -> dict:
 def test_simple_result():
     client = AgentTestClient(agent)
     result = client.call("greet", name="World")
-    # Simple values support direct equality
     assert result == "Hello, World!"
 
 
 def test_dict_result():
     client = AgentTestClient(agent)
     result = client.call("info", name="Alice", age=30)
-    # Access dict results via .data
     assert result.data["name"] == "Alice"
     assert result.data["age"] == 30
 
 
-def test_text_access():
-    client = AgentTestClient(agent)
-    result = client.call("greet", name="World")
-    # .text gives the raw string
-    assert result.text == '"Hello, World!"'
-
-
 def test_list_skills():
     client = AgentTestClient(agent)
     skills = client.list_skills()
@@ -377,59 +362,19 @@ def test_streaming():
 
 ---
 
-## Task Store
-
-The `TaskStore` provides async-safe task lifecycle management:
-
-```python
-from a2a_lite.tasks import TaskStore, TaskStatus
-
-store = TaskStore()
-
-# All operations are async and thread-safe
-task = await store.create(task_id="task-1", skill="process")
-task = await store.get("task-1")
-await store.update("task-1", status=TaskStatus.WORKING, progress=0.5)
-tasks = await store.list()
-await store.delete("task-1")
-```
-
----
-
-## Agent Discovery
-
-Find agents on your local network via mDNS:
-
-```python
-from a2a_lite import AgentDiscovery
-
-# Advertise your agent
-agent.run(port=8787, enable_discovery=True)
-
-# Discover other agents
-discovery = AgentDiscovery()
-agents = await discovery.discover(timeout=5.0)
-for a in agents:
-    print(f"{a.name} at {a.url}")
-```
-
----
-
 ## CLI
 
 ```bash
 a2a-lite init my-agent          # Create new project
 a2a-lite serve agent.py         # Run agent from file
-a2a-lite serve agent.py -r      # Run with hot reload
 a2a-lite inspect http://...     # View agent capabilities
 a2a-lite test http://... skill  # Test a skill
-a2a-lite discover               # Find local agents
 a2a-lite version                # Show version
 ```
 
 ---
 
-## Full API Reference
+## API Reference
 
 ### Agent
 
@@ -443,7 +388,6 @@ Agent(
     task_store: str | TaskStore = None, # "memory" or custom TaskStore
     cors_origins: List[str] = None,     # CORS allowed origins
     production: bool = False,           # Enable production warnings
-    enable_discovery: bool = False,     # mDNS discovery
 )
 ```
 
@@ -453,8 +397,11 @@ Agent(
 |--------|-------------|
 | `@agent.skill(name, **config)` | Register a skill via decorator |
 | `@agent.middleware` | Register middleware via decorator |
-| `agent.use(middleware)` | Register middleware function |
+| `agent.add_middleware(fn)` | Register middleware function |
 | `@agent.on_complete` | Register completion hook |
+| `@agent.on_startup` | Register startup hook |
+| `@agent.on_shutdown` | Register shutdown hook |
+| `@agent.on_error` | Register error handler |
 | `agent.run(port=8787)` | Start the server |
 | `agent.get_app()` | Get the ASGI app (for custom deployment) |
 
@@ -462,7 +409,7 @@ Agent(
 
 ```python
 @agent.skill(
-    name: str,                    # Skill name (required)
+    name: str = None,             # Skill name (defaults to function name)
     description: str = None,      # Human-readable description
     tags: List[str] = None,       # Categorization tags
     streaming: bool = False,      # Enable streaming
@@ -475,18 +422,19 @@ Agent(
 |----------|-------|
 | `APIKeyAuth(keys=[...])` | API key auth (keys hashed with SHA-256) |
 | `BearerAuth(secret=...)` | JWT/Bearer token auth |
-| `OAuth2Auth(issuer=..., audience=...)` | OAuth2 auth |
+| `OAuth2Auth(issuer=..., audience=...)` | OAuth2 auth (requires `a2a-lite[oauth]`) |
 | `NoAuth()` | No auth (default) |
 
 ### Special Parameter Types
 
-These are auto-injected when detected in skill signatures:
+These are auto-injected when detected in skill function signatures:
 
 | Type | Description |
 |------|-------------|
 | `TaskContext` | Task lifecycle management (requires `task_store`) |
-| `InteractionContext` | Human-in-the-loop interactions |
+| `AuthResult` | Authentication result injection |
 | `FilePart` | File upload handling |
+| `DataPart` | Structured data handling |
 
 ---
 
@@ -499,10 +447,9 @@ These are auto-injected when detected in skill signatures:
 | [06_pydantic_models.py](examples/06_pydantic_models.py) | Auto Pydantic conversion |
 | [08_streaming.py](examples/08_streaming.py) | Streaming responses |
 | [09_testing.py](examples/09_testing.py) | Testing your agents |
-| [11_human_in_the_loop.py](examples/11_human_in_the_loop.py) | Ask user questions |
-| [12_file_handling.py](examples/12_file_handling.py) | Handle files |
-| [13_task_tracking.py](examples/13_task_tracking.py) | Progress updates |
-| [14_with_auth.py](examples/14_with_auth.py) | Authentication |
+| [10_file_handling.py](examples/10_file_handling.py) | Handle files |
+| [11_task_tracking.py](examples/11_task_tracking.py) | Progress updates |
+| [12_with_auth.py](examples/12_with_auth.py) | Authentication |
 
 ---
 
@@ -514,9 +461,9 @@ A2A Lite wraps the official A2A Python SDK. Every feature maps to real A2A proto
 |----------|--------------|
 | `@agent.skill()` | Agent Skills |
 | `streaming=True` | SSE Streaming |
-| `InteractionContext.ask()` | `input-required` state |
 | `TaskContext.update()` | Task lifecycle states |
 | `FilePart` | A2A File parts |
+| `DataPart` | A2A Data parts |
 | `APIKeyAuth` / `BearerAuth` | Security schemes |
 
 ---

a2a_lite-0.2.3.dist-info/RECORD

@@ -0,0 +1,16 @@
+a2a_lite/__init__.py,sha256=92WTqzqJV2FTBV0peiuLUdB9y5BZe37jvCT7E2k_vNQ,2642
+a2a_lite/agent.py,sha256=2x4t4ysh10Qwjwp4S8pQehf8O3sINGoJIvq9YN9G-q0,15321
+a2a_lite/auth.py,sha256=A1AMncM0cuWEAcysjumAhjd0lI_jMLmJpeYWuAPj30A,10181
+a2a_lite/cli.py,sha256=5oAwZLR59ebxhuJjIYSHd_AskZqToQLRz45jAAlEBp0,7680
+a2a_lite/decorators.py,sha256=rRzpeVW6mZTAFpNzZYDdraShbDdq7S32c4Z16edD4xw,979
+a2a_lite/executor.py,sha256=pZx_3_JV34qWgbNDkzCWmT1qJRuBAxhrmz4luU7bDBc,13163
+a2a_lite/middleware.py,sha256=c6jb9aFfyTf-JY6KjqaSgFJmpzqbHLC6Q1h9NNteqzo,5545
+a2a_lite/parts.py,sha256=qVRiD-H9_NlMPk-R0gTUiGVQ77E2poiuBWAUyAyAoTI,6177
+a2a_lite/streaming.py,sha256=6qLMUlbk61HBhtotxhSz5Yh4wL9OHeGt24o_J--RJvI,1626
+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-0.2.3.dist-info/METADATA,sha256=x9DLlTzprZMk8qT210WM4dMxHiDaWjX5h9GFnbVxGrY,11392
+a2a_lite-0.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+a2a_lite-0.2.3.dist-info/entry_points.txt,sha256=BONfFqZbCntNal2iwlTJAE09gCUvurfvqslMYVYh4is,46
+a2a_lite-0.2.3.dist-info/RECORD,,

a2a_lite/discovery.py

@@ -1,152 +0,0 @@
-"""
-mDNS-based local agent discovery using Zeroconf.
-"""
-from __future__ import annotations
-
-import asyncio
-import logging
-import socket
-from typing import Dict, List, Optional
-from dataclasses import dataclass
-
-logger = logging.getLogger(__name__)
-
-from zeroconf import ServiceBrowser, ServiceListener, Zeroconf, ServiceInfo
-from zeroconf.asyncio import AsyncZeroconf, AsyncServiceBrowser
-
-
-SERVICE_TYPE = "_a2a-agent._tcp.local."
-
-
-@dataclass
-class DiscoveredAgent:
-    """Information about a discovered A2A agent."""
-    name: str
-    host: str
-    port: int
-    url: str
-    properties: Dict[str, str]
-
-
-class AgentDiscovery:
-    """
-    Discover A2A agents on the local network using mDNS.
-    """
-
-    def __init__(self):
-        self._discovered: Dict[str, DiscoveredAgent] = {}
-        self._zeroconf: Optional[Zeroconf] = None
-        self._service_info: Optional[ServiceInfo] = None
-
-    async def discover(self, timeout: float = 5.0) -> List[DiscoveredAgent]:
-        """
-        Discover agents on the local network.
-
-        Args:
-            timeout: How long to wait for discoveries
-
-        Returns:
-            List of discovered agents
-        """
-        self._discovered.clear()
-
-        async with AsyncZeroconf() as azc:
-            listener = _DiscoveryListener(self._discovered)
-            browser = AsyncServiceBrowser(
-                azc.zeroconf, SERVICE_TYPE, listener
-            )
-
-            await asyncio.sleep(timeout)
-            await browser.async_cancel()
-
-        return list(self._discovered.values())
-
-    def register(
-        self,
-        name: str,
-        port: int,
-        properties: Optional[Dict[str, str]] = None,
-    ) -> ServiceInfo:
-        """
-        Register this agent for discovery.
-
-        Args:
-            name: Agent name
-            port: Port the agent is running on
-            properties: Additional properties to advertise
-
-        Returns:
-            ServiceInfo for unregistration
-        """
-        hostname = socket.gethostname()
-        local_ip = self._get_local_ip()
-
-        # Sanitize name for mDNS (remove spaces and special chars)
-        safe_name = "".join(c if c.isalnum() or c == "-" else "-" for c in name)
-
-        info = ServiceInfo(
-            SERVICE_TYPE,
-            f"{safe_name}.{SERVICE_TYPE}",
-            addresses=[socket.inet_aton(local_ip)],
-            port=port,
-            properties=properties or {},
-            server=f"{hostname}.local.",
-        )
-
-        self._zeroconf = Zeroconf()
-        self._zeroconf.register_service(info)
-        self._service_info = info
-
-        return info
-
-    def unregister(self) -> None:
-        """Unregister this agent from discovery."""
-        if self._zeroconf and self._service_info:
-            self._zeroconf.unregister_service(self._service_info)
-            self._zeroconf.close()
-            self._zeroconf = None
-            self._service_info = None
-
-    def _get_local_ip(self) -> str:
-        """Get local IP address."""
-        try:
-            s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-            s.connect(("8.8.8.8", 80))
-            ip = s.getsockname()[0]
-            s.close()
-            return ip
-        except Exception:
-            logger.debug("Could not detect local IP, falling back to 127.0.0.1")
-            return "127.0.0.1"
-
-
-class _DiscoveryListener(ServiceListener):
-    """Internal listener for mDNS service discovery."""
-
-    def __init__(self, discovered: Dict[str, DiscoveredAgent]):
-        self.discovered = discovered
-
-    def add_service(self, zc: Zeroconf, type_: str, name: str) -> None:
-        info = zc.get_service_info(type_, name)
-        if info:
-            agent_name = name.replace(f".{SERVICE_TYPE}", "")
-            host = socket.inet_ntoa(info.addresses[0]) if info.addresses else "localhost"
-            port = info.port
-
-            self.discovered[name] = DiscoveredAgent(
-                name=agent_name,
-                host=host,
-                port=port,
-                url=f"http://{host}:{port}",
-                properties={
-                    k.decode() if isinstance(k, bytes) else k:
-                    v.decode() if isinstance(v, bytes) else str(v)
-                    for k, v in info.properties.items()
-                },
-            )
-
-    def remove_service(self, zc: Zeroconf, type_: str, name: str) -> None:
-        self.discovered.pop(name, None)
-
-    def update_service(self, zc: Zeroconf, type_: str, name: str) -> None:
-        self.add_service(zc, type_, name)