From 4b322494ba0c32b22bb8b5471a5ddd24d002b419 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=AD=99=E8=AF=9A?= Date: Tue, 30 Dec 2025 18:49:46 +0800 Subject: [PATCH] fix --- Service/EmailHandleService.cs | 3 + Service/GlobalUsings.cs | 4 +- Service/SmartClassify.cs | 246 ++++++++++++++++++ Web/src/components/PopupContainer.vue | 17 +- Web/src/components/ReasonGroupList.vue | 2 +- Web/src/components/SmartClassifyButton.vue | 83 ++++-- Web/src/views/CalendarView.vue | 4 +- Web/src/views/StatisticsView.vue | 24 +- .../TransactionRecordController.cs | 213 +-------------- 9 files changed, 365 insertions(+), 231 deletions(-) create mode 100644 Service/SmartClassify.cs diff --git a/Service/EmailHandleService.cs b/Service/EmailHandleService.cs index 4351e50..145efe8 100644 --- a/Service/EmailHandleService.cs +++ b/Service/EmailHandleService.cs @@ -75,6 +75,9 @@ public class EmailHandleService( ); logger.LogInformation("成功解析邮件，共 {Count} 条交易记录", parsed.Length); + // TODO 接入AI分类 + // 目前已经 + bool allSuccess = true; foreach (var (card, reason, amount, balance, type, occurredAt) in parsed) { diff --git a/Service/GlobalUsings.cs b/Service/GlobalUsings.cs index 76859c8..e78ee92 100644 --- a/Service/GlobalUsings.cs +++ b/Service/GlobalUsings.cs @@ -9,5 +9,5 @@ global using System.Text.Json; global using Entity; global using FreeSql; global using System.Linq; -global using System.Security.Cryptography; -global using Service.AppSettingModel; \ No newline at end of file +global using Service.AppSettingModel; +global using System.Text.Json.Serialization; \ No newline at end of file diff --git a/Service/SmartClassify.cs b/Service/SmartClassify.cs new file mode 100644 index 0000000..c7da11c --- /dev/null +++ b/Service/SmartClassify.cs @@ -0,0 +1,246 @@ +namespace Service; + +public interface ISmartHandleService +{ + Task SmartClassifyAsync(long[] transactionIds, Action<(string , string)> chunkAction); +} + +public class SmartHandleService( + ITransactionRecordRepository transactionRepository, + ITextSegmentService textSegmentService, + ILogger logger, + ITransactionCategoryRepository categoryRepository, + IOpenAiService openAiService +) : ISmartHandleService +{ + public async Task SmartClassifyAsync(long[] transactionIds, Action<(string , string)> chunkAction) + { + try + { + // 获取指定ID的账单（作为样本） + var sampleRecords = await transactionRepository.GetByIdsAsync(transactionIds); + + if (sampleRecords.Length == 0) + { + // await WriteEventAsync("error", "找不到指定的账单"); + chunkAction(("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>(); + 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 + chunkAction(("start", $"开始分类，共 {sampleRecords.Length} 条账单")); + + // 用于存储AI返回的分组分类结果 + var classifyResults = new List<(string Reason, string Classify, TransactionType Type)>(); + var buffer = new StringBuilder(); + var sendedIds = new HashSet(); + 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(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 }); + chunkAction(("data", resultJson)); + } + } + } + } + } + catch (Exception ex) + { + logger.LogWarning(ex, "解析AI分类结果失败: {JsonStr}", jsonStr); + } + + startIdx = closeBrace + 1; + } + } + + chunkAction(("end", "分类完成")); + } + catch (Exception ex) + { + logger.LogError(ex, "智能分类失败"); + chunkAction(("error", $"智能分类失败: {ex.Message}")); + } + } + + /// + /// 查找匹配的右括号 + /// + private static int FindMatchingBrace(string str, int startPos) + { + int braceCount = 0; + for (int i = startPos; i < str.Length; i++) + { + if (str[i] == '{') braceCount++; + else if (str[i] == '}') + { + braceCount--; + if (braceCount == 0) return i; + } + } + return -1; + } + + private static string GetTypeName(TransactionType type) + { + return type switch + { + TransactionType.Expense => "支出", + TransactionType.Income => "收入", + TransactionType.None => "不计入收支", + _ => "未知" + }; + } +} + +/// +/// 分组分类结果DTO（用于AI返回结果解析） +/// +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; } +} + diff --git a/Web/src/components/PopupContainer.vue b/Web/src/components/PopupContainer.vue index f4f72a3..bb2cd3f 100644 --- a/Web/src/components/PopupContainer.vue +++ b/Web/src/components/PopupContainer.vue @@ -90,12 +90,17 @@ const hasActions = computed(() => !!slots['header-actions']) margin: 0; text-align: center; color: var(--van-text-color, #323233); + /*超出长度*/ + overflow: hidden; + text-overflow: ellipsis; + white-space: nowrap; + } .header-stats { - display: flex; + display: grid; + grid-template-columns: 1fr auto 1fr; align-items: center; - justify-content: space-between; gap: 12px; margin-top: 12px; } @@ -104,10 +109,16 @@ const hasActions = computed(() => !!slots['header-actions']) margin: 0; font-size: 14px; color: var(--van-text-color-2, #646566); - flex: 1; + grid-column: 2; text-align: center; } +/* 按钮区域放在右侧 */ +.header-stats :deep(> :last-child:not(.stats-text)) { + grid-column: 3; + justify-self: end; +} + .popup-scroll-content { flex: 1; overflow-y: auto; diff --git a/Web/src/components/ReasonGroupList.vue b/Web/src/components/ReasonGroupList.vue index 1657ac6..44a5e4c 100644 --- a/Web/src/components/ReasonGroupList.vue +++ b/Web/src/components/ReasonGroupList.vue @@ -211,7 +211,7 @@ const props = defineProps({ // 每页数量 pageSize: { type: Number, - default: 20 + default: 5 } }) diff --git a/Web/src/components/SmartClassifyButton.vue b/Web/src/components/SmartClassifyButton.vue index f1ea0a7..44c69b2 100644 --- a/Web/src/components/SmartClassifyButton.vue +++ b/Web/src/components/SmartClassifyButton.vue @@ -1,7 +1,7 @@