Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2026-01-08 03:28:51 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

refactor: get balance config primarily from appConfig

Browse source
This commit is contained in:
Danny Avila 2025-08-18 03:45:02 -04:00
parent d853c10920
commit 1d2be247cf
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
11 changed files with 89 additions and 79 deletions
Download patch file Download diff file Expand all files Collapse all files

6
packages/api/src/middleware/balance.ts
View file

@ -5,7 +5,7 @@ import type { Model } from 'mongoose';
import type { BalanceUpdateFields } from '~/types';
export interface BalanceMiddlewareOptions {
getBalanceConfig: () => Promise<BalanceConfig | null>;
getBalanceConfig: ({ role }?: { role?: string }) => Promise<BalanceConfig | null>;
Balance: Model<IBalance>;
}
@ -82,7 +82,8 @@ export function createSetBalanceConfig({
) => Promise<void> {
return async (req: ServerRequest, res: ServerResponse, next: NextFunction): Promise<void> => {
try {
const balanceConfig = await getBalanceConfig();
const user = req.user as IUser & { _id: string | ObjectId };
const balanceConfig = await getBalanceConfig({ role: user?.role });
if (!balanceConfig?.enabled) {
return next();
}
@ -90,7 +91,6 @@ export function createSetBalanceConfig({
return next();
}
const user = req.user as IUser & { _id: string | ObjectId };
if (!user || !user._id) {
return next();
}