@tiens.nguyen/gonext-local-worker 1.0.8 → 1.0.9

package/gonext-local-worker.mjs

@@ -1,18 +1,87 @@
 #!/usr/bin/env node
 /**
- * Polls GoNext Cloud API for pending local-LLM chat jobs, runs them against the
- * URLs embedded in the job payload (your LAN Ollama / OpenAI-compatible server),
- * then PATCHes completion back to the API.
- *
- * Usage (from api/):
- *   export GONEXT_API_BASE=https://xxxx.execute-api....amazonaws.com
- *   export GONEXT_WORKER_KEY=<plaintext secret from Settings → Worker API key>
- *   node scripts/local-llm-worker.mjs
- *
- * Requires Node 18+ (global fetch). Uses the OpenAI SDK from this package.
+ * GoNext local worker:
+ * - `gonext-local-worker set <workerKey> [--api-base URL] [--poll-ms 1500]`
+ *   writes ~/.gonext/worker.env
+ * - `gonext-local-worker` starts polling loop
  */
+import { mkdir, readFile, writeFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import dotenv from "dotenv";
 import OpenAI from "openai";
 
+const ENV_FILE = join(homedir(), ".gonext", "worker.env");
+dotenv.config({ path: ENV_FILE });
+dotenv.config();
+
+const args = process.argv.slice(2);
+
+function printHelp() {
+  console.log(`
+gonext-local-worker
+
+Usage:
+  gonext-local-worker
+  gonext-local-worker set <workerKey> [--api-base <url>] [--poll-ms <ms>]
+
+Examples:
+  gonext-local-worker set abc123 --api-base https://hwohu56e8d.execute-api.ap-southeast-1.amazonaws.com
+  gonext-local-worker
+`);
+}
+
+function parseFlag(name) {
+  const idx = args.indexOf(name);
+  if (idx >= 0 && args[idx + 1]) {
+    return args[idx + 1];
+  }
+  return undefined;
+}
+
+async function setConfig() {
+  const workerKey = args[1]?.trim();
+  if (!workerKey) {
+    console.error("Missing worker key. Usage: gonext-local-worker set <workerKey>");
+    process.exit(1);
+  }
+  const currentRaw = await readFile(ENV_FILE, "utf8").catch(() => "");
+  const current = dotenv.parse(currentRaw);
+  const apiBaseFromFlag = parseFlag("--api-base");
+  const pollMsFromFlag = parseFlag("--poll-ms");
+  const next = {
+    GONEXT_API_BASE: (
+      apiBaseFromFlag ??
+      current.GONEXT_API_BASE ??
+      process.env.GONEXT_API_BASE ??
+      ""
+    ).replace(/\/+$/, ""),
+    GONEXT_WORKER_KEY: workerKey,
+    GONEXT_POLL_MS:
+      pollMsFromFlag ?? current.GONEXT_POLL_MS ?? process.env.GONEXT_POLL_MS ?? "1500",
+  };
+  await mkdir(join(homedir(), ".gonext"), { recursive: true });
+  await writeFile(
+    ENV_FILE,
+    `GONEXT_API_BASE=${next.GONEXT_API_BASE}\nGONEXT_WORKER_KEY=${next.GONEXT_WORKER_KEY}\nGONEXT_POLL_MS=${next.GONEXT_POLL_MS}\n`,
+    "utf8"
+  );
+  console.log(`Saved ${ENV_FILE}`);
+  if (!next.GONEXT_API_BASE) {
+    console.log("Tip: set API base too: gonext-local-worker set <workerKey> --api-base <https-url>");
+  }
+  return;
+}
+
+if (args.includes("--help") || args.includes("-h")) {
+  printHelp();
+  process.exit(0);
+}
+if (args[0] === "set") {
+  await setConfig();
+  process.exit(0);
+}
+
 const apiBase = (process.env.GONEXT_API_BASE ?? "").replace(/\/+$/, "");
 const workerKey = process.env.GONEXT_WORKER_KEY ?? "";
 const pollMs = Number(process.env.GONEXT_POLL_MS ?? "1500") || 1500;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tiens.nguyen/gonext-local-worker",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "Polls GoNext cloud API for async local LLM jobs and runs them against Ollama/OpenAI-compatible servers on this Mac",
   "type": "module",
   "license": "MIT",