Commit 3a9f41ee by CaIon

fix: temp disable /messages/count_tokens

parent 32c26192
......@@ -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()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment