@neutrome/open-ai-router 0.1.2 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neutrome/open-ai-router",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "type": "module",
   "exports": {
     ".": "./src/index.ts"

package/src/auth/env.ts

@@ -1,12 +1,14 @@
 import type { AuthDriver, TargetAuthContext, AuthResult } from "./types.ts";
 
 export function envAuthDriver(suffix = "_API_KEY", prefix = ""): AuthDriver {
+  const counters = new Map<string, number>();
+
   return {
     name: "env",
     collectTarget(ctx: TargetAuthContext): AuthResult | null {
-      const key =
-        ctx.env[`${prefix}${ctx.providerName.toUpperCase()}${suffix}`];
-      if (typeof key !== "string" || !key) return null;
+      const base = `${prefix}${ctx.providerName.toUpperCase()}`;
+      const key = resolveKey(ctx.env, base, suffix, counters);
+      if (!key) return null;
       return {
         headers: { authorization: `Bearer ${key}` },
         source: "env",
@@ -14,3 +16,26 @@ export function envAuthDriver(suffix = "_API_KEY", prefix = ""): AuthDriver {
     },
   };
 }
+
+function resolveKey(
+  env: Record<string, unknown>,
+  base: string,
+  suffix: string,
+  counters: Map<string, number>,
+): string | null {
+  const nValue = env[`${base}${suffix}S_N`];
+  if (nValue) {
+    const n = typeof nValue === "string" ? parseInt(nValue, 10) : typeof nValue === "number" ? nValue : 0;
+    if (n > 0) {
+      const prev = counters.get(base) ?? 0;
+      const index = (prev % n) + 1;
+      counters.set(base, prev + 1);
+      const key = env[`${base}${suffix}_${index}`];
+      if (typeof key === "string" && key) return key;
+    }
+  }
+
+  const single = env[`${base}${suffix}`];
+  if (typeof single === "string" && single) return single;
+  return null;
+}