61ce809634
- 支持 MiniMax/OpenAI/Google Gemini/智谱/Kimi 五个平台 - 插件化 Provider 架构，自动发现注册 - 多维度 QuotaRule 额度追踪（固定间隔/自然周期/API同步/手动） - OpenAI + Anthropic 兼容 API 代理，SSE 流式转发 - Model 路由表 + 额度耗尽自动 fallback - 多媒体任务队列（图片/语音/视频） - Vue3 + Tailwind 单文件 Web 仪表盘 - Docker 一键部署 Made-with: Cursor
134 lines
4.3 KiB
Python
134 lines
4.3 KiB
Python
"""Plan + QuotaRule + ModelRoute 管理 API"""
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
|
|
from app import database as db
|
|
from app.models import (
|
|
PlanCreate, PlanUpdate, PlanOut, PlanWithRules,
|
|
QuotaRuleCreate, QuotaRuleUpdate, QuotaRuleOut,
|
|
ModelRouteBase, ModelRouteOut,
|
|
)
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
# ── Plan CRUD ─────────────────────────────────────────
|
|
|
|
@router.get("", response_model=list[PlanWithRules])
|
|
async def list_plans(enabled_only: bool = False):
|
|
plans = await db.list_plans(enabled_only=enabled_only)
|
|
result = []
|
|
for p in plans:
|
|
rules = await db.list_quota_rules(p["id"])
|
|
result.append({**p, "quota_rules": rules})
|
|
return result
|
|
|
|
|
|
@router.post("", response_model=dict)
|
|
async def create_plan(body: PlanCreate):
|
|
return await db.create_plan(
|
|
name=body.name,
|
|
provider_name=body.provider_name,
|
|
api_key=body.api_key,
|
|
api_base=body.api_base,
|
|
plan_type=body.plan_type,
|
|
supported_models=body.supported_models,
|
|
extra_headers=body.extra_headers,
|
|
extra_config=body.extra_config,
|
|
enabled=body.enabled,
|
|
)
|
|
|
|
|
|
@router.get("/{plan_id}", response_model=PlanWithRules)
|
|
async def get_plan(plan_id: str):
|
|
p = await db.get_plan(plan_id)
|
|
if not p:
|
|
raise HTTPException(404, "Plan not found")
|
|
rules = await db.list_quota_rules(plan_id)
|
|
return {**p, "quota_rules": rules}
|
|
|
|
|
|
@router.patch("/{plan_id}")
|
|
async def update_plan(plan_id: str, body: PlanUpdate):
|
|
fields = body.model_dump(exclude_unset=True)
|
|
if not fields:
|
|
raise HTTPException(400, "No fields to update")
|
|
ok = await db.update_plan(plan_id, **fields)
|
|
if not ok:
|
|
raise HTTPException(404, "Plan not found or no change")
|
|
return {"ok": True}
|
|
|
|
|
|
@router.delete("/{plan_id}")
|
|
async def delete_plan(plan_id: str):
|
|
ok = await db.delete_plan(plan_id)
|
|
if not ok:
|
|
raise HTTPException(404, "Plan not found")
|
|
return {"ok": True}
|
|
|
|
|
|
# ── QuotaRule CRUD ────────────────────────────────────
|
|
|
|
@router.get("/{plan_id}/rules", response_model=list[QuotaRuleOut])
|
|
async def list_rules(plan_id: str):
|
|
return await db.list_quota_rules(plan_id)
|
|
|
|
|
|
@router.post("/{plan_id}/rules", response_model=dict)
|
|
async def create_rule(plan_id: str, body: QuotaRuleCreate):
|
|
p = await db.get_plan(plan_id)
|
|
if not p:
|
|
raise HTTPException(404, "Plan not found")
|
|
return await db.create_quota_rule(
|
|
plan_id=plan_id,
|
|
rule_name=body.rule_name,
|
|
quota_total=body.quota_total,
|
|
quota_unit=body.quota_unit,
|
|
refresh_type=body.refresh_type.value,
|
|
interval_hours=body.interval_hours,
|
|
calendar_unit=body.calendar_unit,
|
|
calendar_anchor=body.calendar_anchor,
|
|
enabled=body.enabled,
|
|
)
|
|
|
|
|
|
@router.patch("/rules/{rule_id}")
|
|
async def update_rule(rule_id: str, body: QuotaRuleUpdate):
|
|
fields = body.model_dump(exclude_unset=True)
|
|
if "refresh_type" in fields and fields["refresh_type"] is not None:
|
|
fields["refresh_type"] = fields["refresh_type"].value
|
|
ok = await db.update_quota_rule(rule_id, **fields)
|
|
if not ok:
|
|
raise HTTPException(404, "Rule not found or no change")
|
|
return {"ok": True}
|
|
|
|
|
|
@router.delete("/rules/{rule_id}")
|
|
async def delete_rule(rule_id: str):
|
|
d = await db.get_db()
|
|
cur = await d.execute("DELETE FROM quota_rules WHERE id=?", (rule_id,))
|
|
await d.commit()
|
|
if cur.rowcount == 0:
|
|
raise HTTPException(404, "Rule not found")
|
|
return {"ok": True}
|
|
|
|
|
|
# ── Model Routes ──────────────────────────────────────
|
|
|
|
@router.get("/routes/models", response_model=list[ModelRouteOut])
|
|
async def get_model_routes():
|
|
return await db.list_model_routes()
|
|
|
|
|
|
@router.post("/routes/models", response_model=dict)
|
|
async def set_model_route(body: ModelRouteBase):
|
|
return await db.set_model_route(body.model_name, body.plan_id, body.priority)
|
|
|
|
|
|
@router.delete("/routes/models/{route_id}")
|
|
async def delete_model_route(route_id: str):
|
|
ok = await db.delete_model_route(route_id)
|
|
if not ok:
|
|
raise HTTPException(404, "Route not found")
|
|
return {"ok": True}
|