feat(image): 按 ModelFamily 调度 + 按账号 image 用量统计

调度层
- scheduler/family.go 新增 ModelFamily(platform, model) 把 gpt-image-* 折叠
  到 "gpt-image" 家族；FamilyCooldown 把限流冷却按"账号-家族"维度隔离，
  避免 gpt-image 撞限流影响同账号 chat 模型可用性
- scheduler/scheduler.go Scheduler 注入 familyCooldown

转发层
- plugin/{types,request}.go forwardState 新增 realtime 字段，stream/
  realtime 解耦：流式响应和实时透传互相独立
- plugin/forwarder_test.go 配套 realtime 测试

后台账号面
- app/account/{stats,types}.go BuildStatsResult 在已有总量基础上新增
  image_count / image_cost 维度，按 isImageModel 切分
- infra/store/account_store.go 仓储层支持新字段聚合
- server/{dto,handler/account*}.go HTTP 层暴露 image 字段

前端管理页
- pages/admin/{AccountsPage,AccountStatsModal}.tsx 新增 image 列与图表
- shared/{api/accounts,types}.ts 类型与请求扩展
- i18n/{en,zh}.json 文案
- app/{layout/AppShell,router}.tsx + pages/PluginPage.tsx 配套调整

Author: DouDOU-start

backend/internal/app/account/service_test.go

@@ -78,6 +78,10 @@ func (s stubRepository) BatchWindowStats(context.Context, []int, time.Time) (map
 	return nil, nil
 }
 
+func (s stubRepository) BatchImageStats(context.Context, []int, time.Time) (map[int]AccountImageStats, error) {
+	return nil, nil
+}
+
 func (s stubRepository) SaveCredentials(context.Context, int, map[string]string) error { return nil }
 
 // stubStateWriter 捕获 StateWriter 调用。

backend/internal/app/account/stats.go

@@ -2,11 +2,19 @@ package account
 
 import (
 	"sort"
+	"strings"
 	"time"
 
 	"github.com/DouDOU-start/airgate-core/internal/pkg/timezone"
 )
 
