diff --git a/Service/SmartHandleService.cs b/Service/SmartHandleService.cs index 0fd50d3..6de2a03 100644 --- a/Service/SmartHandleService.cs +++ b/Service/SmartHandleService.cs @@ -123,10 +123,13 @@ public class SmartHandleService( 3. 如果无法确定分类,可以选择"其他" 4. 每个分组可能包含多条账单,你需要为整个分组选择一个分类 - 请对每个分组进行分类,每次输出一个分组的分类结果,格式如下: - {"reason": "交易摘要", "type": 0:支出/1:收入/2:不计入收支(Type为Number枚举值) ,"classify": "分类名称"} + 输出格式要求(强制): + - 请使用 NDJSON(每行一个独立的 JSON 对象,末尾以换行符分隔),不要输出数组。 + - 每行的JSON格式严格为:{"reason": "交易摘要", "type": 0, "classify": "分类名称"} + - 不要输出任何解释性文字、编号、标点或多余的文本 + - 如果无法判断分类,请将 "classify" 设为 "其他",并确保仍然输出 JSON 行 - 只输出JSON,不要有其他文字说明。 + 只输出按行的JSON对象(NDJSON),不要有其他文字说明。 """; var userPrompt = $$""" @@ -140,59 +143,101 @@ public class SmartHandleService( // 流式调用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)) + + // 将流解析逻辑提取为本地函数以减少嵌套 + void HandleResult(GroupClassifyResult? result) { - buffer.Append(chunk); - - // 尝试解析完整的JSON对象 - var bufferStr = buffer.ToString(); - var startIdx = 0; - while (startIdx < bufferStr.Length) + if (result is null || string.IsNullOrEmpty(result.Reason)) return; + classifyResults.Add((result.Reason, result.Classify ?? string.Empty, result.Type)); + var group = groupedRecords.FirstOrDefault(g => g.Reason == result.Reason); + if (group == null) return; + foreach (var id in group.Ids) { - 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 + if (sendedIds.Add(id)) { - 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)); - } - } - } - } + 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", "分类完成")); + // 解析缓冲区中的所有完整 JSON 对象或数组 + void FlushBuffer(StringBuilder buffer) + { + var buf = buffer.ToString(); + if (string.IsNullOrWhiteSpace(buf)) return; + + // 优先尝试解析完整数组 + var trimmed = buf.TrimStart(); + if (trimmed.Length > 0 && trimmed[0] == '[') + { + var lastArrEnd = buf.LastIndexOf(']'); + if (lastArrEnd > -1) + { + var arrJson = buf.Substring(0, lastArrEnd + 1); + try + { + var results = JsonSerializer.Deserialize(arrJson); + if (results != null) + { + foreach (var r in results) HandleResult(r); + } + buffer.Remove(0, lastArrEnd + 1); + buf = buffer.ToString(); + } + catch (Exception exArr) + { + logger.LogDebug(exArr, "按数组解析AI返回失败,回退到逐对象解析。预览: {Preview}", arrJson?.Length > 200 ? arrJson.Substring(0, 200) + "..." : arrJson); + } + } + } + + // 逐对象解析 + var startIdx = 0; + while (startIdx < buf.Length) + { + var openBrace = buf.IndexOf('{', startIdx); + if (openBrace == -1) break; + var closeBrace = FindMatchingBrace(buf, openBrace); + if (closeBrace == -1) break; + var jsonStr = buf.Substring(openBrace, closeBrace - openBrace + 1); + try + { + var result = JsonSerializer.Deserialize(jsonStr); + HandleResult(result); + } + catch (Exception ex) + { + logger.LogWarning(ex, "解析AI分类结果失败: {JsonStr}", jsonStr.Length > 200 ? jsonStr.Substring(0, 200) + "..." : jsonStr); + } + startIdx = closeBrace + 1; + } + + if (startIdx > 0) + { + buffer.Remove(0, startIdx); + } + } + + var buffer = new StringBuilder(); + await foreach (var chunk in openAiService.ChatStreamAsync(systemPrompt, userPrompt)) + { + buffer.Append(chunk); + FlushBuffer(buffer); + } + + // 如果AI流结束但没有任何分类结果,发出错误提示 + if (classifyResults.Count == 0) + { + logger.LogWarning("AI未返回任何分类结果,buffer最终内容: {BufferPreview}", buffer.ToString().Length > 500 ? buffer.ToString().Substring(0, 500) + "..." : buffer.ToString()); + chunkAction(("error", "智能分类未返回任何结果,请重试或手动分类")); + } + else + { + chunkAction(("end", "分类完成")); + } } catch (Exception ex) { diff --git a/Web/src/components/ReasonGroupList.vue b/Web/src/components/ReasonGroupList.vue index 44a5e4c..6a3f455 100644 --- a/Web/src/components/ReasonGroupList.vue +++ b/Web/src/components/ReasonGroupList.vue @@ -75,6 +75,7 @@ :finished="transactionFinished" @load="loadGroupTransactions" @click="handleTransactionClick" + @delete="handleGroupTransactionDelete" /> @@ -188,7 +189,7 @@