update:格式化代码

This commit is contained in:
hrz
2025-02-14 00:54:59 +08:00
parent eae590eb26
commit c5180df9ac
11 changed files with 34 additions and 34 deletions
+4 -1
View File
@@ -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}")