hugbot
Version:
Chatbot maker for HuggingFace Inference API and other AI API providers and backends.
48 lines • 3.03 kB
JavaScript
;
var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
if (kind === "m") throw new TypeError("Private method is not writable");
if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
};
var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
};
var _RateLimiter_instances, _RateLimiter_t1, _RateLimiter_delay, _RateLimiter_cooldown, _RateLimiter_setCooldown;
Object.defineProperty(exports, "__esModule", { value: true });
exports.RateLimiter = void 0;
class RateLimiter {
constructor(minInterval = 1000) {
_RateLimiter_instances.add(this);
_RateLimiter_t1.set(this, void 0);
_RateLimiter_delay.set(this, void 0);
_RateLimiter_cooldown.set(this, true);
__classPrivateFieldSet(this, _RateLimiter_delay, minInterval, "f");
__classPrivateFieldSet(this, _RateLimiter_t1, performance.now(), "f");
}
check() {
const t2 = performance.now();
const deltaT = t2 - __classPrivateFieldGet(this, _RateLimiter_t1, "f");
__classPrivateFieldSet(this, _RateLimiter_t1, performance.now(), "f");
if (deltaT > __classPrivateFieldGet(this, _RateLimiter_delay, "f")) {
__classPrivateFieldGet(this, _RateLimiter_instances, "m", _RateLimiter_setCooldown).call(this);
return;
}
else if (__classPrivateFieldGet(this, _RateLimiter_cooldown, "f")) {
__classPrivateFieldSet(this, _RateLimiter_cooldown, false, "f");
return;
}
else {
__classPrivateFieldSet(this, _RateLimiter_cooldown, false, "f");
throw new Error("Rate limiter: You are doing this too often. "
+ `Try again in ${__classPrivateFieldGet(this, _RateLimiter_delay, "f") - deltaT}ms`);
}
}
}
exports.RateLimiter = RateLimiter;
_RateLimiter_t1 = new WeakMap(), _RateLimiter_delay = new WeakMap(), _RateLimiter_cooldown = new WeakMap(), _RateLimiter_instances = new WeakSet(), _RateLimiter_setCooldown = function _RateLimiter_setCooldown() {
setTimeout(() => { __classPrivateFieldSet(this, _RateLimiter_cooldown, true, "f"); }, __classPrivateFieldGet(this, _RateLimiter_delay, "f"));
};
//# sourceMappingURL=RateLimiter.js.map