Compare commits

..

2 Commits

Author SHA1 Message Date
JustSong
3af4649b52 fix: only check model when request path in whitelist 2024-04-06 20:42:35 +08:00
GAI Group
52c32c0b4a chore: resolve the issue of onclick event scope for custom Lark button (#1281)
chore: Resolve the issue of onclick event scope for custom Lark button
2024-04-06 20:08:05 +08:00
2 changed files with 20 additions and 2 deletions

View File

@@ -116,7 +116,7 @@ func TokenAuth() func(c *gin.Context) {
return return
} }
requestModel, err := getRequestModel(c) requestModel, err := getRequestModel(c)
if err != nil && !strings.HasPrefix(c.Request.URL.Path, "/v1/models") { if err != nil && shouldCheckModel(c) {
abortWithMessage(c, http.StatusBadRequest, err.Error()) abortWithMessage(c, http.StatusBadRequest, err.Error())
return return
} }
@@ -142,3 +142,19 @@ func TokenAuth() func(c *gin.Context) {
c.Next() c.Next()
} }
} }
func shouldCheckModel(c *gin.Context) bool {
if strings.HasPrefix(c.Request.URL.Path, "/v1/completions") {
return true
}
if strings.HasPrefix(c.Request.URL.Path, "/v1/chat/completions") {
return true
}
if strings.HasPrefix(c.Request.URL.Path, "/v1/images") {
return true
}
if strings.HasPrefix(c.Request.URL.Path, "/v1/audio") {
return true
}
return false
}

View File

@@ -157,7 +157,9 @@ const LoginForm = () => {
borderRadius: "10em", borderRadius: "10em",
display: "flex", display: "flex",
cursor: "pointer" cursor: "pointer"
}}> }}
onClick={() => onLarkOAuthClicked(status.lark_client_id)}
>
<Image <Image
src={larkIcon} src={larkIcon}
avatar avatar