12 lines
371 B
Go
12 lines
371 B
Go
|
package openai
|
||
|
|
|
||
|
|
import "github.com/songquanpeng/one-api/relay/model"
|
||
|
|
|
||
|
|
func ResponseText2Usage(responseText string, modeName string, promptTokens int) *model.Usage {
|
||
|
|
usage := &model.Usage{}
|
||
|
|
usage.PromptTokens = promptTokens
|
||
|
|
usage.CompletionTokens = CountTokenText(responseText, modeName)
|
||
|
|
usage.TotalTokens = usage.PromptTokens + usage.CompletionTokens
|
||
|
|
return usage
|
||
|
|
}
|