Skip to content

Commit 9706c95

Browse files
authored
Merge pull request #851 from iceljc/master
fix cost calculation
2 parents aef830d + 80df255 commit 9706c95

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/Infrastructure/BotSharp.Core/Conversations/Services/TokenStatistics.cs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ public void AddToken(TokenStatsModel stats, RoleDialogModel message)
4242
var settings = settingsService.GetSetting(stats.Provider, _model);
4343

4444
var deltaPromptCost = stats.PromptCount / 1000f * settings.PromptCost;
45-
var deltaCompletionCost = stats.CompletionCount / 1000 * settings.CompletionCost;
45+
var deltaCompletionCost = stats.CompletionCount / 1000f * settings.CompletionCost;
46+
var deltaTotal = deltaPromptCost + deltaCompletionCost;
4647
_promptCost += deltaPromptCost;
4748
_completionCost += deltaCompletionCost;
4849

@@ -55,10 +56,10 @@ public void AddToken(TokenStatsModel stats, RoleDialogModel message)
5556

5657
// Total cost
5758
var total_cost = float.Parse(stat.GetState("llm_total_cost", "0"));
58-
total_cost += Cost;
59+
total_cost += deltaTotal;
5960
stat.SetState("llm_total_cost", total_cost, isNeedVersion: false, source: StateSource.Application);
6061

61-
62+
// Save stats
6263
var globalStats = _services.GetRequiredService<IBotSharpStatsService>();
6364
var body = new BotSharpStatsInput
6465
{

0 commit comments

Comments
 (0)