Spaces:
Paused
Paused
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 {
|