github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2025-12-26 06:28:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

jina-ai: fix account balance constraint

Browse Source
This commit is contained in:
yanlong.wang 2025-02-13 11:08:18 +08:00
parent bd77535dd9
commit fa4dccc94d
No known key found for this signature in database
GPG Key ID: C0A623C0BADF9F37
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jina-ai/src/patch-express.ts
View File

@ -8,6 +8,7 @@ import { JinaEmbeddingsAuthDTO } from "./dto/jina-embeddings-auth";
import rateLimitControl, { API_CALL_STATUS, RateLimitDesc } from "./rate-limit"; import rateLimitControl, { API_CALL_STATUS, RateLimitDesc } from "./rate-limit";
import asyncLocalContext from "./lib/async-context"; import asyncLocalContext from "./lib/async-context";
import globalLogger from "./lib/logger"; import globalLogger from "./lib/logger";
import { InsufficientBalanceError } from "./lib/errors";
globalLogger.serviceReady(); globalLogger.serviceReady();
const logger = globalLogger.child({ service: 'BillingMiddleware' }); const logger = globalLogger.child({ service: 'BillingMiddleware' });
@ -35,6 +36,9 @@ export const jinaAiBillingMiddleware = (req: Request, res: Response, next: NextF
}); });
const user = await authDto.assertUser(); const user = await authDto.assertUser();
if (!(user.wallet.total_balance > 0)) {
throw new InsufficientBalanceError(`Account balance not enough to run this query, please recharge.`);
}
await rateLimitControl.serviceReady(); await rateLimitControl.serviceReady();
const rateLimitPolicy = authDto.getRateLimits(appName) || [ const rateLimitPolicy = authDto.getRateLimits(appName) || [
parseInt(user.metadata?.speed_level) >= 2 ? parseInt(user.metadata?.speed_level) >= 2 ?