feat: 添加 AI 记账服务文件。

internal/service/ai_bookkeeping_service.go

@@ -21,6 +21,7 @@ import (
 	"accounting-app/internal/models"
 	"accounting-app/internal/repository"
 
+	"github.com/redis/go-redis/v9"
 	"gorm.io/gorm"
 )
 
@@ -31,27 +32,21 @@ type TranscriptionResult struct {
 	Duration float64 `json:"duration,omitempty"`
 }
 
-type cachedInsight struct {
-	Content   string
-	Timestamp time.Time
-}
-
 // ... (existing types)
 
 // AIBookkeepingService orchestrates AI bookkeeping functionality
 type AIBookkeepingService struct {
-	whisperService    *WhisperService
-	llmService        *LLMService
-	transactionRepo   *repository.TransactionRepository
-	accountRepo       *repository.AccountRepository
-	categoryRepo      *repository.CategoryRepository
-	userSettingsRepo  *repository.UserSettingsRepository
-	db                *gorm.DB
-	sessions          map[string]*AISession
-	sessionMutex      sync.RWMutex
-	config            *config.Config
-	insightCache      map[string]cachedInsight
-	insightCacheMutex sync.RWMutex
+	whisperService   *WhisperService
+	llmService       *LLMService
+	transactionRepo  *repository.TransactionRepository
+	accountRepo      *repository.AccountRepository
+	categoryRepo     *repository.CategoryRepository
+	userSettingsRepo *repository.UserSettingsRepository
+	db               *gorm.DB
+	sessions         map[string]*AISession
+	sessionMutex     sync.RWMutex
+	config           *config.Config
+	redisClient      *redis.Client
 }
 
 // NewAIBookkeepingService creates a new AIBookkeepingService
