threewzrd 1.0.2 → 1.0.3

package/dist/cli.js

@@ -3,6 +3,7 @@ import { Command } from 'commander';
 import { homedir } from 'os';
 import { startCommand } from './commands/start.js';
 import { configCommand } from './commands/config.js';
+import { modelCommand } from './commands/model.js';
 // Safely get current working directory, fallback to home
 function safeGetCwd() {
     try {
@@ -22,6 +23,7 @@ program
     .command('start')
     .description('Start the wizard REPL')
     .option('-d, --directory <path>', 'Working directory for the wizard', safeGetCwd())
+    .option('-m, --model <model>', 'Model to use (sonnet, opus, haiku, opus-4.6)')
     .action(startCommand);
 program
     .command('config')
@@ -30,4 +32,8 @@ program
     .option('-d, --delete', 'Delete saved API key')
     .option('-p, --path', 'Show config file path')
     .action(configCommand);
+program
+    .command('model [name]')
+    .description('View or set the default model')
+    .action(modelCommand);
 program.parse();

package/dist/commands/model.d.ts

@@ -0,0 +1 @@
+export declare function modelCommand(name?: string): Promise<void>;

package/dist/commands/model.js

@@ -0,0 +1,62 @@
+import { homedir } from 'os';
+import { join } from 'path';
+import { mkdir, writeFile, readFile } from 'fs/promises';
+import chalk from 'chalk';
+import { MODEL_MAP, DEFAULT_MODEL } from '../core/types.js';
+const CONFIG_DIR = join(homedir(), '.threewzrd');
+const CONFIG_PATH = join(CONFIG_DIR, 'config.json');
+const VALID_MODELS = ['sonnet', 'opus', 'haiku', 'opus-4.6'];
+async function loadConfig() {
+    try {
+        const content = await readFile(CONFIG_PATH, 'utf-8');
+        return JSON.parse(content);
+    }
+    catch {
+        return {};
+    }
+}
+async function saveConfig(config) {
+    await mkdir(CONFIG_DIR, { recursive: true, mode: 0o700 });
+    await writeFile(CONFIG_PATH, JSON.stringify(config, null, 2) + '\n', { mode: 0o600 });
+}
+export async function modelCommand(name) {
+    console.log();
+    if (!name) {
+        // Show current model
+        const config = await loadConfig();
+        const currentModel = config.model || DEFAULT_MODEL;
+        console.log(chalk.cyan('  Model Configuration'));
+        console.log(chalk.gray('  ─────────────────────────────────────────'));
+        console.log();
+        console.log(chalk.gray('  Current default: ') + chalk.green(currentModel));
+        console.log(chalk.gray('  Model ID: ') + chalk.gray(MODEL_MAP[currentModel]));
+        console.log();
+        console.log(chalk.gray('  Available models:'));
+        for (const model of VALID_MODELS) {
+            const marker = model === currentModel ? chalk.green(' (default)') : '';
+            console.log(chalk.gray('    - ') + chalk.white(model) + marker);
+        }
+        console.log();
+        console.log(chalk.gray('  To change: ') + chalk.cyan('threewzrd model <name>'));
+        console.log();
+        return;
+    }
+    // Set model
+    const modelName = name.toLowerCase();
+    if (!VALID_MODELS.includes(modelName)) {
+        console.log(chalk.red(`  Unknown model: ${name}`));
+        console.log();
+        console.log(chalk.gray('  Valid models:'));
+        for (const model of VALID_MODELS) {
+            console.log(chalk.gray('    - ') + chalk.white(model));
+        }
+        console.log();
+        return;
+    }
+    const config = await loadConfig();
+    config.model = modelName;
+    await saveConfig(config);
+    console.log(chalk.green(`  Default model set to: ${modelName}`));
+    console.log(chalk.gray(`  Model ID: ${MODEL_MAP[modelName]}`));
+    console.log();
+}

package/dist/commands/start.d.ts

@@ -1,5 +1,6 @@
 interface StartOptions {
     directory: string;
+    model?: string;
 }
 export declare function startCommand(options: StartOptions): Promise<void>;
 export {};

package/dist/commands/start.js

@@ -1,10 +1,25 @@
 import dotenv from 'dotenv';
 import { homedir } from 'os';
 import { join } from 'path';
-import { mkdir, writeFile } from 'fs/promises';
+import { mkdir, writeFile, readFile } from 'fs/promises';
 import * as readline from 'readline';
 import chalk from 'chalk';
 import { ThreeJsWizard } from '../core/ThreeJsWizard.js';
+const VALID_MODELS = ['sonnet', 'opus', 'haiku', 'opus-4.6'];
+async function getConfiguredModel() {
+    try {
+        const configPath = join(homedir(), '.threewzrd', 'config.json');
+        const content = await readFile(configPath, 'utf-8');
+        const config = JSON.parse(content);
+        if (config.model && VALID_MODELS.includes(config.model)) {
+            return config.model;
+        }
+    }
+    catch {
+        // No config file or invalid config
+    }
+    return undefined;
+}
 function loadEnvFiles(workingDir) {
     // Load from multiple locations (later ones don't override earlier)
     // 1. Current working directory
@@ -128,8 +143,23 @@ export async function startCommand(options) {
             await saveApiKey(apiKey);
         }
     }
+    // Determine which model to use (CLI flag > config > default)
+    let model;
+    if (options.model) {
+        if (VALID_MODELS.includes(options.model)) {
+            model = options.model;
+        }
+        else {
+            console.error(chalk.red(`Invalid model: ${options.model}`));
+            console.error(chalk.gray(`Valid models: ${VALID_MODELS.join(', ')}`));
+            process.exit(1);
+        }
+    }
+    else {
+        model = await getConfiguredModel();
+    }
     // Create and start the wizard
-    const wizard = new ThreeJsWizard();
+    const wizard = new ThreeJsWizard({ model });
     // Handle graceful shutdown
     process.on('SIGINT', () => {
         console.log('\nShutting down...');

package/dist/core/AgentEngine.js

@@ -1,5 +1,5 @@
 import Anthropic from '@anthropic-ai/sdk';
-import { MODEL_MAP } from './types.js';
+import { MODEL_MAP, DEFAULT_MODEL } from './types.js';
 import { toolDefinitions } from '../tools/definitions.js';
 import { ToolExecutor } from '../tools/ToolExecutor.js';
 import { THREEJS_SYSTEM_PROMPT } from '../prompts/system.js';
@@ -10,7 +10,7 @@ const MAX_RETRIES = 3;
 const RETRY_DELAY_MS = 5000; // 5 seconds base delay
 export class AgentEngine {
     client;
-    model = 'sonnet';
+    model = DEFAULT_MODEL;
     conversationHistory = [];
     toolExecutor;
     ui;

package/dist/core/ThreeJsWizard.d.ts

@@ -1,3 +1,7 @@
+import { ModelId } from './types.js';
+export interface WizardOptions {
+    model?: ModelId;
+}
 export declare class ThreeJsWizard {
     private ui;
     private engine;
@@ -5,7 +9,7 @@ export declare class ThreeJsWizard {
     private workingDirectory;
     private isRunning;
     private hasOnboarded;
-    constructor();
+    constructor(options?: WizardOptions);
     start(): Promise<void>;
     private handleCommand;
     stop(): void;

package/dist/core/ThreeJsWizard.js

@@ -9,11 +9,14 @@ export class ThreeJsWizard {
     workingDirectory;
     isRunning = false;
     hasOnboarded = false;
-    constructor() {
+    constructor(options) {
         this.workingDirectory = process.cwd();
         this.ui = new TerminalUI();
         this.engine = new AgentEngine(this.ui, this.workingDirectory);
         this.projectManager = new ProjectManager(this.workingDirectory);
+        if (options?.model) {
+            this.engine.setModel(options.model);
+        }
     }
     async start() {
         this.isRunning = true;
@@ -100,16 +103,16 @@ export class ThreeJsWizard {
             case 'model':
                 if (args.length === 0) {
                     this.ui.printInfo(`Current model: ${this.engine.getModel()}`);
-                    this.ui.printInfo('Available models: sonnet, opus, haiku');
+                    this.ui.printInfo('Available models: sonnet, opus, haiku, opus-4.6');
                 }
                 else {
                     const modelName = args[0].toLowerCase();
-                    if (['sonnet', 'opus', 'haiku'].includes(modelName)) {
+                    if (['sonnet', 'opus', 'haiku', 'opus-4.6'].includes(modelName)) {
                         this.engine.setModel(modelName);
                         this.ui.printModelSwitch(modelName);
                     }
                     else {
-                        this.ui.printError(`Unknown model: ${modelName}. Use: sonnet, opus, or haiku`);
+                        this.ui.printError(`Unknown model: ${modelName}. Use: sonnet, opus, haiku, or opus-4.6`);
                     }
                 }
                 break;

package/dist/core/types.d.ts

@@ -1,6 +1,7 @@
 import Anthropic from '@anthropic-ai/sdk';
-export type ModelId = 'sonnet' | 'opus' | 'haiku';
+export type ModelId = 'sonnet' | 'opus' | 'haiku' | 'opus-4.6';
 export declare const MODEL_MAP: Record<ModelId, string>;
+export declare const DEFAULT_MODEL: ModelId;
 export type ProjectLanguage = 'javascript' | 'typescript';
 export type ProjectTarget = 'browser' | 'mobile' | 'desktop';
 export interface ProjectPreferences {

package/dist/core/types.js

@@ -2,4 +2,6 @@ export const MODEL_MAP = {
     sonnet: 'claude-sonnet-4-20250514',
     opus: 'claude-opus-4-20250514',
     haiku: 'claude-haiku-4-20250514',
+    'opus-4.6': 'claude-opus-4-6-20260201',
 };
+export const DEFAULT_MODEL = 'opus-4.6';

package/dist/ui/TerminalUI.js

@@ -110,6 +110,7 @@ export class TerminalUI {
             sonnet: 'Claude Sonnet 4',
             opus: 'Claude Opus 4',
             haiku: 'Claude Haiku 4',
+            'opus-4.6': 'Claude Opus 4.6',
         };
         console.log(chalk.green(`Switched to ${modelNames[model]}`));
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "threewzrd",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "AI-powered CLI for generating Three.js projects from natural language",
   "main": "dist/cli.js",
   "bin": {