track tokens openrouter with apikey

internal/handler/chat.go

@@ -52,7 +52,7 @@ func HandleChatCompletions(w http.ResponseWriter, r *http.Request) {
 	if m, ok := tempReq["model"].(string); ok && IsOpenRouterModel(m) {
 		ok2, reason := CheckAndTrack(apiKey, 0)
 		if !ok2 { http.Error(w, reason, 429); return }
-		HandleOpenRouter(w, r, body)
+		HandleOpenRouter(w, r, body, apiKey)
 		return
 	}
 	var req model.ChatRequest

internal/handler/openrouter.go

@@ -11,10 +11,10 @@ import (
 var openRouterKey = os.Getenv("OP_Rowproxy_key")
 
 var openRouterModels = map[string]string{
-	"deepseek-r1":           "deepseek/deepseek-r1",
-	"deepseek-v3":           "deepseek/deepseek-v3.2",
-	"deepseek-r1-0528":      "deepseek/deepseek-r1-0528",
-	"deepseek-v3-turbo":     "deepseek/deepseek-v3.1-terminus",
+	"deepseek-r1":       "deepseek/deepseek-r1",
+	"deepseek-v3":       "deepseek/deepseek-v3.2",
+	"deepseek-r1-0528":  "deepseek/deepseek-r1-0528",
+	"deepseek-v3-turbo": "deepseek/deepseek-v3.1-terminus",
 }
 
 func IsOpenRouterModel(model string) bool {
@@ -22,7 +22,7 @@ func IsOpenRouterModel(model string) bool {
 	return ok
 }
 
-func HandleOpenRouter(w http.ResponseWriter, r *http.Request, body []byte) {
+func HandleOpenRouter(w http.ResponseWriter, r *http.Request, body []byte, apiKey string) {
 	var req map[string]interface{}
 	json.Unmarshal(body, &req)
 	if m, ok := req["model"].(string); ok {
@@ -36,12 +36,18 @@ func HandleOpenRouter(w http.ResponseWriter, r *http.Request, body []byte) {
 	proxyReq.Header.Set("Content-Type", "application/json")
 	client := &http.Client{}
 	resp, err := client.Do(proxyReq)
-	if err != nil {
-		http.Error(w, "OpenRouter error", 500)
-		return
-	}
+	if err != nil { http.Error(w, "OpenRouter error", 500); return }
 	defer resp.Body.Close()
+	respBody, _ := io.ReadAll(resp.Body)
 	w.Header().Set("Content-Type", "application/json")
 	w.WriteHeader(resp.StatusCode)
-	io.Copy(w, resp.Body)
+	w.Write(respBody)
+	var result map[string]interface{}
+	if json.Unmarshal(respBody, &result) == nil {
+		if usage, ok := result["usage"].(map[string]interface{}); ok {
+			if total, ok := usage["total_tokens"].(float64); ok {
+				TrackUsage(apiKey, int(total))
+			}
+		}
+	}
 }