mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-06-07 08:41:02 +08:00
refactor(backend): 重构后端异常处理和模型管理
- 新增自定义异常类 BizException、NoteError 和 ProviderError - 优化了模型管理相关的逻辑,包括加载、删除和测试连接等功能 - 改进了 Douyin 下载器的错误处理 - 调整了任务重试逻辑和笔记生成的异常处理- 更新了相关组件和页面以适应新的异常处理机制
This commit is contained in:
@@ -2,6 +2,9 @@ from typing import Optional, Union
|
||||
|
||||
from openai import OpenAI
|
||||
|
||||
from app.utils.logger import get_logger
|
||||
|
||||
logging= get_logger(__name__)
|
||||
class OpenAICompatibleProvider:
|
||||
def __init__(self, api_key: str, base_url: str, model: Union[str, None]=None):
|
||||
self.client = OpenAI(api_key=api_key, base_url=base_url)
|
||||
@@ -13,11 +16,16 @@ class OpenAICompatibleProvider:
|
||||
|
||||
@staticmethod
|
||||
def test_connection(api_key: str, base_url: str) -> bool:
|
||||
print(api_key)
|
||||
try:
|
||||
client = OpenAI(api_key=api_key, base_url=base_url)
|
||||
client.models.list()
|
||||
model = client.models.list()
|
||||
# for segment in model:
|
||||
# print(segment)
|
||||
# print(model)
|
||||
logging.info("连通性测试成功")
|
||||
return True
|
||||
except Exception as e:
|
||||
print(f"Error connecting to OpenAI API: {e}")
|
||||
logging.info(f"连通性测试失败:{e}")
|
||||
|
||||
# print(f"Error connecting to OpenAI API: {e}")
|
||||
return False
|
||||
Reference in New Issue
Block a user