deny allow list if unknown model

This commit is contained in:
2024-01-21 15:37:12 +08:00
parent b1ab9c3d7b
commit 1fc17daa35

View File

@@ -59,6 +59,11 @@ func processRequest(c *gin.Context, upstream *OPENAI_UPSTREAM, record *Record, s
// record chat message from user
record.Body = string(inBody)
requestBody, requestBodyOK := ParseRequestBody(inBody)
// if there allow list and unknown body, return error
if len(upstream.Allow) > 0 && requestBodyOK != nil {
errCtx = append(errCtx, errors.New("[proxy.rewrite]: allow list but unknown model"))
return
}
// record if parse success
if requestBodyOK == nil {
record.Model = requestBody.Model