mirror of
https://gh-proxy.org/https://github.com/klinsmaya/new-api.git
synced 2026-11-08 01:17:09 +08:00
fix: temp disable /messages/count_tokens
This commit is contained in:
@@ -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)
|
||||
})
|
||||
|
||||
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user