+// isImageModel 判断 usage_log.model 是否属于生图家族。
+// 与 scheduler.ModelFamily 的 "gpt-image-*" 规则保持一致 —— 不直接 import scheduler
+// 包是为了避免 stats 这一层依赖调度模块。两边都改时记得同步更新。
+func isImageModel(model string) bool {
+	return strings.HasPrefix(strings.ToLower(strings.TrimSpace(model)), "gpt-image")
+}
+
 const (
 	defaultPage     = 1
 	defaultPageSize = 20
@@ -85,6 +93,7 @@ func BuildStatsResult(account Account, logs []UsageLog, now, startDate, endDate
 	for _, log := range logs {
 		// 按用户时区对齐日期 key，避免 UTC 切换导致跨日错位
 		dateKey := log.CreatedAt.In(location).Format("2006-01-02")
+		isImage := isImageModel(log.Model)
 
 		result.Range.Count++
 		result.Range.InputTokens += log.InputTokens
@@ -93,6 +102,10 @@ func BuildStatsResult(account Account, logs []UsageLog, now, startDate, endDate
 		result.Range.AccountCost += log.AccountCost
 		result.Range.ActualCost += log.ActualCost
 		totalDurationMs += log.DurationMs
+		if isImage {
+			result.Range.ImageCount++
+			result.Range.ImageCost += log.AccountCost
+		}
 
 		if !log.CreatedAt.Before(today) {
 			result.Today.Count++
@@ -101,6 +114,10 @@ func BuildStatsResult(account Account, logs []UsageLog, now, startDate, endDate
 			result.Today.TotalCost += log.TotalCost
 			result.Today.AccountCost += log.AccountCost
 			result.Today.ActualCost += log.ActualCost
+			if isImage {
+				result.Today.ImageCount++
+				result.Today.ImageCost += log.AccountCost
+			}
 		}
 
 		if stats, ok := dailyMap[dateKey]; ok {

backend/internal/app/account/types.go

@@ -42,6 +42,9 @@ type Account struct {
 	Extra          map[string]any
 	CreatedAt      time.Time
 	UpdatedAt      time.Time
+	// ImageStats 仅 OpenAI 平台账号在列表查询路径上填充；其它平台 / 详情查询路径为 nil。
+	// 取自 usage_log model 名前缀 "gpt-image" 的子集聚合。
+	ImageStats *AccountImageStats
 }
 
 // AccountWindowStats 单个账号在某个时间窗口内的聚合统计。
@@ -53,6 +56,17 @@ type AccountWindowStats struct {
 	UserCost    float64 // SUM(actual_cost)，用户扣费总额（平台计费）
 }
 
+// AccountImageStats 单账号生图请求计数。
+//
+// 用于账号列表页"今日 N · 累计 M"展示。仅 OpenAI 平台账号填充（Claude / Anthropic
+// 等平台没有图像生成 endpoint，调用方按零值跳过）。
+//
+// "生图" 的判定与 stats.go::isImageModel 保持一致：model 名前缀 "gpt-image"。
+type AccountImageStats struct {
+	TodayCount int64 // 今日 00:00（服务器本地时区）至今的生图请求数
+	TotalCount int64 // 全部历史生图请求数
+}
+
 // UsageLog 使用记录聚合输入。
 type UsageLog struct {
 	Model        string
@@ -209,13 +223,18 @@ type StatsQuery struct {
 //   - TotalCost   = SUM(usage_log.total_cost)   原始上游定价
 //   - AccountCost = SUM(usage_log.account_cost) 账号实际成本 = total × account_rate（"账号计费"）
 //   - ActualCost  = SUM(usage_log.actual_cost)  用户扣费     = total × billing_rate
+//
+// ImageCount / ImageCost 把 model 名为 "gpt-image*" 的请求单独再聚一次，
+// 不影响 Count/cost（它们仍然是全部请求总和）。给后台展示"今日生图 N 张 / $X"用。
 type PeriodStats struct {
 	Count        int     `json:"count"`
 	InputTokens  int64   `json:"input_tokens"`
 	OutputTokens int64   `json:"output_tokens"`
 	TotalCost    float64 `json:"total_cost"`
 	AccountCost  float64 `json:"account_cost"`
 	ActualCost   float64 `json:"actual_cost"`
+	ImageCount   int     `json:"image_count"`
+	ImageCost    float64 `json:"image_cost"`
 }
 
 // DailyStats 每日统计。
@@ -314,5 +333,8 @@ type Repository interface {
 	FindUsageLogs(context.Context, int, time.Time, time.Time) ([]UsageLog, error)
 	// BatchWindowStats 批量聚合统计，没有记录的账号不出现在返回 map 中。
 	BatchWindowStats(ctx context.Context, accountIDs []int, startTime time.Time) (map[int]AccountWindowStats, error)
+	// BatchImageStats 批量统计指定账号的生图请求数（model 前缀 "gpt-image"）。
+	// 同时返回 [todayStart, now] 区间和 全部历史 两个计数。无记录的账号不出现在返回 map 中。
+	BatchImageStats(ctx context.Context, accountIDs []int, todayStart time.Time) (map[int]AccountImageStats, error)
 	SaveCredentials(context.Context, int, map[string]string) error
 }

backend/internal/infra/store/account_store.go

@@ -320,6 +320,76 @@ func (s *AccountStore) BatchWindowStats(ctx context.Context, accountIDs []int, s
 	return result, nil
 }
 
+// imageModelPrefix 与 app/account/stats.go::isImageModel 保持一致。
+// 改这里时记得同步那边，避免"统计口径"和"列表展示"对不上。
+const imageModelPrefix = "gpt-image"
+
+// BatchImageStats 一次拿"今日生图数"和"累计生图数"两组聚合，
+// 仅算 model 前缀 "gpt-image" 的请求。两条 GROUP BY 查询：
+//
+//	SELECT account_id, COUNT(*) FROM usage_log
+//	  WHERE account_id IN (...) AND model LIKE 'gpt-image%' [AND created_at >= ?]
+//	  GROUP BY account_id
+//
+// 调用方传 openai 平台的账号 ID 子集即可——chat-only 平台传进来不会出错（match 0 行），
+// 只是浪费一次查询。
+func (s *AccountStore) BatchImageStats(ctx context.Context, accountIDs []int, todayStart time.Time) (map[int]appaccount.AccountImageStats, error) {
+	result := map[int]appaccount.AccountImageStats{}
+	if len(accountIDs) == 0 {
+		return result, nil
+	}
+
+	type row struct {
+		AccountID int `json:"account_usage_logs"`
+		Count     int `json:"count"`
+	}
+
+	// 累计：不带 created_at 限制
+	var totalRows []row
+	if err := s.db.UsageLog.Query().
+		Where(
+			entusagelog.HasAccountWith(entaccount.IDIn(accountIDs...)),
+			entusagelog.ModelHasPrefix(imageModelPrefix),
+		).
+		GroupBy(entusagelog.AccountColumn).
+		Aggregate(ent.Count()).
+		Scan(ctx, &totalRows); err != nil {
+		return nil, err
+	}
+	for _, r := range totalRows {
+		if r.AccountID == 0 {
+			continue
+		}
+		entry := result[r.AccountID]
+		entry.TotalCount = int64(r.Count)
+		result[r.AccountID] = entry
+	}
+
+	// 今日：created_at >= todayStart
+	var todayRows []row
+	if err := s.db.UsageLog.Query().
+		Where(
+			entusagelog.HasAccountWith(entaccount.IDIn(accountIDs...)),
+			entusagelog.ModelHasPrefix(imageModelPrefix),
+			entusagelog.CreatedAtGTE(todayStart),
+		).
+		GroupBy(entusagelog.AccountColumn).
+		Aggregate(ent.Count()).
+		Scan(ctx, &todayRows); err != nil {
+		return nil, err
+	}
+	for _, r := range todayRows {
+		if r.AccountID == 0 {
+			continue
+		}
+		entry := result[r.AccountID]
+		entry.TodayCount = int64(r.Count)
+		result[r.AccountID] = entry
+	}
+
+	return result, nil
+}
+
 // SaveCredentials 保存账号凭证。
 func (s *AccountStore) SaveCredentials(ctx context.Context, id int, credentials map[string]string) error {
 	if err := s.db.Account.UpdateOneID(id).

backend/internal/plugin/forwarder_test.go

@@ -1,8 +1,13 @@
 package plugin
 
 import (
+	"net/http"
+	"net/http/httptest"
 	"testing"
 
+	"github.com/gin-gonic/gin"
+
+	"github.com/DouDOU-start/airgate-core/ent"
 	"github.com/DouDOU-start/airgate-core/internal/auth"
 	"github.com/DouDOU-start/airgate-core/internal/routing"
 )
@@ -41,6 +46,72 @@ func TestParseBody_StreamTrue(t *testing.T) {
 	}
 }
 
+func TestBuildPluginRequestUsesWriterForStreamRequest(t *testing.T) {
+	t.Parallel()
+
+	recorder := httptest.NewRecorder()
+	c, _ := gin.CreateTestContext(recorder)
+	c.Request = httptest.NewRequest(http.MethodPost, "/v1/images/generations", nil)
+	state := &forwardState{
+		requestPath: "/v1/images/generations",
+		stream:      true,
+		realtime:    true,
+		keyInfo:     &auth.APIKeyInfo{},
+		account:     &ent.Account{},
+	}
+
+	req := buildPluginRequest(c, state)
+	if !req.Stream {
+		t.Fatalf("Stream = false, want true")
+	}
+	if req.Writer == nil {
+		t.Fatalf("Writer = nil, want stream writer")
+	}
+}
+
+func TestBuildPluginRequestOmitsWriterForPlainNonStreamRequest(t *testing.T) {
+	t.Parallel()
+
+	recorder := httptest.NewRecorder()
+	c, _ := gin.CreateTestContext(recorder)
+	c.Request = httptest.NewRequest(http.MethodPost, "/v1/chat/completions", nil)
+	state := &forwardState{
+		requestPath: "/v1/chat/completions",
+		stream:      false,
+		realtime:    false,
+		keyInfo:     &auth.APIKeyInfo{},
+		account:     &ent.Account{},
+	}
+
+	req := buildPluginRequest(c, state)
+	if req.Writer != nil {
+		t.Fatalf("Writer = %T, want nil", req.Writer)
+	}
+}
+
+func TestBuildPluginRequestOmitsWriterForNonStreamImagesRequest(t *testing.T) {
+	t.Parallel()
+
+	recorder := httptest.NewRecorder()
+	c, _ := gin.CreateTestContext(recorder)
+	c.Request = httptest.NewRequest(http.MethodPost, "/v1/images/generations", nil)
+	state := &forwardState{
+		requestPath: "/v1/images/generations",
+		stream:      false,
+		realtime:    false,
+		keyInfo:     &auth.APIKeyInfo{},
+		account:     &ent.Account{},
+	}
+
+	req := buildPluginRequest(c, state)
+	if req.Stream {
+		t.Fatalf("Stream = true, want false")
+	}
+	if req.Writer != nil {
+		t.Fatalf("Writer = %T, want nil", req.Writer)
+	}
+}
+
 func TestRoutesForAPIKeyUsesBoundGroupOnly(t *testing.T) {
 	t.Parallel()
 

backend/internal/plugin/request.go

@@ -56,6 +56,7 @@ func (f *Forwarder) parseRequest(c *gin.Context) (*forwardState, bool) {
 		body:              body,
 		model:             parsed.Model,
 		stream:            parsed.Stream,
+		realtime:          parsed.Stream,
 		sessionID:         parsed.SessionID,
 		requestedPlatform: requestedPlatform,
 		keyInfo:           keyInfo,
@@ -216,7 +217,7 @@ func buildPluginRequest(c *gin.Context, state *forwardState) *sdk.ForwardRequest
 		Model:   state.model,
 		Stream:  state.stream,
 	}
-	if state.stream {
+	if state.realtime {
 		req.Writer = c.Writer
 	}
 	return req

backend/internal/plugin/types.go

@@ -19,6 +19,7 @@ type forwardState struct {
 	body      []byte
 	model     string
 	stream    bool
+	realtime  bool
 	sessionID string
 
 	requestedPlatform string