feat: Always buffer request body in ai-proxy plugin (#1155)

This commit is contained in:
Kent Dong
2024-07-25 19:35:39 +08:00
committed by GitHub
parent 10569f49ae
commit b6d07a157c
11 changed files with 30 additions and 72 deletions

View File

@@ -44,13 +44,7 @@ func (m *yiProvider) OnRequestHeaders(ctx wrapper.HttpContext, apiName ApiName,
_ = util.OverwriteRequestPath(yiChatCompletionPath)
_ = util.OverwriteRequestHost(yiDomain)
_ = util.OverwriteRequestAuthorization("Bearer " + m.config.GetRandomToken())
if m.contextCache == nil {
ctx.DontReadRequestBody()
} else {
_ = proxywasm.RemoveHttpRequestHeader("Content-Length")
}
_ = proxywasm.RemoveHttpRequestHeader("Content-Length")
return types.ActionContinue, nil
}