feat: support custom base url

everpcpc · Apr 22, 2024 · 72400a1 · 72400a1
1 parent bd634d0
commit 72400a1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/telegram/telegram.go b/telegram/telegram.go
@@ -38,6 +38,7 @@ type Config struct {
 	WhitelistChats []int64 `json:"whitelistChats"`
 	ComicPath      string  `json:"comicPath"`
 	DeleteDelay    string  `json:"deleteDelay"`
+	OpenAIBaseURL  string  `json:"openAIBaseURL"`
 	OpenAIKey      string  `json:"openAIKey"`
 }
 
@@ -113,8 +114,10 @@ func (b *Bot) WithMeilisearch(meili *meilisearch.Client) *Bot {
 	return b
 }
 
-func (b *Bot) WithOpenAI(key string) *Bot {
-	b.ai = openai.NewClient(key)
+func (b *Bot) WithOpenAI(baseURL, key string) *Bot {
+	config := openai.DefaultConfig(key)
+	config.BaseURL = baseURL
+	b.ai = openai.NewClientWithConfig(config)
 	return b
 }
 

diff --git a/yubari.go b/yubari.go
@@ -88,7 +88,7 @@ func main() {
 	telegramBot = telegramBot.WithPixiv(pixivBot)
 	if cfg.Telegram.OpenAIKey != "" {
 		logger.Debug("bot: openai: enabled")
-		telegramBot = telegramBot.WithOpenAI(cfg.Telegram.OpenAIKey)
+		telegramBot = telegramBot.WithOpenAI(cfg.Telegram.OpenAIBaseURL, cfg.Telegram.OpenAIKey)
 	}
 
 	rssUpdate := make(chan string)