sharkcode 0.2.0 → 0.3.0

package/README.md

@@ -1,49 +1,46 @@
 # Shark Code
 
-> Local First, open-source AI coding agent.
+> Local First, open-source AI coding agent.  
 > A small CLI inspired by Claude Code / OpenCode, with DeepSeek as the default provider.
 
 ## Install
 
-### Global install from npm
-
 ```bash
 npm install -g sharkcode
 ```
 
-Then run:
+Launch interactive mode:
 
 ```bash
-sharkcode "explain this codebase"
-sharkcode "fix the null pointer bug in auth.ts"
-sharkcode "add error handling to the API routes"
+sharkcode
 ```
 
-### Run from source
+Or single-shot mode:
 
 ```bash
-git clone https://github.com/syy-shark/sharkcode.git
-cd sharkcode
-bun install
-bun run start "explain this codebase"
+sharkcode "explain this codebase"
+sharkcode "fix the null pointer bug in auth.ts"
 ```
 
-## Configure API Key
+## Providers
 
-Shark Code reads `DEEPSEEK_API_KEY` from either an environment variable or `~/.sharkcode/config.toml`.
+Shark Code supports multiple providers. Switch between them anytime with `/provider`.
 
-### Option 1: Environment variable
+| Provider | Description | Docs |
+|----------|-------------|------|
+| `deepseek` | DeepSeek 官网 (default) | [platform.deepseek.com](https://platform.deepseek.com) |
+| `ark` | 火山引擎 方舟 Coding Plan | [volcengine.com/activity/codingplan](https://www.volcengine.com/activity/codingplan) |
 
-macOS / Linux:
+## Configure
 
-```bash
-export DEEPSEEK_API_KEY=sk-xxxxxx
-```
+### Option 1: Slash commands (recommended)
 
-Windows PowerShell:
+Start `sharkcode`, then:
 
-```powershell
-$env:DEEPSEEK_API_KEY="sk-xxxxxx"
+```
+◆ /provider ark              # switch to 方舟 Coding Plan
+◆ /key sk-xxxxxxxxxx         # set API key (saved automatically)
+◆ /model ark-code-latest      # optionally change model
 ```
 
 ### Option 2: Config file
@@ -51,37 +48,60 @@ $env:DEEPSEEK_API_KEY="sk-xxxxxx"
 `~/.sharkcode/config.toml`
 
 ```toml
-[api]
+[default]
+provider = "deepseek"   # or "ark"
+
+[providers.deepseek]
+# API key from https://platform.deepseek.com
 key = "sk-xxxxxx"
 model = "deepseek-chat"
-base_url = "https://api.deepseek.com/v1"
-```
 
-## Upgrade
+[providers.ark]
+# API key from https://ark.cn-beijing.volces.com (方舟 Coding Plan)
+key = "sk-xxxxxx"
+model = "ark-code-latest"
+```
 
-When you publish a new version to npm, users can upgrade with:
+### Option 3: Environment variables
 
-```bash
-npm update -g sharkcode
+```powershell
+$env:DEEPSEEK_API_KEY="sk-xxxxxx"   # deepseek provider
+$env:ARK_API_KEY="sk-xxxxxx"         # ark provider
 ```
 
-## Publish
+## Slash Commands
+
+Type `/` commands anytime inside the interactive REPL:
+
+| Command | Description |
+|---------|-------------|
+| `/provider` | show current provider & key status |
+| `/provider <name>` | switch provider (`deepseek` \| `ark`) |
+| `/key <api-key>` | set API key for current provider |
+| `/model <model-id>` | set model for current provider |
+| `/clear` | clear conversation history |
+| `/help` | show command list |
+| `/exit` | quit |
+
+## Run from source
 
 ```bash
-npm login
-npm publish
+git clone https://github.com/syy-shark/sharkcode.git
+cd sharkcode
+bun install
+bun run start
 ```
 
-Every code update is published as a new npm version. Typical flow:
+## Upgrade
 
-1. Update code.
-2. Bump `version` in `package.json`.
-3. Run `npm publish`.
+```bash
+npm update -g sharkcode
+```
 
 ## How It Works
 
-```text
-User input -> Prompt + Tools -> LLM -> Tool execution -> Result -> Repeat
+```
+User input → Prompt + Tools → LLM → Tool execution → Result → Repeat
 ```
 
 Built-in tools:
@@ -97,7 +117,7 @@ Built-in tools:
 
 - Bun + TypeScript
 - Vercel AI SDK
-- DeepSeek API
+- DeepSeek API / 火山引擎 方舟 API
 
 ## License
 

package/dist/cli.mjs

@@ -9,44 +9,100 @@ import { parse } from "smol-toml";
 import { existsSync, readFileSync, mkdirSync, writeFileSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
+var PROVIDERS = {
+  deepseek: {
+    baseURL: "https://api.deepseek.com/v1",
+    defaultModel: "deepseek-chat",
+    label: "DeepSeek \u5B98\u7F51"
+  },
+  ark: {
+    // Coding Plan uses a dedicated endpoint — do NOT use /api/v3 (that's the pay-per-use endpoint)
+    baseURL: "https://ark.cn-beijing.volces.com/api/coding/v3",
+    defaultModel: "ark-code-latest",
+    label: "\u65B9\u821F Coding Plan"
+  }
+};
 var CONFIG_DIR = join(homedir(), ".sharkcode");
 var CONFIG_FILE = join(CONFIG_DIR, "config.toml");
-var DEFAULT_CONFIG = `# Shark Code Configuration
-# https://github.com/syy-ex/sharkcode
-
-[api]
-# Get your API key from https://platform.deepseek.com
-# Or set DEEPSEEK_API_KEY environment variable
-key = ""
-model = "deepseek-chat"
-base_url = "https://api.deepseek.com/v1"
-`;
-function ensureConfigDir() {
+function serializeConfig(mc) {
+  const dp = mc.providers.deepseek;
+  const ap = mc.providers.ark;
+  return [
+    "# Shark Code Configuration",
+    "# https://github.com/syy-shark/sharkcode",
+    "",
+    "[default]",
+    `provider = "${mc.activeProvider}"`,
+    "",
+    "[providers.deepseek]",
+    "# API key from https://platform.deepseek.com",
+    `key = "${dp?.key ?? ""}"`,
+    `model = "${dp?.model ?? PROVIDERS.deepseek.defaultModel}"`,
+    "",
+    "[providers.ark]",
+    "# API key from https://ark.cn-beijing.volces.com (\u65B9\u821F Coding Plan)",
+    `key = "${ap?.key ?? ""}"`,
+    `model = "${ap?.model ?? PROVIDERS.ark.defaultModel}"`,
+    ""
+  ].join("\n");
+}
+function ensureConfig() {
   if (!existsSync(CONFIG_DIR)) {
     mkdirSync(CONFIG_DIR, { recursive: true });
   }
   if (!existsSync(CONFIG_FILE)) {
-    writeFileSync(CONFIG_FILE, DEFAULT_CONFIG, "utf-8");
+    const initial = {
+      activeProvider: "deepseek",
+      providers: {
+        deepseek: { key: "", model: PROVIDERS.deepseek.defaultModel },
+        ark: { key: "", model: PROVIDERS.ark.defaultModel }
+      }
+    };
+    writeFileSync(CONFIG_FILE, serializeConfig(initial), "utf-8");
   }
 }
-function loadConfig() {
-  ensureConfigDir();
+function readMultiConfig() {
+  ensureConfig();
   let toml = {};
   try {
     const raw = readFileSync(CONFIG_FILE, "utf-8");
     toml = parse(raw);
   } catch {
   }
-  const api = toml.api ?? {};
-  const apiKey = process.env.DEEPSEEK_API_KEY || api.key || "";
-  const model = api.model || "deepseek-chat";
-  const baseURL = api.base_url || "https://api.deepseek.com/v1";
-  if (!apiKey) {
-    console.error("\u274C No API key found.");
-    console.error(`   Set DEEPSEEK_API_KEY env var or edit ${CONFIG_FILE}`);
-    process.exit(1);
-  }
-  return { apiKey, model, baseURL };
+  const legacy = toml.api;
+  const providersRaw = toml.providers ?? {};
+  const defaultSection = toml.default ?? {};
+  const deepseekKey = process.env.DEEPSEEK_API_KEY || providersRaw.deepseek?.key || legacy?.key || "";
+  const arkKey = process.env.ARK_API_KEY || providersRaw.ark?.key || "";
+  const activeProvider = defaultSection.provider || (legacy ? "deepseek" : "deepseek");
+  return {
+    activeProvider,
+    providers: {
+      deepseek: {
+        key: deepseekKey,
+        model: providersRaw.deepseek?.model || legacy?.model || PROVIDERS.deepseek.defaultModel
+      },
+      ark: {
+        key: arkKey,
+        model: providersRaw.ark?.model || PROVIDERS.ark.defaultModel
+      }
+    }
+  };
+}
+function saveMultiConfig(mc) {
+  ensureConfig();
+  writeFileSync(CONFIG_FILE, serializeConfig(mc), "utf-8");
+}
+function resolveConfig(mc) {
+  const name = mc.activeProvider;
+  const entry = mc.providers[name];
+  const meta = PROVIDERS[name];
+  return {
+    providerName: name,
+    apiKey: entry?.key || "",
+    model: entry?.model || meta?.defaultModel || "deepseek-chat",
+    baseURL: meta?.baseURL || PROVIDERS.deepseek.baseURL
+  };
 }
 
 // src/agent.ts
@@ -210,12 +266,36 @@ function createProvider(config) {
   const provider = createOpenAI({
     baseURL: config.baseURL,
     apiKey: config.apiKey,
-    name: "deepseek"
+    // ARK requires a non-default name to avoid SDK header overrides
+    name: config.providerName === "ark" ? "ark" : "deepseek"
   });
   return provider.chat(config.model);
 }
 
 // src/agent.ts
+var PURPLE = chalk2.hex("#a855f7");
+function createSpinner() {
+  const frames = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
+  let i = 0;
+  let timer = null;
+  return {
+    start() {
+      process.stdout.write("\n");
+      timer = setInterval(() => {
+        process.stdout.write(
+          `\r${PURPLE(frames[i++ % frames.length])} ${chalk2.gray("thinking...")}`
+        );
+      }, 80);
+    },
+    stop() {
+      if (timer) {
+        clearInterval(timer);
+        timer = null;
+        process.stdout.write("\r\x1B[K");
+      }
+    }
+  };
+}
 function buildSystemPrompt() {
   return `You are Shark Code, an AI coding assistant. You help users with coding tasks by reading, writing, and editing files, and running shell commands.
 
@@ -246,7 +326,14 @@ async function runAgent(messages, config) {
     stopWhen: stepCountIs(30)
   });
   let currentStep = 0;
+  const spinner = createSpinner();
+  let spinnerStopped = false;
+  spinner.start();
   for await (const event of result.fullStream) {
+    if (!spinnerStopped && (event.type === "text-delta" || event.type === "tool-call" || event.type === "error")) {
+      spinner.stop();
+      spinnerStopped = true;
+    }
     switch (event.type) {
       case "text-delta":
         process.stdout.write(event.text);
@@ -282,6 +369,9 @@ async function runAgent(messages, config) {
         break;
     }
   }
+  if (!spinnerStopped) {
+    spinner.stop();
+  }
   process.stdout.write("\n");
   const usage = await result.totalUsage;
   if (usage) {
@@ -314,7 +404,12 @@ function truncate(str, max) {
 }
 
 // src/cli.ts
-var PURPLE = chalk3.hex("#a855f7");
+var PURPLE2 = chalk3.hex("#a855f7");
+var GRAY = chalk3.gray;
+var YELLOW = chalk3.yellow;
+var GREEN = chalk3.green;
+var RED = chalk3.red;
+var CYAN = chalk3.cyan;
 var GLYPHS = {
   S: [[0, 1, 1, 1, 1], [1, 0, 0, 0, 0], [0, 1, 1, 1, 0], [0, 0, 0, 0, 1], [1, 1, 1, 1, 0]],
   H: [[1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 1, 1, 1, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1]],
@@ -338,11 +433,137 @@ function renderWord(word, padLeft = 2) {
       }
       if (i < letters.length - 1) line += "  ";
     }
-    rows.push(PURPLE(line));
+    rows.push(PURPLE2(line));
   }
   return rows;
 }
 var BANNER = ["", ...renderWord("shark", 2), "", ...renderWord("code", 8), ""].join("\n");
+function printSlashHelp() {
+  const pad = (s, n) => s + " ".repeat(Math.max(0, n - s.length));
+  console.log(PURPLE2("\n  \u25C6 Slash Commands\n"));
+  const cmds = [
+    ["/provider", "show current provider & list options"],
+    ["/provider <name>", "switch provider  (deepseek | ark)"],
+    ["/key <api-key>", "set API key for current provider"],
+    ["/model <model-id>", "set model for current provider"],
+    ["/clear", "clear conversation history"],
+    ["/help", "show this menu"],
+    ["/exit", "quit"]
+  ];
+  for (const [cmd, desc] of cmds) {
+    console.log("  " + PURPLE2(pad(cmd, 26)) + GRAY(desc));
+  }
+  console.log();
+}
+function handleSlash(input, multiConfig) {
+  const parts = input.trim().split(/\s+/);
+  const cmd = parts[0].toLowerCase();
+  const arg = parts.slice(1).join(" ");
+  if (cmd === "/" || cmd === "/help") {
+    printSlashHelp();
+    return { multiConfig, config: resolveConfig(multiConfig) };
+  }
+  if (cmd === "/exit" || cmd === "/quit") {
+    console.log(GRAY("Bye! \u{1F988}"));
+    return { multiConfig, config: resolveConfig(multiConfig), exit: true };
+  }
+  if (cmd === "/clear") {
+    console.log(GRAY("  \u2713 Conversation cleared."));
+    return { multiConfig, config: resolveConfig(multiConfig), clearHistory: true };
+  }
+  if (cmd === "/provider") {
+    if (!arg) {
+      const current = multiConfig.activeProvider;
+      console.log(PURPLE2("\n  \u25C6 Providers\n"));
+      for (const [name2, meta] of Object.entries(PROVIDERS)) {
+        const entry = multiConfig.providers[name2];
+        const active = name2 === current;
+        const hasKey = !!entry?.key;
+        const marker = active ? PURPLE2("\u25B6") : " ";
+        const keyStatus = hasKey ? GREEN("\u2713 key set") : YELLOW("\u2717 no key");
+        console.log(
+          `  ${marker} ${active ? PURPLE2(name2) : GRAY(name2)}  ${GRAY(meta.label)}  ${keyStatus}` + (active ? `  ${GRAY("model: " + (entry?.model ?? meta.defaultModel))}` : "")
+        );
+      }
+      console.log(
+        `
+  ${GRAY("Usage:")} ${PURPLE2("/provider deepseek")} ${GRAY("or")} ${PURPLE2("/provider ark")}
+`
+      );
+      return { multiConfig, config: resolveConfig(multiConfig) };
+    }
+    const name = arg.toLowerCase();
+    if (!PROVIDERS[name]) {
+      console.log(RED(`  \u2717 Unknown provider: "${name}". Available: ${Object.keys(PROVIDERS).join(", ")}`));
+      return { multiConfig, config: resolveConfig(multiConfig) };
+    }
+    const updated = { ...multiConfig, activeProvider: name };
+    saveMultiConfig(updated);
+    const newConfig = resolveConfig(updated);
+    console.log(
+      GREEN(`
+  \u2713 Switched to ${PROVIDERS[name].label}`) + GRAY(` (${name}) \u2014 model: ${newConfig.model}`)
+    );
+    if (!newConfig.apiKey) {
+      console.log(YELLOW(`  \u26A0 No API key set. Use /key <your-key> to configure it.
+`));
+    } else {
+      console.log();
+    }
+    return { multiConfig: updated, config: newConfig };
+  }
+  if (cmd === "/key") {
+    if (!arg) {
+      console.log(YELLOW(`  Usage: /key <your-api-key>`));
+      return { multiConfig, config: resolveConfig(multiConfig) };
+    }
+    const name = multiConfig.activeProvider;
+    const updated = {
+      ...multiConfig,
+      providers: {
+        ...multiConfig.providers,
+        [name]: {
+          ...multiConfig.providers[name] ?? { model: PROVIDERS[name]?.defaultModel ?? "" },
+          key: arg
+        }
+      }
+    };
+    saveMultiConfig(updated);
+    const masked = arg.slice(0, 6) + "\u2022".repeat(Math.max(0, arg.length - 10)) + arg.slice(-4);
+    console.log(GREEN(`  \u2713 API key saved for ${name}: ${GRAY(masked)}
+`));
+    return { multiConfig: updated, config: resolveConfig(updated) };
+  }
+  if (cmd === "/model") {
+    if (!arg) {
+      const name2 = multiConfig.activeProvider;
+      const current = multiConfig.providers[name2]?.model ?? PROVIDERS[name2]?.defaultModel;
+      console.log(GRAY(`  Current model: `) + PURPLE2(current ?? "unknown"));
+      console.log(GRAY(`  Usage: /model <model-id>
+`));
+      return { multiConfig, config: resolveConfig(multiConfig) };
+    }
+    const name = multiConfig.activeProvider;
+    const updated = {
+      ...multiConfig,
+      providers: {
+        ...multiConfig.providers,
+        [name]: {
+          ...multiConfig.providers[name] ?? { key: "" },
+          model: arg
+        }
+      }
+    };
+    saveMultiConfig(updated);
+    console.log(GREEN(`  \u2713 Model set to ${PURPLE2(arg)} for ${name}
+`));
+    return { multiConfig: updated, config: resolveConfig(updated) };
+  }
+  console.log(RED(`  \u2717 Unknown command: "${cmd}"`));
+  console.log(GRAY(`  Type /help to see available commands.
+`));
+  return { multiConfig, config: resolveConfig(multiConfig) };
+}
 async function readLine(prompt) {
   return new Promise((resolve4) => {
     const rl = readline2.createInterface({ input: process.stdin, output: process.stdout });
@@ -357,60 +578,93 @@ async function readLine(prompt) {
     });
   });
 }
+function statusLine(config) {
+  const providerLabel = PROVIDERS[config.providerName]?.label ?? config.providerName;
+  return PURPLE2("  \u25C6") + GRAY(` ${providerLabel}`) + CYAN(` [${config.model}]`) + GRAY("   type /help for commands\n");
+}
 async function main() {
   const args = process.argv.slice(2);
   if (args[0] === "--help" || args[0] === "-h") {
     console.log(BANNER);
-    console.log(PURPLE("  Usage:"));
-    console.log("    " + PURPLE("sharkcode") + chalk3.gray("                   \u2014 interactive mode"));
-    console.log("    " + PURPLE("sharkcode") + chalk3.yellow(' "prompt"') + chalk3.gray("         \u2014 single-shot mode\n"));
+    console.log(PURPLE2("  Usage:"));
+    console.log("    " + PURPLE2("sharkcode") + GRAY("                   \u2014 interactive mode"));
+    console.log("    " + PURPLE2("sharkcode") + YELLOW(' "prompt"') + GRAY("         \u2014 single-shot mode"));
+    console.log("\n  " + PURPLE2("Slash commands (in interactive mode):"));
+    printSlashHelp();
     return;
   }
   if (args[0] === "--version" || args[0] === "-v") {
-    console.log("sharkcode v0.2.0");
+    console.log("sharkcode v0.3.0");
     return;
   }
-  const config = loadConfig();
   if (args.length > 0) {
-    console.log(PURPLE("\n\u{1F988} SharkCode") + chalk3.gray(` | model: ${config.model}
+    const mc = readMultiConfig();
+    const config2 = resolveConfig(mc);
+    if (!config2.apiKey) {
+      console.error(RED("\u274C No API key found for provider: " + config2.providerName));
+      console.error(GRAY("   Run sharkcode in interactive mode and use /key <your-key>"));
+      process.exit(1);
+    }
+    console.log(PURPLE2("\n\u{1F988} SharkCode") + GRAY(` | ${PROVIDERS[config2.providerName]?.label ?? config2.providerName} | model: ${config2.model}
 `));
     const messages2 = [{ role: "user", content: args.join(" ") }];
-    await runAgent(messages2, config);
+    await runAgent(messages2, config2);
     return;
   }
   console.log(BANNER);
-  console.log(
-    PURPLE("  \u25C6") + chalk3.gray(` model: ${config.model}`) + chalk3.gray('   type "exit" to quit\n')
-  );
+  let multiConfig = readMultiConfig();
+  let config = resolveConfig(multiConfig);
+  console.log(statusLine(config));
+  if (!config.apiKey) {
+    console.log(
+      YELLOW("  \u26A0 No API key configured for ") + PURPLE2(PROVIDERS[config.providerName]?.label ?? config.providerName) + YELLOW(".")
+    );
+    console.log(GRAY("    Use /key <your-api-key> to set it, or /provider to switch.\n"));
+  }
   let messages = [];
   while (true) {
-    const input = await readLine(PURPLE("\n\u25C6 "));
+    const input = await readLine(PURPLE2("\n\u25C6 "));
     if (input === null) {
-      console.log(chalk3.gray("\nBye! \u{1F988}"));
+      console.log(GRAY("\nBye! \u{1F988}"));
       break;
     }
     const trimmed = input.trim();
     if (!trimmed) continue;
-    if (trimmed === "exit" || trimmed === "quit" || trimmed === "/exit") {
-      console.log(chalk3.gray("Bye! \u{1F988}"));
+    if (trimmed === "exit" || trimmed === "quit") {
+      console.log(GRAY("Bye! \u{1F988}"));
       break;
     }
+    if (trimmed.startsWith("/")) {
+      const result = handleSlash(trimmed, multiConfig);
+      if (!result) continue;
+      multiConfig = result.multiConfig;
+      config = result.config;
+      if (result.clearHistory) messages = [];
+      if (result.exit) break;
+      continue;
+    }
+    if (!config.apiKey) {
+      console.log(
+        YELLOW("\n  \u26A0 No API key for ") + PURPLE2(PROVIDERS[config.providerName]?.label ?? config.providerName) + YELLOW(". Set it with /key <your-api-key>\n")
+      );
+      continue;
+    }
     messages.push({ role: "user", content: trimmed });
     try {
       messages = await runAgent(messages, config);
     } catch (err) {
       const error = err;
-      console.error(chalk3.red(`
+      console.error(RED(`
 \u274C Error: ${error.message}`));
       if (error.message?.includes("401") || error.message?.includes("Unauthorized")) {
-        console.error(chalk3.yellow("   Check your API key in ~/.sharkcode/config.toml"));
+        console.error(YELLOW("   Check your API key with /key <your-api-key>"));
       }
       messages = messages.slice(0, -1);
     }
   }
 }
 main().catch((err) => {
-  console.error(chalk3.red(`
+  console.error(RED(`
 \u274C Fatal: ${err.message}`));
   process.exit(1);
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sharkcode",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Local First, open-source AI Coding Agent",
   "type": "module",
   "bin": {