SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

refine

Browse source
This commit is contained in:
Jicheng Lu 2025-01-24 15:22:23 -06:00
parent b66ac96f51
commit 66e77a0078
3 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/Infrastructure/BotSharp.Abstraction/Conversations/ITokenStatistics.cs
View file

@ -7,6 +7,6 @@ public interface ITokenStatistics
float Cost { get; }
void StartTimer();
void StopTimer();
void AddToken(TokenStatsModel stats);
void AddToken(TokenStatsModel stats, RoleDialogModel message);
void PrintStatistics();
}

4
src/Infrastructure/BotSharp.Core/Conversations/Services/TokenStatistics.cs
View file

@ -32,7 +32,7 @@ public class TokenStatistics : ITokenStatistics
_logger = logger;
}
public void AddToken(TokenStatsModel stats)
public void AddToken(TokenStatsModel stats, RoleDialogModel message)
{
_model = stats.Model;
_promptTokenCount += stats.PromptCount;
@ -63,7 +63,7 @@ public class TokenStatistics : ITokenStatistics
var body = new BotSharpStats
{
Category = StatCategory.LlmCost,
Group = $"Provider: {stats.Provider} | Model: {stats.Model}",
Group = $"Agent: {message.CurrentAgentId}",
Data = new Dictionary<string, object>
{
{ "prompt_token_count_total", stats.PromptCount },

2
src/Infrastructure/BotSharp.Logger/Hooks/TokenStatsConversationHook.cs
View file

@ -18,7 +18,7 @@ public class TokenStatsConversationHook : IContentGeneratingHook
public async Task AfterGenerated(RoleDialogModel message, TokenStatsModel tokenStats)
{
_tokenStatistics.StopTimer();
_tokenStatistics.AddToken(tokenStats);
_tokenStatistics.AddToken(tokenStats, message);
await Task.CompletedTask;
}
}