chainlesschain 0.38.1 → 0.40.2

package/src/lib/task-model-selector.js

@@ -0,0 +1,232 @@
+/**
+ * Task-based intelligent model selector for CLI
+ *
+ * Detects task type from user messages and recommends the best model
+ * for each LLM provider. Enables automatic model switching based on
+ * what the user is trying to accomplish.
+ */
+
+/**
+ * Task types supported by the selector
+ */
+export const TaskType = {
+  CHAT: "chat",
+  CODE: "code",
+  REASONING: "reasoning",
+  FAST: "fast",
+  TRANSLATE: "translate",
+  CREATIVE: "creative",
+};
+
+/**
+ * Task type → recommended model per provider
+ * Each provider maps to the best model for that task type.
+ */
+const TASK_MODEL_MAP = {
+  [TaskType.CHAT]: {
+    volcengine: "doubao-seed-1-6-flash-250828",
+    openai: "gpt-4o-mini",
+    anthropic: "claude-sonnet-4-6",
+    deepseek: "deepseek-chat",
+    dashscope: "qwen-plus",
+    gemini: "gemini-2.0-flash",
+    mistral: "mistral-medium-latest",
+    ollama: "qwen2:7b",
+  },
+  [TaskType.CODE]: {
+    volcengine: "doubao-seed-code",
+    openai: "gpt-4o",
+    anthropic: "claude-sonnet-4-6",
+    deepseek: "deepseek-coder",
+    dashscope: "qwen-max",
+    gemini: "gemini-2.0-pro",
+    mistral: "mistral-large-latest",
+    ollama: "codellama:7b",
+  },
+  [TaskType.REASONING]: {
+    volcengine: "doubao-seed-1-6-251015",
+    openai: "o1",
+    anthropic: "claude-opus-4-6",
+    deepseek: "deepseek-reasoner",
+    dashscope: "qwen-max",
+    gemini: "gemini-2.0-pro",
+    mistral: "mistral-large-latest",
+    ollama: "qwen2:7b",
+  },
+  [TaskType.FAST]: {
+    volcengine: "doubao-seed-1-6-lite-251015",
+    openai: "gpt-4o-mini",
+    anthropic: "claude-haiku-4-5-20251001",
+    deepseek: "deepseek-chat",
+    dashscope: "qwen-turbo",
+    gemini: "gemini-2.0-flash",
+    mistral: "mistral-small-latest",
+    ollama: "qwen2:7b",
+  },
+  [TaskType.TRANSLATE]: {
+    volcengine: "doubao-seed-1-6-251015",
+    openai: "gpt-4o",
+    anthropic: "claude-sonnet-4-6",
+    deepseek: "deepseek-chat",
+    dashscope: "qwen-plus",
+    gemini: "gemini-2.0-flash",
+    mistral: "mistral-large-latest",
+    ollama: "qwen2:7b",
+  },
+  [TaskType.CREATIVE]: {
+    volcengine: "doubao-seed-1-6-251015",
+    openai: "gpt-4o",
+    anthropic: "claude-opus-4-6",
+    deepseek: "deepseek-chat",
+    dashscope: "qwen-max",
+    gemini: "gemini-2.0-pro",
+    mistral: "mistral-large-latest",
+    ollama: "qwen2:7b",
+  },
+};
+
+/**
+ * Task type display names (Chinese + English)
+ */
+const TASK_NAMES = {
+  [TaskType.CHAT]: "日常对话",
+  [TaskType.CODE]: "代码任务",
+  [TaskType.REASONING]: "复杂推理",
+  [TaskType.FAST]: "快速响应",
+  [TaskType.TRANSLATE]: "翻译任务",
+  [TaskType.CREATIVE]: "创意写作",
+};
+
+/**
+ * Keyword patterns for detecting task type from user message.
+ * Each pattern is [regex, taskType, priority].
+ * Higher priority wins when multiple patterns match.
+ */
+const TASK_PATTERNS = [
+  // Code patterns (priority 10) — English with word boundaries
+  [
+    /\b(code|coding|program|function|class|bug|debug|refactor|implement)\b/i,
+    TaskType.CODE,
+    10,
+  ],
+  [
+    /\b(javascript|typescript|python|java|rust|go|c\+\+|sql|html|css|react|vue|node|npm|git|api|endpoint|database)\b/i,
+    TaskType.CODE,
+    10,
+  ],
+  [/```[\s\S]*```/, TaskType.CODE, 10],
+  // Code patterns — Chinese (no \b, Chinese chars are not word-boundary compatible)
+  [
+    /(代码|编程|函数|调试|重构|实现|写[一个]*[代码函数方法])/,
+    TaskType.CODE,
+    10,
+  ],
+
+  // Reasoning patterns (priority 8)
+  [
+    /\b(analyze|reason|explain why|prove|compare|evaluate)\b/i,
+    TaskType.REASONING,
+    8,
+  ],
+  [/\b(step.by.step|think.*through)\b/i, TaskType.REASONING, 8],
+  [
+    /(分析|推理|解释为什么|证明|比较|评估|深度思考|逻辑|逐步|一步一步)/,
+    TaskType.REASONING,
+    8,
+  ],
+
+  // Translation patterns (priority 9)
+  [/\b(translate|translation|translate.*to)\b/i, TaskType.TRANSLATE, 9],
+  [/(翻译|转换.*语言|英译中|中译英)/, TaskType.TRANSLATE, 9],
+
+  // Creative patterns (priority 7)
+  [
+    /\b(write|create|compose|story|poem|essay|blog|article)\b/i,
+    TaskType.CREATIVE,
+    7,
+  ],
+  [/(写[一篇]*.*[故事诗歌文章博客]|创作|小说|剧本)/, TaskType.CREATIVE, 7],
+
+  // Fast patterns (priority 5)
+  [/\b(quick|brief|short)\b/i, TaskType.FAST, 5],
+  [/(简短|快速|简单回答|一句话)/, TaskType.FAST, 5],
+];
+
+/**
+ * Detect the task type from a user message using keyword matching.
+ *
+ * @param {string} message - User's input message
+ * @returns {{ taskType: string, confidence: number, name: string }}
+ */
+export function detectTaskType(message) {
+  if (!message || typeof message !== "string") {
+    return {
+      taskType: TaskType.CHAT,
+      confidence: 0,
+      name: TASK_NAMES[TaskType.CHAT],
+    };
+  }
+
+  let bestMatch = null;
+  let bestPriority = -1;
+  let matchCount = 0;
+
+  for (const [pattern, taskType, priority] of TASK_PATTERNS) {
+    if (pattern.test(message)) {
+      matchCount++;
+      if (priority > bestPriority) {
+        bestPriority = priority;
+        bestMatch = taskType;
+      }
+    }
+  }
+
+  if (!bestMatch) {
+    return {
+      taskType: TaskType.CHAT,
+      confidence: 0,
+      name: TASK_NAMES[TaskType.CHAT],
+    };
+  }
+
+  // Confidence based on match count and priority
+  const confidence = Math.min(1, matchCount * 0.3 + bestPriority * 0.07);
+
+  return {
+    taskType: bestMatch,
+    confidence,
+    name: TASK_NAMES[bestMatch],
+  };
+}
+
+/**
+ * Select the best model for a given provider and task type.
+ *
+ * @param {string} provider - LLM provider name
+ * @param {string} taskType - Task type from TaskType enum
+ * @returns {string|null} Model ID, or null if no recommendation
+ */
+export function selectModelForTask(provider, taskType) {
+  const taskMap = TASK_MODEL_MAP[taskType];
+  if (!taskMap) return null;
+  return taskMap[provider] || null;
+}
+
+/**
+ * Get a human-readable task name.
+ *
+ * @param {string} taskType - Task type
+ * @returns {string}
+ */
+export function getTaskName(taskType) {
+  return TASK_NAMES[taskType] || taskType;
+}
+
+/**
+ * Get all supported task types.
+ *
+ * @returns {Object} TaskType enum
+ */
+export function getTaskTypes() {
+  return { ...TaskType };
+}

