agent_bootstrap.sh

#!/bin/bash
# Agent bootstrap: creates agents from accumulated cortex knowledge
# Called from SessionStart hook (async) — doesn't block the session
#
# Flow:
#   1. Quick check: does current project need agents? (fast, no LLM)
#   2. Cooldown check: already bootstrapped today? (skip if so)
#   3. Collect cortex memories for the project (full content, grouped)
#   4. ONE call to claude -p (haiku) to propose agents
#   5. fleet_create.py creates with semantic dedup (threshold 0.55)
#
# Safety:
#   - Max 3 agents per run (hard cap in fleet_create.py's agents[:5] + prompt)
#   - Cooldown: once per project per day
#   - Semantic dedup: cosine distance < 0.55 = skip
#   - Total cap: if project already has 5+ agents, skip entirely

INPUT=$(cat 2>/dev/null)
LIB="$(dirname "$0")/lib"
COOLDOWN_DIR="$HOME/.claude/.cortex_bootstrap_cooldown"
mkdir -p "$COOLDOWN_DIR"

# Extract cwd from hook input
CWD=$(echo "$INPUT" | /usr/bin/python3 -c "
import sys, json
try: print(json.loads(sys.stdin.read()).get('cwd', ''))
except: print('')
" 2>/dev/null)

if [ -z "$CWD" ]; then
    CWD=$(pwd)
fi

# ================================================================
# Phase 1: Quick check — does this project need agents?
# Returns: project_needs,global_needs or empty
# ================================================================
NEEDS=$(/usr/bin/python3 -W ignore - "$CWD" 2>/dev/null <<'PYEOF'
import sys, os, glob, warnings
warnings.filterwarnings("ignore")
os.environ["ONNXRUNTIME_DISABLE_TELEMETRY"] = "1"
os.environ["ORT_LOG_LEVEL"] = "ERROR"

_fd = os.dup(2)
_dn = os.open(os.devnull, os.O_WRONLY)
os.dup2(_dn, 2); os.close(_dn)
try:
    import chromadb
finally:
    os.dup2(_fd, 2); os.close(_fd)

DB_PATH = os.path.expanduser("~/.claude/cortex-db")
cwd = sys.argv[1] if len(sys.argv) > 1 else ""

try:
    client = chromadb.PersistentClient(path=DB_PATH)
    col = client.get_or_create_collection("claude_memories")
    if col.count() == 0:
        sys.exit(0)

    all_data = col.get(include=["metadatas"])

    # Count project/reference memories per project
    project_mem_count = {}
    for m in all_data["metadatas"]:
        p = m.get("project", "")
        t = m.get("type", "")
        if t == "agent_eval":
            continue
        if p and t in ("project", "reference"):
            project_mem_count[p] = project_mem_count.get(p, 0) + 1

    # Match cwd to cortex project names
    cwd_lower = cwd.lower()
    matched = [p for p in project_mem_count if p != "global" and p.lower() in cwd_lower]

    needs = []

    for proj in matched:
        mem_count = project_mem_count.get(proj, 0)
        if mem_count < 3:
            continue

        # Count existing project agents
        agent_dir = os.path.join(cwd, ".claude", "agents")
        proj_agents = 0
        if os.path.isdir(agent_dir):
            proj_agents = len([f for f in os.listdir(agent_dir)
                              if f.endswith(".md") and not f.startswith(".")])

        # Hard cap: skip if already 5+ project agents
        if proj_agents >= 5:
            continue

        # Need more agents if: 3+ memories but less than 2 agents
        if proj_agents < 2:
            needs.append(proj)

    # Check global: user-level agents needed?
    user_agent_dir = os.path.expanduser("~/.claude/agents")
    user_agents = 0
    if os.path.isdir(user_agent_dir):
        user_agents = len([f for f in os.listdir(user_agent_dir)
                          if f.endswith(".md") and not f.startswith(".")])

    # Only bootstrap global if no user agents at all AND enough feedback
    feedback_count = sum(1 for m in all_data["metadatas"]
                        if m.get("type") == "feedback")
    if user_agents == 0 and feedback_count >= 5:
        needs.append("global")

    print(",".join(needs) if needs else "")

except Exception:
    print("")
PYEOF
)

if [ -z "$NEEDS" ]; then
    exit 0
fi

# ================================================================
# Phase 2: Cooldown — once per project per day
# ================================================================
TODAY=$(date +%Y-%m-%d)

ALL_COOLED=true
for proj in $(echo "$NEEDS" | tr ',' ' '); do
    COOLDOWN_FILE="$COOLDOWN_DIR/${proj}_${TODAY}"
    if [ ! -f "$COOLDOWN_FILE" ]; then
        ALL_COOLED=false
        break
    fi
done

if [ "$ALL_COOLED" = true ]; then
    exit 0
fi

echo "[cortex bootstrap] Projects needing agents: $NEEDS"

# ================================================================
# Phase 3: Collect memories + existing agents
# ================================================================
MEMORIES=$("$LIB/collect_memories_full.py" "$NEEDS" 2>/dev/null)
EXISTING_AGENTS=$("$LIB/collect_agents.py" 2>/dev/null)

EXISTING_NAMES=$(echo "$EXISTING_AGENTS" | /usr/bin/python3 -c "
import sys, json
try:
    agents = json.loads(sys.stdin.read())
    for a in agents:
        desc = ''
        for line in a['content'].split('\n'):
            if line.strip().startswith('description:'):
                desc = line.split(':', 1)[1].strip()
                break
        print(f'  - {a[\"name\"]} ({a[\"scope\"]}): {desc}')
except: pass
" 2>/dev/null)

if [ -z "$MEMORIES" ]; then
    echo "[cortex bootstrap] No memories to work with"
    exit 0
fi

# ================================================================
# Phase 4: SINGLE call to haiku — propose agents
# ================================================================
CREATE_RESULT=$(cat <<PROMPT_EOF | claude -p --model haiku --mcp-config '{}' --strict-mcp-config 2>/dev/null
You are an agent architect for Claude Code. Analyze these accumulated vector memories and propose EXACTLY the right number of reusable subagents.

=== VMEM MEMORIES ===
$MEMORIES

=== EXISTING AGENTS ===
${EXISTING_NAMES:-  (none)}

=== CONTEXT ===
Working directory: $CWD
Projects needing agents: $NEEDS

## CRITICAL RULES

1. Output a JSON array. NO markdown wrapping, NO explanation — JUST the JSON array.
2. Maximum 3 agents TOTAL across all projects and global combined.
3. Each agent must cover a DISTINCT domain — no overlapping scope.
4. Do NOT create an agent if an existing one already covers that domain.
5. If no agents are needed, return: []

## Two Scopes

- "project": specific to this project's workflows (goes in .claude/agents/)
- "user": cross-project patterns useful everywhere (goes in ~/.claude/agents/)

## Agent Quality Requirements

- description: SPECIFIC trigger conditions ("Use when..." or "Use for...")
- System prompt: include ACTUAL knowledge from memories — paths, commands, gotchas, patterns
- tools: [Read, Edit, Write, Bash, Grep, Glob] for most agents
- model: opus (always)

## Output Format

[{"scope": "project", "filename": "kebab-case.md", "content": "---\nname: agent-name\ndescription: When to use this\ntools:\n  - Read\n  - Edit\n  - Write\n  - Bash\n  - Grep\n  - Glob\nmodel: opus\n---\n\nSystem prompt here with real knowledge..."}]
PROMPT_EOF
)

if [ -z "$CREATE_RESULT" ]; then
    echo "[cortex bootstrap] No agent proposals generated"
    # Still set cooldown to avoid re-trying a failing call
    for proj in $(echo "$NEEDS" | tr ',' ' '); do
        touch "$COOLDOWN_DIR/${proj}_${TODAY}"
    done
    exit 0
fi

# ================================================================
# Phase 5: Create via fleet_create.py (semantic dedup at 0.55)
# ================================================================
CREATED=$("$LIB/fleet_create.py" "$CREATE_RESULT" "$CWD" 2>/dev/null)

if [ "$CREATED" -gt 0 ] 2>/dev/null; then
    echo "[cortex bootstrap] Created $CREATED new agent(s)"

    # Update activity indicator
    ACTIVITY_FILE="$HOME/.claude/.cortex_activity"
    EXISTING_ACTIVITY=""
    if [ -f "$ACTIVITY_FILE" ]; then
        EXISTING_ACTIVITY=$(cat "$ACTIVITY_FILE" 2>/dev/null)
    fi
    if [ -n "$EXISTING_ACTIVITY" ]; then
        echo "$EXISTING_ACTIVITY | bootstrap: +$CREATED agents" > "$ACTIVITY_FILE"
    else
        echo "bootstrap: +$CREATED agents" > "$ACTIVITY_FILE"
    fi
else
    echo "[cortex bootstrap] No new agents created (existing coverage sufficient)"
fi

# Set cooldown for all projects
for proj in $(echo "$NEEDS" | tr ',' ' '); do
    touch "$COOLDOWN_DIR/${proj}_${TODAY}"
done

# Clean old cooldown files (>7 days)
find "$COOLDOWN_DIR" -name "*_20*" -mtime +7 -delete 2>/dev/null

exit 0