@llmgateway/models 1.194.0 → 1.195.1

package/dist/provider.spec.js

@@ -0,0 +1,136 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { getOrganizationEnvVariant, getProviderEnvValue, getRegionSpecificEnvVarName, getVariantEnvVarName, getVariantEnvVarNameFor, } from "./provider.js";
+const BASE = "LLM_ALIBABA_API_KEY";
+const ENTERPRISE = `${BASE}__ENTERPRISE`;
+const PLANS = `${BASE}__PLANS`;
+const REGIONAL = `${BASE}__US_VIRGINIA`;
+const ENTERPRISE_REGIONAL = `${BASE}__ENTERPRISE__US_VIRGINIA`;
+const PLANS_REGIONAL = `${BASE}__PLANS__US_VIRGINIA`;
+const VERTEX_PROJECT = "LLM_GOOGLE_CLOUD_PROJECT";
+describe("variant env var helpers", () => {
+    beforeEach(() => {
+        for (const name of [
+            BASE,
+            ENTERPRISE,
+            PLANS,
+            REGIONAL,
+            ENTERPRISE_REGIONAL,
+            PLANS_REGIONAL,
+            VERTEX_PROJECT,
+            `${VERTEX_PROJECT}__ENTERPRISE`,
+            `${VERTEX_PROJECT}__PLANS`,
+        ]) {
+            vi.stubEnv(name, undefined);
+        }
+    });
+    afterEach(() => {
+        vi.unstubAllEnvs();
+    });
+    describe("getVariantEnvVarName", () => {
+        it("returns the variant var name only when it is set", () => {
+            expect(getVariantEnvVarName("alibaba", "enterprise")).toBeUndefined();
+            expect(getVariantEnvVarName("alibaba", "plans")).toBeUndefined();
+            vi.stubEnv(ENTERPRISE, "sk-ent");
+            vi.stubEnv(PLANS, "sk-dev");
+            expect(getVariantEnvVarName("alibaba", "enterprise")).toBe(ENTERPRISE);
+            expect(getVariantEnvVarName("alibaba", "plans")).toBe(PLANS);
+        });
+        it("returns undefined without a variant or for unknown providers", () => {
+            vi.stubEnv(ENTERPRISE, "sk-ent");
+            expect(getVariantEnvVarName("alibaba", undefined)).toBeUndefined();
+            expect(getVariantEnvVarName("not-a-provider", "enterprise")).toBeUndefined();
+        });
+    });
+    describe("getVariantEnvVarNameFor", () => {
+        it("applies to arbitrary env var names", () => {
+            vi.stubEnv(`${VERTEX_PROJECT}__PLANS`, "plans-project");
+            expect(getVariantEnvVarNameFor(VERTEX_PROJECT, "plans")).toBe(`${VERTEX_PROJECT}__PLANS`);
+            expect(getVariantEnvVarNameFor(VERTEX_PROJECT, "enterprise")).toBeUndefined();
+        });
+    });
+    describe("getProviderEnvValue with variant", () => {
+        it("reads the variant list at the config index", () => {
+            vi.stubEnv(VERTEX_PROJECT, "base-project-a,base-project-b");
+            vi.stubEnv(`${VERTEX_PROJECT}__ENTERPRISE`, "ent-project-a,ent-project-b");
+            expect(getProviderEnvValue("google-vertex", "project", 1)).toBe("base-project-b");
+            expect(getProviderEnvValue("google-vertex", "project", 1, undefined, "enterprise")).toBe("ent-project-b");
+            expect(getProviderEnvValue("google-vertex", "project", 0, undefined, "plans")).toBe("base-project-a");
+        });
+        it("returns the default when neither variant nor base var is set", () => {
+            expect(getProviderEnvValue("google-vertex", "region", 0, "global", "plans")).toBe("global");
+        });
+    });
+    describe("getRegionSpecificEnvVarName with variant", () => {
+        it("prefers the variant-regional var for matching orgs", () => {
+            vi.stubEnv(REGIONAL, "sk-region");
+            vi.stubEnv(ENTERPRISE_REGIONAL, "sk-ent-region");
+            vi.stubEnv(PLANS_REGIONAL, "sk-dev-region");
+            expect(getRegionSpecificEnvVarName("alibaba", "us-virginia", "enterprise")).toBe(ENTERPRISE_REGIONAL);
+            expect(getRegionSpecificEnvVarName("alibaba", "us-virginia", "plans")).toBe(PLANS_REGIONAL);
+        });
+        it("falls back to the shared regional var when no variant-regional var is set", () => {
+            vi.stubEnv(REGIONAL, "sk-region");
+            expect(getRegionSpecificEnvVarName("alibaba", "us-virginia", "enterprise")).toBe(REGIONAL);
+        });
+        it("never returns a variant-regional name without a variant", () => {
+            vi.stubEnv(REGIONAL, "sk-region");
+            vi.stubEnv(ENTERPRISE_REGIONAL, "sk-ent-region");
+            expect(getRegionSpecificEnvVarName("alibaba", "us-virginia")).toBe(REGIONAL);
+        });
+        it("returns undefined when no regional var is set, even with a plain variant var", () => {
+            vi.stubEnv(ENTERPRISE, "sk-ent");
+            expect(getRegionSpecificEnvVarName("alibaba", "us-virginia", "enterprise")).toBeUndefined();
+        });
+    });
+    describe("getOrganizationEnvVariant", () => {
+        it("maps enterprise-plan orgs to the enterprise variant", () => {
+            expect(getOrganizationEnvVariant({
+                plan: "enterprise",
+                kind: "default",
+                devPlan: "none",
+            })).toBe("enterprise");
+        });
+        it("maps DevPass orgs to the plans variant", () => {
+            expect(getOrganizationEnvVariant({
+                plan: "free",
+                kind: "devpass",
+                devPlan: "pro",
+            })).toBe("plans");
+        });
+        it("maps Chat plan orgs to the plans variant", () => {
+            expect(getOrganizationEnvVariant({
+                plan: "free",
+                kind: "chat",
+                devPlan: "none",
+                chatPlan: "plus",
+            })).toBe("plans");
+        });
+        it("prefers enterprise when an org matches both", () => {
+            expect(getOrganizationEnvVariant({
+                plan: "enterprise",
+                kind: "devpass",
+                devPlan: "pro",
+            })).toBe("enterprise");
+        });
+        it("returns undefined for regular orgs, inactive plans, and missing orgs", () => {
+            expect(getOrganizationEnvVariant({
+                plan: "pro",
+                kind: "default",
+                devPlan: "none",
+            })).toBeUndefined();
+            expect(getOrganizationEnvVariant({
+                plan: "free",
+                kind: "devpass",
+                devPlan: "none",
+            })).toBeUndefined();
+            expect(getOrganizationEnvVariant({
+                plan: "free",
+                kind: "chat",
+                chatPlan: "none",
+            })).toBeUndefined();
+            expect(getOrganizationEnvVariant(null)).toBeUndefined();
+            expect(getOrganizationEnvVariant(undefined)).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=provider.spec.js.map

package/dist/provider.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.spec.js","sourceRoot":"","sources":["../src/provider.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EACN,yBAAyB,EACzB,mBAAmB,EACnB,2BAA2B,EAC3B,oBAAoB,EACpB,uBAAuB,GACvB,MAAM,eAAe,CAAC;AAEvB,MAAM,IAAI,GAAG,qBAAqB,CAAC;AACnC,MAAM,UAAU,GAAG,GAAG,IAAI,cAAc,CAAC;AACzC,MAAM,KAAK,GAAG,GAAG,IAAI,SAAS,CAAC;AAC/B,MAAM,QAAQ,GAAG,GAAG,IAAI,eAAe,CAAC;AACxC,MAAM,mBAAmB,GAAG,GAAG,IAAI,2BAA2B,CAAC;AAC/D,MAAM,cAAc,GAAG,GAAG,IAAI,sBAAsB,CAAC;AACrD,MAAM,cAAc,GAAG,0BAA0B,CAAC;AAElD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACxC,UAAU,CAAC,GAAG,EAAE;QACf,KAAK,MAAM,IAAI,IAAI;YAClB,IAAI;YACJ,UAAU;YACV,KAAK;YACL,QAAQ;YACR,mBAAmB;YACnB,cAAc;YACd,cAAc;YACd,GAAG,cAAc,cAAc;YAC/B,GAAG,cAAc,SAAS;SAC1B,EAAE,CAAC;YACH,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC7B,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YACtE,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YACjE,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACjC,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;YAC5B,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACvE,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACvE,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACjC,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YACnE,MAAM,CACL,oBAAoB,CAAC,gBAAgB,EAAE,YAAY,CAAC,CACpD,CAAC,aAAa,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC7C,EAAE,CAAC,OAAO,CAAC,GAAG,cAAc,SAAS,EAAE,eAAe,CAAC,CAAC;YACxD,MAAM,CAAC,uBAAuB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAC5D,GAAG,cAAc,SAAS,CAC1B,CAAC;YACF,MAAM,CACL,uBAAuB,CAAC,cAAc,EAAE,YAAY,CAAC,CACrD,CAAC,aAAa,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QACjD,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACrD,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,+BAA+B,CAAC,CAAC;YAC5D,EAAE,CAAC,OAAO,CACT,GAAG,cAAc,cAAc,EAC/B,6BAA6B,CAC7B,CAAC;YAEF,MAAM,CAAC,mBAAmB,CAAC,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAC9D,gBAAgB,CAChB,CAAC;YACF,MAAM,CACL,mBAAmB,CAClB,eAAe,EACf,SAAS,EACT,CAAC,EACD,SAAS,EACT,YAAY,CACZ,CACD,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACxB,MAAM,CACL,mBAAmB,CAAC,eAAe,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,CACtE,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACvE,MAAM,CACL,mBAAmB,CAAC,eAAe,EAAE,QAAQ,EAAE,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,CACpE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC7D,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YAClC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAC;YACjD,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC;YAC5C,MAAM,CACL,2BAA2B,CAAC,SAAS,EAAE,aAAa,EAAE,YAAY,CAAC,CACnE,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC5B,MAAM,CACL,2BAA2B,CAAC,SAAS,EAAE,aAAa,EAAE,OAAO,CAAC,CAC9D,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACpF,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YAClC,MAAM,CACL,2BAA2B,CAAC,SAAS,EAAE,aAAa,EAAE,YAAY,CAAC,CACnE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YAClE,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YAClC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAC;YACjD,MAAM,CAAC,2BAA2B,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC,CAAC,IAAI,CACjE,QAAQ,CACR,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACvF,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACjC,MAAM,CACL,2BAA2B,CAAC,SAAS,EAAE,aAAa,EAAE,YAAY,CAAC,CACnE,CAAC,aAAa,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC9D,MAAM,CACL,yBAAyB,CAAC;gBACzB,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,MAAM;aACf,CAAC,CACF,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YACjD,MAAM,CACL,yBAAyB,CAAC;gBACzB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,KAAK;aACd,CAAC,CACF,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YACnD,MAAM,CACL,yBAAyB,CAAC;gBACzB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,MAAM;gBACf,QAAQ,EAAE,MAAM;aAChB,CAAC,CACF,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACtD,MAAM,CACL,yBAAyB,CAAC;gBACzB,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,KAAK;aACd,CAAC,CACF,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC/E,MAAM,CACL,yBAAyB,CAAC;gBACzB,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,MAAM;aACf,CAAC,CACF,CAAC,aAAa,EAAE,CAAC;YAClB,MAAM,CACL,yBAAyB,CAAC;gBACzB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,MAAM;aACf,CAAC,CACF,CAAC,aAAa,EAAE,CAAC;YAClB,MAAM,CACL,yBAAyB,CAAC;gBACzB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,MAAM;aAChB,CAAC,CACF,CAAC,aAAa,EAAE,CAAC;YAClB,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YACxD,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC9D,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@llmgateway/models",
-	"version": "1.194.0",
+	"version": "1.195.1",
 	"private": false,
 	"repository": {
 		"type": "git",