48 lines
1.5 KiB
C#
48 lines
1.5 KiB
C#
|
namespace MemScanEDR.Models;
|
|||
|
|
|
|||
|
|
public enum AIProvider
|
|||
|
|
{
|
|||
|
|
Heuristic,
|
|||
|
|
OpenAICompatible,
|
|||
|
|
Claude,
|
|||
|
|
Local
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public class AIConfig
|
|||
|
|
{
|
|||
|
|
public AIProvider Provider { get; set; } = AIProvider.Heuristic;
|
|||
|
|
public string ApiKey { get; set; } = "";
|
|||
|
|
public string BaseUrl { get; set; } = "";
|
|||
|
|
public string Model { get; set; } = "";
|
|||
|
|
public int MaxTokens { get; set; } = 2000;
|
|||
|
|
public double Temperature { get; set; } = 0.3;
|
|||
|
|
public int Timeout { get; set; } = 30;
|
|||
|
|
/// <summary>S2.0: 启用本地AI内存分析（无监督+监督）</summary>
|
|||
|
|
public bool EnableLocalAI { get; set; } = true;
|
|||
|
|
/// <summary>S2.0: 启用大模型辅助研判</summary>
|
|||
|
|
public bool EnableLLMAssist { get; set; } = false;
|
|||
|
|
|
|||
|
|
public static AIConfig DefaultFor(AIProvider provider) => provider switch
|
|||
|
|
{
|
|||
|
|
AIProvider.OpenAICompatible => new AIConfig
|
|||
|
|
{
|
|||
|
|
Provider = AIProvider.OpenAICompatible,
|
|||
|
|
BaseUrl = "https://api.openai.com/v1",
|
|||
|
|
Model = "gpt-4o-mini"
|
|||
|
|
},
|
|||
|
|
AIProvider.Claude => new AIConfig
|
|||
|
|
{
|
|||
|
|
Provider = AIProvider.Claude,
|
|||
|
|
BaseUrl = "https://api.anthropic.com",
|
|||
|
|
Model = "claude-3-5-sonnet-20241022"
|
|||
|
|
},
|
|||
|
|
AIProvider.Local => new AIConfig
|
|||
|
|
{
|
|||
|
|
Provider = AIProvider.Local,
|
|||
|
|
BaseUrl = "http://127.0.0.1:11434",
|
|||
|
|
Model = "llama3"
|
|||
|
|
},
|
|||
|
|
_ => new AIConfig { Provider = AIProvider.Heuristic }
|
|||
|
|
};
|
|||
|
|
}
|