fix: fix usage & ratio

2025-11-21 07:26:47 +08:00 · 2024-04-04 23:32:28 +08:00
parent 02539e8cd6
commit 17081393fc
2 changed files with 6 additions and 1 deletions
--- a/relay/channel/zhipu/main.go
+++ b/relay/channel/zhipu/main.go
@@ -285,7 +285,11 @@ func embeddingResponseZhipu2OpenAI(response *EmbeddingRespone) *openai.Embedding
 		Object: "list",
 		Data:   make([]openai.EmbeddingResponseItem, 0, len(response.Embeddings)),
 		Model:  response.Model,
-		Usage:  model.Usage{TotalTokens: response.Usage.TotalTokens},
+		Usage: model.Usage{
+			PromptTokens:     response.PromptTokens,
+			CompletionTokens: response.CompletionTokens,
+			TotalTokens:      response.Usage.TotalTokens,
+		},
 	}

 	for _, item := range response.Embeddings {