sharkcode 0.2.1 → 0.3.1

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

@@ -3,50 +3,107 @@
 // src/cli.ts
 import chalk3 from "chalk";
 import * as readline2 from "readline";
+import { select, input } from "@inquirer/prompts";
 
 // src/config.ts
 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,7 +267,8 @@ 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);
 }
@@ -348,6 +406,11 @@ function truncate(str, max) {
 
 // src/cli.ts
 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]],
@@ -376,6 +439,80 @@ function renderWord(word, padLeft = 2) {
   return rows;
 }
 var BANNER = ["", ...renderWord("shark", 2), "", ...renderWord("code", 8), ""].join("\n");
+async function showCommandMenu(multiConfig) {
+  console.log();
+  try {
+    const action = await select({
+      message: PURPLE2("\u25C6 \u9009\u62E9\u64CD\u4F5C"),
+      choices: [
+        { name: "\u{1F50C}  \u5207\u6362 / \u914D\u7F6E Provider", value: "provider" },
+        { name: "\u{1F5D1}\uFE0F  \u6E05\u7A7A\u5BF9\u8BDD\u5386\u53F2", value: "clear" },
+        { name: "\u{1F6AA}  \u9000\u51FA", value: "exit" }
+      ]
+    });
+    switch (action) {
+      case "provider":
+        return showSetupFlow(multiConfig);
+      case "clear":
+        console.log(GRAY("\n  \u2713 \u5BF9\u8BDD\u5DF2\u6E05\u7A7A\n"));
+        return { multiConfig, config: resolveConfig(multiConfig), clearHistory: true };
+      case "exit":
+        console.log(GRAY("\nBye! \u{1F988}"));
+        return { multiConfig, config: resolveConfig(multiConfig), exit: true };
+    }
+  } catch {
+    console.log(GRAY("\n  \u53D6\u6D88\n"));
+  }
+  return { multiConfig, config: resolveConfig(multiConfig) };
+}
+async function showSetupFlow(multiConfig) {
+  console.log();
+  try {
+    const providerChoices = Object.entries(PROVIDERS).map(([id, meta]) => {
+      const hasKey = !!multiConfig.providers[id]?.key;
+      const badge = hasKey ? GREEN("\u2713 \u5DF2\u914D\u7F6E") : YELLOW("\u2717 \u672A\u914D\u7F6E");
+      return { name: `${meta.label}   ${badge}`, value: id };
+    });
+    const selectedProvider = await select({
+      message: PURPLE2("\u25C6 \u9009\u62E9 Provider"),
+      choices: providerChoices,
+      default: multiConfig.activeProvider
+    });
+    const currentKey = multiConfig.providers[selectedProvider]?.key ?? "";
+    const hint = currentKey ? GRAY("(\u56DE\u8F66\u4FDD\u7559  " + currentKey.slice(0, 6) + "\u2022\u2022\u2022)") : GRAY("(\u5FC5\u586B)");
+    const rawKey = await input({
+      message: PURPLE2("\u25C6 API Key ") + hint,
+      default: currentKey || void 0
+    });
+    const newKey = rawKey.trim() || currentKey;
+    let updated = { ...multiConfig, activeProvider: selectedProvider };
+    if (newKey) {
+      updated = {
+        ...updated,
+        providers: {
+          ...updated.providers,
+          [selectedProvider]: {
+            model: PROVIDERS[selectedProvider]?.defaultModel ?? "",
+            ...updated.providers[selectedProvider] ?? {},
+            key: newKey
+          }
+        }
+      };
+    }
+    saveMultiConfig(updated);
+    const newConfig = resolveConfig(updated);
+    const keyMsg = newKey && newKey !== currentKey ? "\uFF0CAPI Key \u5DF2\u4FDD\u5B58" : "";
+    console.log(GREEN(`
+  \u2713 \u5DF2\u5207\u6362\u5230 ${PROVIDERS[selectedProvider].label}${keyMsg}`) + "\n");
+    if (!newConfig.apiKey) {
+      console.log(YELLOW("  \u26A0 \u8FD8\u672A\u586B\u5199 API Key\uFF0C\u65E0\u6CD5\u53D1\u9001\u6D88\u606F\n"));
+    }
+    return { multiConfig: updated, config: newConfig };
+  } catch {
+    console.log(GRAY("\n  \u53D6\u6D88\n"));
+    return { multiConfig, config: resolveConfig(multiConfig) };
+  }
+}
 async function readLine(prompt) {
   return new Promise((resolve4) => {
     const rl = readline2.createInterface({ input: process.stdin, output: process.stdout });
@@ -390,60 +527,95 @@ async function readLine(prompt) {
     });
   });
 }
