suncheng/EmailBill
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix
Some checks failed
Docker Build & Deploy / Build Docker Image (push) Failing after 6s
Details
Docker Build & Deploy / Deploy to Production (push) Has been skipped
Details

Browse Source
This commit is contained in:
孙诚
2026-01-01 11:58:21 +08:00
parent 0444218898
commit c1aa4df4f3
11 changed files with 442 additions and 165 deletions
Download Patch File Download Diff File Expand all files Collapse all files

139
Service/SmartHandleService.cs
View File

@@ -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<long>();
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<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 });
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<GroupClassifyResult[]>(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<GroupClassifyResult>(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)
{