Skip to main content

Base agent class for all Beast Mode agents

Project description

Beast Agent

Base agent class for ALL Beast Mode agents

PyPI version Python Versions License: MIT Code style: black


🎯 Purpose

The foundational base class (BaseAgent) that EVERY Beast Mode agent inherits from.

This package provides:

  • ✅ Standardized agent lifecycle (startup, shutdown, health checks)
  • ✅ Message handling via beast-mailbox-core integration
  • ✅ Agent registration and discovery
  • ✅ Capability declaration and management
  • ✅ Optional logging/telemetry hooks
  • ✅ Configuration management

Architectural Clarity:

  • beast-agent = Low-level base class (every agent IS-A beast-agent)
  • beast-agentic-framework = High-level orchestration (multi-agent coordination)

🚀 Quick Start

Installation

pip install beast-agent

Create Your First Agent

from beast_agent import BaseAgent
from beast_agent.decorators import capability

class MyAgent(BaseAgent):
    """Simple agent with a single capability"""
    
    def __init__(self):
        super().__init__(
            agent_id="my-agent",
            capabilities=["process_data"]
        )
    
    @capability("process_data")
    async def process_data(self, data: dict) -> dict:
        """Process data and return results"""
        # Your agent logic here
        return {"status": "processed", "result": data}

# Run the agent
async def main():
    agent = MyAgent()
    await agent.startup()
    # Agent is now ready to handle messages
    
if __name__ == "__main__":
    import asyncio
    asyncio.run(main())

📋 Agent Patterns

Per-Repo Agent Pattern

class RepoAgent(BaseAgent):
    """Agent that monitors a single repository"""
    
    def __init__(self, repo_name: str):
        super().__init__(
            agent_id=f"repo-agent-{repo_name}",
            capabilities=["code_review", "pr_validation", "security_scan"]
        )
        self.repo_name = repo_name
    
    @capability("code_review")
    async def review_code(self, pr_number: int) -> dict:
        """Review code in PR"""
        # Review logic
        return {"status": "reviewed", "pr": pr_number}

Per-Branch Agent Pattern

class BranchAgent(BaseAgent):
    """Agent specific to a branch"""
    
    def __init__(self, repo_name: str, branch_name: str):
        super().__init__(
            agent_id=f"branch-agent-{repo_name}-{branch_name}",
            capabilities=["deployment", "testing", "monitoring"]
        )
        self.repo = repo_name
        self.branch = branch_name
    
    @capability("deployment")
    async def deploy(self, environment: str) -> dict:
        """Deploy branch to environment"""
        # Deployment logic
        return {"status": "deployed", "environment": environment}

🔧 Features

Agent Lifecycle

agent = MyAgent()
await agent.startup()  # Initialize and register
# ... agent is running ...
await agent.shutdown()  # Graceful cleanup

Message Handling

# Register handler for message type
agent.register_handler("TASK_REQUEST", handle_task_request)

# Send message to another agent
await agent.send_message(
    target="other-agent-id",
    message_type="HELP_REQUEST",
    content={"task": "analyze_code"}
)

Health Checks

health = agent.health_check()
print(f"Healthy: {health.healthy}")
print(f"State: {health.state}")

📚 Documentation


🧪 Testing

# Run all tests
pytest

# Run with coverage
pytest --cov=src/beast_agent --cov-report=html

# Run specific test
pytest tests/test_base_agent.py

# Run integration tests (requires Redis and beast-mailbox-core)
# Note: Integration tests are automatically skipped if dependencies unavailable
pytest tests/test_mailbox_integration.py

# Run all tests including integration (if Redis available)
pytest tests/

Integration Testing

Integration tests require:

  • Redis: Running locally (via Docker) or in CI (via service containers)
  • beast-mailbox-core: Installed as dependency

Local Testing:

  • Redis Docker container is automatically managed via conftest.py fixtures
  • Tests gracefully skip if Redis/Docker unavailable
  • Uses separate test database (db=15) to avoid conflicts

CI Testing:

  • Redis service container automatically provided in GitHub Actions
  • Integration tests run against real Redis in CI

🤝 Integration

Required Dependencies

  • beast-mailbox-core >= 0.3.0 - Messaging and discovery

Optional Dependencies

  • beast-observability - Enhanced telemetry
  • Any cloud platform (works with AWS, GCP, Azure, on-prem)

📦 Package Status

Tier: 1 (Foundation)
Phase: Development
Coverage: Target 90%+
Quality: Target zero defects


🔗 Related Packages

  • beast-mailbox-core - Redis-backed mailbox utilities
  • beast-agentic-framework - Multi-agent orchestration
  • beast-observability - Unified telemetry
  • beast-redaction-client - Data classification

📄 License

MIT License - see LICENSE for details.


🙏 Acknowledgments

Part of the Beast Mode multi-agent framework ecosystem.


Built with ❤️ by the Beast Mode team

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

beast_agent-0.1.2.tar.gz (22.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

beast_agent-0.1.2-py3-none-any.whl (12.1 kB view details)

Uploaded Python 3

File details

Details for the file beast_agent-0.1.2.tar.gz.

File metadata

  • Download URL: beast_agent-0.1.2.tar.gz
  • Upload date:
  • Size: 22.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for beast_agent-0.1.2.tar.gz
Algorithm Hash digest
SHA256 4427ab450793e2fa2e1f9a667c0d8ff9d75810278a74b98a5943125c48c9a213
MD5 261eced3b48f7d864dc435eea7422b00
BLAKE2b-256 d2ffaa5301c6ca2af0f36afc32d86ed18c4e9fcb78dd7b4ddc782cf5e56d391d

See more details on using hashes here.

Provenance

The following attestation bundles were made for beast_agent-0.1.2.tar.gz:

Publisher: publish.yml on nkllon/beast-agent

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file beast_agent-0.1.2-py3-none-any.whl.

File metadata

  • Download URL: beast_agent-0.1.2-py3-none-any.whl
  • Upload date:
  • Size: 12.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for beast_agent-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 19cfab052c980e6f3c9b1d8eaf0c29d02724e8505bac6b3b275c35717b1609c8
MD5 be39a24ae7cf8d1846db52930108a79d
BLAKE2b-256 f2997cf9c268cfefbaa153e480c20b224d6380738b8a9b380235cc4fd62488e7

See more details on using hashes here.

Provenance

The following attestation bundles were made for beast_agent-0.1.2-py3-none-any.whl:

Publisher: publish.yml on nkllon/beast-agent

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page