Mirrors
/
oai-reverse-proxy
mirror of https://gitgud.io/khanon/oai-reverse-proxy.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixes post-startup checks

This commit is contained in:
nai-degen 2023-05-01 18:35:35 -05:00
parent 40eac590af
commit 94720b51e7
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/key-management/key-checker.ts
View File

@ -120,10 +120,11 @@ export class KeyChecker {
} }
this.log.info({ key: key.hash }, "Checking key..."); this.log.info({ key: key.hash }, "Checking key...");
let isInitialCheck = !key.lastChecked;
try { try {
// During the initial check we need to get the subscription first because // During the initial check we need to get the subscription first because
// trials have different behavior. // trials have different behavior.
if (!key.lastChecked) { if (isInitialCheck) {
const subscription = await this.getSubscription(key); const subscription = await this.getSubscription(key);
this.updateKey(key.hash, { isTrial: !subscription.has_payment_method }); this.updateKey(key.hash, { isTrial: !subscription.has_payment_method });
const [provisionedModels, usage] = await Promise.all([ const [provisionedModels, usage] = await Promise.all([
@ -164,6 +165,11 @@ export class KeyChecker {
} }
this.lastCheck = Date.now(); this.lastCheck = Date.now();
// Only enqueue the next check if this wasn't a startup check, since those
// are batched together elsewhere.
if (!isInitialCheck) {
this.scheduleNextCheck();
}
} }
private async getProvisionedModels( private async getProvisionedModels(