|
| 1 | +#!/bin/bash |
| 2 | + |
| 3 | +# Auto-Learning Utility |
| 4 | +# Adds keywords to a skill's configuration for improved auto-activation |
| 5 | + |
| 6 | +set -euo pipefail |
| 7 | + |
| 8 | +# Usage check |
| 9 | +if [ $# -lt 2 ]; then |
| 10 | + echo "Usage: $0 <skill-name> <keyword1> [keyword2] [keyword3] ..." |
| 11 | + echo "" |
| 12 | + echo "Example:" |
| 13 | + echo " $0 localization-developer \"membership\" \"tiers\" \"settings\"" |
| 14 | + exit 1 |
| 15 | +fi |
| 16 | + |
| 17 | +SKILL_NAME=$1 |
| 18 | +shift |
| 19 | +NEW_KEYWORDS=("$@") |
| 20 | + |
| 21 | +# Get script directory |
| 22 | +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
| 23 | +SKILL_RULES="$SCRIPT_DIR/../skill-rules.json" |
| 24 | +LOG_DIR="$SCRIPT_DIR/../../logs" |
| 25 | +LOG_FILE="$LOG_DIR/skill-learning.log" |
| 26 | + |
| 27 | +# Ensure log directory exists |
| 28 | +mkdir -p "$LOG_DIR" |
| 29 | + |
| 30 | +# Validate skill exists |
| 31 | +if ! jq -e ".skills[\"$SKILL_NAME\"]" "$SKILL_RULES" > /dev/null 2>&1; then |
| 32 | + echo "❌ Error: Skill '$SKILL_NAME' not found in skill-rules.json" |
| 33 | + echo "" |
| 34 | + echo "Available skills:" |
| 35 | + jq -r '.skills | keys[]' "$SKILL_RULES" | sed 's/^/ - /' |
| 36 | + exit 1 |
| 37 | +fi |
| 38 | + |
| 39 | +echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━" |
| 40 | +echo "📚 AUTO-LEARNING: Adding keywords to $SKILL_NAME" |
| 41 | +echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━" |
| 42 | +echo "" |
| 43 | + |
| 44 | +# Get existing keywords |
| 45 | +EXISTING_KEYWORDS=$(jq -r ".skills[\"$SKILL_NAME\"].promptTriggers.keywords[]" "$SKILL_RULES" 2>/dev/null || echo "") |
| 46 | + |
| 47 | +# Check which keywords are new |
| 48 | +TRULY_NEW_KEYWORDS=() |
| 49 | +for keyword in "${NEW_KEYWORDS[@]}"; do |
| 50 | + keyword_lower=$(echo "$keyword" | tr '[:upper:]' '[:lower:]') |
| 51 | + |
| 52 | + IS_DUPLICATE=false |
| 53 | + while IFS= read -r existing; do |
| 54 | + existing_lower=$(echo "$existing" | tr '[:upper:]' '[:lower:]') |
| 55 | + if [ "$keyword_lower" = "$existing_lower" ]; then |
| 56 | + IS_DUPLICATE=true |
| 57 | + echo "⏭️ Skipping '$keyword' (already exists)" |
| 58 | + break |
| 59 | + fi |
| 60 | + done <<< "$EXISTING_KEYWORDS" |
| 61 | + |
| 62 | + if [ "$IS_DUPLICATE" = false ]; then |
| 63 | + TRULY_NEW_KEYWORDS+=("$keyword") |
| 64 | + echo "✅ Adding '$keyword'" |
| 65 | + fi |
| 66 | +done |
| 67 | + |
| 68 | +# Exit if no new keywords |
| 69 | +if [ ${#TRULY_NEW_KEYWORDS[@]} -eq 0 ]; then |
| 70 | + echo "" |
| 71 | + echo "ℹ️ No new keywords to add - all provided keywords already exist" |
| 72 | + exit 0 |
| 73 | +fi |
| 74 | + |
| 75 | +echo "" |
| 76 | +echo "Updating skill-rules.json..." |
| 77 | + |
| 78 | +# Create backup |
| 79 | +cp "$SKILL_RULES" "$SKILL_RULES.backup" |
| 80 | + |
| 81 | +# Build jq update command |
| 82 | +JQ_FILTER=".skills[\"$SKILL_NAME\"].promptTriggers.keywords += [" |
| 83 | +for i in "${!TRULY_NEW_KEYWORDS[@]}"; do |
| 84 | + if [ $i -gt 0 ]; then |
| 85 | + JQ_FILTER+=", " |
| 86 | + fi |
| 87 | + JQ_FILTER+="\"${TRULY_NEW_KEYWORDS[$i]}\"" |
| 88 | +done |
| 89 | +JQ_FILTER+="] | .skills[\"$SKILL_NAME\"].promptTriggers.keywords |= unique" |
| 90 | + |
| 91 | +# Update skill-rules.json |
| 92 | +jq "$JQ_FILTER" "$SKILL_RULES" > "$SKILL_RULES.tmp" |
| 93 | + |
| 94 | +# Validate JSON |
| 95 | +if jq empty "$SKILL_RULES.tmp" 2>/dev/null; then |
| 96 | + mv "$SKILL_RULES.tmp" "$SKILL_RULES" |
| 97 | + echo "✅ Updated skill-rules.json" |
| 98 | + |
| 99 | + # Remove backup |
| 100 | + rm "$SKILL_RULES.backup" |
| 101 | + |
| 102 | + # Log the update |
| 103 | + echo "[$(date '+%Y-%m-%d %H:%M:%S')] Added keywords to $SKILL_NAME" >> "$LOG_FILE" |
| 104 | + for keyword in "${TRULY_NEW_KEYWORDS[@]}"; do |
| 105 | + echo " + $keyword" >> "$LOG_FILE" |
| 106 | + done |
| 107 | + echo "" >> "$LOG_FILE" |
| 108 | + |
| 109 | + echo "" |
| 110 | + echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━" |
| 111 | + echo "✨ Success! Added ${#TRULY_NEW_KEYWORDS[@]} new keyword(s)" |
| 112 | + echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━" |
| 113 | + echo "" |
| 114 | + echo "💡 The system will now auto-activate '$SKILL_NAME' for prompts" |
| 115 | + echo " containing these keywords." |
| 116 | + echo "" |
| 117 | +else |
| 118 | + echo "❌ Error: Generated invalid JSON, restoring backup" |
| 119 | + mv "$SKILL_RULES.backup" "$SKILL_RULES" |
| 120 | + rm -f "$SKILL_RULES.tmp" |
| 121 | + exit 1 |
| 122 | +fi |
0 commit comments