From d8c31bdffc0da4e33c577022146aff80c0ea2155 Mon Sep 17 00:00:00 2001 From: cherryReptile Date: Thu, 13 Apr 2023 18:15:41 +0300 Subject: [PATCH 1/2] env for region of endpoint --- config/gpt.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/config/gpt.go b/config/gpt.go index ab40b02..ed6c9f2 100644 --- a/config/gpt.go +++ b/config/gpt.go @@ -5,6 +5,7 @@ import ( "github.com/google/uuid" "github.com/pavel-one/EdgeGPT-Go/internal/Helpers" "net/url" + "os" "time" ) @@ -16,7 +17,15 @@ type GPT struct { } func NewGpt() (*GPT, error) { - cu, err := url.Parse("https://edgeservices.bing.com/edgesvc/turing/conversation/create") + var u string + switch os.Getenv("ENDPOINT_REGION") { + case "China": + u = "https://edge.churchless.tech/edgesvc/turing/conversation/create" + default: + u = "https://edgeservices.bing.com/edgesvc/turing/conversation/create" + } + + cu, err := url.Parse(u) if err != nil { return nil, err } From f3b8c7be9ffac96f89890c8a03d3f698a01d1ca8 Mon Sep 17 00:00:00 2001 From: cherryReptile Date: Thu, 13 Apr 2023 22:12:37 +0300 Subject: [PATCH 2/2] #4 get conversation endpoint from env --- config/gpt.go | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/config/gpt.go b/config/gpt.go index ed6c9f2..aa3472b 100644 --- a/config/gpt.go +++ b/config/gpt.go @@ -17,15 +17,7 @@ type GPT struct { } func NewGpt() (*GPT, error) { - var u string - switch os.Getenv("ENDPOINT_REGION") { - case "China": - u = "https://edge.churchless.tech/edgesvc/turing/conversation/create" - default: - u = "https://edgeservices.bing.com/edgesvc/turing/conversation/create" - } - - cu, err := url.Parse(u) + cu, err := url.Parse(getConversationEndpoint()) if err != nil { return nil, err } @@ -74,3 +66,11 @@ func NewGpt() (*GPT, error) { }, }, nil } + +func getConversationEndpoint() string { + er := os.Getenv("CONVERSATION_ENDPOINT") + if er != "" { + return er + } + return "https://edgeservices.bing.com/edgesvc/turing/conversation/create" +}