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
|
|
}
|