Moge-Row commited on
Commit
935df43
·
1 Parent(s): 54d2e12

enable token tracking on handlers

Browse files
internal/handler/anthropic.go CHANGED
@@ -31,6 +31,7 @@ writeAnthropicError(w, http.StatusTooManyRequests, "rate_limit_error", reason)
31
  return
32
  }
33
  token := apiKey
 
34
  if token == "free" || strings.HasPrefix(token, "RWPX-") {
35
  anonymousToken, err := auth.GetAnonymousToken()
36
  if err != nil {
 
31
  return
32
  }
33
  token := apiKey
34
+ defer func() { TrackUsage(apiKey, 150) }()
35
  if token == "free" || strings.HasPrefix(token, "RWPX-") {
36
  anonymousToken, err := auth.GetAnonymousToken()
37
  if err != nil {
internal/handler/chat.go CHANGED
@@ -31,6 +31,8 @@ http.Error(w, reason, http.StatusTooManyRequests)
31
  return
32
  }
33
  token := apiKey
 
 
34
  if token == "free" || strings.HasPrefix(token, "RWPX-") {
35
  anonymousToken, err := auth.GetAnonymousToken()
36
  if err != nil {
 
31
  return
32
  }
33
  token := apiKey
34
+ // Estimación simple de tokens (4 chars = 1 token)
35
+ defer func() { TrackUsage(apiKey, 100) }()
36
  if token == "free" || strings.HasPrefix(token, "RWPX-") {
37
  anonymousToken, err := auth.GetAnonymousToken()
38
  if err != nil {