Bladeren bron

小bug修复,更新groq检测

GXX 3 weken geleden
bovenliggende
commit
8ad88c539a
3 gewijzigde bestanden met toevoegingen van 14 en 2 verwijderingen
  1. 5 1
      chat_model.py
  2. 7 1
      main.py
  3. 2 0
      tts.py

+ 5 - 1
chat_model.py

@@ -72,7 +72,7 @@ def AI_chat(group_id ,word ,prompt):
             break
     return ans
 
-def groq_chat(messages, model = "deepseek-r1-distill-llama-70b"):
+def groq_chat(messages, model = "qwen/qwen3-32b"):
     """
     主对话通道,https://api.gxx12138.space/groq/v1
     """
@@ -177,6 +177,8 @@ def AI_lora_getpic_prompt(word):
     群聊id自动获取群聊记录来获取绘画提示词
     版本2
     """
+    if len(word) == 0:
+        return ""
     messages = [{'role': 'system', 'content': prompt_lora}]
     messages.append({'role': 'user', 'content': word})
     ans = siliflow_chat(messages).replace(", ", ",")
@@ -195,11 +197,13 @@ def get_ja_trans(group_id):
     for i in range(len(df) - 1, -1, -1):
         if df.loc[i, 'user_name'] == "猫猫":
             cat_msg = df.loc[i, 'message']
+            break
     if cat_msg == "":
         return "失败"
     # 开始进行翻译
     prompt = config_get('pro_trans') + cat_msg
     messages = [{'role': 'user', 'content': prompt}]
+    # ans = siliflow_chat(messages, "deepseek-ai/DeepSeek-V3")
     ans = siliflow_chat(messages)
     return ans.strip().replace("\n", "")
 

+ 7 - 1
main.py

@@ -270,9 +270,15 @@ def chat():
             else:
                 change2setu(raw_message, group_id, user_id)
         elif str(raw_message) == "看不懂中文":
+            print("开始日语语音合成")
             ja_text = get_ja_trans(group_id)
             bs64 = get_gpu_tts(ja_text)
-            send_group_vioce(group_id, "base64://"+bs64)
+            if bs64 == "获取失败":
+                print(bs64)
+                send_group_message_word(group_id, "转写失败")
+            else:
+                print("发送语音开始!")
+                send_group_vioce(group_id, "base64://"+bs64)
         elif my_name in raw_message:
             on_my_message(group_id, user_id, user_name, raw_message, message)
             print(

+ 2 - 0
tts.py

@@ -84,6 +84,8 @@ def get_gpu_tts(text, text_lang="ja",
     }
     # api返回wav文件,转为base64编码返回
     response = requests.post(url, json=paload)
+    if response.status_code != 200:
+        return "获取失败"
     bs64 = base64.b64encode(response.content).decode('utf-8')
     return bs64