|
|
@@ -140,6 +140,7 @@ func testChannel(channel *model.Channel, testModel string) (err error, openAIErr
|
|
|
if err != nil {
|
|
|
return err, nil
|
|
|
}
|
|
|
+ info.PromptTokens = usage.PromptTokens
|
|
|
priceData, err := helper.ModelPriceHelper(c, info, usage.PromptTokens, int(request.MaxTokens))
|
|
|
if err != nil {
|
|
|
return err, nil
|
|
|
@@ -159,7 +160,7 @@ func testChannel(channel *model.Channel, testModel string) (err error, openAIErr
|
|
|
consumedTime := float64(milliseconds) / 1000.0
|
|
|
other := service.GenerateTextOtherInfo(c, info, priceData.ModelRatio, priceData.GroupRatio, priceData.CompletionRatio, priceData.ModelPrice)
|
|
|
model.RecordConsumeLog(c, 1, channel.Id, usage.PromptTokens, usage.CompletionTokens, testModel, "模型测试",
|
|
|
- quota, "模型测试", 0, quota, int(consumedTime), false, "default", other)
|
|
|
+ quota, "模型测试", 0, quota, int(consumedTime), false, info.Group, other)
|
|
|
common.SysLog(fmt.Sprintf("testing channel #%d, response: \n%s", channel.Id, string(respBody)))
|
|
|
return nil, nil
|
|
|
}
|