feat(ark-context): update prompt token details

volcengine · Dec 17, 2024 · fd9d7a6 · fd9d7a6
1 parent ea91b35
commit fd9d7a6
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/service/arkruntime/model/common.go b/service/arkruntime/model/common.go
@@ -21,15 +21,15 @@ const (
 	ErrorRetryMaxDelay  = 8 * time.Second
 )
 
-type TokensDetail struct {
+type PromptTokensDetail struct {
 	CachedTokens int `json:"cached_tokens"`
 }
 
 type Usage struct {
-	PromptTokens        int           `json:"prompt_tokens"`
-	CompletionTokens    int           `json:"completion_tokens"`
-	TotalTokens         int           `json:"total_tokens"`
-	PromptTokensDetails *TokensDetail `json:"prompt_tokens_details,omitempty"`
+	PromptTokens        int                `json:"prompt_tokens"`
+	CompletionTokens    int                `json:"completion_tokens"`
+	TotalTokens         int                `json:"total_tokens"`
+	PromptTokensDetails PromptTokensDetail `json:"prompt_tokens_details"`
 }
 
 type Response interface {