gencode-ai 0.1.3 → 0.3.0

package/src/planning/types.ts

@@ -0,0 +1,150 @@
+/**
+ * Plan Mode Type Definitions
+ *
+ * Types for Plan Mode - a read-only exploration mode that allows
+ * the LLM to plan implementations before executing changes.
+ */
+
+// ============================================================================
+// Plan Mode State Types
+// ============================================================================
+
+/**
+ * Plan mode phases following Claude Code's workflow
+ */
+export type PlanPhase =
+  | 'understanding' // Initial exploration
+  | 'design' // Planning approach
+  | 'review' // Clarifying with user
+  | 'final' // Writing plan file
+  | 'approval'; // Waiting for user approval
+
+/**
+ * User approval options for plan mode
+ */
+export type PlanApprovalOption =
+  | 'approve' // Accept plan and execute with auto-accept edits
+  | 'approve_clear' // Accept plan, clear context, auto-accept edits
+  | 'approve_manual' // Accept plan but manually approve each edit
+  | 'modify' // Go back to modify the plan
+  | 'cancel'; // Cancel plan mode entirely
+
+/**
+ * Pre-approved permission request (Claude Code ExitPlanMode style)
+ */
+export interface AllowedPrompt {
+  tool: 'Bash';
+  prompt: string; // Semantic description, e.g., "run tests", "install dependencies"
+}
+
+/**
+ * Plan mode state
+ */
+export interface PlanModeState {
+  /** Whether plan mode is currently active */
+  active: boolean;
+
+  /** Current phase of planning */
+  phase: PlanPhase;
+
+  /** Path to the plan file */
+  planFilePath: string | null;
+
+  /** User's original request that triggered plan mode */
+  originalRequest: string | null;
+
+  /** Requested permissions for execution phase */
+  requestedPermissions: AllowedPrompt[];
+
+  /** Timestamp when plan mode was entered */
+  enteredAt: Date | null;
+}
+
+/**
+ * Plan file structure
+ */
+export interface PlanFile {
+  /** Plan file path */
+  path: string;
+
+  /** Plan content in markdown */
+  content: string;
+
+  /** Creation timestamp */
+  createdAt: Date;
+
+  /** Last update timestamp */
+  updatedAt: Date;
+}
+
+// ============================================================================
+// Tool Filtering Types
+// ============================================================================
+
+/**
+ * Tools allowed in plan mode (read-only + planning tools)
+ */
+export const PLAN_MODE_ALLOWED_TOOLS = [
+  'Read',
+  'Glob',
+  'Grep',
+  'WebFetch',
+  'WebSearch',
+  'TodoWrite',
+  'AskUserQuestion',
+  'EnterPlanMode', // Can re-enter if needed
+  'ExitPlanMode', // To exit plan mode
+] as const;
+
+/**
+ * Tools blocked in plan mode (write/execute operations)
+ */
+export const PLAN_MODE_BLOCKED_TOOLS = ['Write', 'Edit', 'Bash'] as const;
+
+export type PlanModeAllowedTool = (typeof PLAN_MODE_ALLOWED_TOOLS)[number];
+export type PlanModeBlockedTool = (typeof PLAN_MODE_BLOCKED_TOOLS)[number];
+
+// ============================================================================
+// UI Types
+// ============================================================================
+
+/**
+ * Operating mode (cycle with Shift+Tab: normal → plan → accept → normal)
+ * - normal: Default mode, edits require confirmation
+ * - plan: Read-only exploration mode, edits blocked
+ * - accept: Auto-accept mode, edits approved without confirmation
+ */
+export type ModeType = 'normal' | 'plan' | 'accept';
+
+/**
+ * Plan approval UI state
+ */
+export interface PlanApprovalState {
+  /** Plan content for display */
+  planContent: string;
+
+  /** Summary of files to change */
+  filesToChange: Array<{
+    path: string;
+    action: 'create' | 'modify' | 'delete';
+  }>;
+
+  /** Requested permissions */
+  requestedPermissions: AllowedPrompt[];
+
+  /** Callback when user makes decision */
+  onDecision: (option: PlanApprovalOption) => void;
+}
+
+// ============================================================================
+// Event Types
+// ============================================================================
+
+/**
+ * Plan mode events for UI updates
+ */
+export type PlanModeEvent =
+  | { type: 'enter'; planFilePath: string }
+  | { type: 'phase_change'; phase: PlanPhase }
+  | { type: 'exit'; approved: boolean }
+  | { type: 'toggle' }; // Shift+Tab toggle

