fix record usage

llm/types.go

@@ -2,6 +2,7 @@ package llm
 
 import (
 	"fmt"
+	"opencatd-open/internal/model"
 
 	"github.com/sashabaranov/go-openai"
 )
@@ -15,11 +16,14 @@ type StreamChatResponse openai.ChatCompletionStreamResponse
 type ChatMessage openai.ChatCompletionMessage
 
 type TokenUsage struct {
+	User             *model.User
+	TokenID          int64
 	Model            string `json:"model"`
-	PromptTokens     int    `json:"prompt_tokens"`
-	CompletionTokens int    `json:"completion_tokens"`
-	ToolsTokens      int    `json:"tools_tokens"`
-	TotalTokens      int    `json:"total_tokens"`
+	Stream           bool
+	PromptTokens     int `json:"prompt_tokens"`
+	CompletionTokens int `json:"completion_tokens"`
+	ToolsTokens      int `json:"tools_tokens"`
+	TotalTokens      int `json:"total_tokens"`
 }
 
 type ErrorResponse struct {