fix

WebApi/Controllers/TransactionRecordController.cs

@@ -8,7 +8,7 @@ public class TransactionRecordController(
     ITransactionRecordRepository transactionRepository,
     ITransactionCategoryRepository categoryRepository,
     IOpenAiService openAiService,
-    ITextSegmentService textSegmentService,
+    ISmartHandleService smartHandleService,
     ILogger<TransactionRecordController> logger
 ) : ControllerBase
 {
@@ -572,194 +572,20 @@ public class TransactionRecordController(
         Response.Headers.Append("Cache-Control", "no-cache");
         Response.Headers.Append("Connection", "keep-alive");
 
-        try
+        // 验证账单ID列表
+        if (request.TransactionIds == null || request.TransactionIds.Count == 0)
         {
-            // 验证账单ID列表
-            if (request.TransactionIds == null || request.TransactionIds.Count == 0)
-            {
-                await WriteEventAsync("error", "请提供要分类的账单ID");
-                return;
-            }
-
-            // 获取指定ID的账单（作为样本）
-            var sampleRecords = await transactionRepository.GetByIdsAsync(request.TransactionIds.ToArray());
-
-            if (sampleRecords.Length == 0)
-            {
-                await WriteEventAsync("error", "找不到指定的账单");
-                return;
-            }
-
-            // 重新按Reason分组所有待分类账单
-            var groupedRecords = sampleRecords
-                .GroupBy(r => r.Reason)
-                .Select(g => new
-                {
-                    Reason = g.Key,
-                    Ids = g.Select(r => r.Id).ToList(),
-                    Count = g.Count(),
-                    TotalAmount = g.Sum(r => r.Amount),
-                    SampleType = g.First().Type
-                })
-                .OrderByDescending(g => Math.Abs(g.TotalAmount))
-                .ToList();
-
-            // 【增强功能】对每个分组的摘要进行分词，查询已分类的相似账单
-            var referenceRecords = new Dictionary<string, List<TransactionRecord>>();
-            foreach (var group in groupedRecords)
-            {
-                // 使用专业分词库提取关键词
-                var keywords = textSegmentService.ExtractKeywords(group.Reason);
-                
-                if (keywords.Count > 0)
-                {
-                    // 查询包含这些关键词且已分类的账单（带相关度评分）
-                    // minMatchRate=0.4 表示至少匹配40%的关键词才被认为是相似的
-                    var similarClassifiedWithScore = await transactionRepository.GetClassifiedByKeywordsWithScoreAsync(keywords, minMatchRate: 0.4, limit: 10);
-                    
-                    if (similarClassifiedWithScore.Count > 0)
-                    {
-                        // 只取前5个最相关的
-                        var topSimilar = similarClassifiedWithScore.Take(5).Select(x => x.record).ToList();
-                        referenceRecords[group.Reason] = topSimilar;
-                        
-                        // 记录调试信息
-                        logger.LogDebug("摘要 '{Reason}' 提取关键词: {Keywords}, 找到 {Count} 个相似账单，相关度分数: {Scores}",
-                            group.Reason,
-                            string.Join(", ", keywords),
-                            similarClassifiedWithScore.Count,
-                            string.Join(", ", similarClassifiedWithScore.Select(x => $"{x.record.Reason}({x.relevanceScore:F2})")));
-                    }
-                    else
-                    {
-                        logger.LogDebug("摘要 '{Reason}' 提取关键词: {Keywords}, 未找到高相关度的相似账单",
-                            group.Reason,
-                            string.Join(", ", keywords));
-                    }
-                }
-            }
-
-            // 获取所有分类
-            var categories = await categoryRepository.GetAllAsync();
-
-            // 构建分类信息
-            var categoryInfo = new StringBuilder();
-            foreach (var type in new[] { 0, 1, 2 })
-            {
-                var typeName = GetTypeName((TransactionType)type);
-                categoryInfo.AppendLine($"{typeName}: ");
-                var categoriesOfType = categories.Where(c => (int)c.Type == type).ToList();
-                foreach (var category in categoriesOfType)
-                {
-                    categoryInfo.AppendLine($"- {category.Name}");
-                }
-            }
-
-            // 构建账单分组信息
-            var billsInfo = new StringBuilder();
-            foreach (var (group, index) in groupedRecords.Select((g, i) => (g, i)))
-            {
-                billsInfo.AppendLine($"{index + 1}. 摘要={group.Reason}, 当前类型={GetTypeName(group.SampleType)}, 当前分类={(string.IsNullOrEmpty(group.SampleType.ToString()) ? "未分类" : group.SampleType.ToString())}, 涉及金额={group.TotalAmount}");
-                
-                // 如果有相似的已分类账单，添加参考信息
-                if (referenceRecords.TryGetValue(group.Reason, out var references))
-                {
-                    billsInfo.AppendLine("   【参考】相似且已分类的账单：");
-                    foreach (var refer in references.Take(3)) // 最多显示3个参考
-                    {
-                        billsInfo.AppendLine($"      - 摘要={refer.Reason}, 分类={refer.Classify}, 类型={GetTypeName(refer.Type)}, 金额={refer.Amount}");
-                    }
-                }
-            }
-
-            var systemPrompt = $$"""
-                你是一个专业的账单分类助手。请根据提供的账单分组信息和分类列表，为每个分组选择最合适的分类。
-
-                可用的分类列表：
-                {{categoryInfo}}
-
-                分类规则：
-                1. 根据账单的摘要和涉及金额，选择最匹配的分类
-                2. 如果提供了【参考】信息，优先参考相似账单的分类，这些是历史上已分类的相似账单
-                3. 如果无法确定分类，可以选择"其他"
-                4. 每个分组可能包含多条账单，你需要为整个分组选择一个分类
-
-                请对每个分组进行分类，每次输出一个分组的分类结果，格式如下：
-                {"reason": "交易摘要", "type": 0:支出/1:收入/2:不计入收支(Type为Number枚举值) ,"classify": "分类名称"}
-
-                只输出JSON，不要有其他文字说明。
-                """;
-
-            var userPrompt = $$"""
-                请为以下账单分组进行分类：
-
-                {{billsInfo}}
-
-                请逐个输出分类结果。
-                """;
-
-            // 流式调用AI
-            await WriteEventAsync("start", $"开始分类，共 {sampleRecords.Length} 条账单");
-
-            // 用于存储AI返回的分组分类结果
-            var classifyResults = new List<(string Reason, string Classify, TransactionType Type)>();
-            var buffer = new StringBuilder();
-            var sendedIds = new HashSet<long>();
-            await foreach (var chunk in openAiService.ChatStreamAsync(systemPrompt, userPrompt))
-            {
-                buffer.Append(chunk);
-
-                // 尝试解析完整的JSON对象
-                var bufferStr = buffer.ToString();
-                var startIdx = 0;
-                while (startIdx < bufferStr.Length)
-                {
-                    var openBrace = bufferStr.IndexOf('{', startIdx);
-                    if (openBrace == -1) break;
-
-                    var closeBrace = FindMatchingBrace(bufferStr, openBrace);
-                    if (closeBrace == -1) break;
-
-                    var jsonStr = bufferStr.Substring(openBrace, closeBrace - openBrace + 1);
-                    try
-                    {
-                        var result = JsonSerializer.Deserialize<GroupClassifyResult>(jsonStr);
-                        if (result != null && !string.IsNullOrEmpty(result.Reason))
-                        {
-                            classifyResults.Add((result.Reason, result.Classify ?? "", result.Type));
-                            // 每一条结果单独通知
-                            var group = groupedRecords.FirstOrDefault(g => g.Reason == result.Reason);
-                            if (group != null)
-                            {
-                                // 为该分组的所有账单ID返回分类结果
-                                foreach (var id in group.Ids)
-                                {
-                                    if (!sendedIds.Contains(id))
-                                    {
-                                        sendedIds.Add(id);
-                                        var resultJson = JsonSerializer.Serialize(new { id, result.Classify, result.Type });
-                                        await WriteEventAsync("data", resultJson);
-                                    }
-                                }
-                            }
-                        }
-                    }
-                    catch (Exception ex)
-                    {
-                        logger.LogWarning(ex, "解析AI分类结果失败: {JsonStr}", jsonStr);
-                    }
-
-                    startIdx = closeBrace + 1;
-                }
-            }
-
-            await WriteEventAsync("end", "分类完成");
+            await WriteEventAsync("error", "请提供要分类的账单ID");
+            return;
         }
-        catch (Exception ex)
+
+        await smartHandleService.SmartClassifyAsync(request.TransactionIds.ToArray(), async (chunk) =>
         {
-            logger.LogError(ex, "智能分类失败");
-            await WriteEventAsync("error", $"智能分类失败: {ex.Message}");
-        }
+            var (eventType, content) = chunk;
+            await WriteEventAsync(eventType, content);
+        });
+
+        await Response.Body.FlushAsync();
     }
 
     /// <summary>
@@ -1074,21 +900,6 @@ public record SmartClassifyRequest(
     List<long>? TransactionIds = null
 );
 
-/// <summary>
-/// 分组分类结果DTO（用于AI返回结果解析）
-/// </summary>
-public record GroupClassifyResult
-{
-    [JsonPropertyName("reason")]
-    public string Reason { get; set; } = string.Empty;
-
-    [JsonPropertyName("classify")]
-    public string? Classify { get; set; }
-
-    [JsonPropertyName("type")]
-    public TransactionType Type { get; set; }
-}
-
 /// <summary>
 /// 批量更新分类项DTO
 /// </summary>