(t *testing.T, summary codersdk.ChatCostSummary, modelConfigID, chatID uuid.UUID)
| 10169 | } |
| 10170 | |
| 10171 | func assertChatCostSummary(t *testing.T, summary codersdk.ChatCostSummary, modelConfigID, chatID uuid.UUID) { |
| 10172 | t.Helper() |
| 10173 | |
| 10174 | require.Equal(t, int64(1000), summary.TotalCostMicros) |
| 10175 | require.Equal(t, int64(2), summary.PricedMessageCount) |
| 10176 | require.Equal(t, int64(0), summary.UnpricedMessageCount) |
| 10177 | require.Equal(t, int64(200), summary.TotalInputTokens) |
| 10178 | require.Equal(t, int64(100), summary.TotalOutputTokens) |
| 10179 | require.Equal(t, int64(4000), summary.TotalRuntimeMs) |
| 10180 | |
| 10181 | require.Len(t, summary.ByModel, 1) |
| 10182 | require.Equal(t, modelConfigID, summary.ByModel[0].ModelConfigID) |
| 10183 | require.Equal(t, int64(1000), summary.ByModel[0].TotalCostMicros) |
| 10184 | require.Equal(t, int64(2), summary.ByModel[0].MessageCount) |
| 10185 | require.Equal(t, int64(4000), summary.ByModel[0].TotalRuntimeMs) |
| 10186 | |
| 10187 | require.Len(t, summary.ByChat, 1) |
| 10188 | require.Equal(t, chatID, summary.ByChat[0].RootChatID) |
| 10189 | require.Equal(t, int64(1000), summary.ByChat[0].TotalCostMicros) |
| 10190 | require.Equal(t, int64(2), summary.ByChat[0].MessageCount) |
| 10191 | require.Equal(t, int64(4000), summary.ByChat[0].TotalRuntimeMs) |
| 10192 | } |
| 10193 | |
| 10194 | func TestChatCostSummary(t *testing.T) { |
| 10195 | t.Parallel() |
no test coverage detected