package/src/lib/ws-server.js

@@ -0,0 +1,474 @@
+/**
+ * ChainlessChain WebSocket Server
+ *
+ * Exposes CLI commands over WebSocket for remote access by IDE plugins,
+ * web frontends, automation scripts, etc. Commands are executed by spawning
+ * child processes — all 60+ CLI commands are available immediately.
+ */
+
+import { EventEmitter } from "node:events";
+import { spawn } from "node:child_process";
+import { fileURLToPath } from "node:url";
+import { dirname, join } from "node:path";
+import { WebSocketServer } from "ws";
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+/** Absolute path to the CLI entry point */
+const BIN_PATH = join(__dirname, "..", "..", "bin", "chainlesschain.js");
+
+/** Commands that must not be executed via WebSocket */
+const BLOCKED_COMMANDS = new Set(["serve", "chat", "agent", "setup"]);
+
+/** Heartbeat interval (ms) */
+const HEARTBEAT_INTERVAL = 30_000;
+
+/**
+ * Tokenize a command string into an array of arguments.
+ * Handles double-quoted and single-quoted strings. Does NOT invoke a shell.
+ */
+export function tokenizeCommand(input) {
+  const args = [];
+  let current = "";
+  let inDouble = false;
+  let inSingle = false;
+  let escape = false;
+
+  for (const ch of input) {
+    if (escape) {
+      current += ch;
+      escape = false;
+      continue;
+    }
+    if (ch === "\\" && inDouble) {
+      escape = true;
+      continue;
+    }
+    if (ch === '"' && !inSingle) {
+      inDouble = !inDouble;
+      continue;
+    }
+    if (ch === "'" && !inDouble) {
+      inSingle = !inSingle;
+      continue;
+    }
+    if ((ch === " " || ch === "\t") && !inDouble && !inSingle) {
+      if (current.length > 0) {
+        args.push(current);
+        current = "";
+      }
+      continue;
+    }
+    current += ch;
+  }
+  if (current.length > 0) {
+    args.push(current);
+  }
+  return args;
+}
+
+export class ChainlessChainWSServer extends EventEmitter {
+  /**
+   * @param {object} options
+   * @param {number}  [options.port=18800]
+   * @param {string}  [options.host="127.0.0.1"]
+   * @param {string}  [options.token]           - If set, clients must authenticate first
+   * @param {number}  [options.maxConnections=10]
+   * @param {number}  [options.timeout=30000]   - Command execution timeout (ms)
+   */
+  constructor(options = {}) {
+    super();
+    this.port = options.port || 18800;
+    this.host = options.host || "127.0.0.1";
+    this.token = options.token || null;
+    this.maxConnections = options.maxConnections || 10;
+    this.timeout = options.timeout || 30000;
+
+    /** @type {WebSocketServer|null} */
+    this.wss = null;
+
+    /** Connected clients: clientId → { ws, authenticated, connectedAt } */
+    this.clients = new Map();
+
+    /** Running child processes: requestId → ChildProcess */
+    this.processes = new Map();
+
+    this._heartbeatTimer = null;
+    this._clientCounter = 0;
+  }
+
+  /** Start the WebSocket server */
+  start() {
+    return new Promise((resolve, reject) => {
+      this.wss = new WebSocketServer({
+        port: this.port,
+        host: this.host,
+      });
+
+      this.wss.on("listening", () => {
+        this._startHeartbeat();
+        this.emit("listening", { port: this.port, host: this.host });
+        resolve();
+      });
+
+      this.wss.on("error", (err) => {
+        this.emit("error", err);
+        reject(err);
+      });
+
+      this.wss.on("connection", (ws, req) => this._handleConnection(ws, req));
+    });
+  }
+
+  /** Stop the server and clean up */
+  async stop() {
+    if (this._heartbeatTimer) {
+      clearInterval(this._heartbeatTimer);
+      this._heartbeatTimer = null;
+    }
+
+    // Kill all running child processes
+    for (const [id, child] of this.processes) {
+      try {
+        child.kill("SIGTERM");
+      } catch (_err) {
+        // Process may have already exited
+      }
+      this.processes.delete(id);
+    }
+
+    // Close all client connections
+    for (const [, client] of this.clients) {
+      try {
+        client.ws.close(1001, "Server shutting down");
+      } catch (_err) {
+        // Connection may already be closed
+      }
+    }
+    this.clients.clear();
+
+    // Close the server
+    if (this.wss) {
+      await new Promise((resolve) => {
+        this.wss.close(() => resolve());
+      });
+      this.wss = null;
+    }
+
+    this.emit("stopped");
+  }
+
+  /** @private */
+  _handleConnection(ws, req) {
+    if (this.clients.size >= this.maxConnections) {
+      ws.close(1013, "Max connections reached");
+      return;
+    }
+
+    const clientId = `client-${++this._clientCounter}`;
+    const clientIp =
+      req.socket.remoteAddress || req.headers["x-forwarded-for"] || "unknown";
+
+    this.clients.set(clientId, {
+      ws,
+      authenticated: !this.token, // If no token required, auto-authenticated
+      connectedAt: Date.now(),
+      ip: clientIp,
+      alive: true,
+    });
+
+    this.emit("connection", { clientId, ip: clientIp });
+
+    ws.on("message", (data) => {
+      try {
+        const message = JSON.parse(data.toString("utf8"));
+        this._handleMessage(clientId, ws, message);
+      } catch (_err) {
+        this._send(ws, {
+          type: "error",
+          code: "INVALID_JSON",
+          message: "Failed to parse message as JSON",
+        });
+      }
+    });
+
+    ws.on("close", () => {
+      this.clients.delete(clientId);
+      this.emit("disconnection", { clientId });
+    });
+
+    ws.on("pong", () => {
+      const client = this.clients.get(clientId);
+      if (client) client.alive = true;
+    });
+  }
+
+  /** @private */
+  _handleMessage(clientId, ws, message) {
+    const { id, type } = message;
+
+    if (!id) {
+      this._send(ws, {
+        type: "error",
+        code: "MISSING_ID",
+        message: 'Message must include an "id" field',
+      });
+      return;
+    }
+
+    // Check authentication
+    const client = this.clients.get(clientId);
+    if (this.token && !client.authenticated && type !== "auth") {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "AUTH_REQUIRED",
+        message: "Authentication required. Send an auth message first.",
+      });
+      return;
+    }
+
+    switch (type) {
+      case "auth":
+        this._handleAuth(clientId, ws, message);
+        break;
+      case "ping":
+        this._send(ws, { id, type: "pong", serverTime: Date.now() });
+        break;
+      case "execute":
+        this._executeCommand(id, ws, message.command, false);
+        break;
+      case "stream":
+        this._executeCommand(id, ws, message.command, true);
+        break;
+      case "cancel":
+        this._cancelRequest(id, ws);
+        break;
+      default:
+        this._send(ws, {
+          id,
+          type: "error",
+          code: "UNKNOWN_TYPE",
+          message: `Unknown message type: ${type}`,
+        });
+    }
+  }
+
+  /** @private */
+  _handleAuth(clientId, ws, message) {
+    const { id, token } = message;
+    const success = token === this.token;
+    const client = this.clients.get(clientId);
+
+    if (success && client) {
+      client.authenticated = true;
+    }
+
+    this._send(ws, {
+      id,
+      type: "auth-result",
+      success,
+      ...(success ? {} : { message: "Invalid token" }),
+    });
+
+    if (!success) {
+      // Disconnect after failed auth
+      setTimeout(() => ws.close(4001, "Authentication failed"), 100);
+    }
+  }
+
+  /** @private */
+  _executeCommand(id, ws, command, stream) {
+    if (!command || typeof command !== "string") {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "INVALID_COMMAND",
+        message: "Command must be a non-empty string",
+      });
+      return;
+    }
+
+    const args = tokenizeCommand(command.trim());
+    if (args.length === 0) {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "INVALID_COMMAND",
+        message: "Empty command",
+      });
+      return;
+    }
+
+    // Block dangerous/interactive commands
+    const baseCmd = args[0];
+    if (BLOCKED_COMMANDS.has(baseCmd)) {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "COMMAND_BLOCKED",
+        message: `Command "${baseCmd}" cannot be executed via WebSocket (interactive or recursive)`,
+      });
+      return;
+    }
+
+    const child = spawn(process.execPath, [BIN_PATH, ...args], {
+      env: {
+        ...process.env,
+        FORCE_COLOR: "0",
+        NO_SPINNER: "1",
+      },
+      stdio: ["pipe", "pipe", "pipe"],
+      windowsHide: true,
+    });
+
+    this.processes.set(id, child);
+    this.emit("command:start", { id, command, stream });
+
+    // Timeout handling
+    const timer = setTimeout(() => {
+      if (this.processes.has(id)) {
+        try {
+          child.kill("SIGTERM");
+        } catch (_err) {
+          // Process may have already exited
+        }
+        this.processes.delete(id);
+        this._send(ws, {
+          id,
+          type: "error",
+          code: "COMMAND_TIMEOUT",
+          message: `Command timed out after ${this.timeout}ms`,
+        });
+      }
+    }, this.timeout);
+
+    if (stream) {
+      // Stream mode: send chunks as they arrive
+      child.stdout.on("data", (data) => {
+        this._send(ws, {
+          id,
+          type: "stream-data",
+          channel: "stdout",
+          data: data.toString("utf8"),
+        });
+      });
+
+      child.stderr.on("data", (data) => {
+        this._send(ws, {
+          id,
+          type: "stream-data",
+          channel: "stderr",
+          data: data.toString("utf8"),
+        });
+      });
+
+      child.on("close", (exitCode) => {
+        clearTimeout(timer);
+        this.processes.delete(id);
+        this._send(ws, {
+          id,
+          type: "stream-end",
+          exitCode: exitCode ?? 1,
+        });
+        this.emit("command:end", { id, exitCode });
+      });
+    } else {
+      // Buffered mode: collect all output then send result
+      const stdoutChunks = [];
+      const stderrChunks = [];
+
+      child.stdout.on("data", (data) => stdoutChunks.push(data));
+      child.stderr.on("data", (data) => stderrChunks.push(data));
+
+      child.on("close", (exitCode) => {
+        clearTimeout(timer);
+        this.processes.delete(id);
+
+        const stdout = Buffer.concat(stdoutChunks).toString("utf8");
+        const stderr = Buffer.concat(stderrChunks).toString("utf8");
+
+        this._send(ws, {
+          id,
+          type: "result",
+          success: exitCode === 0,
+          exitCode: exitCode ?? 1,
+          stdout,
+          stderr,
+        });
+        this.emit("command:end", { id, exitCode });
+      });
+    }
+
+    child.on("error", (err) => {
+      clearTimeout(timer);
+      this.processes.delete(id);
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "SPAWN_ERROR",
+        message: err.message,
+      });
+    });
+  }
+
+  /** @private */
+  _cancelRequest(id, ws) {
+    const child = this.processes.get(id);
+    if (child) {
+      try {
+        child.kill("SIGTERM");
+      } catch (_err) {
+        // Process may have already exited
+      }
+      this.processes.delete(id);
+      this._send(ws, {
+        id,
+        type: "result",
+        success: false,
+        exitCode: -1,
+        stdout: "",
+        stderr: "Cancelled by client",
+      });
+    } else {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "NOT_FOUND",
+        message: `No running command with id "${id}"`,
+      });
+    }
+  }
+
+  /** @private — ping/pong heartbeat to detect dead connections */
+  _startHeartbeat() {
+    this._heartbeatTimer = setInterval(() => {
+      for (const [clientId, client] of this.clients) {
+        if (!client.alive) {
+          client.ws.terminate();
+          this.clients.delete(clientId);
+          this.emit("disconnection", { clientId, reason: "heartbeat timeout" });
+          continue;
+        }
+        client.alive = false;
+        try {
+          client.ws.ping();
+        } catch (_err) {
+          // Connection may be closing
+        }
+      }
+    }, HEARTBEAT_INTERVAL);
+  }
+
+  /** @private — safe JSON send */
+  _send(ws, data) {
+    if (ws.readyState === ws.OPEN) {
+      try {
+        ws.send(JSON.stringify(data));
+      } catch (_err) {
+        // Connection may have just closed
+      }
+    }
+  }
+}