mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-07 16:08:02 +08:00
Randark
901a0439f1
后端： - 协议凭证维度 credentials[api_protocol] ∈ chat_completions(默认)/anthropic/responses(deepseek) - /v1/messages 零转换直通原生 Anthropic 端点（kimi/zhipu/deepseek），CC/Responses 入站交叉组合走 apicompat 双向转换链（responses/chat_completions anthropic-native 转发器） - count_tokens：anthropic 协议透传原生端点；其余 CN 协议本地 tiktoken 估算 - Coding Plan 额度探测（5h/weekly 滚动窗口）+ payg 余额探测（kimi/deepseek）， deepseek 双币种 CNY+USD 明细，任一币种达标不停调 - 周期任务 [CNBalance] 并发探测 + 预算随工作量放大；响应式 429 冷却到最早窗口 重置点；余额不足可恢复临时停调；智谱 CREDIT_LIMIT 不污染窗口解析 - CC→anthropic 流式客户端断开后继续排水上游保住 usage 计量 前端： - 创建/编辑弹窗 account_mode + api_protocol + base_url 联动预设（含 watcher 竞态防护） - 用量单元格：kimi/zhipu coding 显示 5h/weekly 窗口，kimi/deepseek payg 显示余额， 多币种并列展示；探测失败保留快照；挂载自动探测 5min 去抖 - 调度阈值设置面板补 kimi/zhipu 平台（对齐后端 AllowedSchedulingThresholdPlatforms）
70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
package admin
|
||
|
||
import (
|
||
"strconv"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/pkg/response"
|
||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// CNProviderHandler 暴露国产供应商（kimi/zhipu/deepseek）的额度与余额查询端点。
|
||
//
|
||
// - GET /admin/cn-providers/accounts/:id/quota Coding Plan 滚动窗口用量（kimi/zhipu）
|
||
// - GET /admin/cn-providers/accounts/:id/balance payg 账号余额（kimi/deepseek）
|
||
//
|
||
// 智谱（zhipu）无余额端点，故同一账号仅 quota 或 balance 其一可用：服务端按账号
|
||
// platform + account_mode 校验并返回明确错误（见 CNProvider*Service 的 load*Account）。
|
||
type CNProviderHandler struct {
|
||
quotaService *service.CNProviderQuotaService
|
||
balanceService *service.CNProviderBalanceService
|
||
}
|
||
|
||
func NewCNProviderHandler(
|
||
quotaService *service.CNProviderQuotaService,
|
||
balanceService *service.CNProviderBalanceService,
|
||
) *CNProviderHandler {
|
||
return &CNProviderHandler{
|
||
quotaService: quotaService,
|
||
balanceService: balanceService,
|
||
}
|
||
}
|
||
|
||
// QueryQuota 查询 Coding Plan 滚动窗口用量（5h + weekly）。
|
||
func (h *CNProviderHandler) QueryQuota(c *gin.Context) {
|
||
accountID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
response.BadRequest(c, "Invalid account ID")
|
||
return
|
||
}
|
||
if h == nil || h.quotaService == nil {
|
||
response.BadRequest(c, "cn provider quota service is not enabled")
|
||
return
|
||
}
|
||
result, err := h.quotaService.QueryUsage(c.Request.Context(), accountID)
|
||
if err != nil {
|
||
response.ErrorFrom(c, err)
|
||
return
|
||
}
|
||
response.Success(c, result)
|
||
}
|
||
|
||
// QueryBalance 查询 payg 账号余额。
|
||
func (h *CNProviderHandler) QueryBalance(c *gin.Context) {
|
||
accountID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
response.BadRequest(c, "Invalid account ID")
|
||
return
|
||
}
|
||
if h == nil || h.balanceService == nil {
|
||
response.BadRequest(c, "cn provider balance service is not enabled")
|
||
return
|
||
}
|
||
result, err := h.balanceService.QueryBalance(c.Request.Context(), accountID)
|
||
if err != nil {
|
||
response.ErrorFrom(c, err)
|
||
return
|
||
}
|
||
response.Success(c, result)
|
||
}
|