feat: Enhance token usage computation with improved message handling and context management

This commit is contained in:
Marco Beretta 2025-12-15 17:56:50 +01:00
parent e0610b0d14
commit 0420ed2e68
No known key found for this signature in database
GPG key ID: D918033D8E74CC11
3 changed files with 27 additions and 7 deletions

View file

@ -806,7 +806,6 @@ class BaseClient {
user,
);
this.savedMessageIds.add(responseMessage.messageId);
delete responseMessage.tokenCount;
return responseMessage;
}

View file

@ -240,6 +240,8 @@ class AgentClient extends BaseClient {
Object.assign(
{
endpoint: this.options.endpoint,
endpointType: this.options.endpointType,
model: this.options.agent?.model_parameters?.model,
agent_id: this.options.agent.id,
modelLabel: this.options.modelLabel,
maxContextTokens: this.options.maxContextTokens,