mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-03 13:00:52 +08:00
3dd5095792
- Implemented `TokenCount` transform method across translators to calculate token usage. - Integrated token counting logic into executor pipelines for Claude, Gemini, and CLI translators. - Added corresponding API endpoints and handlers (`/messages/count_tokens`) for token usage retrieval. - Enhanced translation registry to support `TokenCount` functionality alongside existing response types.
21 lines
492 B
Go
21 lines
492 B
Go
package claude
|
|
|
|
import (
|
|
. "github.com/router-for-me/CLIProxyAPI/v6/internal/constant"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/interfaces"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/translator/translator"
|
|
)
|
|
|
|
func init() {
|
|
translator.Register(
|
|
Claude,
|
|
Gemini,
|
|
ConvertClaudeRequestToGemini,
|
|
interfaces.TranslateResponse{
|
|
Stream: ConvertGeminiResponseToClaude,
|
|
NonStream: ConvertGeminiResponseToClaudeNonStream,
|
|
TokenCount: ClaudeTokenCount,
|
|
},
|
|
)
|
|
}
|