package com.vincent.rsf.server.ai.service.provider;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.vincent.rsf.server.ai.model.AiDiagnosticToolResult;
|
import com.vincent.rsf.server.ai.model.AiPromptContext;
|
import com.vincent.rsf.server.system.entity.AiCallLog;
|
import com.vincent.rsf.server.system.service.AiCallLogService;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@Service
|
public class AiApiFailureSummaryService implements AiDiagnosticDataProvider {
|
|
private static final String TOOL_CODE = "ai_call_failure";
|
private static final String TOOL_NAME = "AI调用失败";
|
|
@Resource
|
private AiCallLogService aiCallLogService;
|
@Resource
|
private com.vincent.rsf.server.ai.config.AiProperties aiProperties;
|
|
/**
|
* 返回 AI 调用失败工具默认顺序。
|
*/
|
@Override
|
public int getOrder() {
|
return 50;
|
}
|
|
/**
|
* 返回 AI 调用失败工具编码。
|
*/
|
@Override
|
public String getToolCode() {
|
return TOOL_CODE;
|
}
|
|
/**
|
* 返回 AI 调用失败工具展示名。
|
*/
|
@Override
|
public String getToolName() {
|
return TOOL_NAME;
|
}
|
|
/**
|
* 返回 AI 调用失败工具默认说明。
|
*/
|
@Override
|
public String getDefaultToolPrompt() {
|
return "重点识别最近 AI 调用失败的模型、错误类型和时间窗口。";
|
}
|
|
/**
|
* 汇总最近一段时间内的 AI 调用失败记录。
|
*/
|
@Override
|
public AiDiagnosticToolResult buildDiagnosticData(AiPromptContext context) {
|
Date start = new Date(System.currentTimeMillis() - aiProperties.getApiFailureWindowHours() * 3600_000L);
|
List<AiCallLog> records = aiCallLogService.list(new LambdaQueryWrapper<AiCallLog>()
|
.eq(AiCallLog::getTenantId, context.getTenantId())
|
.eq(AiCallLog::getResult, 0)
|
.ge(AiCallLog::getCreateTime, start)
|
.orderByDesc(AiCallLog::getCreateTime)
|
.last("limit 10"));
|
Map<String, Object> meta = new LinkedHashMap<>();
|
meta.put("count", records.size());
|
if (records.isEmpty()) {
|
return new AiDiagnosticToolResult()
|
.setToolCode(getToolCode())
|
.setToolName(getToolName())
|
.setSeverity("INFO")
|
.setSummaryText("最近 " + aiProperties.getApiFailureWindowHours() + " 小时未发现 AI 调用失败记录。")
|
.setRawMeta(meta);
|
}
|
List<String> parts = new ArrayList<>();
|
for (AiCallLog item : records) {
|
parts.add((item.getModelCode() == null ? "未知模型" : item.getModelCode())
|
+ "（"
|
+ (item.getErr() == null ? "无异常描述" : item.getErr())
|
+ "）");
|
}
|
meta.put("latestErrors", parts);
|
return new AiDiagnosticToolResult()
|
.setToolCode(getToolCode())
|
.setToolName(getToolName())
|
.setSeverity("WARN")
|
.setSummaryText("最近 " + aiProperties.getApiFailureWindowHours() + " 小时发现 " + records.size() + " 条 AI 调用失败记录，最近失败包括：" + String.join("；", parts))
|
.setRawMeta(meta);
|
}
|
|
}
|
