From e5ba26febe8a61d460f2a611d37167ad20a75b82 Mon Sep 17 00:00:00 2001
From: zhayujie <yjzha1996@163.com>
Date: Wed, 6 Dec 2023 00:31:31 +0800
Subject: [PATCH] fix: tts voice base url

---
 bot/linkai/link_ai_bot.py    | 4 +++-
 voice/openai/openai_voice.py | 3 ++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/bot/linkai/link_ai_bot.py b/bot/linkai/link_ai_bot.py
index f308370..cd743b4 100644
--- a/bot/linkai/link_ai_bot.py
+++ b/bot/linkai/link_ai_bot.py
@@ -94,7 +94,7 @@ class LinkAIBot(Bot):
             file_id = context.kwargs.get("file_id")
             if file_id:
                 body["file_id"] = file_id
-            logger.info(f"[LINKAI] query={query}, app_code={app_code}, mode={body.get('model')}, file_id={file_id}")
+            logger.info(f"[LINKAI] query={query}, app_code={app_code}, model={body.get('model')}, file_id={file_id}")
             headers = {"Authorization": "Bearer " + linkai_api_key}
 
             # do http request
@@ -120,6 +120,8 @@ class LinkAIBot(Bot):
                 if response["choices"][0].get("img_urls"):
                     thread = threading.Thread(target=self._send_image, args=(context.get("channel"), context, response["choices"][0].get("img_urls")))
                     thread.start()
+                    if response["choices"][0].get("text_content"):
+                        reply_content = response["choices"][0].get("text_content")
                 return Reply(ReplyType.TEXT, reply_content)
 
             else:
diff --git a/voice/openai/openai_voice.py b/voice/openai/openai_voice.py
index 2dd3cbe..767353e 100644
--- a/voice/openai/openai_voice.py
+++ b/voice/openai/openai_voice.py
@@ -33,7 +33,8 @@ class OpenaiVoice(Voice):
 
     def textToVoice(self, text):
         try:
-            url = 'https://api.openai.com/v1/audio/speech'
+            api_base = conf().get("open_ai_api_base") or "https://api.openai.com/v1"
+            url = f'{api_base}/audio/speech'
             headers = {
                 'Authorization': 'Bearer ' + conf().get("open_ai_api_key"),
                 'Content-Type': 'application/json'