update:格式化代码

core/providers/llm/base.py

@@ -1,7 +1,8 @@
 from abc import ABC, abstractmethod
 
+
 class LLMProviderBase(ABC):
     @abstractmethod
     def response(self, session_id, dialogue):
         """LLM response generator"""
-        pass
+        pass

core/providers/llm/dify/dify.py

@@ -5,6 +5,7 @@ from core.providers.llm.base import LLMProviderBase
 
 logger = logging.getLogger(__name__)
 
+
 class LLMProvider(LLMProviderBase):
     def __init__(self, config):
         self.api_key = config["api_key"]
@@ -35,4 +36,4 @@ class LLMProvider(LLMProviderBase):
 
         except Exception as e:
             logger.error(f"Error in response generation: {e}")
-            yield "【服务响应异常】"
+            yield "【服务响应异常】"

core/providers/llm/openai/openai.py

@@ -4,6 +4,7 @@ from core.providers.llm.base import LLMProviderBase
 
 logger = logging.getLogger(__name__)
 
+
 class LLMProvider(LLMProviderBase):
     def __init__(self, config):
         self.model_name = config.get("model_name")
@@ -12,6 +13,8 @@ class LLMProvider(LLMProviderBase):
             self.base_url = config.get("base_url")
         else:
             self.base_url = config.get("url")
+        if "你" in self.api_key:
+            logger.error("你还没配置LLM的密钥，请在配置文件中配置密钥，否则无法正常工作")
         self.client = openai.OpenAI(api_key=self.api_key, base_url=self.base_url)
 
     def response(self, session_id, dialogue):
@@ -29,4 +32,4 @@ class LLMProvider(LLMProviderBase):
                     if content:  # 仅在content非空时生成
                         yield content
         except Exception as e:
-            logger.error(f"Error in response generation: {e}")
+            logger.error(f"Error in response generation: {e}")

core/providers/tts/base.py

@@ -1,20 +1,14 @@
 import asyncio
 import logging
 import os
-import json
-import uuid
-import base64
-# from datetime import datetime
-# import edge_tts
 import numpy as np
 import opuslib
-# import requests
-# from core.utils.util import read_config, get_project_dir
 from pydub import AudioSegment
 from abc import ABC, abstractmethod
 
 logger = logging.getLogger(__name__)
 
+
 class TTSProviderBase(ABC):
     def __init__(self, config, delete_audio_file):
         self.delete_audio_file = delete_audio_file
@@ -34,8 +28,8 @@ class TTSProviderBase(ABC):
                     max_repeat_time = max_repeat_time - 1
                     logger.error(f"语音生成失败: {text}:{tmp_file}，再试{max_repeat_time}次")
 
-            if max_repeat_time>0:
-                logger.info(f"语音生成成功: {text}:{tmp_file}，重试{5-max_repeat_time}次")
+            if max_repeat_time > 0:
+                logger.info(f"语音生成成功: {text}:{tmp_file}，重试{5 - max_repeat_time}次")
 
             return tmp_file
         except Exception as e:
@@ -86,4 +80,4 @@ class TTSProviderBase(ABC):
             opus_data = encoder.encode(np_frame.tobytes(), frame_size)
             opus_datas.append(opus_data)
 
-        return opus_datas, duration
+        return opus_datas, duration

core/providers/tts/doubao.py

@@ -6,6 +6,7 @@ import requests
 from datetime import datetime
 from core.providers.tts.base import TTSProviderBase
 
+
 class TTSProvider(TTSProviderBase):
     def __init__(self, config, delete_audio_file):
         super().__init__(config, delete_audio_file)
@@ -52,4 +53,4 @@ class TTSProvider(TTSProviderBase):
         if "data" in resp.json():
             data = resp.json()["data"]
             file_to_save = open(output_file, "wb")
-            file_to_save.write(base64.b64decode(data))
+            file_to_save.write(base64.b64decode(data))

core/providers/tts/edge.py

@@ -4,6 +4,7 @@ import edge_tts
 from datetime import datetime
 from core.providers.tts.base import TTSProviderBase
 
+
 class TTSProvider(TTSProviderBase):
     def __init__(self, config, delete_audio_file):
         super().__init__(config, delete_audio_file)
@@ -14,4 +15,4 @@ class TTSProvider(TTSProviderBase):
 
     async def text_to_speak(self, text, output_file):
         communicate = edge_tts.Communicate(text, voice=self.voice)  # Use your preferred voice
-        await communicate.save(output_file)
+        await communicate.save(output_file)