claude-code-model-switch 1.0.0

package/example-config.json

@@ -0,0 +1,24 @@
+{
+  "models": {
+    "deepseek-chat": {
+      "ANTHROPIC_AUTH_TOKEN": "sk-caf695360e7f4b018bb1b308cbf992eb",
+      "ANTHROPIC_BASE_URL": "https://api.deepseek.com/anthropic",
+      "ANTHROPIC_MODEL": "deepseek-chat",
+      "ANTHROPIC_SMALL_FAST_MODEL": "deepseek-chat",
+      "ANTHROPIC_DEFAULT_SONNET_MODEL": "deepseek-chat",
+      "ANTHROPIC_DEFAULT_OPUS_MODEL": "deepseek-chat",
+      "CLAUDE_CODE_MAX_OUTPUT_TOKENS": "8192",
+      "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": "1"
+    },
+    "claude-3-5-sonnet": {
+      "ANTHROPIC_AUTH_TOKEN": "your-anthropic-token-here",
+      "ANTHROPIC_BASE_URL": "https://api.anthropic.com",
+      "ANTHROPIC_MODEL": "claude-3-5-sonnet-20241022",
+      "ANTHROPIC_SMALL_FAST_MODEL": "claude-3-haiku-20240307",
+      "ANTHROPIC_DEFAULT_SONNET_MODEL": "claude-3-5-sonnet-20241022",
+      "ANTHROPIC_DEFAULT_OPUS_MODEL": "claude-3-opus-20240229",
+      "CLAUDE_CODE_MAX_OUTPUT_TOKENS": "8192",
+      "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": "0"
+    }
+  }
+}

package/index.js

