fix: estimate image gen tokens from prompt length instead of hardcoding

2026-04-27 10:28:39 -04:00
parent 5ee539d95c
commit b1a72f5a10
1 changed files with 8 additions and 2 deletions
@@ -558,10 +558,16 @@ func (s *Server) handleImageGenerations(c *gin.Context) {
 		return
 	}

+	// Estimate tokens from prompt text (~4 chars per token)
+	promptTokens := uint32(len(req.Prompt) / 4)
+	if promptTokens < 1 {
+		promptTokens = 1
+	}
+
 	s.logRequest(startTime, clientID, providerName, req.Model, &models.Usage{
-		PromptTokens:     1,
+		PromptTokens:     promptTokens,
 		CompletionTokens: uint32(len(resp.Data)),
-		TotalTokens:      1 + uint32(len(resp.Data)),
+		TotalTokens:      promptTokens + uint32(len(resp.Data)),
 	}, nil, false)

 	c.JSON(http.StatusOK, resp)