mirror of
https://github.com/danny-avila/LibreChat.git
synced 2026-09-11 09:06:58 +00:00
e696b07619
`AgentClient.recordTokenUsage` had no `transactions` parameter, so the setting never reached `createTransaction`, whose guard reads it from the object it is handed. `transactions?.enabled === false` saw `undefined` and the write went ahead. This path is reached only from `BaseClient`'s fallback branch, when the provider returns no usable stream usage, so the bulk path masked it wherever usage is reported. Where it is not, the setting had no effect at all.
69 lines
2.1 KiB
JavaScript
69 lines
2.1 KiB
JavaScript
const mockSpendTokens = jest.fn().mockResolvedValue(undefined);
|
|
|
|
jest.mock('~/models', () => ({
|
|
spendTokens: mockSpendTokens,
|
|
}));
|
|
|
|
jest.mock('@librechat/data-schemas', () => ({
|
|
logger: { debug: jest.fn(), error: jest.fn(), warn: jest.fn(), info: jest.fn() },
|
|
}));
|
|
|
|
const AgentClient = require('../client');
|
|
|
|
describe('AgentClient#recordTokenUsage transactions config', () => {
|
|
/** @returns {AgentClient} */
|
|
const createClient = () =>
|
|
Object.create(AgentClient.prototype, {
|
|
user: { value: 'user-1' },
|
|
conversationId: { value: 'convo-1' },
|
|
responseMessageId: { value: 'msg-1' },
|
|
options: { value: { req: { user: { id: 'user-1' } }, endpointTokenConfig: undefined } },
|
|
});
|
|
|
|
beforeEach(() => {
|
|
mockSpendTokens.mockClear();
|
|
});
|
|
|
|
it('forwards the transactions config to spendTokens', async () => {
|
|
await createClient().recordTokenUsage({
|
|
model: 'gpt-4o-mini',
|
|
promptTokens: 10,
|
|
completionTokens: 5,
|
|
balance: { enabled: false },
|
|
transactions: { enabled: false },
|
|
});
|
|
|
|
expect(mockSpendTokens).toHaveBeenCalledTimes(1);
|
|
expect(mockSpendTokens.mock.calls[0][0]).toEqual(
|
|
expect.objectContaining({ transactions: { enabled: false } }),
|
|
);
|
|
});
|
|
|
|
it('forwards the transactions config on the reasoning-token spend', async () => {
|
|
await createClient().recordTokenUsage({
|
|
model: 'gpt-4o-mini',
|
|
promptTokens: 10,
|
|
completionTokens: 5,
|
|
transactions: { enabled: false },
|
|
usage: { reasoning_tokens: 7 },
|
|
});
|
|
|
|
expect(mockSpendTokens).toHaveBeenCalledTimes(2);
|
|
expect(mockSpendTokens.mock.calls[1][0]).toEqual(
|
|
expect.objectContaining({ context: 'reasoning', transactions: { enabled: false } }),
|
|
);
|
|
});
|
|
|
|
it('still records when transactions are enabled', async () => {
|
|
await createClient().recordTokenUsage({
|
|
model: 'gpt-4o-mini',
|
|
promptTokens: 10,
|
|
completionTokens: 5,
|
|
transactions: { enabled: true },
|
|
});
|
|
|
|
expect(mockSpendTokens.mock.calls[0][0]).toEqual(
|
|
expect.objectContaining({ transactions: { enabled: true } }),
|
|
);
|
|
});
|
|
});
|