coo7 commited on
Commit
bea8fae
·
verified ·
1 Parent(s): 506d434

Update internal/apiserver/router.go

Browse files
Files changed (1) hide show
  1. internal/apiserver/router.go +8 -8
internal/apiserver/router.go CHANGED
@@ -14,16 +14,16 @@ import (
14
 
15
  // RegisterRoutes 注册 Echo 路由
16
  func RegisterRoutes(e *echo.Echo) {
17
- // 添加Bearer Token认证中间件
18
- e.Use(middleware.BearerAuth())
19
-
20
- // 添加主页路由
21
  e.GET("/", handleHome)
22
 
23
- // ChatGPT 风格的请求转发到 /v1/chat/completions
24
- e.POST("/hf/v1/chat/completions", handleChatCompletion)
25
- // 获取支持的模型列表
26
- e.GET("/hf/v1/models", handleListModels)
 
 
 
27
  }
28
 
29
  // handleHome 处理主页请求
 
14
 
15
  // RegisterRoutes 注册 Echo 路由
16
  func RegisterRoutes(e *echo.Echo) {
17
+ // 主页路由不需要认证,所以要放在认证中间件之前
 
 
 
18
  e.GET("/", handleHome)
19
 
20
+ // 添加Bearer Token认证中间件,只对API路由生效
21
+ apiGroup := e.Group("/hf")
22
+ apiGroup.Use(middleware.BearerAuth())
23
+
24
+ // API路由
25
+ apiGroup.POST("/v1/chat/completions", handleChatCompletion)
26
+ apiGroup.GET("/v1/models", handleListModels)
27
  }
28
 
29
  // handleHome 处理主页请求