dual-brain 0.2.30 → 0.3.1

package/src/subscription.mjs

@@ -1,212 +0,0 @@
-// subscription.mjs — Subscription-aware routing defaults
-import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
-import { join } from 'node:path';
-
-export const SUBSCRIPTIONS = {
-  // Claude subscriptions
-  'claude-pro': {
-    label: 'Claude Pro ($20/mo)',
-    provider: 'claude',
-    tokenBudget: 'moderate', // 5-hr rolling window, weekly cap
-    recommendedProfile: 'balanced',
-    modelWeights: { haiku: 0.4, sonnet: 0.5, opus: 0.1 },
-    notes: 'One extended Opus session can use 20% of your allocation. Prefer sonnet for routine work.',
-  },
-  'claude-max-5x': {
-    label: 'Claude Max 5x ($100/mo)',
-    provider: 'claude',
-    tokenBudget: 'generous',
-    recommendedProfile: 'quality-first',
-    modelWeights: { haiku: 0.2, sonnet: 0.5, opus: 0.3 },
-    notes: '5x Pro capacity. Opus is available for complex/creative work without worry.',
-  },
-  'claude-max-20x': {
-    label: 'Claude Max 20x ($200/mo)',
-    provider: 'claude',
-    tokenBudget: 'unlimited',
-    recommendedProfile: 'quality-first',
-    modelWeights: { haiku: 0.1, sonnet: 0.4, opus: 0.5 },
-    notes: 'Effectively unlimited. Use the best model for every task.',
-  },
-  'claude-team': {
-    label: 'Claude Team ($30/seat/mo)',
-    provider: 'claude',
-    tokenBudget: 'moderate',
-    recommendedProfile: 'balanced',
-    modelWeights: { haiku: 0.3, sonnet: 0.5, opus: 0.2 },
-    notes: 'Team tier with admin controls. Collaboration triggers recommended.',
-  },
-  // ChatGPT subscriptions
-  'chatgpt-plus': {
-    label: 'ChatGPT Plus ($20/mo)',
-    provider: 'openai',
-    tokenBudget: 'limited', // 50 o3/day on Plus
-    recommendedProfile: 'cost-saver',
-    modelWeights: { 'o4-mini': 0.6, 'gpt-4.1': 0.3, 'o3': 0.1 },
-    notes: '50 o3 messages/day limit. Heavy on o4-mini for routine, save o3 for critical decisions.',
-  },
-  'chatgpt-pro': {
-    label: 'ChatGPT Pro ($200/mo)',
-    provider: 'openai',
-    tokenBudget: 'generous',
-    recommendedProfile: 'quality-first',
-    modelWeights: { 'o4-mini': 0.3, 'gpt-4.1': 0.4, 'o3': 0.3 },
-    notes: 'Unlimited access to all models. Use o3 freely for complex reasoning.',
-  },
-  // Dual subscription (both providers)
-  'dual-pro': {
-    label: 'Both Pro tiers',
-    provider: 'both',
-    tokenBudget: 'moderate',
-    recommendedProfile: 'balanced',
-    modelWeights: { haiku: 0.2, sonnet: 0.3, 'gpt-4.1': 0.3, 'o4-mini': 0.2 },
-    notes: 'Split load across providers. Route by model strength: Claude for code, GPT for reasoning.',
-  },
-  'dual-max': {
-    label: 'Max + Pro (or both Max)',
-    provider: 'both',
-    tokenBudget: 'unlimited',
-    recommendedProfile: 'quality-first',
-    modelWeights: { sonnet: 0.3, opus: 0.2, 'gpt-4.1': 0.2, 'o3': 0.3 },
-    notes: 'Full power from both providers. Route by task fit, not by cost.',
-  },
-};
-
-const DEFAULT_WEIGHTS = {
-  modelWeights: { haiku: 0.3, sonnet: 0.5, opus: 0.2 },
-  profile: 'balanced',
-  notes: 'No subscription configured. Using balanced defaults.',
-};
-
-function subFile(cwd) {
-  return join(cwd || process.cwd(), '.dualbrain', 'subscription.json');
-}
-
-/** Returns the subscription config object or null. */
-export function getSubscription(subType) {
-  return SUBSCRIPTIONS[subType] ?? null;
-}
-
-/** Returns { modelWeights, profile, notes } for the subscription. Falls back to balanced defaults. */
-export function getRecommendedWeights(subType) {
-  const sub = SUBSCRIPTIONS[subType];
-  if (!sub) return DEFAULT_WEIGHTS;
-  return {
-    modelWeights: sub.modelWeights,
-    profile: sub.recommendedProfile,
-    notes: sub.notes,
-  };
-}
-
-/** Writes { subscription, configuredAt } to .dualbrain/subscription.json. */
-export function saveUserSubscription(subType, cwd) {
-  const dir = join(cwd || process.cwd(), '.dualbrain');
-  if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
-  writeFileSync(
-    subFile(cwd),
-    JSON.stringify({ subscription: subType, configuredAt: new Date().toISOString() }, null, 2),
-    'utf8'
-  );
-}
-
-/** Reads the saved subscription. Returns subType string or null. */
-export function loadUserSubscription(cwd) {
-  try {
-    const p = subFile(cwd);
-    if (!existsSync(p)) return null;
-    const data = JSON.parse(readFileSync(p, 'utf8'));
-    return data.subscription ?? null;
-  } catch {
-    return null;
-  }
-}
-
-/**
- * Generates a text recommendation based on subscription + current routing stats.
- * routingStats: return value of getRoutingStats() from routing-advisor.mjs
- */
-export function generateRecommendation(subType, routingStats) {
-  const sub = SUBSCRIPTIONS[subType];
-  if (!sub) return 'No subscription configured. Run `dual-brain subscription set <type>` to enable smart routing defaults.';
-
-  // Tally actual model usage from routing stats cells
-  const actualUsage = {};
-  let totalObs = 0;
-  for (const models of Object.values(routingStats?.cells ?? {})) {
-    for (const [model, entry] of Object.entries(models)) {
-      actualUsage[model] = (actualUsage[model] ?? 0) + (entry.observations ?? 0);
-      totalObs += entry.observations ?? 0;
-    }
-  }
-
-  if (totalObs === 0) {
-    return `You're on ${sub.label}. No routing history yet — recommended profile is ${sub.recommendedProfile}. ${sub.notes}`;
-  }
-
-  // Compute actual share per model
-  const actualShare = {};
-  for (const [model, count] of Object.entries(actualUsage)) {
-    actualShare[model] = count / totalObs;
-  }
-
-  const rec = sub.modelWeights;
-  const budget = sub.tokenBudget;
-  const lines = [];
-
-  // Check expensive model utilization vs. recommended
-  const expensiveModels = ['opus', 'o3'];
-  for (const model of expensiveModels) {
-    const recW = rec[model] ?? 0;
-    const actW = actualShare[model] ?? 0;
-
-    if (recW > 0 && budget === 'unlimited' && actW < recW * 0.5) {
-      lines.push(
-        `You're on ${sub.label} but only using ${model} ${Math.round(actW * 100)}% of the time` +
-        ` (recommended: ${Math.round(recW * 100)}%). You're paying for capacity you're not using.` +
-        ` Consider switching to ${sub.recommendedProfile} mode.`
-      );
-    } else if (recW < 0.2 && budget === 'limited' && actW > recW * 2 && actW > 0.1) {
-      lines.push(
-        `Your ${sub.label} subscription is token-limited. ${model} usage at ${Math.round(actW * 100)}%` +
-        ` may exhaust daily limits — recommended cap is ~${Math.round(recW * 100)}%.`
-      );
-    }
-  }
-
-  // Cheap model suggestions on budget-constrained plans
-  const cheapModels = ['haiku', 'o4-mini'];
-  if (budget === 'moderate' || budget === 'limited') {
-    for (const model of cheapModels) {
-      const recW = rec[model] ?? 0;
-      const actW = actualShare[model] ?? 0;
-      if (recW > 0 && actW < recW * 0.5) {
-        lines.push(
-          `Your ${sub.label} subscription has a ${budget} budget. Increasing ${model} usage` +
-          ` (currently ${Math.round(actW * 100)}%, recommended ${Math.round(recW * 100)}%)` +
-          ` for search and routine tasks would preserve your allocation.`
-        );
-        break; // one cheap-model tip is enough
-      }
-    }
-  }
-
-  // Dominant model confirmation — find the most-used model
-  const topModel = Object.entries(actualShare).sort((a, b) => b[1] - a[1])[0];
-  if (lines.length === 0 && topModel) {
-    lines.push(
-      `Your ${sub.label} subscription is well-matched. ${Math.round(topModel[1] * 100)}% of dispatches` +
-      ` use ${topModel[0]} — a good fit for your ${budget} budget. ${sub.notes}`
-    );
-  }
-
-  return lines.slice(0, 3).join(' ');
-}
-
-/** Returns array of { key, label, provider } for display in UX. */
-export function listSubscriptions() {
-  return Object.entries(SUBSCRIPTIONS).map(([key, sub]) => ({
-    key,
-    label: sub.label,
-    provider: sub.provider,
-  }));
-}

