// ai/context/budget.go
if hasRetrieval {
    budget.ShortTermMemory = int(float64(remaining) * 0.40)
    budget.LongTermMemory = int(float64(remaining) * 0.15)
    budget.Retrieval = int(float64(remaining) * 0.45)
}

type AdaptiveBudgetAllocator struct {
    profiles map[Intent]*BudgetProfile
}

type BudgetProfile struct {
    ShortTermRatio float64
    LongTermRatio  float64
    RetrievalRatio float64
    UserPrefsRatio float64
}

var defaultProfiles = map[Intent]*BudgetProfile{
    IntentQuery:  {0.25, 0.10, 0.55, 0.10}, // 查询: 检索优先
    IntentCreate: {0.50, 0.20, 0.20, 0.10}, // 创建: 上下文优先
    IntentUpdate: {0.45, 0.15, 0.30, 0.10}, // 更新: 平衡
    IntentChat:   {0.60, 0.25, 0.05, 0.10}, // 闲聊: 最小化检索
    IntentDelete: {0.40, 0.10, 0.40, 0.10}, // 删除: 确认上下文
}

func (a *AdaptiveBudgetAllocator) Allocate(total int, intent Intent) *TokenBudget {
    profile := a.profiles[intent]
    if profile == nil {
        profile = a.profiles[IntentChat] // fallback
    }
    
    remaining := total - DefaultSystemPrompt
    return &TokenBudget{
        Total:           total,
        SystemPrompt:    DefaultSystemPrompt,
        ShortTermMemory: int(float64(remaining) * profile.ShortTermRatio),
        LongTermMemory:  int(float64(remaining) * profile.LongTermRatio),
        Retrieval:       int(float64(remaining) * profile.RetrievalRatio),
        UserPrefs:       int(float64(remaining) * profile.UserPrefsRatio),
    }
}