修改openai web chat可以设置代理api url

zatxm · May 13, 2024 · 287d8c3 · 287d8c3
1 parent f51004f
commit 287d8c3
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 6 deletions.
diff --git a/etc/c.yaml b/etc/c.yaml
@@ -23,6 +23,10 @@ openai:
     # auth_proxy_url: http://127.0.0.1:1081
     # web登录后放置cookie的文件夹
     cookie_path: /anp/data/cookies
+    # openai web chat url，可以修改为任意代理地址
+    # 不设置默认官方https://chatgpt.com容易出盾
+    # 目前chat.openai.com还能用，建议设置成这个
+    chat_web_url: https://chat.openai.com
 
 # 谷歌gemini接口
 # https://makersuite.google.com/app/apikey申请

diff --git a/internal/config/config.go b/internal/config/config.go
@@ -30,6 +30,7 @@ type httpsInfo struct {
 type openai struct {
 	AuthProxyUrl string `yaml:"auth_proxy_url"`
 	CookiePath   string `yaml:"cookie_path"`
+	ChatWebUrl   string `yaml:"chat_web_url"`
 }
 
 type gemini struct {
@@ -139,3 +140,7 @@ func ClaudeProxyUrl() string {
 func CozeProxyUrl() string {
 	return cfg.Coze.ProxyUrl
 }
+
+func OpenaiChatWebUrl() string {
+	return cfg.Openai.ChatWebUrl
+}
diff --git a/internal/openai/api/web.go b/internal/openai/api/web.go
@@ -134,8 +134,12 @@ func askConversationWebHttp(p types.OpenAiCompletionChatRequest, mt, auth string
 		}
 	}
 
+	webChatUrl := config.OpenaiChatWebUrl()
+	if webChatUrl == "" {
+		webChatUrl = cst.ChatOriginUrl
+	}
 	// anon token
-	requirementsUrl := chatCfg["requirementsUrl"]
+	requirementsUrl := webChatUrl + chatCfg["requirementsPath"]
 	req, err := http.NewRequest(http.MethodPost, requirementsUrl, nil)
 	if err != nil {
 		fhblade.Log.Error("chat-requirements new req err",
@@ -217,7 +221,7 @@ func askConversationWebHttp(p types.OpenAiCompletionChatRequest, mt, auth string
 	p.ForceRateLimit = false
 	p.WebsocketRequestId = uuid.NewString()
 	reqJson, _ := fhblade.Json.Marshal(p)
-	chatUrl := chatCfg["askUrl"]
+	chatUrl := webChatUrl + chatCfg["askPath"]
 	req, err = http.NewRequest(http.MethodPost, chatUrl, bytes.NewReader(reqJson))
 	if err != nil {
 		client.CcPool.Put(gClient)

diff --git a/internal/openai/cst/c.go b/internal/openai/cst/c.go
@@ -37,12 +37,12 @@ var (
 	OaiDeviceId = uuid.NewString()
 	ChatAskMap  = map[string]map[string]string{
 		"backend-anon": map[string]string{
-			"requirementsUrl": "https://chatgpt.com/backend-anon/sentinel/chat-requirements",
-			"askUrl":          "https://chatgpt.com/backend-anon/conversation",
+			"requirementsPath": "/backend-anon/sentinel/chat-requirements",
+			"askPath":          "/backend-anon/conversation",
 		},
 		"backend-api": map[string]string{
-			"requirementsUrl": "https://chatgpt.com/backend-api/sentinel/chat-requirements",
-			"askUrl":          "https://chatgpt.com/backend-api/conversation",
+			"requirementsPath": "/backend-api/sentinel/chat-requirements",
+			"askPath":          "/backend-api/conversation",
 		},
 	}
 )