open-agents-ai 0.184.69 → 0.184.70

package/dist/index.js

@@ -28399,21 +28399,35 @@ ${transcript}`
       thinking;
       /** Abort signal — set by the runner so /stop can cancel in-flight requests */
       _abortSignal = null;
+      /** Multi-key pool — round-robin rotation per request for load distribution */
+      _keyPool = [];
+      _keyIndex = 0;
       constructor(baseUrl, model, apiKey, thinking) {
         this.baseUrl = normalizeBaseUrl(baseUrl);
         this.model = model;
         this.apiKey = apiKey ?? "";
         this.thinking = thinking ?? true;
       }
+      /** Set multiple API keys for round-robin rotation per request */
+      setKeyPool(keys) {
+        this._keyPool = keys.filter((k) => k.length > 0);
+        this._keyIndex = 0;
+      }
       /** Set the abort signal from the runner (called at run start) */
       setAbortSignal(signal) {
         this._abortSignal = signal;
       }
-      /** Build auth headers — all providers use standard Bearer token auth. */
+      /** Build auth headers — all providers use standard Bearer token auth.
+       *  When a key pool is set, round-robins through keys per request. */
       authHeaders() {
         const headers = { "Content-Type": "application/json" };
-        if (this.apiKey) {
-          headers["Authorization"] = `Bearer ${this.apiKey}`;
+        let key = this.apiKey;
+        if (this._keyPool.length > 0) {
+          key = this._keyPool[this._keyIndex % this._keyPool.length];
+          this._keyIndex++;
+        }
+        if (key) {
+          headers["Authorization"] = `Bearer ${key}`;
         }
         return headers;
       }
@@ -49969,6 +49983,85 @@ async function handleEndpoint(arg, ctx, local = false) {
     await handleSponsoredEndpoint(ctx, local);
     return;
   }
+  if (arg.startsWith("add ")) {
+    const addArg = arg.slice(4).replace(/\u2014/g, "--").replace(/\u2013/g, "--");
+    const addParts = addArg.split(/\s+/);
+    const addUrl = addParts[0];
+    const addAuthIdx = addParts.indexOf("--auth") !== -1 ? addParts.indexOf("--auth") : addParts.indexOf("-auth");
+    const addKey = addAuthIdx !== -1 && addParts[addAuthIdx + 1] ? addParts[addAuthIdx + 1] : "";
+    if (!addKey) {
+      renderError("Usage: /endpoint add <url> --auth <key>");
+      return;
+    }
+    const normalizedAddUrl = normalizeBaseUrl(addUrl);
+    const suffix = addKey.slice(-4);
+    const provider2 = detectProvider(addUrl);
+    const settings = resolveSettings(ctx.repoRoot);
+    const pool = settings.endpointKeys ?? [];
+    if (pool.some((k) => k.suffix === suffix)) {
+      renderWarning(`Key ending ...${suffix} is already in the pool.`);
+      return;
+    }
+    pool.push({ key: addKey, suffix });
+    const backendType2 = provider2.id === "ollama" ? "ollama" : "vllm";
+    ctx.setEndpoint(normalizedAddUrl, backendType2, addKey);
+    setConfigValue("backendUrl", normalizedAddUrl);
+    setConfigValue("backendType", backendType2);
+    setConfigValue("apiKey", addKey);
+    ctx.saveSettings({ backendUrl: normalizedAddUrl, backendType: backendType2, apiKey: addKey, endpointKeys: pool });
+    saveGlobalSettings({ endpointKeys: pool });
+    if (ctx.setKeyPool)
+      ctx.setKeyPool(pool.map((k) => k.key));
+    recordUsage("endpoint", normalizedAddUrl, {
+      meta: { provider: provider2.label, backendType: backendType2, authHint: addKey.slice(0, 4) + "...", authKey: addKey }
+    });
+    renderInfo(`Added key ...${suffix} to pool (${pool.length} key${pool.length > 1 ? "s" : ""} total)`);
+    for (const k of pool) {
+      process.stdout.write(`    ${c2.dim("\u25CF")} ${provider2.label}-${k.suffix}  ${c2.dim(k.key.slice(0, 4) + "..." + k.key.slice(-4))}
+`);
+    }
+    process.stdout.write("\n");
+    return;
+  }
+  if (arg === "keys" || arg === "pool") {
+    const settings = resolveSettings(ctx.repoRoot);
+    const pool = settings.endpointKeys ?? [];
+    if (pool.length === 0) {
+      renderInfo("No key pool configured. Use /endpoint add <url> --auth <key> to add keys.");
+      return;
+    }
+    const provider2 = detectProvider(ctx.config.backendUrl);
+    renderInfo(`Key pool for ${c2.bold(ctx.config.backendUrl)} (${pool.length} key${pool.length > 1 ? "s" : ""}):`);
+    for (const k of pool) {
+      process.stdout.write(`    ${c2.green("\u25CF")} ${provider2.label}-${k.suffix}  ${c2.dim(k.key.slice(0, 4) + "..." + k.key.slice(-4))}
+`);
+    }
+    process.stdout.write(`
+  ${c2.dim("Remove: /endpoint remove <suffix>")}
+
+`);
+    return;
+  }
+  if (arg.startsWith("remove ") || arg.startsWith("rm ")) {
+    const suffix = arg.split(/\s+/)[1];
+    const settings = resolveSettings(ctx.repoRoot);
+    const pool = settings.endpointKeys ?? [];
+    const idx = pool.findIndex((k) => k.suffix === suffix);
+    if (idx === -1) {
+      renderError(`No key with suffix "${suffix}" in the pool.`);
+      return;
+    }
+    pool.splice(idx, 1);
+    ctx.saveSettings({ endpointKeys: pool });
+    saveGlobalSettings({ endpointKeys: pool });
+    if (ctx.setKeyPool)
+      ctx.setKeyPool(pool.map((k) => k.key));
+    renderInfo(`Removed key ...${suffix} from pool (${pool.length} remaining).`);
+    if (pool.length === 0) {
+      renderInfo("Pool empty \u2014 using single API key from config.");
+    }
+    return;
+  }
   const normalizedArg = arg.replace(/\u2014/g, "--").replace(/\u2013/g, "--");
   const parts = normalizedArg.split(/\s+/);
   const url = parts[0];
@@ -64049,6 +64142,10 @@ ${lines.join("\n")}
     backend = new NexusAgenticBackend(sendFn, config.model, targetPeer, config.apiKey, thinkingEnabled);
   } else {
     backend = new OllamaAgenticBackend(config.backendUrl, config.model, config.apiKey, thinkingEnabled);
+    const poolKeys = config._keyPool;
+    if (poolKeys && poolKeys.length > 0 && backend instanceof OllamaAgenticBackend) {
+      backend.setKeyPool(poolKeys);
+    }
   }
   try {
     const endpointHistory = loadUsageHistory("endpoint", repoRoot);
@@ -64934,6 +65031,9 @@ async function startInteractive(config, repoPath) {
   }
   if (savedSettings.apiKey)
     config = { ...config, apiKey: savedSettings.apiKey };
+  if (savedSettings.endpointKeys && savedSettings.endpointKeys.length > 0) {
+    config._keyPool = savedSettings.endpointKeys.map((k) => k.key);
+  }
   if (savedSettings.verbose !== void 0)
     config = { ...config, verbose: savedSettings.verbose };
   if (savedSettings.maxRetries !== void 0)
@@ -65982,6 +66082,14 @@ Rationale: ${proposal.rationale}${provenanceNote}`;
         statusBar.stopRemoteMetricsPolling();
       }
     },
+    setKeyPool(keys) {
+      if (activeTask) {
+        const backend = activeTask.runner._backend;
+        if (backend?.setKeyPool)
+          backend.setKeyPool(keys);
+      }
+      currentConfig._keyPool = keys;
+    },
     clearScreen() {
       if (isNeovimActive()) {
         writeToNeovimOutput("[screen cleared]\n");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "open-agents-ai",
-  "version": "0.184.69",
+  "version": "0.184.70",
   "description": "AI coding agent powered by open-source models (Ollama/vLLM) — interactive TUI with agentic tool-calling loop",
   "type": "module",
   "main": "./dist/index.js",