From fe6354528c8b0b8f496bdff36029a810c83dce96 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=A1=A3=E6=98=8E=E5=BF=97?= Date: Mon, 17 Apr 2023 02:58:11 +0800 Subject: [PATCH] support api key like openai style --- pkg/azure/proxy.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pkg/azure/proxy.go b/pkg/azure/proxy.go index 15927f2..432a7a5 100644 --- a/pkg/azure/proxy.go +++ b/pkg/azure/proxy.go @@ -81,6 +81,10 @@ func NewOpenAIReverseProxy() *httputil.ReverseProxy { token = AzureOpenAIToken } else { token = strings.ReplaceAll(req.Header.Get("Authorization"), "Bearer ", "") + // if start with 'sk-' then remove it + if strings.HasPrefix(token, "sk-") { + token = strings.Replace(token, "sk-", "", 1) + } } req.Header.Set("api-key", token)