[Bug]: Google Endpoint Balance Issues

[mattj2332]

What happened?

When CHECK_BALANCE is set to true, users are able to bypass the lack of tokens assigned to their account by switching to Google. OpenAI, Anthropic and all other providers honour the CHECK_BALANCE setting.

Currently running 0.7.5-rc2

Steps to Reproduce

1. Set CHECK_BALANCE=true in .env
2. Ensure that a valid API Key is present for Google and any other providers that will be tested.
3. Start LibreChat
4. Login with a user who has no balance assigned or insufficient balance
5. Change the provider to Google
6. Send a message

Step 6 should respond with insufficient funds.

Repeating steps 5 and 6 with OpenAI results in Insufficient Funds message as per screenshot

What browsers are you seeing the problem on?

No response

Relevant log output

No response

Screenshots

Code of Conduct

• I agree to follow this project's Code of Conduct

[chihebIA]

---

Issue: Balance Check Not Triggered for Google Endpoint

Problem:
The balance check functionality wasn't triggered for the Google endpoint due to the default configuration in supportsBalanceCheck not including Google.

Affected Code Snippet (api/app/clients/BaseClient.js):

if ( isEnabled(process.env.CHECK_BALANCE) && supportsBalanceCheck[this.options.endpointType ?? this.options.endpoint] ) { await checkBalance({ req: this.options.req, res: this.options.res, txData: { user: this.user, tokenType: 'prompt', amount: promptTokens, model: this.modelOptions.model, endpoint: this.options.endpoint, endpointTokenConfig: this.options.endpointTokenConfig, }, }); }

By default, the supportsBalanceCheck object didn't include Google as an endpoint for balance checking.

Solution:
To resolve this, I modified the supportsBalanceCheck configuration in packages/data-provider/src/config.ts to include Google. The change looks like this:

typescript

export const supportsBalanceCheck = { [EModelEndpoint.custom]: true, [EModelEndpoint.openAI]: true, [EModelEndpoint.anthropic]: true, [EModelEndpoint.gptPlugins]: true, [EModelEndpoint.assistants]: true, [EModelEndpoint.agents]: true, [EModelEndpoint.azureAssistants]: true, [EModelEndpoint.azureOpenAI]: true, [EModelEndpoint.bedrock]: true, [EModelEndpoint.google]: true, // Added Google endpoint };

With this addition, the balance check now works for the Google endpoint as expected.

---

[danny-avila]

Also #4580

The root cause of these issues is that Google had not published their prices for a long time when it was implemented, as their models were in "free trial."

I have a series of changes planned for the Google Endpoint that should address this.