@slkiser/opencode-quota 2.0.0 → 2.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +102 -46
- package/dist/lib/command-handled.d.ts +4 -0
- package/dist/lib/command-handled.d.ts.map +1 -1
- package/dist/lib/command-handled.js +6 -0
- package/dist/lib/command-handled.js.map +1 -1
- package/dist/lib/config.d.ts.map +1 -1
- package/dist/lib/config.js +2 -0
- package/dist/lib/config.js.map +1 -1
- package/dist/lib/copilot.d.ts.map +1 -1
- package/dist/lib/copilot.js +105 -18
- package/dist/lib/copilot.js.map +1 -1
- package/dist/lib/entries.d.ts +2 -0
- package/dist/lib/entries.d.ts.map +1 -1
- package/dist/lib/google.d.ts.map +1 -1
- package/dist/lib/google.js +7 -2
- package/dist/lib/google.js.map +1 -1
- package/dist/lib/modelsdev-pricing.d.ts +4 -0
- package/dist/lib/modelsdev-pricing.d.ts.map +1 -1
- package/dist/lib/modelsdev-pricing.js +14 -1
- package/dist/lib/modelsdev-pricing.js.map +1 -1
- package/dist/lib/opencode-auth.d.ts +9 -0
- package/dist/lib/opencode-auth.d.ts.map +1 -1
- package/dist/lib/opencode-auth.js +38 -0
- package/dist/lib/opencode-auth.js.map +1 -1
- package/dist/lib/provider-availability.d.ts +7 -0
- package/dist/lib/provider-availability.d.ts.map +1 -0
- package/dist/lib/provider-availability.js +12 -0
- package/dist/lib/provider-availability.js.map +1 -0
- package/dist/lib/quota-stats-format.d.ts.map +1 -1
- package/dist/lib/quota-stats-format.js +9 -1
- package/dist/lib/quota-stats-format.js.map +1 -1
- package/dist/lib/quota-stats.d.ts +15 -0
- package/dist/lib/quota-stats.d.ts.map +1 -1
- package/dist/lib/quota-stats.js +168 -73
- package/dist/lib/quota-stats.js.map +1 -1
- package/dist/lib/quota-status.d.ts.map +1 -1
- package/dist/lib/quota-status.js +34 -3
- package/dist/lib/quota-status.js.map +1 -1
- package/dist/lib/qwen-auth.d.ts +8 -0
- package/dist/lib/qwen-auth.d.ts.map +1 -0
- package/dist/lib/qwen-auth.js +19 -0
- package/dist/lib/qwen-auth.js.map +1 -0
- package/dist/lib/qwen-local-quota.d.ts +35 -0
- package/dist/lib/qwen-local-quota.d.ts.map +1 -0
- package/dist/lib/qwen-local-quota.js +162 -0
- package/dist/lib/qwen-local-quota.js.map +1 -0
- package/dist/lib/types.d.ts +62 -2
- package/dist/lib/types.d.ts.map +1 -1
- package/dist/lib/types.js +2 -2
- package/dist/lib/types.js.map +1 -1
- package/dist/lib/zai.d.ts +9 -0
- package/dist/lib/zai.d.ts.map +1 -0
- package/dist/lib/zai.js +86 -0
- package/dist/lib/zai.js.map +1 -0
- package/dist/plugin.d.ts.map +1 -1
- package/dist/plugin.js +282 -127
- package/dist/plugin.js.map +1 -1
- package/dist/providers/openai.d.ts.map +1 -1
- package/dist/providers/openai.js +7 -10
- package/dist/providers/openai.js.map +1 -1
- package/dist/providers/qwen-code.d.ts +3 -0
- package/dist/providers/qwen-code.d.ts.map +1 -0
- package/dist/providers/qwen-code.js +59 -0
- package/dist/providers/qwen-code.js.map +1 -0
- package/dist/providers/registry.d.ts.map +1 -1
- package/dist/providers/registry.js +4 -0
- package/dist/providers/registry.js.map +1 -1
- package/dist/providers/zai.d.ts +8 -0
- package/dist/providers/zai.d.ts.map +1 -0
- package/dist/providers/zai.js +112 -0
- package/dist/providers/zai.js.map +1 -0
- package/package.json +1 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,aAAa,EAA6C,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,aAAa,EAA6C,MAAM,mBAAmB,CAAC;AAalG,eAAO,MAAM,cAAc,EAAE,aAgH5B,CAAC"}
|
package/dist/providers/openai.js
CHANGED
|
@@ -2,19 +2,16 @@
|
|
|
2
2
|
* OpenAI (Plus/Pro) provider wrapper.
|
|
3
3
|
*/
|
|
4
4
|
import { queryOpenAIQuota } from "../lib/openai.js";
|
|
5
|
+
import { isAnyProviderIdAvailable } from "../lib/provider-availability.js";
|
|
5
6
|
export const openaiProvider = {
|
|
6
7
|
id: "openai",
|
|
7
8
|
async isAvailable(ctx) {
|
|
8
|
-
// Best-effort: if
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
}
|
|
15
|
-
catch {
|
|
16
|
-
return true;
|
|
17
|
-
}
|
|
9
|
+
// Best-effort: if provider lookup errors, preserve current permissive fallback.
|
|
10
|
+
return isAnyProviderIdAvailable({
|
|
11
|
+
ctx,
|
|
12
|
+
candidateIds: ["openai", "chatgpt", "codex", "opencode"],
|
|
13
|
+
fallbackOnError: true,
|
|
14
|
+
});
|
|
18
15
|
},
|
|
19
16
|
matchesCurrentModel(model) {
|
|
20
17
|
const provider = model.split("/")[0]?.toLowerCase();
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAW3E,MAAM,CAAC,MAAM,cAAc,GAAkB;IAC3C,EAAE,EAAE,QAAQ;IAEZ,KAAK,CAAC,WAAW,CAAC,GAAyB;QACzC,gFAAgF;QAChF,OAAO,wBAAwB,CAAC;YAC9B,GAAG;YACH,YAAY,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,CAAC;YACxD,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB,CAAC,KAAa;QAC/B,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC;QACpD,IAAI,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC5B,OAAO,CACL,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1F,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAA0B;QACpC,MAAM,MAAM,GAAG,MAAM,gBAAgB,EAAE,CAAC;QAExC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,EAAE;gBACX,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;aACrD,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,SAAS,CAAC;QAElD,4FAA4F;QAC5F,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG;gBACd,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE;gBACrE,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE;gBACrE,MAAM,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE;aACnF,CAAC,MAAM,CAAC,OAAO,CAA6E,CAAC;YAE9F,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO;oBACL,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC;oBACtD,MAAM,EAAE,EAAE;iBACX,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,gBAAgB,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC;YAChE,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;YAE1B,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM,CAAC,KAAK;wBAClB,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;wBACxC,YAAY,EAAE,KAAK,CAAC,YAAY;qBACjC;iBACF;gBACD,MAAM,EAAE,EAAE;aACX,CAAC;QACJ,CAAC;QAED,qCAAqC;QACrC,MAAM,OAAO,GAAwB,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAE3B,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QACrC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,GAAG,KAAK,SAAS;gBACvB,KAAK;gBACL,KAAK,EAAE,SAAS;gBAChB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QACrC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,GAAG,KAAK,SAAS;gBACvB,KAAK;gBACL,KAAK,EAAE,SAAS;gBAChB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;QAC7C,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,GAAG,KAAK,cAAc;gBAC5B,KAAK;gBACL,KAAK,EAAE,cAAc;gBACrB,gBAAgB,EAAE,UAAU,CAAC,gBAAgB;gBAC7C,YAAY,EAAE,UAAU,CAAC,YAAY;aACtC,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,SAAS,EAAE,IAAI;YACf,OAAO;YACP,MAAM,EAAE,EAAE;SACX,CAAC;IACJ,CAAC;CACF,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"qwen-code.d.ts","sourceRoot":"","sources":["../../src/providers/qwen-code.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAA6C,MAAM,mBAAmB,CAAC;AAkBlG,eAAO,MAAM,gBAAgB,EAAE,aA6D9B,CAAC"}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
import { computeQwenQuota, readQwenLocalQuotaState } from "../lib/qwen-local-quota.js";
|
|
2
|
+
import { readAuthFileCached } from "../lib/opencode-auth.js";
|
|
3
|
+
import { DEFAULT_QWEN_AUTH_CACHE_MAX_AGE_MS, hasQwenOAuthAuth, isQwenCodeModelId, } from "../lib/qwen-auth.js";
|
|
4
|
+
export const qwenCodeProvider = {
|
|
5
|
+
id: "qwen-code",
|
|
6
|
+
async isAvailable(_ctx) {
|
|
7
|
+
const auth = await readAuthFileCached({ maxAgeMs: DEFAULT_QWEN_AUTH_CACHE_MAX_AGE_MS });
|
|
8
|
+
return hasQwenOAuthAuth(auth);
|
|
9
|
+
},
|
|
10
|
+
matchesCurrentModel(model) {
|
|
11
|
+
return isQwenCodeModelId(model);
|
|
12
|
+
},
|
|
13
|
+
async fetch(ctx) {
|
|
14
|
+
const auth = await readAuthFileCached({ maxAgeMs: DEFAULT_QWEN_AUTH_CACHE_MAX_AGE_MS });
|
|
15
|
+
if (!hasQwenOAuthAuth(auth)) {
|
|
16
|
+
return { attempted: false, entries: [], errors: [] };
|
|
17
|
+
}
|
|
18
|
+
const quota = computeQwenQuota({ state: await readQwenLocalQuotaState() });
|
|
19
|
+
const style = ctx.config.toastStyle ?? "classic";
|
|
20
|
+
if (style === "grouped") {
|
|
21
|
+
const entries = [
|
|
22
|
+
{
|
|
23
|
+
name: "Qwen Daily",
|
|
24
|
+
group: "Qwen (OAuth)",
|
|
25
|
+
label: "Daily:",
|
|
26
|
+
right: `${quota.day.used}/${quota.day.limit}`,
|
|
27
|
+
percentRemaining: quota.day.percentRemaining,
|
|
28
|
+
resetTimeIso: quota.day.resetTimeIso,
|
|
29
|
+
},
|
|
30
|
+
{
|
|
31
|
+
name: "Qwen RPM",
|
|
32
|
+
group: "Qwen (OAuth)",
|
|
33
|
+
label: "RPM:",
|
|
34
|
+
right: `${quota.rpm.used}/${quota.rpm.limit}`,
|
|
35
|
+
percentRemaining: quota.rpm.percentRemaining,
|
|
36
|
+
resetTimeIso: quota.rpm.resetTimeIso,
|
|
37
|
+
},
|
|
38
|
+
];
|
|
39
|
+
return { attempted: true, entries, errors: [] };
|
|
40
|
+
}
|
|
41
|
+
return {
|
|
42
|
+
attempted: true,
|
|
43
|
+
entries: [
|
|
44
|
+
{
|
|
45
|
+
name: "Qwen Daily",
|
|
46
|
+
percentRemaining: quota.day.percentRemaining,
|
|
47
|
+
resetTimeIso: quota.day.resetTimeIso,
|
|
48
|
+
},
|
|
49
|
+
{
|
|
50
|
+
name: "Qwen RPM",
|
|
51
|
+
percentRemaining: quota.rpm.percentRemaining,
|
|
52
|
+
resetTimeIso: quota.rpm.resetTimeIso,
|
|
53
|
+
},
|
|
54
|
+
],
|
|
55
|
+
errors: [],
|
|
56
|
+
};
|
|
57
|
+
},
|
|
58
|
+
};
|
|
59
|
+
//# sourceMappingURL=qwen-code.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"qwen-code.js","sourceRoot":"","sources":["../../src/providers/qwen-code.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACvF,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EACL,kCAAkC,EAClC,gBAAgB,EAChB,iBAAiB,GAClB,MAAM,qBAAqB,CAAC;AAW7B,MAAM,CAAC,MAAM,gBAAgB,GAAkB;IAC7C,EAAE,EAAE,WAAW;IAEf,KAAK,CAAC,WAAW,CAAC,IAA0B;QAC1C,MAAM,IAAI,GAAG,MAAM,kBAAkB,CAAC,EAAE,QAAQ,EAAE,kCAAkC,EAAE,CAAC,CAAC;QACxF,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,mBAAmB,CAAC,KAAa;QAC/B,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAyB;QACnC,MAAM,IAAI,GAAG,MAAM,kBAAkB,CAAC,EAAE,QAAQ,EAAE,kCAAkC,EAAE,CAAC,CAAC;QACxF,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QACvD,CAAC;QAED,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,KAAK,EAAE,MAAM,uBAAuB,EAAE,EAAE,CAAC,CAAC;QAC3E,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,SAAS,CAAC;QAEjD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,OAAO,GAAwB;gBACnC;oBACE,IAAI,EAAE,YAAY;oBAClB,KAAK,EAAE,cAAc;oBACrB,KAAK,EAAE,QAAQ;oBACf,KAAK,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE;oBAC7C,gBAAgB,EAAE,KAAK,CAAC,GAAG,CAAC,gBAAgB;oBAC5C,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,YAAY;iBACrC;gBACD;oBACE,IAAI,EAAE,UAAU;oBAChB,KAAK,EAAE,cAAc;oBACrB,KAAK,EAAE,MAAM;oBACb,KAAK,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE;oBAC7C,gBAAgB,EAAE,KAAK,CAAC,GAAG,CAAC,gBAAgB;oBAC5C,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,YAAY;iBACrC;aACF,CAAC;YAEF,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QAClD,CAAC;QAED,OAAO;YACL,SAAS,EAAE,IAAI;YACf,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,YAAY;oBAClB,gBAAgB,EAAE,KAAK,CAAC,GAAG,CAAC,gBAAgB;oBAC5C,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,YAAY;iBACrC;gBACD;oBACE,IAAI,EAAE,UAAU;oBAChB,gBAAgB,EAAE,KAAK,CAAC,GAAG,CAAC,gBAAgB;oBAC5C,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,YAAY;iBACrC;aACF;YACD,MAAM,EAAE,EAAE;SACX,CAAC;IACJ,CAAC;CACF,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../src/providers/registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../src/providers/registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AASvD,wBAAgB,YAAY,IAAI,aAAa,EAAE,CAW9C"}
|
|
@@ -8,14 +8,18 @@ import { openaiProvider } from "./openai.js";
|
|
|
8
8
|
import { googleAntigravityProvider } from "./google-antigravity.js";
|
|
9
9
|
import { firmwareProvider } from "./firmware.js";
|
|
10
10
|
import { chutesProvider } from "./chutes.js";
|
|
11
|
+
import { qwenCodeProvider } from "./qwen-code.js";
|
|
12
|
+
import { zaiProvider } from "./zai.js";
|
|
11
13
|
export function getProviders() {
|
|
12
14
|
// Order here defines display ordering in the toast.
|
|
13
15
|
return [
|
|
14
16
|
copilotProvider,
|
|
15
17
|
openaiProvider,
|
|
18
|
+
qwenCodeProvider,
|
|
16
19
|
firmwareProvider,
|
|
17
20
|
chutesProvider,
|
|
18
21
|
googleAntigravityProvider,
|
|
22
|
+
zaiProvider,
|
|
19
23
|
];
|
|
20
24
|
}
|
|
21
25
|
//# sourceMappingURL=registry.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/providers/registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/providers/registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAEvC,MAAM,UAAU,YAAY;IAC1B,oDAAoD;IACpD,OAAO;QACL,eAAe;QACf,cAAc;QACd,gBAAgB;QAChB,gBAAgB;QAChB,cAAc;QACd,yBAAyB;QACzB,WAAW;KACZ,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"zai.d.ts","sourceRoot":"","sources":["../../src/providers/zai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,aAAa,EAA6C,MAAM,mBAAmB,CAAC;AAIlG,eAAO,MAAM,WAAW,EAAE,aAuHzB,CAAC"}
|
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Z.ai provider wrapper.
|
|
3
|
+
*
|
|
4
|
+
* Normalizes Z.ai quota into generic toast entries.
|
|
5
|
+
*/
|
|
6
|
+
import { queryZaiQuota } from "../lib/zai.js";
|
|
7
|
+
import { isAnyProviderIdAvailable } from "../lib/provider-availability.js";
|
|
8
|
+
export const zaiProvider = {
|
|
9
|
+
id: "zai",
|
|
10
|
+
async isAvailable(ctx) {
|
|
11
|
+
return isAnyProviderIdAvailable({
|
|
12
|
+
ctx,
|
|
13
|
+
// Z.ai models typically use "zai" or "glm" provider ids.
|
|
14
|
+
candidateIds: ["zai", "glm", "zai-coding-plan"],
|
|
15
|
+
fallbackOnError: false,
|
|
16
|
+
});
|
|
17
|
+
},
|
|
18
|
+
matchesCurrentModel(model) {
|
|
19
|
+
const lower = model.toLowerCase();
|
|
20
|
+
const provider = lower.split("/")[0];
|
|
21
|
+
if (provider && (provider.includes("zai") || provider.includes("glm"))) {
|
|
22
|
+
return true;
|
|
23
|
+
}
|
|
24
|
+
return lower.includes("glm");
|
|
25
|
+
},
|
|
26
|
+
async fetch(ctx) {
|
|
27
|
+
const result = await queryZaiQuota();
|
|
28
|
+
if (!result) {
|
|
29
|
+
return { attempted: false, entries: [], errors: [] };
|
|
30
|
+
}
|
|
31
|
+
if (!result.success) {
|
|
32
|
+
return {
|
|
33
|
+
attempted: true,
|
|
34
|
+
entries: [],
|
|
35
|
+
errors: [{ label: "Z.ai", message: result.error }],
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
const style = ctx.config.toastStyle ?? "classic";
|
|
39
|
+
// Classic toast: show a single entry based on the worst remaining window
|
|
40
|
+
if (style === "classic") {
|
|
41
|
+
const windows = [];
|
|
42
|
+
if (result.windows.hourly) {
|
|
43
|
+
windows.push({ name: "Hourly", ...result.windows.hourly });
|
|
44
|
+
}
|
|
45
|
+
if (result.windows.weekly) {
|
|
46
|
+
windows.push({ name: "Weekly", ...result.windows.weekly });
|
|
47
|
+
}
|
|
48
|
+
if (result.windows.mcp) {
|
|
49
|
+
windows.push({ name: "MCP", ...result.windows.mcp });
|
|
50
|
+
}
|
|
51
|
+
if (windows.length === 0) {
|
|
52
|
+
return {
|
|
53
|
+
attempted: true,
|
|
54
|
+
entries: [{ name: result.label, percentRemaining: 0 }],
|
|
55
|
+
errors: [],
|
|
56
|
+
};
|
|
57
|
+
}
|
|
58
|
+
windows.sort((a, b) => a.percentRemaining - b.percentRemaining);
|
|
59
|
+
const worst = windows[0];
|
|
60
|
+
return {
|
|
61
|
+
attempted: true,
|
|
62
|
+
entries: [
|
|
63
|
+
{
|
|
64
|
+
name: result.label,
|
|
65
|
+
percentRemaining: worst.percentRemaining,
|
|
66
|
+
resetTimeIso: worst.resetTimeIso,
|
|
67
|
+
},
|
|
68
|
+
],
|
|
69
|
+
errors: [],
|
|
70
|
+
};
|
|
71
|
+
}
|
|
72
|
+
// Grouped style: expose all windows
|
|
73
|
+
const entries = [];
|
|
74
|
+
const group = result.label;
|
|
75
|
+
const hourly = result.windows.hourly;
|
|
76
|
+
if (hourly) {
|
|
77
|
+
entries.push({
|
|
78
|
+
name: `${group} Hourly`,
|
|
79
|
+
group,
|
|
80
|
+
label: "Hourly:",
|
|
81
|
+
percentRemaining: hourly.percentRemaining,
|
|
82
|
+
resetTimeIso: hourly.resetTimeIso,
|
|
83
|
+
});
|
|
84
|
+
}
|
|
85
|
+
const weekly = result.windows.weekly;
|
|
86
|
+
if (weekly) {
|
|
87
|
+
entries.push({
|
|
88
|
+
name: `${group} Weekly`,
|
|
89
|
+
group,
|
|
90
|
+
label: "Weekly:",
|
|
91
|
+
percentRemaining: weekly.percentRemaining,
|
|
92
|
+
resetTimeIso: weekly.resetTimeIso,
|
|
93
|
+
});
|
|
94
|
+
}
|
|
95
|
+
const mcp = result.windows.mcp;
|
|
96
|
+
if (mcp) {
|
|
97
|
+
entries.push({
|
|
98
|
+
name: `${group} MCP`,
|
|
99
|
+
group,
|
|
100
|
+
label: "MCP:",
|
|
101
|
+
percentRemaining: mcp.percentRemaining,
|
|
102
|
+
resetTimeIso: mcp.resetTimeIso,
|
|
103
|
+
});
|
|
104
|
+
}
|
|
105
|
+
return {
|
|
106
|
+
attempted: true,
|
|
107
|
+
entries,
|
|
108
|
+
errors: [],
|
|
109
|
+
};
|
|
110
|
+
},
|
|
111
|
+
};
|
|
112
|
+
//# sourceMappingURL=zai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"zai.js","sourceRoot":"","sources":["../../src/providers/zai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAE3E,MAAM,CAAC,MAAM,WAAW,GAAkB;IACxC,EAAE,EAAE,KAAK;IAET,KAAK,CAAC,WAAW,CAAC,GAAyB;QACzC,OAAO,wBAAwB,CAAC;YAC9B,GAAG;YACH,yDAAyD;YACzD,YAAY,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,iBAAiB,CAAC;YAC/C,eAAe,EAAE,KAAK;SACvB,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB,CAAC,KAAa;QAC/B,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QAClC,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrC,IAAI,QAAQ,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACvE,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAyB;QACnC,MAAM,MAAM,GAAG,MAAM,aAAa,EAAE,CAAC;QAErC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,EAAE;gBACX,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;aACnD,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,SAAS,CAAC;QAEjD,yEAAyE;QACzE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,OAAO,GAA6E,EAAE,CAAC;YAE7F,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC1B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7D,CAAC;YACD,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC1B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7D,CAAC;YACD,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;gBACvB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO;oBACL,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC;oBACtD,MAAM,EAAE,EAAE;iBACX,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,gBAAgB,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC;YAChE,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;YAE1B,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM,CAAC,KAAK;wBAClB,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;wBACxC,YAAY,EAAE,KAAK,CAAC,YAAY;qBACjC;iBACF;gBACD,MAAM,EAAE,EAAE;aACX,CAAC;QACJ,CAAC;QAED,oCAAoC;QACpC,MAAM,OAAO,GAA6G,EAAE,CAAC;QAC7H,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAE3B,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QACrC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,GAAG,KAAK,SAAS;gBACvB,KAAK;gBACL,KAAK,EAAE,SAAS;gBAChB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QACrC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,GAAG,KAAK,SAAS;gBACvB,KAAK;gBACL,KAAK,EAAE,SAAS;gBAChB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;QAC/B,IAAI,GAAG,EAAE,CAAC;YACR,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,GAAG,KAAK,MAAM;gBACpB,KAAK;gBACL,KAAK,EAAE,MAAM;gBACb,gBAAgB,EAAE,GAAG,CAAC,gBAAgB;gBACtC,YAAY,EAAE,GAAG,CAAC,YAAY;aAC/B,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,SAAS,EAAE,IAAI;YACf,OAAO;YACP,MAAM,EAAE,EAAE;SACX,CAAC;IACJ,CAAC;CACF,CAAC"}
|
package/package.json
CHANGED