@@ -0,0 +1,240 @@
+#!/usr/bin/env node
+
+const fs = require('fs');
+const path = require('path');
+const { program } = require('commander');
+
+// Configuration file paths
+const CLAUDE_SETTINGS_PATH = path.join(process.env.HOME, '.claude', 'settings.json');
+const MODEL_CONFIG_PATH = path.join(process.env.HOME, '.claude-code-model-switch', 'settings.json');
+
+// Default values
+const DEFAULT_MAX_OUTPUT_TOKENS = '8192';
+const DEFAULT_DISABLE_NONESSENTIAL_TRAFFIC = '1';
+
+// Ensure configuration directory exists
+function ensureConfigDir() {
+  const configDir = path.dirname(MODEL_CONFIG_PATH);
+  if (!fs.existsSync(configDir)) {
+    fs.mkdirSync(configDir, { recursive: true });
+  }
+}
+
+// Load model configuration
+function loadModelConfig() {
+  ensureConfigDir();
+
+  if (!fs.existsSync(MODEL_CONFIG_PATH)) {
+    // Create empty default configuration file
+    const defaultConfig = { models: {} };
+    fs.writeFileSync(MODEL_CONFIG_PATH, JSON.stringify(defaultConfig, null, 2));
+    return defaultConfig;
+  }
+
+  try {
+    const configContent = fs.readFileSync(MODEL_CONFIG_PATH, 'utf8');
+    return JSON.parse(configContent);
+  } catch (error) {
+    console.error('Error: Failed to parse model configuration file');
+    process.exit(1);
+  }
+}
+
+// Load Claude settings
+function loadClaudeSettings() {
+  if (!fs.existsSync(CLAUDE_SETTINGS_PATH)) {
+    console.error('Error: Claude settings file not found');
+    process.exit(1);
+  }
+
+  try {
+    const settingsContent = fs.readFileSync(CLAUDE_SETTINGS_PATH, 'utf8');
+    return JSON.parse(settingsContent);
+  } catch (error) {
+    console.error('Error: Failed to parse Claude settings file');
+    process.exit(1);
+  }
+}
+
+// Save Claude settings
+function saveClaudeSettings(settings) {
+  try {
+    fs.writeFileSync(CLAUDE_SETTINGS_PATH, JSON.stringify(settings, null, 2));
+  } catch (error) {
+    console.error('Error: Failed to save Claude settings file');
+    process.exit(1);
+  }
+}
+
+// Get model configuration
+function getModelConfig(modelName) {
+  const config = loadModelConfig();
+
+  if (!config.models || !config.models[modelName]) {
+    console.error(`Error: Model "${modelName}" not found`);
+    console.log('Available models:', Object.keys(config.models || {}).join(', ') || 'None');
+    process.exit(1);
+  }
+
+  const modelConfig = config.models[modelName];
+
+  // Check if model configuration is empty
+  if (Object.keys(modelConfig).length === 0) {
+    console.error(`Error: Model "${modelName}" configuration is empty`);
+    process.exit(1);
+  }
+
+  return modelConfig;
+}
+
+// Apply model configuration
+function applyModelConfig(modelConfig) {
+  const settings = loadClaudeSettings();
+
+  // Ensure env object exists
+  if (!settings.env) {
+    settings.env = {};
+  }
+
+  // Only update model-related environment variables
+  const modelEnvVars = [
+    'ANTHROPIC_AUTH_TOKEN',
+    'ANTHROPIC_BASE_URL',
+    'ANTHROPIC_MODEL',
+    'ANTHROPIC_SMALL_FAST_MODEL',
+    'ANTHROPIC_DEFAULT_SONNET_MODEL',
+    'ANTHROPIC_DEFAULT_OPUS_MODEL',
+    'CLAUDE_CODE_MAX_OUTPUT_TOKENS',
+    'CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC'
+  ];
+
+  // Apply configuration
+  for (const key of modelEnvVars) {
+    if (modelConfig[key] !== undefined) {
+      settings.env[key] = modelConfig[key];
+    }
+  }
+
+  // Handle default value logic
+  // Check required environment variables
+  const requiredEnvVars = [
+    'ANTHROPIC_MODEL',
+    'ANTHROPIC_AUTH_TOKEN',
+    'ANTHROPIC_BASE_URL'
+  ];
+
+  for (const envVar of requiredEnvVars) {
+    if (!settings.env[envVar]) {
+      console.error(`Error: ${envVar} is required`);
+      process.exit(1);
+    }
+  }
+
+  // If other model fields are not set, use ANTHROPIC_MODEL value
+  const derivedModelFields = [
+    'ANTHROPIC_SMALL_FAST_MODEL',
+    'ANTHROPIC_DEFAULT_SONNET_MODEL',
+    'ANTHROPIC_DEFAULT_OPUS_MODEL'
+  ];
+
+  for (const field of derivedModelFields) {
+    if (!settings.env[field]) {
+      settings.env[field] = settings.env.ANTHROPIC_MODEL;
+    }
+  }
+
+  // Handle special environment variable defaults
+  if (!settings.env.CLAUDE_CODE_MAX_OUTPUT_TOKENS) {
+    settings.env.CLAUDE_CODE_MAX_OUTPUT_TOKENS = DEFAULT_MAX_OUTPUT_TOKENS;
+  }
+
+  if (!settings.env.CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC) {
+    settings.env.CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC = DEFAULT_DISABLE_NONESSENTIAL_TRAFFIC;
+  }
+
+  saveClaudeSettings(settings);
+  console.log(`✅ Switched to model: ${settings.env.ANTHROPIC_MODEL}`);
+}
+
+// List all models
+function listModels() {
+  const config = loadModelConfig();
+  const models = Object.keys(config.models || {});
+
+  if (models.length === 0) {
+    console.log('No models configured');
+    return;
+  }
+
+  console.log('Available models:');
+  models.forEach(model => {
+    console.log(`  - ${model}`);
+  });
+}
+
+// Check if model name is valid
+function isValidModel(modelName) {
+  const config = loadModelConfig();
+  return config.models &&
+         config.models[modelName] &&
+         Object.keys(config.models[modelName]).length > 0;
+}
+
+// Main program
+program
+  .name('ccms')
+  .description('Claude Code Model Switch - Switch Claude Code models')
+  .version('1.0.0')
+  .argument('[model]', 'Model name to switch to')
+  .action((model) => {
+    if (model) {
+      // If model name argument provided, attempt to switch model
+      if (isValidModel(model)) {
+        const modelConfig = getModelConfig(model);
+        applyModelConfig(modelConfig);
+      } else {
+        const config = loadModelConfig();
+        const modelConfig = config.models && config.models[model];
+
+        if (modelConfig && Object.keys(modelConfig).length === 0) {
+          console.error(`Error: Model "${model}" configuration is empty`);
+        } else {
+          console.error(`Error: Model "${model}" not found`);
+        }
+
+        const validModels = Object.keys(config.models || {}).filter(name =>
+          config.models[name] && Object.keys(config.models[name]).length > 0
+        );
+        console.log('Available models:', validModels.join(', ') || 'None');
+        process.exit(1);
+      }
+    } else {
+      // If no arguments provided, show help information
+      program.help();
+    }
+  });
+
+program
+  .command('switch <model>')
+  .description('Switch to specified model')
+  .action((model) => {
+    const modelConfig = getModelConfig(model);
+    applyModelConfig(modelConfig);
+  });
+
+program
+  .command('list')
+  .description('List all available models')
+  .action(() => {
+    listModels();
+  });
+
+program
+  .command('config-path')
+  .description('Display configuration file paths')
+  .action(() => {
+    console.log('Model configuration file path:', MODEL_CONFIG_PATH);
+    console.log('Claude settings file path:', CLAUDE_SETTINGS_PATH);
+  });
+
+program.parse();

