mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-06-12 19:20:00 +08:00
41f17592c2
- whisper: model.bin 截断/损坏时删目录重下重试一次，修「Unable to
open file model.bin」死循环；mlx 同样按 config.json 判完整性
- /generate_note 加就绪门禁：本地转写引擎模型没下好直接拦截，返回
reason=transcriber_model_not_ready，不让任务静默卡在首次下载
- 全局代理：新增 ProxyConfigManager（JSON 配置 + HTTP_PROXY env 兜底）
+ build_openai_client，统一注入代理到 LLM/Groq 客户端；yt-dlp 与
youtube-transcript-api 也走代理
- build_openai_client 校验 api_key 非空，空 key 给「xxx 的 API Key
未配置」而不是天书般的 Illegal header value b'Bearer '
- universal_gpt: 模型拒绝自定义 temperature（o1/o3/gpt-5 系列）时
就地去掉参数重试，不消耗重试预算
- connect_test 改用真实 chat completion 而非 /v1/models 探测
- main.py: lifespan 拆 [startup 1/5..5/5] 分段日志 + 异常清晰定位
- /sys_health 重构为结构化返回 {backend,ffmpeg,db,whisper_model}
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
from typing import Optional
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
|
|
from app.exceptions.provider import ProviderError
|
|
from app.models.model_config import ModelConfig
|
|
from app.services.model import ModelService
|
|
from app.utils.response import ResponseWrapper as R
|
|
from app.services.provider import ProviderService
|
|
|
|
router = APIRouter()
|
|
|
|
# 新增 type 字段
|
|
class ProviderRequest(BaseModel):
|
|
name: str
|
|
api_key: str
|
|
base_url: str
|
|
logo: Optional[str] = None
|
|
type: str
|
|
|
|
class TestRequest(BaseModel):
|
|
id: str
|
|
# 可选：指定用哪个 model 跑连通性测试；不传则用该 provider 在 DB 里的第一个模型
|
|
model: Optional[str] = None
|
|
class ProviderUpdateRequest(BaseModel):
|
|
id: str
|
|
name: Optional[str] = None
|
|
api_key: Optional[str] = None
|
|
base_url: Optional[str] = None
|
|
logo: Optional[str] = None
|
|
type: Optional[str] = None
|
|
enabled:Optional[int] = None
|
|
|
|
@router.post("/add_provider")
|
|
def add_provider(data: ProviderRequest):
|
|
try:
|
|
res = ProviderService.add_provider(
|
|
name=data.name,
|
|
api_key=data.api_key,
|
|
base_url=data.base_url,
|
|
logo=data.logo,
|
|
type_=data.type
|
|
)
|
|
return R.success(msg='添加模型供应商成功',data=res)
|
|
except Exception as e:
|
|
return R.error(msg=e)
|
|
|
|
@router.get("/get_all_providers")
|
|
def get_all_providers():
|
|
try:
|
|
res = ProviderService.get_all_providers_safe()
|
|
return R.success(data=res)
|
|
except Exception as e:
|
|
return R.error(msg=e)
|
|
|
|
@router.get("/get_provider_by_id/{id}")
|
|
def get_provider_by_id(id: str):
|
|
try:
|
|
res = ProviderService.get_provider_by_id_safe(id)
|
|
return R.success(data=res)
|
|
except Exception as e:
|
|
return R.error(msg=e)
|
|
#
|
|
# @router.get("/get_provider_by_name/{name}")
|
|
# def get_provider_by_name(name: str):
|
|
# try:
|
|
# res = ProviderService.get_provider_by_name(name)
|
|
# return R.success(data=res)
|
|
# except Exception as e:
|
|
# return R.error(msg=e)
|
|
|
|
|
|
@router.post("/update_provider")
|
|
def update_provider(data: ProviderUpdateRequest):
|
|
try:
|
|
if all(
|
|
field is None
|
|
for field in [data.name, data.api_key, data.base_url, data.logo, data.type,data.enabled]
|
|
):
|
|
return R.error(msg='请至少填写一个参数')
|
|
|
|
updated_provider =ProviderService.update_provider(
|
|
id=data.id,
|
|
data=dict(data)
|
|
)
|
|
if updated_provider:
|
|
return R.success(msg='更新模型供应商成功', data=updated_provider)
|
|
else:
|
|
return R.error(msg='更新模型供应商失败')
|
|
except Exception as e:
|
|
print(e)
|
|
return R.error(msg=str(e))
|
|
|
|
@router.post('/connect_test')
|
|
def gpt_connect_test(data: TestRequest):
|
|
ModelService().connect_test(data.id, model=data.model)
|
|
return R.success(msg='连接成功')
|