package/src/pricing/calculator.ts

@@ -0,0 +1,71 @@
+/**
+ * Cost calculation utilities
+ */
+
+import { getModelPricing } from './models.js';
+import { CostEstimate, TokenUsage } from './types.js';
+
+/**
+ * Calculate cost from token usage
+ */
+export function calculateCost(
+  provider: string,
+  model: string,
+  tokens: TokenUsage
+): CostEstimate {
+  const pricing = getModelPricing(provider, model);
+
+  // If no pricing found, return zero cost
+  if (!pricing) {
+    return {
+      inputCost: 0,
+      outputCost: 0,
+      totalCost: 0,
+      currency: 'USD',
+    };
+  }
+
+  // Calculate cost per token type
+  const inputCost = (tokens.inputTokens / 1_000_000) * pricing.inputPer1M;
+  const outputCost = (tokens.outputTokens / 1_000_000) * pricing.outputPer1M;
+
+  return {
+    inputCost,
+    outputCost,
+    totalCost: inputCost + outputCost,
+    currency: 'USD',
+  };
+}
+
+/**
+ * Format cost for display
+ */
+export function formatCost(cost: number): string {
+  if (cost === 0) {
+    return '$0.00';
+  }
+  if (cost < 0.01) {
+    return '<$0.01';
+  }
+  return `$${cost.toFixed(2)}`;
+}
+
+/**
+ * Format token count for display
+ */
+export function formatTokens(count: number): string {
+  if (count >= 1_000_000) {
+    return `${(count / 1_000_000).toFixed(1)}M`;
+  }
+  if (count >= 1_000) {
+    return `${(count / 1_000).toFixed(1)}K`;
+  }
+  return count.toString();
+}
+
+/**
+ * Format cost estimate for display
+ */
+export function formatCostEstimate(estimate: CostEstimate): string {
+  return formatCost(estimate.totalCost);
+}

package/src/pricing/index.ts

@@ -0,0 +1,7 @@
+/**
+ * Pricing and cost tracking module
+ */
+
+export * from './types.js';
+export * from './models.js';
+export * from './calculator.js';

package/src/pricing/models.ts

