a2a-lite 0.2.1__tar.gz → 0.2.3__tar.gz

{a2a_lite-0.2.1 → a2a_lite-0.2.3}/PKG-INFO

@@ -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.1 → a2a_lite-0.2.3}/README.md

@@ -1,6 +1,6 @@
 # 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.
 
@@ -116,7 +116,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}
 ```
 
@@ -155,28 +154,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
@@ -187,7 +171,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
@@ -204,7 +188,7 @@ async def process(data: str, task: TaskContext) -> str:
     return "Done!"
 ```
 
-### Level 8: Authentication
+### Level 7: Authentication
 
 ```python
 from a2a_lite import Agent, APIKeyAuth
@@ -229,14 +213,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(
@@ -247,11 +241,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}")
 ```
 
@@ -280,25 +274,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()
@@ -346,59 +331,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
 
@@ -412,7 +357,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
 )
 ```
 
@@ -422,8 +366,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) |
 
@@ -431,7 +378,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
@@ -444,18 +391,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 |
 
 ---
 
@@ -468,10 +416,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 |
 
 ---
 
@@ -483,9 +430,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.1 → a2a_lite-0.2.3}/examples/06_pydantic_models.py

@@ -6,7 +6,7 @@ This is the SIMPLEST way to handle complex data - just use Pydantic!
 Run: python examples/06_pydantic_models.py
 Test: a2a-lite test http://localhost:8787 create_user -p '{"user": {"name": "Alice", "email": "alice@example.com", "age": 30}}'
 """
-from pydantic import BaseModel, EmailStr
+from pydantic import BaseModel
 from typing import List, Optional
 from a2a_lite import Agent
 

a2a_lite-0.2.1/examples/12_file_handling.py → a2a_lite-0.2.3/examples/10_file_handling.py

@@ -3,7 +3,7 @@ Example: Multi-modal file handling.
 
 Handle files, images, documents - not just text!
 
-Run: python examples/12_file_handling.py
+Run: python examples/10_file_handling.py
 """
 from a2a_lite import Agent, FilePart, DataPart, Artifact
 

a2a_lite-0.2.1/examples/13_task_tracking.py → a2a_lite-0.2.3/examples/11_task_tracking.py

@@ -3,7 +3,7 @@ Example: Task lifecycle and progress tracking.
 
 Show users real-time progress for long-running tasks.
 
-Run: python examples/13_task_tracking.py
+Run: python examples/11_task_tracking.py
 """
 import asyncio
 from a2a_lite import Agent, TaskContext

a2a_lite-0.2.1/examples/14_with_auth.py → a2a_lite-0.2.3/examples/12_with_auth.py

@@ -3,7 +3,7 @@ Example: Authentication (optional).
 
 Add auth when you need it - skip it when you don't.
 
-Run: python examples/14_with_auth.py
+Run: python examples/12_with_auth.py
 Test: curl -H "X-API-Key: secret-key" http://localhost:8787/...
 """
 from a2a_lite import Agent, APIKeyAuth

{a2a_lite-0.2.1 → a2a_lite-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "a2a-lite"
-version = "0.2.1"
+version = "0.2.3"
 description = "Simplified wrapper for Google's A2A Protocol SDK"
 readme = "README.md"
 license = "Apache-2.0"
@@ -24,14 +24,15 @@ dependencies = [
     "pydantic>=2.0",
     "typer>=0.9.0",
     "rich>=13.0",
-    "watchfiles>=0.20.0",
-    "zeroconf>=0.80.0",
     "uvicorn>=0.30.0",
     "httpx>=0.25.0",
     "starlette>=0.40.0",
 ]
 
 [project.optional-dependencies]
+oauth = [
+    "pyjwt[crypto]>=2.0",
+]
 dev = [
     "pytest>=7.0",
     "pytest-asyncio>=0.21",