package com.vincent.rsf.server.ai.config;
|
|
import com.vincent.rsf.server.ai.constant.AiSceneCode;
|
import lombok.Data;
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
import org.springframework.context.annotation.Configuration;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@Data
|
@Configuration
|
@ConfigurationProperties(prefix = "ai")
|
public class AiProperties {
|
|
private String gatewayBaseUrl = "http://127.0.0.1:8086";
|
|
private Integer sessionTtlSeconds = 86400;
|
|
private Integer maxContextMessages = 12;
|
|
private String systemPrompt = "你是WMS系统内的智能助手,回答时优先保持准确、简洁,并结合上下文帮助用户理解仓储业务。";
|
|
private String diagnosisSystemPrompt = "你是一名资深WMS智能诊断助手,目标是结合当前系统上下文对仓库运行情况做巡检分析。"
|
+ "回答时禁止凭空猜测,必须优先依据提供的实时摘要进行判断。"
|
+ "请优先按以下顺序分析:先总结库存、任务、设备站点的实时状态,指出是否存在明显异常;"
|
+ "如果发现异常,请给出异常现象、可能原因、影响范围、建议处理步骤;"
|
+ "如果数据正常,请明确说明当前未发现明显异常,并提醒仍需人工结合现场状态复核。"
|
+ "回答尽量引用你拿到的实时数据,不要编造未查询到的设备状态或业务事实。"
|
+ "请按“问题概述、关键证据、可能原因、建议动作、风险评估”的结构输出,并优先给出可执行建议。";
|
|
private Integer routeFailThreshold = 3;
|
|
private Integer routeCooldownMinutes = 10;
|
|
private Integer diagnosticLogWindowHours = 24;
|
|
private Integer apiFailureWindowHours = 24;
|
|
private String defaultModelCode = "deepseek-ai/DeepSeek-V3.2";
|
|
private List<ModelConfig> models = new ArrayList<>();
|
|
public List<ModelConfig> getEnabledModels() {
|
return models.stream().filter(model -> Boolean.TRUE.equals(model.getEnabled())).collect(Collectors.toList());
|
}
|
|
public String resolveDefaultModelCode() {
|
if (defaultModelCode != null && !defaultModelCode.trim().isEmpty()) {
|
return defaultModelCode;
|
}
|
return getEnabledModels().isEmpty() ? "deepseek-ai/DeepSeek-V3.2" : getEnabledModels().get(0).getCode();
|
}
|
|
public String buildScenePrompt(String sceneCode, String basePrompt) {
|
String prompt = basePrompt == null ? null : basePrompt.trim();
|
if (AiSceneCode.SYSTEM_DIAGNOSE.equals(sceneCode)) {
|
if (prompt == null || prompt.isEmpty()) {
|
return diagnosisSystemPrompt;
|
}
|
return prompt + "\n\n" + diagnosisSystemPrompt;
|
}
|
return prompt;
|
}
|
|
@Data
|
public static class ModelConfig {
|
private String code;
|
private String name;
|
private String provider;
|
private Boolean enabled = true;
|
}
|
|
}
|