@@ -0,0 +1,334 @@
+/**
+ * Model pricing database
+ * Prices are per 1M tokens in USD
+ * Updated: January 2025
+ */
+
+import { ModelPricing } from './types.js';
+
+/**
+ * Pricing data for all supported models
+ * Source: Official provider pricing pages as of January 2025
+ */
+export const MODEL_PRICING: ModelPricing[] = [
+  // Anthropic Claude Models
+  {
+    provider: 'anthropic',
+    model: 'claude-opus-4-5',
+    inputPer1M: 15.0,
+    outputPer1M: 75.0,
+    effectiveDate: '2025-01-01',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-opus-4',
+    inputPer1M: 15.0,
+    outputPer1M: 75.0,
+    effectiveDate: '2024-11-01',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-sonnet-4-5',
+    inputPer1M: 3.0,
+    outputPer1M: 15.0,
+    effectiveDate: '2025-01-01',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-sonnet-4',
+    inputPer1M: 3.0,
+    outputPer1M: 15.0,
+    effectiveDate: '2024-10-22',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-3-5-sonnet-20241022',
+    inputPer1M: 3.0,
+    outputPer1M: 15.0,
+    effectiveDate: '2024-10-22',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-3-5-sonnet-20240620',
+    inputPer1M: 3.0,
+    outputPer1M: 15.0,
+    effectiveDate: '2024-06-20',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-haiku-3-5',
+    inputPer1M: 0.8,
+    outputPer1M: 4.0,
+    effectiveDate: '2024-11-01',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-3-5-haiku-20241022',
+    inputPer1M: 0.8,
+    outputPer1M: 4.0,
+    effectiveDate: '2024-10-22',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-3-haiku-20240307',
+    inputPer1M: 0.25,
+    outputPer1M: 1.25,
+    effectiveDate: '2024-03-07',
+  },
+  {
+    provider: 'anthropic',
+    model: 'claude-3-opus-20240229',
+    inputPer1M: 15.0,
+    outputPer1M: 75.0,
+    effectiveDate: '2024-02-29',
+  },
+
+  // OpenAI Models
+  {
+    provider: 'openai',
+    model: 'gpt-4o',
+    inputPer1M: 2.5,
+    outputPer1M: 10.0,
+    effectiveDate: '2024-08-06',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4o-2024-11-20',
+    inputPer1M: 2.5,
+    outputPer1M: 10.0,
+    effectiveDate: '2024-11-20',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4o-2024-08-06',
+    inputPer1M: 2.5,
+    outputPer1M: 10.0,
+    effectiveDate: '2024-08-06',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4o-2024-05-13',
+    inputPer1M: 5.0,
+    outputPer1M: 15.0,
+    effectiveDate: '2024-05-13',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4o-mini',
+    inputPer1M: 0.15,
+    outputPer1M: 0.6,
+    effectiveDate: '2024-07-18',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4o-mini-2024-07-18',
+    inputPer1M: 0.15,
+    outputPer1M: 0.6,
+    effectiveDate: '2024-07-18',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4-turbo',
+    inputPer1M: 10.0,
+    outputPer1M: 30.0,
+    effectiveDate: '2024-04-09',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4-turbo-2024-04-09',
+    inputPer1M: 10.0,
+    outputPer1M: 30.0,
+    effectiveDate: '2024-04-09',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-4',
+    inputPer1M: 30.0,
+    outputPer1M: 60.0,
+    effectiveDate: '2023-03-14',
+  },
+  {
+    provider: 'openai',
+    model: 'gpt-3.5-turbo',
+    inputPer1M: 0.5,
+    outputPer1M: 1.5,
+    effectiveDate: '2023-11-06',
+  },
+  {
+    provider: 'openai',
+    model: 'o1',
+    inputPer1M: 15.0,
+    outputPer1M: 60.0,
+    effectiveDate: '2024-12-17',
+  },
+  {
+    provider: 'openai',
+    model: 'o1-2024-12-17',
+    inputPer1M: 15.0,
+    outputPer1M: 60.0,
+    effectiveDate: '2024-12-17',
+  },
+  {
+    provider: 'openai',
+    model: 'o1-preview',
+    inputPer1M: 15.0,
+    outputPer1M: 60.0,
+    effectiveDate: '2024-09-12',
+  },
+  {
+    provider: 'openai',
+    model: 'o1-preview-2024-09-12',
+    inputPer1M: 15.0,
+    outputPer1M: 60.0,
+    effectiveDate: '2024-09-12',
+  },
+  {
+    provider: 'openai',
+    model: 'o1-mini',
+    inputPer1M: 3.0,
+    outputPer1M: 12.0,
+    effectiveDate: '2024-09-12',
+  },
+  {
+    provider: 'openai',
+    model: 'o1-mini-2024-09-12',
+    inputPer1M: 3.0,
+    outputPer1M: 12.0,
+    effectiveDate: '2024-09-12',
+  },
+
+  // Google Gemini Models
+  {
+    provider: 'gemini',
+    model: 'gemini-2.0-flash-exp',
+    inputPer1M: 0.0,
+    outputPer1M: 0.0,
+    effectiveDate: '2024-12-11',
+  },
+  {
+    provider: 'gemini',
+    model: 'gemini-2.0-flash',
+    inputPer1M: 0.075,
+    outputPer1M: 0.3,
+    effectiveDate: '2025-01-01',
+  },
+  {
+    provider: 'gemini',
+    model: 'gemini-exp-1206',
+    inputPer1M: 0.0,
+    outputPer1M: 0.0,
+    effectiveDate: '2024-12-06',
+  },
+  {
+    provider: 'gemini',
+    model: 'gemini-1.5-pro',
+    inputPer1M: 1.25,
+    outputPer1M: 5.0,
+    effectiveDate: '2024-05-14',
+  },
+  {
+    provider: 'gemini',
+    model: 'gemini-1.5-pro-002',
+    inputPer1M: 1.25,
+    outputPer1M: 5.0,
+    effectiveDate: '2024-09-24',
+  },
+  {
+    provider: 'gemini',
+    model: 'gemini-1.5-flash',
+    inputPer1M: 0.075,
+    outputPer1M: 0.3,
+    effectiveDate: '2024-05-14',
+  },
+  {
+    provider: 'gemini',
+    model: 'gemini-1.5-flash-002',
+    inputPer1M: 0.075,
+    outputPer1M: 0.3,
+    effectiveDate: '2024-09-24',
+  },
+  {
+    provider: 'gemini',
+    model: 'gemini-1.5-flash-8b',
+    inputPer1M: 0.0375,
+    outputPer1M: 0.15,
+    effectiveDate: '2024-10-03',
+  },
+
+  // Google Vertex AI (same pricing as Gemini)
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-2.0-flash-exp',
+    inputPer1M: 0.0,
+    outputPer1M: 0.0,
+    effectiveDate: '2024-12-11',
+  },
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-2.0-flash',
+    inputPer1M: 0.075,
+    outputPer1M: 0.3,
+    effectiveDate: '2025-01-01',
+  },
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-exp-1206',
+    inputPer1M: 0.0,
+    outputPer1M: 0.0,
+    effectiveDate: '2024-12-06',
+  },
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-1.5-pro',
+    inputPer1M: 1.25,
+    outputPer1M: 5.0,
+    effectiveDate: '2024-05-14',
+  },
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-1.5-pro-002',
+    inputPer1M: 1.25,
+    outputPer1M: 5.0,
+    effectiveDate: '2024-09-24',
+  },
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-1.5-flash',
+    inputPer1M: 0.075,
+    outputPer1M: 0.3,
+    effectiveDate: '2024-05-14',
+  },
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-1.5-flash-002',
+    inputPer1M: 0.075,
+    outputPer1M: 0.3,
+    effectiveDate: '2024-09-24',
+  },
+  {
+    provider: 'vertex-ai',
+    model: 'gemini-1.5-flash-8b',
+    inputPer1M: 0.0375,
+    outputPer1M: 0.15,
+    effectiveDate: '2024-10-03',
+  },
+];
+
+/**
+ * Get pricing for a specific model
+ */
+export function getModelPricing(
+  provider: string,
+  model: string
+): ModelPricing | undefined {
+  return MODEL_PRICING.find(
+    (p) => p.provider === provider && p.model === model
+  );
+}
+
+/**
+ * Get all pricing for a provider
+ */
+export function getProviderPricing(provider: string): ModelPricing[] {
+  return MODEL_PRICING.filter((p) => p.provider === provider);
+}

