@goodandready/dsh-key-rotation 0.5.4 → 0.6.0

package/lib/index.js

@@ -80,7 +80,7 @@ export const Config = Schema.object({
   providers: Schema.array(Schema.object({
     provider: Schema.string().required(),
     keys: Schema.array(Schema.string()).default([]),
-    cooldownMs: Schema.number().optional(),
+    cooldownMs: Schema.number(),
   })).default([...DEFAULT_PROVIDERS]),
 });
 
@@ -617,6 +617,32 @@ export function apply(ctx, config = {}) {
     return rotate(options, pool);
   });
 
+  // Safety net for non-stream requests (agent/request-error waterfall).
+  // llm/stream covers streaming calls; sync calls (embeddings, batch) go
+  // through agent/request and surface errors here. If the error is
+  // switchable, mark the key and ask the agent loop to retry.
+  ctx.on('agent/request-error', async (payload, next) => {
+    const provider = payload?.provider ?? payload?.failure?.provider ?? '';
+    if (!provider) return next();
+    const { providerToPool, switchCodes } = buildRuntime();
+    const pool = providerToPool.get(provider);
+    if (!pool) return next();
+    const code = String(payload?.failure?.code ?? payload?.code ?? '');
+    const message = String(payload?.failure?.message ?? payload?.message ?? '');
+    const switchable = switchCodes.has(code) || SWITCHABLE_MESSAGE_PATTERN.test(message);
+    if (!switchable) return next();
+    const ref = pool.state.lastUsed;
+    if (ref) {
+      const backoff = recordFailure(pool, ref, Date.now(), pool.cooldownMs ?? 60000);
+      pushEvent(pool, ref, code || 'UNKNOWN', backoff);
+      pool.state.switches = (pool.state.switches ?? 0) + 1;
+      pool.state.lastReason = code || 'UNKNOWN';
+      pool.state.lastSwitchAt = Date.now();
+      console.warn(`[dsh-key-rotation] ${provider}: key ${String(ref)} failed via agent/request-error (${String(code)} ${String(message).slice(0, 80)}) — retry`);
+    }
+    return { kind: 'retry' };
+  });
+
   ctx.inject(['settings'], (sctx) => {
     const scope = sctx.settings.register(NS, Config, { base: config });
     getConfig = () => scope.get() ?? config;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goodandready/dsh-key-rotation",
-  "version": "0.5.4",
+  "version": "0.6.0",
   "description": "Per-provider API key rotation for DeepSeek Harness: a key pool per provider, auto-created clone routes, and switching to the next key on quota/rate-limit errors. Includes a Settings section (Key Rotation) to edit the key pools, cooldown and switch codes.",
   "keywords": [
     "deepseek-harness",