feat(HEL-560): 数据中枢接管数据源/模型池/会员,注册改一次性邀请码
主站 - 新增 m0006 invite_codes 迁移;注册强制邀请码(首个管理员除外),消码与建号 同一事务,并发提交只有一个能成功 - 新增 /api/hub-admin/* 服务端点(共享 HUB_ADMIN_TOKEN,先于鉴权校验),供数据 中枢桥接读写会话/密码/模型池/会员/邀请码,并提供供应商模型列表拉取 - 前端:注册表单加邀请码(桌面 login、index.html、移动端);「系统管理」改为 「数据中枢」入口指向 8766,原模型池与会员管理分区移除,仅留「行情管理」; 随之清理陈旧 CSS 数据中枢 - 取消独立账号:删除 hub_admin/hub_sessions 与登录、改密、锁定逻辑,改为校验 主站 xiaobai_session,仅管理员可进,CSRF 由会话派生,危险操作二次确认走主站 - 控制台新增数据源凭证可编辑区(原有内容一项不删)、供应商制模型池(自动拉取 /models,失败退回卡内手动录入)、会员管理与邀请码页 - 日夜双主题:颜色收敛为同名 token 换值,SVG 改用 inline style 以吃到变量 自测 - 主站 verify_baseline 通过(498 项);数据中枢 235 项通过 - tools/verify_datahub_console.py 端到端跑通两服务真实对话; tools/verify_datahub_console_ui.py 浏览器跑通门禁/凭证/模型池/会员/主题/1030 窄屏 Co-authored-by: multica-agent <github@multica.ai>
This commit is contained in:
@@ -6,6 +6,7 @@ from urllib.parse import urlparse
|
||||
|
||||
from backend.bootstrap.config import validate_text
|
||||
from backend.features.screener.compiler import LLMCompilerError, test_llm_connection
|
||||
from backend.llm import transport as llm_transport
|
||||
|
||||
|
||||
class LLMServiceMixin:
|
||||
@@ -208,6 +209,37 @@ class LLMServiceMixin:
|
||||
start.isoformat(timespec="seconds"),
|
||||
)
|
||||
|
||||
def fetch_llm_models(self, base_url: str, api_key: str) -> list[str]:
|
||||
base_url = str(base_url or "").strip().rstrip("/")
|
||||
parsed = urlparse(base_url)
|
||||
if parsed.scheme not in {"http", "https"} or not parsed.netloc:
|
||||
raise ValueError("Base URL 格式不正确。")
|
||||
key = str(api_key or "").strip()
|
||||
if not key:
|
||||
key = self._stored_api_key(base_url)
|
||||
if not key:
|
||||
raise ValueError("该供应商尚未保存 API Key,请先填写后再拉取模型列表。")
|
||||
try:
|
||||
return llm_transport.list_models(
|
||||
api_key=key,
|
||||
base_url=base_url,
|
||||
timeout=15,
|
||||
user_agent="XiaobaiReviewWeb/0.5",
|
||||
)
|
||||
except llm_transport.OpenAIHTTPError as exc:
|
||||
raise ValueError(exc.describe("模型列表拉取失败")) from exc
|
||||
except llm_transport.OpenAITransportError as exc:
|
||||
raise ValueError(f"模型列表拉取失败:{exc}") from exc
|
||||
|
||||
def _stored_api_key(self, base_url: str) -> str:
|
||||
for item in self._system_credentials.get("llm_models") or []:
|
||||
if not isinstance(item, dict):
|
||||
continue
|
||||
stored = str(item.get("base_url") or "").strip().rstrip("/")
|
||||
if stored == base_url and item.get("api_key"):
|
||||
return str(item["api_key"])
|
||||
return ""
|
||||
|
||||
def test_system_llm_profile(self, model_id: str, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
current = next(
|
||||
(
|
||||
|
||||
@@ -67,6 +67,37 @@ def chat_completion(
|
||||
)
|
||||
|
||||
|
||||
def list_models(
|
||||
*,
|
||||
api_key: str,
|
||||
base_url: str,
|
||||
timeout: int,
|
||||
user_agent: str,
|
||||
) -> list[str]:
|
||||
request = urllib.request.Request(
|
||||
f"{base_url.rstrip('/')}/models",
|
||||
headers={
|
||||
"Authorization": f"Bearer {api_key}",
|
||||
"User-Agent": user_agent,
|
||||
},
|
||||
method="GET",
|
||||
)
|
||||
try:
|
||||
with urllib.request.urlopen(request, timeout=timeout) as response:
|
||||
payload = json.loads(response.read().decode("utf-8"))
|
||||
except urllib.error.HTTPError as exc:
|
||||
raise OpenAIHTTPError(exc.code, _http_error_detail(exc)) from exc
|
||||
except (urllib.error.URLError, TimeoutError, json.JSONDecodeError, OSError) as exc:
|
||||
raise OpenAITransportError(str(exc)) from exc
|
||||
items = payload.get("data") if isinstance(payload, dict) else payload
|
||||
models = []
|
||||
for item in items or []:
|
||||
name = str((item or {}).get("id") or "") if isinstance(item, dict) else str(item or "")
|
||||
if name and name not in models:
|
||||
models.append(name)
|
||||
return models
|
||||
|
||||
|
||||
def stream_chat_completion(
|
||||
*,
|
||||
api_key: str,
|
||||
|
||||
Reference in New Issue
Block a user