package/src/pricing/types.ts

@@ -0,0 +1,32 @@
+/**
+ * Pricing and cost tracking types
+ */
+
+/**
+ * Token usage information
+ */
+export interface TokenUsage {
+  inputTokens: number;
+  outputTokens: number;
+}
+
+/**
+ * Cost estimate in USD
+ */
+export interface CostEstimate {
+  inputCost: number;
+  outputCost: number;
+  totalCost: number;
+  currency: string; // 'USD'
+}
+
+/**
+ * Model pricing per 1M tokens (USD)
+ */
+export interface ModelPricing {
+  provider: string;
+  model: string;
+  inputPer1M: number; // USD per 1M input tokens
+  outputPer1M: number; // USD per 1M output tokens
+  effectiveDate?: string; // When this pricing became effective
+}

package/src/prompts/index.ts

@@ -14,7 +14,7 @@ import * as os from 'os';
 const __dirname = dirname(fileURLToPath(import.meta.url));
 
 // Path to providers.json config
-const PROVIDERS_CONFIG_PATH = join(homedir(), '.gencode', 'providers.json');
+const PROVIDERS_CONFIG_PATH = join(homedir(), '.gen', 'providers.json');
 
 // Resolve prompts directory - check both src and dist locations
 function getPromptsDir(): string {
@@ -33,7 +33,7 @@ const promptsDir = getPromptsDir();
 export type ProviderType = 'anthropic' | 'openai' | 'gemini' | 'generic';
 
 /**
- * Providers config structure from ~/.gencode/providers.json
+ * Providers config structure from ~/.gen/providers.json
  */
 interface ProvidersConfig {
   connections: Record<string, unknown>;
@@ -41,7 +41,7 @@ interface ProvidersConfig {
 }
 
 /**
- * Load providers config from ~/.gencode/providers.json
+ * Load providers config from ~/.gen/providers.json
  */
 function loadProvidersConfig(): ProvidersConfig | null {
   try {
@@ -57,7 +57,7 @@ function loadProvidersConfig(): ProvidersConfig | null {
 
 /**
  * Look up which provider owns a given model ID
- * Searches through ~/.gencode/providers.json to find the provider
+ * Searches through ~/.gen/providers.json to find the provider
  *
  * @param model - The model ID (e.g., "claude-sonnet-4-5@20250929")
  * @returns The provider name (e.g., "anthropic") or null if not found
@@ -80,9 +80,13 @@ export function getProviderForModel(model: string): string | null {
 /**
  * Map provider names to prompt types
  * Falls back to 'generic' for unknown providers
+ * Handles both "provider" and "provider:authMethod" formats
  */
 export function mapProviderToPromptType(provider: string): ProviderType {
-  switch (provider) {
+  // Extract provider prefix (e.g., "gemini:api_key" → "gemini")
+  const providerPrefix = provider.split(':')[0];
+
+  switch (providerPrefix) {
     case 'anthropic':
       return 'anthropic';
     case 'openai':
@@ -200,7 +204,7 @@ export function buildSystemPrompt(
 
 /**
  * Format memory context for injection into system prompt
- * Uses <claudeMd> tag for Claude Code compatibility
+ * Uses <claudeMd> tag for optimal compatibility with Claude models
  */
 export function formatMemoryContext(memoryContext: string): string {
   if (!memoryContext) {
@@ -240,7 +244,7 @@ export function buildSystemPromptWithMemory(
  * Flow: model → provider (from providers.json) → prompt
  *
  * This is the recommended way to build system prompts as it automatically
- * looks up the provider for the given model from ~/.gencode/providers.json
+ * looks up the provider for the given model from ~/.gen/providers.json
  *
  * @param model - The model ID (e.g., "claude-sonnet-4-5@20250929")
  * @param cwd - Current working directory
@@ -261,10 +265,10 @@ export function buildSystemPromptForModel(
 
 /**
  * Debug utility to verify prompt loading at runtime
- * Set GENCODE_DEBUG_PROMPTS=1 for summary, GENCODE_DEBUG_PROMPTS=2 for full content
+ * Set GEN_DEBUG_PROMPTS=1 for summary, GEN_DEBUG_PROMPTS=2 for full content
  */
 export function debugPromptLoading(model: string, fallbackProvider?: string): void {
-  const debugLevel = process.env.GENCODE_DEBUG_PROMPTS;
+  const debugLevel = process.env.GEN_DEBUG_PROMPTS;
   if (!debugLevel || debugLevel === '0') {
     return;
   }