1bc0e60235
Some checks failed
Deploy to Staging / Build Images (pull_request) Has been cancelled
Deploy to Staging / Deploy to Staging (pull_request) Has been cancelled
Deploy to Staging / Verify Staging (pull_request) Has been cancelled
Deploy to Staging / Notify Staging Ready (pull_request) Has been cancelled
Deploy to Staging / Notify Staging Failure (pull_request) Has been cancelled
🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
59 lines
1.5 KiB
Bash
Executable File
59 lines
1.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Enforces correct model usage for Task tool based on agent definitions
|
|
# Blocks Task calls that don't specify the correct model for the subagent_type
|
|
|
|
# Read tool input from stdin
|
|
INPUT=$(cat)
|
|
|
|
# Extract subagent_type and model from the input
|
|
SUBAGENT_TYPE=$(echo "$INPUT" | jq -r '.subagent_type // empty')
|
|
MODEL=$(echo "$INPUT" | jq -r '.model // empty')
|
|
|
|
# If no subagent_type, allow (not an agent call)
|
|
if [[ -z "$SUBAGENT_TYPE" ]]; then
|
|
exit 0
|
|
fi
|
|
|
|
# Get expected model for agent type
|
|
# Most agents use sonnet, quality-reviewer uses opus
|
|
get_expected_model() {
|
|
case "$1" in
|
|
# Custom project agents
|
|
feature-agent|first-frontend-agent|platform-agent|quality-agent)
|
|
echo "sonnet"
|
|
;;
|
|
# Role agents
|
|
developer|technical-writer|debugger)
|
|
echo "sonnet"
|
|
;;
|
|
quality-reviewer)
|
|
echo "opus"
|
|
;;
|
|
# Built-in agents - default to sonnet for cost efficiency
|
|
Explore|Plan|Bash|general-purpose)
|
|
echo "sonnet"
|
|
;;
|
|
*)
|
|
# Unknown agent, no enforcement
|
|
echo ""
|
|
;;
|
|
esac
|
|
}
|
|
|
|
EXPECTED_MODEL=$(get_expected_model "$SUBAGENT_TYPE")
|
|
|
|
# If agent not in mapping, allow (unknown agent type)
|
|
if [[ -z "$EXPECTED_MODEL" ]]; then
|
|
exit 0
|
|
fi
|
|
|
|
# Check if model matches expected
|
|
if [[ "$MODEL" != "$EXPECTED_MODEL" ]]; then
|
|
echo "BLOCKED: Agent '$SUBAGENT_TYPE' requires model: '$EXPECTED_MODEL' but got '${MODEL:-<not specified>}'."
|
|
echo "Retry with: model: \"$EXPECTED_MODEL\""
|
|
exit 1
|
|
fi
|
|
|
|
# Model matches, allow the call
|
|
exit 0
|