+function statusLine(config) {
+  const label = PROVIDERS[config.providerName]?.label ?? config.providerName;
+  return PURPLE2("  \u25C6") + GRAY(` ${label}`) + CYAN(`  [${config.model}]`) + GRAY("   \u8F93\u5165 / \u8C03\u51FA\u6307\u4EE4\u83DC\u5355\n");
+}
 async function main() {
   const args = process.argv.slice(2);
   if (args[0] === "--help" || args[0] === "-h") {
     console.log(BANNER);
     console.log(PURPLE2("  Usage:"));
-    console.log("    " + PURPLE2("sharkcode") + chalk3.gray("                   \u2014 interactive mode"));
-    console.log("    " + PURPLE2("sharkcode") + chalk3.yellow(' "prompt"') + chalk3.gray("         \u2014 single-shot mode\n"));
+    console.log("    " + PURPLE2("sharkcode") + GRAY("          \u2014 \u4EA4\u4E92\u6A21\u5F0F\uFF08\u76F4\u63A5\u542F\u52A8\uFF09"));
+    console.log("    " + PURPLE2("sharkcode") + YELLOW(' "prompt"') + GRAY("  \u2014 \u5355\u6B21\u6267\u884C"));
+    console.log(GRAY("\n  \u4EA4\u4E92\u6A21\u5F0F\u5185\u8F93\u5165 / \u8C03\u51FA\u6307\u4EE4\u83DC\u5355\n"));
     return;
   }
   if (args[0] === "--version" || args[0] === "-v") {
-    console.log("sharkcode v0.2.0");
+    console.log("sharkcode v0.3.1");
     return;
   }
-  const config = loadConfig();
   if (args.length > 0) {
-    console.log(PURPLE2("\n\u{1F988} SharkCode") + chalk3.gray(` | model: ${config.model}
-`));
-    const messages2 = [{ role: "user", content: args.join(" ") }];
-    await runAgent(messages2, config);
+    const mc = readMultiConfig();
+    const config2 = resolveConfig(mc);
+    if (!config2.apiKey) {
+      console.error(RED("\u274C \u672A\u914D\u7F6E API Key\u3002\u8BF7\u5148\u8FD0\u884C sharkcode \u5E76\u8F93\u5165 / \u914D\u7F6E Provider\u3002"));
+      process.exit(1);
+    }
+    console.log(
+      PURPLE2("\n\u{1F988} SharkCode") + GRAY(` | ${PROVIDERS[config2.providerName]?.label ?? config2.providerName} | ${config2.model}
+`)
+    );
+    await runAgent([{ role: "user", content: args.join(" ") }], config2);
     return;
   }
   console.log(BANNER);
-  console.log(
-    PURPLE2("  \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 \u5C1A\u672A\u914D\u7F6E API Key\u3002") + GRAY("\u8F93\u5165 / \u7136\u540E\u9009\u62E9\u300C\u5207\u6362 / \u914D\u7F6E Provider\u300D\n")
+    );
+  }
   let messages = [];
   while (true) {
-    const input = await readLine(PURPLE2("\n\u25C6 "));
-    if (input === null) {
-      console.log(chalk3.gray("\nBye! \u{1F988}"));
+    const raw = await readLine(PURPLE2("\n\u25C6 "));
+    if (raw === null) {
+      console.log(GRAY("\nBye! \u{1F988}"));
       break;
     }
-    const trimmed = input.trim();
+    const trimmed = raw.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 === "/") {
+      const r = await showCommandMenu(multiConfig);
+      multiConfig = r.multiConfig;
+      config = r.config;
+      if (r.clearHistory) messages = [];
+      if (r.exit) break;
+      continue;
+    }
+    if (trimmed === "/provider" || trimmed.startsWith("/provider ") || trimmed === "/model" || trimmed.startsWith("/model ") || trimmed === "/key" || trimmed.startsWith("/key ")) {
+      const r = await showSetupFlow(multiConfig);
+      multiConfig = r.multiConfig;
+      config = r.config;
+      if (r.exit) break;
+      continue;
+    }
+    if (trimmed.startsWith("/")) {
+      console.log(GRAY("  \u672A\u77E5\u547D\u4EE4\u3002\u8F93\u5165 / \u8C03\u51FA\u6307\u4EE4\u83DC\u5355\n"));
+      continue;
+    }
+    if (!config.apiKey) {
+      console.log(YELLOW("  \u26A0 \u8FD8\u672A\u586B\u5199 API Key\u3002\u8F93\u5165 / \u2192 \u5207\u6362 / \u914D\u7F6E Provider\n"));
+      continue;
+    }
     messages.push({ role: "user", content: trimmed });
     try {
       messages = await runAgent(messages, config);
     } catch (err) {
-      const error = err;
-      console.error(chalk3.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"));
-      }
-      messages = messages.slice(0, -1);
+      console.error(RED(`
+\u274C ${String(err)}
+`));
+      messages.pop();
     }
   }
 }
 main().catch((err) => {
-  console.error(chalk3.red(`
-\u274C Fatal: ${err.message}`));
+  console.error(chalk3.red(`Fatal: ${String(err)}`));
   process.exit(1);
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sharkcode",
-  "version": "0.2.1",
+  "version": "0.3.1",
   "description": "Local First, open-source AI Coding Agent",
   "type": "module",
   "bin": {
@@ -26,6 +26,7 @@
   },
   "dependencies": {
     "@ai-sdk/openai": "^3.0.48",
+    "@inquirer/prompts": "^8.3.2",
     "ai": "^6.0.141",
     "chalk": "^5.6.2",
     "smol-toml": "^1.6.1",