mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-06-12 03:00:09 +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>
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
import json
|
||
import os
|
||
from pathlib import Path
|
||
from typing import Any, Dict, Optional
|
||
|
||
|
||
class ProxyConfigManager:
|
||
"""全局代理配置，存 JSON 文件，支持前端动态修改。
|
||
|
||
作用范围：LLM API + 转写 API（Groq 等）+ yt-dlp 视频下载。
|
||
优先级：配置文件里 enabled=true 的 url > 环境变量 HTTP_PROXY/HTTPS_PROXY/ALL_PROXY。
|
||
这样桌面端/web 用户在设置页填，docker/服务器部署用环境变量兜底。
|
||
"""
|
||
|
||
def __init__(self, filepath: str = "config/proxy.json"):
|
||
self.path = Path(filepath)
|
||
self.path.parent.mkdir(parents=True, exist_ok=True)
|
||
|
||
def _read(self) -> Dict[str, Any]:
|
||
if not self.path.exists():
|
||
return {}
|
||
try:
|
||
with self.path.open("r", encoding="utf-8") as f:
|
||
return json.load(f)
|
||
except Exception:
|
||
return {}
|
||
|
||
def _write(self, data: Dict[str, Any]):
|
||
with self.path.open("w", encoding="utf-8") as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||
|
||
def get_config(self) -> Dict[str, Any]:
|
||
data = self._read()
|
||
return {
|
||
"enabled": bool(data.get("enabled", False)),
|
||
"url": data.get("url", "") or "",
|
||
}
|
||
|
||
def update_config(self, enabled: bool, url: Optional[str] = None) -> Dict[str, Any]:
|
||
data = self._read()
|
||
data["enabled"] = bool(enabled)
|
||
if url is not None:
|
||
data["url"] = url.strip()
|
||
self._write(data)
|
||
return self.get_config()
|
||
|
||
def get_proxy_url(self) -> Optional[str]:
|
||
"""返回当前生效的代理 URL；没有则 None。
|
||
|
||
- 配置文件 enabled=true 且 url 非空 → 用配置的 url
|
||
- 否则回退到环境变量（标准的 HTTP_PROXY / HTTPS_PROXY / ALL_PROXY，大小写都认）
|
||
"""
|
||
cfg = self.get_config()
|
||
if cfg["enabled"] and cfg["url"]:
|
||
return cfg["url"]
|
||
for key in ("HTTPS_PROXY", "https_proxy", "HTTP_PROXY", "http_proxy", "ALL_PROXY", "all_proxy"):
|
||
val = os.environ.get(key)
|
||
if val:
|
||
return val
|
||
return None
|