#AI

zhou zhou

3 天以前 e6369c9b64a82eeada6b6f0658d2ae786787e101

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package com.vincent.rsf.server.ai.config;
 
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 defaultModelCode = "mock-general";
 
    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() ? "mock-general" : getEnabledModels().get(0).getCode();
    }
 
    @Data
    public static class ModelConfig {
        private String code;
        private String name;
        private String provider;
        private Boolean enabled = true;
    }
 
}