package/src/templates.mjs

@@ -1,260 +0,0 @@
-import { existsSync, readFileSync } from 'node:fs';
-import { join } from 'node:path';
-
-/**
- * Task contract — every dispatch must have one.
- * @typedef {{
- *   id: string,
- *   objective: string,
- *   scope: string[],
- *   nonGoals?: string[],
- *   risk: 'low'|'medium'|'high'|'critical',
- *   acceptanceCriteria: string[],
- *   allowedOperations?: string[],
- *   context?: string,
- *   files?: string[],
- *   timeoutMs?: number,
- * }} TaskContract
- */
-
-/**
- * Validate a task contract has all required fields.
- * Returns { valid, missing }
- */
-export function validateContract(contract) {
-  const required = ['objective', 'scope', 'risk', 'acceptanceCriteria'];
-  const missing = required.filter(f => !contract?.[f] || (Array.isArray(contract[f]) && contract[f].length === 0));
-  return {
-    valid: missing.length === 0,
-    missing,
-    contract: missing.length === 0 ? { ...contract, id: contract.id || Date.now().toString(36) } : null,
-  };
-}
-
-// ── Template definitions ─────────────────────────────────────────────────────
-
-const TEMPLATES = {
-  search: {
-    id: 'search',
-    version: '1.0',
-    tier: 'search',
-    description: 'Read-only lookups, grep, explore. Returns files found, line refs, confidence.',
-    requiredFields: ['objective', 'scope'],
-    render(contract, context = {}) {
-      const lines = [];
-      lines.push(`Find: ${contract.objective}`);
-      lines.push('');
-      if (contract.scope.length) lines.push(`Scope: ${contract.scope.join(', ')}`);
-      if (contract.files?.length) lines.push(`Start with: ${contract.files.join(', ')}`);
-      if (contract.context) lines.push(`Context: ${contract.context}`);
-      lines.push('');
-      lines.push('Return: file paths, line numbers, relevant code snippets, and confidence level.');
-      if (contract.nonGoals?.length) lines.push(`Do NOT: ${contract.nonGoals.join('; ')}`);
-      return lines.join('\n');
-    },
-  },
-
-  execute: {
-    id: 'execute',
-    version: '1.0',
-    tier: 'execute',
-    description: 'Edits, tests, git ops. Returns files changed, tests run, edge cases.',
-    requiredFields: ['objective', 'scope', 'acceptanceCriteria'],
-    render(contract, context = {}) {
-      const lines = [];
-      lines.push(contract.objective);
-      lines.push('');
-      if (contract.scope.length) lines.push(`Files in scope: ${contract.scope.join(', ')}`);
-      if (contract.files?.length) lines.push(`Read first: ${contract.files.join(', ')}`);
-      if (contract.context) lines.push(`Context: ${contract.context}`);
-      lines.push('');
-      lines.push('Acceptance criteria:');
-      for (const c of contract.acceptanceCriteria) {
-        lines.push(`- ${c}`);
-      }
-      if (contract.nonGoals?.length) {
-        lines.push('');
-        lines.push('Non-goals (do NOT do these):');
-        for (const ng of contract.nonGoals) lines.push(`- ${ng}`);
-      }
-      if (contract.allowedOperations?.length) {
-        lines.push('');
-        lines.push(`Allowed operations: ${contract.allowedOperations.join(', ')}`);
-      }
-      lines.push('');
-      lines.push('Return: files changed, tests run, edge cases found.');
-      return lines.join('\n');
-    },
-  },
-
-  think: {
-    id: 'think',
-    version: '1.0',
-    tier: 'think',
-    description: 'Architecture decisions, design review, planning.',
-    requiredFields: ['objective'],
-    render(contract, context = {}) {
-      const lines = [];
-      lines.push(contract.objective);
-      lines.push('');
-      if (contract.scope?.length) lines.push(`Relevant modules: ${contract.scope.join(', ')}`);
-      if (contract.context) lines.push(`Background: ${contract.context}`);
-      if (contract.files?.length) lines.push(`Key files: ${contract.files.join(', ')}`);
-      lines.push('');
-      lines.push('Provide: recommendation, rationale, alternatives considered, risks, and confidence level.');
-      if (contract.acceptanceCriteria?.length) {
-        lines.push('');
-        lines.push('Decision criteria:');
-        for (const c of contract.acceptanceCriteria) lines.push(`- ${c}`);
-      }
-      return lines.join('\n');
-    },
-  },
-
-  review: {
-    id: 'review',
-    version: '1.0',
-    tier: 'review',
-    description: 'Code review with severity, line refs, test gaps, security concerns.',
-    requiredFields: ['objective', 'scope'],
-    render(contract, context = {}) {
-      const lines = [];
-      lines.push(`Review: ${contract.objective}`);
-      lines.push('');
-      if (contract.scope.length) lines.push(`Files to review: ${contract.scope.join(', ')}`);
-      if (contract.context) lines.push(`Context: ${contract.context}`);
-      lines.push('');
-      lines.push('Check for:');
-      lines.push('- Correctness and edge cases');
-      lines.push('- Security vulnerabilities (OWASP top 10)');
-      lines.push('- Test coverage gaps');
-      lines.push('- Architectural drift');
-      lines.push('- Performance concerns');
-      if (contract.acceptanceCriteria?.length) {
-        lines.push('');
-        lines.push('Specific concerns:');
-        for (const c of contract.acceptanceCriteria) lines.push(`- ${c}`);
-      }
-      lines.push('');
-      lines.push('Return: findings with severity (critical/high/medium/low), file:line refs, and suggested fixes.');
-      return lines.join('\n');
-    },
-  },
-};
-
-// ── Output schemas ───────────────────────────────────────────────────────────
-
-const OUTPUT_SCHEMAS = {
-  think: '{ "decision": "string", "confidence": 0.0-1.0, "reasoning": "string", "workSpec": { "objective": "string", "files": ["path"], "criteria": ["string"] } }',
-  execute: '{ "filesChanged": ["path"], "testsRun": boolean, "issues": ["string"] }',
-  review: '{ "pass": boolean, "findings": [{ "severity": "critical|high|medium|low", "file": "path", "line": number, "issue": "string", "fix": "string" }] }',
-  search: '{ "found": [{ "file": "path", "line": number, "snippet": "string" }], "confidence": 0.0-1.0 }',
-};
-
-// ── Model render hints ────────────────────────────────────────────────────────
-
-const MODEL_RENDER_HINTS = {
-  xml: ['claude', 'sonnet', 'haiku', 'opus'],
-  markdown: ['gpt', 'gpt-4', 'gpt-4.1', 'gpt-4o'],
-  prose: ['o3', 'o4-mini'],
-};
-
-// ── Template API ─────────────────────────────────────────────────────────────
-
-/**
- * Get the structured output schema for a tier.
- */
-export function getOutputSchema(tier) {
-  return OUTPUT_SCHEMAS[tier] || null;
-}
-
-/**
- * Get the preferred prompt rendering format for a given model ID.
- */
-export function getRenderHint(modelId) {
-  if (!modelId) return 'markdown';
-  const normalized = String(modelId).toLowerCase();
-  for (const [format, patterns] of Object.entries(MODEL_RENDER_HINTS)) {
-    if (patterns.some(p => normalized.includes(p))) return format;
-  }
-  return 'markdown';
-}
-
-/**
- * Get a template by tier name.
- */
-export function getTemplate(tier) {
-  return TEMPLATES[tier] || null;
-}
-
-/**
- * List all available templates.
- */
-export function listTemplates() {
-  return Object.values(TEMPLATES).map(t => ({
-    id: t.id,
-    version: t.version,
-    tier: t.tier,
-    description: t.description,
-    requiredFields: t.requiredFields,
-  }));
-}
-
-/**
- * Render a prompt from a template and task contract.
- * Validates contract first. Returns { prompt, template, contract, valid, errors }
- */
-export function renderPrompt(tier, contract, context = {}) {
-  const template = TEMPLATES[tier];
-  if (!template) {
-    return { prompt: null, valid: false, errors: [`Unknown template tier: ${tier}`] };
-  }
-
-  // Validate required fields
-  const missing = template.requiredFields.filter(f => !contract?.[f] || (Array.isArray(contract[f]) && contract[f].length === 0));
-  if (missing.length > 0) {
-    return {
-      prompt: null,
-      valid: false,
-      errors: missing.map(f => `Missing required field: ${f}`),
-      template: { id: template.id, version: template.version },
-    };
-  }
-
-  const prompt = template.render(contract, context);
-
-  return {
-    prompt,
-    valid: true,
-    errors: [],
-    template: { id: template.id, version: template.version },
-    contract: { ...contract, id: contract.id || Date.now().toString(36) },
-    outputSchema: OUTPUT_SCHEMAS[tier] || null,
-    stats: {
-      words: prompt.split(/\s+/).length,
-      chars: prompt.length,
-      estimatedTokens: Math.ceil(prompt.length / 4),
-    },
-  };
-}
-
-/**
- * Quick render: build a contract from minimal inputs and render.
- * For when HEAD knows the tier and objective but hasn't built a full contract.
- */
-export function quickRender(tier, objective, opts = {}) {
-  const { scope = [], files = [], risk = 'medium', criteria = [], nonGoals = [], context = '' } = opts;
-
-  const contract = {
-    objective,
-    scope,
-    files,
-    risk,
-    acceptanceCriteria: criteria.length ? criteria : [`${objective} is complete and working`],
-    nonGoals,
-    context,
-    allowedOperations: tier === 'search' ? ['read'] : tier === 'execute' ? ['read', 'write', 'test'] : ['read', 'analyze'],
-  };
-
-  return renderPrompt(tier, contract);
-}