System Architecture

async def get_or_create_team(user_id: str, client=None):
    """
    Get existing team for a user or create a new one.
    Uses LRU eviction if cache exceeds MAX_AGENTS.
    Implements proper resource cleanup when evicting teams.
    """

async def async_ask_junkie(user_text: str, user_id: str, session_id: str, images: list = None, client=None) -> str:
    """
    Run the user's Team with improved error handling and response validation.
    """
    team = await get_or_create_team(user_id, client=client)
    result = await team.arun(
        input=user_text, user_id=user_id, session_id=session_id, images=images
    )

CREATE TABLE messages (
    message_id BIGINT PRIMARY KEY,
    channel_id BIGINT NOT NULL,
    author_id BIGINT NOT NULL,
    author_name TEXT NOT NULL,
    content TEXT NOT NULL,
    created_at TIMESTAMP WITH TIME ZONE NOT NULL,
    timestamp_str TEXT NOT NULL
);

CREATE INDEX idx_messages_channel_created
ON messages (channel_id, created_at DESC);

async def get_recent_context(channel, limit: int = 500, before_message=None) -> List[str]:
    # 1. Try DB first
    db_messages = await get_messages(channel_id, limit)
    
    # 2. If DB has insufficient data, fetch from API
    if len(db_messages) == 0:
        return await fetch_and_cache_from_api(channel, limit, before_message)

# 1. User sends message with "!" prefix
# 2. Resolve Discord mentions to readable format
processed_content = resolve_mentions(message)

# 3. Build context-aware prompt
prompt = await build_context_prompt(
    message, raw_prompt, 
    limit=TEAM_LEADER_CONTEXT_LIMIT, 
    reply_to_message=reply_to_message
)

# 4. Set execution context for tools
set_current_channel_id(message.channel.id)
set_current_channel(message.channel)

# 5. Run team with context
reply = await async_ask_junkie(
    prompt, user_id=user_id, session_id=session_id, images=images
)

if len(_user_teams) >= MAX_AGENTS:
    oldest_user, oldest_team = _user_teams.popitem(last=False)
    
    # Cleanup MCP connections
    if hasattr(oldest_team, 'members'):
        for member in oldest_team.members:
            if hasattr(member, 'tools'):
                for tool in member.tools:
                    if hasattr(tool, 'close'):
                        await tool.close()

pool: Optional[asyncpg.Pool] = None

async def init_db():
    global pool
    pool = await asyncpg.create_pool(POSTGRES_URL)
    await create_schema()

async def close_db():
    if pool:
        await pool.close()

# Context variable to store the current channel ID
_current_channel_id: contextvars.ContextVar[Optional[int]] = 
    contextvars.ContextVar("current_channel_id", default=None)

def set_current_channel_id(channel_id: int):
    _current_channel_id.set(channel_id)

def get_current_channel_id() -> Optional[int]:
    return _current_channel_id.get()