小鱼开发
b597d715c8
后端： - 修复 get_current_user 未校验 is_active，被封禁用户仍可用旧 Token - auth.py 捕获 ValueError 转 HTTPException（验证码错误、账号被封、Token 无效等不再返回 500） - 修正 SMS 每日上限注释（3次 → 10次） - 修复迁移脚本外键引用错误：users.id → mjk_users.id - 新建积分系统 4 张表的迁移（mjk_user_points/batches/transactions/recharge_orders） - pyproject.toml 补充 alembic + psycopg2-binary 依赖 - ruff 格式修复（import 排序等） 前端： - 修复 doRefreshToken 成功后不持久化新 Token 的严重 bug - 修复应用重启后 SSE 不自动重连（收不到踢人通知） - 修复 App.tsx handleLogout 未 await - client.ts 统一从 utils/env 导入 isTauri，默认 base URL 兜底 localhost:8000 - 清理 ~20 个未使用的 hooks/utils/api 模块/组件导出 - 修复所有 ESLint 警告（206 → 0）和 TSC 错误 - 测试通过（5/5） 其他： - 更新 requirements.lock 和 uv.lock
116 lines
2.5 KiB
Python
116 lines
2.5 KiB
Python
"""
|
||
LLM Provider 抽象基类
|
||
=====================
|
||
|
||
定义所有 AI 模型提供商的统一接口。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from abc import ABC, abstractmethod
|
||
|
||
from pydantic import BaseModel
|
||
|
||
|
||
class ModelHealth(BaseModel):
|
||
"""模型健康状态"""
|
||
|
||
id: str
|
||
name: str
|
||
is_available: bool
|
||
response_time: float # 毫秒
|
||
last_error: str | None = None
|
||
|
||
|
||
class GenerationResult(BaseModel):
|
||
"""生成结果"""
|
||
|
||
content: str
|
||
usage: dict | None = None # token 用量等
|
||
model: str # 实际使用的模型
|
||
|
||
|
||
class LLMProvider(ABC):
|
||
"""
|
||
LLM 提供商抽象基类
|
||
|
||
所有 AI 模型提供商（OpenAI、文心一言、通义千问等）需实现此接口。
|
||
"""
|
||
|
||
# 提供商标识
|
||
provider_id: str = ""
|
||
provider_name: str = ""
|
||
|
||
def __init__(self, api_key: str | None = None, base_url: str | None = None, **kwargs):
|
||
"""
|
||
初始化 Provider
|
||
|
||
Args:
|
||
api_key: API 密钥
|
||
base_url: 自定义 Base URL（用于代理或私有部署）
|
||
**kwargs: 其他配置参数
|
||
"""
|
||
self.api_key = api_key
|
||
self.base_url = base_url
|
||
self.config = kwargs
|
||
|
||
@abstractmethod
|
||
async def generate(
|
||
self,
|
||
prompt: str,
|
||
model: str | None = None,
|
||
temperature: float = 0.7,
|
||
max_tokens: int | None = None,
|
||
**kwargs,
|
||
) -> GenerationResult:
|
||
"""
|
||
同步生成文本
|
||
|
||
Args:
|
||
prompt: 提示词
|
||
model: 模型名称，None 则使用默认模型
|
||
temperature: 随机性（0-2）
|
||
max_tokens: 最大生成 token 数
|
||
**kwargs: 额外参数
|
||
|
||
Returns:
|
||
GenerationResult: 生成结果
|
||
"""
|
||
pass
|
||
|
||
@abstractmethod
|
||
async def health_check(self, model: str | None = None) -> ModelHealth:
|
||
"""
|
||
健康检查
|
||
|
||
Args:
|
||
model: 指定模型，None 则检查默认模型
|
||
|
||
Returns:
|
||
ModelHealth: 健康状态
|
||
"""
|
||
pass
|
||
|
||
@property
|
||
@abstractmethod
|
||
def available_models(self) -> list[str]:
|
||
"""返回可用的模型列表"""
|
||
pass
|
||
|
||
|
||
class ProviderError(Exception):
|
||
"""Provider 调用异常"""
|
||
|
||
def __init__(
|
||
self, message: str, provider_id: str = "", original_error: Exception | None = None
|
||
):
|
||
super().__init__(message)
|
||
self.provider_id = provider_id
|
||
self.original_error = original_error
|
||
|
||
|
||
class ModelUnavailableError(ProviderError):
|
||
"""模型不可用异常"""
|
||
|
||
pass
|