ajarbot/examples/sub_agent_example.py

"""
Example: Using Sub-Agent Orchestration

This example demonstrates how to use the sub-agent system to delegate
specialized tasks to focused agents.
"""

import sys
from pathlib import Path

# Add parent directory to path
sys.path.insert(0, str(Path(__file__).parent.parent))

from agent import Agent


def example_1_manual_spawning():
    """Example 1: Manually spawn and use a specialist."""
    print("=== Example 1: Manual Spawning ===\n")

    # Create main agent
    agent = Agent(provider="claude")

    # Spawn a zettelkasten specialist
    zettel_specialist = agent.spawn_sub_agent(
        specialist_prompt="""You are a zettelkasten expert. Your ONLY job is:
        - Process fleeting notes into permanent notes
        - Find semantic connections using hybrid search
        - Create wiki-style links between related concepts

        Stay focused on knowledge management. Be concise.""",
        agent_id="zettelkasten_processor"  # Cached for reuse
    )

    # Use the specialist
    result = zettel_specialist.chat(
        "Search for all fleeting notes tagged 'AI' and show me what you find.",
        username="jordan"
    )

    print(f"Specialist Response:\n{result}\n")

    # Reuse the cached specialist
    result2 = zettel_specialist.chat(
        "Now create a permanent note summarizing key AI concepts.",
        username="jordan"
    )

    print(f"Second Response:\n{result2}\n")


def example_2_delegation():
    """Example 2: One-off delegation (convenience method)."""
    print("=== Example 2: Delegation ===\n")

    agent = Agent(provider="claude")

    # One-off delegation (specialist not cached)
    result = agent.delegate(
        task="List all files in the memory_workspace/obsidian directory",
        specialist_prompt="""You are a file system expert. Your job is to:
        - Navigate directories efficiently
        - Provide clear, organized file listings

        Be concise and focused.""",
        username="jordan"
    )

    print(f"Delegation Result:\n{result}\n")


def example_3_cached_delegation():
    """Example 3: Cached delegation (reuse specialist)."""
    print("=== Example 3: Cached Delegation ===\n")

    agent = Agent(provider="claude")

    # First call: Creates and caches the specialist
    result1 = agent.delegate(
        task="Search the zettelkasten vault for notes about 'architecture'",
        specialist_prompt="""You are a zettelkasten search expert. Your job is:
        - Use hybrid search to find relevant notes
        - Summarize key findings concisely

        Stay focused on search and retrieval.""",
        username="jordan",
        agent_id="zettel_search"  # This specialist will be cached
    )

    print(f"First Search:\n{result1}\n")

    # Second call: Reuses the cached specialist
    result2 = agent.delegate(
        task="Now search for notes about 'design patterns'",
        specialist_prompt="(ignored - using cached specialist)",
        username="jordan",
        agent_id="zettel_search"  # Same ID = reuse cached specialist
    )

    print(f"Second Search:\n{result2}\n")


def example_4_multiple_specialists():
    """Example 4: Use multiple specialists for different tasks."""
    print("=== Example 4: Multiple Specialists ===\n")

    agent = Agent(provider="claude")

    # Email specialist
    email_result = agent.delegate(
        task="Check if there are any unread emails in the last 24 hours",
        specialist_prompt="""You are an email analyst. Your job is:
        - Search and filter emails efficiently
        - Summarize key information concisely

        Focus on email intelligence.""",
        username="jordan",
        agent_id="email_analyst"
    )

    print(f"Email Analysis:\n{email_result}\n")

    # Calendar specialist
    calendar_result = agent.delegate(
        task="Show me my calendar events for the next 3 days",
        specialist_prompt="""You are a calendar expert. Your job is:
        - Retrieve calendar events efficiently
        - Present schedules clearly

        Focus on time management.""",
        username="jordan",
        agent_id="calendar_manager"
    )

    print(f"Calendar Review:\n{calendar_result}\n")


def example_5_isolated_memory():
    """Example 5: Create specialist with isolated memory."""
    print("=== Example 5: Isolated Memory ===\n")

    agent = Agent(provider="claude")

    # Specialist with its own memory workspace
    specialist = agent.spawn_sub_agent(
        specialist_prompt="You are a research assistant. Focus on gathering information.",
        agent_id="researcher",
        share_memory=False  # Isolated workspace
    )

    # This specialist's memory is stored in:
    # memory_workspace/sub_agents/researcher/

    result = specialist.chat(
        "Research the concept of 'emergence' and save findings.",
        username="jordan"
    )

    print(f"Research Result:\n{result}\n")


if __name__ == "__main__":
    # Run examples
    # Uncomment the examples you want to try:

    # example_1_manual_spawning()
    # example_2_delegation()
    # example_3_cached_delegation()
    # example_4_multiple_specialists()
    # example_5_isolated_memory()

    print("\nℹ️  Uncomment the examples you want to run in the __main__ block")
    print("ℹ️  Note: Some examples require Google OAuth setup and active API keys")