Rename WorkerPool -> Pool, better Pipeline validation, better Pipeline

flow tests.

Author: tcdent

README.md

@@ -1,6 +1,10 @@
 # `agentexec`
 
-[![codecov](https://codecov.io/gh/Agent-CI/agentexec/branch/main/graph/badge.svg)](https://codecov.io/gh/Agent-CI/agentexec)
+[![PyPI](https://img.shields.io/pypi/v/agentexec?color=blue)](https://pypi.org/project/agentexec/)
+[![Python](https://img.shields.io/badge/python-3.12+-blue)](https://www.python.org/)
+[![License](https://img.shields.io/badge/license-MIT-green)](LICENSE)
+[![Type Checked](https://img.shields.io/badge/type%20checked-ty-blue)](https://github.com/astral-sh/ty)
+[![Code Style](https://img.shields.io/badge/code%20style-ruff-orange)](https://github.com/astral-sh/ruff)
 
 **Production-ready orchestration for OpenAI Agents SDK** with Redis-backed task queues, SQLAlchemy activity tracking, and multiprocessing worker pools.
 
@@ -38,7 +42,7 @@ uv add agentexec
 ```
 
 **Requirements:**
-- Python 3.11+
+- Python 3.12+
 - Redis (for task queue)
 - SQLAlchemy-compatible database (for activity tracking)
 - Agents that you want to parallelize!
@@ -68,7 +72,7 @@ class ResearchContext(BaseModel):
 engine = create_engine("sqlite:///agents.db")
 
 # Create worker pool
-pool = ax.WorkerPool(engine=engine)
+pool = ax.Pool(engine=engine)
 
 
 @pool.task("research_company")
@@ -338,7 +342,7 @@ class MyContext(BaseModel):
     param: str
 
 
-pool = ax.WorkerPool(engine=engine)
+pool = ax.Pool(engine=engine)
 
 
 @pool.task("task_name")
@@ -435,7 +439,7 @@ ENV AGENTEXEC_WORKER_MODULE=src.worker
 import os
 import agentexec as ax
 
-pool = ax.WorkerPool(database_url=os.environ["DATABASE_URL"])
+pool = ax.Pool(database_url=os.environ["DATABASE_URL"])
 
 @pool.task("my_task")
 async def my_task(agent_id, context):
@@ -472,7 +476,7 @@ uv sync
 uv run pytest
 
 # Type checking
-uv run mypy src/agentexec
+uv run ty check
 
 # Linting
 uv run ruff check src/

docker/README.md

@@ -11,7 +11,7 @@ Base Docker image for deploying agentexec workers.
 import os
 import agentexec as ax
 
-pool = ax.WorkerPool(database_url=os.environ["DATABASE_URL"])
+pool = ax.Pool(database_url=os.environ["DATABASE_URL"])
 
 @pool.task("my_task")
 async def my_task(agent_id, context):
@@ -71,13 +71,13 @@ Your worker module must expose either:
 
 1. A `pool` variable (recommended):
    ```python
-   pool = ax.WorkerPool(database_url=os.environ["DATABASE_URL"])
+   pool = ax.Pool(database_url=os.environ["DATABASE_URL"])
    ```
 
 2. Or a `create_pool()` function:
    ```python
    def create_pool():
-       return ax.WorkerPool(database_url=os.environ["DATABASE_URL"])
+       return ax.Pool(database_url=os.environ["DATABASE_URL"])
    ```
 
 ## Docker Compose Example

docker/entrypoint.py

@@ -2,7 +2,7 @@
 """Worker entrypoint for agentexec Docker container.
 
 This script dynamically imports the user's worker module and starts the pool.
-The module must expose a `pool` variable (WorkerPool instance) or a
+The module must expose a `pool` variable (Pool instance) or a
 `create_pool()` function that returns one.
 
 Environment Variables:
@@ -21,12 +21,12 @@
 
 
 def get_pool():
-    """Import user module and extract the WorkerPool instance."""
+    """Import user module and extract the Pool instance."""
     module_path = os.environ.get("AGENTEXEC_WORKER_MODULE")
 
     if not module_path:
         print("Error: AGENTEXEC_WORKER_MODULE environment variable is required")
-        print("Set it to your Python module containing the WorkerPool instance")
+        print("Set it to your Python module containing the Pool instance")
         print("Example: AGENTEXEC_WORKER_MODULE=myapp.worker")
         sys.exit(1)
 
@@ -54,7 +54,7 @@ def get_pool():
     else:
         print(f"Error: Module '{module_path}' must expose 'pool' or 'create_pool()'")
         print("Example:")
-        print("  pool = ax.WorkerPool(database_url=os.environ['DATABASE_URL'])")
+        print("  pool = ax.Pool(database_url=os.environ['DATABASE_URL'])")
         sys.exit(1)
 
     return pool

docs/api-reference/core.md

@@ -16,7 +16,7 @@ ax.get_result()
 ax.Priority
 
 # Worker
-ax.WorkerPool
+ax.Pool
 
 # Activity
 ax.activity
@@ -235,12 +235,12 @@ Execute the task handler (called by workers).
 
 ---
 
-## WorkerPool
+## Pool
 
 Manages multi-process worker execution.
 
 ```python
-class WorkerPool:
+class Pool:
     def __init__(
         self,
         engine: Engine,
@@ -315,7 +315,7 @@ import agentexec as ax
 engine = create_engine("sqlite:///agents.db")
 ax.Base.metadata.create_all(engine)
 
-pool = ax.WorkerPool(
+pool = ax.Pool(
     engine=engine,
     database_url="sqlite:///agents.db"
 )

docs/api-reference/pipeline.md

@@ -8,21 +8,21 @@ Orchestrates multi-step task workflows.
 
 ```python
 class Pipeline:
-    def __init__(self, pool: WorkerPool)
+    def __init__(self, pool: Pool)
 ```
 
 ### Parameters
 
 | Parameter | Type | Description |
 |-----------|------|-------------|
-| `pool` | `WorkerPool` | Worker pool for task registration and enqueueing |
+| `pool` | `Pool` | Worker pool for task registration and enqueueing |
 
 ### Example
 
 ```python
 import agentexec as ax
 
-pool = ax.WorkerPool(database_url="sqlite:///agents.db")
+pool = ax.Pool(database_url="sqlite:///agents.db")
 pipeline = ax.Pipeline(pool)
 
 class MyPipeline(pipeline.Base):
@@ -352,7 +352,7 @@ class ReportContext(BaseModel):
     format: str
 
 # Create pool and pipeline
-pool = ax.WorkerPool(database_url="sqlite:///agents.db")
+pool = ax.Pool(database_url="sqlite:///agents.db")
 pipeline = ax.Pipeline(pool)
 
 class ResearchPipeline(pipeline.Base):

docs/concepts/architecture.md

@@ -50,7 +50,7 @@ Redis provides atomic operations, persistence options, and excellent performance
 The worker pool manages multiple Python processes:
 
 ```python
-pool = ax.WorkerPool(engine=engine, database_url=DATABASE_URL)
+pool = ax.Pool(engine=engine, database_url=DATABASE_URL)
 
 @pool.task("my_task")
 async def my_task(agent_id: UUID, context: MyContext):
@@ -256,10 +256,10 @@ Use multiple queues for different workloads:
 
 ```python
 # High-priority pool
-pool_high = ax.WorkerPool(queue_name="high_priority", ...)
+pool_high = ax.Pool(queue_name="high_priority", ...)
 
 # Low-priority pool
-pool_low = ax.WorkerPool(queue_name="low_priority", ...)
+pool_low = ax.Pool(queue_name="low_priority", ...)
 
 # Enqueue to specific queue
 await ax.enqueue("urgent_task", ctx, queue_name="high_priority")

docs/concepts/task-lifecycle.md

@@ -100,7 +100,7 @@ class MyContext(BaseModel):
     company: str
     priority: int = 1
 
-pool = ax.WorkerPool(engine=engine, database_url=DATABASE_URL)
+pool = ax.Pool(engine=engine, database_url=DATABASE_URL)
 
 @pool.task("research_company")
 async def research_company(agent_id: UUID, context: MyContext) -> str:

docs/concepts/worker-pool.md

@@ -6,7 +6,7 @@ The worker pool is the execution engine of agentexec. It manages multiple Python
 
 ```
 ┌─────────────────────────────────────────────────────────┐
-│                    WorkerPool                           │
+│                    Pool                           │
 │  ┌─────────────────────────────────────────────────┐   │
 │  │                Main Process                      │   │
 │  │  • Spawns worker processes                       │   │
@@ -39,7 +39,7 @@ engine = create_engine("postgresql://user:pass@localhost/mydb")
 ax.Base.metadata.create_all(engine)
 
 # Create worker pool
-pool = ax.WorkerPool(
+pool = ax.Pool(
     engine=engine,
     database_url="postgresql://user:pass@localhost/mydb",
     queue_name=None,  # Uses CONF.queue_name by default
@@ -351,7 +351,7 @@ Run multiple pools for different workloads:
 
 ```python
 # High-priority tasks
-pool_high = ax.WorkerPool(
+pool_high = ax.Pool(
     engine=engine,
     database_url=DATABASE_URL,
     queue_name="high_priority",
@@ -362,7 +362,7 @@ async def urgent_task(agent_id: UUID, context: UrgentContext):
     ...
 
 # Low-priority tasks
-pool_low = ax.WorkerPool(
+pool_low = ax.Pool(
     engine=engine,
     database_url=DATABASE_URL,
     queue_name="low_priority",

docs/contributing.md

@@ -177,7 +177,7 @@ Use clear, descriptive commit messages:
 feat: add support for custom queue names
 
 - Allow specifying queue_name in enqueue()
-- Update WorkerPool to accept queue_name parameter
+- Update Pool to accept queue_name parameter
 - Add tests for custom queue functionality
 ```
 

docs/getting-started/configuration.md

@@ -28,7 +28,7 @@ These settings must be provided - there are no defaults:
 |----------|-------------|---------|
 | `REDIS_URL` or `AGENTEXEC_REDIS_URL` | Redis connection URL | `redis://localhost:6379/0` |
 
-The `DATABASE_URL` is passed directly to `WorkerPool`, not through configuration.
+The `DATABASE_URL` is passed directly to `Pool`, not through configuration.
 
 ## Optional Settings
 
@@ -127,7 +127,7 @@ Create a `.env` file in your project root:
 # Required
 REDIS_URL=redis://localhost:6379/0
 
-# Database (passed to WorkerPool)
+# Database (passed to Pool)
 DATABASE_URL=postgresql://user:password@localhost:5432/myapp
 
 # OpenAI (for agents)
@@ -269,7 +269,7 @@ REDIS_URL=rediss://localhost:6379/0
 
 ## Database URL Format
 
-The `DATABASE_URL` passed to `WorkerPool` follows SQLAlchemy conventions:
+The `DATABASE_URL` passed to `Pool` follows SQLAlchemy conventions:
 
 ```bash
 # PostgreSQL