mirror of
https://github.com/fawney19/Aether.git
synced 2026-01-03 08:12:26 +08:00
33265b4b13
将模型配置从多个固定字段（description, official_url, icon_url, default_supports_* 等） 统一为灵活的 config JSON 字段，提高扩展性。同时优化前端模型创建表单，支持从 models-dev 列表直接选择模型快速填充。 主要变更： - 后端：模型表迁移，支持 config JSON 存储模型能力和元信息 - 前端：GlobalModelFormDialog 支持两种创建方式（列表选择/手动填写） - API 类型更新，对齐新的数据结构
287 lines
10 KiB
Python
287 lines
10 KiB
Python
"""
|
||
GlobalModel 服务层
|
||
|
||
提供 GlobalModel 的 CRUD 操作、查询和统计功能
|
||
"""
|
||
|
||
from typing import Dict, List, Optional
|
||
|
||
from sqlalchemy import and_, func
|
||
from sqlalchemy.orm import Session, joinedload
|
||
|
||
from src.core.exceptions import InvalidRequestException, NotFoundException
|
||
from src.core.logger import logger
|
||
from src.models.database import GlobalModel, Model
|
||
from src.models.pydantic_models import GlobalModelUpdate
|
||
|
||
|
||
|
||
class GlobalModelService:
|
||
"""GlobalModel 服务"""
|
||
|
||
@staticmethod
|
||
def get_global_model(db: Session, global_model_id: str) -> GlobalModel:
|
||
"""
|
||
获取单个 GlobalModel
|
||
|
||
Args:
|
||
global_model_id: GlobalModel 的 UUID 或 name
|
||
"""
|
||
# 先尝试通过 ID 查找
|
||
global_model = db.query(GlobalModel).filter(GlobalModel.id == global_model_id).first()
|
||
|
||
# 如果没找到，尝试通过 name 查找
|
||
if not global_model:
|
||
global_model = db.query(GlobalModel).filter(GlobalModel.name == global_model_id).first()
|
||
|
||
if not global_model:
|
||
raise NotFoundException(f"GlobalModel {global_model_id} not found")
|
||
return global_model
|
||
|
||
@staticmethod
|
||
def get_global_model_by_name(db: Session, name: str) -> Optional[GlobalModel]:
|
||
"""通过名称获取 GlobalModel"""
|
||
return db.query(GlobalModel).filter(GlobalModel.name == name).first()
|
||
|
||
@staticmethod
|
||
def list_global_models(
|
||
db: Session,
|
||
skip: int = 0,
|
||
limit: int = 100,
|
||
is_active: Optional[bool] = None,
|
||
search: Optional[str] = None,
|
||
) -> List[GlobalModel]:
|
||
"""列出 GlobalModel"""
|
||
query = db.query(GlobalModel)
|
||
|
||
if is_active is not None:
|
||
query = query.filter(GlobalModel.is_active == is_active)
|
||
|
||
if search:
|
||
search_pattern = f"%{search}%"
|
||
query = query.filter(
|
||
(GlobalModel.name.ilike(search_pattern))
|
||
| (GlobalModel.display_name.ilike(search_pattern))
|
||
)
|
||
|
||
# 按名称排序
|
||
query = query.order_by(GlobalModel.name)
|
||
|
||
return query.offset(skip).limit(limit).all()
|
||
|
||
@staticmethod
|
||
def create_global_model(
|
||
db: Session,
|
||
name: str,
|
||
display_name: str,
|
||
is_active: Optional[bool] = True,
|
||
# 按次计费配置
|
||
default_price_per_request: Optional[float] = None,
|
||
# 阶梯计费配置（必填）
|
||
default_tiered_pricing: dict = None,
|
||
# Key 能力配置
|
||
supported_capabilities: Optional[List[str]] = None,
|
||
# 模型配置（JSON）
|
||
config: Optional[dict] = None,
|
||
) -> GlobalModel:
|
||
"""创建 GlobalModel"""
|
||
# 检查名称是否已存在
|
||
existing = GlobalModelService.get_global_model_by_name(db, name)
|
||
if existing:
|
||
raise InvalidRequestException(f"GlobalModel with name '{name}' already exists")
|
||
|
||
global_model = GlobalModel(
|
||
name=name,
|
||
display_name=display_name,
|
||
is_active=is_active,
|
||
# 按次计费配置
|
||
default_price_per_request=default_price_per_request,
|
||
# 阶梯计费配置
|
||
default_tiered_pricing=default_tiered_pricing,
|
||
# Key 能力配置
|
||
supported_capabilities=supported_capabilities,
|
||
# 模型配置（JSON）
|
||
config=config,
|
||
)
|
||
|
||
db.add(global_model)
|
||
db.commit()
|
||
db.refresh(global_model)
|
||
|
||
return global_model
|
||
|
||
@staticmethod
|
||
def update_global_model(
|
||
db: Session,
|
||
global_model_id: str,
|
||
update_data: GlobalModelUpdate,
|
||
) -> GlobalModel:
|
||
"""
|
||
更新 GlobalModel
|
||
|
||
使用 exclude_unset=True 来区分"未提供字段"和"显式设置为 None"：
|
||
- 未提供的字段不会被更新
|
||
- 显式设置为 None 的字段会被更新为 None（置空）
|
||
"""
|
||
global_model = GlobalModelService.get_global_model(db, global_model_id)
|
||
|
||
# 只更新显式设置的字段（包括显式设置为 None 的情况）
|
||
data_dict = update_data.model_dump(exclude_unset=True)
|
||
|
||
# 处理阶梯计费配置：如果是 TieredPricingConfig 对象，转换为 dict
|
||
if "default_tiered_pricing" in data_dict:
|
||
tiered_pricing = data_dict["default_tiered_pricing"]
|
||
if tiered_pricing is not None and hasattr(tiered_pricing, "model_dump"):
|
||
data_dict["default_tiered_pricing"] = tiered_pricing.model_dump()
|
||
|
||
for field, value in data_dict.items():
|
||
setattr(global_model, field, value)
|
||
|
||
db.commit()
|
||
db.refresh(global_model)
|
||
|
||
return global_model
|
||
|
||
@staticmethod
|
||
def delete_global_model(db: Session, global_model_id: str) -> None:
|
||
"""
|
||
删除 GlobalModel
|
||
|
||
默认行为: 级联删除所有关联的 Provider 模型实现
|
||
"""
|
||
global_model = GlobalModelService.get_global_model(db, global_model_id)
|
||
|
||
# 查找所有关联的 Model（使用 global_model.id，预加载 provider 关联）
|
||
associated_models = (
|
||
db.query(Model)
|
||
.options(joinedload(Model.provider))
|
||
.filter(Model.global_model_id == global_model.id)
|
||
.all()
|
||
)
|
||
|
||
# 级联删除所有关联的 Provider 模型实现
|
||
if associated_models:
|
||
logger.info(f"删除 GlobalModel {global_model.name} 的 {len(associated_models)} 个关联 Provider 模型")
|
||
for model in associated_models:
|
||
db.delete(model)
|
||
|
||
# 删除 GlobalModel
|
||
db.delete(global_model)
|
||
db.commit()
|
||
|
||
@staticmethod
|
||
def get_global_model_stats(db: Session, global_model_id: str) -> Dict:
|
||
"""获取 GlobalModel 统计信息"""
|
||
global_model = GlobalModelService.get_global_model(db, global_model_id)
|
||
|
||
# 统计关联的 Model 数量（使用 global_model.id，预加载 provider 关联）
|
||
models = (
|
||
db.query(Model)
|
||
.options(joinedload(Model.provider))
|
||
.filter(Model.global_model_id == global_model.id)
|
||
.all()
|
||
)
|
||
|
||
# 统计支持的 Provider 数量
|
||
provider_ids = set(model.provider_id for model in models)
|
||
|
||
# 从阶梯计费中提取价格范围
|
||
input_prices = []
|
||
output_prices = []
|
||
for m in models:
|
||
tiered = m.get_effective_tiered_pricing()
|
||
if tiered and tiered.get("tiers"):
|
||
first_tier = tiered["tiers"][0]
|
||
if first_tier.get("input_price_per_1m") is not None:
|
||
input_prices.append(first_tier["input_price_per_1m"])
|
||
if first_tier.get("output_price_per_1m") is not None:
|
||
output_prices.append(first_tier["output_price_per_1m"])
|
||
|
||
return {
|
||
"global_model_id": global_model.id,
|
||
"name": global_model.name,
|
||
"total_models": len(models),
|
||
"total_providers": len(provider_ids),
|
||
"price_range": {
|
||
"min_input": min(input_prices) if input_prices else None,
|
||
"max_input": max(input_prices) if input_prices else None,
|
||
"min_output": min(output_prices) if output_prices else None,
|
||
"max_output": max(output_prices) if output_prices else None,
|
||
},
|
||
}
|
||
|
||
@staticmethod
|
||
def batch_assign_to_providers(
|
||
db: Session,
|
||
global_model_id: str,
|
||
provider_ids: List[str],
|
||
create_models: bool = False,
|
||
) -> Dict:
|
||
"""批量为多个 Provider 添加 GlobalModel 实现"""
|
||
from .service import ModelService
|
||
|
||
global_model = GlobalModelService.get_global_model(db, global_model_id)
|
||
|
||
results = {
|
||
"success": [],
|
||
"errors": [],
|
||
}
|
||
|
||
for provider_id in provider_ids:
|
||
try:
|
||
# 检查该 Provider 是否已有该 GlobalModel 的实现（使用 global_model.id）
|
||
existing_model = (
|
||
db.query(Model)
|
||
.filter(
|
||
Model.provider_id == provider_id,
|
||
Model.global_model_id == global_model.id,
|
||
)
|
||
.first()
|
||
)
|
||
|
||
if existing_model:
|
||
results["errors"].append(
|
||
{
|
||
"provider_id": provider_id,
|
||
"error": "Model already exists for this provider",
|
||
}
|
||
)
|
||
continue
|
||
|
||
if create_models:
|
||
# 创建新的 Model（价格和能力设为 None，继承 GlobalModel 默认值）
|
||
model = Model(
|
||
provider_id=provider_id,
|
||
global_model_id=global_model.id,
|
||
provider_model_name=global_model.name, # 默认使用 GlobalModel name
|
||
# 计费设为 None，使用 GlobalModel 默认值
|
||
price_per_request=None,
|
||
tiered_pricing=None,
|
||
# 能力设为 None，使用 GlobalModel 默认值
|
||
supports_vision=None,
|
||
supports_function_calling=None,
|
||
supports_streaming=None,
|
||
supports_extended_thinking=None,
|
||
is_active=True,
|
||
)
|
||
db.add(model)
|
||
db.commit()
|
||
|
||
results["success"].append(
|
||
{"provider_id": provider_id, "model_id": model.id, "created": True}
|
||
)
|
||
else:
|
||
results["errors"].append(
|
||
{
|
||
"provider_id": provider_id,
|
||
"error": "create_models=False, no existing model found",
|
||
}
|
||
)
|
||
|
||
except Exception as e:
|
||
db.rollback()
|
||
results["errors"].append({"provider_id": provider_id, "error": str(e)})
|
||
|
||
db.commit()
|
||
return results
|