@mathew-macharia/black-mamba 1.0.1 → 1.0.3

package/README.md

@@ -8,7 +8,7 @@ Requires Node.js 18 or newer.
 npm install -g @mathew-macharia/black-mamba
 ```
 
-Run `bm` in the folder that holds your data. That folder is where it reads `accounts/`, `.env`, and `sha3_wasm_bg.wasm`.
+Run `bm` in the folder that holds your data. That folder is where it reads `accounts/` and `.env`. The proof-of-work file `sha3_wasm_bg.wasm` ships inside the package.
 
 ```
 accounts/<name>/api-key.txt

package/docs/config.md

@@ -37,4 +37,4 @@ Set to `1` or `true` to enable the logger's debug method. Any other value leaves
 
 DeepSeek web origin. Default `https://chat.deepseek.com`. Trailing slashes are removed. Login, chat create, the proof-of-work challenge, and completion are called under this origin at `/api/v0/users/current`, `/api/v0/chat_session/create`, `/api/v0/chat/create_pow_challenge`, and `/api/v0/chat/completion`.
 
-The proof-of-work wasm file is separate. If `sha3_wasm_bg.wasm` is missing from the working directory, it is downloaded from DeepSeek's static file host and saved there. That download URL does not follow `DEEPSEEK_BASE_URL`.
+The proof-of-work file `sha3_wasm_bg.wasm` lives in `src/deepseek/` and is published with `src`. If that installed file is missing, startup downloads it from DeepSeek's static file host and saves it there. That download URL does not follow `DEEPSEEK_BASE_URL`.

package/docs/install.md

@@ -6,12 +6,11 @@ Requires Node.js 18 or newer.
 npm install -g @mathew-macharia/black-mamba
 ```
 
-The command is `bm`. Run it in the directory that holds your data. That working directory (`process.cwd()`) is where `bm` reads `accounts/`, `.env`, and `sha3_wasm_bg.wasm`.
+The command is `bm`. Run it in the directory that holds your data. That working directory (`process.cwd()`) is where `bm` reads `accounts/` and `.env`. The proof-of-work file `sha3_wasm_bg.wasm` is installed with the package.
 
 ```
 <data-directory>/
   .env                  optional
-  sha3_wasm_bg.wasm     created on first start if missing
   accounts/
     <name>/
       api-key.txt
@@ -36,8 +35,6 @@ From the data directory:
 bm
 ```
 
-If `sha3_wasm_bg.wasm` is missing from that directory, startup downloads it and saves it there. The terminal shows two `[WASM]` lines: one when the download starts, and one when the file is saved.
-
 When the server is up, stdout has:
 
 ```

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mathew-macharia/black-mamba",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "OpenAI-compatible proxy that maps chat sessions onto DeepSeek web chat",
   "type": "module",
   "main": "index.js",

package/src/config.js

@@ -2,7 +2,9 @@
 
 import "dotenv/config";
 import path from "node:path";
+import { fileURLToPath } from "node:url";
 
+const SRC_DIR = path.dirname(fileURLToPath(import.meta.url));
 const ROOT_DIR = process.cwd();
 
 function readNumber(raw, fallback) {
@@ -38,7 +40,7 @@ export function loadConfig(env = process.env, rootDir = ROOT_DIR) {
       chatCompletionUrl: `${deepseekBaseUrl}/api/v0/chat/completion`,
       powTargetPath: "/api/v0/chat/completion",
       wasmUrl: "https://fe-static.deepseek.com/chat/static/sha3_wasm_bg.7b9ca65ddd.wasm",
-      wasmPath: path.join(rootDir, "sha3_wasm_bg.wasm"),
+      wasmPath: path.join(SRC_DIR, "deepseek", "sha3_wasm_bg.wasm"),
       healthCheckIntervalMs: 30 * 60 * 1000,
       estimatedWafTtlMs: 4 * 24 * 60 * 60 * 1000,
       estimatedSessionTtlMs: 30 * 24 * 60 * 60 * 1000,

package/src/pi/prompt.js

@@ -19,12 +19,19 @@ function primaryToolExample(tools) {
 
 function toolCallRules(tools) {
   const { primaryTool, exampleParam } = primaryToolExample(tools);
-  return `1. You do not have direct terminal, shell, or file system access. To inspect files, run commands, or edit code, you request execution by emitting a DSML block:
+  return `1. You do not have direct terminal, shell, or file system access. To inspect files, run commands, or edit code, you request execution by emitting a DSML block. The format must match exactly like this:
 <｜DSML｜tool_calls>
 <｜DSML｜invoke name="${primaryTool}">
 <｜DSML｜parameter ${exampleParam}</｜DSML｜parameter>
 </｜DSML｜invoke>
 </｜DSML｜tool_calls>
+
+Explanation:
+<｜DSML｜tool_calls> opens the block.
+<｜DSML｜invoke name="${primaryTool}"> names the tool. ${primaryTool} is the tool name.
+<｜DSML｜parameter name="..." string="true"> holds the value, and </｜DSML｜parameter> closes it.
+</｜DSML｜invoke> then </｜DSML｜tool_calls> close the block.
+
 2. Output EXACTLY ONE DSML block at the end of your response and end your turn immediately. Pi will execute it locally on the user's machine and return the result in a DSML result block.
 3. Tool requests must strictly use the DSML block. Do not format tool calls with any other syntax.
 4. If the user request does not require any tools, respond directly in normal text.`;
@@ -78,7 +85,15 @@ Pi is connected to chat.deepseek and sends and listens to https://chat.deepseek.
 }
 
 export function formatToolTurnDirective(_tools = []) {
-  return "[Tool output received from user's machine. To take another action, output a DSML block and end turn. If complete, respond directly to the user.]";
+  return `[Tool output received from user's machine. To take another action, output a DSML block and end turn. If complete, respond directly to the user.]
+
+REMINDER: When calling a tool, four tags must be there, in this order:
+
+<｜DSML｜tool_calls>
+<｜DSML｜invoke name="TOOL">
+<｜DSML｜parameter name="PARAM" string="true">VALUE</｜DSML｜parameter>
+</｜DSML｜invoke>
+</｜DSML｜tool_calls>`;
 }
 
 export const USER_TOOL_REMINDER =