package/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "claude-code-model-switch",
+  "version": "1.0.0",
+  "description": "A CLI tool to switch Claude Code models",
+  "main": "index.js",
+  "bin": {
+    "ccms": "./index.js"
+  },
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": ["claude", "code", "model", "switch"],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "commander": "^11.1.0"
+  }
+}

package/prompt.md

@@ -0,0 +1,45 @@
+我需要实现一个claude code的模型切换器,这个切换器用node.js实现,可以通过npm安装.
+
+实现逻辑非常简单.读取`~/.claude/settings.json`,切换里面和模型有关的环境变量。除了这些env，禁止修改配置文件的任何其他字段。
+
+```json
+{
+  "env": {
+    "ANTHROPIC_AUTH_TOKEN": "sk-caf695360e7f4b018bb1b308cbf992eb",
+    "ANTHROPIC_BASE_URL": "https://api.deepseek.com/anthropic",
+    "ANTHROPIC_MODEL": "deepseek-chat",
+    "ANTHROPIC_SMALL_FAST_MODEL": "deepseek-chat",
+    "ANTHROPIC_DEFAULT_SONNET_MODEL": "deepseek-chat",
+    "ANTHROPIC_DEFAULT_OPUS_MODEL": "deepseek-chat",
+    "CLAUDE_CODE_MAX_OUTPUT_TOKENS": "8192",
+    "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": "1"
+  }
+}
+```
+
+这个工具本身通过~/.claude-code-models.json配置多个模型,格式为
+
+```json
+{
+  "models":{
+    "模型名称":{
+        "ANTHROPIC_AUTH_TOKEN": "sk-caf695360e7f4b018bb1b308cbf992eb",
+        "ANTHROPIC_BASE_URL": "https://api.deepseek.com/anthropic",
+        "ANTHROPIC_MODEL": "deepseek-chat",
+        "ANTHROPIC_SMALL_FAST_MODEL": "deepseek-chat",
+        "ANTHROPIC_DEFAULT_SONNET_MODEL": "deepseek-chat",
+        "ANTHROPIC_DEFAULT_OPUS_MODEL": "deepseek-chat",
+        "CLAUDE_CODE_MAX_OUTPUT_TOKENS": "8192",
+        "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": "1"
+    }
+  }
+}
+```
+
+当用户执行`ccms 模型`命令时，读取`~/.claude-code-model-swich/settings.json`文件，检查是否有这个模型名称，没有报错，有就进行`~/.claude/settings.json`替换
+
+重点：
+1. `CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC`是一个特殊的环境变量，当用户没有设置这个环境变量时，**默认为1**，当用户设置为0时，表示允许非必要的流量，对应修改成0
+2. `CLAUDE_CODE_MAX_OUTPUT_TOKENS`用户没有设置的时候，默认为8192
+3. ANTHROPIC_SMALL_FAST_MODEL、ANTHROPIC_DEFAULT_SONNET_MODEL、ANTHROPIC_DEFAULT_OPUS_MODEL都可以不设置，只有ANTHROPIC_MODEL是必填的，其余三个环境变量在没有设置的时候，沿用ANTHROPIC_MODEL的值。
+4. `~/.claude-code-model-swich/settings.json`文件不存在的时候，创建一个空的默认配置文件`{"models": {}}`，并按照模型不存在的错误提示用户（认为加载了这个空配置文件）