From 955b35682df2b7ac297d985ceeb48a557edbfc5e Mon Sep 17 00:00:00 2001
From: Hao Wu <wuhaoecho@gmail.com>
Date: Thu, 20 Apr 2023 12:39:54 +0800
Subject: [PATCH] update (#165)

---
 api/chat_main_handler.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/api/chat_main_handler.go b/api/chat_main_handler.go
index f620d444..49f5bce5 100644
--- a/api/chat_main_handler.go
+++ b/api/chat_main_handler.go
@@ -322,7 +322,15 @@ func (h *ChatHandler) chatStream(w http.ResponseWriter, chatSession sqlc_queries
 		RespondWithError(w, http.StatusInternalServerError, eris.Wrap(err, "get chat model").Error(), err)
 		return "", "", true
 	} else {
-		config.BaseURL = chat_model.Url
+		index := strings.Index(chat_model.Url, "/chat/")
+		if index != -1 {
+			// is full url https://api.openai.com/v1/chat/completions
+			baseUrl := chat_model.Url[:index]
+			config.BaseURL = baseUrl
+		} else {
+			config.BaseURL = chat_model.Url
+		}
+
 	}
 	client := openai.NewClientWithConfig(config)