#4 get conversation endpoint from env

pavel-one · Apr 13, 2023 · f3b8c7b · f3b8c7b
1 parent d8c31bd
commit f3b8c7b
Showing 1 changed file with 9 additions and 9 deletions.
diff --git 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"
+}