@@ -66,6 +61,16 @@ func NewAIBookkeepingService(
 	whisperService := NewWhisperService(cfg)
 	llmService := NewLLMService(cfg, accountRepo, categoryRepo)
 
+	// Initialize Redis client
+	var rdb *redis.Client
+	if cfg.RedisAddr != "" {
+		rdb = redis.NewClient(&redis.Options{
+			Addr:     cfg.RedisAddr,
+			Password: cfg.RedisPassword,
+			DB:       cfg.RedisDB,
+		})
+	}
+
 	svc := &AIBookkeepingService{
 		whisperService:   whisperService,
 		llmService:       llmService,
@@ -75,8 +80,8 @@ func NewAIBookkeepingService(
 		userSettingsRepo: userSettingsRepo,
 		db:               db,
 		sessions:         make(map[string]*AISession),
-		insightCache:     make(map[string]cachedInsight),
 		config:           cfg,
+		redisClient:      rdb,
 	}
 
 	// Start session cleanup goroutine
@@ -99,18 +104,33 @@ func (s *AIBookkeepingService) GenerateDailyInsight(ctx context.Context, userID
 	hasher := md5.New()
 	hasher.Write(dataBytes)
 	dataHash := hex.EncodeToString(hasher.Sum(nil))
-	cacheKey := fmt.Sprintf("%d:%s", userID, dataHash)
 
-	// Check cache
-	s.insightCacheMutex.RLock()
-	if cached, exists := s.insightCache[cacheKey]; exists {
-		// Cache valid for 24 hours (or just rely on hash change)
-		if time.Since(cached.Timestamp) < 24*time.Hour {
-			s.insightCacheMutex.RUnlock()
-			return cached.Content, nil
+	// Cache keys
+	cacheKey := fmt.Sprintf("ai:insight:daily:%d:%s", userID, dataHash)
+	lastInsightKey := fmt.Sprintf("ai:insight:last:%d", userID)
+
+	// Check Redis cache if available
+	if s.redisClient != nil {
+		val, err := s.redisClient.Get(ctx, cacheKey).Result()
+		if err == nil {
+			return val, nil
+		}
+	}
+
+	// Retrieve last insight for context awareness
+	var historyContext string
+	if s.redisClient != nil {
+		lastInsight, err := s.redisClient.Get(ctx, lastInsightKey).Result()
+		if err == nil && lastInsight != "" {
+			// Extract a summary or just use the whole thing if it's short enough.
+			// For simplicity and effectiveness, we inject it directly.
+			// Limit length to avoid context overflow if necessary, but 200-300 chars is fine.
+			if len(lastInsight) > 500 {
+				lastInsight = lastInsight[:500] + "..."
+			}
+			historyContext = fmt.Sprintf("\n【上下文参考】\n你上一次对该用户的评价是：\n%s\n\n请参考历史评价保持人设的一致性。如果用户有改观请表扬，如果依然如故请继续'毒舌'吐槽。", lastInsight)
 		}
 	}
-	s.insightCacheMutex.RUnlock()
 
 	// 2. Construct Prompt
 	prompt := fmt.Sprintf(`System: 你是 Novault 的首席财务AI「小金」。
@@ -119,7 +139,7 @@ func (s *AIBookkeepingService) GenerateDailyInsight(ctx context.Context, userID
 - 说话必须像个真实的当代年轻人，要**接地气**，拒绝客服味，拒绝播音腔。
 - 只有我们俩的时候，尽情释放你的"毒舌"属性，看到不合理的消费要**毫不留情地针砭时弊**，但最后给个"虽然但是"的台阶下。
 - 擅长用 emoji、语气词(哎哟/我去/啧啧/绝了)和网络梗来增加"人味"和"节目效果"。
-- **像个老朋友一样碎碎念**，不要端着，可以适度阴阳怪气。
+- **像个老朋友一样碎碎念**，不要端着，可以适度阴阳怪气。%s
 
 语气示例：
 - "啧啧，这周外卖点得飞起啊，咱家的锅是用来积灰的吗？还是说锅也需要放假？"
@@ -139,7 +159,7 @@ func (s *AIBookkeepingService) GenerateDailyInsight(ctx context.Context, userID
 - 看到 'last7DaysSpend' -> 请说 "最近7天花销" 或 "这周的战绩"
 - 看到 'top3Categories' -> 请说 "消费大头" 或 "钱都花哪儿了"
 
-1. "spending": 今日支出点评（100-150字）
+1. "spending": 今日支出点评（70-100字）
    *点评指南（尽量多写点，发挥你的戏精本色）：*
    - 看到 streakDays >= 3：疯狂打call，吹爆用户的坚持，用词要夸张，比如"史诗级成就"。
    - 看到 streakDays == 0：阴阳怪气地问是不是把记账这事儿忘了，或者是被外星人抓走了。
@@ -157,7 +177,7 @@ func (s *AIBookkeepingService) GenerateDailyInsight(ctx context.Context, userID
      * 是 0：直接颁发"诺贝尔省钱学奖"。
    - **关键原则：字数要够！内容要足！不要三言两语就打发了！要像个话痨朋友一样多说几句！**
 
-2. "budget": 预算建议（80-120字）
+2. "budget": 预算建议（50-70字）
    *建议指南（多点真诚的建议，也多点调侃）：*
    - 预算快超了：发出高能预警，比如"警告警告，余额正在报警，请立即停止剁手行为"。建议吃土、喝风。
    - 预算还多：鼓励适当奖励自己，比如"稍微吃顿好的也没事，人生苦短，及时行乐（在预算内）"。
@@ -172,7 +192,7 @@ func (s *AIBookkeepingService) GenerateDailyInsight(ctx context.Context, userID
    - 或者："听说'不买立省100%%'是致富捷径，建议全文背诵。"
 
 输出格式（纯 JSON）：
-{"spending": "...", "budget": "...", "emoji": "...", "tip": "..."}`, string(dataBytes))
+{"spending": "...", "budget": "...", "emoji": "...", "tip": "..."}`, historyContext, string(dataBytes))
 
 	// 3. Call LLM
 	report, err := s.llmService.GenerateReport(ctx, prompt)
@@ -192,13 +212,17 @@ func (s *AIBookkeepingService) GenerateDailyInsight(ctx context.Context, userID
 	}
 	report = strings.TrimSpace(report)
 
-	// Update cache
-	s.insightCacheMutex.Lock()
-	s.insightCache[cacheKey] = cachedInsight{
-		Content:   report,
-		Timestamp: time.Now(),
+	// Update cache if Redis is available
+	if s.redisClient != nil {
+		// Set short-term cache (5 min)
+		s.redisClient.Set(ctx, cacheKey, report, 5*time.Minute)
+
+		// Set long-term history context (7 days)
+		// We fire this asynchronously to avoid blocking
+		go func() {
+			s.redisClient.Set(context.Background(), lastInsightKey, report, 7*24*time.Hour)
+		}()
 	}
-	s.insightCacheMutex.Unlock()
 
 	return report, nil
 }