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

Merge pull request #905 from iceljc/master

Browse source 
add agent id in resource key
This commit is contained in:
iceljc 2025-02-28 13:03:44 -06:00 committed by GitHub
parent b5eade4c5e 152811cc72
commit 5c90f2cd8a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -60,12 +60,13 @@ public class TokenStatistics : ITokenStatistics
stat.SetState("llm_total_cost", total_cost, isNeedVersion: false, source: StateSource.Application); stat.SetState("llm_total_cost", total_cost, isNeedVersion: false, source: StateSource.Application);
// Save stats // Save stats
var agentId = message.CurrentAgentId ?? string.Empty;
var globalStats = _services.GetRequiredService<IBotSharpStatsService>(); var globalStats = _services.GetRequiredService<IBotSharpStatsService>();
var body = new BotSharpStatsInput var body = new BotSharpStatsInput
{ {
Metric = StatsMetric.AgentLlmCost, Metric = StatsMetric.AgentLlmCost,
Dimension = "agent", Dimension = "agent",
DimRefVal = message.CurrentAgentId, DimRefVal = agentId,
RecordTime = DateTime.UtcNow, RecordTime = DateTime.UtcNow,
IntervalType = StatsInterval.Day, IntervalType = StatsInterval.Day,
Data = [ Data = [
@ -75,7 +76,7 @@ public class TokenStatistics : ITokenStatistics
new StatsKeyValuePair("completion_cost_total", deltaCompletionCost) new StatsKeyValuePair("completion_cost_total", deltaCompletionCost)
] ]
}; };
globalStats.UpdateStats("global-llm-cost", body); globalStats.UpdateStats($"global-llm-cost-{agentId}", body);
} }
public void PrintStatistics() public void PrintStatistics()

5
src/Infrastructure/BotSharp.Logger/Hooks/GlobalStatsConversationHook.cs
View file

@ -25,17 +25,18 @@ public class GlobalStatsConversationHook : IContentGeneratingHook
// record agent call // record agent call
var globalStats = _services.GetRequiredService<IBotSharpStatsService>(); var globalStats = _services.GetRequiredService<IBotSharpStatsService>();
var agentId = message.CurrentAgentId ?? string.Empty;
var body = new BotSharpStatsInput var body = new BotSharpStatsInput
{ {
Metric = StatsMetric.AgentCall, Metric = StatsMetric.AgentCall,
Dimension = "agent", Dimension = "agent",
DimRefVal = message.CurrentAgentId ?? string.Empty, DimRefVal = agentId,
RecordTime = DateTime.UtcNow, RecordTime = DateTime.UtcNow,
IntervalType = StatsInterval.Day, IntervalType = StatsInterval.Day,
Data = [ Data = [
new StatsKeyValuePair("agent_call_count", 1) new StatsKeyValuePair("agent_call_count", 1)
] ]
}; };
globalStats.UpdateStats("global-agent-call", body); globalStats.UpdateStats($"global-agent-call-{agentId}", body);
} }
} }