From 3a9f41ee85cc369f5b8d7fe6e62ff4e7bf3a9ec8 Mon Sep 17 00:00:00 2001 From: CaIon Date: Fri, 4 Sep 2026 10:07:03 +0800 Subject: [PATCH] fix: temp disable /messages/count_tokens --- router/relay-router.go | 3 ++- router/relay_router_test.go | 14 -------------- 2 files changed, 2 insertions(+), 15 deletions(-) diff --git a/router/relay-router.go b/router/relay-router.go index dd759fb8b..42b1882f0 100644 --- a/router/relay-router.go +++ b/router/relay-router.go @@ -85,7 +85,8 @@ func SetRelayRouter(router *gin.Engine) { httpRouter.Use(middleware.Distribute()) // claude related routes - httpRouter.POST("/messages/count_tokens", controller.CountClaudeTokens) + // TODO: /messages/count_tokens is disabled. The current controller.CountClaudeTokens + // httpRouter.POST("/messages/count_tokens", controller.CountClaudeTokens) httpRouter.POST("/messages", func(c *gin.Context) { controller.Relay(c, types.RelayFormatClaude) }) diff --git a/router/relay_router_test.go b/router/relay_router_test.go index 96ff09a29..579bd7bff 100644 --- a/router/relay_router_test.go +++ b/router/relay_router_test.go @@ -89,20 +89,6 @@ func TestListModelsSupportsOpenAIAndGeminiAuthentication(t *testing.T) { } } -func TestRelayRouterRegistersClaudeTokenCountingEndpoint(t *testing.T) { - gin.SetMode(gin.TestMode) - engine := gin.New() - SetRelayRouter(engine) - - for _, route := range engine.Routes() { - if route.Method == http.MethodPost && route.Path == "/v1/messages/count_tokens" { - return - } - } - - t.Fatal("POST /v1/messages/count_tokens route is not registered") -} - func setupRelayRouterTestDB(t *testing.T) { t.Helper()