opencode-antigravity-autopilot 2.0.0 → 2.1.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/dist/AsyncLogger.d.ts +5 -0
- package/dist/AsyncLogger.d.ts.map +1 -0
- package/dist/AsyncLogger.js +78 -0
- package/dist/AsyncLogger.js.map +1 -0
- package/dist/auth/AccountRotator.d.ts +1 -1
- package/dist/auth/AccountRotator.d.ts.map +1 -1
- package/dist/auth/AccountRotator.js +13 -3
- package/dist/auth/AccountRotator.js.map +1 -1
- package/dist/logger.d.ts +2 -0
- package/dist/logger.d.ts.map +1 -0
- package/dist/logger.js +30 -0
- package/dist/logger.js.map +1 -0
- package/dist/manager.d.ts +4 -3
- package/dist/manager.d.ts.map +1 -1
- package/dist/manager.js +72 -19
- package/dist/manager.js.map +1 -1
- package/dist/plugin-tools.d.ts +3 -1
- package/dist/plugin-tools.d.ts.map +1 -1
- package/dist/plugin-tools.js +64 -34
- package/dist/plugin-tools.js.map +1 -1
- package/dist/quota/ApiQuotaPoller.d.ts +3 -0
- package/dist/quota/ApiQuotaPoller.d.ts.map +1 -1
- package/dist/quota/ApiQuotaPoller.js +100 -29
- package/dist/quota/ApiQuotaPoller.js.map +1 -1
- package/dist/quota/QuotaCacheUpdater.d.ts +2 -0
- package/dist/quota/QuotaCacheUpdater.d.ts.map +1 -1
- package/dist/quota/QuotaCacheUpdater.js +73 -29
- package/dist/quota/QuotaCacheUpdater.js.map +1 -1
- package/dist/rotation/HardLimitDetector.js +1 -1
- package/dist/rotation/HardLimitDetector.js.map +1 -1
- package/package.json +1 -1
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AsyncLogger.d.ts","sourceRoot":"","sources":["../src/AsyncLogger.ts"],"names":[],"mappings":"AAIA,qBAAa,WAAW;IACpB,OAAO,CAAC,MAAM,CAAC,OAAO,CAAoC;WAE7C,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;CAqBpE"}
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
|
|
2
|
+
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
|
|
3
|
+
return new (P || (P = Promise))(function (resolve, reject) {
|
|
4
|
+
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
|
|
5
|
+
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
|
|
6
|
+
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
|
|
7
|
+
step((generator = generator.apply(thisArg, _arguments || [])).next());
|
|
8
|
+
});
|
|
9
|
+
};
|
|
10
|
+
var __generator = (this && this.__generator) || function (thisArg, body) {
|
|
11
|
+
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
|
|
12
|
+
return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
|
|
13
|
+
function verb(n) { return function (v) { return step([n, v]); }; }
|
|
14
|
+
function step(op) {
|
|
15
|
+
if (f) throw new TypeError("Generator is already executing.");
|
|
16
|
+
while (g && (g = 0, op[0] && (_ = 0)), _) try {
|
|
17
|
+
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
|
|
18
|
+
if (y = 0, t) op = [op[0] & 2, t.value];
|
|
19
|
+
switch (op[0]) {
|
|
20
|
+
case 0: case 1: t = op; break;
|
|
21
|
+
case 4: _.label++; return { value: op[1], done: false };
|
|
22
|
+
case 5: _.label++; y = op[1]; op = [0]; continue;
|
|
23
|
+
case 7: op = _.ops.pop(); _.trys.pop(); continue;
|
|
24
|
+
default:
|
|
25
|
+
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
|
|
26
|
+
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
|
|
27
|
+
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
|
|
28
|
+
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
|
|
29
|
+
if (t[2]) _.ops.pop();
|
|
30
|
+
_.trys.pop(); continue;
|
|
31
|
+
}
|
|
32
|
+
op = body.call(thisArg, _);
|
|
33
|
+
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
|
|
34
|
+
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
|
|
35
|
+
}
|
|
36
|
+
};
|
|
37
|
+
import fs from 'fs';
|
|
38
|
+
var AsyncLogger = (function () {
|
|
39
|
+
function AsyncLogger() {
|
|
40
|
+
}
|
|
41
|
+
AsyncLogger.logAsync = function (message, data) {
|
|
42
|
+
return __awaiter(this, void 0, void 0, function () {
|
|
43
|
+
var timestamp, logMessage, error_1;
|
|
44
|
+
return __generator(this, function (_a) {
|
|
45
|
+
switch (_a.label) {
|
|
46
|
+
case 0:
|
|
47
|
+
timestamp = new Date().toISOString();
|
|
48
|
+
logMessage = "[".concat(timestamp, "] ").concat(message);
|
|
49
|
+
if (data) {
|
|
50
|
+
try {
|
|
51
|
+
logMessage += " ".concat(JSON.stringify(data));
|
|
52
|
+
}
|
|
53
|
+
catch (e) {
|
|
54
|
+
logMessage += " [Circular/Unserializable]";
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
logMessage += '\n';
|
|
58
|
+
_a.label = 1;
|
|
59
|
+
case 1:
|
|
60
|
+
_a.trys.push([1, 3, , 4]);
|
|
61
|
+
return [4, fs.promises.appendFile(this.logFile, logMessage, 'utf8')];
|
|
62
|
+
case 2:
|
|
63
|
+
_a.sent();
|
|
64
|
+
return [3, 4];
|
|
65
|
+
case 3:
|
|
66
|
+
error_1 = _a.sent();
|
|
67
|
+
console.warn('Failed to write to log file:', error_1);
|
|
68
|
+
return [3, 4];
|
|
69
|
+
case 4: return [2];
|
|
70
|
+
}
|
|
71
|
+
});
|
|
72
|
+
});
|
|
73
|
+
};
|
|
74
|
+
AsyncLogger.logFile = '/tmp/antigravity-autopilot.log';
|
|
75
|
+
return AsyncLogger;
|
|
76
|
+
}());
|
|
77
|
+
export { AsyncLogger };
|
|
78
|
+
//# sourceMappingURL=AsyncLogger.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AsyncLogger.js","sourceRoot":"","sources":["../src/AsyncLogger.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AAIpB;IAAA;IAwBA,CAAC;IArBgB,oBAAQ,GAArB,UAAsB,OAAe,EAAE,IAAU;;;;;;wBACvC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;wBACvC,UAAU,GAAG,WAAI,SAAS,eAAK,OAAO,CAAE,CAAC;wBAE7C,IAAI,IAAI,EAAE,CAAC;4BACP,IAAI,CAAC;gCACD,UAAU,IAAI,WAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAE,CAAC;4BAC7C,CAAC;4BAAC,OAAO,CAAC,EAAE,CAAC;gCACT,UAAU,IAAI,4BAA4B,CAAC;4BAC/C,CAAC;wBACL,CAAC;wBAED,UAAU,IAAI,IAAI,CAAC;;;;wBAGf,WAAM,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,EAAA;;wBAA9D,SAA8D,CAAC;;;;wBAG/D,OAAO,CAAC,IAAI,CAAC,8BAA8B,EAAE,OAAK,CAAC,CAAC;;;;;;KAE3D;IAtBc,mBAAO,GAAG,gCAAgC,CAAC;IAuB9D,kBAAC;CAAA,AAxBD,IAwBC;SAxBY,WAAW"}
|
|
@@ -4,7 +4,7 @@ export declare class AccountRotator {
|
|
|
4
4
|
private activeIndex;
|
|
5
5
|
constructor(accounts: AccountMetadataV3[], initialIndex: number);
|
|
6
6
|
getCurrentAccount(): AccountMetadataV3 | null;
|
|
7
|
-
markCurrentExhausted(cooldownMs?: number): number;
|
|
7
|
+
markCurrentExhausted(cooldownMs?: number, resetTimeISO?: string): number;
|
|
8
8
|
private saveToDisk;
|
|
9
9
|
private getStoragePath;
|
|
10
10
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AccountRotator.d.ts","sourceRoot":"","sources":["../../src/auth/AccountRotator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAoB,MAAM,sBAAsB,CAAC;AAK3E,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAuB;IACvC,OAAO,CAAC,WAAW,CAAU;gBAEjB,QAAQ,EAAE,iBAAiB,EAAE,EAAE,YAAY,EAAE,MAAM;IASxD,iBAAiB,IAAI,iBAAiB,GAAG,IAAI;IAqC7C,oBAAoB,CAAC,UAAU,
|
|
1
|
+
{"version":3,"file":"AccountRotator.d.ts","sourceRoot":"","sources":["../../src/auth/AccountRotator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAoB,MAAM,sBAAsB,CAAC;AAK3E,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAuB;IACvC,OAAO,CAAC,WAAW,CAAU;gBAEjB,QAAQ,EAAE,iBAAiB,EAAE,EAAE,YAAY,EAAE,MAAM;IASxD,iBAAiB,IAAI,iBAAiB,GAAG,IAAI;IAqC7C,oBAAoB,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM;IAyB/E,OAAO,CAAC,UAAU;IA2BlB,OAAO,CAAC,cAAc;CASvB"}
|
|
@@ -63,13 +63,23 @@ var AccountRotator = (function () {
|
|
|
63
63
|
}
|
|
64
64
|
return this.accounts[this.activeIndex];
|
|
65
65
|
};
|
|
66
|
-
AccountRotator.prototype.markCurrentExhausted = function (cooldownMs) {
|
|
67
|
-
if (cooldownMs === void 0) { cooldownMs = 30 * 60 * 1000; }
|
|
66
|
+
AccountRotator.prototype.markCurrentExhausted = function (cooldownMs, resetTimeISO) {
|
|
68
67
|
if (this.accounts.length === 0) {
|
|
69
68
|
return -1;
|
|
70
69
|
}
|
|
71
70
|
var currentAccount = this.accounts[this.activeIndex];
|
|
72
|
-
|
|
71
|
+
if (resetTimeISO) {
|
|
72
|
+
var resetTime = new Date(resetTimeISO);
|
|
73
|
+
if (!Number.isNaN(resetTime.getTime())) {
|
|
74
|
+
currentAccount.coolingDownUntil = resetTime.getTime();
|
|
75
|
+
}
|
|
76
|
+
else {
|
|
77
|
+
currentAccount.coolingDownUntil = Date.now() + (cooldownMs || 30 * 60 * 1000);
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
else {
|
|
81
|
+
currentAccount.coolingDownUntil = Date.now() + (cooldownMs || 30 * 60 * 1000);
|
|
82
|
+
}
|
|
73
83
|
this.activeIndex = (this.activeIndex + 1) % this.accounts.length;
|
|
74
84
|
this.saveToDisk();
|
|
75
85
|
return this.activeIndex;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AccountRotator.js","sourceRoot":"","sources":["../../src/auth/AccountRotator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AACA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB;IAIE,wBAAY,QAA6B,EAAE,YAAoB;QAC7D,IAAI,CAAC,CAAC,IAAI,YAAY,cAAc,CAAC,EAAE,CAAC;YAEtC,OAAO,IAAI,cAAc,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,CAAC,QAAQ,4BAAO,QAAQ,SAAC,CAAC;QAC9B,IAAI,CAAC,WAAW,GAAG,YAAY,CAAC;IAClC,CAAC;IAEM,0CAAiB,GAAxB;QACE,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QACnC,IAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,IAAM,KAAK,GAAG,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;YAC7C,IAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAErC,IAAI,CAAC,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,IAAI,GAAG,EAAE,CAAC;gBACjE,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;gBACzB,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;QAED,IAAI,WAAW,GAAG,QAAQ,CAAC;QAC3B,IAAI,QAAQ,GAAG,CAAC,CAAC,CAAC;QAElB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,IAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,GAAG,WAAW,EAAE,CAAC;gBACvE,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC;gBACvC,QAAQ,GAAG,CAAC,CAAC;YACf,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,QAAQ,CAAC;YAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC;QAED,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC;IAEM,6CAAoB,GAA3B,UAA4B,
|
|
1
|
+
{"version":3,"file":"AccountRotator.js","sourceRoot":"","sources":["../../src/auth/AccountRotator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AACA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB;IAIE,wBAAY,QAA6B,EAAE,YAAoB;QAC7D,IAAI,CAAC,CAAC,IAAI,YAAY,cAAc,CAAC,EAAE,CAAC;YAEtC,OAAO,IAAI,cAAc,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,CAAC,QAAQ,4BAAO,QAAQ,SAAC,CAAC;QAC9B,IAAI,CAAC,WAAW,GAAG,YAAY,CAAC;IAClC,CAAC;IAEM,0CAAiB,GAAxB;QACE,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QACnC,IAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,IAAM,KAAK,GAAG,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;YAC7C,IAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAErC,IAAI,CAAC,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,IAAI,GAAG,EAAE,CAAC;gBACjE,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;gBACzB,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;QAED,IAAI,WAAW,GAAG,QAAQ,CAAC;QAC3B,IAAI,QAAQ,GAAG,CAAC,CAAC,CAAC;QAElB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,IAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,GAAG,WAAW,EAAE,CAAC;gBACvE,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC;gBACvC,QAAQ,GAAG,CAAC,CAAC;YACf,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,QAAQ,CAAC;YAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC;QAED,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC;IAEM,6CAAoB,GAA3B,UAA4B,UAAmB,EAAE,YAAqB;QACpE,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,CAAC,CAAC;QACZ,CAAC;QAED,IAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAEvD,IAAI,YAAY,EAAE,CAAC;YACjB,IAAM,SAAS,GAAG,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBACvC,cAAc,CAAC,gBAAgB,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,cAAc,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,UAAU,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAChF,CAAC;QACH,CAAC;aAAM,CAAC;YACN,cAAc,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,UAAU,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAEjE,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEO,mCAAU,GAAlB;QACE,IAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAG1C,IAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACzC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;gBAC3D,OAAO;YACT,CAAC;QACH,CAAC;QAED,IAAM,OAAO,GAAqB;YAChC,OAAO,EAAE,CAAC;YACV,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;QAEF,IAAI,CAAC;YACH,EAAE,CAAC,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAC3E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,uCAAc,GAAtB;QACE,IAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;QAClD,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,2BAA2B,CAAC,CAAC;QAC3E,CAAC;QAED,IAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,2BAA2B,CAAC,CAAC;IAChF,CAAC;IACH,qBAAC;AAAD,CAAC,AA/GD,IA+GC"}
|
package/dist/logger.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAMA,wBAAgB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,GAAG,QAsBnD"}
|
package/dist/logger.js
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import * as fs from 'fs';
|
|
2
|
+
import * as path from 'path';
|
|
3
|
+
import * as os from 'os';
|
|
4
|
+
var LOG_FILE_PATH = path.join(os.tmpdir(), 'antigravity-autopilot.log');
|
|
5
|
+
export function logDebug(message, data) {
|
|
6
|
+
try {
|
|
7
|
+
var timestamp = new Date().toISOString();
|
|
8
|
+
var logEntry = "[".concat(timestamp, "] ").concat(message);
|
|
9
|
+
if (data !== undefined) {
|
|
10
|
+
if (data instanceof Error) {
|
|
11
|
+
logEntry += "\nError: ".concat(data.message, "\nStack: ").concat(data.stack);
|
|
12
|
+
}
|
|
13
|
+
else {
|
|
14
|
+
try {
|
|
15
|
+
var serialized = JSON.stringify(data, null, 2);
|
|
16
|
+
logEntry += "\nData: ".concat(serialized);
|
|
17
|
+
}
|
|
18
|
+
catch (e) {
|
|
19
|
+
logEntry += "\nData: [Circular or Unserializable]";
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
logEntry += '\n';
|
|
24
|
+
fs.appendFileSync(LOG_FILE_PATH, logEntry);
|
|
25
|
+
}
|
|
26
|
+
catch (error) {
|
|
27
|
+
console.error('Failed to write to log file:', error);
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=logger.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logger.js","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB,IAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC;AAE1E,MAAM,UAAU,QAAQ,CAAC,OAAe,EAAE,IAAU;IAChD,IAAI,CAAC;QACD,IAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,IAAI,QAAQ,GAAG,WAAI,SAAS,eAAK,OAAO,CAAE,CAAC;QAC3C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACrB,IAAI,IAAI,YAAY,KAAK,EAAE,CAAC;gBACxB,QAAQ,IAAI,mBAAY,IAAI,CAAC,OAAO,sBAAY,IAAI,CAAC,KAAK,CAAE,CAAC;YACjE,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC;oBACD,IAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;oBACjD,QAAQ,IAAI,kBAAW,UAAU,CAAE,CAAC;gBACxC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACT,QAAQ,IAAI,sCAAsC,CAAC;gBACvD,CAAC;YACL,CAAC;QACL,CAAC;QACD,QAAQ,IAAI,IAAI,CAAC;QACjB,EAAE,CAAC,cAAc,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAEb,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;IACzD,CAAC;AACL,CAAC"}
|
package/dist/manager.d.ts
CHANGED
|
@@ -10,13 +10,14 @@ export declare class QuotaManager {
|
|
|
10
10
|
private quotaTracker;
|
|
11
11
|
private modelSelector;
|
|
12
12
|
private lspProcess;
|
|
13
|
+
private lastSelectedModel;
|
|
13
14
|
constructor(config?: PluginConfig);
|
|
14
15
|
initialize(): Promise<void>;
|
|
15
16
|
getToken(): Promise<string | null>;
|
|
16
17
|
getQuota(): Promise<QuotaInfo | null>;
|
|
17
|
-
getQuotaViaApi(modelName?: string): Promise<QuotaInfo | null>;
|
|
18
|
-
getAllQuotasViaApi(): Promise<Map<string, QuotaInfo>>;
|
|
19
|
-
rotateAccount(): Promise<void>;
|
|
18
|
+
getQuotaViaApi(modelName?: string, retries?: number): Promise<QuotaInfo | null>;
|
|
19
|
+
getAllQuotasViaApi(retries?: number): Promise<Map<string, QuotaInfo>>;
|
|
20
|
+
rotateAccount(resetTimeISO?: string): Promise<void>;
|
|
20
21
|
selectBestModel(): string | null;
|
|
21
22
|
updateQuotaForModel(model: string, quota: QuotaInfo): void;
|
|
22
23
|
getQuotaTracker(): QuotaTracker;
|
package/dist/manager.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../src/manager.ts"],"names":[],"mappings":"AAGA,OAAO,EAAe,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAE7D,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAEvD,OAAO,EAAE,YAAY,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;
|
|
1
|
+
{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../src/manager.ts"],"names":[],"mappings":"AAGA,OAAO,EAAe,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAE7D,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAEvD,OAAO,EAAE,YAAY,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAG9D,qBAAa,YAAY;IACvB,OAAO,CAAC,WAAW,CAAsB;IACzC,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,SAAS,CAAa;IAC9B,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,SAAS,CAAkB;IACnC,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,aAAa,CAAwB;IAC7C,OAAO,CAAC,UAAU,CAA2D;IAC7E,OAAO,CAAC,iBAAiB,CAAiB;gBAE9B,MAAM,CAAC,EAAE,YAAY;IA4B3B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,QAAQ,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAKlC,QAAQ,IAAI,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAqBrC,cAAc,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,OAAO,SAAI,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IA4C1E,kBAAkB,CAAC,OAAO,SAAI,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAsBhE,aAAa,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKzD,eAAe,IAAI,MAAM,GAAG,IAAI;IAQhC,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,GAAG,IAAI;IAI1D,eAAe,IAAI,YAAY;IAI/B,wBAAwB,CAAC,QAAQ,EAAE,qBAAqB,GAAG,IAAI;CAGhE;AAED,wBAAsB,QAAQ,CAAC,MAAM,CAAC,EAAE,YAAY;gBAK9B,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;gBAItB,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;qBAIpB,OAAO,CAAC,IAAI,CAAC;uBAIjB,MAAM,GAAG,IAAI;+BAIL,MAAM,SAAS,SAAS,GAAG,IAAI;uBAIvC,YAAY;uCAII,qBAAqB,GAAG,IAAI;GAIlE"}
|
package/dist/manager.js
CHANGED
|
@@ -38,14 +38,16 @@ import { TokenStorageReader } from './auth/TokenStorageReader';
|
|
|
38
38
|
import { AccountRotator } from './auth/AccountRotator';
|
|
39
39
|
import { LSPFinder } from './quota/LSPFinder';
|
|
40
40
|
import { QuotaPoller } from './quota/QuotaPoller';
|
|
41
|
-
import { ApiQuotaPoller } from './quota/ApiQuotaPoller';
|
|
41
|
+
import { ApiQuotaPoller, AuthenticationError } from './quota/ApiQuotaPoller';
|
|
42
42
|
import { QuotaTracker } from './rotation/QuotaTracker';
|
|
43
43
|
import { ModelSelector } from './rotation/ModelSelector';
|
|
44
|
+
import { writeQuotaToCache } from './quota/QuotaCacheUpdater';
|
|
44
45
|
var QuotaManager = (function () {
|
|
45
46
|
function QuotaManager(config) {
|
|
46
47
|
if (!(this instanceof QuotaManager)) {
|
|
47
48
|
return new QuotaManager(config);
|
|
48
49
|
}
|
|
50
|
+
this.lastSelectedModel = null;
|
|
49
51
|
this.modelSelector = null;
|
|
50
52
|
this.lspProcess = null;
|
|
51
53
|
this.tokenReader = new TokenStorageReader();
|
|
@@ -116,48 +118,95 @@ var QuotaManager = (function () {
|
|
|
116
118
|
});
|
|
117
119
|
});
|
|
118
120
|
};
|
|
119
|
-
QuotaManager.prototype.getQuotaViaApi = function (
|
|
120
|
-
return __awaiter(this,
|
|
121
|
-
var account, quotas,
|
|
121
|
+
QuotaManager.prototype.getQuotaViaApi = function (modelName_1) {
|
|
122
|
+
return __awaiter(this, arguments, void 0, function (modelName, retries) {
|
|
123
|
+
var account, result, targetModel, quotas, error_1;
|
|
124
|
+
if (retries === void 0) { retries = 3; }
|
|
122
125
|
return __generator(this, function (_a) {
|
|
123
126
|
switch (_a.label) {
|
|
124
127
|
case 0:
|
|
125
128
|
account = this.rotator.getCurrentAccount();
|
|
126
|
-
if (!account)
|
|
129
|
+
if (!account) {
|
|
127
130
|
return [2, null];
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
case 1:
|
|
131
|
-
|
|
132
|
-
|
|
131
|
+
}
|
|
132
|
+
_a.label = 1;
|
|
133
|
+
case 1:
|
|
134
|
+
_a.trys.push([1, 8, , 11]);
|
|
135
|
+
result = null;
|
|
136
|
+
targetModel = modelName || this.lastSelectedModel;
|
|
137
|
+
if (!targetModel) return [3, 3];
|
|
138
|
+
return [4, this.apiPoller.checkQuotaForModel(account, targetModel)];
|
|
139
|
+
case 2:
|
|
140
|
+
result = _a.sent();
|
|
141
|
+
return [3, 5];
|
|
142
|
+
case 3: return [4, this.apiPoller.getAllQuotas(account)];
|
|
143
|
+
case 4:
|
|
133
144
|
quotas = _a.sent();
|
|
134
|
-
if (quotas.size === 0)
|
|
145
|
+
if (quotas.size === 0) {
|
|
135
146
|
return [2, null];
|
|
136
|
-
|
|
137
|
-
|
|
147
|
+
}
|
|
148
|
+
result = quotas.values().next().value || null;
|
|
149
|
+
_a.label = 5;
|
|
150
|
+
case 5:
|
|
151
|
+
if (!result) return [3, 7];
|
|
152
|
+
return [4, writeQuotaToCache(result)];
|
|
153
|
+
case 6:
|
|
154
|
+
_a.sent();
|
|
155
|
+
_a.label = 7;
|
|
156
|
+
case 7: return [2, result];
|
|
157
|
+
case 8:
|
|
158
|
+
error_1 = _a.sent();
|
|
159
|
+
if (!(error_1 instanceof AuthenticationError && retries > 0)) return [3, 10];
|
|
160
|
+
return [4, this.rotateAccount()];
|
|
161
|
+
case 9:
|
|
162
|
+
_a.sent();
|
|
163
|
+
return [2, this.getQuotaViaApi(modelName, retries - 1)];
|
|
164
|
+
case 10: throw error_1;
|
|
165
|
+
case 11: return [2];
|
|
138
166
|
}
|
|
139
167
|
});
|
|
140
168
|
});
|
|
141
169
|
};
|
|
142
170
|
QuotaManager.prototype.getAllQuotasViaApi = function () {
|
|
143
|
-
return __awaiter(this,
|
|
144
|
-
var account;
|
|
171
|
+
return __awaiter(this, arguments, void 0, function (retries) {
|
|
172
|
+
var account, quotas, first, error_2;
|
|
173
|
+
if (retries === void 0) { retries = 3; }
|
|
145
174
|
return __generator(this, function (_a) {
|
|
146
175
|
switch (_a.label) {
|
|
147
176
|
case 0:
|
|
148
177
|
account = this.rotator.getCurrentAccount();
|
|
149
178
|
if (!account)
|
|
150
179
|
return [2, new Map()];
|
|
180
|
+
_a.label = 1;
|
|
181
|
+
case 1:
|
|
182
|
+
_a.trys.push([1, 5, , 8]);
|
|
151
183
|
return [4, this.apiPoller.getAllQuotas(account)];
|
|
152
|
-
case
|
|
184
|
+
case 2:
|
|
185
|
+
quotas = _a.sent();
|
|
186
|
+
first = quotas.values().next().value;
|
|
187
|
+
if (!first) return [3, 4];
|
|
188
|
+
return [4, writeQuotaToCache(first)];
|
|
189
|
+
case 3:
|
|
190
|
+
_a.sent();
|
|
191
|
+
_a.label = 4;
|
|
192
|
+
case 4: return [2, quotas];
|
|
193
|
+
case 5:
|
|
194
|
+
error_2 = _a.sent();
|
|
195
|
+
if (!(error_2 instanceof AuthenticationError && retries > 0)) return [3, 7];
|
|
196
|
+
return [4, this.rotateAccount()];
|
|
197
|
+
case 6:
|
|
198
|
+
_a.sent();
|
|
199
|
+
return [2, this.getAllQuotasViaApi(retries - 1)];
|
|
200
|
+
case 7: throw error_2;
|
|
201
|
+
case 8: return [2];
|
|
153
202
|
}
|
|
154
203
|
});
|
|
155
204
|
});
|
|
156
205
|
};
|
|
157
|
-
QuotaManager.prototype.rotateAccount = function () {
|
|
206
|
+
QuotaManager.prototype.rotateAccount = function (resetTimeISO) {
|
|
158
207
|
return __awaiter(this, void 0, void 0, function () {
|
|
159
208
|
return __generator(this, function (_a) {
|
|
160
|
-
this.rotator.markCurrentExhausted();
|
|
209
|
+
this.rotator.markCurrentExhausted(undefined, resetTimeISO);
|
|
161
210
|
this.quotaTracker.clearAll();
|
|
162
211
|
return [2];
|
|
163
212
|
});
|
|
@@ -165,7 +214,11 @@ var QuotaManager = (function () {
|
|
|
165
214
|
};
|
|
166
215
|
QuotaManager.prototype.selectBestModel = function () {
|
|
167
216
|
var _a;
|
|
168
|
-
|
|
217
|
+
var model = ((_a = this.modelSelector) === null || _a === void 0 ? void 0 : _a.selectModel()) || null;
|
|
218
|
+
if (model) {
|
|
219
|
+
this.lastSelectedModel = model;
|
|
220
|
+
}
|
|
221
|
+
return model;
|
|
169
222
|
};
|
|
170
223
|
QuotaManager.prototype.updateQuotaForModel = function (model, quota) {
|
|
171
224
|
this.quotaTracker.updateQuota(model, quota);
|
package/dist/manager.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../src/manager.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAa,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;
|
|
1
|
+
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../src/manager.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAa,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7E,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAEzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAE9D;IAWE,sBAAY,MAAqB;QAC/B,IAAI,CAAC,CAAC,IAAI,YAAY,YAAY,CAAC,EAAE,CAAC;YAEpC,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;QAC9B,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,WAAW,GAAG,IAAI,kBAAkB,EAAE,CAAC;QAC5C,IAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QAChD,IAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC;QAEtD,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACzD,IAAI,CAAC,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;QACtC,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,cAAc,KAAI,GAAG,CAAC,CAAC;QAEpE,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,eAAe,EAAE,CAAC;YAC5B,IAAM,QAAQ,GAA0B;gBACtC,eAAe,EAAE,MAAM,CAAC,eAAe;gBACvC,cAAc,EAAE,EAAE;gBAClB,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,GAAG;aAC7C,CAAC;YACF,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAEK,iCAAU,GAAhB;;;;;;wBACE,KAAA,IAAI,CAAA;wBAAc,WAAM,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAAA;;wBAApD,GAAK,UAAU,GAAG,SAAkC,CAAC;;;;;KACtD;IAEK,+BAAQ,GAAd;;;;gBACQ,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;gBACjD,WAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,EAAC;;;KAC9C;IAEK,+BAAQ,GAAd;;;;;;6BACM,CAAC,IAAI,CAAC,UAAU,EAAhB,cAAgB;wBAClB,KAAA,IAAI,CAAA;wBAAc,WAAM,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAAA;;wBAApD,GAAK,UAAU,GAAG,SAAkC,CAAC;;;wBAGvD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;4BACrB,WAAO,IAAI,EAAC;wBACd,CAAC;wBAEa,WAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CACxC,IAAI,CAAC,UAAU,CAAC,IAAI,EACpB,IAAI,CAAC,UAAU,CAAC,SAAS,CAC1B,EAAA;;wBAHK,KAAK,GAAG,SAGb;wBAED,IAAI,CAAC,KAAK,EAAE,CAAC;4BACX,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;wBACzB,CAAC;wBAED,WAAO,KAAK,EAAC;;;;KACd;IAEK,qCAAc,GAApB;4DAAqB,SAAkB,EAAE,OAAW;;YAAX,wBAAA,EAAA,WAAW;;;;wBAC5C,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;wBACjD,IAAI,CAAC,OAAO,EAAE,CAAC;4BAEb,WAAO,IAAI,EAAC;wBACd,CAAC;;;;wBAKK,MAAM,GAAqB,IAAI,CAAC;wBAE9B,WAAW,GAAG,SAAS,IAAI,IAAI,CAAC,iBAAiB,CAAC;6BAEpD,WAAW,EAAX,cAAW;wBACJ,WAAM,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO,EAAE,WAAW,CAAC,EAAA;;wBAAtE,MAAM,GAAG,SAA6D,CAAC;;4BAGxD,WAAM,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,EAAA;;wBAAnD,MAAM,GAAG,SAA0C;wBACzD,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;4BAEtB,WAAO,IAAI,EAAC;wBACd,CAAC;wBACD,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,IAAI,IAAI,CAAC;;;6BAG5C,MAAM,EAAN,cAAM;wBAGR,WAAM,iBAAiB,CAAC,MAAM,CAAC,EAAA;;wBAA/B,SAA+B,CAAC;;4BAElC,WAAO,MAAM,EAAC;;;6BAIV,CAAA,OAAK,YAAY,mBAAmB,IAAI,OAAO,GAAG,CAAC,CAAA,EAAnD,eAAmD;wBAErD,WAAM,IAAI,CAAC,aAAa,EAAE,EAAA;;wBAA1B,SAA0B,CAAC;wBAC3B,WAAO,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,GAAG,CAAC,CAAC,EAAC;6BAErD,MAAM,OAAK,CAAC;;;;;KAEf;IAEK,yCAAkB,GAAxB;4DAAyB,OAAW;;YAAX,wBAAA,EAAA,WAAW;;;;wBAC5B,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;wBACjD,IAAI,CAAC,OAAO;4BAAE,WAAO,IAAI,GAAG,EAAE,EAAC;;;;wBAGd,WAAM,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,EAAA;;wBAAnD,MAAM,GAAG,SAA0C;wBAEnD,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;6BACvC,KAAK,EAAL,cAAK;wBACP,WAAM,iBAAiB,CAAC,KAAK,CAAC,EAAA;;wBAA9B,SAA8B,CAAC;;4BAEjC,WAAO,MAAM,EAAC;;;6BAEV,CAAA,OAAK,YAAY,mBAAmB,IAAI,OAAO,GAAG,CAAC,CAAA,EAAnD,cAAmD;wBAErD,WAAM,IAAI,CAAC,aAAa,EAAE,EAAA;;wBAA1B,SAA0B,CAAC;wBAC3B,WAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,GAAG,CAAC,CAAC,EAAC;4BAE9C,MAAM,OAAK,CAAC;;;;;KAEf;IAEK,oCAAa,GAAnB,UAAoB,YAAqB;;;gBACvC,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;gBAC3D,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;;;;KAC9B;IAED,sCAAe,GAAf;;QACE,IAAM,KAAK,GAAG,CAAA,MAAA,IAAI,CAAC,aAAa,0CAAE,WAAW,EAAE,KAAI,IAAI,CAAC;QACxD,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC;QACjC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,0CAAmB,GAAnB,UAAoB,KAAa,EAAE,KAAgB;QACjD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC9C,CAAC;IAED,sCAAe,GAAf;QACE,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,+CAAwB,GAAxB,UAAyB,QAA+B;QACtD,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACtE,CAAC;IACH,mBAAC;AAAD,CAAC,AA/JD,IA+JC;;AAED,MAAM,UAAgB,QAAQ,CAAC,MAAqB;;;;;;oBAC5C,OAAO,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;oBACzC,WAAM,OAAO,CAAC,UAAU,EAAE,EAAA;;oBAA1B,SAA0B,CAAC;oBAE3B,WAAO;4BACC,QAAQ;;;wCACZ,WAAO,OAAO,CAAC,QAAQ,EAAE,EAAC;;;6BAC3B;4BAEK,QAAQ;;;wCACZ,WAAO,OAAO,CAAC,QAAQ,EAAE,EAAC;;;6BAC3B;4BAEK,aAAa;;;wCACjB,WAAO,OAAO,CAAC,aAAa,EAAE,EAAC;;;6BAChC;4BAED,eAAe;gCACb,OAAO,OAAO,CAAC,eAAe,EAAE,CAAC;4BACnC,CAAC;4BAED,mBAAmB,YAAC,KAAa,EAAE,KAAgB;gCACjD,OAAO,CAAC,mBAAmB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;4BAC5C,CAAC;4BAED,eAAe;gCACb,OAAO,OAAO,CAAC,eAAe,EAAE,CAAC;4BACnC,CAAC;4BAED,wBAAwB,YAAC,QAA+B;gCACtD,OAAO,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;4BAC7C,CAAC;yBACF,EAAC;;;;CACH"}
|
package/dist/plugin-tools.d.ts
CHANGED
|
@@ -17,7 +17,9 @@ export declare function createQuotaTools(config?: PluginConfig): {
|
|
|
17
17
|
quota_rotate_account: {
|
|
18
18
|
description: string;
|
|
19
19
|
args: {};
|
|
20
|
-
execute(args?:
|
|
20
|
+
execute(args?: {
|
|
21
|
+
resetTimeISO?: string;
|
|
22
|
+
}, ctx?: any): Promise<string>;
|
|
21
23
|
};
|
|
22
24
|
};
|
|
23
25
|
//# sourceMappingURL=plugin-tools.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"plugin-tools.d.ts","sourceRoot":"","sources":["../src/plugin-tools.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAE5C,wBAAgB,gBAAgB,CAAC,MAAM,CAAC,EAAE,YAAY;;;;uBAQ5B;YAAE,QAAQ,CAAC,EAAE,OAAO,CAAA;SAAE,QAAa,GAAG;;;;;
|
|
1
|
+
{"version":3,"file":"plugin-tools.d.ts","sourceRoot":"","sources":["../src/plugin-tools.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAE5C,wBAAgB,gBAAgB,CAAC,MAAM,CAAC,EAAE,YAAY;;;;uBAQ5B;YAAE,QAAQ,CAAC,EAAE,OAAO,CAAA;SAAE,QAAa,GAAG;;;;;sBAyEtC;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,QAAQ,GAAG;;;;;uBAoC3B;YAAE,YAAY,CAAC,EAAE,MAAM,CAAA;SAAE,QAAQ,GAAG;;EAc9D"}
|
package/dist/plugin-tools.js
CHANGED
|
@@ -34,17 +34,6 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
|
|
|
34
34
|
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
|
|
35
35
|
}
|
|
36
36
|
};
|
|
37
|
-
var __values = (this && this.__values) || function(o) {
|
|
38
|
-
var s = typeof Symbol === "function" && Symbol.iterator, m = s && o[s], i = 0;
|
|
39
|
-
if (m) return m.call(o);
|
|
40
|
-
if (o && typeof o.length === "number") return {
|
|
41
|
-
next: function () {
|
|
42
|
-
if (o && i >= o.length) o = void 0;
|
|
43
|
-
return { value: o && o[i++], done: !o };
|
|
44
|
-
}
|
|
45
|
-
};
|
|
46
|
-
throw new TypeError(s ? "Object is not iterable." : "Symbol.iterator is not defined.");
|
|
47
|
-
};
|
|
48
37
|
var __read = (this && this.__read) || function (o, n) {
|
|
49
38
|
var m = typeof Symbol === "function" && o[Symbol.iterator];
|
|
50
39
|
if (!m) return o;
|
|
@@ -61,6 +50,17 @@ var __read = (this && this.__read) || function (o, n) {
|
|
|
61
50
|
}
|
|
62
51
|
return ar;
|
|
63
52
|
};
|
|
53
|
+
var __values = (this && this.__values) || function(o) {
|
|
54
|
+
var s = typeof Symbol === "function" && Symbol.iterator, m = s && o[s], i = 0;
|
|
55
|
+
if (m) return m.call(o);
|
|
56
|
+
if (o && typeof o.length === "number") return {
|
|
57
|
+
next: function () {
|
|
58
|
+
if (o && i >= o.length) o = void 0;
|
|
59
|
+
return { value: o && o[i++], done: !o };
|
|
60
|
+
}
|
|
61
|
+
};
|
|
62
|
+
throw new TypeError(s ? "Object is not iterable." : "Symbol.iterator is not defined.");
|
|
63
|
+
};
|
|
64
64
|
import { QuotaManager } from './manager';
|
|
65
65
|
import { HardLimitDetector } from './rotation/HardLimitDetector';
|
|
66
66
|
export function createQuotaTools(config) {
|
|
@@ -72,17 +72,17 @@ export function createQuotaTools(config) {
|
|
|
72
72
|
args: {},
|
|
73
73
|
execute: function () {
|
|
74
74
|
return __awaiter(this, arguments, void 0, function (args, ctx) {
|
|
75
|
-
var currentQuota, accounts, activeIndex, percentage, statusIcon, statusText, output, allQuotas,
|
|
76
|
-
var e_1,
|
|
75
|
+
var currentQuota, accounts, activeIndex, percentage, statusIcon, statusText, output, allQuotas, sortedModels, sortedModels_1, sortedModels_1_1, _a, modelName, quota, pct, icon;
|
|
76
|
+
var e_1, _b;
|
|
77
77
|
if (args === void 0) { args = {}; }
|
|
78
|
-
return __generator(this, function (
|
|
79
|
-
switch (
|
|
78
|
+
return __generator(this, function (_c) {
|
|
79
|
+
switch (_c.label) {
|
|
80
80
|
case 0: return [4, manager.initialize()];
|
|
81
81
|
case 1:
|
|
82
|
-
|
|
82
|
+
_c.sent();
|
|
83
83
|
return [4, manager.getQuotaViaApi()];
|
|
84
84
|
case 2:
|
|
85
|
-
currentQuota =
|
|
85
|
+
currentQuota = _c.sent();
|
|
86
86
|
accounts = manager['tokenReader'].getAccounts();
|
|
87
87
|
activeIndex = manager['tokenReader'].getActiveIndex();
|
|
88
88
|
if (!currentQuota) {
|
|
@@ -91,45 +91,68 @@ export function createQuotaTools(config) {
|
|
|
91
91
|
percentage = Math.round(currentQuota.remainingFraction * 100);
|
|
92
92
|
statusIcon = '';
|
|
93
93
|
statusText = '';
|
|
94
|
-
if (percentage
|
|
95
|
-
statusIcon = '
|
|
96
|
-
statusText = '
|
|
94
|
+
if (percentage <= 0) {
|
|
95
|
+
statusIcon = '❌';
|
|
96
|
+
statusText = 'exhausted';
|
|
97
97
|
}
|
|
98
|
-
else if (percentage
|
|
98
|
+
else if (percentage < 20) {
|
|
99
|
+
statusIcon = '❌';
|
|
100
|
+
statusText = 'critical';
|
|
101
|
+
}
|
|
102
|
+
else if (percentage < 50) {
|
|
99
103
|
statusIcon = '⚠';
|
|
100
104
|
statusText = 'running low';
|
|
101
105
|
}
|
|
102
106
|
else {
|
|
103
|
-
statusIcon = '
|
|
104
|
-
statusText = '
|
|
107
|
+
statusIcon = '✓';
|
|
108
|
+
statusText = 'healthy';
|
|
105
109
|
}
|
|
106
|
-
output = "# Quota Status\n\n"
|
|
110
|
+
output = "# Quota Status\n\n";
|
|
111
|
+
output += "".concat(statusIcon, " **Current Model:** ").concat(currentQuota.model || 'Unknown', "\n");
|
|
112
|
+
output += "**Status:** ".concat(statusText.toUpperCase(), " (").concat(percentage, "% remaining)\n");
|
|
113
|
+
output += "**Account:** ".concat(activeIndex + 1, " of ").concat(accounts.length, " active\n");
|
|
107
114
|
if (!args.detailed) return [3, 4];
|
|
108
115
|
return [4, manager.getAllQuotasViaApi()];
|
|
109
116
|
case 3:
|
|
110
|
-
allQuotas =
|
|
117
|
+
allQuotas = _c.sent();
|
|
111
118
|
if (allQuotas.size > 0) {
|
|
112
|
-
output += "\n##
|
|
119
|
+
output += "\n## Available Models\n\n";
|
|
120
|
+
sortedModels = Array.from(allQuotas.entries())
|
|
121
|
+
.filter(function (_a) {
|
|
122
|
+
var _b = __read(_a, 1), name = _b[0];
|
|
123
|
+
var lower = name.toLowerCase();
|
|
124
|
+
return !lower.startsWith('chat_') &&
|
|
125
|
+
!lower.startsWith('rev19') &&
|
|
126
|
+
!lower.includes('gemini 2.5') &&
|
|
127
|
+
!lower.includes('gemini 3 pro image');
|
|
128
|
+
})
|
|
129
|
+
.sort(function (a, b) { return a[0].localeCompare(b[0]); });
|
|
113
130
|
try {
|
|
114
|
-
for (
|
|
115
|
-
|
|
131
|
+
for (sortedModels_1 = __values(sortedModels), sortedModels_1_1 = sortedModels_1.next(); !sortedModels_1_1.done; sortedModels_1_1 = sortedModels_1.next()) {
|
|
132
|
+
_a = __read(sortedModels_1_1.value, 2), modelName = _a[0], quota = _a[1];
|
|
116
133
|
pct = Math.round(quota.remainingFraction * 100);
|
|
117
|
-
icon =
|
|
134
|
+
icon = '✓';
|
|
135
|
+
if (pct <= 0)
|
|
136
|
+
icon = '❌';
|
|
137
|
+
else if (pct < 20)
|
|
138
|
+
icon = '❌';
|
|
139
|
+
else if (pct < 50)
|
|
140
|
+
icon = '⚠';
|
|
118
141
|
output += "- ".concat(icon, " **").concat(modelName, "**: ").concat(pct, "%\n");
|
|
119
142
|
}
|
|
120
143
|
}
|
|
121
144
|
catch (e_1_1) { e_1 = { error: e_1_1 }; }
|
|
122
145
|
finally {
|
|
123
146
|
try {
|
|
124
|
-
if (
|
|
147
|
+
if (sortedModels_1_1 && !sortedModels_1_1.done && (_b = sortedModels_1.return)) _b.call(sortedModels_1);
|
|
125
148
|
}
|
|
126
149
|
finally { if (e_1) throw e_1.error; }
|
|
127
150
|
}
|
|
128
151
|
}
|
|
129
152
|
return [3, 5];
|
|
130
153
|
case 4:
|
|
131
|
-
output += "\n*
|
|
132
|
-
|
|
154
|
+
output += "\n*Tip: Run with { detailed: true } to see all models*\n";
|
|
155
|
+
_c.label = 5;
|
|
133
156
|
case 5: return [2, output];
|
|
134
157
|
}
|
|
135
158
|
});
|
|
@@ -179,15 +202,22 @@ export function createQuotaTools(config) {
|
|
|
179
202
|
},
|
|
180
203
|
},
|
|
181
204
|
quota_rotate_account: {
|
|
182
|
-
description: 'Manually rotate to the next available account when current is exhausted.',
|
|
205
|
+
description: 'Manually rotate to the next available account when current is exhausted. Optionally provide resetTimeISO for accurate cooldown.',
|
|
183
206
|
args: {},
|
|
184
207
|
execute: function (args, ctx) {
|
|
185
208
|
return __awaiter(this, void 0, void 0, function () {
|
|
209
|
+
var resetDate, now, minutesUntilReset;
|
|
186
210
|
return __generator(this, function (_a) {
|
|
187
211
|
switch (_a.label) {
|
|
188
|
-
case 0: return [4, manager.rotateAccount()];
|
|
212
|
+
case 0: return [4, manager.rotateAccount(args === null || args === void 0 ? void 0 : args.resetTimeISO)];
|
|
189
213
|
case 1:
|
|
190
214
|
_a.sent();
|
|
215
|
+
if (args === null || args === void 0 ? void 0 : args.resetTimeISO) {
|
|
216
|
+
resetDate = new Date(args.resetTimeISO);
|
|
217
|
+
now = new Date();
|
|
218
|
+
minutesUntilReset = Math.round((resetDate.getTime() - now.getTime()) / (60 * 1000));
|
|
219
|
+
return [2, "\u2713 Rotated to next account. Previous account will be available again at ".concat(resetDate.toISOString(), " (~").concat(minutesUntilReset, " minutes).")];
|
|
220
|
+
}
|
|
191
221
|
return [2, "\u2713 Rotated to next account. Previous account marked as exhausted for 30 minutes."];
|
|
192
222
|
}
|
|
193
223
|
});
|
package/dist/plugin-tools.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"plugin-tools.js","sourceRoot":"","sources":["../src/plugin-tools.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAGjE,MAAM,UAAU,gBAAgB,CAAC,MAAqB;IACpD,IAAM,OAAO,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IACzC,IAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAE/C,OAAO;QACL,YAAY,EAAE;YACZ,WAAW,EAAE,sFAAsF;YACnG,IAAI,EAAE,EAAE;YACF,OAAO;oEAAC,IAAiC,EAAE,GAAS;;;oBAA5C,qBAAA,EAAA,SAAiC;;;oCAC7C,WAAM,OAAO,CAAC,UAAU,EAAE,EAAA;;gCAA1B,SAA0B,CAAC;gCAEN,WAAM,OAAO,CAAC,cAAc,EAAE,EAAA;;gCAA7C,YAAY,GAAG,SAA8B;gCAC7C,QAAQ,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;gCAChD,WAAW,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,cAAc,EAAE,CAAC;gCAE5D,IAAI,CAAC,YAAY,EAAE,CAAC;oCAClB,WAAO,oFAAoF,EAAC;gCAC9F,CAAC;gCAEK,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,iBAAiB,GAAG,GAAG,CAAC,CAAC;gCAChE,UAAU,GAAG,EAAE,CAAC;gCAChB,UAAU,GAAG,EAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"plugin-tools.js","sourceRoot":"","sources":["../src/plugin-tools.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAGjE,MAAM,UAAU,gBAAgB,CAAC,MAAqB;IACpD,IAAM,OAAO,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IACzC,IAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAE/C,OAAO;QACL,YAAY,EAAE;YACZ,WAAW,EAAE,sFAAsF;YACnG,IAAI,EAAE,EAAE;YACF,OAAO;oEAAC,IAAiC,EAAE,GAAS;;;oBAA5C,qBAAA,EAAA,SAAiC;;;oCAC7C,WAAM,OAAO,CAAC,UAAU,EAAE,EAAA;;gCAA1B,SAA0B,CAAC;gCAEN,WAAM,OAAO,CAAC,cAAc,EAAE,EAAA;;gCAA7C,YAAY,GAAG,SAA8B;gCAC7C,QAAQ,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;gCAChD,WAAW,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,cAAc,EAAE,CAAC;gCAE5D,IAAI,CAAC,YAAY,EAAE,CAAC;oCAClB,WAAO,oFAAoF,EAAC;gCAC9F,CAAC;gCAEK,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,iBAAiB,GAAG,GAAG,CAAC,CAAC;gCAChE,UAAU,GAAG,EAAE,CAAC;gCAChB,UAAU,GAAG,EAAE,CAAC;gCAGpB,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;oCACpB,UAAU,GAAG,GAAG,CAAC;oCACjB,UAAU,GAAG,WAAW,CAAC;gCAC3B,CAAC;qCAAM,IAAI,UAAU,GAAG,EAAE,EAAE,CAAC;oCAC3B,UAAU,GAAG,GAAG,CAAC;oCACjB,UAAU,GAAG,UAAU,CAAC;gCAC1B,CAAC;qCAAM,IAAI,UAAU,GAAG,EAAE,EAAE,CAAC;oCAC3B,UAAU,GAAG,GAAG,CAAC;oCACjB,UAAU,GAAG,aAAa,CAAC;gCAC7B,CAAC;qCAAM,CAAC;oCACN,UAAU,GAAG,GAAG,CAAC;oCACjB,UAAU,GAAG,SAAS,CAAC;gCACzB,CAAC;gCAEG,MAAM,GAAG,oBAAoB,CAAC;gCAClC,MAAM,IAAI,UAAG,UAAU,iCAAuB,YAAY,CAAC,KAAK,IAAI,SAAS,OAAI,CAAC;gCAClF,MAAM,IAAI,sBAAe,UAAU,CAAC,WAAW,EAAE,eAAK,UAAU,mBAAgB,CAAC;gCACjF,MAAM,IAAI,uBAAgB,WAAW,GAAG,CAAC,iBAAO,QAAQ,CAAC,MAAM,cAAW,CAAC;qCAEvE,IAAI,CAAC,QAAQ,EAAb,cAAa;gCACG,WAAM,OAAO,CAAC,kBAAkB,EAAE,EAAA;;gCAA9C,SAAS,GAAG,SAAkC;gCAEpD,IAAI,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;oCACvB,MAAM,IAAI,2BAA2B,CAAC;oCAGhC,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;yCACjD,MAAM,CAAC,UAAC,EAAM;4CAAN,KAAA,aAAM,EAAL,IAAI,QAAA;wCACZ,IAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;wCACjC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC;4CAC/B,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC;4CAC1B,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC;4CAC7B,CAAC,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC;oCAC1C,CAAC,CAAC;yCACD,IAAI,CAAC,UAAC,CAAC,EAAE,CAAC,IAAK,OAAA,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAxB,CAAwB,CAAC,CAAC;;wCAE5C,KAAiC,iBAAA,SAAA,YAAY,CAAA,8GAAE,CAAC;4CAArC,KAAA,iCAAkB,EAAjB,SAAS,QAAA,EAAE,KAAK,QAAA;4CACpB,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,iBAAiB,GAAG,GAAG,CAAC,CAAC;4CAClD,IAAI,GAAG,GAAG,CAAC;4CACf,IAAI,GAAG,IAAI,CAAC;gDAAE,IAAI,GAAG,GAAG,CAAC;iDACpB,IAAI,GAAG,GAAG,EAAE;gDAAE,IAAI,GAAG,GAAG,CAAC;iDACzB,IAAI,GAAG,GAAG,EAAE;gDAAE,IAAI,GAAG,GAAG,CAAC;4CAE9B,MAAM,IAAI,YAAK,IAAI,gBAAM,SAAS,iBAAO,GAAG,QAAK,CAAC;wCACpD,CAAC;;;;;;;;;gCACH,CAAC;;;gCAED,MAAM,IAAI,0DAA0D,CAAC;;oCAGvE,WAAO,MAAM,EAAC;;;;aACf;SACF;QAED,iBAAiB,EAAE;YACjB,WAAW,EAAE,6EAA6E;YAC1F,IAAI,EAAE,EAAE;YACF,OAAO,YAAC,IAAuB,EAAE,GAAS;;;;;;gCAC9C,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;oCAChB,WAAO,wFAAwF,EAAC;gCAClG,CAAC;gCAEc,WAAM,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,EAAA;;gCAAlD,MAAM,GAAG,SAAyC;gCAEpD,IAAI,GAAG,EAAE,CAAC;gCACd,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;oCACvB,IAAI,GAAG,GAAG,CAAC;gCACb,CAAC;qCAAM,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oCAC/B,IAAI,GAAG,GAAG,CAAC;gCACb,CAAC;qCAAM,CAAC;oCACN,IAAI,GAAG,GAAG,CAAC;gCACb,CAAC;gCAEG,MAAM,GAAG,+BAAwB,IAAI,CAAC,KAAK,SAAM,CAAC;gCACtD,MAAM,IAAI,UAAG,IAAI,gBAAM,MAAM,CAAC,OAAO,WAAQ,CAAC;gCAE9C,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oCACxB,MAAM,IAAI,8BAAuB,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,oBAAa,MAAM,CAAC,SAAS,CAAE,CAAC,CAAC,CAAC,wBAAwB,SAAM,CAAC;gCACvH,CAAC;qCAAM,CAAC;oCACN,MAAM,IAAI,+BAA+B,CAAC;gCAC5C,CAAC;gCAED,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;oCACvB,MAAM,IAAI,iEAA4D,CAAC;gCACzE,CAAC;gCAED,WAAO,MAAM,EAAC;;;;aACf;SACF;QAED,oBAAoB,EAAE;YACpB,WAAW,EAAE,iIAAiI;YAC9I,IAAI,EAAE,EAAE;YACF,OAAO,YAAC,IAAgC,EAAE,GAAS;;;;;oCACvD,WAAM,OAAO,CAAC,aAAa,CAAC,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,YAAY,CAAC,EAAA;;gCAA/C,SAA+C,CAAC;gCAEhD,IAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,YAAY,EAAE,CAAC;oCACjB,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oCACxC,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;oCACjB,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;oCAC1F,WAAO,sFAA0E,SAAS,CAAC,WAAW,EAAE,gBAAM,iBAAiB,eAAY,EAAC;gCAC9I,CAAC;gCAED,WAAO,sFAAiF,EAAC;;;;aAC1F;SACF;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -11,6 +11,9 @@ export interface ModelQuotaInfo {
|
|
|
11
11
|
isExhausted: boolean;
|
|
12
12
|
resetTime?: string;
|
|
13
13
|
}
|
|
14
|
+
export declare class AuthenticationError extends Error {
|
|
15
|
+
constructor(message: string);
|
|
16
|
+
}
|
|
14
17
|
export declare class ApiQuotaPoller {
|
|
15
18
|
constructor();
|
|
16
19
|
private refreshAccessToken;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ApiQuotaPoller.d.ts","sourceRoot":"","sources":["../../src/quota/ApiQuotaPoller.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;
|
|
1
|
+
{"version":3,"file":"ApiQuotaPoller.d.ts","sourceRoot":"","sources":["../../src/quota/ApiQuotaPoller.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAoClD,MAAM,WAAW,SAAS;IACxB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,WAAW,EAAE,OAAO,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAaD,qBAAa,mBAAoB,SAAQ,KAAK;gBAChC,OAAO,EAAE,MAAM;CAI5B;AAID,qBAAa,cAAc;;YAQX,kBAAkB;YAuClB,cAAc;IA4B5B,OAAO,CAAC,gBAAgB;YASV,oBAAoB;IAgC5B,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAuDjE,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAwB5F,YAAY,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;CAqBhF"}
|
|
@@ -1,3 +1,18 @@
|
|
|
1
|
+
var __extends = (this && this.__extends) || (function () {
|
|
2
|
+
var extendStatics = function (d, b) {
|
|
3
|
+
extendStatics = Object.setPrototypeOf ||
|
|
4
|
+
({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
|
|
5
|
+
function (d, b) { for (var p in b) if (Object.prototype.hasOwnProperty.call(b, p)) d[p] = b[p]; };
|
|
6
|
+
return extendStatics(d, b);
|
|
7
|
+
};
|
|
8
|
+
return function (d, b) {
|
|
9
|
+
if (typeof b !== "function" && b !== null)
|
|
10
|
+
throw new TypeError("Class extends value " + String(b) + " is not a constructor or null");
|
|
11
|
+
extendStatics(d, b);
|
|
12
|
+
function __() { this.constructor = d; }
|
|
13
|
+
d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
|
|
14
|
+
};
|
|
15
|
+
})();
|
|
1
16
|
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
|
|
2
17
|
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
|
|
3
18
|
return new (P || (P = Promise))(function (resolve, reject) {
|
|
@@ -61,16 +76,26 @@ var __read = (this && this.__read) || function (o, n) {
|
|
|
61
76
|
}
|
|
62
77
|
return ar;
|
|
63
78
|
};
|
|
64
|
-
var ANTIGRAVITY_CLIENT_ID = '
|
|
65
|
-
var ANTIGRAVITY_CLIENT_SECRET = '
|
|
79
|
+
var ANTIGRAVITY_CLIENT_ID = '1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com';
|
|
80
|
+
var ANTIGRAVITY_CLIENT_SECRET = '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';
|
|
66
81
|
var GOOGLE_TOKEN_URL = 'https://oauth2.googleapis.com/token';
|
|
67
|
-
var CLOUDCODE_BASE_URL = 'https://cloudcode.googleapis.com';
|
|
82
|
+
var CLOUDCODE_BASE_URL = 'https://cloudcode-pa.googleapis.com';
|
|
68
83
|
var CLOUDCODE_METADATA = {
|
|
69
84
|
clientName: 'code-oss',
|
|
70
85
|
versionString: '1.95.2',
|
|
71
86
|
extensionVersion: '1.11.13',
|
|
72
87
|
ideName: 'vscode',
|
|
73
88
|
};
|
|
89
|
+
var AuthenticationError = (function (_super) {
|
|
90
|
+
__extends(AuthenticationError, _super);
|
|
91
|
+
function AuthenticationError(message) {
|
|
92
|
+
var _this = _super.call(this, message) || this;
|
|
93
|
+
_this.name = 'AuthenticationError';
|
|
94
|
+
return _this;
|
|
95
|
+
}
|
|
96
|
+
return AuthenticationError;
|
|
97
|
+
}(Error));
|
|
98
|
+
export { AuthenticationError };
|
|
74
99
|
var ApiQuotaPoller = (function () {
|
|
75
100
|
function ApiQuotaPoller() {
|
|
76
101
|
if (!(this instanceof ApiQuotaPoller)) {
|
|
@@ -79,7 +104,7 @@ var ApiQuotaPoller = (function () {
|
|
|
79
104
|
}
|
|
80
105
|
ApiQuotaPoller.prototype.refreshAccessToken = function (refreshToken) {
|
|
81
106
|
return __awaiter(this, void 0, void 0, function () {
|
|
82
|
-
var params, response, data;
|
|
107
|
+
var params, controller, timeout, response, data, e_1;
|
|
83
108
|
return __generator(this, function (_a) {
|
|
84
109
|
switch (_a.label) {
|
|
85
110
|
case 0:
|
|
@@ -89,44 +114,75 @@ var ApiQuotaPoller = (function () {
|
|
|
89
114
|
refresh_token: refreshToken,
|
|
90
115
|
grant_type: 'refresh_token',
|
|
91
116
|
});
|
|
117
|
+
controller = new AbortController();
|
|
118
|
+
timeout = setTimeout(function () { return controller.abort(); }, 10000);
|
|
119
|
+
_a.label = 1;
|
|
120
|
+
case 1:
|
|
121
|
+
_a.trys.push([1, 4, 5, 6]);
|
|
92
122
|
return [4, fetch(GOOGLE_TOKEN_URL, {
|
|
93
123
|
method: 'POST',
|
|
94
124
|
headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
|
|
95
125
|
body: params.toString(),
|
|
126
|
+
signal: controller.signal,
|
|
96
127
|
})];
|
|
97
|
-
case
|
|
128
|
+
case 2:
|
|
98
129
|
response = _a.sent();
|
|
99
130
|
if (!response.ok) {
|
|
131
|
+
console.warn("refreshAccessToken: Failed ".concat(response.status));
|
|
132
|
+
if (response.status === 401 || response.status === 403) {
|
|
133
|
+
throw new AuthenticationError("Token refresh failed (".concat(response.status, ")"));
|
|
134
|
+
}
|
|
100
135
|
throw new Error("Token refresh failed (".concat(response.status, ")"));
|
|
101
136
|
}
|
|
102
137
|
return [4, response.json()];
|
|
103
|
-
case
|
|
138
|
+
case 3:
|
|
104
139
|
data = (_a.sent());
|
|
105
140
|
return [2, data.access_token];
|
|
141
|
+
case 4:
|
|
142
|
+
e_1 = _a.sent();
|
|
143
|
+
console.warn('refreshAccessToken: Error', e_1.message);
|
|
144
|
+
throw e_1;
|
|
145
|
+
case 5:
|
|
146
|
+
clearTimeout(timeout);
|
|
147
|
+
return [7];
|
|
148
|
+
case 6: return [2];
|
|
106
149
|
}
|
|
107
150
|
});
|
|
108
151
|
});
|
|
109
152
|
};
|
|
110
153
|
ApiQuotaPoller.prototype.loadCodeAssist = function (accessToken) {
|
|
111
154
|
return __awaiter(this, void 0, void 0, function () {
|
|
112
|
-
var response;
|
|
155
|
+
var controller, timeout, response;
|
|
113
156
|
return __generator(this, function (_a) {
|
|
114
157
|
switch (_a.label) {
|
|
115
|
-
case 0:
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
'Content-Type': 'application/json',
|
|
120
|
-
'User-Agent': 'antigravity',
|
|
121
|
-
},
|
|
122
|
-
body: JSON.stringify({ metadata: CLOUDCODE_METADATA }),
|
|
123
|
-
})];
|
|
158
|
+
case 0:
|
|
159
|
+
controller = new AbortController();
|
|
160
|
+
timeout = setTimeout(function () { return controller.abort(); }, 10000);
|
|
161
|
+
_a.label = 1;
|
|
124
162
|
case 1:
|
|
163
|
+
_a.trys.push([1, , 4, 5]);
|
|
164
|
+
return [4, fetch("".concat(CLOUDCODE_BASE_URL, "/v1internal:loadCodeAssist"), {
|
|
165
|
+
method: 'POST',
|
|
166
|
+
headers: {
|
|
167
|
+
Authorization: "Bearer ".concat(accessToken),
|
|
168
|
+
'Content-Type': 'application/json',
|
|
169
|
+
'User-Agent': 'antigravity',
|
|
170
|
+
},
|
|
171
|
+
body: JSON.stringify({ metadata: CLOUDCODE_METADATA }),
|
|
172
|
+
signal: controller.signal,
|
|
173
|
+
})];
|
|
174
|
+
case 2:
|
|
125
175
|
response = _a.sent();
|
|
126
|
-
if (!response.ok)
|
|
176
|
+
if (!response.ok) {
|
|
177
|
+
console.warn("loadCodeAssist: Failed ".concat(response.status));
|
|
127
178
|
throw new Error("loadCodeAssist failed (".concat(response.status, ")"));
|
|
179
|
+
}
|
|
128
180
|
return [4, response.json()];
|
|
129
|
-
case
|
|
181
|
+
case 3: return [2, (_a.sent())];
|
|
182
|
+
case 4:
|
|
183
|
+
clearTimeout(timeout);
|
|
184
|
+
return [7];
|
|
185
|
+
case 5: return [2];
|
|
130
186
|
}
|
|
131
187
|
});
|
|
132
188
|
});
|
|
@@ -142,11 +198,16 @@ var ApiQuotaPoller = (function () {
|
|
|
142
198
|
};
|
|
143
199
|
ApiQuotaPoller.prototype.fetchAvailableModels = function (accessToken, projectId) {
|
|
144
200
|
return __awaiter(this, void 0, void 0, function () {
|
|
145
|
-
var payload, response;
|
|
201
|
+
var payload, controller, timeout, response;
|
|
146
202
|
return __generator(this, function (_a) {
|
|
147
203
|
switch (_a.label) {
|
|
148
204
|
case 0:
|
|
149
205
|
payload = projectId ? { project: projectId } : {};
|
|
206
|
+
controller = new AbortController();
|
|
207
|
+
timeout = setTimeout(function () { return controller.abort(); }, 10000);
|
|
208
|
+
_a.label = 1;
|
|
209
|
+
case 1:
|
|
210
|
+
_a.trys.push([1, , 4, 5]);
|
|
150
211
|
return [4, fetch("".concat(CLOUDCODE_BASE_URL, "/v1internal:fetchAvailableModels"), {
|
|
151
212
|
method: 'POST',
|
|
152
213
|
headers: {
|
|
@@ -155,13 +216,20 @@ var ApiQuotaPoller = (function () {
|
|
|
155
216
|
'User-Agent': 'antigravity',
|
|
156
217
|
},
|
|
157
218
|
body: JSON.stringify(payload),
|
|
219
|
+
signal: controller.signal,
|
|
158
220
|
})];
|
|
159
|
-
case
|
|
221
|
+
case 2:
|
|
160
222
|
response = _a.sent();
|
|
161
|
-
if (!response.ok)
|
|
223
|
+
if (!response.ok) {
|
|
224
|
+
console.warn("fetchAvailableModels: Failed ".concat(response.status));
|
|
162
225
|
throw new Error("fetchModels failed (".concat(response.status, ")"));
|
|
226
|
+
}
|
|
163
227
|
return [4, response.json()];
|
|
164
|
-
case
|
|
228
|
+
case 3: return [2, (_a.sent())];
|
|
229
|
+
case 4:
|
|
230
|
+
clearTimeout(timeout);
|
|
231
|
+
return [7];
|
|
232
|
+
case 5: return [2];
|
|
165
233
|
}
|
|
166
234
|
});
|
|
167
235
|
});
|
|
@@ -169,7 +237,7 @@ var ApiQuotaPoller = (function () {
|
|
|
169
237
|
ApiQuotaPoller.prototype.checkQuota = function (account) {
|
|
170
238
|
return __awaiter(this, void 0, void 0, function () {
|
|
171
239
|
var accessToken, projectId, codeAssist, quotaResponse, models, _a, _b, _c, modelKey, info, quotaInfo, label, lowerLabel, remainingFraction, error_1;
|
|
172
|
-
var
|
|
240
|
+
var e_2, _d;
|
|
173
241
|
var _e;
|
|
174
242
|
return __generator(this, function (_f) {
|
|
175
243
|
switch (_f.label) {
|
|
@@ -215,17 +283,20 @@ var ApiQuotaPoller = (function () {
|
|
|
215
283
|
});
|
|
216
284
|
}
|
|
217
285
|
}
|
|
218
|
-
catch (
|
|
286
|
+
catch (e_2_1) { e_2 = { error: e_2_1 }; }
|
|
219
287
|
finally {
|
|
220
288
|
try {
|
|
221
289
|
if (_b && !_b.done && (_d = _a.return)) _d.call(_a);
|
|
222
290
|
}
|
|
223
|
-
finally { if (
|
|
291
|
+
finally { if (e_2) throw e_2.error; }
|
|
224
292
|
}
|
|
225
293
|
return [2, models];
|
|
226
294
|
case 5:
|
|
227
295
|
error_1 = _f.sent();
|
|
228
|
-
|
|
296
|
+
if (error_1 instanceof AuthenticationError) {
|
|
297
|
+
throw error_1;
|
|
298
|
+
}
|
|
299
|
+
console.warn('Failed to check quota via API:', error_1);
|
|
229
300
|
return [2, []];
|
|
230
301
|
case 6: return [2];
|
|
231
302
|
}
|
|
@@ -262,7 +333,7 @@ var ApiQuotaPoller = (function () {
|
|
|
262
333
|
ApiQuotaPoller.prototype.getAllQuotas = function (account) {
|
|
263
334
|
return __awaiter(this, void 0, void 0, function () {
|
|
264
335
|
var models, quotaMap, models_1, models_1_1, model;
|
|
265
|
-
var
|
|
336
|
+
var e_3, _a;
|
|
266
337
|
return __generator(this, function (_b) {
|
|
267
338
|
switch (_b.label) {
|
|
268
339
|
case 0: return [4, this.checkQuota(account)];
|
|
@@ -284,12 +355,12 @@ var ApiQuotaPoller = (function () {
|
|
|
284
355
|
});
|
|
285
356
|
}
|
|
286
357
|
}
|
|
287
|
-
catch (
|
|
358
|
+
catch (e_3_1) { e_3 = { error: e_3_1 }; }
|
|
288
359
|
finally {
|
|
289
360
|
try {
|
|
290
361
|
if (models_1_1 && !models_1_1.done && (_a = models_1.return)) _a.call(models_1);
|
|
291
362
|
}
|
|
292
|
-
finally { if (
|
|
363
|
+
finally { if (e_3) throw e_3.error; }
|
|
293
364
|
}
|
|
294
365
|
return [2, quotaMap];
|
|
295
366
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ApiQuotaPoller.js","sourceRoot":"","sources":["../../src/quota/ApiQuotaPoller.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"ApiQuotaPoller.js","sourceRoot":"","sources":["../../src/quota/ApiQuotaPoller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkDA,IAAM,qBAAqB,GAAG,2EAA2E,CAAC;AAC1G,IAAM,yBAAyB,GAAG,qCAAqC,CAAC;AACxE,IAAM,gBAAgB,GAAG,qCAAqC,CAAC;AAC/D,IAAM,kBAAkB,GAAG,qCAAqC,CAAC;AACjE,IAAM,kBAAkB,GAAG;IACzB,UAAU,EAAE,UAAU;IACtB,aAAa,EAAE,QAAQ;IACvB,gBAAgB,EAAE,SAAS;IAC3B,OAAO,EAAE,QAAQ;CAClB,CAAC;AAEF;IAAyC,uCAAK;IAC5C,6BAAY,OAAe;QACzB,YAAA,MAAK,YAAC,OAAO,CAAC,SAAC;QACf,KAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC;;IACpC,CAAC;IACH,0BAAC;AAAD,CAAC,AALD,CAAyC,KAAK,GAK7C;;AAID;IACE;QACE,IAAI,CAAC,CAAC,IAAI,YAAY,cAAc,CAAC,EAAE,CAAC;YAEtC,OAAO,IAAI,cAAc,EAAE,CAAC;QAC9B,CAAC;IACH,CAAC;IAEa,2CAAkB,GAAhC,UAAiC,YAAoB;;;;;;wBAE7C,MAAM,GAAG,IAAI,eAAe,CAAC;4BACjC,SAAS,EAAE,qBAAqB;4BAChC,aAAa,EAAE,yBAAyB;4BACxC,aAAa,EAAE,YAAY;4BAC3B,UAAU,EAAE,eAAe;yBAC5B,CAAC,CAAC;wBAEG,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;wBACnC,OAAO,GAAG,UAAU,CAAC,cAAM,OAAA,UAAU,CAAC,KAAK,EAAE,EAAlB,CAAkB,EAAE,KAAK,CAAC,CAAC;;;;wBAGzC,WAAM,KAAK,CAAC,gBAAgB,EAAE;gCAC7C,MAAM,EAAE,MAAM;gCACd,OAAO,EAAE,EAAE,cAAc,EAAE,mCAAmC,EAAE;gCAChE,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;gCACvB,MAAM,EAAE,UAAU,CAAC,MAAM;6BAC1B,CAAC,EAAA;;wBALI,QAAQ,GAAG,SAKf;wBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;4BACjB,OAAO,CAAC,IAAI,CAAC,qCAA8B,QAAQ,CAAC,MAAM,CAAE,CAAC,CAAC;4BAC9D,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gCACvD,MAAM,IAAI,mBAAmB,CAAC,gCAAyB,QAAQ,CAAC,MAAM,MAAG,CAAC,CAAC;4BAC7E,CAAC;4BACD,MAAM,IAAI,KAAK,CAAC,gCAAyB,QAAQ,CAAC,MAAM,MAAG,CAAC,CAAC;wBAC/D,CAAC;wBAEa,WAAM,QAAQ,CAAC,IAAI,EAAE,EAAA;;wBAA7B,IAAI,GAAG,CAAC,SAAqB,CAAkB;wBAErD,WAAO,IAAI,CAAC,YAAY,EAAC;;;wBAEzB,OAAO,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAC,CAAC,OAAO,CAAC,CAAC;wBACrD,MAAM,GAAC,CAAC;;wBAER,YAAY,CAAC,OAAO,CAAC,CAAC;;;;;;KAEzB;IAEa,uCAAc,GAA5B,UAA6B,WAAmB;;;;;;wBAExC,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;wBACnC,OAAO,GAAG,UAAU,CAAC,cAAM,OAAA,UAAU,CAAC,KAAK,EAAE,EAAlB,CAAkB,EAAE,KAAK,CAAC,CAAC;;;;wBAGzC,WAAM,KAAK,CAAC,UAAG,kBAAkB,+BAA4B,EAAE;gCAC9E,MAAM,EAAE,MAAM;gCACd,OAAO,EAAE;oCACP,aAAa,EAAE,iBAAU,WAAW,CAAE;oCACtC,cAAc,EAAE,kBAAkB;oCAClC,YAAY,EAAE,aAAa;iCAC5B;gCACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,kBAAkB,EAAE,CAAC;gCACtD,MAAM,EAAE,UAAU,CAAC,MAAM;6BAC1B,CAAC,EAAA;;wBATI,QAAQ,GAAG,SASf;wBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;4BACjB,OAAO,CAAC,IAAI,CAAC,iCAA0B,QAAQ,CAAC,MAAM,CAAE,CAAC,CAAC;4BAC1D,MAAM,IAAI,KAAK,CAAC,iCAA0B,QAAQ,CAAC,MAAM,MAAG,CAAC,CAAC;wBAChE,CAAC;wBAEO,WAAM,QAAQ,CAAC,IAAI,EAAE,EAAA;4BAA7B,WAAO,CAAC,SAAqB,CAA2B,EAAC;;wBAEzD,YAAY,CAAC,OAAO,CAAC,CAAC;;;;;;KAEzB;IAEO,yCAAgB,GAAxB,UAAyB,uBAAgC;QACvD,IAAI,CAAC,uBAAuB;YAAE,OAAO,SAAS,CAAC;QAC/C,IAAI,OAAO,uBAAuB,KAAK,QAAQ,EAAE,CAAC;YAChD,IAAM,KAAK,GAAG,uBAAuB,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACjE,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACtC,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEa,6CAAoB,GAAlC,UACE,WAAmB,EACnB,SAAkB;;;;;;wBAGZ,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBAClD,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;wBACnC,OAAO,GAAG,UAAU,CAAC,cAAM,OAAA,UAAU,CAAC,KAAK,EAAE,EAAlB,CAAkB,EAAE,KAAK,CAAC,CAAC;;;;wBAGzC,WAAM,KAAK,CAAC,UAAG,kBAAkB,qCAAkC,EAAE;gCACpF,MAAM,EAAE,MAAM;gCACd,OAAO,EAAE;oCACP,aAAa,EAAE,iBAAU,WAAW,CAAE;oCACtC,cAAc,EAAE,kBAAkB;oCAClC,YAAY,EAAE,aAAa;iCAC5B;gCACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gCAC7B,MAAM,EAAE,UAAU,CAAC,MAAM;6BAC1B,CAAC,EAAA;;wBATI,QAAQ,GAAG,SASf;wBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;4BACjB,OAAO,CAAC,IAAI,CAAC,uCAAgC,QAAQ,CAAC,MAAM,CAAE,CAAC,CAAC;4BAChE,MAAM,IAAI,KAAK,CAAC,8BAAuB,QAAQ,CAAC,MAAM,MAAG,CAAC,CAAC;wBAC7D,CAAC;wBAEO,WAAM,QAAQ,CAAC,IAAI,EAAE,EAAA;4BAA7B,WAAO,CAAC,SAAqB,CAA2B,EAAC;;wBAEzD,YAAY,CAAC,OAAO,CAAC,CAAC;;;;;;KAEzB;IAEK,mCAAU,GAAhB,UAAiB,OAA0B;;;;;;;;;wBAGnB,WAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC,EAAA;;wBAAjE,WAAW,GAAG,SAAmD;wBACnE,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,gBAAgB,CAAC;6BAE1D,CAAC,SAAS,EAAV,cAAU;wBAEO,WAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,EAAA;;wBAAnD,UAAU,GAAG,SAAsC;wBACzD,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC;;4BAGlD,WAAM,IAAI,CAAC,oBAAoB,CAAC,WAAW,EAAE,SAAS,CAAC,EAAA;;wBAAvE,aAAa,GAAG,SAAuD;wBAC7E,IAAI,CAAC,aAAa,CAAC,MAAM;4BAAE,WAAO,EAAE,EAAC;wBAE/B,MAAM,GAAqB,EAAE,CAAC;;4BAEpC,KAA+B,KAAA,SAAA,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA,4CAAE,CAAC;gCAA3D,KAAA,mBAAgB,EAAf,QAAQ,QAAA,EAAE,IAAI,QAAA;gCAClB,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;gCACjC,IAAI,CAAC,SAAS;oCAAE,SAAS;gCAEnB,KAAK,GAAG,IAAI,CAAC,WAAW,IAAI,QAAQ,CAAC;gCACrC,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;gCAGvC,IACE,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC;oCAC9B,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC;oCAC9B,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC;oCACjC,UAAU,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EACzC,CAAC;oCACD,SAAS;gCACX,CAAC;gCAEK,iBAAiB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAA,SAAS,CAAC,iBAAiB,mCAAI,CAAC,CAAC,CAAC,CAAC;gCAErF,MAAM,CAAC,IAAI,CAAC;oCACV,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,QAAQ;oCAC7B,WAAW,EAAE,KAAK;oCAClB,iBAAiB,mBAAA;oCACjB,WAAW,EAAE,iBAAiB,IAAI,CAAC;oCACnC,SAAS,EAAE,SAAS,CAAC,SAAS;iCAC/B,CAAC,CAAC;4BACL,CAAC;;;;;;;;;wBAED,WAAO,MAAM,EAAC;;;wBAEd,IAAI,OAAK,YAAY,mBAAmB,EAAE,CAAC;4BACzC,MAAM,OAAK,CAAC;wBACd,CAAC;wBACD,OAAO,CAAC,IAAI,CAAC,gCAAgC,EAAE,OAAK,CAAC,CAAC;wBACtD,WAAO,EAAE,EAAC;;;;;KAEb;IAEK,2CAAkB,GAAxB,UAAyB,OAA0B,EAAE,SAAiB;;;;;4BACrD,WAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAA;;wBAAvC,MAAM,GAAG,SAA8B;wBAGzC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,UAAA,CAAC,IAAI,OAAA,CAAC,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS,EAApD,CAAoD,CAAC,CAAC;wBAGnF,IAAI,CAAC,KAAK,EAAE,CAAC;4BACL,qBAAmB,SAAS,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;4BACvE,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,UAAA,CAAC;gCACnB,OAAA,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,kBAAgB,CAAC;oCAChD,CAAC,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,kBAAgB,CAAC;4BADtD,CACsD,CACvD,CAAC;wBACJ,CAAC;wBAED,IAAI,CAAC,KAAK;4BAAE,WAAO,IAAI,EAAC;wBAExB,WAAO;gCACL,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;gCAC1C,SAAS,EAAE,KAAK,CAAC,SAAS;gCAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;6BACnB,EAAC;;;;KACH;IAEK,qCAAY,GAAlB,UAAmB,OAA0B;;;;;;4BAC5B,WAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAA;;wBAAvC,MAAM,GAAG,SAA8B;wBACvC,QAAQ,GAAG,IAAI,GAAG,EAAqB,CAAC;;4BAE9C,KAAoB,WAAA,SAAA,MAAM,CAAA,gFAAE,CAAC;gCAAlB,KAAK;gCACd,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE;oCACxB,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;oCAC1C,SAAS,EAAE,KAAK,CAAC,SAAS;oCAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;iCACnB,CAAC,CAAC;gCAGH,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE;oCAC9B,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;oCAC1C,SAAS,EAAE,KAAK,CAAC,SAAS;oCAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;iCACnB,CAAC,CAAC;4BACL,CAAC;;;;;;;;;wBAED,WAAO,QAAQ,EAAC;;;;KACjB;IACH,qBAAC;AAAD,CAAC,AAxND,IAwNC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"QuotaCacheUpdater.d.ts","sourceRoot":"","sources":["../../src/quota/QuotaCacheUpdater.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"QuotaCacheUpdater.d.ts","sourceRoot":"","sources":["../../src/quota/QuotaCacheUpdater.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAoBrC,wBAAsB,iBAAiB,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAoBvE;AAED,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,OAAO,CAAgB;IAC/B,OAAO,CAAC,UAAU,CAAyB;IAC3C,OAAO,CAAC,gBAAgB,CAAU;gBAEtB,OAAO,EAAE,YAAY,EAAE,gBAAgB,GAAE,MAAe;IAmB9D,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAsBlC,KAAK,IAAI,IAAI;IASb,IAAI,IAAI,IAAI;CAMb;AAED,wBAAsB,sBAAsB,CAAC,gBAAgB,GAAE,MAAe,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAQ1G"}
|
|
@@ -46,57 +46,101 @@ function getQuotaCachePath() {
|
|
|
46
46
|
var homeDir = os.homedir();
|
|
47
47
|
return path.join(homeDir, '.config', 'opencode', 'quota-cache.json');
|
|
48
48
|
}
|
|
49
|
+
export function writeQuotaToCache(quota) {
|
|
50
|
+
return __awaiter(this, void 0, void 0, function () {
|
|
51
|
+
var cache, cachePath, cacheDir, error_1;
|
|
52
|
+
return __generator(this, function (_a) {
|
|
53
|
+
switch (_a.label) {
|
|
54
|
+
case 0:
|
|
55
|
+
_a.trys.push([0, 2, , 3]);
|
|
56
|
+
cache = {
|
|
57
|
+
percentage: Math.round(quota.remainingFraction * 100),
|
|
58
|
+
model: quota.model || 'unknown',
|
|
59
|
+
timestamp: Date.now(),
|
|
60
|
+
};
|
|
61
|
+
cachePath = getQuotaCachePath();
|
|
62
|
+
cacheDir = path.dirname(cachePath);
|
|
63
|
+
if (!fs.existsSync(cacheDir)) {
|
|
64
|
+
fs.mkdirSync(cacheDir, { recursive: true });
|
|
65
|
+
}
|
|
66
|
+
return [4, fs.promises.writeFile(cachePath, JSON.stringify(cache, null, 2), 'utf-8')];
|
|
67
|
+
case 1:
|
|
68
|
+
_a.sent();
|
|
69
|
+
return [3, 3];
|
|
70
|
+
case 2:
|
|
71
|
+
error_1 = _a.sent();
|
|
72
|
+
console.warn('Failed to write quota cache', error_1);
|
|
73
|
+
return [3, 3];
|
|
74
|
+
case 3: return [2];
|
|
75
|
+
}
|
|
76
|
+
});
|
|
77
|
+
});
|
|
78
|
+
}
|
|
49
79
|
var QuotaCacheUpdater = (function () {
|
|
50
80
|
function QuotaCacheUpdater(manager, updateIntervalMs) {
|
|
51
|
-
if (updateIntervalMs === void 0) { updateIntervalMs =
|
|
81
|
+
if (updateIntervalMs === void 0) { updateIntervalMs = 300000; }
|
|
82
|
+
if (!(this instanceof QuotaCacheUpdater)) {
|
|
83
|
+
return new QuotaCacheUpdater(manager, updateIntervalMs);
|
|
84
|
+
}
|
|
52
85
|
this.intervalId = null;
|
|
53
86
|
this.manager = manager;
|
|
54
|
-
|
|
87
|
+
if (!updateIntervalMs || isNaN(updateIntervalMs) || updateIntervalMs < 300000) {
|
|
88
|
+
this.updateIntervalMs = 300000;
|
|
89
|
+
}
|
|
90
|
+
else {
|
|
91
|
+
this.updateIntervalMs = updateIntervalMs;
|
|
92
|
+
}
|
|
93
|
+
this.updateCache = this.updateCache.bind(this);
|
|
94
|
+
this.start = this.start.bind(this);
|
|
95
|
+
this.stop = this.stop.bind(this);
|
|
55
96
|
}
|
|
56
97
|
QuotaCacheUpdater.prototype.updateCache = function () {
|
|
57
98
|
return __awaiter(this, void 0, void 0, function () {
|
|
58
|
-
var
|
|
99
|
+
var interval, quota, error_2;
|
|
100
|
+
var _this = this;
|
|
59
101
|
return __generator(this, function (_a) {
|
|
60
102
|
switch (_a.label) {
|
|
61
103
|
case 0:
|
|
62
|
-
|
|
63
|
-
|
|
104
|
+
interval = this.updateIntervalMs;
|
|
105
|
+
_a.label = 1;
|
|
64
106
|
case 1:
|
|
107
|
+
_a.trys.push([1, 5, 6, 7]);
|
|
108
|
+
return [4, this.manager.getQuotaViaApi()];
|
|
109
|
+
case 2:
|
|
65
110
|
quota = _a.sent();
|
|
66
|
-
if (!quota)
|
|
67
|
-
|
|
111
|
+
if (!quota) return [3, 4];
|
|
112
|
+
return [4, writeQuotaToCache(quota)];
|
|
113
|
+
case 3:
|
|
114
|
+
_a.sent();
|
|
115
|
+
_a.label = 4;
|
|
116
|
+
case 4: return [3, 7];
|
|
117
|
+
case 5:
|
|
118
|
+
error_2 = _a.sent();
|
|
119
|
+
console.warn('Failed to update quota cache (poller)', error_2);
|
|
120
|
+
return [3, 7];
|
|
121
|
+
case 6:
|
|
122
|
+
if (this.intervalId !== null) {
|
|
123
|
+
this.intervalId = setTimeout(function () {
|
|
124
|
+
_this.updateCache();
|
|
125
|
+
}, interval);
|
|
68
126
|
}
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
model: quota.model || 'unknown',
|
|
72
|
-
timestamp: Date.now(),
|
|
73
|
-
};
|
|
74
|
-
cachePath = getQuotaCachePath();
|
|
75
|
-
cacheDir = path.dirname(cachePath);
|
|
76
|
-
if (!fs.existsSync(cacheDir)) {
|
|
77
|
-
fs.mkdirSync(cacheDir, { recursive: true });
|
|
78
|
-
}
|
|
79
|
-
fs.writeFileSync(cachePath, JSON.stringify(cache, null, 2), 'utf-8');
|
|
80
|
-
return [3, 3];
|
|
81
|
-
case 2:
|
|
82
|
-
error_1 = _a.sent();
|
|
83
|
-
console.error('Failed to update quota cache:', error_1);
|
|
84
|
-
return [3, 3];
|
|
85
|
-
case 3: return [2];
|
|
127
|
+
return [7];
|
|
128
|
+
case 7: return [2];
|
|
86
129
|
}
|
|
87
130
|
});
|
|
88
131
|
});
|
|
89
132
|
};
|
|
90
133
|
QuotaCacheUpdater.prototype.start = function () {
|
|
91
134
|
var _this = this;
|
|
92
|
-
this.
|
|
93
|
-
|
|
135
|
+
if (this.intervalId)
|
|
136
|
+
return;
|
|
137
|
+
this.intervalId = setTimeout(function () {
|
|
94
138
|
_this.updateCache();
|
|
95
|
-
},
|
|
139
|
+
}, 100);
|
|
96
140
|
};
|
|
97
141
|
QuotaCacheUpdater.prototype.stop = function () {
|
|
98
142
|
if (this.intervalId) {
|
|
99
|
-
|
|
143
|
+
clearTimeout(this.intervalId);
|
|
100
144
|
this.intervalId = null;
|
|
101
145
|
}
|
|
102
146
|
};
|
|
@@ -106,7 +150,7 @@ export { QuotaCacheUpdater };
|
|
|
106
150
|
export function startQuotaCacheService() {
|
|
107
151
|
return __awaiter(this, arguments, void 0, function (updateIntervalMs) {
|
|
108
152
|
var manager, updater;
|
|
109
|
-
if (updateIntervalMs === void 0) { updateIntervalMs =
|
|
153
|
+
if (updateIntervalMs === void 0) { updateIntervalMs = 300000; }
|
|
110
154
|
return __generator(this, function (_a) {
|
|
111
155
|
switch (_a.label) {
|
|
112
156
|
case 0:
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"QuotaCacheUpdater.js","sourceRoot":"","sources":["../../src/quota/QuotaCacheUpdater.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"QuotaCacheUpdater.js","sourceRoot":"","sources":["../../src/quota/QuotaCacheUpdater.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAG1C,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AAQpB,SAAS,iBAAiB;IACxB,IAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAClD,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,kBAAkB,CAAC,CAAC;IAClE,CAAC;IACD,IAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,kBAAkB,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,UAAgB,iBAAiB,CAAC,KAAgB;;;;;;;oBAE9C,KAAK,GAAe;wBACxB,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,iBAAiB,GAAG,GAAG,CAAC;wBACrD,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,SAAS;wBAC/B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACtB,CAAC;oBAEI,SAAS,GAAG,iBAAiB,EAAE,CAAC;oBAChC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBAEzC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC7B,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC9C,CAAC;oBAED,WAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,EAAA;;oBAA/E,SAA+E,CAAC;;;;oBAGhF,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE,OAAK,CAAC,CAAC;;;;;;CAEtD;AAED;IAKE,2BAAY,OAAqB,EAAE,gBAAiC;QAAjC,iCAAA,EAAA,yBAAiC;QAClE,IAAI,CAAC,CAAC,IAAI,YAAY,iBAAiB,CAAC,EAAE,CAAC;YACzC,OAAO,IAAK,iBAAyB,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAGvB,IAAI,CAAC,gBAAgB,IAAI,KAAK,CAAC,gBAAgB,CAAC,IAAI,gBAAgB,GAAG,MAAM,EAAE,CAAC;YAC9E,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAEK,uCAAW,GAAjB;;;;;;;wBACQ,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC;;;;wBAGvB,WAAM,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAA;;wBAA3C,KAAK,GAAG,SAAmC;6BAE7C,KAAK,EAAL,cAAK;wBACP,WAAM,iBAAiB,CAAC,KAAK,CAAC,EAAA;;wBAA9B,SAA8B,CAAC;;;;;wBAGjC,OAAO,CAAC,IAAI,CAAC,uCAAuC,EAAE,OAAK,CAAC,CAAC;;;wBAI7D,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,EAAE,CAAC;4BAC7B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;gCAC3B,KAAI,CAAC,WAAW,EAAE,CAAC;4BACrB,CAAC,EAAE,QAAQ,CAAC,CAAC;wBACf,CAAC;;;;;;KAEJ;IAED,iCAAK,GAAL;QAAA,iBAOC;QANC,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QAG5B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;YAC3B,KAAI,CAAC,WAAW,EAAE,CAAC;QACrB,CAAC,EAAE,GAAG,CAAC,CAAC;IACV,CAAC;IAED,gCAAI,GAAJ;QACE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;IACH,CAAC;IACH,wBAAC;AAAD,CAAC,AA7DD,IA6DC;;AAED,MAAM,UAAgB,sBAAsB;wDAAC,gBAAiC;;QAAjC,iCAAA,EAAA,yBAAiC;;;;oBACtE,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;oBACnC,WAAM,OAAO,CAAC,UAAU,EAAE,EAAA;;oBAA1B,SAA0B,CAAC;oBAErB,OAAO,GAAG,IAAI,iBAAiB,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;oBACjE,OAAO,CAAC,KAAK,EAAE,CAAC;oBAEhB,WAAO,OAAO,EAAC;;;;CAChB"}
|
|
@@ -117,7 +117,7 @@ var HardLimitDetector = (function () {
|
|
|
117
117
|
if (quota.remainingFraction <= 0) {
|
|
118
118
|
nextModel = (_a = this.modelSelector) === null || _a === void 0 ? void 0 : _a.selectModel();
|
|
119
119
|
if (!nextModel) {
|
|
120
|
-
this.rotator.markCurrentExhausted();
|
|
120
|
+
this.rotator.markCurrentExhausted(undefined, quota.resetTime);
|
|
121
121
|
return [2, {
|
|
122
122
|
isExhausted: true,
|
|
123
123
|
shouldRotate: true,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"HardLimitDetector.js","sourceRoot":"","sources":["../../src/rotation/HardLimitDetector.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAU1D;IAQE,2BAAY,MAAqB;QAC/B,IAAI,CAAC,CAAC,IAAI,YAAY,iBAAiB,CAAC,EAAE,CAAC;YAEzC,OAAO,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,kBAAkB,EAAE,CAAC;QAC5C,IAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QAChD,IAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC;QAEtD,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACzD,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;QACtC,IAAI,CAAC,cAAc,GAAG,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,cAAc,KAAI,GAAG,CAAC;QACpD,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC1D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAE1B,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,eAAe,EAAE,CAAC;YAC5B,IAAM,QAAQ,GAA0B;gBACtC,eAAe,EAAE,MAAM,CAAC,eAAe;gBACvC,cAAc,EAAE,EAAE;gBAClB,cAAc,EAAE,IAAI,CAAC,cAAc;aACpC,CAAC;YACF,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAEK,0CAAc,GAApB,UAAqB,YAAoB;;;;;;;wBACjC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;wBACjD,IAAI,CAAC,OAAO,EAAE,CAAC;4BACb,WAAO;oCACL,WAAW,EAAE,KAAK;oCAClB,YAAY,EAAE,KAAK;oCACnB,OAAO,EAAE,yBAAyB;iCACnC,EAAC;wBACJ,CAAC;wBAEa,WAAM,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO,EAAE,YAAY,CAAC,EAAA;;wBAAtE,KAAK,GAAG,SAA8D;wBAE5E,IAAI,CAAC,KAAK,EAAE,CAAC;4BACX,WAAO;oCACL,WAAW,EAAE,KAAK;oCAClB,YAAY,EAAE,KAAK;oCACnB,OAAO,EAAE,mCAAmC;iCAC7C,EAAC;wBACJ,CAAC;wBAED,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;wBAEnD,IAAI,KAAK,CAAC,iBAAiB,IAAI,CAAC,EAAE,CAAC;4BAC3B,SAAS,GAAG,MAAA,IAAI,CAAC,aAAa,0CAAE,WAAW,EAAE,CAAC;4BAEpD,IAAI,CAAC,SAAS,EAAE,CAAC;gCACf,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"HardLimitDetector.js","sourceRoot":"","sources":["../../src/rotation/HardLimitDetector.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAU1D;IAQE,2BAAY,MAAqB;QAC/B,IAAI,CAAC,CAAC,IAAI,YAAY,iBAAiB,CAAC,EAAE,CAAC;YAEzC,OAAO,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,kBAAkB,EAAE,CAAC;QAC5C,IAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QAChD,IAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC;QAEtD,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACzD,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;QACtC,IAAI,CAAC,cAAc,GAAG,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,cAAc,KAAI,GAAG,CAAC;QACpD,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC1D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAE1B,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,eAAe,EAAE,CAAC;YAC5B,IAAM,QAAQ,GAA0B;gBACtC,eAAe,EAAE,MAAM,CAAC,eAAe;gBACvC,cAAc,EAAE,EAAE;gBAClB,cAAc,EAAE,IAAI,CAAC,cAAc;aACpC,CAAC;YACF,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAEK,0CAAc,GAApB,UAAqB,YAAoB;;;;;;;wBACjC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;wBACjD,IAAI,CAAC,OAAO,EAAE,CAAC;4BACb,WAAO;oCACL,WAAW,EAAE,KAAK;oCAClB,YAAY,EAAE,KAAK;oCACnB,OAAO,EAAE,yBAAyB;iCACnC,EAAC;wBACJ,CAAC;wBAEa,WAAM,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO,EAAE,YAAY,CAAC,EAAA;;wBAAtE,KAAK,GAAG,SAA8D;wBAE5E,IAAI,CAAC,KAAK,EAAE,CAAC;4BACX,WAAO;oCACL,WAAW,EAAE,KAAK;oCAClB,YAAY,EAAE,KAAK;oCACnB,OAAO,EAAE,mCAAmC;iCAC7C,EAAC;wBACJ,CAAC;wBAED,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;wBAEnD,IAAI,KAAK,CAAC,iBAAiB,IAAI,CAAC,EAAE,CAAC;4BAC3B,SAAS,GAAG,MAAA,IAAI,CAAC,aAAa,0CAAE,WAAW,EAAE,CAAC;4BAEpD,IAAI,CAAC,SAAS,EAAE,CAAC;gCACf,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;gCAE9D,WAAO;wCACL,WAAW,EAAE,IAAI;wCACjB,YAAY,EAAE,IAAI;wCAClB,OAAO,EAAE,gBAAS,YAAY,oDAAiD;qCAChF,EAAC;4BACJ,CAAC;4BAED,WAAO;oCACL,WAAW,EAAE,IAAI;oCACjB,YAAY,EAAE,IAAI;oCAClB,SAAS,WAAA;oCACT,OAAO,EAAE,gBAAS,YAAY,iDAAuC,SAAS,MAAG;iCAClF,EAAC;wBACJ,CAAC;wBAED,IAAI,KAAK,CAAC,iBAAiB,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;4BAC5C,SAAS,GAAG,MAAA,IAAI,CAAC,aAAa,0CAAE,WAAW,EAAE,CAAC;4BAEpD,IAAI,SAAS,IAAI,SAAS,KAAK,YAAY,EAAE,CAAC;gCAC5C,WAAO;wCACL,WAAW,EAAE,KAAK;wCAClB,YAAY,EAAE,IAAI;wCAClB,SAAS,WAAA;wCACT,OAAO,EAAE,gBAAS,YAAY,+BAAqB,CAAC,KAAK,CAAC,iBAAiB,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,8BAAoB,SAAS,MAAG;qCAC9H,EAAC;4BACJ,CAAC;wBACH,CAAC;wBAED,WAAO;gCACL,WAAW,EAAE,KAAK;gCAClB,YAAY,EAAE,KAAK;gCACnB,OAAO,EAAE,gBAAS,YAAY,kBAAQ,CAAC,KAAK,CAAC,iBAAiB,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAoB;6BACrG,EAAC;;;;KACH;IAEK,2CAAe,GAArB;;;;;;;wBACQ,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;wBACjD,IAAI,CAAC,OAAO;4BAAE,WAAO;wBAEN,WAAM,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,EAAA;;wBAAnD,MAAM,GAAG,SAA0C;;4BAEzD,KAA6B,KAAA,SAAA,MAAM,CAAC,OAAO,EAAE,CAAA,4CAAE,CAAC;gCAArC,KAAA,mBAAc,EAAb,KAAK,QAAA,EAAE,KAAK,QAAA;gCACtB,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;4BAC9C,CAAC;;;;;;;;;;;;;KACF;IAED,oDAAwB,GAAxB,UAAyB,QAA+B;QACtD,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACtE,CAAC;IAED,2CAAe,GAAf;QACE,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAEK,yCAAa,GAAnB;;;gBACE,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;gBACpC,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;;;;KAC9B;IACH,wBAAC;AAAD,CAAC,AAxHD,IAwHC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "opencode-antigravity-autopilot",
|
|
3
|
-
"version": "2.
|
|
3
|
+
"version": "2.1.0",
|
|
4
4
|
"description": "Intelligent quota monitoring and model auto-switching for OpenCode with Antigravity. Plugin-based, zero source modifications required.",
|
|
5
5
|
"main": "./dist/index.js",
|
|
6
6
|
"types": "./dist/index.d.ts",
|